// CSE143 Su01 Homework 2 Sample Solution
// wordpro.cpp - implementation of WordPro data structure and functions
// SRB  6/20

#include "stringUtil.h"
#include "wordpro.h"

using namespace std;

bool WordPro::loadDocument(string fileName)  {
	string sentence;
	ifstream data(fileName.c_str());

	if (!data) 
		return false;
	getline(data,sentence);
	// read data from file
	
	while (sentence != ". .") {
		mDocument.setLine(mDocument.getSize(), sentence);
		mDocument.setSize(mDocument.getSize() + 1);
		getline(data,sentence);
	}
	return true;
}

void WordPro::saveDocument(string fileName)  {
	
	ofstream oFile(fileName.c_str() );
	for(int x=0;x<mDocument.getSize();x++)
		oFile<<mDocument.getLine(x)<<endl;
}

void WordPro::printDocument() {
	for(int x=0;x<mDocument.getSize();x++) {
		cout<<mDocument.getLine(x);
		cout<<endl;
	}
	cout<<endl;
}

void WordPro::spellCheck()  {
	
	string correctSentence;		// string representing the correctly spelled word
	int numErrors=0;			// total number of errors
	string seps = "  ,!.?";		// denotes seperator for the tokens 
	string token;				// points to the current token
	string correctWord; 
	cout<<endl<<endl;
	for(int x=0;x<mDocument.getSize();x++) {
		token = newStrtok( mDocument.getLine(x), seps );
	
		while( token != "" )	   {
			//Retrieves the correctly spelled version of the word
			correctWord = mDictionary.correctWord(token);
			if(correctWord!=token)
				numErrors++;
			correctSentence = correctSentence + correctWord + " ";
			token = newStrtok( "", seps );
		}

		mDocument.setLine(x, correctSentence);
		correctSentence.erase(0,correctSentence.size() );
	}

	cout<<endl<<"Number of errors = "<<numErrors<<endl<<endl;
	printDocument();
}
	

void WordPro::findWord(string word) {
	int lineFound=0;
	bool found=false;
	for(int x=0;x<mDocument.getSize();x++) {
		if(mDocument.getLine(x).find(word)!=string::npos ) {
			lineFound=x;
			found = true;
			break;
		}
	}
	if(found) {
		cout<<endl<<"The first occurance of text \""<<word<<"\" was found in line "<<lineFound+1<<endl;
		cout<<mDocument.getLine(lineFound)<<endl<<endl;
	}
	else 
		cout<<endl<<"The text \""<<word<<"\" was not found in the document"<<endl<<endl;
}

void WordPro::continueDocument() {
	string currSentence;
	cout<<"Please enter text:"<<endl;
	getline(cin,currSentence);
	while(currSentence!=".") {
		mDocument.setLine(mDocument.getSize(), currSentence);
		mDocument.setSize(mDocument.getSize() + 1);
		getline(cin,currSentence);
	}
}


void WordPro::startNewDocument()  {
	mDocument.setSize(0);
	continueDocument();
}