view python/safe.py @ 9:0d1670ab37df

towards useful output
author Henry S. Thompson <ht@inf.ed.ac.uk>
date Wed, 26 May 2021 13:52:53 -0400
parents fcef94b6324c
children 73bb35b96624
line wrap: on
line source

#!/usr/bin/python3
'''Find a safe landing space (i.e. in air, not above lava) near a cluster'''
import sys, math

from util import *
from bisect import insort_left as insort

def usage():
  print("""Usage:  safe.py [-d n]
  n is maximum distance of landing to some point in the cluster, default is 10.
  
  Feed stdin with lines from the output of cluster.py, having produced
  air.tsv with the same settings as the cluster input, and lava.tsv
  with a lower-bound on Y of 0""")

if len(sys.argv)>1:
  if sys.argv.pop(1)!='-d':
    usage()
    exit(1)
  n=float(sys.argv.pop(1))
else:
  n=10.0

class Block:
  '''Shared by Air and Lava,
  holds points and columns.
  Columns are dictionaries indexed by (x,z) coords,
  value is upward-ordered list of pairs, closed intervals of y'''
  def __init__(self,filename):
    with open(filename,'r') as file:
      self.readHeaders(file)
      self.readPoints(file)

    cc=self.columns={}
    tc={}
    for p in self.points:
      k=(p[0],p[2])
      y=p[1]
      try:
        yy=tc[k]
        insort(yy,y)
      except KeyError:
        tc[k]=[y]
    # convert to lists of intervals
    # Simplest case first
    for k,yy in tc.items():
      tyy=[yy]
      ii=[]
      while True:
        clean=True
        for j,yy in enumerate(tyy):
          if len(yy)==1+(yy[-1]-yy[0]):
            ii.append((yy[0],yy[-1]))
          else:
            clean=False
            for i in range(len(yy)-1):
              if yy[i]+1!=yy[i+1]:
                ii+=(yy[0],yy[i])
                tyy=[yy[i+1:]]+tyy[j+1:]
            break
        if clean:
          break
      self.columns[k]=ii

  def readPoints(self,file):
    file.readline()
    self.points=[[float(i) for i in l.split('\t')[2].split(',')] for l in file]

Block.readHeaders=readHeaders # from util

class Air(Block):
  pass

A=Air('air.tsv')

class Lava(Block):
  pass

L=Lava('lava.tsv')

def d(p1,p2):
  dx=p1[0]-p2[0]
  dz=p1[1]-p2[1]
  dy=p1[2]-p2[2]
  return math.sqrt((dx*dx)+(dy*dy)+(dz*dz))

def safety(p):
  ka=[(a[0],a[2]) for a in A.points if d(a,p)<=n]
  return [(k,A.columns[k],L.columns[k]) for k in ka if k in L.columns]

readHeaders(sys.modules['__main__'],sys.stdin,False)
for l in sys.stdin:
  c=eval(l)
  ss=[]
  hits={}
  misses=[]
  for p in c:
    s=safety(p)
    if s:
      pass
    else:
      misses.append(p)
  print(c)
  print(' %s nearby landing zones'%(
    (lambda nm,nc:'No' if nm==nc else nc-nm)(len(misses),len(c))))