/*
   BobToolz plugin for GtkRadiant
   Copyright (C) 2001 Gordon Biggans

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

   This library 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
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with this library; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "StdAfx.h"

#include "funchandlers.h"

#include "IntersectDialog.h"
#include "PolygonDialog.h"
#include "StairDialog.h"
#include "DoorDialog.h"
#include "IntersectInfoDialog.h"
#include "BrushCheckDialog.h"
#include "AutoCaulkDialog.h"
#include "AutoCaulkStartDialog.h"
#include "TextureResetDialog.h"
#include "pathplotterdialog.h"

#include "DEntity.h"
#include "shapes.h"
#include "lists.h"
#include "misc.h"
#include "DShape.h"

// for autocaulk
std::list<Str> exclusionList;       // whole brush exclusion
std::list<Str> exclusionList_Face;  // single face exclusion

BOOL el1Loaded;
BOOL el2Loaded;

DBobView*   g_PathView = NULL;
// -------------

/************************
    Global Variables
************************/

CPolygonDialog polygonDlg;
CIntersectDialog intrDlg;
CStairDialog stairDlg;
CDoorDialog doorDlg;
CAutoCaulkStartDialog autocaulkDlg;
CTextureResetDialog texRstDlg;
CPathPlotterDialog ppDlg;

/************************
    --Main Functions--
************************/

void LoadLists(){
	char buffer[256];

	if ( !el1Loaded ) {
		el1Loaded = LoadExclusionList( GetFilename( buffer, "bt\\bt-el1.txt" ), &exclusionList );
	}
	if ( !el2Loaded ) {
		el2Loaded = LoadExclusionList( GetFilename( buffer, "bt\\bt-el2.txt" ), &exclusionList );
	}
}

void PolygonBuilder( vec3_t vMin, vec3_t vMax ){
	// ensure we have something selected
	if ( g_FuncTable.m_pfnSelectedBrushCount() != 1 ) {
		MessageBox( NULL, "Invalid number of brushes selected, chose 1 only", "Error", MB_OK );
		return;
	}

	// tell Radiant we want to access the selected brushes
	g_FuncTable.m_pfnAllocateSelectedBrushHandles();

	// get handle to size definition brush
	brush_t *brush = (brush_t*)g_FuncTable.m_pfnGetSelectedBrushHandle( 0 );
	// cant release until we delete the brush, if we do...

	// ask user for type, size, etc....
	if ( polygonDlg.DoModal() == IDOK ) {
		DShape poly;

		g_FuncTable.m_pfnDeleteBrushHandle( brush );

		if ( polygonDlg.m_bInverse ) {
			poly.BuildInversePrism( vMin, vMax, polygonDlg.m_nSideCount, polygonDlg.m_bAlignTop );
		}
		else
		{
			if ( polygonDlg.m_bBorder ) {
				poly.BuildBorderedPrism( vMin, vMax, polygonDlg.m_nSideCount, polygonDlg.m_nBorderSize, polygonDlg.m_bAlignTop );
			}
			else{
				poly.BuildRegularPrism( vMin, vMax, polygonDlg.m_nSideCount, polygonDlg.m_bAlignTop );
			}
		}

		poly.Commit();
	}


	g_FuncTable.m_pfnReleaseSelectedBrushHandles();
}


void IntersectionFinder(){
	if ( intrDlg.DoModal() == IDCANCEL ) {
		return;
	}

	if ( intrDlg.m_nBrushOptions == BRUSH_OPT_SELECTED ) {
		// ensure we have enough brushes selected
		if ( g_FuncTable.m_pfnSelectedBrushCount() < 2 ) {
			MessageBox( NULL, "Invalid number of brushes selected, choose at least 2", "Error", MB_OK );
			return;
		}
	}

	CIntersectInfoDialog*   intrInfoDlg = new CIntersectInfoDialog();
	intrInfoDlg->Create( IDD_INTERSECT_INFO_DIALOG );

	DEntity world;

	switch ( intrDlg.m_nBrushOptions )
	{
	case BRUSH_OPT_SELECTED:
	{
		world.LoadSelectedBrushes( &intrInfoDlg->m_prog1 );
		break;
	}
	case BRUSH_OPT_WHOLE_MAP:
	{
		world.LoadFromEntity( 0, &intrInfoDlg->m_prog1 );
		break;
	}
	}

	world.RemoveNonCheckBrushes( &exclusionList, intrDlg.m_bUseDetail );
	BOOL* pbSelectList;
	if ( intrDlg.m_bDuplicateOnly ) {
		pbSelectList = world.BuildDuplicateList();
	}
	else{
		pbSelectList = world.BuildIntersectList();
	}

	world.SelectBrushes( pbSelectList );

	intrInfoDlg->DestroyWindow();
	delete[] pbSelectList;
}

void StairBuilder( vec3_t vMin, vec3_t vMax ){
	// ensure we have something selected
	if ( g_FuncTable.m_pfnSelectedBrushCount() != 1 ) {
		MessageBox( NULL, "Invalid number of brushes selected, chose 1 only", "Error", MB_OK );
		return;
	}

	// tell Radiant we want to access the selected brushes
	g_FuncTable.m_pfnAllocateSelectedBrushHandles();

	// get handle to size definition brush
	brush_t *brush = (brush_t*)g_FuncTable.m_pfnGetSelectedBrushHandle( 0 );
	// cant release until we delete the brush, if we do...


	// ask user for type, size, etc....
	if ( stairDlg.DoModal() == IDOK ) {

		// calc brush size
		vec3_t size;
		_VectorSubtract( vMax, vMin, size );


		if ( ( (int)size[2] % stairDlg.m_nStairHeight ) != 0 ) {
			// stairs must fit evenly into brush
			MessageBox( NULL, "Invalid stair height\nHeight of block must be divisable by stair height", "Error", MB_OK );
		}
		else
		{

			// Remove Size Brush
			g_FuncTable.m_pfnDeleteBrushHandle( brush );


			// Get Step Count, Direction of Stairs, Stair Style
			int numSteps = (int)size[2] / stairDlg.m_nStairHeight;
			int direction = stairDlg.m_StairDir;
			int style = stairDlg.m_StairStyle;

			if ( stairDlg.m_StairStyle == STYLE_CORNER ) {
				BuildCornerStairs( vMin, vMax, numSteps, "textures/common/caulk", (LPCTSTR)stairDlg.m_riserTexture );
			}
			else
			{
				// Get Step Dimensions
				float stairHeight = (float)stairDlg.m_nStairHeight;
				float stairWidth;
				if ( ( direction == MOVE_EAST ) || ( direction == MOVE_WEST ) ) {
					stairWidth = ( size[0] ) / numSteps;
				}
				else{
					stairWidth = ( size[1] ) / numSteps;
				}


				// Build Base For Stair (bob's style)
				if ( style == STYLE_BOB ) {
					Build_Wedge( direction, vMin, vMax, TRUE );
				}


				// Set First Step Starting Position
				vMax[2] = vMin[2] + stairHeight;
				SetInitialStairPos( direction, vMin, vMax, stairWidth );


				// Build The Steps
				for ( int i = 0; i < numSteps; i++ )
				{
					if ( style == STYLE_BOB ) {
						Build_StairStep_Wedge( direction, vMin, vMax, "textures/common/caulk", (LPCTSTR)stairDlg.m_riserTexture, stairDlg.m_bDetail );
					}
					else if ( style == STYLE_ORIGINAL ) {
						Build_StairStep( vMin, vMax, "textures/common/caulk", (LPCTSTR)stairDlg.m_riserTexture, direction );
					}

					// get step into next position
					MoveBlock( direction, vMin, vMax, stairWidth );
					vMax[2] += stairHeight;
					if ( style == STYLE_BOB ) {
						vMin[2] += stairHeight; // wedge bottom must be raised
					}
				}
			}
		}
	}

	g_FuncTable.m_pfnReleaseSelectedBrushHandles();
}

void DoorBuilder( vec3_t vMin, vec3_t vMax ){
	// ensure we have something selected
	if ( g_FuncTable.m_pfnSelectedBrushCount() != 1 ) {
		MessageBox( NULL, "Invalid number of brushes selected, chose 1 only", "Error", MB_OK );
		return;
	}

	// tell Radiant we want to access the selected brushes
	g_FuncTable.m_pfnAllocateSelectedBrushHandles();

	// get handle to size definition brush
	brush_t *brush = (brush_t*)g_FuncTable.m_pfnGetSelectedBrushHandle( 0 );
	// cant release until we delete the brush, if we do...



	strcpy( doorDlg.m_fbTextureName.GetBuffer( 256 ), g_FuncTable.m_pfnGetCurrentTexture() );

	if ( doorDlg.DoModal() == IDOK ) {
		g_FuncTable.m_pfnDeleteBrushHandle( brush );

		BuildDoorsX2( vMin, vMax,
					  doorDlg.m_bSclMainHor, doorDlg.m_bSclMainVert,
					  doorDlg.m_bSclTrimHor, doorDlg.m_bSclTrimVert,
					  (LPCTSTR)doorDlg.m_fbTextureName,
					  (LPCTSTR)doorDlg.m_trimTextureName,
					  doorDlg.m_doorDirection );
	}

	g_FuncTable.m_pfnReleaseSelectedBrushHandles();
}

void FixBrushes(){
	DEntity world;

	CIntersectInfoDialog*   intrInfoDlg = new CIntersectInfoDialog();
	intrInfoDlg->Create( IDD_INTERSECT_INFO_DIALOG );

	world.LoadFromEntity( 0, &intrInfoDlg->m_prog1 );

	intrInfoDlg->DestroyWindow();

	CBrushCheckDialog*  chkDlg = new CBrushCheckDialog();
	chkDlg->Create( IDD_BRUSHCHECKER_DIALOG );

	int count = world.FixBrushes( TRUE, &chkDlg->m_prog1 );

	chkDlg->DestroyWindow();

	Sys_Printf( "%i invalid/duplicate planes removed\n", count );
}

void AutoCaulk(){
	UndoableCommand undo( "bobToolz.autoCaulk" );

	if ( !el1Loaded ) {
		autocaulkDlg.m_Warning1 = "WARNING: Brush exclusion list not found\n, ALL BRUSHES WILL BE USED";
	}

	if ( autocaulkDlg.DoModal() == IDCANCEL ) {
		return;
	}

	if ( autocaulkDlg.m_nMode == MODE_AC_BUILD_MINI_PRT ) {
		BuildMiniPrt( &exclusionList );
		return;
	}

	CAutoCaulkDialog* acDlg = new CAutoCaulkDialog;
	acDlg->Create( IDD_AUTOCAULK_DIALOG );

	char filename[1204];

	if ( autocaulkDlg.m_nMode == MODE_AC_NORMAL ) {
		char* rad_filename = g_BSPTable.m_pfnGetMapName();
		if ( !rad_filename ) {
			MessageBox( NULL, "An Error Occurred While Trying To Get The Map Filename", "Error", MB_OK );
			acDlg->DestroyWindow();
			return;
		}

		strcpy( filename, rad_filename );

		char* ext = strrchr( filename, '.' ) + 1;
		strcpy( ext, "prt" ); // rename the extension
	}
	else
	{
		IEpair* pEp = g_EpairTable.m_pfnIEpairForProjectKeys();
		char *pn = pEp->ValueForKey( "mapspath" );
		pEp->DecRef();

		strcpy( filename, pn );
		strcat( filename, "/ac_prt.prt" );
	}

	DEntity portals;
	if ( !portals.LoadFromPrt( filename, &acDlg->m_prog1 ) ) {
		MessageBox( NULL, "Failed To Load Portal File", "Error", MB_OK );
		acDlg->DestroyWindow();
		return;
	}
	// load portal file

	CIntersectInfoDialog*   intrInfoDlg = new CIntersectInfoDialog();
	intrInfoDlg->Create( IDD_INTERSECT_INFO_DIALOG );

	DEntity world;

	world.LoadFromEntity( 0, &intrInfoDlg->m_prog1 );
	intrInfoDlg->DestroyWindow();

	if ( autocaulkDlg.m_nMode == MODE_AC_NORMAL ) {
		world.RemoveNonCheckBrushes( &exclusionList, FALSE );
	}
	else{
		world.RemoveNonCheckBrushes( &exclusionList, TRUE );
	}

	world.ResetChecks( &exclusionList_Face );

	int caulkedCount = 0;
	int killCnt = world.AutoCaulk( &portals, autocaulkDlg.m_bAllowDestruction, &caulkedCount, &acDlg->m_prog2 );

	if ( autocaulkDlg.m_bAllowDestruction ) {
		Sys_Printf( "%i unrequired brush(es) killed\n", killCnt );
	}
	Sys_Printf( "%i face(s) caulked\n", caulkedCount );

	acDlg->DestroyWindow();
}

void ResetTextures(){
	texRstDlg.m_TextureName = GetCurrentTexture();
	texRstDlg.m_NewTextureName = GetCurrentTexture();

	if ( texRstDlg.DoModal() == IDCANCEL ) {
		return;
	}

	float fScale[2];
	float fShift[2];
	fScale[1] = texRstDlg.m_fScaleVertical;
	fScale[0] = texRstDlg.m_fScaleHorizontal;

	fShift[1] = (float)texRstDlg.m_nShiftVertical;
	fShift[0] = (float)texRstDlg.m_nShiftHorizontal;

	DEntity world;
	world.LoadFromEntity( 0, NULL );

	if ( texRstDlg.m_bAllTextures ) {
		world.ResetTextures( NULL, fScale, fShift, texRstDlg.m_nRotation, texRstDlg.m_NewTextureName, texRstDlg.m_bOnlyTexture );
	}
	else{
		world.ResetTextures( texRstDlg.m_TextureName, fScale, fShift, texRstDlg.m_nRotation, texRstDlg.m_NewTextureName, texRstDlg.m_bOnlyTexture );
	}
}

void PathPlotter(){
	int ret = ppDlg.DoModal();
	if ( ret == IDCANCEL ) {
		return;
	}
	if ( ret == IDNO ) {
		if ( g_PathView ) {
			delete g_PathView;
		}

		return;
	}

	if ( g_FuncTable.m_pfnSelectedBrushCount() != 1 ) {
		MessageBox( NULL, "Invalid number of brushes selected, chose 1 only", "Error", MB_OK );
		return;
	}

	// tell Radiant we want to access the selected brushes
	g_FuncTable.m_pfnAllocateSelectedBrushHandles();

	// get handle to size definition brush
	brush_t *brush = (brush_t*)g_FuncTable.m_pfnGetSelectedBrushHandle( 0 );
	// cant release until we delete the brush, if we do...

	DEntity world;
	world.LoadEPairList( *g_FuncTable.m_pfnGetEntityKeyValList( brush->owner ) );

	DEPair* trigger_ep = world.FindEPairByKey( "targetname" );

	if ( trigger_ep ) {
		if ( !strcmp( world.m_Classname, "trigger_push" ) ) {
			DEPair* target_ep = world.FindEPairByKey( "target" );
			if ( target_ep ) {
				entity_s* entTarget = FindEntityFromTargetname( target_ep->value );
				if ( entTarget ) {
					if ( g_PathView ) {
						delete g_PathView;
					}
					g_PathView = new DBobView;

					g_PathView->Begin( trigger_ep->value, target_ep->value, ppDlg.m_fMultiplier, ppDlg.m_nPoints, ppDlg.m_fGravity, ppDlg.m_bNoUpdate, ppDlg.m_bShowExtra );
				}
				else{
					MessageBox( NULL, "trigger_push target could not be found.", "Error", MB_OK );
				}
			}
			else{
				MessageBox( NULL, "trigger_push has no target.", "Error", MB_OK );
			}
		}
		else{
			MessageBox( NULL, "You must select a 'trigger_push' entity.", "Error", MB_OK );
		}
	}
	else{
		MessageBox( NULL, "Entity must have a targetname", "Error", MB_OK );
	}

	g_FuncTable.m_pfnReleaseSelectedBrushHandles();
}

void PitBuilder( vec3_t vMin, vec3_t vMax ){
	// ensure we have something selected
	if ( g_FuncTable.m_pfnSelectedBrushCount() != 1 ) {
		MessageBox( NULL, "Invalid number of brushes selected, chose 1 only", "Error", MB_OK );
		return;
	}

	// tell Radiant we want to access the selected brushes
	g_FuncTable.m_pfnAllocateSelectedBrushHandles();

	// get handle to size definition brush
	brush_t *brush = (brush_t*)g_FuncTable.m_pfnGetSelectedBrushHandle( 0 );
	// cant release until we delete the brush, if we do...

	DShape pit;

	if ( pit.BuildPit( vMin, vMax ) ) {
		pit.Commit();

		g_FuncTable.m_pfnDeleteBrushHandle( brush );
	}
	else{
		MessageBox( NULL, "Failed To Make Pit\nTry Making The Brush Bigger", "Error", MB_OK );
	}

	g_FuncTable.m_pfnReleaseSelectedBrushHandles();
}