/*
* UFO Saved Game Editor
* Copyright (C) 2010 Christopher Davoren
*
* 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 <http://www.gnu.org/licenses/>.
*/
package net.rubikscomplex.ufosge.data;
import java.io.EOFException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import net.rubikscomplex.ufosge.util.UFOInputStream;
import net.rubikscomplex.ufosge.util.UFOOutputStream;
/**
*
* @author Chris Davoren
*/
public class Base {
public static final byte FACILITY_EMPTY = -1;
public static final byte FACILITY_ACCESS_LIFT = 0;
public static final byte FACILITY_LIVING_QUARTERS = 1;
public static final byte FACILITY_LABORATORY = 2;
public static final byte FACILITY_WORKSHOP = 3;
public static final byte FACILITY_SMALL_RADAR = 4;
public static final byte FACILITY_LARGE_RADAR = 5;
public static final byte FACILITY_MISSILE_DEFENSE = 6;
public static final byte FACILITY_GENERAL_STORES = 7;
public static final byte FACILITY_ALIEN_CONT = 8;
public static final byte FACILITY_LASER_DEFENSE = 9;
public static final byte FACILITY_PLASMA_DEFENSE = 10;
public static final byte FACILITY_FUSION_DEFENSE = 11;
public static final byte FACILITY_GRAV_SHIELD = 12;
public static final byte FACILITY_MIND_SHIELD = 13;
public static final byte FACILITY_PSIONIC_LAB = 14;
public static final byte FACILITY_HYPERWAVE = 15;
public static final byte FACILITY_HANGAR_TL = 16;
public static final byte FACILITY_HANGAR_TR = 17;
public static final byte FACILITY_HANGAR_BL = 18;
public static final byte FACILITY_HANGAR_BR = 19;
public static final int NAME_LEN = 16;
public static final int BASE_SIZE = 6;
public static final int INVENTORY_SIZE = 96;
public static final int UNKNOWN_SIZE = 3;
public static final int HF_TL = 0x01;
public static final int HF_TR = 0x02;
public static final int HF_BL = 0x04;
public static final int HF_BR = 0x08;
public static final int HF_TOP = HF_TL | HF_TR;
public static final int HF_BOTTOM = HF_BL | HF_BR;
public static final int HF_LEFT = HF_TL | HF_BL;
public static final int HF_RIGHT = HF_TR | HF_BR;
public static final int HF_ANY = HF_LEFT | HF_RIGHT;
public static final int NUM_FACILITY_TYPES = 20;
public static final int MAX_DAYS_LEFT = 127;
public static final String[] INV_NAMES = {
"Stingray Launcher",
"Avalanche Launcher",
"Cannon",
"Fusion Ball Launcher",
"Laser Cannon",
"Plasma Beam",
"Stingray Missile",
"Avalanche Missile",
"Cannon Rounds",
"Fusion Balls",
"Tank/Cannon",
"Tank/Rocket Launcher",
"Tank/Laser Cannon",
"Hovertank/Plasma",
"Hovertank/Launcher",
"Pistol",
"Pistol Clip",
"Rifle",
"Rifle Clip",
"Heavy Cannon",
"HC-AP Ammo",
"HC-HE Ammo",
"HC-IN Ammo",
"Auto Cannon",
"AC-AP Ammo",
"AC-HE Ammo",
"AC-IN Ammo",
"Rocket Launcher",
"Small Rocket",
"Large Rocket",
"Incendiary Rocket",
"Laser Pistol",
"Laser Rifle",
"Heavy Laser",
"Grenade",
"Smoke Grenade",
"Proximity Grenade",
"High Explosive",
"Motion Scanner",
"Medi-Kit",
"Psi-Amp",
"Stun Rod",
"Electro Flare",
"empty *",
"empty*",
"empty*",
"CORPSE*",
"CORPSE & ARMOUR*",
"CORPSE & POWER SUIT*",
"Heavy Plasma",
"Heavy Plasma Clip",
"Plasma Rifle",
"Plasma Rifle Clip",
"Plasma Pistol",
"Plasma Pistol Clip",
"Blaster Launcher",
"Blaster Bomb",
"Small Launcher",
"Stun Bomb",
"Alien Grenade",
"Elerium-115",
"Mind Probe",
">>UNDEFINED <<*",
">> empty <<*",
">> empty << *",
"Sectoid Corpse",
"Snakeman Corpse",
"Ethereal Corpse",
"Muton Corpse",
"Floater Corpse",
"Celatid Corpse",
"Silacoid Corpse",
"Chryssalid Corpse",
"Reaper Corpse",
"Sectopod Corpse",
"Cyberdisc Corpse",
"Hovertank Corpse",
"Tank Corpse",
"Male Civilian Corpse*",
"Female Civilian Corpse*",
"UFO Power Source",
"UFO Navigation",
"UFO Construction",
"Alien Food",
"Alien Reproduction",
"Alien Entertainment",
"Alien Surgery",
"Examination Room",
"Alien Alloys",
"Alien Habitat",
"Personal Armour",
"Power Suit",
"Flying Suit",
"HWP Cannon Shell",
"HWP Rockets",
"HWP Fusion Bomb",
};
public int index;
public String name;
public short shortRange;
public short longRange;
public short hyperwaveRange;
public byte[][] facilities;
public byte[][] daysLeft;
public short engineers;
public short scientists;
public short[] inventory;
public boolean isActive;
byte[] unknown;
public static String facilityToString(byte facility) {
switch (facility) {
case FACILITY_EMPTY: return "Empty";
case FACILITY_ACCESS_LIFT: return "Access Lift";
case FACILITY_LIVING_QUARTERS: return "Living Quarters";
case FACILITY_LABORATORY: return "Laboratory";
case FACILITY_WORKSHOP: return "Workshop";
case FACILITY_SMALL_RADAR: return "Small Radar";
case FACILITY_LARGE_RADAR: return "Large Radar";
case FACILITY_MISSILE_DEFENSE: return "Missile Defense";
case FACILITY_GENERAL_STORES: return "General Stores";
case FACILITY_ALIEN_CONT: return "Alient Containment";
case FACILITY_LASER_DEFENSE: return "Laser Defense";
case FACILITY_PLASMA_DEFENSE: return "Plasma Defense";
case FACILITY_FUSION_DEFENSE: return "Fusion Defense";
case FACILITY_GRAV_SHIELD: return "Grav Shield";
case FACILITY_MIND_SHIELD: return "Mind Shield";
case FACILITY_PSIONIC_LAB: return "Psionic Laboratory";
case FACILITY_HYPERWAVE: return "Hyperwave Decoder";
case FACILITY_HANGAR_TL:
case FACILITY_HANGAR_TR:
case FACILITY_HANGAR_BL:
case FACILITY_HANGAR_BR: return "Hangar";
default: return "Unknown";
}
}
public static byte stringToFacility(String str) {
if (str.equalsIgnoreCase("Empty")) {
return FACILITY_EMPTY;
}
else if (str.equalsIgnoreCase("Access Lift")) {
return FACILITY_ACCESS_LIFT;
}
else if (str.equalsIgnoreCase("Living Quarters")) {
return FACILITY_LIVING_QUARTERS;
}
else if (str.equalsIgnoreCase("Laboratory")) {
return FACILITY_LABORATORY;
}
else if (str.equalsIgnoreCase("Workshop")) {
return FACILITY_WORKSHOP;
}
else if (str.equalsIgnoreCase("Small Radar")) {
return FACILITY_SMALL_RADAR;
}
else if (str.equalsIgnoreCase("Large Radar")) {
return FACILITY_LARGE_RADAR;
}
else if (str.equalsIgnoreCase("Missile Defense")) {
return FACILITY_MISSILE_DEFENSE;
}
else if (str.equalsIgnoreCase("General Stores")) {
return FACILITY_GENERAL_STORES;
}
else if (str.equalsIgnoreCase("Alient Containment")) {
return FACILITY_ALIEN_CONT;
}
else if (str.equalsIgnoreCase("Laser Defense")) {
return FACILITY_LASER_DEFENSE;
}
else if (str.equalsIgnoreCase("Plasma Defense")) {
return FACILITY_PLASMA_DEFENSE;
}
else if (str.equalsIgnoreCase("Fusion Defense")) {
return FACILITY_FUSION_DEFENSE;
}
else if (str.equalsIgnoreCase("Grav Shield")) {
return FACILITY_GRAV_SHIELD;
}
else if (str.equalsIgnoreCase("Mind Shield")) {
return FACILITY_MIND_SHIELD;
}
else if (str.equalsIgnoreCase("Psionic Lab")) {
return FACILITY_PSIONIC_LAB;
}
else if (str.equalsIgnoreCase("Hyperwave Decoder")) {
return FACILITY_HYPERWAVE;
}
else if (str.equalsIgnoreCase("Hangar")) {
return FACILITY_HANGAR_TL;
}
else {
return FACILITY_EMPTY;
}
}
public static boolean isHangar(int facilityType) {
return isHangar(facilityType, HF_ANY);
}
public static boolean isHangar(int facilityType, int flags) {
return ((facilityType == FACILITY_HANGAR_TL) && ((flags & HF_TL) != 0)) ||
((facilityType == FACILITY_HANGAR_TR) && ((flags & HF_TR) != 0)) ||
((facilityType == FACILITY_HANGAR_BL) && ((flags & HF_BL) != 0)) ||
((facilityType == FACILITY_HANGAR_BR) && ((flags & HF_BR) != 0));
}
public static Map<Byte, String> getFacilityMap() {
HashMap<Byte, String> map = new HashMap<Byte, String>();
map.put(FACILITY_EMPTY, facilityToString(FACILITY_EMPTY));
map.put(FACILITY_ACCESS_LIFT, facilityToString(FACILITY_ACCESS_LIFT));
map.put(FACILITY_LIVING_QUARTERS, facilityToString(FACILITY_LIVING_QUARTERS));
map.put(FACILITY_LABORATORY, facilityToString(FACILITY_LABORATORY));
map.put(FACILITY_WORKSHOP, facilityToString(FACILITY_WORKSHOP));
map.put(FACILITY_SMALL_RADAR, facilityToString(FACILITY_SMALL_RADAR));
map.put(FACILITY_LARGE_RADAR, facilityToString(FACILITY_LARGE_RADAR));
map.put(FACILITY_MISSILE_DEFENSE, facilityToString(FACILITY_MISSILE_DEFENSE));
map.put(FACILITY_GENERAL_STORES, facilityToString(FACILITY_GENERAL_STORES));
map.put(FACILITY_ALIEN_CONT, facilityToString(FACILITY_ALIEN_CONT));
map.put(FACILITY_LASER_DEFENSE, facilityToString(FACILITY_LASER_DEFENSE));
map.put(FACILITY_PLASMA_DEFENSE, facilityToString(FACILITY_PLASMA_DEFENSE));
map.put(FACILITY_FUSION_DEFENSE, facilityToString(FACILITY_FUSION_DEFENSE));
map.put(FACILITY_GRAV_SHIELD, facilityToString(FACILITY_GRAV_SHIELD));
map.put(FACILITY_MIND_SHIELD, facilityToString(FACILITY_MIND_SHIELD));
map.put(FACILITY_PSIONIC_LAB, facilityToString(FACILITY_PSIONIC_LAB));
map.put(FACILITY_HYPERWAVE, facilityToString(FACILITY_HYPERWAVE));
map.put(FACILITY_HANGAR_TL, facilityToString(FACILITY_HANGAR_TL));
// map.put(FACILITY_HANGAR_TR, facilityToString(FACILITY_HANGAR_TR));
// map.put(FACILITY_HANGAR_BL, facilityToString(FACILITY_HANGAR_BL));
// map.put(FACILITY_HANGAR_BR, facilityToString(FACILITY_HANGAR_BR));
return map;
}
public static ArrayList<String> getFacilityNameList(boolean sort, boolean emptyAtTop) {
ArrayList<String> list = new ArrayList<String>();
list.add(facilityToString(FACILITY_ACCESS_LIFT));
list.add(facilityToString(FACILITY_LIVING_QUARTERS));
list.add(facilityToString(FACILITY_LABORATORY));
list.add(facilityToString(FACILITY_WORKSHOP));
list.add(facilityToString(FACILITY_SMALL_RADAR));
list.add(facilityToString(FACILITY_LARGE_RADAR));
list.add(facilityToString(FACILITY_MISSILE_DEFENSE));
list.add(facilityToString(FACILITY_GENERAL_STORES));
list.add(facilityToString(FACILITY_ALIEN_CONT));
list.add(facilityToString(FACILITY_LASER_DEFENSE));
list.add(facilityToString(FACILITY_PLASMA_DEFENSE));
list.add(facilityToString(FACILITY_FUSION_DEFENSE));
list.add(facilityToString(FACILITY_GRAV_SHIELD));
list.add(facilityToString(FACILITY_MIND_SHIELD));
list.add(facilityToString(FACILITY_PSIONIC_LAB));
list.add(facilityToString(FACILITY_HYPERWAVE));
list.add(facilityToString(FACILITY_HANGAR_TL));
if (!sort || !emptyAtTop) {
list.add(0, facilityToString(FACILITY_EMPTY));
}
if (sort) {
Collections.sort(list);
}
if (sort && emptyAtTop) {
list.add(0, facilityToString(FACILITY_EMPTY));
}
return list;
}
public void setFacilityType(int x, int y, byte facilityType) {
setFacilityType(x, y, facilityType, facilities, daysLeft);
}
public static void setFacilityType(int x, int y, byte facilityType, byte[][] facilities, byte daysLeft[][]) {
if (facilityType == FACILITY_HANGAR_TL && (x > BASE_SIZE - 2 || y > BASE_SIZE - 2)) {
return;
}
if (facilityType != FACILITY_EMPTY) {
if (daysLeft[y][x] < 0) {
daysLeft[y][x] = 0;
}
}
else {
daysLeft[y][x] = -1;
}
if (facilityType != FACILITY_HANGAR_TL) {
if (facilities[y][x] == FACILITY_HANGAR_TL) {
facilities[y + 1][x] = FACILITY_EMPTY;
facilities[y][x + 1] = FACILITY_EMPTY;
facilities[y + 1][x + 1] = FACILITY_EMPTY;
daysLeft[y + 1][x] = -1;
daysLeft[y][x + 1] = -1;
daysLeft[y + 1][x + 1] = -1;
}
facilities[y][x] = facilityType;
}
else {
facilities[y][x] = FACILITY_HANGAR_TL;
if (x < BASE_SIZE - 1) {
facilities[y][x + 1] = FACILITY_HANGAR_TR;
daysLeft[y][x + 1] = daysLeft[y][x];
}
if (y < BASE_SIZE - 1) {
facilities[y + 1][x] = FACILITY_HANGAR_BL;
daysLeft[y + 1][x] = daysLeft[y][x];
}
if (x < BASE_SIZE - 1 && y < BASE_SIZE - 1) {
facilities[y + 1][x + 1] = FACILITY_HANGAR_BR;
daysLeft[y + 1][x + 1] = daysLeft[y][x];
}
}
}
public static void setDaysLeft(int x, int y, byte daysLeftNew, byte[][] facilities, byte[][] daysLeft) {
if (facilities[y][x] == FACILITY_EMPTY) {
daysLeft[y][x] = -1;
return;
}
daysLeft[y][x] = daysLeftNew >= 0 ? daysLeftNew : 0;
if (!Base.isHangar(facilities[y][x])) {
return;
}
else {
if (x < BASE_SIZE - 1) {
daysLeft[y][x + 1] = daysLeftNew;
}
if (y < BASE_SIZE - 1) {
daysLeft[y + 1][x] = daysLeftNew;
}
if (x < BASE_SIZE - 1 && y < BASE_SIZE - 1) {
daysLeft[y + 1][x + 1] = daysLeftNew;
}
}
}
public boolean readFromFile(UFOInputStream in, int index) throws IOException {
try {
name = in.readUFOString(NAME_LEN);
shortRange = in.readUFOShort();
longRange = in.readUFOShort();
hyperwaveRange = in.readUFOShort();
facilities = new byte[BASE_SIZE][BASE_SIZE];
for (int i = 0; i < BASE_SIZE; i++) {
in.read(facilities[i]);
}
daysLeft = new byte[BASE_SIZE][BASE_SIZE];
for (int i = 0; i < BASE_SIZE; i++) {
in.read(daysLeft[i]);
}
engineers = in.readUFOByte();
scientists = in.readUFOByte();
inventory = new short[INVENTORY_SIZE];
for (int i = 0; i < INVENTORY_SIZE; i++) {
inventory[i] = in.readUFOShort();
}
isActive = in.readByte() == 0;
unknown = new byte[UNKNOWN_SIZE];
in.read(unknown);
this.index = index;
return true;
}
catch (EOFException e) {
return false;
}
}
public void writeToFile(UFOOutputStream out) throws IOException {
out.writeUFOString(name, NAME_LEN);
out.writeUFOShort(shortRange);
out.writeUFOShort(longRange);
out.writeUFOShort(hyperwaveRange);
for(int i = 0; i < BASE_SIZE; i++) {
out.write(facilities[i]);
}
for (int i = 0; i < BASE_SIZE; i++) {
out.write(daysLeft[i]);
}
out.writeByte(engineers);
out.writeByte(scientists);
for(int i = 0; i < INVENTORY_SIZE; i++) {
out.writeUFOShort(inventory[i]);
}
out.writeByte(isActive ? 0 : 1);
out.write(unknown);
}
public static ArrayList<Base> readList(UFOInputStream in) throws IOException {
ArrayList<Base> list = new ArrayList<Base>();
while (true) {
Base base = new Base();
if (base.readFromFile(in, list.size())) {
list.add(base);
}
else {
break;
}
}
return list;
}
public static void writeList(UFOOutputStream out, ArrayList<Base> list) throws IOException {
Iterator<Base> bi = list.iterator();
while(bi.hasNext()) {
bi.next().writeToFile(out);
}
}
public boolean isUsed() {
return isActive || name.length() > 0;
}
public void recalculateRadarValues() {
shortRange = 0;
longRange = 0;
hyperwaveRange = 0;
for(int i = 0; i < BASE_SIZE; i++) {
for(int j = 0; j < BASE_SIZE; j++) {
if(daysLeft[i][j] != 0) {
continue;
}
if(facilities[i][j] == FACILITY_SMALL_RADAR) {
shortRange += shortRange == 0 || shortRange == 20 ? 10 : 0;
}
else if(facilities[i][j] == FACILITY_LARGE_RADAR) {
longRange += longRange == 0 ? 20 : 0;
shortRange += shortRange == 0 || shortRange == 10 ? 20 : 0;
}
else if (facilities[i][j] == FACILITY_HYPERWAVE) {
hyperwaveRange = 100;
}
}
}
}
@Override
public String toString() {
return name;
}
}