BagDriver.cpp

Go to the documentation of this file.

//-----------------------
// C++ includes
//-----------------------
#include <iostream>
#include <fstream>
#include <string>
#include <vector>

//-----------------------
// Application includes
//-----------------------
#include "ArrayBag.h"
#include "io_functions.h"

//-----------------------
// Using statements
//-----------------------
using namespace std;

//----------------------
// Functions Prototypes
//----------------------
int getBagNum();                 //prompts user for bag number
void processOption(char& c, ArrayBag<string> bags[]);  //processes menu input
void displayBag(ArrayBag<string>& bag);   //display the bag's entire contents
void restoreBag(ArrayBag<string>& bag);   //prompts for filename with items
void saveBag(ArrayBag<string>& bag);   //prompts for filename to store items in

int main()
{
  int bagNum;              //The bag number to use
  ArrayBag<string> bag[3];//bags to test, bag 0 used for intersection/union
  string restoreBagFlg;  // holds response to restore bag from a file
  string menuOption;    // holds menu option input from user

//Opening Message
  cout<< "Bag Driver - Tests the Bag class using two bags" << endl;

//Restore the bag from a file
  cout<< "----------------------------------" << endl; 
  cout<< "Would you like to restore a bag from a file? (Y/N) ";
   
//Read user input - use a string to accept all inputs
  getline(cin, restoreBagFlg);
  if(restoreBagFlg[0] == 'Y' || restoreBagFlg[0] == 'y'){ 
    bagNum = getBagNum();
    restoreBag(bag[bagNum]);            // opens/reads file into aBag
  }

//Menu driven part of main program
//Initial menu display and loop to process menu option
  printMenu();
  do {
    cout << "\nEnter menu selection (0 for menu): ";
    getline(cin, menuOption);    //Use a string in case user enters chars
    processOption(menuOption[0], bag);
  }while(menuOption[0] != '9');
   
  return 0;            //Exit program
}//main()


//---------------------------
// Local function definitions
//---------------------------

int getBagNum()
{
  int bagNum=0;  
  while(bagNum != 1 && bagNum != 2)
  {
    cout << "Which bag (1 or 2)? ";
    bagNum = getInt();
  }
  return bagNum;
} //getBagNum()

void displayBag(ArrayBag<string>& bag) 
{
  string item;
  vector<string> bagVector = bag.toVector();

  if(bagVector.size() > 0)
  {
    cout << "The bag contains: " << endl;
    for(unsigned int i=0; i<bagVector.size(); i++){
      cout << bagVector[i] << ", ";
    }
    cout<<endl;
  }
  else
  {
    cout << "The bag is empty." << endl;
  }
}


void processOption(char &menuChar, ArrayBag<string> bag[])
{
  int bagNum;
  string item;
  string saveBagFlg;

   switch (menuChar)
   {
    case '0':               // Display menu options
       printMenu();
       break;

    case '1':               // Output size of the bag
       cout << "\nBag Size is: " << bag[getBagNum()].getCurrentSize() << endl;
       break;

    case '2':               // Add items to the bag 
       cout << "Add Item:" << endl;
       item = getItem();
       bagNum = getBagNum();
       if(bag[bagNum].add(item))
       {
         cout << item << " was successfully added to bag " << bagNum << endl;
       }
       else
       {
         cout << item << " was not added to the bag." << endl;
       }
       break;

//NOTE that options 3 - 8 are left for the student to complete

    case '8':               // Display entire contents of bag
       displayBag(bag[getBagNum()]);
       break;

    case '9':               // Exit the program
    case 'Q':               // handles multiple quit inputs
    case 'q':
       cout<< "Would you like to save the bag to a file? (Y/N) ";
       getline(cin, saveBagFlg);
       if(saveBagFlg[0] == 'Y' || saveBagFlg[0] == 'y')
         saveBag(bag[getBagNum()]);         // save bag to file
       cout<< "Goodbye!"<< endl;  // Exit message
       menuChar = '9';
       break;
    default:                      // Invalid menu option entered
       cout << "\nError!  Invalid option.  Please try again.\n\n";
       printMenu();
    }
}//processOption()

void restoreBag(ArrayBag<string>& bag)
{
  bool success = true;
  ifstream fin;
  string filename;
  string item;

  //Data validation to get filename and open it
  do{
    cout<<"Enter the filename that contains the bag: ";
    getline(cin, filename);
    fin.clear();
    fin.open(filename.c_str());
    if(fin == 0){
      cout<<"ERROR  - could not open file " << filename<<endl;
    }
  } while (!fin);

  //read file until eof reached
  while(success && getline(fin,item)){
    success = bag.add(item);
  }
  if (!success) cout << "Not all items added to the bag!" << endl;
  cout<< bag.getCurrentSize() << " items read into the bag\n\n";
  fin.close();
}//restoreBag()

void saveBag(ArrayBag<string>& bag)
{
   //NOTE this is a function stub. You need to implement it. 
   cout << "Bag will be saved when implemented" << endl;
}//saveBag()