self.angered = False
def angry(self):
self.angered = True
+ def at(self, q):
+ return (q.i, q.j) == (self.i, self.j)
class Planet:
def __init__(self):
self.pclass = None # could be ""M", "N", "O", or "destroyed"
self.crystals = "absent"# could be "mined", "present", "absent"
self.known = "unknown" # could be "unknown", "known", "shuttle_down"
- self.inhabited = False # is it inhabites?
+ self.inhabited = False # is it inhabited?
def __str__(self):
return self.name
class Page:
def __init__(self):
self.stars = None
- self.starbase = None
+ self.starbase = False
self.klingons = None
+ def __repr__(self):
+ return "<%s,%s,%s>" % (self.klingons, self.starbase, self.stars)
def fill2d(size, fillfun):
"Fill an empty list in 2D."
def __init__(self, etype=None, loc=None, power=None):
self.type = etype
self.location = Coord()
+ self.kdist = None
+ self.kavgd = None
if loc:
self.move(loc)
self.power = power # enemy energy level
self.score = 0.0 # overall score
self.perdate = 0.0 # rate of kills
self.idebug = False # Debugging instrumentation enabled?
+ self.statekscmdr = None # No SuperCommander coordinates yet.
def recompute(self):
# Stas thinks this should be (C expression):
# game.state.remkl + len(game.state.kcmdr) > 0 ?
# avoid intruding on another commander's territory
if enemy.type == 'C':
if iq in game.state.kcmdr:
- return False
+ return []
# refuse to leave if currently attacking starbase
if game.battle == game.quadrant:
- return False
+ return []
# don't leave if over 1000 units of energy
if enemy.power > 1000.0:
- return False
- # emit escape message and move out of quadrant.
- # we know this if either short or long range sensors are working
- if not damaged(DSRSENS) or not damaged(DLRSENS) or \
- game.condition == "docked":
- prout(crmena(True, enemy.type, "sector", enemy.location) + \
- (_(" escapes to Quadrant %s (and regains strength).") % iq))
+ return []
+ oldloc = copy.copy(enemy.location)
# handle local matters related to escape
enemy.move(None)
game.klhere -= 1
if cmdr == game.quadrant:
game.state.kcmdr.append(iq)
break
- return True # success
+ # report move out of quadrant.
+ return [(True, enemy, oldloc, ibq)]
# The bad-guy movement algorithm:
#
nbaddys = (((game.quadrant in game.state.kcmdr)*2 + (game.state.kscmdr==game.quadrant)*2+game.klhere*1.23+game.irhere*1.5)/2.0)
else:
nbaddys = (game.quadrant in game.state.kcmdr) + (game.state.kscmdr==game.quadrant)
- dist1 = enemy.kdist
- mdist = int(dist1 + 0.5) # Nearest integer distance
+ old_dist = enemy.kdist
+ mdist = int(old_dist + 0.5) # Nearest integer distance
# If SC, check with spy to see if should hi-tail it
if enemy.type == 'S' and \
(enemy.power <= 500.0 or (game.condition=="docked" and not damaged(DPHOTON))):
motion = ((forces + randreal(200))/150.0) - 5.0
else:
if forces > 1000.0: # Very strong -- move in for kill
- motion = (1.0 - randreal())**2 * dist1 + 1.0
+ motion = (1.0 - randreal())**2 * old_dist + 1.0
if game.condition == "docked" and (game.options & OPTION_BASE): # protected by base -- back off !
motion -= game.skill*(2.0-randreal()**2)
if game.idebug:
proutn("=== MOTION = %d, FORCES = %1.2f, " % (motion, forces))
# don't move if no motion
if motion == 0:
- return
+ return []
# Limit motion according to skill
if abs(motion) > game.skill:
if motion < 0:
while attempts < 20 and not success:
attempts += 1
if look.i < 0 or look.i >= QUADSIZE:
- if motion < 0 and tryexit(enemy, look, irun):
- return
+ if motion < 0:
+ return tryexit(enemy, look, irun)
if krawli == m.i or m.j == 0:
break
look.i = goto.i + krawli
krawli = -krawli
elif look.j < 0 or look.j >= QUADSIZE:
- if motion < 0 and tryexit(enemy, look, irun):
- return
+ if motion < 0:
+ return tryexit(enemy, look, irun)
if krawlj == m.j or m.i == 0:
break
look.j = goto.j + krawlj
if game.quad[look.i][look.j] == game.ship and \
(enemy.type == 'C' or enemy.type == 'S'):
collision(rammed=True, enemy=enemy)
- return
+ return []
if krawli != m.i and m.j != 0:
look.i = goto.i + krawli
krawli = -krawli
break # done early
if game.idebug:
skip(1)
- if enemy.move(goto):
- if not damaged(DSRSENS) or game.condition == "docked":
- proutn(_("*** %s from Sector %s") % (cramen(enemy.type), enemy.location))
- if enemy.kdist < dist1:
- proutn(_(" advances to "))
- else:
- proutn(_(" retreats to "))
- prout("Sector %s." % goto)
+ # Enemy moved, but is still in sector
+ return [(False, enemy, old_dist, goto)]
def moveklings():
"Sequence Klingon tactical movement."
prout("== MOVCOM")
# Figure out which Klingon is the commander (or Supercommander)
# and do move
+ tacmoves = []
if game.quadrant in game.state.kcmdr:
for enemy in game.enemies:
if enemy.type == 'C':
- movebaddy(enemy)
+ tacmoves += movebaddy(enemy)
if game.state.kscmdr == game.quadrant:
for enemy in game.enemies:
if enemy.type == 'S':
- movebaddy(enemy)
+ tacmoves += movebaddy(enemy)
break
# If skill level is high, move other Klingons and Romulans too!
# Move these last so they can base their actions on what the
if game.skill >= SKILL_EXPERT and (game.options & OPTION_MVBADDY):
for enemy in game.enemies:
if enemy.type in ('K', 'R'):
- movebaddy(enemy)
- sortenemies()
+ tacmoves += movebaddy(enemy)
+ return tacmoves
def movescom(iq, avoid):
"Commander movement helper."
unschedule(FSCDBAS)
for enemy in game.enemies:
if enemy.type == 'S':
- break
- enemy.move(None)
+ enemy.move(None)
game.klhere -= 1
if game.condition != "docked":
newcnd()
return None # we should never get here
def collision(rammed, enemy):
- "Collision handling fot rammong events."
+ "Collision handling for rammong events."
prouts(_("***RED ALERT! RED ALERT!"))
skip(1)
prout(_("***COLLISION IMMINENT."))
if iquad == '.':
continue
# hit something
- if not damaged(DSRSENS) or game.condition == "docked":
+ setwnd(message_window)
+ if not damaged(DSRSENS) or game.condition == "docked":
skip(1) # start new line after text track
if iquad in ('E', 'F'): # Hit our ship
skip(1)
return None
for enemy in game.enemies:
if w == enemy.location:
- break
- kp = math.fabs(enemy.power)
- h1 = 700.0 + randrange(100) - \
- 1000.0 * (w-origin).distance() * math.fabs(math.sin(bullseye-track.angle))
- h1 = math.fabs(h1)
- if kp < h1:
- h1 = kp
- if enemy.power < 0:
- enemy.power -= -h1
- else:
- enemy.power -= h1
- if enemy.power == 0:
- deadkl(w, iquad, w)
- return None
- proutn(crmena(True, iquad, "sector", w))
- displacement = course(track.bearing+randreal(-2.4, 2.4), distance=2**0.5)
- displacement.next()
- bumpto = displacement.sector()
- if not bumpto.valid_sector():
- prout(_(" damaged but not destroyed."))
- return
- if game.quad[bumpto.i][bumpto.j] == ' ':
- prout(_(" buffeted into black hole."))
- deadkl(w, iquad, bumpto)
- if game.quad[bumpto.i][bumpto.j] != '.':
- prout(_(" damaged but not destroyed."))
+ kp = math.fabs(enemy.power)
+ h1 = 700.0 + randrange(100) - \
+ 1000.0 * (w-origin).distance() * math.fabs(math.sin(bullseye-track.angle))
+ h1 = math.fabs(h1)
+ if kp < h1:
+ h1 = kp
+ if enemy.power < 0:
+ enemy.power -= -h1
+ else:
+ enemy.power -= h1
+ if enemy.power == 0:
+ deadkl(w, iquad, w)
+ return None
+ proutn(crmena(True, iquad, "sector", w))
+ displacement = course(track.bearing+randreal(-2.4, 2.4), distance=2**0.5)
+ displacement.next()
+ bumpto = displacement.sector()
+ if not bumpto.valid_sector():
+ prout(_(" damaged but not destroyed."))
+ return
+ if game.quad[bumpto.i][bumpto.j] == ' ':
+ prout(_(" buffeted into black hole."))
+ deadkl(w, iquad, bumpto)
+ if game.quad[bumpto.i][bumpto.j] != '.':
+ prout(_(" damaged but not destroyed."))
+ else:
+ prout(_(" damaged-- displaced by blast to Sector %s ")%bumpto)
+ enemy.location = bumpto
+ game.quad[w.i][w.j] = '.'
+ game.quad[bumpto.i][bumpto.j] = iquad
+ for enemy in game.enemies:
+ enemy.kdist = enemy.kavgd = (game.sector-enemy.location).distance()
+ sortenemies()
+ break
else:
- prout(_(" damaged-- displaced by blast to Sector %s ")%bumpto)
- enemy.location = bumpto
- game.quad[w.i][w.j] = '.'
- game.quad[bumpto.i][bumpto.j] = iquad
- for enemy in game.enemies:
- enemy.kdist = enemy.kavgd = (game.sector-enemy.location).distance()
- sortenemies()
+ prout("Internal error, no enemy where expected!")
+ raise SystemExit, 1
return None
elif iquad == 'B': # Hit a base
skip(1)
game.state.baseq = filter(lambda x: x != game.quadrant, game.state.baseq)
game.quad[w.i][w.j] = '.'
game.base.invalidate()
- game.state.galaxy[game.quadrant.i][game.quadrant.j].starbase -= 1
- game.state.chart[game.quadrant.i][game.quadrant.j].starbase -= 1
+ game.state.galaxy[game.quadrant.i][game.quadrant.j].starbase = False
+ game.state.chart[game.quadrant.i][game.quadrant.j].starbase = False
game.state.basekl += 1
newcnd()
return None
return None
break
skip(1)
+ setwnd(message_window)
prout(_("Torpedo missed."))
return None
# Select devices and cause damage
cdam = []
while ncrit > 0:
- ncrit -= 1
while True:
j = randdevice()
# Cheat to prevent shuttle damage unless on ship
cdam.append(j)
extradm = (hit*game.damfac)/(ncrit*randreal(75, 100))
game.damage[j] += extradm
+ ncrit -= 1
skipcount = 0
for (i, j) in enumerate(cdam):
proutn(device[j])
return
# commanders get a chance to tac-move towards you
if (((game.quadrant in game.state.kcmdr or game.state.kscmdr == game.quadrant) and not game.justin) or game.skill == SKILL_EMERITUS) and torps_ok:
- moveklings()
+ for (bugout, enemy, old, goto) in moveklings():
+ if bugout:
+ # we know about this if either short or long range
+ # sensors are working
+ if damaged(DSRSENS) and damaged(DLRSENS) \
+ and game.condition != "docked":
+ prout(crmena(True, enemy.type, "sector", old) + \
+ (_(" escapes to Quadrant %s (and regains strength).") % goto))
+ else: # Enemy still in-sector
+ if enemy.move(goto):
+ if not damaged(DSRSENS) or game.condition == "docked":
+ proutn(_("*** %s from Sector %s") % (cramen(enemy.type), enemy.location))
+ if enemy.kdist < old:
+ proutn(_(" advances to "))
+ else:
+ proutn(_(" retreats to "))
+ prout("Sector %s." % goto)
+ sortenemies()
# if no enemies remain after movement, we're done
- if len(game.enemies) == 0 or (len(game.enemies) == 1 and thing == game.quadrant and not thing.angered):
+ if len(game.enemies) == 0 or (len(game.enemies) == 1 and thing.at(game.quadrant) and not thing.angered):
return
# set up partial hits if attack happens during shield status change
pfac = 1.0/game.inshld
"Register a phaser hit on Klingons and Romulans."
w = Coord()
skip(1)
- for (kk, wham) in enumerate(hits):
+ kk = 0
+ for wham in hits:
if wham == 0:
continue
dustfac = randreal(0.9, 1.0)
else:
prout(_("(Shields not currently useable.)"))
newqad()
- # Adjust finish time to time of tractor beaming
- fintim = game.state.date+game.optime
+ # Adjust finish time to time of tractor beaming?
+ # fintim = game.state.date+game.optime
attack(torps_ok=False)
if not game.state.kcmdr:
unschedule(FTBEAM)
#announce()
skip(1)
prout(_("Lt. Uhura- \"The deep space probe is now in Quadrant %s.\"") % game.probe.quadrant())
- pdest = game.state.galaxy[game.probe.quadrant().i][game.probe.quadrant().j]
+ pquad = game.probe.quadrant()
+ pdest = game.state.galaxy[pquad.i][pquad.j]
if communicating():
- chp = game.state.chart[game.probe.quadrant().i][game.probe.quadrant().j]
- chp.klingons = pdest.klingons
- chp.starbase = pdest.starbase
- chp.stars = pdest.stars
+ game.state.chart[pquad.i][pquad.j].klingons = pdest.klingons
+ game.state.chart[pquad.i][pquad.j].starbase = pdest.starbase
+ game.state.chart[pquad.i][pquad.j].stars = pdest.stars
pdest.charted = True
game.probe.moves -= 1 # One less to travel
if game.probe.arrived() and game.isarmed and pdest.stars:
supernova(game.probe) # fire in the hole!
unschedule(FDSPROB)
- if game.state.galaxy[game.quadrant().i][game.quadrant().j].supernova:
+ if game.state.galaxy[pquad.i][pquad.j].supernova:
return
elif evcode == FDISTR: # inhabited system issues distress call
unschedule(FDISTR)
global linecount
sys.stdout.write('\n')
proutn(prompt)
- raw_input()
+ if not replayfp:
+ raw_input()
sys.stdout.write('\n' * rows)
linecount = 0
if curwnd == message_window and y >= my - 2:
pause_game()
clrscr()
+ # Uncomment this to debug curses problems
+ if logfp:
+ logfp.write("#curses: at %s proutn(%s)\n" % ((y, x), repr(line)))
curwnd.addstr(line)
curwnd.refresh()
else:
"Change windows -- OK for this to be a no-op in tty mode."
global curwnd
if game.options & OPTION_CURSES:
+ # Uncomment this to debug curses problems
+ if logfp:
+ if wnd == fullscreen_window:
+ legend = "fullscreen"
+ elif wnd == srscan_window:
+ legend = "srscan"
+ elif wnd == report_window:
+ legend = "report"
+ elif wnd == status_window:
+ legend = "status"
+ elif wnd == lrscan_window:
+ legend = "lrscan"
+ elif wnd == message_window:
+ legend = "message"
+ elif wnd == prompt_window:
+ legend = "prompt"
+ else:
+ legend = "unknown"
+ logfp.write("#curses: setwnd(%s)\n" % legend)
curwnd = wnd
- curses.curs_set(wnd == fullscreen_window or wnd == message_window or wnd == prompt_window)
+ # Some curses implementations get confused when you try this.
+ try:
+ curses.curs_set(wnd in (fullscreen_window, message_window, prompt_window))
+ except curses.error:
+ pass
def clreol():
"Clear to end of line -- can be a no-op in tty mode"
scanner.chew()
return
# Make sure enough time is left for the trip
- game.optime = course.dist/0.095
+ game.optime = course.distance/0.095
if game.optime >= game.state.remtime:
prout(_("First Officer Spock- \"Captain, our speed under impulse"))
prout(_("power is only 0.95 sectors per stardate. Are you sure"))
game.ididit = True
if game.alldone:
return
- power = 20.0 + 100.0*course.dist
+ power = 20.0 + 100.0*course.distance
game.energy -= power
- game.optime = course.dist/0.095
+ game.optime = course.distance/0.095
if game.energy <= 0:
finish(FNRG)
return
skip(1)
prout(_("Engineering to bridge--"))
if not game.shldup or 0.5*wcourse.power(game.warpfac) > game.energy:
- iwarp = (game.energy/(wcourse.dist+0.05)) ** 0.333333333
+ iwarp = (game.energy/(wcourse.distance+0.05)) ** 0.333333333
if iwarp <= 0:
prout(_("We can't do it, Captain. We don't have enough energy."))
else:
game.instar = 0
for i in range(GALSIZE):
for j in range(GALSIZE):
- k = randrange(1, QUADSIZE**2/10+1)
+ # Can't have more stars per quadrant than fit in one decimal digit,
+ # if we do the chart representation will break.
+ k = randrange(1, min(10, QUADSIZE**2/10))
game.instar += k
game.state.galaxy[i][j].stars = k
# Locate star bases in galaxy
+ if game.idebug:
+ prout("=== Allocating %d bases" % game.inbase)
for i in range(game.inbase):
while True:
while True:
prout("=== Saving base #%d, close to #%d" % (i, j))
if not contflag:
break
+ if game.idebug:
+ prout("=== Placing base #%d in quadrant %s" % (i, w))
game.state.baseq.append(w)
game.state.galaxy[w.i][w.j].starbase = game.state.chart[w.i][w.j].starbase = True
# Position ordinary Klingon Battle Cruisers
prout(_("LEAVE AT ONCE, OR YOU WILL BE DESTROYED!"))
# Put in THING if needed
if thing == game.quadrant:
- Enemy(type='?', loc=dropin(),
+ Enemy(etype='?', loc=dropin(),
power=randreal(6000,6500.0)+250.0*game.skill)
if not damaged(DSRSENS):
skip(1)
w.j = withprob(0.5) * (QUADSIZE-1)
if game.quad[w.i][w.j] == '.':
break
- game.tholian = Enemy(type='T', loc=w,
+ game.tholian = Enemy(etype='T', loc=w,
power=randrange(100, 500) + 25.0*game.skill)
# Reserve unoccupied corners
if game.quad[0][0]=='.':
game.options |= OPTION_TTY
seed = int(time.time())
(options, arguments) = getopt.getopt(sys.argv[1:], "r:s:txV")
+ replay = False
for (switch, val) in options:
if switch == '-r':
try:
raise SystemExit, 1
try:
line = replayfp.readline().strip()
- (leader, _, seed) = line.split()
+ (leader, __, seed) = line.split()
seed = eval(seed)
sys.stderr.write("sst2k: seed set to %s\n" % seed)
line = replayfp.readline().strip()
arguments += line.split()[2:]
+ replay = True
except ValueError:
sys.stderr.write("sst: replay file %s is ill-formed\n"% val)
raise SystemExit(1)
if logfp:
logfp.write("# seed %s\n" % seed)
logfp.write("# options %s\n" % " ".join(arguments))
+ logfp.write("# SST2K version %s\n" % version)
logfp.write("# recorded by %s@%s on %s\n" % \
(getpass.getuser(),socket.gethostname(),time.ctime()))
random.seed(seed)
game.alldone = False
else:
makemoves()
+ if replay:
+ break
skip(1)
stars()
skip(1)
projects / super-star-trek.git / blobdiff