bungo-chat-main.cpp

#include<windows.h>
#include<vector>
#include<map>
#include<iostream>
#include<time.h>
#include<cstdio>
#include<math.h>
#include<vector>
#include "DxLib.h"
#include<queue>
#include<stack>
#include<set>
#include<string>
#include <regex>
#include"mecab.h"
#include"filemaker.h"


/////
#include "MTable.h"
#include "Analyze.h"
#include "RuleBase.h"
#include "Character.h"
#include "Main.h"
/////


#ifdef _DEBUG
	#pragma comment(lib,"libmecab.lib")
#else
	#pragma comment(lib,"libmecab.lib")
#endif



const int M_X=40,M_Y=40;
const int MAX_X=60,MAX_Y=60;
const int MAP_X=640,MAP_Y=480;
const float walkspeed=0.12;
const float mwalkspeed=0.05;
using namespace std;
#define rep(i,n) for(int i=0;i<n;i++)
const int INF =100100100;

#ifdef _DEBUG
	#pragma comment(lib,"libmecab.lib")
#else
	#pragma comment(lib,"libmecab.lib")
#endif


int cnt=0;



int pointer=0;
vector<pair<string,int>> strlog;

#define MAX_INPUT_SIZE 1000
#define CURSOR_X 10
#define CURSOR_Y 600
#define W_space 10
#define WIN_SIDE_S 20
#define LINE_S (WIN_SIDE_S+50)//80以上
#define LINE_K 20
#define LINE_L 50//TAIL_SIZE以上
#define LINE_R 430 //480-TAIL_SIZE以下
#define TAIL_SIZE 40
#define MAX_MOJI 13

int WINAPI WinMain( HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow )
{
	int startmode=0;
    // ＤＸライブラリの初期化
	ChangeWindowMode(TRUE);
	SetGraphMode(480,640,32);
    if( DxLib_Init() < 0 )
    {
        // エラーが発生したら直ちに終了
        return -1 ;
    }
	
	FILE *logfile=fopen("logfile.txt","a");
	char kuchoname[100],yasanname[100];
	sprintf(kuchoname,"オウム");sprintf(yasanname,"だれか");
	fprintf(logfile,"\n\n************************************\n");
	int whiteP = GetColor(255,255,255 ) ;
	int yellowP = GetColor(200,200,0 ) ;
	int blackP = GetColor(0,0,0 ) ;
	int blueP = GetColor(0,0,255 ) ;
	DrawString(200, 310 ,"起動中…",whiteP);
	setlocale(LC_CTYPE, "");

	srand((unsigned) time(NULL));
 
	char md[200];
	while( ProcessMessage() == 0 && CheckHitKey( KEY_INPUT_ESCAPE ) == 0 )
    {
		ClearDrawScreen() ;
		DrawString(90, 310 ,"***Enter to start***",whiteP);
		DrawString(0, 480 ,"何も入力せずエンターで通常開始",yellowP);
		DrawString(0, 500 ,"1を入力してエンターでmode:1",yellowP);
		DrawString(0, 520 ,"2を入力してエンターでmode:2",yellowP);
		if(KeyInputString( CURSOR_X, CURSOR_Y,100/*文字数*2*/ , md ,FALSE/*escキャンセル*/ )==1){
			break;
		}
		ScreenFlip();
	}
	sscanf(md,"%d",&startmode);
	if(startmode<0||startmode>2){startmode=0;}

	char filename[200];
	char mname[100];
	bool ifonline=false;
	vector<string> fileview;
	int index=1001001;
	Character * sensei;
	if(startmode!=2){
		fileview=FIND_FILES();
		while( ProcessMessage() == 0 && CheckHitKey( KEY_INPUT_ESCAPE ) == 0 )
		{
			ClearDrawScreen() ;
			DrawString(0, 0 ,"語句の番号を入れてください(.txtは不要です)",whiteP);
			DrawString(0, 40 ,"＊＊＊＊＊＊ファイル名一覧＊＊＊＊＊",yellowP);
			rep(i,fileview.size()){
				if(i>=fileview.size())break;
				char v[300];
				sprintf(v,"%d:%s",i,fileview[i].c_str());
				DrawString(0, 60+i*20 ,v,yellowP);			
			}
			if(KeyInputString( CURSOR_X, CURSOR_Y,100/*文字数*2*/ , mname ,FALSE/*escキャンセル*/ )==1){
				break;
			}
			ScreenFlip();
		}
		DrawString(180, 310 ,"読み込み中…",whiteP);
		ScreenFlip();
		int index=1001001;sscanf(mname,"%d",&index);
		if(index<fileview.size())sprintf(mname,"%s",fileview[index].c_str());
		fprintf(logfile,"語句：%s\t",mname);
		sprintf(filename,"text\\%s.txt",mname);
		MIKE_BUNGO_DICTIONALY(filename);
		while( ProcessMessage() == 0 && CheckHitKey( KEY_INPUT_ESCAPE ) == 0 )
		{
			ClearDrawScreen() ;
			DrawString(0, 0 ,"口調の番号を入れてください(.txtは不要です)",whiteP);
			DrawString(0, 40 ,"＊＊＊＊＊＊ファイル名一覧＊＊＊＊＊",yellowP);
			rep(i,fileview.size()){
				if(i>=fileview.size())break;
				char v[300];
				sprintf(v,"%d:%s",i,fileview[i].c_str());
				DrawString(0, 60+i*20 ,v,yellowP);			
			}
			if(KeyInputString( CURSOR_X, CURSOR_Y,100/*文字数*2*/ , mname ,FALSE/*escキャンセル*/ )==1){
				break;
			}
			ScreenFlip();
		}
		DrawString(180, 310 ,"読み込み中…",whiteP);
		ScreenFlip();
		index=1001001;sscanf(mname,"%d",&index);
		if(index<fileview.size())sprintf(mname,"%s",fileview[index].c_str());
		sprintf(kuchoname,"%s",mname);
		fprintf(logfile,"口調：%s\t",mname);
		sprintf(filename,"text\\%s.txt",mname);
			MIKE_BUNGO_LEARN(filename);
		while( ProcessMessage() == 0 && CheckHitKey( KEY_INPUT_ESCAPE ) == 0 )
		{
			ClearDrawScreen() ;
			DrawString(0, 0 ,"自分の口調も学習させますか？",whiteP);
			DrawString(0, 40 ,"(「あなたのオウム」の時は必ず学習させてください)",yellowP);
			DrawString(0, 80 ,"0=しない　1=今までの入力に続けて学習させる",yellowP);
			DrawString(0, 120 ,"2=今までの学習はリセットし、学習させる(今までの学習は消えます)",yellowP);
			if(KeyInputString( CURSOR_X, CURSOR_Y,100/*文字数*2*/ , mname ,FALSE/*escキャンセル*/ )==1){
				break;
			}
			ScreenFlip();
		}
		sscanf(mname,"%d",&index);
		if(index==0){
			ifonline=false;
		}else if(index==2){
			FILE *newf;
			newf=fopen("text\\あなたのオウム.txt","w");
			fclose(newf);
		}else{
			ifonline=true;
		}
		ScreenFlip();
		fileview.clear();
	}
	if(startmode!=1){
		fileview=FIND_YFILES();
		while( ProcessMessage() == 0 && CheckHitKey( KEY_INPUT_ESCAPE ) == 0 )
		{
			ClearDrawScreen() ;
			DrawString(0, 0 ,"二人目のファイル番号を入れてください(.txtは不要です)",whiteP);
			DrawString(0, 40 ,"＊＊＊＊＊＊ファイル名一覧＊＊＊＊＊",yellowP);
			rep(i,fileview.size()){
				if(i>=fileview.size())break;
				char v[300];
				sprintf(v,"%d:%s",i,fileview[i].c_str());
				DrawString(0, 60+i*20 ,v,yellowP);			
			}
			if(KeyInputString( CURSOR_X, CURSOR_Y,100/*文字数*2*/ , mname ,FALSE/*escキャンセル*/ )==1){
				break;
			}
			ScreenFlip();
		}
		DrawString(180, 310 ,"読み込み中…",whiteP);
		ScreenFlip();
		index=1001001;sscanf(mname,"%d",&index);
		if(index<fileview.size())sprintf(mname,"%s",fileview[index].c_str());
		sprintf(yasanname,"%s",mname);
		char markovf[200];char rulebasef[200];char _seriff[200];
		fprintf(logfile,"二人目：%s\n",mname);
		sprintf(markovf,"data\\%s.markov",mname);
		sprintf(rulebasef,"RuleBasePatterns.rx");
		sprintf(_seriff,"data\\%s.serif",mname);
		sensei = new Character(std::string(markovf), std::string(rulebasef),std::string(_seriff));
	}

	char input[MAX_INPUT_SIZE];	
	int haikei= LoadGraph( "picture\\PC2016bg2.jpg" ) ;
	int window1=LoadGraph("picture\\PC2016lic2.png");
	int window2=LoadGraph("picture\\PC2016lic.png");
	int tail1=LoadGraph("picture\\PC2016litail2.png");
	int tail2=LoadGraph("picture\\PC2016litail.png");
	int litwin1=LoadGraph("picture\\PC2016lit.png");
	int litwin2=LoadGraph("picture\\PC2016lib.png");
	int litwin3=LoadGraph("picture\\PC2016lit2.png");
	int litwin4=LoadGraph("picture\\PC2016lib2.png");
	while( ProcessMessage() == 0 && CheckHitKey( KEY_INPUT_ESCAPE ) == 0 )
    {
		ClearDrawScreen() ;
		DrawGraph(0,0,haikei,FALSE);
		int L=1;
		rep(i,strlog.size()){
			pair<string,int> h=strlog[strlog.size()-i-1];
			int hsize=h.first.size();
			vector<string> Hyoji;
			int G=1;
			while(h.first.size()>MAX_MOJI*2){
				string fm=h.first.substr(0,MAX_MOJI*2);
				h.first.replace(0,MAX_MOJI*2,"");
				Hyoji.push_back(fm);
				G++;
				hsize=MAX_MOJI*2;
			}
			hsize=26;
			Hyoji.push_back(h.first);
			hsize*=10;
			int LUhasi= CURSOR_Y-LINE_S*(i+1)-LINE_K*(L+G)-W_space;
			
			if(h.second==0){
				DrawExtendGraph(LINE_R-hsize, LUhasi,LINE_R, LUhasi+LINE_K*G+W_space,window1,TRUE);
				DrawGraph(LINE_R, LUhasi ,tail1,TRUE);
				DrawGraph(LINE_R-hsize, LUhasi-WIN_SIDE_S ,litwin3,TRUE);
				DrawGraph(LINE_R-hsize,LUhasi+LINE_K*G+W_space ,litwin4,TRUE);
				SetFontSize(10);
				DrawString(LINE_R-hsize,LUhasi-WIN_SIDE_S ,"あなた",blackP);
				SetFontSize(20);
				rep(j,Hyoji.size()){
					DrawString(LINE_R-hsize, LUhasi+LINE_K*j+W_space ,Hyoji[j].c_str(),blackP);
				}
				
			}
			if(h.second==1){
				DrawExtendGraph(LINE_L, LUhasi, LINE_L+hsize, LUhasi+LINE_K*G+W_space,window2,TRUE);
				DrawGraph(LINE_L-TAIL_SIZE, LUhasi ,tail2,TRUE);
				DrawGraph(LINE_L, LUhasi-WIN_SIDE_S ,litwin1,TRUE);
				DrawGraph(LINE_L, LUhasi+LINE_K*G+W_space ,litwin2,TRUE);
				SetFontSize(10);
				DrawString(LINE_L, LUhasi-WIN_SIDE_S , yasanname,blackP);
				SetFontSize(20);
				rep(j,Hyoji.size()){
					DrawString(LINE_L, LUhasi+LINE_K*j+W_space ,Hyoji[j].c_str(),blackP);
				}
				
			}
			if(h.second==2){
				DrawExtendGraph(LINE_L, LUhasi, LINE_L+hsize, LUhasi+LINE_K*G+W_space,window2,TRUE);
				DrawGraph(LINE_L-TAIL_SIZE, LUhasi ,tail2,TRUE);
				DrawGraph(LINE_L, LUhasi-WIN_SIDE_S ,litwin1,TRUE);
				DrawGraph(LINE_L, LUhasi+LINE_K*G+W_space ,litwin2,TRUE);
				SetFontSize(10);
				DrawString(LINE_L, LUhasi-WIN_SIDE_S ,kuchoname,blackP);
				SetFontSize(20);
				rep(j,Hyoji.size()){
					DrawString(LINE_L, LUhasi+LINE_K*j+W_space ,Hyoji[j].c_str(),blueP);
				}
				
			}
			L+=G;

		}
		if(KeyInputString( CURSOR_X, CURSOR_Y,MAX_INPUT_SIZE/*文字数*2*/ , input ,FALSE/*escキャンセル*/ )==1){
			
			
				fprintf(logfile,"あなた:\t%s\n",input);
				const MeCab::Node* node=nodecreate(input);
				pair<string,int> pinp;
				pinp.first=input;
				if(pinp.first!=""){
					pinp.second=0;
						strlog.push_back(pinp);
					if(startmode==0){
						
						pinp.first=sensei->Respond(std::string(input));
						pinp.second=1;
						if(pinp.first.size()==0){
							pinp.first=MIKE_BUNGO_REPLY(input,ifonline);
							pinp.second=2;
						}
						if(pinp.first.empty()||pinp.first=="。"||pinp.first=="！"||pinp.first=="!"){
							pinp.first=sensei->ForceRespond(input);
							pinp.second=1;
						}
					}else if(startmode==1){
						int br=0;
						pinp.first=MIKE_BUNGO_REPLY(input,ifonline);
						pinp.second=2;
						while(pinp.first.empty()||pinp.first=="。"||pinp.first=="！"||pinp.first=="!"){
							pinp.first=MIKE_BUNGO_REPLY(input,false);
							pinp.second=2;
							br++;
							if(br>10)break;
						}br=0;
						while(pinp.first.empty()||pinp.first=="。"||pinp.first=="！"||pinp.first=="!"){
							pinp.first=MIKE_BUNGO_REPLY("今日は",false);
							pinp.second=2;
							if(br>15)break;
						}
					}else if(startmode==2){
						pinp.first=sensei->RespondF(input);
						pinp.second=1;
						if(pinp.first.empty()||pinp.first=="。"||pinp.first=="！"||pinp.first=="!"){
							pinp.first=sensei->ForceRespond(input);
							pinp.second=1;
						}
					}	
					fprintf(logfile,"システム%d:\t%s\n",pinp.second,pinp.first.c_str());
						strlog.push_back(pinp);
				}
					int p=0;
		}
		ScreenFlip();
	}

    DxLib_End() ;
    return 0 ;
}