shipsystem.go

package main

var statInformation map[StatType]StatInfo

type SystemType int

const (
	SYS_PROPULSION SystemType = iota
	SYS_POWER
	SYS_COMPUTER
	SYS_COMMS
	SYS_NAVIGATION
	SYS_SCANNER
	SYS_LIFESUPPORT
	SYS_STORAGE
	SYS_WEAPON
	SYS_SHIELD
	SYS_MAXSYSTEMS
)

type ShipSystem interface {
	GetStat(StatType) int
	GetAllStats() map[StatType]RoomStat
	AddStat(RoomStat)
	RemoveStat(RoomStat)
	InitStats()
	Update(int)
	OnStatUpdate()
}

type SystemStats struct {
	Stats map[StatType]RoomStat
}

func (ss SystemStats) GetStat(s StatType) int {
	if _, ok := ss.Stats[s]; ok {
		return ss.Stats[s].Modifier
	} else {
		return 0
	}
}

func (ss SystemStats) GetAllStats() map[StatType]RoomStat {
	return ss.Stats
}

func (ss *SystemStats) AddStat(s RoomStat) {
	if _, ok := ss.Stats[s.Stat]; ok {
		ss.Stats[s.Stat] = RoomStat{s.Stat, ss.Stats[s.Stat].Modifier + s.Modifier}
	} else {
		ss.Stats[s.Stat] = s
	}
}

func (ss *SystemStats) RemoveStat(s RoomStat) {
	if _, ok := ss.Stats[s.Stat]; ok {
		ss.Stats[s.Stat] = RoomStat{s.Stat, ss.Stats[s.Stat].Modifier - s.Modifier}
		if ss.Stats[s.Stat].Modifier <= 0 {
			delete(ss.Stats, s.Stat)
		}
	}
}

func (ss *SystemStats) InitStats() {
	ss.Stats = make(map[StatType]RoomStat)
}

func (ss *SystemStats) OnStatUpdate() {
	//no op for now.
}

type StatType int

const (
	//SYS_PROPULSION stats
	STAT_SUBLIGHT_THRUST StatType = iota
	STAT_SUBLIGHT_FUELUSE
	STAT_SUBLIGHT_POWER
	STAT_FTL_THRUST
	STAT_FTL_FUELUSE
	STAT_FTL_POWER

	//SYS_POWER stats
	STAT_POWER_GEN
	STAT_POWER_DRAW

	//SYS_NAVIGATION stats
	STAT_NAV_POWER
	STAT_NAV_COMPUTER

	//SYS_LIFESUPPORT
	STAT_PRESSURIZATION_RATE
	STAT_LS_MODULE_CAP
	STAT_CO2_SCRUBRATE

	//SYS_STORAGE
	STAT_GENERAL_STORAGE //total volume for general item storage (L)
	STAT_LIQUID_STORAGE  //total volume for liquid storage (L)
	STAT_GAS_STORAGE     //total volume for gas storage (L)

	STAT_MAXSTATS
)

type StatInfo struct {
	Name        string
	Description string
	System      SystemType
	Stat        StatType
}

type RoomStat struct {
	Stat     StatType
	Modifier int
}

func (rs RoomStat) GetName() string {
	return statInformation[rs.Stat].Name
}

func (rs RoomStat) GetDesc() string {
	return statInformation[rs.Stat].Description
}

func (rs RoomStat) GetSystem() SystemType {
	return statInformation[rs.Stat].System
}

func init() {
	statInformation = make(map[StatType]StatInfo)

	statInformation[STAT_SUBLIGHT_THRUST] = StatInfo{
		Name:        "Sublight Thrust Factor",
		Description: "The thrust provided by the engine for sublight-travel",
		System:      SYS_PROPULSION,
		Stat:        STAT_SUBLIGHT_THRUST,
	}
	statInformation[STAT_SUBLIGHT_FUELUSE] = StatInfo{
		Name:        "Sublight Fuel Use",
		Description: "The fuel used by the engine for sublight-travel",
		System:      SYS_PROPULSION,
		Stat:        STAT_SUBLIGHT_FUELUSE,
	}
	statInformation[STAT_SUBLIGHT_POWER] = StatInfo{
		Name:        "Sublight Power Draw",
		Description: "The power used by the engine for sublight-travel",
		System:      SYS_PROPULSION,
		Stat:        STAT_SUBLIGHT_POWER,
	}
	statInformation[STAT_FTL_THRUST] = StatInfo{
		Name:        "FTL Thrust Factor",
		Description: "The thrust provided by the engine for FTL-travel",
		System:      SYS_PROPULSION,
		Stat:        STAT_FTL_THRUST,
	}
	statInformation[STAT_FTL_FUELUSE] = StatInfo{
		Name:        "FTL Fuel Use",
		Description: "The fuel used by the engine for FTL-travel",
		System:      SYS_PROPULSION,
		Stat:        STAT_FTL_FUELUSE,
	}
	statInformation[STAT_FTL_POWER] = StatInfo{
		Name:        "FTL Power Draw",
		Description: "The power used by the engine for FTL-travel",
		System:      SYS_PROPULSION,
		Stat:        STAT_FTL_POWER,
	}
	statInformation[STAT_POWER_GEN] = StatInfo{
		Name:        "Power Generation",
		Description: "Ship power generated by the module.",
		System:      SYS_POWER,
		Stat:        STAT_POWER_GEN,
	}
	statInformation[STAT_POWER_DRAW] = StatInfo{
		Name:        "Power Draw",
		Description: "Ship power drawn by the module on a regular basis.",
		System:      SYS_POWER,
		Stat:        STAT_POWER_DRAW,
	}
	statInformation[STAT_LS_MODULE_CAP] = StatInfo{
		Name:        "Life Support Module Limit",
		Description: "Number of modules life support system can safely support. Trying to support mode modules than this cap will affect efficiency and performance of all life support systems.",
		System:      SYS_LIFESUPPORT,
		Stat:        STAT_LS_MODULE_CAP,
	}
	statInformation[STAT_PRESSURIZATION_RATE] = StatInfo{
		Name:        "Pressurization Rate",
		Description: "Rate at which life support can repressurize a room.",
		System:      SYS_LIFESUPPORT,
		Stat:        STAT_PRESSURIZATION_RATE,
	}
	statInformation[STAT_CO2_SCRUBRATE] = StatInfo{
		Name:        "CO2 Scrub Rate",
		Description: "Rate at which life support can scrub CO2 from atmosphere and replace with life giving O2.",
		System:      SYS_LIFESUPPORT,
		Stat:        STAT_CO2_SCRUBRATE,
	}
	statInformation[STAT_GENERAL_STORAGE] = StatInfo{
		Name:        "General Storage",
		Description: "Store all the things you love, in Litres.",
		System:      SYS_STORAGE,
		Stat:        STAT_GENERAL_STORAGE,
	}
	statInformation[STAT_LIQUID_STORAGE] = StatInfo{
		Name:        "Liquid Storage",
		Description: "Storage for liquids (water, fuel, etc.), in Litres ",
		System:      SYS_STORAGE,
		Stat:        STAT_LIQUID_STORAGE,
	}
	statInformation[STAT_GAS_STORAGE] = StatInfo{
		Name:        "Gas Storage",
		Description: "Storage for gasses, in Litres.",
		System:      SYS_STORAGE,
		Stat:        STAT_GAS_STORAGE,
	}
}