Mercurial > sypikslang
view gamelib/schematics.py @ 146:2587f8c34f84
Science juggling and visualization improvements.
author | Jeremy Thurgood <firxen@gmail.com> |
---|---|
date | Fri, 11 May 2012 16:20:49 +0200 |
parents | 53277724645b |
children | 3abb05f7e720 |
line wrap: on
line source
# -*- test-case-name: gamelib.tests.test_schematics -*- from gamelib import research from gamelib.game_base import Science class SchematicCategorySet(object): def __init__(self, *categories): for cat in categories: setattr(self, cat, cat.lower().replace('_', ' ')) cat = SchematicCategorySet( 'HAND_WEAPON', 'VEHICLE', 'DOOMSDAY_DEVICE', 'PATHOGEN', 'MIND_CONTROL', 'BEAST', 'AQUATIC', 'INTELLIGENCE', 'AI', ) class Schematic(Science): # For all Schematics SCIENCE_TYPE = 'schematic' # Acquisition STARTING_PRODUCT = False ACQUISITION_CHANCE = 0.8 CATEGORIES = () # Costs UPGRADE_REQUIREMENT = 1 COST = None MAX_UPGRADE = 5 # Power and reliability PROTOTYPE_RELIABILITY = 0.4 PRODUCTION_RELIABILITY = 0.8 BASE_POWER = None POWER_INCREMENT = None # FIXME: Icons for equipment. IMAGE_NAME = "physics" def can_spend(self, lab, spend): if self.points + spend > self.MAX_UPGRADE: return False extra = self.UPGRADE_REQUIREMENT * (self.points + spend) return lab.meet_requirements(self, extra) def is_a(self, category): return category in self.CATEGORIES def reliability(self): if self.points: exp = 1 + 0.5 * (self.points - 1) return 1.0 - ((1.0 - self.PRODUCTION_RELIABILITY) ** exp) return self.PROTOTYPE_RELIABILITY def power(self): power = self.BASE_POWER if None not in (power, self.POWER_INCREMENT): power += self.POWER_INCREMENT * self.points return power @classmethod def sanity_check(cls): for science, points in cls.PREREQUISITES: assert issubclass(science, Science) assert isinstance(points, int) assert isinstance(cls.CATEGORIES, tuple) class MachineGun(Schematic): NAME = "machine gun" COST = 100 CATEGORIES = (cat.HAND_WEAPON,) STARTING_PRODUCT = True BASE_POWER = 10 PRODUCTION_RELIABILITY = 1.0 def __init__(self, points=0): self.points = 1 class LightningGun(Schematic): NAME = "lightning gun" COST = 300 CATEGORIES = (cat.HAND_WEAPON,) PREREQUISITES = ( (research.Electrickery, 1), ) BASE_POWER = 10 class TeslaTank(Schematic): NAME = "tesla tank" COST = 40000 CATEGORIES = (cat.VEHICLE,) PREREQUISITES = ( (research.Electrickery, 5), (research.Robotics, 1), ) BASE_POWER = 100 POWER_INCREMENT = 10 class DoomsdayVirus(Schematic): NAME = "doomsday virus" COST = 100000 CATEGORIES = (cat.DOOMSDAY_DEVICE, cat.PATHOGEN) PREREQUISITES = ( (research.Biogenetics, 5), ) class PropagandaMachine(Schematic): NAME = "propaganda machine" COST = 1000 CATEGORIES = (cat.MIND_CONTROL,) PREREQUISITES = ( (research.Psychology, 2), ) class GiantSquid(Schematic): NAME = "giant squid" COST = 50000 CATEGORIES = (cat.BEAST, cat.AQUATIC) PREREQUISITES = ( (research.MarineBiology, 20), (research.Biogenetics, 15), ) class SharksWithFrickinLasers(Schematic): NAME = "sharks with frickin' lasers" COST = 20000 CATEGORIES = (cat.BEAST, cat.AQUATIC) PREREQUISITES = ( (research.MarineBiology, 10), (research.Lasers, 7), ) class GiantRobot(Schematic): NAME = "giant robot" COST = 2500000 CATEGORIES = (cat.DOOMSDAY_DEVICE, cat.AI) PREREQUISITES = ( (research.Robotics, 15), (research.ArtificialIntelligence, 10), ) class ObservationDrone(Schematic): NAME = "observation drone" COST = 6000 CATEGORIES = (cat.INTELLIGENCE,) PREREQUISITES = ( (research.Robotics, 3), ) class SpySatellite(Schematic): NAME = "spy satellite" COST = 100000 CATEGORIES = (cat.INTELLIGENCE,) PREREQUISITES = ( (research.Rocketry, 5), (research.Space, 1), ) class OrbitalLaserPlatform(Schematic): NAME = "orbital laser platform" COST = 10000000 CATEGORIES = (cat.DOOMSDAY_DEVICE,) PREREQUISITES = ( (SpySatellite, 3), (research.Lasers, 15), (research.Space, 5), ) class TacticalAI(Schematic): NAME = "tactical AI" COST = 10000 CATEGORIES = (cat.AI, cat.INTELLIGENCE) PREREQUISITES = ( (research.ArtificialIntelligence, 3), ) class LaserGun(Schematic): NAME = "Laser Gun" COST = 300 CATEGORIES = (cat.HAND_WEAPON,) PREREQUISITES = ( (research.Lasers, 2), )