tests/compiler/sloc-counter/driver.cxx


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

// file      : tests/compiler/sloc-counter/driver.cxx
// license   : MIT; see accompanying LICENSE file

#include <fstream>
#include <sstream>
#include <iostream>

#include <libcutl/compiler/code-stream.hxx>
#include <libcutl/compiler/sloc-counter.hxx>

#undef NDEBUG
#include <cassert>

using namespace std;
using namespace cutl::compiler;

int
main (int argc, char* argv[])
{
  ifstream ifs;

  if (argc > 1)
    ifs.open (argv[1]);

  istream& in (ifs.is_open () ? ifs : cin);

  ostringstream os1, os2;
  ostream_filter<sloc_counter, char> filt (os1);

  for (istream::int_type i (in.get ());
       i != istream::traits_type::eof ();
       i = in.get ())
  {
    char c (istream::traits_type::to_char_type (i));
    os1.put (c);
    os2.put (c);
  }

  assert (os1.str () == os2.str ());
  cout << filt.stream ().count () << endl;
}