#include <iostream.h>
#include <fstream.h>
#include <string.h>

class TruckClass {
	char Driver[80];
	float TripMiles[5];
public:
	void EnterDriver(void);
	void EnterTrip(int);
	void BlankOut(void);
	void DisplayInfo(void);
	char* GetDriver(void);
	float GetTrip(int);
	void SetDriver(char*);
	void SetTrip(int,float);
};


void TruckClass::EnterDriver(){
	cout << "\n";
	cout << "Enter the driver's name: ";
	cin.get(Driver,79);
	char nl;
	cin.get(nl);
}

void TruckClass::EnterTrip(int tripnumber){
	cout << "\n";
	cout << "Enter the mileage for trip number " << tripnumber << "\n";
	cin >> TripMiles[tripnumber-1];
	char nl;
	cin.get(nl);
}

void TruckClass::BlankOut(){
	strcpy(Driver," ");
	int t;
	for(t=0;t<5;t++) TripMiles[t]=0;
}

void TruckClass::DisplayInfo(){
	cout << "Driver Name: " << Driver << "\n";
	int t;
	for(t=0;t<5;t++) cout << "Trip Number "<< t+1 << " mileage was: " <<TripMiles[t] << "\n";
	cout << "\n\n";
}

char* TruckClass::GetDriver(){
	return(Driver);
}

float TruckClass::GetTrip(int tnumber){
	return(TripMiles[tnumber]);
}

void TruckClass::SetDriver(char* dname){
	strcpy(Driver,dname);
}

void TruckClass::SetTrip(int trip, float miles){
	TripMiles[trip]=miles;
}


int main(){
	TruckClass Trucks[10];
	int i,t;
	char nl;
	int DriverNumber, TripNum;
	char tempdriver[80];
	float tempmiles;
	fstream myinfile, myoutfile;

	
	for(i=0;i<10;i++) Trucks[i].BlankOut();

	int menuchoice=0;

	while(menuchoice<6) {
		cout << "\n\n";
		cout << "K & R Trucking Company\n";
		cout << "Main Menu\n\n";
		cout << "1. Enter a driver's name.\n";
		cout << "2. Enter mileage for a trip.\n";
		cout << "3. Display information for a driver.\n";
		cout << "4. Save all information to disk.\n";
		cout << "5. Retrieve all information from disk.\n";
		cout << "6. Exit the program.\n\n";
		cout << "Please make a selection (1-6) and hit <Enter>: ";
		cin >> menuchoice;
		cin.get(nl);

		switch(menuchoice){
		case 1: 
			cout << "\n\n";
			cout << "Enter driver number (1-10): ";
			cin >> DriverNumber;
			cin.get(nl);
			Trucks[DriverNumber-1].EnterDriver();
			break;
		case 2: 
			cout << "\n\n";
			cout << "Enter driver number (1-10): ";
			cin >> DriverNumber;
			cin.get(nl);
			cout << "\n";
			cout << "Enter trip number (1-5): ";
			cin >> TripNum;
			cin.get(nl);
			Trucks[DriverNumber-1].EnterTrip(TripNum);
			break;
		case 3: 
			cout << "\n\n";
			cout << "Enter driver number (1-10): ";
			cin >> DriverNumber;
			cin.get(nl);
			Trucks[DriverNumber-1].DisplayInfo();
			break;
		case 4:
			myoutfile.open("truck.dat",ios::out);
			for(i=0;i<10;i++){
				myoutfile << Trucks[i].GetDriver() << "\n";
				for(t=0;t<5;t++){
					myoutfile << Trucks[i].GetTrip(t) << "\n";
				}
			}
			myoutfile.close();
		case 5:
			myinfile.open("truck.dat",ios::in);
			for(i=0;i<10;i++){
				myinfile.get(tempdriver,79);
				myinfile.get(nl);
				Trucks[i].SetDriver(tempdriver);
				for(t=0;t<5;t++){
					myinfile >> tempmiles;
					myinfile.get(nl);
					Trucks[i].SetTrip(t,tempmiles);
				}
			}
			myinfile.close();
		
		case 6:
			break;
		default:
			cout << "Menu choice must be a number in the range 1 to 6.";
		}

		
	}
	return(0);
}