1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// Copyright Hugh Perkins 2006, 2009
// hughperkins@gmail.com http://manageddreams.com
//
// 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 2 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 in the file licence.txt; if not, write to the
// Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-
// 1307 USA
// You can find the licence also on the web at:
// http://www.opensource.org/licenses/gpl-license.php
//
// ======================================================================================
//
package hughai.unitdata;
import java.util.*;
import java.util.Map;
import com.springrts.ai.*;
import com.springrts.ai.oo.clb.*;
import hughai.PlayerObjects;
import hughai.basictypes.*;
import hughai.*;
import hughai.mapping.*;
import hughai.unitdata.*;
import hughai.utils.*;
// a group of some type of unit, such as level 1 tanks
public abstract class UnitList
{
//public HashMap< Integer, UnitDef> defbyid = new HashMap< Integer, UnitDef>();
HashSet<Unit> units = new HashSet<Unit>();
UnitController unitController;
public UnitList( PlayerObjects playerObjects )
{
unitController = playerObjects.getUnitController();
unitController.registerListener(new UnitControllerHandler());
}
// list of lower-case names that the derived concrete class will handle
abstract List<String> getUnitNames();
public class UnitControllerHandler extends UnitController.UnitAdapter {
@Override
public void UnitRemoved(Unit unit)
{
units.remove(unit);
}
@Override
public void UnitAdded(Unit unit )
{
if (!units.contains(unit))
{
String name = unit.getDef().getName().toLowerCase();
if ( getUnitNames().contains(name))
{
units.add( unit );
}
}
}
}
public HashSet<Unit> getUnits() {
return units;
}
}
|
