#include "display.h"
#include "dfa.h"
#include <cstdlib>
#include <string>


void input_str(char *str);
int main()
{
	Display msg;	//信息类
	DFA adfa;		//dfa类
	char str[100];	//待测字符串
	string choice;//设置为字符串型防止输入非法字符时死循环
	do{
		system("cls");
		msg.main_msg();

		cin>>choice;
		if(choice=="1")
		{
			msg.dfa_in();	//显示说明信息
			adfa.input();	//输入dfa
			if(adfa.check())	//DFA正确性检查
				adfa.print_dfa(); //输出dfa
			system("pause");
		}
		else if(choice=="2")
		{
			msg.dfa_rw(); //显示说明信息
			string tmp;	//选择
			
			cout<<"1.将内存中dfa数据写入data.dfa文件;\n2.输入dfa并写入data.dfa;\n3.读取data.dfa文件中的dfa"<<endl;
			cout<<"请选择:";
			cin>>tmp;
			if(tmp=="1")
			{
				if(adfa.num_q==0)
				{
					cout<<"内存内容为空，您还没有输入过DFA，请先输入:"<<endl;
					adfa.input();	//输入dfa
					if(adfa.check())	//DFA正确性检查
					{
						adfa.write();	//写dfa到文件
						cout<<"DFA已成功写入data.dfa !"<<endl;
					}
				}
				else
				{
					adfa.write();	//写dfa到文件
					cout<<"DFA已成功写入data.dfa !"<<endl;
				}
			}
			
			else if(tmp=="2")
			{
				adfa.input();	//输入dfa
				if(adfa.check())	//DFA正确性检查
				{
					adfa.write();	//写dfa到文件
					cout<<"DFA已成功写入data.dfa !"<<endl;
				}
			}
			
			else if(tmp=="3")
			{
				cout<<"文件data.dfa的内容为:"<<endl;
				if(adfa.read())
				{
					if(adfa.check())	//DFA正确性检查
						adfa.print_dfa();	//输出dfa
				}
			}
			else
			{
				cout<<"输入错误!"<<endl;
				break;
			}
			system("pause");
		}
		else if(choice=="3")
		{
			string tmp;		//选择
			cout<<"1.使用内存中的DFA.\n2.使用data.dfa中的DFA\n3.输入DFA"<<endl;
			cout<<"请选择:";
			cin>>tmp;
			if(tmp=="1")
			{
				if(adfa.num_q==0)
				{
					cout<<"内存内容为空，您还没有输入过DFA，请先输入:"<<endl;
					adfa.input();	//输入dfa
					adfa.print_dfa();	//输出dfa
					cout<<endl;
					adfa.check();	//DFA正确性检查
					cout<<endl;
				}
				else
				{
					adfa.print_dfa();	//输出dfa
					cout<<endl;
					adfa.check();	//DFA正确性检查
					cout<<endl;
				}
			}
			
			else if(tmp=="2")
			{
				cout<<"文件data.dfa的内容为:"<<endl;
				if(adfa.read())
				{
					adfa.print_dfa();	//输出dfa
					cout<<endl;
					adfa.check();	//DFA正确性检查
					cout<<endl;
				}
			}
			
			else if(tmp=="3")
			{
				adfa.input();	//输入DFA
				adfa.print_dfa();	//输出dfa
				adfa.check();	//DFA正确性检查
			}
			else
			{
				cout<<"输入错误!"<<endl;
				break;
			}
			system("pause");
		}
		else if(choice=="4")
		{
			string tmp;	//选择
			cout<<"1.使用内存中的DFA.\n2.使用data.dfa中的DFA\n3.输入DFA"<<endl;
			cin>>tmp;
			if(tmp=="1")
			{
				if(adfa.num_q==0)
				{
					cout<<"内存内容为空，您还没有输入过DFA，请先输入:"<<endl;
					adfa.input();	//输入dfa
					adfa.print_dfa();	//输出dfa
					if(adfa.check())	//DFA正确性检查
					{
						adfa.collection();	//语言集列表
					}
				}
				else
				{
					adfa.print_dfa();	//输出dfa
					if(adfa.check())	//DFA正确性检查
					{
						adfa.collection();	//语言集列表
					}
				}
			}
			
			else if(tmp=="2")
			{
				cout<<"文件data.dfa的内容为:"<<endl;
				if(adfa.read())
				{
					adfa.print_dfa();	//输出dfa
					if(adfa.check())	//DFA正确性检查
					{
						adfa.collection();	//语言集列表
					}
				}
			}
			
			else if(tmp=="3")
			{
				adfa.input();	//输入DFA
				adfa.print_dfa();	//输出dfa
				if(adfa.check())	//DFA正确性检查
				{
					adfa.collection();	//语言集列表
				}
			}
			else
			{
				cout<<"输入错误!"<<endl;
				break;
			}
			system("pause");
		}
		else if(choice=="5")
		{
			string tmp;		//选择
			cout<<"1.使用内存中的DFA.\n2.使用data.dfa中的DFA\n3.输入DFA"<<endl;
			cout<<"请选择:";
			cin>>tmp;
			if(tmp=="1")
			{
				if(adfa.num_q==0)
				{
					cout<<"内存内容为空，您还没有输入过DFA，请先输入:"<<endl;
					adfa.input();	//输入dfa
					adfa.print_dfa();	//输出dfa
					if(adfa.check())	//DFA正确性检查
					{
						input_str(str);	//输入字符串
						adfa.judge(str);		//规则字符串判断
					}
				}
				else
				{
					adfa.print_dfa();	//输出dfa
					if(adfa.check())	//DFA正确性检查
					{
						input_str(str);	//输入字符串
						adfa.judge(str);		//规则字符串判断
					}
				}
			}
			
			else if(tmp=="2")
			{
				cout<<"文件data.dfa的内容为:"<<endl;
				if(adfa.read())
				{
					adfa.print_dfa();	//输出dfa
					if(adfa.check())	//DFA正确性检查
					{
						input_str(str);
						adfa.judge(str);		//规则字符串判断
					}
				}
			}
			
			else if(tmp=="3")
			{
				adfa.input();	//输入DFA
				adfa.print_dfa();	//输出dfa
				if(adfa.check())	//DFA正确性检查
				{
					input_str(str);
					adfa.judge(str);		//规则字符串判断
				}
			}
			else
			{
				cout<<"输入错误!"<<endl;
				break;
			}
			system("pause");
		}
		else if(choice=="0")
		{
			break;
		}
		else
		{
			cout<<"输入错误！"<<endl;
			system("pause");
		}
		//adfa.release();
	}while(1);

	system("pause");
	return 0;
}

void input_str(char *str)
{
	cout<<"请输入待测字符串:";
	cin>>str;
}