diff python/safe.py @ 7:3ee329b129c6

rename block properties, slightly revise result format
author Henry S. Thompson <ht@inf.ed.ac.uk>
date Wed, 26 May 2021 09:18:49 -0400
parents a01ff74f9fd7
children fcef94b6324c
line wrap: on
line diff
--- a/python/safe.py	Tue May 25 17:20:04 2021 -0400
+++ b/python/safe.py	Wed May 26 09:18:49 2021 -0400
@@ -30,8 +30,8 @@
     with open(filename,'r') as air:
       Block.__init__(self,air)
       air.readline()
-      aa=self.aa=[[float(i) for i in l.split('\t')[2].split(',')] for l in air]
-    ad=self.ad={}
+      aa=self.blocks=[[float(i) for i in l.split('\t')[2].split(',')] for l in air]
+    ad=self.columns={}
     for a in aa:
       k=(a[0],a[2])
       h=ad.get(k,500.0)
@@ -45,12 +45,12 @@
     with open(filename,'r') as lava:
       Block.__init__(self,lava)
       lava.readline()
-      ll=self.ll={}
+      ll=self.columns={}
       for l in lava:
         x=[float(i) for i in l.split('\t')[2].split(',')]
         k=(x[0],x[2])
         kk=ll.setdefault(k,list())
-        kk.append(x)
+        kk.append(x[1])
   
 L=Lava('lava.tsv')
 
@@ -60,14 +60,14 @@
   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.aa if d(a,p)<=n]
-  return [(k,ll[k]) for k in ka if k in L.ll]
+def unsafe(p):
+  ka=[(a[0],a[2]) for a in A.blocks 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)
-  s=[(p,safety(p)) for p in c]
+  s=[(p,unsafe(p)) for p in c]
   print(s if s else 'No air',c)