C/ngrams.cpp

   1 #pragma once
   2
   3 #include <iostream>
   4 #include <fstream>
   5 #include <sstream>
   6
   7 #include <map>
   8 #include <string>
   9
  10 struct NGram: public std::map<std::string, long> {
  11         NGram(const std::string &file) {
  12                 std::ifstream f(file.c_str());
  13
  14                 long frequency;
  15                 for(std::string gram; f >> frequency;) {
  16                         f.get(); // drop one space
  17                         getline(f, gram); // get rest of line
  18
  19                         this->insert(make_pair(gram, frequency));
  20                 }
  21
  22                 f.close();
  23         }
  24 };
  25
  26
  27 /*int main(int argc, char **argv) {
  28         // check options
  29         if(argc < 2) {
  30                 cerr << "Usage: " << argv[0] << " <filename>" << endl;
  31                 return 1;
  32         }
  33
  34         NGram ngrams(argv[1]);
  35
  36         for(NGram::const_iterator it = ngrams.begin();
  37         it != ngrams.end(); ++it) {
  38                 cout << it->first << " " << it->second << endl;
  39         }
  40
  41         return 0;
  42 }*/