// Filename   : geraline.cpp
// Last Change: 2021-01-29 金 16:30:52.
//
#include <wx/dir.h>
#include <wx/msgdlg.h>
#include <wx/textfile.h>
#include <wx/filename.h>
#include <wx/datetime.h>
#include "geraline.h"

#define GRID_SZ 9
#define GRID_W 100
#define GRID_H  60
#define GRID_DY 30 // offset
#define ORGIMG_W 2480
#define ORGIMG_H 3508
#define THUMB_W 400
#define THUMB_H 560

EralineFrame::EralineFrame(wxWindow* parent, wxWindowID id, const wxString& title, const wxPoint& pos, const wxSize& size, long style) 
	: wxFrame(parent, id, title, pos, size, style)
{
	CreateControls();

	// Connect Events
	this->Connect(wxEVT_SIZE, wxSizeEventHandler(EralineFrame::OnSize));
	m_dirPickerInDir->Connect(wxEVT_COMMAND_DIRPICKER_CHANGED, wxFileDirPickerEventHandler(EralineFrame::OnInDirChanged), NULL, this);
	m_dataViewListCtrl->Connect(wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED, wxDataViewEventHandler(EralineFrame::OnItemSelected), NULL, this);
	m_bitmapThumb->Connect(wxEVT_LEFT_DCLICK, wxMouseEventHandler(EralineFrame::OnThumbDClicked), NULL, this);
	m_bitmapZoom->Connect(wxEVT_LEFT_DCLICK, wxMouseEventHandler(EralineFrame::OnZoomDClicked), NULL, this);
	m_gridView->Connect(wxEVT_GRID_RANGE_SELECT, wxGridRangeSelectEventHandler(EralineFrame::OnRangeSelect), NULL, this);
	/*
	m_spinCtrlY->Connect( wxEVT_COMMAND_SPINCTRL_UPDATED, wxSpinEventHandler( EralineFrame::OnSpinY ), NULL, this );
	m_spinCtrlY->Connect( wxEVT_COMMAND_TEXT_UPDATED, wxCommandEventHandler( EralineFrame::OnSpinYText ), NULL, this );
	m_spinCtrlH->Connect( wxEVT_COMMAND_SPINCTRL_UPDATED, wxSpinEventHandler( EralineFrame::OnSpinH ), NULL, this );
	m_spinCtrlH->Connect( wxEVT_COMMAND_TEXT_UPDATED, wxCommandEventHandler( EralineFrame::OnSpinHText ), NULL, this );
	*/
	m_buttonErase->Connect(wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler(EralineFrame::OnErase), NULL, this);

	m_textCtrlMsg->AppendText(wxString::Format(wxT("Scale = %2.4f x %2.4f\n"), 1.0 * THUMB_W / ORGIMG_W, 1.0 * THUMB_H / ORGIMG_H));

	m_file = wxT("image/eraline.tif");
	LoadFile();
}

EralineFrame::~EralineFrame()
{
	// Disconnect Events
	this->Disconnect(wxEVT_SIZE, wxSizeEventHandler(EralineFrame::OnSize));
	m_dirPickerInDir->Disconnect(wxEVT_COMMAND_DIRPICKER_CHANGED, wxFileDirPickerEventHandler(EralineFrame::OnInDirChanged), NULL, this);
	m_dataViewListCtrl->Disconnect(wxEVT_COMMAND_DATAVIEW_ITEM_ACTIVATED, wxDataViewEventHandler(EralineFrame::OnItemSelected), NULL, this);
	m_bitmapThumb->Disconnect(wxEVT_LEFT_DCLICK, wxMouseEventHandler(EralineFrame::OnThumbDClicked), NULL, this);
	m_bitmapZoom->Disconnect(wxEVT_LEFT_DCLICK, wxMouseEventHandler(EralineFrame::OnZoomDClicked), NULL, this);
	m_gridView->Disconnect(wxEVT_GRID_RANGE_SELECT, wxGridRangeSelectEventHandler(EralineFrame::OnRangeSelect), NULL, this);
	/*
	m_spinCtrlY->Disconnect( wxEVT_COMMAND_SPINCTRL_UPDATED, wxSpinEventHandler( EralineFrame::OnSpinY ), NULL, this );
	m_spinCtrlY->Disconnect( wxEVT_COMMAND_TEXT_UPDATED, wxCommandEventHandler( EralineFrame::OnSpinYText ), NULL, this );
	m_spinCtrlH->Disconnect( wxEVT_COMMAND_SPINCTRL_UPDATED, wxSpinEventHandler( EralineFrame::OnSpinH ), NULL, this );
	m_spinCtrlH->Disconnect( wxEVT_COMMAND_TEXT_UPDATED, wxCommandEventHandler( EralineFrame::OnSpinHText ), NULL, this );
	*/
	m_buttonErase->Disconnect(wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler(EralineFrame::OnErase), NULL, this);
}

// Event-Handlers
void EralineFrame::OnSize(wxSizeEvent& event)
{
	event.Skip();
}

void EralineFrame::OnInDirChanged(wxFileDirPickerEvent& event)
{
	m_dataViewListCtrl->DeleteAllItems();
	wxString path = m_dirPickerInDir->GetPath();
	wxDir dir(path);
	if (!dir.IsOpened()) return;

	int n = 0;
	wxString file;
	bool cont = dir.GetFirst(&file, wxT("*.tif"), wxDIR_FILES);

	wxVector<wxVariant> data;
	while (cont) {
		wxFileName fname(path + wxFILE_SEP_PATH + file);
		wxString sz = wxString::Format(wxT("%s KB"), (fname.GetSize() / 1024).ToString());
		wxString dt = fname.GetModificationTime().Format(wxT("%Y-%m-%d %H:%M:%S"));

		data.push_back(wxVariant(file));
		data.push_back(wxVariant(sz));
		data.push_back(wxVariant(dt));
		m_dataViewListCtrl->AppendItem(data);
		data.clear();
		n++;
		cont = dir.GetNext(&file);
	}

	m_textCtrlMsg->AppendText(wxString::Format(wxT("Input Directory Changed (%d files).\n"), n));
	m_dirPickerOutDir->SetPath(path);
}

void EralineFrame::OnErase(wxCommandEvent& WXUNUSED(event))
{
	wxString indir = m_dirPickerInDir->GetPath();
	wxString outdir = m_dirPickerOutDir->GetPath();
	if (indir.IsSameAs(outdir)) {
		m_textCtrlMsg->AppendText(wxT("Error: Output direcoty is same Input directory.\n"));
		wxMessageBox(wxT("Bad Folder"));
		return;
	}

	wxString cmd, infile, outfile;

	wxTextFile textfile;
	wxString batchfile = outdir + wxFILE_SEP_PATH + wxT("a.bat");
	textfile.Create(batchfile);

	textfile.AddLine(wxT("SET INDIR=") + indir);
	textfile.AddLine(wxT("SET OUTDIR=") + outdir);
	textfile.AddLine(wxT("SET ERALINE=") + wxGetCwd() + wxFILE_SEP_PATH + wxT("eraline.exe"));

	// > eraline.exe y h in.tif out.tif
	for (int r = 0; r < m_dataViewListCtrl->GetItemCount(); r++) {
		cmd = wxString::Format(wxT("%%ERALINE%% %d %d %%INDIR%%%s %%OUTDIR%%%s"), 
				//m_spinCtrlY->GetValue(), 1,
				m_spinCtrlY->GetValue(), m_spinCtrlH->GetValue(),
				wxFILE_SEP_PATH + m_dataViewListCtrl->GetTextValue(r, 0),
				wxFILE_SEP_PATH + m_dataViewListCtrl->GetTextValue(r, 0)
				);
		textfile.AddLine(cmd);
	}

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

	wxExecute(batchfile, wxEXEC_SYNC|wxEXEC_HIDE_CONSOLE, NULL, NULL);
	m_textCtrlMsg->AppendText(wxT("Process done.\n"));
	wxMessageBox(wxT("Process done."));
}

void EralineFrame::OnThumbDClicked(wxMouseEvent& event)
{
	int x, y;
	event.GetPosition(&x, &y);
	m_x0 = (int)(1.0 * ORGIMG_W / THUMB_W * x);
	m_y0 = (int)(1.0 * ORGIMG_H / THUMB_H * y);
	m_textCtrlMsg->AppendText(wxString::Format(wxT("Preview: (%d, %d) Double-Clicked. -> [%d, %d]\n"), x, y, m_x0, m_y0));

	int dy = 40; // for centering
	m_scrolledWindow->Scroll(m_x0 / 10 - dy, m_y0 / 10 - dy);
}

void EralineFrame::OnZoomDClicked(wxMouseEvent& event)
{
	int x, y;
	event.GetPosition(&x, &y);
	m_x1 = x;
	m_y1 = y;
	m_textCtrlMsg->AppendText(wxString::Format(wxT("Zoom: (%d, %d) Double-Clicked.\n"), x, y));

	ViewDot();
}

void EralineFrame::OnItemSelected(wxDataViewEvent& WXUNUSED(event))
{
	int r = m_dataViewListCtrl->GetSelectedRow();
	wxString file = m_dataViewListCtrl->GetTextValue(r, 0);

	m_file = m_dirPickerInDir->GetPath() + wxFILE_SEP_PATH + file;
	LoadFile();
}

void EralineFrame::OnRangeSelect(wxGridRangeSelectEvent& event)
{
	int c0 = event.GetLeftCol();
	int c1 = event.GetRightCol();
	int r0 = event.GetTopRow();
	int r1 = event.GetBottomRow();

	m_spinCtrlY->SetValue(m_y1 + r0 - GRID_DY);
	m_spinCtrlH->SetValue(r1 - r0 + 1);
	m_textCtrlMsg->AppendText(wxString::Format(wxT("Grid: (%d, %d) - (%d, %d) [%d x %d], Selected.\n"), r0, c0, r1, c1, c1 - c0 + 1, r1 - r0 + 1));
}

// Functions
void EralineFrame::CreateControls()
{
	this->SetIcon(wxIcon(wxT("sample")));
	this->SetSizeHints(wxDefaultSize, wxDefaultSize);
	this->SetBackgroundColour(wxColour(230, 230, 45));

	wxGridBagSizer* gbSizer = new wxGridBagSizer(0, 0);
	gbSizer->SetFlexibleDirection(wxBOTH);
	gbSizer->SetNonFlexibleGrowMode(wxFLEX_GROWMODE_SPECIFIED);

	// Col 0
	m_staticTextInDir = new wxStaticText(this, wxID_ANY, wxT("Input Folder"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextInDir, wxGBPosition(0, 0), wxGBSpan(1, 1), wxALL|wxALIGN_RIGHT|wxALIGN_CENTER_VERTICAL, 5);

	m_staticTextList = new wxStaticText(this, wxID_ANY, wxT("Files"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextList, wxGBPosition(1, 0), wxGBSpan(1, 1), wxALL|wxALIGN_RIGHT, 5);

	m_staticTextThumb = new wxStaticText(this, wxID_ANY, wxT("Preview"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextThumb, wxGBPosition(2, 0), wxGBSpan(1, 1), wxALL|wxALIGN_RIGHT, 5);

	m_bitmapLogo = new wxStaticBitmap(this, wxID_ANY, wxNullBitmap, wxDefaultPosition, wxSize(300, 1), 0);
	gbSizer->Add(m_bitmapLogo, wxGBPosition(8, 0), wxGBSpan(1, 2), wxALL, 5);

	// Col 1
	m_dirPickerInDir = new wxDirPickerCtrl(this, wxID_ANY, wxEmptyString, wxT("Select a folder"), wxDefaultPosition, wxSize(280, -1), wxDIRP_DEFAULT_STYLE|wxDIRP_DIR_MUST_EXIST|wxDIRP_SMALL|wxDIRP_USE_TEXTCTRL);
	gbSizer->Add(m_dirPickerInDir, wxGBPosition(0, 1), wxGBSpan(1, 1), wxALL|wxALIGN_CENTER_VERTICAL, 5);

	m_dataViewListCtrl = new wxDataViewListCtrl(this, wxID_ANY, wxDefaultPosition, wxSize(THUMB_W, 500), wxDV_ROW_LINES|wxDV_SINGLE);
	m_dataViewListColumnName = m_dataViewListCtrl->AppendTextColumn(wxT("  Name"), wxDATAVIEW_CELL_INERT, 180, static_cast<wxAlignment>(wxALIGN_LEFT),   wxDATAVIEW_COL_RESIZABLE);
	m_dataViewListColumnSize = m_dataViewListCtrl->AppendTextColumn(wxT("Size  "), wxDATAVIEW_CELL_INERT,  80, static_cast<wxAlignment>(wxALIGN_RIGHT),  wxDATAVIEW_COL_RESIZABLE);
	m_dataViewListColumnTime = m_dataViewListCtrl->AppendTextColumn(wxT("Time"),   wxDATAVIEW_CELL_INERT,  -1, static_cast<wxAlignment>(wxALIGN_CENTER), wxDATAVIEW_COL_RESIZABLE);
	gbSizer->Add(m_dataViewListCtrl, wxGBPosition(1, 1), wxGBSpan(1, 1), wxALL|wxEXPAND, 5);

	m_bitmapThumb = new wxStaticBitmap(this, wxID_ANY, wxNullBitmap, wxDefaultPosition, wxSize(THUMB_W, THUMB_H), 0);
	gbSizer->Add(m_bitmapThumb, wxGBPosition(2, 1), wxGBSpan(6, 1), wxALL, 5);

	// Col 2
	m_staticTextZoom = new wxStaticText(this, wxID_ANY, wxT("-->> Zoom"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextZoom, wxGBPosition(2, 2), wxGBSpan(1, 1), wxALL|wxALIGN_TOP|wxALIGN_RIGHT, 5);

	// Col 3
	m_scrolledWindow = new wxScrolledWindow(this, wxID_ANY, wxDefaultPosition, wxSize(600, 500), wxHSCROLL|wxVSCROLL);
	m_scrolledWindow->SetScrollRate(5, 5);
	m_scrolledWindow->SetScrollbars(10, 10, ORGIMG_W / 10, ORGIMG_H / 10);
	m_scrolledWindow->SetBackgroundColour(wxColour(200, 200, 200));
	m_bitmapZoom = new wxStaticBitmap(m_scrolledWindow, wxID_ANY, wxNullBitmap, wxDefaultPosition, wxSize(300, 400), 0);
	gbSizer->Add(m_scrolledWindow, wxGBPosition(0, 3), wxGBSpan(8, 1), wxEXPAND | wxALL, 5);

	// Col 4
	m_staticTextDot = new wxStaticText(this, wxID_ANY, wxT("-->> Dot"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextDot, wxGBPosition(0, 4), wxGBSpan(1, 1), wxALL|wxALIGN_RIGHT|wxALIGN_CENTER_VERTICAL, 5);

	m_staticTextY = new wxStaticText(this, wxID_ANY, wxT("y"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextY, wxGBPosition(3, 4), wxGBSpan(1, 1), wxALL|wxALIGN_RIGHT|wxALIGN_CENTER_VERTICAL, 5);

	m_staticTextH = new wxStaticText(this, wxID_ANY, wxT("h"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextH, wxGBPosition(4, 4), wxGBSpan(1, 1), wxALL|wxALIGN_RIGHT|wxALIGN_CENTER_VERTICAL, 5);

	m_staticTextOutDir = new wxStaticText(this, wxID_ANY, wxT("Output Folder"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextOutDir, wxGBPosition(5, 4), wxGBSpan(1, 1), wxALL|wxALIGN_RIGHT|wxALIGN_CENTER_VERTICAL, 5);

	m_staticTextMsg = new wxStaticText(this, wxID_ANY, wxT("Message"), wxDefaultPosition, wxDefaultSize, 0);
	gbSizer->Add(m_staticTextMsg, wxGBPosition(6, 4), wxGBSpan(1, 1), wxALL|wxALIGN_RIGHT, 5);

	// Col 5
	m_gridView = new wxGrid(this, wxID_ANY, wxDefaultPosition, wxDefaultSize, 0);
	m_gridView->CreateGrid(GRID_W, GRID_H);
	m_gridView->EnableEditing(false);
	m_gridView->EnableGridLines(true);
	m_gridView->EnableDragGridSize(false);
	m_gridView->EnableDragRowSize(false);
	m_gridView->SetMargins(0, 0);
	m_gridView->EnableDragColMove(false);
	m_gridView->EnableDragColSize(false);
	m_gridView->SetRowMinimalAcceptableHeight(GRID_SZ);
	m_gridView->SetColMinimalAcceptableWidth(GRID_SZ);
	m_gridView->SetDefaultColSize(GRID_SZ, true);
	m_gridView->SetDefaultRowSize(GRID_SZ, true);
	m_gridView->SetColLabelSize(25);
	m_gridView->SetRowLabelSize(25);
	m_gridView->SetColLabelAlignment(wxALIGN_CENTER, wxALIGN_CENTER);
	m_gridView->SetRowLabelAlignment(wxALIGN_RIGHT, wxALIGN_CENTER);
	m_gridView->SetDefaultCellAlignment(wxALIGN_LEFT, wxALIGN_TOP);
	m_gridView->SetLabelFont(wxFont(5, wxFONTFAMILY_DEFAULT, wxFONTSTYLE_NORMAL, wxFONTWEIGHT_NORMAL, false, wxEmptyString));
	m_gridView->SetSelectionBackground(*wxRED);
	gbSizer->Add(m_gridView, wxGBPosition(0, 5), wxGBSpan(3, 3), wxALL, 5);

	m_spinCtrlY = new wxSpinCtrl(this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(80, -1), wxALIGN_RIGHT|wxSP_ARROW_KEYS, 0, 3500, 0);
	gbSizer->Add(m_spinCtrlY, wxGBPosition(3, 5), wxGBSpan(1, 1), wxALL|wxALIGN_CENTER_VERTICAL, 5);

	m_spinCtrlH = new wxSpinCtrl(this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(80, -1), wxALIGN_RIGHT|wxSP_ARROW_KEYS, 1, 10, 1);
	gbSizer->Add(m_spinCtrlH, wxGBPosition(4, 5), wxGBSpan(1, 1), wxALL, 5);

	m_dirPickerOutDir = new wxDirPickerCtrl(this, wxID_ANY, wxEmptyString, wxT("Select a folder"), wxDefaultPosition, wxSize(280, -1), wxDIRP_DEFAULT_STYLE|wxDIRP_DIR_MUST_EXIST|wxDIRP_SMALL|wxDIRP_USE_TEXTCTRL);
	gbSizer->Add(m_dirPickerOutDir, wxGBPosition(5, 5), wxGBSpan(1, 2), wxALL|wxALIGN_CENTER_VERTICAL, 5);

	m_textCtrlMsg = new wxTextCtrl(this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxSize(500, 60), wxTE_MULTILINE);
	gbSizer->Add(m_textCtrlMsg, wxGBPosition(6, 5), wxGBSpan(2, 3), wxALL, 5);

	// Col 6
	m_textCtrlRGB = new wxTextCtrl(this, wxID_ANY, wxEmptyString, wxDefaultPosition, wxDefaultSize, wxTE_READONLY);
	gbSizer->Add(m_textCtrlRGB, wxGBPosition(3, 6), wxGBSpan(1, 1), wxALL, 5);

	// Col 7
	m_buttonErase = new wxButton(this, wxID_ANY, wxT("Erase"), wxDefaultPosition, wxSize(60, -1), 0);
	gbSizer->Add(m_buttonErase, wxGBPosition(5, 7), wxGBSpan(1, 1), wxALL|wxALIGN_CENTER_VERTICAL, 5);

	//
	this->SetSizer(gbSizer);
	this->Layout();

	this->Centre(wxBOTH);
}

void EralineFrame::LoadFile()
{
	if (!wxFileExists(m_file)) return;
    m_image.LoadFile(m_file, wxBITMAP_TYPE_TIFF, 0);

    wxBitmap bitmap = wxBitmap(m_image.Scale(THUMB_W, THUMB_H, wxIMAGE_QUALITY_HIGH));
	m_bitmapThumb->SetBitmap(bitmap);

    bitmap = wxBitmap(m_image);
	m_bitmapZoom->SetBitmap(bitmap);

	m_textCtrlMsg->AppendText(wxString::Format(wxT("Image-Size: %d x %d\n"), m_image.GetWidth(), m_image.GetHeight()));
}

void EralineFrame::ViewDot()
{
	wxGridUpdateLocker noUpdates(m_gridView);
	/*
	if (m_image.GetWidth() != m_gridView->GetNumberCols() || m_image.GetHeight() != m_gridView->GetNumberRows()) {
		m_gridView->DeleteCols(0, m_gridView->GetNumberCols(), true);
		m_gridView->DeleteRows(0, m_gridView->GetNumberRows(), true);
		m_gridView->AppendCols(m_image.GetWidth(), true);
		m_gridView->AppendRows(m_image.GetHeight(), true);
	}
	*/
	int rn = m_gridView->GetNumberRows();
	int cn = m_gridView->GetNumberCols();

	int dy = GRID_DY; // for centering
	if (m_y1 < dy) m_y1 = dy;
	if (m_x1 + cn > ORGIMG_W) m_x1 = ORGIMG_W - cn;
	if (m_y1 + rn > ORGIMG_H) m_y1 = ORGIMG_H - rn;

	unsigned char r, g, b;
	for (int y = 0; y < rn; y++) {
		for (int x = 0; x < cn; x++) {
			r = m_image.GetRed(m_x1 + x, m_y1 + y - dy);
			g = m_image.GetGreen(m_x1 + x, m_y1 + y - dy);
			b = m_image.GetBlue(m_x1 + x, m_y1 + y - dy);
			m_gridView->SetCellBackgroundColour(y, x, wxColour(r, g, b));
		}
	}
	m_textCtrlMsg->AppendText(wxString::Format(wxT("Refresh Grid (%d x %d).\n"), m_gridView->GetNumberCols(), m_gridView->GetNumberRows()));
}