#ifndef _GENERATOR_CLASS
#define _GENERATOR_CLASS

#include "DependencyStructure.h"

class Generator
{
public:
	Generator(void);
	~Generator(void);
	string Summarize(DependencyStructure &Article);
private:
	DependencyStructure a;
	void addWord(int index, vector<string> &summary);
	void addSubject(int index, vector<string> &summary);
	void addVerb(int index, vector<string> &summary);

	int getHighestRatedRelated(int index, string tag);
	int getHighestRatedRelatedBy(int index, vector<string> labels);
	void addWordsRelatedTo(int index, vector<string> labels, int limit, vector<string> &summary);
	bool stringIn(string s, vector<string> a);

	void saveAsJavaScript();

	vector<string> adjectiveLabels, seriesLabels, adverbLabels, preVerbLabels, verbLabels;
};

#endif