/*
	FinalSun/FinalAlert 2 Mission Editor

	Copyright (C) 1999-2024 Electronic Arts, Inc.
	Authored by Matthias Wagner

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

// RTPDlg.cpp: Implementierungsdatei
//

#include "stdafx.h"
#include "finalsun.h"
#include "RTPDlg.h"
#include "variables.h"
#include "functions.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// Dialogfeld CRTPDlg 


CRTPDlg::CRTPDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CRTPDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CRTPDlg)
		// HINWEIS: Der Klassen-Assistent fügt hier Elementinitialisierung ein
	//}}AFX_DATA_INIT
}


void CRTPDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CRTPDlg)
	DDX_Control(pDX, IDC_PREVIEW, m_Preview);
	DDX_Control(pDX, IDC_USED, m_Used);
	DDX_Control(pDX, IDC_AVAIL, m_Available);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CRTPDlg, CDialog)
	//{{AFX_MSG_MAP(CRTPDlg)
	ON_BN_CLICKED(IDC_ADD, OnAdd)
	ON_BN_CLICKED(IDC_REMOVE, OnRemove)
	ON_LBN_SELCHANGE(IDC_AVAIL, OnSelchangeAvail)
	ON_LBN_DBLCLK(IDC_AVAIL, OnDblclkAvail)
	ON_LBN_SELCHANGE(IDC_USED, OnSelchangeUsed)
	ON_LBN_DBLCLK(IDC_USED, OnDblclkUsed)
	ON_WM_PAINT()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// Behandlungsroutinen für Nachrichten CRTPDlg 

BOOL CRTPDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	for (auto const& [seq, unitname] : rules["TerrainTypes"]) {
		CString addedString = unitname;

		if (g_data["IgnoreRA2"].HasValue(unitname)) {
			continue;
		}
		addedString = TranslateStringACP(addedString);

		if (unitname.Find("TREE") >= 0) {
			if (unitname.GetLength() > 0 && unitname != "VEINTREE") // out with it :-)
			{
				auto const& theaterType = Map->GetTheater();
				int TreeMin = g_data.GetInteger(theaterType + "Limits", "TreeMin");
				int TreeMax = g_data.GetInteger(theaterType + "Limits", "TreeMax");

				CString id = unitname;
				id.Delete(0, 4);
				int n = atoi(id);


				if (n<TreeMin || n>TreeMax) {
					continue;
				}

				m_Available.AddString(unitname);
			}
		}
	}

	return TRUE;  // return TRUE unless you set the focus to a control
				  // EXCEPTION: OCX-Eigenschaftenseiten sollten FALSE zurückgeben
}

void CRTPDlg::OnAdd()
{
	int sel = m_Available.GetCurSel();
	if (sel < 0) return;

	CString currentname;
	m_Available.GetText(sel, currentname);

	m_Available.DeleteString(sel);
	m_Used.AddString(currentname);
}

void CRTPDlg::OnRemove()
{
	int sel = m_Used.GetCurSel();
	if (sel < 0) return;

	CString currentname;
	m_Used.GetText(sel, currentname);

	m_Used.DeleteString(sel);
	m_Available.AddString(currentname);
}

void CRTPDlg::OnOK()
{
	if (m_Used.GetCount() <= 0) {
		//AfxMessageBox("Please select at least one tree","Error", MB_OK);
		return;
	}

	rndterrainsrc.clear();

	int i;
	for (i = 0; i < m_Used.GetCount(); i++) {
		CString str;
		m_Used.GetText(i, str);
		rndterrainsrc.push_back(str);
	}

	CDialog::OnOK();
}

void CRTPDlg::OnSelchangeAvail()
{
	int sel = m_Available.GetCurSel();

	if (sel < 0) {
		m_LastSelected = "";
		RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
		return;
	}

	CString s;
	m_Available.GetText(sel, s);
	m_LastSelected = s;
	RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);

}

void CRTPDlg::OnDblclkAvail()
{
	int sel = m_Available.GetCurSel();
	if (sel < 0) return;

	CString currentname;
	m_Available.GetText(sel, currentname);

	m_Available.DeleteString(sel);
	m_Used.AddString(currentname);
}

void CRTPDlg::OnSelchangeUsed()
{
	int sel = m_Used.GetCurSel();

	if (sel < 0) {
		m_LastSelected = "";
		RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);
		return;
	}

	CString s;
	m_Used.GetText(sel, s);
	m_LastSelected = s;
	RedrawWindow(NULL, NULL, RDW_INVALIDATE | RDW_UPDATENOW);

}

void CRTPDlg::OnDblclkUsed()
{
	int sel = m_Used.GetCurSel();
	if (sel < 0) return;

	CString currentname;
	m_Used.GetText(sel, currentname);

	m_Used.DeleteString(sel);
	m_Available.AddString(currentname);
}

void CRTPDlg::OnPaint()
{

	if (m_LastSelected == "") return;

	CPaintDC dc(this); // device context for painting

	int id = Map->GetUnitTypeID(m_LastSelected);

	PICDATA* p = &treeinfo[id].pic; //ovrlpics[m_currentOverlay][i];


	if (p->pic == NULL) {
		CString type;
		type = m_LastSelected;

		if (missingimages.find(type) == missingimages.end()) {
			theApp.m_loading->LoadUnitGraphic(type);
			Map->UpdateTreeInfo(&type);
			p = &treeinfo[id].pic;
		}
		if (p->pic == NULL) {
			missingimages[type] = TRUE;
		}
	}

	int curwidth = p->wMaxWidth;
	int curheight = p->wMaxHeight;

	BITMAPINFO biinfo;
	memset(&biinfo, 0, sizeof(BITMAPINFO));
	biinfo.bmiHeader.biBitCount = 24;
	biinfo.bmiHeader.biWidth = curwidth;
	biinfo.bmiHeader.biHeight = curheight;
	biinfo.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
	biinfo.bmiHeader.biClrUsed = 0;
	biinfo.bmiHeader.biPlanes = 1;
	biinfo.bmiHeader.biCompression = BI_RGB;
	biinfo.bmiHeader.biClrImportant = 0;


	int pitch = curwidth * 3;
	if (pitch == 0) return;

	if (pitch % sizeof(DWORD)) {
		pitch += sizeof(DWORD) - (curwidth * 3) % sizeof(DWORD);
	}

	BYTE* colors = new(BYTE[pitch * curheight]);
	memset(colors, 255, pitch * (curheight));

	RGBTRIPLE* pal = palIso;
#ifdef NOSURFACES
	if (p->pal == iPalTheater) pal = palTheater;
	if (p->pal == iPalUnit) pal = palUnit;
#endif

	int k, l;
	for (k = 0; k < curheight; k++) {
		for (l = 0; l < curwidth; l++) {
			if (((BYTE*)p->pic)[l + k * curwidth]) {
				memcpy(&colors[l * 3 + (curheight - k - 1) * pitch], &pal[((BYTE*)p->pic)[l + k * curwidth]], 3);
			}
		}
	}

	RECT pr;
	m_Preview.GetWindowRect(&pr);
	this->ScreenToClient(&pr);

	pr.top += 15;
	pr.bottom -= 5;
	pr.left += 5;
	pr.right -= 5;

	CBrush f;
	f.Attach(GetStockObject(WHITE_BRUSH));
	dc.FillRect(&pr, &f);
	f.Detach();

	auto clientDC = m_Preview.GetWindowDC();
	StretchDIBits(clientDC->GetSafeHdc(), 20, 30, curwidth, curheight,
		0, 0, curwidth, curheight, colors, &biinfo, DIB_RGB_COLORS, SRCCOPY);

	delete[] colors;

	// Kein Aufruf von CDialog::OnPaint() für Zeichnungsnachrichten
}