#include <sys/time.h>
#include <time.h>

#include "QF/quakeio.h"
#include "QF/script.h"
#include "QF/sys.h"

#include "Map.h"
#include "Entity.h"
#include "TexturePalette.h"
#include "SetBrush.h"
#include "XYView.h"
#include "CameraView.h"
#include "QuakeEd.h"
#include "Things.h"
#include "InspectorControl.h"
#include "Project.h"

#define THING Map
#include "THING+NSArray.m"

id  map_i;

@implementation Map
/*
===============================================================================

FILE METHODS

===============================================================================
*/
- (id) init
{
	self = [super init];
	array = [[NSMutableArray alloc] init];
	map_i = self;
	minz = 0;
	maxz = 80;

	oldselection = [[NSMutableArray alloc] init];

	return self;
}

- (id) saveSelected
{
	int     i, c;
	id      o, w;

	[oldselection removeAllObjects];
	w = [self objectAtIndex: 0];
	c = [w count];
	sb_newowner = oldselection;
	for (i = 0; i < c; i++) {
		o = [w objectAtIndex: 0];
		if ([o selected])
			[o moveToEntity];
		else
			[w removeObjectAtIndex: 0];
	}

	c = [self count];
	for (i = 0; i < c; i++) {
		o = [self objectAtIndex: 0];
		[self removeObjectAtIndex: 0];
		[o removeAllObjects];
	}

	return self;
}

- (id) addSelected
{
	int     i, c;
	id      n, w;

	c = [oldselection count];
	w = [self objectAtIndex: 0];    // world object

	sb_newowner = w;
	for (i = 0; i < c; i++) {
		n = [oldselection objectAtIndex: i];
		[n moveToEntity];
		i--;
		c--;
	}
	[oldselection removeAllObjects];

	return self;
}

- (id) newMap
{
	id  ent;

	[self saveSelected];
	ent = [[Entity alloc] initClass: "worldspawn"];
	[self addObject: ent];
	currentEntity = NULL;
	[self setCurrentEntity: ent];
	[self addSelected];

	return self;
}

- (id) currentEntity
{
	return currentEntity;
}

- (id) setCurrentEntity: ent
{
	id  old;

	old = currentEntity;
	currentEntity = ent;
	if (old != ent) {
		[things_i newCurrentEntity];    // update inspector
		[inspcontrol_i setCurrentInspector: i_things];
	}

	return self;
}

- (float) currentMinZ
{
	float  grid;

	grid = [xyview_i gridsize];
	minz = grid * rint (minz / grid);

	return minz;
}

- (id) setCurrentMinZ: (float)m
{
	if (m > -2048)
		minz = m;

	return self;
}

- (float) currentMaxZ
{
	float  grid;

	[self currentMinZ];

	// grid align
	grid = [xyview_i gridsize];
	maxz = grid * rint (maxz / grid);

	if (maxz <= minz)
		maxz = minz + grid;

	return maxz;
}

- (id) setCurrentMaxZ: (float)m
{
	if (m < 2048)
		maxz = m;

	return self;
}

- (void) removeObject: o
{
	[super removeObject: o];

	if (o == currentEntity)         // select the world
		[self setCurrentEntity: [self objectAtIndex: 0]];

	return;
}

#define FN_DEVLOG "/qcache/devlog"
- (id) writeStats
{
	FILE                *f;
	extern int          c_updateall;
	struct timeval      tp;
	struct timezone     tzp;

	gettimeofday (&tp, &tzp);

	f = fopen (FN_DEVLOG, "a");
	fprintf (f, "%i %i\n", (int) tp.tv_sec, c_updateall);
	c_updateall = 0;
	fclose (f);
	return self;
}

- (int) numSelected
{
	int     i, c;
	int     num;

	num = 0;
	c = [currentEntity count];
	for (i = 0; i < c; i++) {
		if ([[currentEntity objectAtIndex: i] selected])
			num++;
	}
	return num;
}

- (id) selectedBrush
{
	int     i, c;
	int     num;

	num = 0;
	c = [currentEntity count];
	for (i = 0; i < c; i++) {
		if ([[currentEntity objectAtIndex: i] selected])
			return [currentEntity objectAtIndex: i];
	}
	return nil;
}

/*
=================
readMapFile
=================
*/
- (id) readMapFile: (NSString *)fname
{
	char        *dat;
	const char  *wad, *cl;
	id          new;
	id          ent;
	int         i, c;
	vec3_t      org;
	float       angle;
	script_t    *script;
	size_t      size;
	NSFileManager *fm = [NSFileManager defaultManager];
	NSData      *contents;

	[self saveSelected];

	NSLog (@"loading %@\n", fname);

	contents = [fm contentsAtPath: fname];
	if (!contents)
		return self;
	size = [contents length];
	dat = malloc (size + 1);
	[contents getBytes: dat];
	dat[size] = 0;

	script = Script_New ();
	Script_Start (script, [fname cString], dat);

	do {
		new = [[Entity alloc] initFromScript: script];
		if (!new)
			break;
		[self addObject: new];
	} while (1);

	free (dat);

	[self addSelected];

	ent = [self objectAtIndex: 0];
	// load the apropriate texture wad
	wad = [ent valueForQKey: "wad"];
	if (wad && wad[0]) {
		// remove old style fullpaths
		if (wad[0] == '/')
			[ent removeKeyPair: "wad"];
		else if (strcmp ([texturepalette_i currentWad], wad))
			[project_i setTextureWad: wad];
	}

	[self setCurrentEntity: ent];

	// center the camera and XY view on the playerstart
	c = [self count];
	for (i = 1; i < c; i++) {
		ent = [self objectAtIndex: i];
		cl = [ent valueForQKey: "classname"];
		if (cl && !strcasecmp (cl, "info_player_start")) {
			angle = atof ([ent valueForQKey: "angle"]);
			angle = angle / 180 * M_PI;
			[ent getVector: org forKey: "origin"];
			[cameraview_i setOrigin: org angle: angle];
			[xyview_i centerOn: org];
			break;
		}
	}
	NSLog (@"%@ loaded\n", fname);

	return self;
}

/*
=================
writeMapFile
=================
*/
- (id) writeMapFile: (const char *)fname
   useRegion: (BOOL)reg
{
	FILE            *f;
	unsigned int    i;

	Sys_Printf ("writeMapFile: %s\n", fname);

	f = fopen (fname, "w");
	if (!f)
		Sys_Error ("couldn't write %s", fname);
	for (i = 0; i < [self count]; i++)
		[[self objectAtIndex: i] writeToFILE: f region: reg];

	fclose (f);

	return self;
}

/*
==============================================================================

DRAWING

==============================================================================
*/

- (void) ZDrawSelf
{
	int  i, count;

	count = [self count];

	for (i = 0; i < count; i++)
		[[self objectAtIndex: i] ZDrawSelf];
}

- (void) RenderSelf: (void (*)(face_t *))callback
{
	int  i, count;

	count = [self count];

	for (i = 0; i < count; i++)
		[[self objectAtIndex: i] RenderSelf: callback];
}

// ============================================================================

/*
===================
entityConnect

A command-shift-click on an entity while an entity is selected will
make a target connection from the original entity.
===================
*/
- (id) entityConnect: (vec3_t)p1
   : (vec3_t)p2
{
	id  oldent, ent;

	oldent = [self currentEntity];
	if (oldent == [self objectAtIndex: 0]) {
		Sys_Printf ("Must have a non-world entity selected to connect\n");
		return self;
	}

	[self selectRay: p1: p2: YES];
	ent = [self currentEntity];
	if (ent == oldent) {
		Sys_Printf ("Must click on a different entity to connect\n");
		return self;
	}

	if (ent == [self objectAtIndex: 0]) {
		Sys_Printf ("Must click on a non-world entity to connect\n");
		return self;
	}

	[oldent setKey: "target" toValue: [ent targetname]];
	[quakeed_i updateAll];

	return self;
}

/*
=================
selectRay

If ef is true, any entity brush along the ray will be selected in preference
to intervening world brushes
=================
*/
- (id) selectRay: (vec3_t)p1
   : (vec3_t)p2
   : (BOOL)ef
{
	int             i, j, c, c2;
	id              ent, bestent;
	id              brush, bestbrush;
	int             face, bestface;
	float           time, besttime;
	texturedef_t    *td;

	bestent = nil;
	bestface = -1;
	bestbrush = nil;
	besttime = 99999;

	c = [self count];
	for (i = c - 1; i >= 0; i--) {
		ent = [self objectAtIndex: i];
		c2 = [ent count];
		for (j = 0; j < c2; j++) {
			brush = [ent objectAtIndex: j];
			[brush hitByRay: p1 : p2 : &time : &face];
			if (time < 0 || time > besttime)
				continue;
			bestent = ent;
			besttime = time;
			bestbrush = brush;
			bestface = face;
		}
		if (i == 1 && ef && bestbrush)  // found an entity, don't check the
			                            // world
			break;
	}

	if (besttime == 99999) {
		Sys_Printf ("trace missed\n");
		return self;
	}

	if ([bestbrush regioned]) {
		Sys_Printf ("WARNING: clicked on regioned brush\n");
		return self;
	}

	if (bestent != currentEntity) {
		[self makeSelectedPerform: @selector (deselect)];
		[self setCurrentEntity: bestent];
	}

	[quakeed_i disableFlushWindow];
	if (![bestbrush selected]) {
		// don't grab texture if others are selected
		if ([map_i numSelected] == 0) {
			td = [bestbrush texturedefForFace: bestface];
			[texturepalette_i setTextureDef: td];
		}

		[bestbrush setSelected: YES];
		Sys_Printf ("selected entity %i brush %i face %i\n",
		            (int) [self indexOfObject: bestent],
		            (int) [bestent indexOfObject: bestbrush], bestface);
	} else {
		[bestbrush setSelected: NO];
		Sys_Printf ("deselected entity %i brush %i face %i\n",
		            (int) [self indexOfObject: bestent],
		            (int) [bestent indexOfObject: bestbrush], bestface);
	}

	[quakeed_i enableFlushWindow];
	[quakeed_i updateAll];

	return self;
}

/*
=================
grabRay

checks only the selected brushes
Returns the brush hit, or nil if missed.
=================
*/
- (id) grabRay: (vec3_t)p1
   : (vec3_t)p2
{
	int     i, j, c, c2;
	id      ent;
	id      brush, bestbrush;
	int     face;
	float   time, besttime;

	bestbrush = nil;
	besttime = 99999;

	c = [self count];
	for (i = 0; i < c; i++) {
		ent = [self objectAtIndex: i];
		c2 = [ent count];
		for (j = 0; j < c2; j++) {
			brush = [ent objectAtIndex: j];
			if (![brush selected])
				continue;
			[brush hitByRay: p1 : p2 : &time : &face];
			if (time < 0 || time > besttime)
				continue;
			besttime = time;
			bestbrush = brush;
		}
	}

	if (besttime == 99999)
		return nil;
	return bestbrush;
}

/*
=================
getTextureRay
=================
*/
- (id) getTextureRay: (vec3_t)p1
   : (vec3_t)p2
{
	int             i, j, c, c2;
	id              ent, bestent;
	id              brush, bestbrush;
	int             face, bestface;
	float           time, besttime;
	texturedef_t    *td;
	vec3_t          mins, maxs;

	bestbrush = nil;
	bestent = nil;
	besttime = 99999;
	bestface = -1;
	c = [self count];
	for (i = 0; i < c; i++) {
		ent = [self objectAtIndex: i];
		c2 = [ent count];
		for (j = 0; j < c2; j++) {
			brush = [ent objectAtIndex: j];
			[brush hitByRay: p1 : p2 : &time : &face];
			if (time < 0 || time > besttime)
				continue;
			bestent = ent;
			bestface = face;
			besttime = time;
			bestbrush = brush;
		}
	}

	if (besttime == 99999)
		return nil;
	if (![bestent modifiable]) {
		Sys_Printf ("can't modify spawned entities\n");
		return self;
	}

	td = [bestbrush texturedefForFace: bestface];
	[texturepalette_i setTextureDef: td];

	Sys_Printf ("grabbed texturedef and sizes\n");

	[bestbrush getMins: mins maxs: maxs];

	minz = mins[2];
	maxz = maxs[2];

	return bestbrush;
}

/*
=================
setTextureRay
=================
*/
- (id) setTextureRay: (vec3_t)p1
   : (vec3_t)p2
   : (BOOL)allsides;
{
	int             i, j, c, c2;
	id              ent, bestent;
	id              brush, bestbrush;
	int             face, bestface;
	float           time, besttime;
	texturedef_t    td;

	bestent = nil;
	bestface = -1;
	bestbrush = nil;
	besttime = 99999;

	c = [self count];
	for (i = 0; i < c; i++) {
		ent = [self objectAtIndex: i];
		c2 = [ent count];
		for (j = 0; j < c2; j++) {
			brush = [ent objectAtIndex: j];
			[brush hitByRay: p1 : p2 : &time : &face];
			if (time < 0 || time > besttime)
				continue;
			bestent = ent;
			besttime = time;
			bestbrush = brush;
			bestface = face;
		}
	}

	if (besttime == 99999) {
		Sys_Printf ("trace missed\n");
		return self;
	}

	if (![bestent modifiable]) {
		Sys_Printf ("can't modify spawned entities\n");
		return self;
	}

	if ([bestbrush regioned]) {
		Sys_Printf ("WANRING: clicked on regioned brush\n");
		return self;
	}

	[texturepalette_i getTextureDef: &td];

	[quakeed_i disableFlushWindow];
	if (allsides) {
		[bestbrush setTexturedef: &td];
		Sys_Printf ("textured entity %i brush %i\n",
		            (int) [self indexOfObject: bestent],
		            (int) [bestent indexOfObject: bestbrush]);
	} else {
		[bestbrush setTexturedef: &td forFace: bestface];
		Sys_Printf ("deselected entity %i brush %i face %i\n",
		            (int) [self indexOfObject: bestent],
		            (int) [bestent indexOfObject: bestbrush],
		            bestface);
	}
	[quakeed_i enableFlushWindow];

	[quakeed_i updateAll];

	return self;
}

/*
==============================================================================

OPERATIONS ON SELECTIONS

==============================================================================
*/

- (id) makeSelectedPerform: (SEL)sel
{
	int     i, j, c, c2;
	id      ent, brush;
	int     total;

	total = 0;
	c = [self count];
	for (i = c - 1; i >= 0; i--) {
		ent = [self objectAtIndex: i];
		c2 = [ent count];
		for (j = c2 - 1; j >= 0; j--) {
			brush = [ent objectAtIndex: j];
			if (![brush selected])
				continue;
			if ([brush regioned])
				continue;
			total++;
			[brush performSelector: sel];
		}
	}

//  if (!total)
//      Sys_Printf ("nothing selected\n");

	return self;
}

- (id) makeUnselectedPerform: (SEL)sel
{
	int     i, j, c, c2;
	id      ent, brush;

	c = [self count];
	for (i = c - 1; i >= 0; i--) {
		ent = [self objectAtIndex: i];
		c2 = [ent count];
		for (j = c2 - 1; j >= 0; j--) {
			brush = [ent objectAtIndex: j];
			if ([brush selected])
				continue;
			if ([brush regioned])
				continue;
			[brush performSelector: sel];
		}
	}

	return self;
}

- (id) makeAllPerform: (SEL)sel
{
	int     i, j, c, c2;
	id      ent, brush;

	c = [self count];
	for (i = c - 1; i >= 0; i--) {
		ent = [self objectAtIndex: i];
		c2 = [ent count];
		for (j = c2 - 1; j >= 0; j--) {
			brush = [ent objectAtIndex: j];
			if ([brush regioned])
				continue;
			[brush performSelector: sel];
		}
	}

	return self;
}

- (id) makeGlobalPerform: (SEL)sel  // in and out of region
{
	int     i, j, c, c2;
	id      ent, brush;

	c = [self count];
	for (i = c - 1; i >= 0; i--) {
		ent = [self objectAtIndex: i];
		c2 = [ent count];
		for (j = c2 - 1; j >= 0; j--) {
			brush = [ent objectAtIndex: j];
			[brush performSelector: sel];
		}
	}

	return self;
}

void
sel_identity (void)
{
	sel_x[0] = 1;
	sel_x[1] = 0;
	sel_x[2] = 0;
	sel_y[0] = 0;
	sel_y[1] = 1;
	sel_y[2] = 0;
	sel_z[0] = 0;
	sel_z[1] = 0;
	sel_z[2] = 1;
}

- (id) transformSelection
{
	if (![currentEntity modifiable]) {
		Sys_Printf ("can't modify spawned entities\n");
		return self;
	}
	// find an origin to apply the transformation to
	sb_mins[0] = sb_mins[1] = sb_mins[2] = 99999;
	sb_maxs[0] = sb_maxs[1] = sb_maxs[2] = -99999;
	[self makeSelectedPerform: @selector (addToBBox)];
	sel_org[0] = [xyview_i snapToGrid: (sb_mins[0] + sb_maxs[0]) / 2];
	sel_org[1] = [xyview_i snapToGrid: (sb_mins[1] + sb_maxs[1]) / 2];
	sel_org[2] = [xyview_i snapToGrid: (sb_mins[2] + sb_maxs[2]) / 2];

	// do it!
	[self makeSelectedPerform: @selector (transform)];

	[quakeed_i updateAll];
	return self;
}

void
swapvectors (vec3_t a, vec3_t b)
{
	vec3_t  temp;

	VectorCopy (a, temp);
	VectorCopy (b, a);
	VectorSubtract (vec3_origin, temp, b);
}

/*
===============================================================================

UI operations

===============================================================================
*/

- (id) rotate_x: sender
{
	sel_identity ();
	swapvectors (sel_y, sel_z);
	[self transformSelection];
	return self;
}

- (id) rotate_y: sender
{
	sel_identity ();
	swapvectors (sel_x, sel_z);
	[self transformSelection];
	return self;
}

- (id) rotate_z: sender
{
	sel_identity ();
	swapvectors (sel_x, sel_y);
	[self transformSelection];
	return self;
}

- (id) flip_x: sender
{
	sel_identity ();
	sel_x[0] = -1;
	[self transformSelection];
	[map_i makeSelectedPerform: @selector (flipNormals)];
	return self;
}

- (id) flip_y: sender
{
	sel_identity ();
	sel_y[1] = -1;
	[self transformSelection];
	[map_i makeSelectedPerform: @selector (flipNormals)];
	return self;
}

- (id) flip_z: sender
{
	sel_identity ();
	sel_z[2] = -1;
	[self transformSelection];
	[map_i makeSelectedPerform: @selector (flipNormals)];
	return self;
}

- (id) cloneSelection: sender
{
	int     i, j, c, originalElements;
	id      o, b;
	id      new;

	sb_translate[0] = sb_translate[1] = [xyview_i gridsize];
	sb_translate[2] = 0;

	// copy individual brushes in the world entity
	o = [self objectAtIndex: 0];
	c = [o count];
	for (i = 0; i < c; i++) {
		b = [o objectAtIndex: i];
		if (![b selected])
			continue;
		// copy the brush, then translate the original
		new = [b copy];
		[new setSelected: YES];
		[new translate];
		[b setSelected: NO];
		[o addObject: new];
	}

	// copy entire entities otherwise
	originalElements = [self count];    // don't copy the new ones
	for (i = 1; i < originalElements; i++) {
		o = [self objectAtIndex: i];
		if (![[o objectAtIndex: 0] selected])
			continue;

		new = [o copy];
		[self addObject: new];

		c = [o count];
		for (j = 0; j < c; j++)
			[[o objectAtIndex: j] setSelected: NO];

		c = [new count];
		for (j = 0; j < c; j++) {
			b = [new objectAtIndex: j];
			[b translate];
			[b setSelected: YES];
		}
	}

	[quakeed_i updateAll];

	return self;
}

- (id) selectCompleteEntity: sender
{
	id      o;
	int     i, c;

	o = [self selectedBrush];
	if (!o) {
		Sys_Printf ("nothing selected\n");
		return self;
	}
	o = [o parent];
	c = [o count];
	for (i = 0; i < c; i++)
		[[o objectAtIndex: i] setSelected: YES];

	Sys_Printf ("%i brushes selected\n", c);

	[quakeed_i updateAll];

	return self;
}

- (id) makeEntity: sender
{
	if (currentEntity != [self objectAtIndex: 0]) {
		Sys_Printf ("ERROR: can't makeEntity inside an entity\n");
		NSBeep ();
		return self;
	}

	if ([self numSelected] == 0) {
		Sys_Printf ("ERROR: must have a seed brush to make an entity\n");
		NSBeep ();
		return self;
	}

	sb_newowner = [[Entity alloc] initClass: [things_i spawnName]];

	if ([sb_newowner modifiable]) {
		[self makeSelectedPerform: @selector (moveToEntity)];
	} else {  // throw out seed brush and select entity fixed brush
		[self makeSelectedPerform: @selector (remove)];
		[[sb_newowner objectAtIndex: 0] setSelected: YES];
	}

	[self addObject: sb_newowner];
	[self setCurrentEntity: sb_newowner];

	[quakeed_i updateAll];

	return self;
}

- (id) selbox: (SEL)selector
{
	id  b;

	if ([self numSelected] != 1) {
		Sys_Printf ("must have a single brush selected\n");
		return self;
	}

	b = [self selectedBrush];
	[b getMins: select_min maxs: select_max];
	[b remove];

	[self makeUnselectedPerform: selector];

	Sys_Printf ("identified contents\n");
	[quakeed_i updateAll];

	return self;
}

- (id) selectCompletelyInside: sender
{
	return [self selbox: @selector (selectComplete)];
}

- (id) selectPartiallyInside: sender
{
	return [self selbox: @selector (selectPartial)];
}

- (id) tallBrush: sender
{
	id              b;
	vec3_t          mins, maxs;
	texturedef_t    td;

	if ([self numSelected] != 1) {
		Sys_Printf ("must have a single brush selected\n");
		return self;
	}

	b = [self selectedBrush];
	td = *[b texturedef];
	[b getMins: mins maxs: maxs];
	[b remove];

	mins[2] = -2048;
	maxs[2] = 2048;

	b = [[SetBrush alloc]  initOwner: [map_i objectAtIndex: 0]
	                            mins: mins
	                            maxs: maxs
	                         texture: &td];
	[[map_i objectAtIndex: 0] addObject: b];
	[b setSelected: YES];
	[quakeed_i updateAll];

	return self;
}

- (id) shortBrush: sender
{
	id              b;
	vec3_t          mins, maxs;
	texturedef_t    td;

	if ([self numSelected] != 1) {
		Sys_Printf ("must have a single brush selected\n");
		return self;
	}

	b = [self selectedBrush];
	td = *[b texturedef];
	[b getMins: mins maxs: maxs];
	[b remove];

	mins[2] = 0;
	maxs[2] = 16;

	b = [[SetBrush alloc]  initOwner: [map_i objectAtIndex: 0]
	                            mins: mins
	                            maxs: maxs
	                         texture: &td];
	[[map_i objectAtIndex: 0] addObject: b];
	[b setSelected: YES];
	[quakeed_i updateAll];

	return self;
}

/*
==================
subtractSelection
==================
*/
- (id) subtractSelection: semder
{
	int     i, j, c, c2;
	id      o, o2;
	id      sellist, sourcelist;

	Sys_Printf ("performing brush subtraction...\n");

	sourcelist = [[NSMutableArray alloc] init];
	sellist = [[NSMutableArray alloc] init];
	carve_in = [[NSMutableArray alloc] init];
	carve_out = [[NSMutableArray alloc] init];

	c = [currentEntity count];
	for (i = 0; i < c; i++) {
		o = [currentEntity objectAtIndex: i];
		if ([o selected])
			[sellist addObject: o];
		else
			[sourcelist addObject: o];
	}

	c = [sellist count];
	for (i = 0; i < c; i++) {
		o = [sellist objectAtIndex: i];
		[o setCarveVars];

		c2 = [sourcelist count];
		for (j = 0; j < c2; j++) {
			o2 = [sourcelist objectAtIndex: j];
			[o2 carve];
			[carve_in removeAllObjects];
		}

		[sourcelist release];  // the individual have been moved/freed
		sourcelist = carve_out;
		carve_out = [[NSMutableArray alloc] init];
	}

	// add the selection back to the remnants
	[currentEntity removeAllObjects];
	[currentEntity addObjectsFromArray: sourcelist];
	[currentEntity addObjectsFromArray: sellist];

	[sourcelist release];
	[sellist release];
	[carve_in release];
	[carve_out release];

	if (![currentEntity count]) {
		o = currentEntity;
		[self removeObject: o];
	}

	Sys_Printf ("subtracted selection\n");
	[quakeed_i updateAll];

	return self;
}

@end