0ad/binaries/data/mods/public/gui/session_new/selection.js

function _setHighlight(ents, alpha)
{
	if (ents.length)
		Engine.GuiInterfaceCall("SetSelectionHighlight", { "entities":ents, "alpha":alpha });
}

function _setMotionOverlay(ents, enabled)
{
	if (ents.length)
		Engine.GuiInterfaceCall("SetMotionDebugOverlay", { "entities":ents, "enabled":enabled });
}

function _playSound(ent)
{
	Engine.GuiInterfaceCall("PlaySound", { "name":"select", "entity":ent });
}

//-------------------------------- -------------------------------- -------------------------------- 
// EntityGroup class for managing grouped entities
//-------------------------------- -------------------------------- -------------------------------- 
function EntityGroup()
{
	this.groupCounts = {};
	this.groupEntIDs = {};
}

EntityGroup.prototype.create = function(entIDs)
{
	this.groupCounts = {};
	var rankedTemplates = {};
	
	for each (var entID in entIDs)
	{
		var entState = GetEntityState(entID);
		var rank = entState.identity.rank;
		var templateName = entState.template;
		var template = GetTemplateData(templateName);
		var unitName = template.name.specific || template.name.generic || "???";

		if (rank)
		{
			if (rankedTemplates[unitName])
			{
				this.groupCounts[rankedTemplates[unitName]] += 1;
				this.groupEntIDs[unitName].push(entID);
			}
			else
			{
				rankedTemplates[unitName] = templateName;
				this.groupCounts[rankedTemplates[unitName]] = 1;
				this.groupEntIDs[unitName] = [entID];
			}
		}
		else
		{
			if (this.groupCounts[templateName])
			{
				this.groupCounts[templateName] += 1;
				this.groupEntIDs[unitName].push(entID);
			}
			else
			{
				this.groupCounts[templateName] = 1;
				this.groupEntIDs[unitName] = [entID];
			}
		}
	}
};

EntityGroup.prototype.remove = function(templateName)
{
	delete this.groupCounts[templateName];
};

EntityGroup.prototype.getCount = function(templateName)
{
	if (this.groupCounts[templateName])
		return this.groupCounts[templateName];
	else
		return 0;
};

EntityGroup.prototype.getTemplateNames = function()
{
	var templateNames = [];
	for (var templateName in this.groupCounts)
		templateNames.push(templateName);
	return templateNames;
};

EntityGroup.prototype.getEntsByUnitName = function(unitName)
{
	return this.groupEntIDs[unitName];
};


EntityGroup.prototype.getEntsByUnitNameInverse = function(unitName)
{
	var ents = [];

	for (var name in this.groupEntIDs)
		if (name != unitName)
			ents = ents.concat(this.groupEntIDs[name]);	

	return ents;
};

//-------------------------------- -------------------------------- -------------------------------- 
// EntitySelection class for managing the entity selection list and the primary selection
//-------------------------------- -------------------------------- -------------------------------- 
function EntitySelection()
{
	// Private properties:
	//--------------------------------
	this.selected = {}; // { id:id, id:id, ... } for each selected entity ID 'id'

	// { id:id, ... } for mouseover-highlighted entity IDs in these, the key is a string and the value is an int; 
	//	we want to use the int form wherever possible since it's more efficient to send to the simulation code)
	this.highlighted = {}; 

	this.motionDebugOverlay = false;

	// Public properties:
	//--------------------------------
	this.dirty = false; // set whenever the selection has changed
	this.groups = new EntityGroup();
}

// Deselect everything but entities of the chosen type if the modifier is true
// otherwise deselect just the chosen entity
EntitySelection.prototype.makePrimarySelection = function(primaryTemplateName, modifierKey)
{
	var selection = this.toList();
	var entID;
	
	// Find an entID of a unit of the same type
	for (var i = 0; i < selection.length; i++)
	{
		var entState = GetEntityState(selection[i]);
		if (!entState)
			continue;
		if (entState.template == primaryTemplateName)
			entID = selection[i];
	}
	
	var primaryEntState = GetEntityState(entID);
	if (!primaryEntState)
		return;
	var primaryTemplate = GetTemplateData(primaryTemplateName);
	var primaryUnitName = primaryTemplate.name.specific || primaryTemplate.name.generic || "???";

	var ents = [];
	if (modifierKey)
		ents = this.groups.getEntsByUnitNameInverse(primaryUnitName);
	else
		ents = this.groups.getEntsByUnitName(primaryUnitName);

	this.reset();
	this.addList(ents);
	this.CreateSelectionGroups();
}

// Get a list of the template names
EntitySelection.prototype.getTemplateNames = function()
{
	var templateNames = [];
	var ents = this.toList();
	
	for each (var ent in ents)
	{
		var entState = GetEntityState(ent);
		if (entState)
			templateNames.push(entState.template);
	}
	return templateNames;
}

// Make selection groups
EntitySelection.prototype.CreateSelectionGroups = function()
{
	this.groups.create(this.toList());
};

// Update the selection to take care of changes (like units that have been killed)
EntitySelection.prototype.update = function()
{
	var numberRemoved = 0;
	for each (var ent in this.selected)
	{
		var entState = GetEntityState(ent);
		if (!entState)
		{
			delete this.selected[ent];
			numberRemoved++;
		}
	}
	if (numberRemoved > 0)
	{
		this.dirty = true;
		this.groups.create(this.toList());
	}
};

EntitySelection.prototype.toggle = function(ent)
{
	if (this.selected[ent])
	{
		_setHighlight([ent], 0);
		_setMotionOverlay([ent], false);
		delete this.selected[ent];
	}
	else
	{
		_setHighlight([ent], 1);
		_setMotionOverlay([ent], this.motionDebugOverlay);
		this.selected[ent] = ent;
	}
	this.dirty = true;
};

EntitySelection.prototype.addList = function(ents)
{
	var added = [];
	for each (var ent in ents)
	{
		if (!this.selected[ent])
		{
			added.push(ent);
			this.selected[ent] = ent;
		}
	}
	_setHighlight(added, 1);
	_setMotionOverlay(added, this.motionDebugOverlay);
	if (added.length)
		_playSound(added[0]);
	this.dirty = true;
};

EntitySelection.prototype.reset = function()
{
	_setHighlight(this.toList(), 0);
	_setMotionOverlay(this.toList(), false);
	this.selected = {};
	this.dirty = true;
	this.groups = new EntityGroup();
};

EntitySelection.prototype.toList = function()
{
	var ents = [];
	for each (var ent in this.selected)
		ents.push(ent);
	return ents;
};

EntitySelection.prototype.setHighlightList = function(ents)
{
	var removed = [];
	var added = [];

	// Remove highlighting for the old units (excluding ones that are actively selected too)
	for each (var ent in this.highlighted)
		if (!this.selected[ent])
			removed.push(ent);
	
	// Add new highlighting
	for each (var ent in ents)
		if (!this.selected[ent])
			added.push(ent);

	_setHighlight(removed, 0);
	_setHighlight(added, 0.5);

	// TODO: this could be a bit more efficient by only changing the ones that
	// have entered/left the highlight list
	
	// Store the new list
	this.highlighted = {};
	for each (var ent in ents)
		this.highlighted[ent] = ent;
};

EntitySelection.prototype.SetMotionDebugOverlay = function(enabled)
{
	this.motionDebugOverlay = enabled;
	_setMotionOverlay(this.toList(), enabled);
};

var g_Selection = new EntitySelection();