You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
4.0 KiB
Python
121 lines
4.0 KiB
Python
1 year ago
|
import os, time
|
||
|
|
||
|
'''
|
||
|
Plugin handler for the Earthworm export plugin.
|
||
|
'''
|
||
|
class SeedlinkPluginHandler:
|
||
|
# Create defaults
|
||
|
def __init__(self):
|
||
|
self.instances = {}
|
||
|
self.channelMap = {}
|
||
|
self.idMap = {}
|
||
|
|
||
|
def push(self, seedlink):
|
||
|
# Check and set defaults
|
||
|
host = seedlink._get('sources.scream_ring.address')
|
||
|
try: tcpport = int(seedlink.param('sources.scream_ring.tcpport'))
|
||
|
except: tcpport = 1567
|
||
|
try: port = int(seedlink.param('sources.scream_ring.port'))
|
||
|
except: port = 1567
|
||
|
|
||
|
seedlink.setParam('sources.scream_ring.tcpport', tcpport);
|
||
|
seedlink.setParam('sources.scream_ring.port', port);
|
||
|
|
||
|
try: rsize = int(seedlink.param('sources.scream_ring.rsize'))
|
||
|
except: rsize = 1000
|
||
|
|
||
|
seedlink.setParam('sources.scream_ring.rsize', rsize);
|
||
|
|
||
|
try:
|
||
|
if seedlink.param('sources.scream_ring.tcp').lower() in ("yes","true","1"):
|
||
|
seedlink.setParam('sources.scream_ring.tcpFlag',' -tcp')
|
||
|
else:
|
||
|
seedlink.setParam('sources.scream_ring.tcpFlag','')
|
||
|
except:
|
||
|
seedlink.setParam('sources.scream_ring.tcpFlag','')
|
||
|
|
||
|
key = (host, tcpport, port, seedlink.param('sources.scream_ring.tcpFlag'))
|
||
|
|
||
|
try:
|
||
|
screamId = self.instances[key]
|
||
|
|
||
|
except KeyError:
|
||
|
screamId = len(self.instances)
|
||
|
self.instances[key] = screamId
|
||
|
self.channelMap[screamId] = []
|
||
|
|
||
|
|
||
|
try:
|
||
|
channelItems = [ x.strip() for x in seedlink.param('sources.scream_ring.channels').split(',') ]
|
||
|
map = os.path.join(seedlink.config_dir, "scream_ring2sl%d.map" % screamId)
|
||
|
seedlink.setParam('sources.scream_ring.mapFlag',map)
|
||
|
|
||
|
except KeyError:
|
||
|
try:
|
||
|
map = seedlink.param('sources.scream_ring.map')
|
||
|
if not os.path.isabs(map):
|
||
|
map = os.path.join(seedlink.config_dir, map)
|
||
|
except: map = os.path.join(seedlink.config_dir, 'scream2sl.map')
|
||
|
|
||
|
seedlink.setParam('sources.scream_ring.mapFlag',map)
|
||
|
channelItems = []
|
||
|
|
||
|
for item in channelItems:
|
||
|
mapping = [x.strip() for x in item.split(':')]
|
||
|
if len(mapping) != 2:
|
||
|
raise Exception("Error: invalid channel mapping '%s' in %s" % (item, seedlink.station_config_file))
|
||
|
if not mapping[0] or not mapping[1]:
|
||
|
raise Exception("Error: invalid channel mapping '%s' in %s" % (item, seedlink.station_config_file))
|
||
|
|
||
|
# Prepend current station code if not explicitely given
|
||
|
if not " " in mapping[1]:
|
||
|
mapping[1] = seedlink._get('seedlink.station.code') + " " + mapping[1]
|
||
|
|
||
|
mapping[1] = [x.strip() for x in mapping[1].split()]
|
||
|
|
||
|
# Add network code
|
||
|
mapping.insert(1, seedlink._get('seedlink.station.network'))
|
||
|
|
||
|
if not mapping in self.channelMap[screamId]:
|
||
|
self.channelMap[screamId].append(mapping)
|
||
|
|
||
|
try:
|
||
|
if not mapping[1] in self.idMap[mapping[0]]:
|
||
|
self.idMap[mapping[0]].append(mapping[1])
|
||
|
except KeyError:
|
||
|
self.idMap[mapping[0]] = [mapping[1]]
|
||
|
|
||
|
return (key,map)
|
||
|
|
||
|
|
||
|
# Flush does nothing
|
||
|
def flush(self, seedlink):
|
||
|
for x in self.idMap.keys():
|
||
|
if len(self.idMap[x]) > 1:
|
||
|
raise Exception("Error: Scream plugin has multiple mappings for id(s) %s" % ", ".join(self.idMap.keys()))
|
||
|
|
||
|
cols = [0,0,0,0]
|
||
|
|
||
|
for x in self.channelMap.keys():
|
||
|
for c in self.channelMap[x]:
|
||
|
vals = [c[0], c[1], c[2][0], c[2][1]]
|
||
|
for i in range(len(vals)):
|
||
|
if len(vals[i]) > cols[i]: cols[i] = len(vals[i])
|
||
|
|
||
|
for x in self.channelMap.keys():
|
||
|
mappings = self.channelMap[x]
|
||
|
if len(mappings) == 0: continue
|
||
|
|
||
|
scream2slmap = os.path.join(seedlink.config_dir, "scream_ring2sl%d.map" % x)
|
||
|
|
||
|
fd = open(scream2slmap, "w")
|
||
|
|
||
|
line_id = 1
|
||
|
fd.write("# Generated at %s - Do not edit!\n" % time.strftime("%Y-%m-%d %H:%M:%S UTC", time.gmtime()))
|
||
|
for c in mappings:
|
||
|
fd.write("ChanInfo %s%s %s%s %s%s %s%s %d\n" % (c[0], " "*(cols[0]-len(c[0])), c[1], " "*(cols[1]-len(c[1])), c[2][0], " "*(cols[2]-len(c[2][0])), c[2][1], " "*(cols[3]-len(c[2][1])), line_id))
|
||
|
line_id = line_id + 1
|
||
|
|
||
|
fd.close()
|
||
|
|