#include"train.h"
void main()
{
	char ch,ch2,choice,start_place[9],end_place[9],train_num[8];
	int t=1,tt=1,start,end;
	graph_country g;
	int pathlength;
	p_city pathinfo;
	train *tr=init_train();
	creat_train(tr);
	while(t==1)
	{
		PR("*---------------------------------------*\n");
		PR("*--旅游交通查詢系統選擇菜單-------------*\n");
		PR("*	    火車信息查詢---------0      *\n");
		PR("*	    城市路徑查詢---------1      *\n");
		PR("*	    查看所有信息---------2      *\n");
		PR("*	    退出-----------------3      *\n");
		PR("*---------------------------------------*\n");
		PR("請選擇: ");
		ch = getch();
		PR("%c\n",ch);
		if(ch=='0')
		{
			while(tt==1)
			{
				PR("*--火車信息查詢選擇菜單-----------------*\n");
				PR("*	 按車次查詢--------------0      *\n");
				PR("*	 按出發地與目的地查詢----1      *\n");
				PR("*	 退出--------------------2      *\n");
				PR("*---------------------------------------*\n");
				PR("請選擇: ");
				ch2 = getch();
				PR("%c\n",ch2);
				if(ch2=='0')
				{
					PR("請輸入列車號: ");
					scanf("%s",train_num);
					find_train_num(tr,train_num);
				}
				else if(ch2=='1')
				{
					PR("請輸入始發站: ");
					scanf("%s",start_place);
					PR("請輸入終點站: ");
					scanf("%s",end_place);
					PR("*----------------------------*\n");
					PR("*--按車次查詢選擇菜單------*\n");
					PR("*	 最快-------0        *\n");
					PR("*	 最省錢-----1        *\n");
					PR("*	 全部-------2        *\n");
					PR("*	 退出-------3        *\n");
					PR("*----------------------------*\n");	
					PR("請選擇: ");
					choice = getch();
					PR("%c\n",choice);
					find_place(tr,start_place,end_place,choice);
				}
				else if(ch2=='2'){
					tt=0;
				}
				else {
					PR("你的輸入有誤\n");
				}
			}//end while
		}//end ch=0:火車信息查詢
		else if(ch=='1')
		{
			init_graph(&g);
			creat_graph(&g);
			PR("請輸入出發地的序號(0-24): \n");
			scanf("%d",&start);
			PR("請輸入目的地的序號(0-24): \n");
			scanf("%d",&end);
		//	if(end>0&&end<
			shortest_path(g,start,end,&pathlength,&pathinfo);
	//		PR("有待完善!\n");
		}
		else if(ch=='2')
		{
			print_train(tr);
		}
		else if(ch=='3')
		{
			t=0;
		}
		else {
			PR("你的輸入有誤\n");
		}
	}//end while
	for(t=0;t<3;t++)
	{
		putchar(getch());
	}
}