aboutsummaryrefslogtreecommitdiff
path: root/tests/compiler/sloc-counter/driver.cxx
blob: 32408cd0aca5e9463eae79fd1ac9216637039fec (plain)
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
// file      : tests/compiler/sloc-counter/driver.cxx
// copyright : Copyright (c) 2009-2017 Code Synthesis Tools CC
// license   : MIT; see accompanying LICENSE file

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

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

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;
}