Mercurial > hg > python

annotate util.py @ 26:5488b5d3ba10

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
use indentation and colour to differentiate safe from unsafe landing place
author Henry Thompson <ht@markup.co.uk>
date Mon, 31 May 2021 15:01:38 +0100
parents
children 668d788cac47
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
26
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
1 '''Utility fns for dealing with block .tsv files'''
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
2 import math,re
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
3
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
4 IPAT=re.compile('-?[0-9][0-9]*$')
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
5 PPAT=re.compile('[ ,;\[\]]')
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
6
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
7 Red=''
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
8 eRed=''
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
9 RedFmt=Red+'%s'+eRed
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
10
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
11 def intsMaybe(ii):
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
12 return (int(i) for i in ii if IPAT.match(i))
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
13
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
14 def readHeaders(host,infile,skipColHdrs=True):
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
15 l=infile.readline().rstrip()
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
16 ff=PPAT.split(l)
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
17 (host.nr,host.ox,host.oy,host.oz)=intsMaybe(ff)
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
18 host.et=ff[9]
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
19 l=infile.readline().rstrip()
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
20 (host.orad,host.ymin,host.ymax)=intsMaybe(PPAT.split(l))
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
21 if skipColHdrs:
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
22 _=infile.readline()
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
23
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
24 def d(p1,p2):
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
25 dx=p1[0]-p2[0]
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
26 dz=p1[1]-p2[1]
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
27 dy=p1[2]-p2[2]
5488b5d3ba10 use indentation and colour to differentiate safe from unsafe landing place
Henry Thompson <ht@markup.co.uk>
parents:
diff changeset
28 return math.sqrt((dx*dx)+(dy*dy)+(dz*dz))