// Filename   : alloc.cpp
// Last Change: 2021-05-10 月 10:53:23.
//
#include <wx/dir.h>
#include <wx/filefn.h> 
#include <wx/textfile.h>
#include <wx/utils.h> 
#include <wx/msgdlg.h>
#include <wx/progdlg.h>
#include <wx/filename.h>
#include <wx/fileconf.h>

#include "alloc.h"

#define WIDTH  2480
#define HEIGHT 3508


AllocFrame::AllocFrame(wxWindow* parent, wxWindowID id, const wxString& title, const wxPoint& pos, const wxSize& size, long style) 
	: wxFrame(parent, id, title, pos, size, style)
{
	CreateControls();
	InitializeControlsValue();
	m_buttonAlloc->Connect(wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler(AllocFrame::OnAlloc), NULL, this);
}

AllocFrame::~AllocFrame()
{
	m_buttonAlloc->Disconnect(wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler(AllocFrame::OnAlloc), NULL, this);
}

/* Functions */
void AllocFrame::CreateControls()
{
	this->SetIcon(wxIcon(wxT("sample")));
	this->SetSizeHints(wxDefaultSize, wxDefaultSize);
	this->SetBackgroundColour(wxColour(196, 234, 170));
	wxSize pksz = wxSize(330, -1);

	wxFlexGridSizer* fgSizer = new wxFlexGridSizer(0, 2, 0, 0);
	fgSizer->SetFlexibleDirection(wxBOTH);
	fgSizer->SetNonFlexibleGrowMode(wxFLEX_GROWMODE_SPECIFIED);

	m_staticTextPdf = new wxStaticText(this, wxID_ANY, wxT("PDFファイル"), wxDefaultPosition, wxDefaultSize, 0);
	fgSizer->Add(m_staticTextPdf, 0, wxALL|wxALIGN_RIGHT|wxALIGN_CENTER_VERTICAL, 5);

	m_filePickerPdf1 = new wxFilePickerCtrl(this, wxID_ANY, wxEmptyString, wxT("Select a file"), wxT("*.pdf"), wxDefaultPosition, pksz, wxFLP_FILE_MUST_EXIST|wxFLP_SMALL|wxFLP_USE_TEXTCTRL);
	fgSizer->Add(m_filePickerPdf1, 0, wxALL|wxALIGN_CENTER_VERTICAL, 5);

	fgSizer->Add(0, 0, 1, wxEXPAND, 5);

	m_filePickerPdf2 = new wxFilePickerCtrl(this, wxID_ANY, wxEmptyString, wxT("Select a file"), wxT("*.pdf"), wxDefaultPosition, pksz, wxFLP_FILE_MUST_EXIST|wxFLP_SMALL|wxFLP_USE_TEXTCTRL);
	fgSizer->Add(m_filePickerPdf2, 0, wxALL|wxALIGN_CENTER_VERTICAL, 5);

	fgSizer->Add(0, 0, 1, wxEXPAND, 5);

	m_checkBoxSkipPdf = new wxCheckBox(this, wxID_ANY, wxT("PDFからの変換をスキップする"), wxDefaultPosition, wxDefaultSize, 0);
	fgSizer->Add(m_checkBoxSkipPdf, 0, wxALL, 5);

	m_staticTextTif = new wxStaticText(this, wxID_ANY, wxT("TIFフォルダ"), wxDefaultPosition, wxDefaultSize, 0);
	fgSizer->Add(m_staticTextTif, 0, wxALL|wxALIGN_CENTER_VERTICAL|wxALIGN_RIGHT, 5);

	m_dirPickerTif = new wxDirPickerCtrl(this, wxID_ANY, wxEmptyString, wxT("Select a folder"), wxDefaultPosition, pksz, wxDIRP_DIR_MUST_EXIST|wxDIRP_SMALL|wxDIRP_USE_TEXTCTRL);
	fgSizer->Add(m_dirPickerTif, 0, wxALL|wxALIGN_CENTER_VERTICAL, 5);

	fgSizer->Add(0, 0, 1, wxEXPAND, 5);

	m_checkBoxJ2T = new wxCheckBox(this, wxID_ANY, wxT("JPEGから変換する"), wxDefaultPosition, wxDefaultSize, 0);
	//m_checkBoxJ2T->SetValue(true);
	fgSizer->Add(m_checkBoxJ2T, 0, wxALL, 5);

	m_staticTextWork = new wxStaticText(this, wxID_ANY, wxT("作業フォルダ"), wxDefaultPosition, wxDefaultSize, 0);
	fgSizer->Add(m_staticTextWork, 0, wxALL|wxALIGN_CENTER_VERTICAL|wxALIGN_RIGHT, 5);

	m_dirPickerWork = new wxDirPickerCtrl(this, wxID_ANY, wxEmptyString, wxT("Select a folder"), wxDefaultPosition, pksz, wxDIRP_DIR_MUST_EXIST|wxDIRP_SMALL|wxDIRP_USE_TEXTCTRL);
	fgSizer->Add(m_dirPickerWork, 0, wxALL|wxALIGN_CENTER_VERTICAL, 5);

	fgSizer->Add(0, 0, 1, wxEXPAND, 5);

	m_buttonAlloc = new wxButton(this, wxID_ANY, wxT("コピー"), wxDefaultPosition, wxDefaultSize, 0);
	fgSizer->Add(m_buttonAlloc, 0, wxALL|wxALIGN_RIGHT, 5);

	this->SetSizer(fgSizer);
	this->Layout();

	this->Centre(wxBOTH);

    m_filePickerPdf1->SetDropTarget(new DnDFile(m_filePickerPdf1));
    m_filePickerPdf2->SetDropTarget(new DnDFile(m_filePickerPdf2));
}

void AllocFrame::InitializeControlsValue()
{
    wxString conf_file = wxGetCwd() + wxFILE_SEP_PATH + wxT("alloc.conf");
    wxFileConfig* conf = new wxFileConfig(wxT("MyApp"), wxT("T.Mutoh"), conf_file, wxEmptyString, wxCONFIG_USE_LOCAL_FILE);

    wxString pdf1, pdf2, tiff, work;
    conf->SetPath(wxT("/Files"));
    conf->Read(wxT("pdf1"), &pdf1);
    conf->Read(wxT("pdf2"), &pdf2);
    conf->SetPath(wxT("/Dirs"));
    conf->Read(wxT("tiff"), &tiff);
    conf->Read(wxT("work"), &work);

	delete conf;
	
	m_filePickerPdf1->SetPath(pdf1);
	m_filePickerPdf2->SetPath(pdf2);
	m_dirPickerTif->SetPath(tiff);
	m_dirPickerWork->SetPath(work);
}

void AllocFrame::Pdf2Jpg()
{
	wxString kanso  = m_filePickerPdf1->GetPath();
	wxString normal = m_filePickerPdf2->GetPath();

	if (!kanso.IsEmpty())
		wxCopyFile(kanso, m_dirPickerWork->GetPath() + wxFILE_SEP_PATH + wxT("1.pdf"));
	wxCopyFile(normal, m_dirPickerWork->GetPath() + wxFILE_SEP_PATH + wxT("2.pdf"));

	wxString batchfile = m_dirPickerWork->GetPath() + wxFILE_SEP_PATH + wxT("alloc.bat");
	wxTextFile textfile;

	wxString cmd;
	textfile.Create(batchfile);

	cmd = wxT("cd ") + m_dirPickerWork->GetPath();
	textfile.AddLine(cmd);

	//cmd = wxT("\"C:\\Program Files\\gs\\gs9.50\\bin\\gswin64c.exe\" -dBATCH -dNOPAUSE -q -sDEVICE=pdfwrite -sOutputFile=a.pdf 1.pdf 2.pdf");
	cmd = wxT("\"C:\\Program Files\\gs\\gs9.50\\bin\\gswin64c.exe\" -dBATCH -dNOPAUSE -q -sDEVICE=pdfwrite -sOutputFile=a.pdf ");
	if (!kanso.IsEmpty()) cmd += wxT("1.pdf ");
	cmd += wxT("2.pdf");
	textfile.AddLine(cmd);

	for (int i = 0; i < 50; i++) {
		cmd = wxString::Format(wxT("magick -density 400 a.pdf[%d] _%02d.jpg"), i, i + 1);
		textfile.AddLine(cmd);
	}

	wxCSConv cust(wxT("cp932"));
	textfile.Write(wxTextFileType_Dos, cust);
	textfile.Close();

	wxExecute(batchfile, wxEXEC_SYNC|wxEXEC_HIDE_CONSOLE, NULL, NULL);
}

bool AllocFrame::AllocateTif()
{
	wxArrayString files;
	if (m_checkBoxJ2T->IsChecked()) 	// scansnap ix1400
		wxDir::GetAllFiles(m_dirPickerTif->GetPath(), &files, wxT("*.jpg"), wxDIR_FILES);
	else {								// im c6000
		wxDir::GetAllFiles(m_dirPickerTif->GetPath(), &files, wxT("*.tif"), wxDIR_FILES);
		if (files.GetCount() == 0)
			wxDir::GetAllFiles(m_dirPickerTif->GetPath(), &files, wxT("*.tiff"), wxDIR_FILES);
	}
	files.Sort();

	wxProgressDialog pd(wxT("Progress Dialog"), wxT("Wait a moment..."), files.GetCount(), this, wxPD_CAN_ABORT|wxPD_APP_MODAL|wxPD_ELAPSED_TIME|wxPD_ESTIMATED_TIME|wxPD_REMAINING_TIME);
	bool cout = true;
	for (int i = 0; i < files.GetCount(); i++ ) {
		wxString work_path = m_dirPickerWork->GetPath() + wxFILE_SEP_PATH + wxString::Format(wxT("%02d"), i / 8 + 1);
		if (i % 8 == 0) wxFileName::Mkdir(work_path, wxS_DIR_DEFAULT, 0);
		if (i % 8 == 3 || i % 8 == 5 || i % 8 == 7) continue;

		wxFileName f(files[i]);
		wxString outtif = work_path + wxFILE_SEP_PATH + f.GetName() + wxT(".tif");

		if (m_checkBoxJ2T->IsChecked()) {	// scansnap ix1400
			wxImage jpg(files[i], wxBITMAP_TYPE_JPEG);
			wxImage tif = jpg.Resize(wxSize(WIDTH, HEIGHT), wxPoint(0, 0), 255, 255, 255);
			//wxImage tif = jpg.Scale(WIDTH, HEIGHT, wxIMAGE_QUALITY_NORMAL);
			tif.SetOption(wxIMAGE_OPTION_RESOLUTIONUNIT, wxIMAGE_RESOLUTION_INCHES);
			tif.SetOption(wxIMAGE_OPTION_RESOLUTION, 300);
			tif.SetOption(wxIMAGE_OPTION_TIFF_BITSPERSAMPLE, 8);	// ビットの深さ 1(b/w) or 8(rgb, greyscale)
			tif.SetOption(wxIMAGE_OPTION_TIFF_PHOTOMETRIC, 1);		// 2(rgb) or 0(greyscale) or 1(grayscale)
			tif.SetOption(wxIMAGE_OPTION_TIFF_SAMPLESPERPIXEL, 1);  // チャンネル数 1(greyscale) or 3(rgb)
			tif.SetOption(wxIMAGE_OPTION_TIFF_COMPRESSION, 5);		// 1(非圧縮), 2(ITU-T), 3(ファクシミリ互換のITU-T Group 3), 4(ファクシミリ互換のITU-T Group 4), 5(固定長コードLZW圧縮), 6(JPEG), 7(JPEG圧縮-2), 8(ZIP圧縮), 32773(Packbits圧縮)

			/*
			tif.SetOption(wxIMAGE_OPTION_TIFF_BITSPERSAMPLE, 1);	// ビットの深さ 1(b/w) or 8(rgb, greyscale)
			tif.SetOption(wxIMAGE_OPTION_TIFF_PHOTOMETRIC, 0);		// 2(rgb) or 0(greyscale) or 1(grayscale)
			//tif.SetOption(wxIMAGE_OPTION_TIFF_SAMPLESPERPIXEL, 3); // チャンネル数 1(greyscale) or 3(rgb)
			tif.SetOption(wxIMAGE_OPTION_TIFF_COMPRESSION, 4);		// 1(非圧縮), 2(ITU-T), 3(ファクシミリ互換のITU-T Group 3), 4(ファクシミリ互換のITU-T Group 4), 5(固定長コードLZW圧縮), 6(JPEG), 7(JPEG圧縮-2), 8(ZIP圧縮), 32773(Packbits圧縮)
			tif.ConvertToGreyscale();
			*/
			tif.SaveFile(outtif, wxBITMAP_TYPE_TIFF);
		} else {	// im c6000 (.tiff -> .tif)
			wxCopyFile(f.GetFullPath(), outtif);
		}

		if (i == files.GetCount())
			cout = pd.Update(i, wxT("Thanks !"));
		if (i == files.GetCount() / 2)
			cout = pd.Update(i, wxT("Only a half left !"));
		else
			cout = pd.Update(i);
		if (!cout) {
			if (wxMessageBox(wxT("Do you really want to cancel ?"), wxT("Question"), wxYES_NO|wxICON_QUESTION) == wxYES)
				break;
			pd.Resume();
		}
	}
	return cout;
}

/* Event Handlers */
void AllocFrame::OnAlloc(wxCommandEvent& WXUNUSED(event))
{
	wxFileName::Rmdir(m_dirPickerWork->GetPath(), wxPATH_RMDIR_RECURSIVE);
	wxFileName::Mkdir(m_dirPickerWork->GetPath(), wxS_DIR_DEFAULT, 0);

	if (AllocateTif()) {
		wxMessageDialog dlg(this, wxT("Mover を起動しますか？"), wxT("Question ?"), wxYES_NO);
		if (dlg.ShowModal() == wxID_YES) {
			wxExecute(wxT("mover.exe"), wxEXEC_ASYNC, NULL, NULL);
		}
	}
	if (!m_checkBoxSkipPdf->IsChecked()) Pdf2Jpg();
	wxMessageBox(wxT("コピー完了"));
}