vector <string> countyNameVector;
vector <int> countyNCount;
vector <int> countyFCount;
vector <int> countyOCount;
int NCount;
int FCount;
int OCount;
int NTotal;
int FTotal;
int OTotal;

void print_results()
{
for(int i = 0; i < countyNameVector.size(); i++)
    {
    cout << countyNameVector[i] << " " << countyNCount[i] << " " << countyFCount[i] << " " << countyOCount[i] << endl;
    }
    for(int i = 0; i < countyNCount.size(); i++)
        {
        NTotal = NTotal + countyNCount[i];
        FTotal = FTotal + countyFCount[i];
        OTotal = OTotal + countyOCount[i];
        }
        cout << NTotal << " " << FTotal << " " << OTotal << endl;
}

void search_county()
{
    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 << countyNameVector[i] << " " << countyNCount[i] << "" << countyFCount[i] << " " << countyOCount[i] << endl;
                }
            else
            {
            cout << "No county was found!" << endl;
            }
        }
}
void add_county_election_file()
{
    int NCount = 0;
    int FCount = 0;
    int OCount = 0;
    int NTotal = 0;
    int FTotal = 0;
    int OTotal = 0;
    char vote;
    string filename;
    string countyName;

    cout << "Enter the county file to process: ";
    cin >> filename;
    std::ifstream input((filename).c_str());
    if(input.is_open())
        {
        input >> countyName;
        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;
                        }
            }
            countyNCount.push_back(NCount);
            countyFCount.push_back(FCount);
            countyOCount.push_back(OCount);
        }
        cout << countyName << endl;
}
char get_menu_choice()
{
    char selection;
    cin >> selection;
    return selection;
}
int main()
{
    char selection;
    // string searchname;

    do
    {
    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: ";
    selection = get_menu_choice();

    if(selection == 'a' || selection == 'A')
        {
        add_county_election_file();
        }

    if(selection == 'p' || selection == 'P')
        {
        print_results();

        }
    if(selection == 'S' || selection == 's')
        {
        search_county();
        }


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


}