testvideostore.cpp

data+structures+using+c的源码

#include <iostream>
#include <fstream>
#include <string>
#include "binarySearchTree.h"
#include "videoType.h"
#include "videoBinaryTree.h"

using namespace std;

void createVideoList(ifstream& infile, 
					 videoBinaryTree& videoList);
void displayMenu();

int main()
{
	videoBinaryTree  videoList;
	int choice;
	char ch;
	string title;

	ifstream infile;

	infile.open("a:\\videoDat.txt");
	if(!infile)
	{
		cout<<"The input file does not exist."<<endl;
		return 1;
	}

	createVideoList(infile, videoList);
	infile.close();

	displayMenu();				//show the menu
	cout<<"Enter your choice: ";		
	cin>>choice;				//get the request
	cin.get(ch);
	cout<<endl;

		//process the request
	while(choice != 9)
	{
		switch(choice)
		{
		case 1: cout<<"Enter the title: ";
				getline(cin,title);
				cout<<endl;
				if(videoList.videoSearch(title))
					cout<<"Title found."<<endl;
				else
					cout<<"The store does not carry "
					    <<"this title."<<endl;
				break;
		case 2: cout<<"Enter the title: ";
				getline(cin,title);
				cout<<endl;
				if(videoList.videoSearch(title))
				{
					if(videoList.isVideoAvailable(title))
					{
						videoList.videoCheckOut(title);
						cout<<"Enjoy your movie: "<<title<<endl;
					}
					else
						cout<<"The video is currently "
 		     				<<"out of stock."<<endl;
				}
				else
					cout<<"The video is not in the store."<<endl;

				break;
		case 3: cout<<"Enter the title: ";
				getline(cin,title);
				cout<<endl;
				if(videoList.videoSearch(title))
				{
					videoList.videoCheckIn(title);
					cout<<"Thanks for returning "<<title<<endl;
				}
				else
					cout<<"This video is not from our store."
					    <<endl;

				break;
		case 4: cout<<"Enter the title: ";
				getline(cin,title);
				cout<<endl;
				if(videoList.videoSearch(title))
				{
					if(videoList.isVideoAvailable(title))
						cout<<"The video is currently in stock."
						    <<endl;
					else
						cout<<"The video is out of stock."<<endl;
				}
 				else
					cout<<"The video is not in the store."<<endl;

				break;
		case 5: videoList.videoPrintTitle();
			    break;
		case 6: videoList.inorderTraversal();
			    break;
		default: cout<<"Bad Selection."<<endl;
		}//end switch

		displayMenu();				//display the menu
	    cout<<"Enter your choice: ";
	    cin>>choice;				//get the next request
		cin.get(ch);
	    cout<<endl;
	}//end while


	return 0;
}

void createVideoList(ifstream& infile, 
					 videoBinaryTree& videoList)
{
	string Title;
	string Star1;
	string Star2;
	string Producer;
	string Director;
	string ProductionCo;
	char   ch;
	int   InStock;

	videoType newVideo;

	getline(infile, Title);
	while(infile)
	{
		getline(infile, Star1);
		getline(infile, Star2);
		getline(infile, Producer);
		getline(infile, Director);
		getline(infile, ProductionCo);
		infile>>InStock;
		infile.get(ch);
		newVideo.setVideoInfo(Title,Star1,Star2,Producer,
			                   Director,ProductionCo,InStock);
		videoList.insert(newVideo);

		getline(infile, Title);
	}//end while

}//end createVideoList

void displayMenu()
{
	cout<<"Select one of the following "<<endl;
    cout<<"1: To check whether a particular video is in "
		<<"the store"<<endl;
    cout<<"2: To check out a video"<<endl;
    cout<<"3: To check in a video"<<endl;
    cout<<"4: To see whether a particular video is in stock"
 		<<endl;
    cout<<"5: To print the titles of all the videos"<<endl;
    cout<<"6: To print a list of all the videos"<<endl;
    cout<<"9: To exit"<<endl;
}