// File: Test_Set.cpp
// Driver for set ADT

#include <assert.h>
#include <console.h>
#include "Set_ADT.h"
#include "char_ops.h"

void Operate(char,SetADT<int> *,int);

void main()
{SetADT<int> *Set;
 int NumSets;
 char Choice;
 cout << "How many sets to work with? >";
 cin >> NumSets;
 assert(NumSets>0);
 cout << "Your sets are numbered 0 to " << NumSets-1 << endl;
 Set=new SetADT<int> [NumSets];
 do {
  cout << "A)dd Item to Set   \n";
  cout << "R)emove Item From Set   \n";
  cout << "S)ize (Cardinality) of Set\n";
  cout << "U)nion of 2 Sets  \n";
  cout << "I)ntersection of 2 Sets \n";
  cout << "D)ifference of 2 Sets \n";
  cout << "M)embership in a Set\n";
  cout << "P)rint the Contents of a Set\n";
  cout << "Q)uit\n";
  Choice=GetChoice("ARSUIDMPQ");
  if (Choice!='Q') Operate(Choice,Set,NumSets);
 } while (Choice!='Q');
}

int SelectSet(char *Msg)
{int SetNum;
 cout << Msg << '>';
 cin  >> SetNum;
 return(SetNum);
}

void Operate(char Op,SetADT<int> *Set,int NumSets)
{int Set1,Set2,Set3,Element;
 switch (Op) {
  case 'A':
   Set1=SelectSet("Add Element to Which Set?");
   assert(Set1>=0 && Set1<NumSets);
   cout << "Element to Add? >";
   cin >> Element;
   if (!Set[Set1].addElt(Element))
    cout << Element << " already in Set\n";
   break;
  case 'R':
   Set1=SelectSet("Remove Element From Which Set?");
   assert(Set1>=0 && Set1<NumSets);
   cout << "Element to Remove? >";
   cin >> Element;
   if (!Set[Set1].removeElt(Element))
    cout << Element << " wasn't in Set\n";
   break;
  case 'S':
   Set1=SelectSet("Size of Which Set?");
   assert(Set1>=0 && Set1<NumSets);
   cout << "Set " << Set1 << " has " << Set[Set1].cardinality() << " Elements\n";
   break;
  case 'U':
   Set1=SelectSet("First Set?");
   assert(Set1>=0 && Set1<NumSets);
   Set2=SelectSet("Second Set?");
   assert(Set2>=0 && Set2<NumSets);
   Set3=SelectSet("Set in Which to Place Result?");
   assert(Set3>=0 && Set3<NumSets);
   Set[Set3]=Set[Set1]+Set[Set2];
   break;
  case 'I':
   Set1=SelectSet("First Set?");
   assert(Set1>=0 && Set1<NumSets);
   Set2=SelectSet("Second Set?");
   assert(Set2>=0 && Set2<NumSets);
   Set3=SelectSet("Set in Which to Place Result?");
   assert(Set3>=0 && Set3<NumSets);
   Set[Set3]=Set[Set1]*Set[Set2];
   break;
  case 'D':
   Set1=SelectSet("First Set?");
   assert(Set1>=0 && Set1<NumSets);
   Set2=SelectSet("Second Set?");
   assert(Set2>=0 && Set2<NumSets);
   Set3=SelectSet("Set in Which to Place Result?");
   assert(Set3>=0 && Set3<NumSets);
   Set[Set3]=Set[Set1]-Set[Set2];
   break;
  case 'M':
   Set1=SelectSet("Check Whether an Element is in Which Set?");
   assert(Set1>=0 && Set1<NumSets);
   cout << "Element to Look For? >";
   cin >> Element;
   if (Set[Set1]>Element)
    cout << Element << " is a member of Set " << Set1 << endl;
   else cout << Element << " is not a member of Set " << Set1 << endl;
   break;
  case 'P':
   Set1=SelectSet("Print Which Set?");
   assert(Set1>=0 && Set1<NumSets);
   cout << Set[Set1];
   break;
 }
 cout << "Press Any Key to Continue\n";
 getch();
 clrscr();
}