Skip to content

Commit

Permalink
[util] pep8
Browse files Browse the repository at this point in the history
  • Loading branch information
@Sec42
Sec42 committed Jun 16, 2024
1 parent e10bfc4 commit a7ffad3
Showing 1 changed file with 55 additions and 47 deletions.
102 changes: 55 additions & 47 deletions util.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -13,32 +13,35 @@ def utcoffset(self, dt):
return datetime.timedelta(0)
def dst(self, dt):
return datetime.timedelta(0)
def tzname(self,dt):
def tzname(self, dt):
return "Z"

Z=Zulu()
Z = Zulu()

def myhex(data, sep):
return sep.join(["%02x"%(x) for x in data])

# XXX: our own slice() is in the way.
slice_=slice
slice_ = slice

#fallback class for python3 <3.8
class mybytes(bytes):
def hex(self, sep=''):
return sep.join(["%02x"%(x) for x in self])

def __getitem__(self, key):
if isinstance(key, slice_):
return mybytes(super().__getitem__(key))
else:
return super().__getitem__(key)

def __add__(self, key):
return mybytes(super().__add__(key))

class mybytearray(bytearray):
def hex(self, sep=''):
return sep.join(["%02x"%(x) for x in self])

def __getitem__(self, key):
if isinstance(key, slice_):
return mybytearray(super().__getitem__(key))
Expand All @@ -57,7 +60,7 @@ def fmt_iritime(iritime):
uxtime= float(iritime)*90/1000+1399818235
if uxtime>1435708799: uxtime-=1 # Leap second: 2015-06-30 23:59:59
if uxtime>1483228799: uxtime-=1 # Leap second: 2016-12-31 23:59:59
strtime=datetime.datetime.fromtimestamp(uxtime,tz=Z).strftime("%Y-%m-%dT%H:%M:%S.{:02.0f}Z".format((uxtime%1)*100))
strtime = datetime.datetime.fromtimestamp(uxtime,tz=Z).strftime("%Y-%m-%dT%H:%M:%S.{:02.0f}Z".format((uxtime%1)*100))
return (uxtime, strtime)

def grouped(iterable, n):
Expand All @@ -71,17 +74,20 @@ def remove_zeros(l):
else:
break

def group(string,n): # similar to grouped, but keeps rest at the end
string=re.sub('(.{%d})'%n,'\\1 ',string)

def group(string, n): # similar to grouped, but keeps rest at the end
string = re.sub('(.{%d})'%n,'\\1 ', string)
return string.rstrip()

def slice_extra(string,n):
blocks=[string[x:x+n] for x in range(0,len(string)+1,n)]
extra=blocks.pop()

def slice_extra(string, n):
blocks = [string[x:x+n] for x in range(0, len(string)+1, n)]
extra = blocks.pop()
return (blocks,extra)

def slice(string,n):
return [string[x:x+n] for x in range(0,len(string),n)]

def slice(string, n):
return [string[x:x+n] for x in range(0, len(string),n)]

def to_ascii(data, dot=False, escape=False, mask=False):
str=""
Expand Down Expand Up @@ -109,45 +115,46 @@ def bitdiff(a, b):

def objprint(q):
for i in dir(q):
attr=getattr(q,i)
attr = getattr(q, i)
if i.startswith('_'):
continue
if isinstance(attr, types.MethodType):
continue
print("%s: %s"%(i,attr))
print("%s: %s"%(i, attr))


def curses_eol(file=sys.stderr):
import curses
curses.setupterm(fd=file.fileno())
el=curses.tigetstr('el')
cr=curses.tigetstr('cr') or b'\r'
nl=curses.tigetstr('nl') or b'\n'
el = curses.tigetstr('el')
cr = curses.tigetstr('cr') or b'\r'
nl = curses.tigetstr('nl') or b'\n'
if el is None:
eol= (nl).decode("ascii")
eolnl=(nl).decode("ascii")
eol = (nl).decode("ascii")
eolnl = (nl).decode("ascii")
else:
eol= (el+cr).decode("ascii")
eolnl=(el+nl).decode("ascii")
eol = (el+cr).decode("ascii")
eolnl = (el+nl).decode("ascii")
return eol

# extract position (x/y/z) from 5 bytes
# skip=(0-4) bits at the beginning
def xyz(data, skip=0):
val=int(data[0:5].hex(),16)
sb=4-skip
val = int(data[0:5].hex(), 16)
sb = 4-skip

loc_x=(val>>(12*2+sb)) & 0xfff
if loc_x > 0x800: loc_x= -(0x1000 - loc_x)
loc_x = (val>>(12*2+sb)) & 0xfff
if loc_x > 0x800: loc_x = -(0x1000 - loc_x)

loc_y=(val>>(12*1+sb)) & 0xfff
if loc_y > 0x800: loc_y= -(0x1000 - loc_y)
loc_y = (val>>(12*1+sb)) & 0xfff
if loc_y > 0x800: loc_y = -(0x1000 - loc_y)

loc_z=(val>>(12*0+sb)) & 0xfff
if loc_z > 0x800: loc_z= -(0x1000 - loc_z)
loc_z = (val>>(12*0+sb)) & 0xfff
if loc_z > 0x800: loc_z = -(0x1000 - loc_z)

# From bitsparser.py: ad-hoc quick conversion
lat = atan2(loc_z,sqrt(loc_x**2+loc_y**2))*180/pi
lon = atan2(loc_y,loc_x)*180/pi
lat = atan2(loc_z, sqrt(loc_x**2+loc_y**2))*180/pi
lon = atan2(loc_y, loc_x)*180/pi
alt = sqrt(loc_x**2+loc_y**2+loc_z**2)*4

return dict(x=loc_x, y=loc_y, z=loc_z, lat=lat, lon=lon, alt=alt)
Expand All @@ -156,21 +163,22 @@ def xyz(data, skip=0):
channel_width=1e7/(30*8) # 30 sub-bands with 8 "frequency accesses" each

def channelize(freq):
fbase=freq-base_freq
freq_chan=int(fbase/channel_width)
foff =fbase%channel_width
freq_off=foff-(channel_width/2)
return (freq_chan,freq_off)
fbase = freq - base_freq
freq_chan = int(fbase / channel_width)
foff = fbase%channel_width
freq_off = foff - (channel_width/2)
return (freq_chan, freq_off)


def channelize_str(freq):
fbase=freq-base_freq
freq_chan=int(fbase/channel_width)
sb=int(freq_chan/8)+1
fa=(freq_chan%8)+1
sx=freq_chan-30*8+1
foff =fbase%channel_width
freq_off=foff-(channel_width/2)
if sb>30:
fbase = freq-base_freq
freq_chan = int(fbase / channel_width)
sb = int(freq_chan/8)+1
fa = (freq_chan % 8)+1
sx = freq_chan-30*8+1
foff = fbase%channel_width
freq_off = foff-(channel_width/2)
if sb > 30:
return f"S.{sx:02}|{freq_off:+06.0f}"
else:
return f"{sb:02}.{fa:1}|{freq_off:+06.0f}"
Expand Down Expand Up @@ -208,14 +216,14 @@ def parse_handoff(lcwstr):

def parse_channel(fstr):
if "|" in fstr:
chan, off=fstr.split('|')
chan, off = fstr.split('|')
if '.' in chan:
sb, fa=chan.split('.')
sb, fa = chan.split('.')
frequency = get_channel(sb, fa, strict=False) + int(off)
else:
frequency=base_freq+channel_width*int(chan)+int(off)+channel_width/2
frequency = base_freq+channel_width*int(chan)+int(off)+channel_width/2
else:
frequency=int(fstr)
frequency = int(fstr)
return int(frequency)

class dt(datetime.datetime):
Expand Down

0 comments on commit a7ffad3

Please sign in to comment.