diff python/util.py @ 4:56508a6033a9

minutor chunk hacking
author Henry S. Thompson <ht@inf.ed.ac.uk>
date Tue, 25 May 2021 14:01:26 -0400
parents
children e411408d64ec
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/python/util.py	Tue May 25 14:01:26 2021 -0400
@@ -0,0 +1,24 @@
+'''Utility fns for dealing with block .tsv files'''
+import math,re
+
+IPAT=re.compile('-?[0-9][0-9]*$')
+PPAT=re.compile('[ ,;\[\]]')
+
+def intsMaybe(ii):
+  return (int(i) for i in ii if IPAT.match(i))
+
+def readHeaders(host,infile,skipColHdrs=True):
+    l=infile.readline().rstrip()
+    ff=PPAT.split(l)
+    (host.nr,host.ox,host.oy,host.oz)=intsMaybe(ff)
+    host.et=ff[9]
+    l=infile.readline().rstrip()
+    (host.orad,host.ymin,host.ymax)=intsMaybe(PPAT.split(l))
+    if skipColHdrs:
+      _=infile.readline()
+
+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))