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.
123 lines
3.8 KiB
Python
123 lines
3.8 KiB
Python
1 year ago
|
import re
|
||
|
|
|
||
|
|
"""
|
||
|
|
Plugin handler for the miscScript plugin.
|
||
|
|
"""
|
||
|
|
class SeedlinkPluginHandler:
|
||
|
|
# Create defaults
|
||
|
|
def __init__(self):
|
||
|
|
self.instances = {}
|
||
|
|
|
||
|
|
def push(self, seedlink):
|
||
|
|
sta = seedlink.param("seedlink.station.id")
|
||
|
|
|
||
|
|
try:
|
||
|
|
key = sta + "." + str(self.instances[sta])
|
||
|
|
self.instances[sta] += 1
|
||
|
|
except KeyError:
|
||
|
|
key = sta + ".0"
|
||
|
|
self.instances[sta] = 1
|
||
|
|
|
||
|
|
# Check and set defaults
|
||
|
|
try:
|
||
|
|
seedlink.param("sources.miscScript.script_path")
|
||
|
|
except:
|
||
|
|
seedlink.setParam("sources.miscScript.script_path", "")
|
||
|
|
|
||
|
|
try:
|
||
|
|
seedlink.param("sources.miscScript.script_args")
|
||
|
|
except:
|
||
|
|
seedlink.setParam(
|
||
|
|
"sources.miscScript.script_args", "default"
|
||
|
|
) # If no argument, set 'default'
|
||
|
|
|
||
|
|
try:
|
||
|
|
seedlink.param("sources.miscScript.proc")
|
||
|
|
except:
|
||
|
|
seedlink.setParam("sources.miscScript.proc", "auto")
|
||
|
|
|
||
|
|
try:
|
||
|
|
seedlink.param("sources.miscScript.sample_frequency")
|
||
|
|
except:
|
||
|
|
seedlink.setParam("sources.miscScript.sample_frequency", "1")
|
||
|
|
|
||
|
|
freq = seedlink.param("sources.miscScript.sample_frequency")
|
||
|
|
|
||
|
|
if re.match("[0-9]+$", freq) != None:
|
||
|
|
seedlink.setParam("sources.miscScript.sample_period", str(1.0 / int(freq)))
|
||
|
|
else:
|
||
|
|
res = re.match("([0-9]+)/([0-9]+)$", freq)
|
||
|
|
if res != None:
|
||
|
|
seedlink.setParam(
|
||
|
|
"sources.miscScript.sample_period",
|
||
|
|
str(float(res.group(2)) / float(res.group(1))),
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
print("Sample frequency invalid !!!")
|
||
|
|
raise Exception
|
||
|
|
|
||
|
|
try:
|
||
|
|
seedlink.param("sources.miscScript.channels")
|
||
|
|
except:
|
||
|
|
seedlink.setParam("sources.miscScript.channels", "HHZ,HHN,HHE")
|
||
|
|
|
||
|
|
splitted_chans = seedlink.param("sources.miscScript.channels").split(",")
|
||
|
|
seedlink.setParam("sources.miscScript.channelsNumber", len(splitted_chans))
|
||
|
|
|
||
|
|
try:
|
||
|
|
seedlink.param("sources.miscScript.flush_period")
|
||
|
|
except:
|
||
|
|
seedlink.setParam("sources.miscScript.flush_period", "0")
|
||
|
|
|
||
|
|
##### Auto-generate proc conf and channel/source_id mapping
|
||
|
|
|
||
|
|
if seedlink.param("sources.miscScript.proc") == "auto":
|
||
|
|
seedlink.setParam("sources.miscScript.proc", "auto:miscScript_%s" % key)
|
||
|
|
trees = ""
|
||
|
|
channels = ""
|
||
|
|
idx = 0
|
||
|
|
|
||
|
|
for chan in splitted_chans:
|
||
|
|
chan = chan.strip()
|
||
|
|
|
||
|
|
if chan == "none":
|
||
|
|
idx += 1
|
||
|
|
continue
|
||
|
|
|
||
|
|
elif len(chan) == 3:
|
||
|
|
location_val = "00"
|
||
|
|
stream_val = chan[0:2]
|
||
|
|
channel_val = chan[2]
|
||
|
|
|
||
|
|
elif len(chan) == 5:
|
||
|
|
location_val = chan[0:2]
|
||
|
|
stream_val = chan[2:4]
|
||
|
|
channel_val = chan[4]
|
||
|
|
|
||
|
|
else:
|
||
|
|
print("Invalid channel name")
|
||
|
|
raise Exception
|
||
|
|
|
||
|
|
trees += " <tree>\n"
|
||
|
|
trees += """ <input name="{}" channel="{}" location="{}" rate="{}"/>\n""".format(
|
||
|
|
idx,
|
||
|
|
channel_val,
|
||
|
|
location_val,
|
||
|
|
seedlink.param("sources.miscScript.sample_frequency"),
|
||
|
|
)
|
||
|
|
trees += """ <node stream="{}"/>\n""".format(stream_val)
|
||
|
|
trees += " </tree>\n"
|
||
|
|
|
||
|
|
channels += "channel {} source_id={}\n".format(idx, idx)
|
||
|
|
|
||
|
|
idx += 1
|
||
|
|
|
||
|
|
seedlink.setParam("sources.miscScript.trees", trees)
|
||
|
|
seedlink.setParam("sources.miscScript.channels", channels)
|
||
|
|
|
||
|
|
return key
|
||
|
|
|
||
|
|
# Flush does nothing
|
||
|
|
def flush(self, seedlink):
|
||
|
|
pass
|