// password.cpp : ÄÜ¼Ö ÀÀ¿ë ÇÁ·Î±×·¥¿¡ ´ëÇÑ ÁøÀÔÁ¡À» Á¤ÀÇÇÕ´Ï´Ù.
//
// VC++·Î Äָܼðµå¿¡¼­ ÄÄÆÄÀÏÇÒ¶§ »ç¿ë
// #include "stdafx.h"

#include <iostream>
#include <stdlib.h>
#include <string>
#include <math.h>
#include <stdio.h>
#include <vector>
#include <utility>

using namespace std;

class Cryptography
{
	static const int MAXROW=3;
	static const int char_size= 128;
	typedef pair<char*,int*> EncodeUnit;
	//int			numrow,irow;
	double		dtemp,pivot;
	double		a[MAXROW][2*MAXROW];
	double		a_1[MAXROW][MAXROW];

	double      originalKey[MAXROW][MAXROW];

	string      original_Code;

	string      password_Code;
	string      password_Decode;

	vector<EncodeUnit> EncodeUnitVec;


	void InverseMatrix()  
	{
		int			i,j,k;

		int irow=0;
		/*..pivot..*/
		for(i=0;i<MAXROW-1;i++)
		{
			irow=i;
			for(k=i+1;k<MAXROW;k++)
			{
				if(fabs(a[irow][i])<fabs(a[k][i])) irow=k;
			}

			for(j=0;j<2*MAXROW;j++)
			{
				dtemp=a[i][j];
				a[i][j]=a[irow][j];
				a[irow][j]=dtemp;
			} // for(j=0;j<2*numrow;j++)
		} // for(i=0;i<numrow-1;i++)


		/*..Gauss-Jordan..*/
		for(i=0;i<MAXROW;i++)
		{
			pivot=a[i][i];
			for(j=0;j<2*MAXROW;j++) a[i][j]/=pivot;
			for(k=0;k<MAXROW;k++)
			{
				if(k==i) continue;
				dtemp=a[k][i];
				for(j=i;j<2*MAXROW;j++)
				{
					a[k][j]-=dtemp*a[i][j];
				}
			} // for(k=0;k<numrow;k++)
		} // for(i=0;i<numrow;i++)

		/*..Inverse Matrix..*/
		for(i=0;i<MAXROW;i++)
			for(j=0;j<MAXROW;j++) a_1[i][j]=a[i][j+MAXROW];
	}
	
	void MakeEncodeVector(vector<char*> Extract3charVec)
	{
		for(vector<char*>::iterator it=Extract3charVec.begin(); it != Extract3charVec.end(); ++it)
		{
			char* pcharUnit= new char[3];
			int*  pDivCount= new int[3];

			cout<<"È®ÀÎ Extract3charVecÀÇ ¿ä¼Ò :"<<*it<<endl;

			for(int i=0; i <MAXROW; ++i)
			{
				pDivCount[i] = (int) ( ((originalKey[i][0] * (*it)[0]) + (originalKey[i][1] * (*it)[1]) +
					(originalKey[i][2] * (*it)[2])) / char_size );

				pcharUnit[i] = (int) ((originalKey[i][0] * (*it)[0]) + (originalKey[i][1] * (*it)[1]) +
					(originalKey[i][2] * (*it)[2]) ) % char_size;

				//È®ÀÎ ÄÚµå
				cout<<"È®ÀÎ pDivCount[i]*char_size + pcharUnit[i] : "<<pDivCount[i]*char_size + pcharUnit[i]<<endl;
				cout<<"È®ÀÎ ((originalKey[i][0] * (*it)[0]) + (originalKey[i][1] * (*it)[1]) + \
					  (originalKey[i][2] * (*it)[2])) : "<<((originalKey[i][0] * (*it)[0]) + 
					  (originalKey[i][1] * (*it)[1]) +
					  (originalKey[i][2] * (*it)[2]))<<endl;

				password_Code.push_back(pcharUnit[i]);
			} // for(int i=0; i <MAXROW; ++i)
			delete [] *it;

			EncodeUnitVec.push_back(make_pair(pcharUnit,pDivCount));
		} // for(vector<char*>::iterator it=Extract3charVec.begin(); it != Extract3charVec.end(); ++it)
	}

	void FreeEncodeUnit()
	{
		for(vector<EncodeUnit>::iterator it=EncodeUnitVec.begin(); it !=EncodeUnitVec.end(); ++it)
		{
			delete [] it->first; delete [] it->second;
		}
	}

	void Encode()
	{
		int mod = original_Code.size()%MAXROW;
		if(mod != 0)
			original_Code.resize(original_Code.size()+MAXROW-mod,' ');

		vector<char*> Extract3charVec;
		// 3°³ ¹®ÀÚ ´ÜÀ§·Î ÃßÃâ
		for(int i=0; i < original_Code.size()/MAXROW; ++i)
		{
			char* pInputChar = new char[4];
			memset(pInputChar,0,4);

			pInputChar[0] = *(original_Code.begin()+(i*MAXROW)+0);
			pInputChar[1] = *(original_Code.begin()+(i*MAXROW)+1);
			pInputChar[2] = *(original_Code.begin()+(i*MAXROW)+2);

			cout<<"ÀÔ·Â ¹ÞÀº ¹®ÀÚ °ª È®ÀÎ : "<<(int)pInputChar[0]<<" "<<(int)pInputChar[1]<<" "<<(int)pInputChar[2]<<endl;

			Extract3charVec.push_back(pInputChar);
		} // for(int i=0; i < original_Code.size()/MAXROW; ++i)

		MakeEncodeVector(Extract3charVec);
	
	}

	void Decode()
	{
		vector<int*> DecodeVec;

		for(vector<EncodeUnit>::iterator it=EncodeUnitVec.begin(); it !=EncodeUnitVec.end(); ++it)
		{
			int* pDecodeUnit= new int[3];
			pDecodeUnit[0] = ((it->second)[0]*char_size) + (it->first)[0];
			pDecodeUnit[1] = ((it->second)[1]*char_size) + (it->first)[1];
			pDecodeUnit[2] = ((it->second)[2]*char_size) + (it->first)[2];

			cout<<"È®ÀÎ pDecodeUnit[0] : "<<pDecodeUnit[0]<<endl;
			cout<<"È®ÀÎ pDecodeUnit[1] : "<<pDecodeUnit[1]<<endl;
			cout<<"È®ÀÎ pDecodeUnit[2] : "<<pDecodeUnit[2]<<endl;

			DecodeVec.push_back(pDecodeUnit);
		} // for(vector<EncodeUnit>::iterator it=EncodeUnitVec.begin(); it !=EncodeUnitVec.end(); ++it)


		for(vector<int*>::iterator it=DecodeVec.begin(); it !=DecodeVec.end(); ++it)
		{
			double temp[MAXROW];
			for(int i=0; i <MAXROW; ++i)
			{
				temp[i] = ( (double)(a_1[i][0] * (*it)[0]) + ((double)a_1[i][1] * (*it)[1]) +
					((double)a_1[i][2] * (*it)[2]) );

				cout<<"È®ÀÎ temp[i] : "<<temp[i]<<endl;
				cout<<"È®ÀÎ ¹®ÀÚ : "<<static_cast<char>(temp[i])<<endl;

				password_Decode.push_back((char)temp[i]);
			} // for(int i=0; i <MAXROW; ++i)
		} // for(vector<EncodeUnit>::iterator it=EncodeUnitVec.begin(); it !=EncodeUnitVec.end(); ++it)

		for(vector<int*>::iterator it=DecodeVec.begin(); it != DecodeVec.end(); ++it)
		{
			delete [] *it;
		} // for(vector<EncodeUnit>::iterator it=EncodeUnitVec.begin(); it !=EncodeUnitVec.end(); ++it)
	}

public:
	Cryptography()
	{	/*..Initialize..*/
		for(int i=0;i<MAXROW;i++)
		{
			for(int j=0;j<MAXROW;j++)   a_1[i][j]=0.;
			for(int j=0;j<2*MAXROW;j++) a[i][j]=0.;
		}
	}


	void InputKeyArray()
	{
		int			i,j;

		cout<<"Å°¸¦ ÀÔ·ÂÇϼ¼¿ä ~"<<endl;
		cout<<"3X3 ¹è¿­ ¹è¿­ÀÔ´Ï´Ù Â÷·Ê·Î 9°³ÀÇ ¼ýÀÚ¸¦ ÀÔ·ÂÇØÁÖ¼¼¿ä"<<endl;
		cout<<"°æ°í(WARNNING) ! : ÀÔ·ÂÇϽŠ¹è¿­ÀÇ ¿ªÇà·ÄÀÇ ¿ä¼ÒÁß¿¡¼­ Á¤¼ö°¡ ¾Æ´Ï¶ó ¼Ò¼ö Áï ½Ç¼ö°ªÀÌ "
			"µé¾î ÀÖÀ» °æ¿ì ¹ö±×°¡ ¹ß»ýÇÕ´Ï´Ù,(±âÁ¸ÀÇ Á¤ÇØ ³õÀº Å° ¹è¿­À» »ç¿ëÇÒ°ÍÀ» ÃßõÇÔ)" <<endl;

	/*..Input Matrix Component..*/
		for(i=0;i<MAXROW;i++)
		{
			for(j=0;j<MAXROW;j++)
			{
				printf("a[%d][%d]=",i,j);
				scanf("%lf",&(a[i][j]));
				originalKey[i][j]=a[i][j];

				if(i==j) a[i][j+MAXROW]=1.;
				else     a[i][j+MAXROW]=0.;
			} // for(j=0;j<numrow;j++)
		} // for(i=0;i<MAXROW;i++)

		InverseMatrix();
	}

	void FixArray()
	{
		int i,j;

		a[0][0]=3; a[0][1]=1; a[0][2]=2; a[1][0]=1; a[1][1]=0; a[1][2]=1; a[2][0]=0; a[2][1]=0; a[2][2]=1;
		originalKey[0][0]=3; originalKey[0][1]=1; originalKey[0][2]=2; originalKey[1][0]=1; originalKey[1][1]=0;
		originalKey[1][2]=1; originalKey[2][0]=0; originalKey[2][1]=0; originalKey[2][2]=1;

		for(i=0;i<MAXROW;i++)
		{
			for(j=0;j<MAXROW;j++)
			{
				if(i==j) a[i][j+MAXROW]=1.;
				else     a[i][j+MAXROW]=0.;
			} // for(j=0;j<numrow;j++)
		} // for(i=0;i<MAXROW;i++)

		InverseMatrix();
	}

	void ShowKeyArray()
	{
		int			i,j;
		cout<<" ÀÔ·ÂÇϽŠŰ ¹è¿­À» º¸¿©ÁÝ´Ï´Ù "<<endl;
		
		for(i=0;i<MAXROW;i++)
		{
			for(j=0;j<MAXROW;j++) printf("%5.3f ",originalKey[i][j]);
	
			printf("\n");
		}
	}

	void ShowInverseArray()
	{
		int			i,j;
		printf("\nInverse Matrix is...\n");

		for(i=0;i<MAXROW;i++)
		{
			for(j=0;j<MAXROW;j++) printf("%5.3f ",a_1[i][j]);
	
			printf("\n");
		}
	}

	void WriteMessage()
	{
		cout<<"¾Ïȣȭ ÇÒ ¹®ÀÚ¸¦ ÀÔ·ÂÇϼ¼¿ä (¶Ù¾î¾²±â´Â Çã¿ëµÇÁö ¾Ê½À´Ï´Ù)"<<endl;

		cin>>original_Code;

		cout<<"ÀÔ·ÂÇϽŠ¹®ÀÚ´Â : "<<original_Code<<" ÀÔ´Ï´Ù"<<endl;

		cout<<"ÀÌ ¹®ÀÚ¿­À» ¾Ïȣȭ ½Ãŵ´Ï´Ù"<<endl;

		Encode();
	}

	void ShowPasswordCode()
	{
		cout<<password_Code<<endl;
	}

	void ShowPasswordDecode()
	{
		Decode();
	
		cout<<"¾ÏÈ£Çص¶ ¿Ï·á(¿øº») : "<<password_Decode<<endl;
	}
};

int main()
{
	Cryptography cry;
	int select;
	cout<< "¹Ì¸® ¼³Á¤µÈ Å°¸¦ »ç¿ëÇϽǷÁ¸é 1¹øÀ»  Á÷Á¢ ¸í·ÉÀ» ÀÔ·ÂÇϽ÷Á¸é 2¹øÀ» ´­·¯´©¼¼¿ä "<<endl;
	cin >> select;

	if( select ==2 )
	{
		cry.InputKeyArray();
		cry.ShowKeyArray();
		cry.ShowInverseArray();
	} // if( select ==1 )
	else if(select == 1)
	{
		cry.FixArray();
		cry.ShowKeyArray();
		cry.ShowInverseArray();
	}
	else
	{
		cout<<"À߸ø ÀÔ·ÂÇϼ̽À´Ï´Ù"<<endl;
		exit(1);
	}

	cry.WriteMessage();

	cout<<"¾Ïȣȭ ½ÃŲ ¹®ÀÚ¿­À» º¾´Ï´Ù"<<endl;
	cry.ShowPasswordCode();

	cout<<"¾Ïȣȭ ½ÃŲ ¹®ÀÚ¿­À» Çص¶ÇÕ´Ï´Ù"<<endl;
	cry.ShowPasswordDecode();

	system("pause");
	return 0;
}