C++ Primer TextQuery 完整代码

最近看这本书, 然后查资料整理了一些这个代码, 并添加了注释 练习一些 IO 和map set吧:

#include "TextQuery.h"

using namespace std;

ifstream &open_file(ifstream &in, const string &file);
void print_results(const set<TextQuery::line_no>& locs, const string& sought, const TextQuery &file);
string make_plural(size_t ctr, const string &word, const string &ending);

int main(int argc, char **argv) {
	ifstream infile;
	if (argc < 2 || !open_file(infile, argv[1])) { //通过参数输入文件 这里如果有文件已经通过openfile打开了文件
		cerr << "No input file!" << endl;
		return EXIT_FAILURE;
	}
	TextQuery tq;
	tq.read_file(infile); //调用TextQuery的函数打开文件
	while (true) {
		cout << "enter word to look for,or q to quit" << endl;
		string s;
		cin >> s;
		if (!cin || s == "q")
			break;
		set<TextQuery::line_no> locs = tq.run_query(s); //输入一个单词, 就可以通过这个函数 输出一个含有这个单词的set
		print_results(locs, s, tq);
	}
	return 0;
}
//打开文件的工具函数
ifstream &open_file(ifstream &in, const string &file)
{
	in.close();       // close in case it was already open
	in.clear();      // clear any existing errors
					 // if the open fails, the stream will be in an invalid state
	in.open(file.c_str()); // open the file we were given 这是真正打开文件的地方
	return in; // condition state is good if open succeeded
}
//打印结果的函数
void print_results(const set<TextQuery::line_no>& locs, const string& sought, const TextQuery &file)
{
	typedef set<TextQuery::line_no> line_nums; //遍历set获取输出的结果
	line_nums::size_type size = locs.size();
	cout << "\n" << sought << " occurs " << size << " "
			<< make_plural(size, "time", "s") << endl;
	line_nums::const_iterator it = locs.begin();
	for (; it != locs.end(); ++it) {
		cout << "\t(line" << (*it) + 1 << ")" << file.text_line(*it) << endl;
	}
}
//显示time 还是times
string make_plural(size_t ctr, const string &word, const string &ending)
{
	return (ctr == 1) ? word : word + ending;
}

#include<iostream>
#include<fstream>       //ifstream 的头文件
#include<set>
#include<string>
#include<vector>
#include<map>
#include <stdlib.h>   //EXIT_FALURE 的头文件
#include<sstream>    //istringstream的头文件
#include<utility> //pairs的头文件

using namespace std;


class TextQuery {
public:
	typedef std::vector<std::string>::size_type line_no;
	//读取文件的 ifstream 并进行逐行读取,然后进行逐个单词 和其对应行号的扫描
	void read_file(std::ifstream &is) {
		store_file(is);
		build_map();
	}
	std::set<line_no> run_query(const std::string &) const;
	std::string text_line(line_no) const;
private:
	void store_file(std::ifstream&);
	void build_map();
	std::vector<std::string> lines_of_text; //存储text文件的每一行
	std::map<std::string, std::set<line_no> > word_map;
};

#include "TextQuery.h"
#include <stdexcept>

//扫描文件中的每一行
void TextQuery::store_file(ifstream &is)
{
	string textline;
	while (getline(is, textline))
		lines_of_text.push_back(textline);
}
//扫描每一行的每个单词
void TextQuery::build_map()
{
	for (line_no line_num = 0; line_num != lines_of_text.size(); ++line_num) {
		istringstream line(lines_of_text[line_num]);
		string word;
		while (line >> word) //遍历每个单词 把num 和word添加到 word_map中
			word_map[word].insert(line_num);
	}
}
//给定一个搜索词, 获取存在该词的行号
set<TextQuery::line_no> TextQuery::run_query(const string &query_word) const
{
	map<string, set<line_no> >::const_iterator loc = word_map.find(query_word);
	if (loc == word_map.end())
		return set<line_no>();
	else
		return loc->second;
}
//根据行号, 把行打印出来
string TextQuery::text_line(line_no line) const
{
	if (line < lines_of_text.size())
		return lines_of_text[line];
	throw std::out_of_range("line number out of range");
}