#include <iostream>
#include <string>
#include <iomanip>
#include <fstream>
#include <sstream>
#include <vector>
#include <algorithm>
using namespace std;

//First function to print the results of the county election file
void print_results(const std::vector<string> & countyNameVector, const std::vector<int> & countyNCount, const vector<int> & countyFCount, const std::vector<int> & countyOCount, int & NTotal, int & FTotal, int & OTotal)
{
	cout << endl;
	cout << endl;
	cout << endl;
	cout << "County" << setw(20) << "Nullifier" << setw(20) << "Fieldite" << setw(20) << "Third Party" << endl;
	cout << "----------------------------------------------------------------------" << endl;
	//CountyNameVector is a vector which holds all the county names of the files that have been loaded
	for(int i = 0; i < countyNameVector.size(); i++)
	{
		cout << countyNameVector[i] << setw(20) << countyNCount[i] << setw(20) << countyFCount[i] << setw(20) << countyOCount[i] << endl;
		//This is totalling up the tallys of votes for each party
		NTotal = NTotal + countyNCount[i];
		FTotal = FTotal + countyFCount[i];
		OTotal = OTotal + countyOCount[i];
	}
	cout << "-----------------------------------------------------------------------" << endl;
	cout << "Total" << setw(20) << NTotal << setw(20) << FTotal << setw(20) << OTotal << endl << endl << endl << endl << endl;
}

//Function that searches for a specific county name in the loaded files
void search_county(const std::vector<string> & countyNameVector, const std::vector<int> & countyNCount, const vector<int> & countyFCount, const std::vector<int> & countyOCount)
{
	string searchname;
	cout << "Enter the county file to search for: ";
	cin >> searchname;
	for(int i = 0; i < countyNameVector.size(); i++)
	{
		if(searchname == countyNameVector[i])
		{
			cout << endl;
			cout << endl;
			cout << "County" << setw(20) << "Nullifier" << setw(20) << "Fieldite" << setw(20) << "Third Party" << endl;
			cout << "-----------------------------------------------------------------------" << endl;
			cout << countyNameVector[i] << setw(20) << countyNCount[i] << setw(20) << countyFCount[i] << setw(20) << countyOCount[i] << endl;
			cout << "-----------------------------------------------------------------------" << endl;
			cout << endl;
			cout << endl;
			return;
		}
	}
	cout << endl;
	//if no county name is found, then it returns a message
	cout << "No county was found!" << endl << endl;
}

//Function that lets the user add a county election file to the program
void add_county_election_file(std::vector<string> & countyNameVector, std::vector<int> & countyNCount, vector<int> & countyFCount, std::vector<int> & countyOCount, int & NCount, int & FCount, int & OCount)
{
	char vote;
    string filename;
    string countyName;
	NCount = 0;
	FCount = 0;
	OCount = 0;

    cout << "Enter the county file to process: ";
    cin >> filename;
    std::ifstream input((filename).c_str());
    if(input.is_open())
    {
        input >> countyName;
		//adds the county name to the vector of names
        countyNameVector.push_back(countyName);
        while(input >> vote)
		{
			if(vote == 'N' || vote == 'n')
			{
                NCount = NCount + 1;
            }
			else if(vote == 'F' || vote == 'f')
			{
				FCount = FCount + 1;
			}
			else
			{
				OCount = OCount + 1;
			}
		}
		//adds the count of party votes for the current file to a vector
        countyNCount.push_back(NCount);
        countyFCount.push_back(FCount);
        countyOCount.push_back(OCount);
    }
	else
	{
	cout << endl;
	cout << "File not found!" << endl;
	cout << endl;
	}
}

//function to get which choice the user is selecting from the menu
char get_menu_choice()
{
	cout << "Add a county election file         A" << endl;
	cout << "Show election totals on screen     P" << endl;
	cout << "Search for county results          S" << endl;
	cout << "Exit the program                   Q" << endl;
	cout << "Please enter your choice: ";
	char selection;
	cin >> selection;
	return selection;
}

int main()
{
	vector <string> countyNameVector;
	vector <int> countyNCount;
	vector <int> countyFCount;
	vector <int> countyOCount;
	int NCount = 0;
	int FCount = 0;
	int OCount = 0;
	int NTotal = 0;
	int FTotal = 0;
	int OTotal = 0;
	char selection;

	do
	{
		//this sets the user choice to a variable, which is used for our if statement
		selection = get_menu_choice();

	if(selection == 'a' || selection == 'A')
	{
		add_county_election_file(countyNameVector, countyNCount, countyFCount, countyOCount, NCount, FCount, OCount);
	}

	if(selection == 'p' || selection == 'P')
	{
		print_results(countyNameVector, countyNCount, countyFCount, countyOCount, NTotal, FTotal, OTotal);

	}
	if(selection == 'S' || selection == 's')
	{
		search_county(countyNameVector, countyNCount, countyFCount, countyOCount);
	}


	}while((selection != 'Q' && selection != 'q'));
	return 0;


}