Mercurial > hg > lib > markup
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))))