import glob
import os

from obspy import read, UTCDateTime, Stream
from obspy.signal.trigger import coincidence_trigger
from pprint import pprint


dta = read('./data/seeds/KLSM.mseed')
tim_ini = UTCDateTime("2025-10-26T10:44:12.4")
tim_fin = UTCDateTime("2025-10-26T10:47:18.7")
trace_ids = {}
event_templates = {"KLSM": [dta.slice(tim_ini, tim_fin)]}
similarity_thresholds = {"KLSM": 0.1}
fil = glob.glob(os.path.join("data/seeds", "*.mseed"))
stt = Stream()
for n in fil:
    trc = read(n)
    trc_net = trc.traces[0].stats.network
    trc_nam = trc.traces[0].stats.station
    trc_loc = "" if (o := trc.traces[0].stats.location) == "Null" else o
    trc_cha = trc.traces[0].stats.channel
    stt += trc
    trace_ids[f"{trc_net}.{trc_nam}.{trc_loc}.{trc_cha}"] = 1
trig = coincidence_trigger(
    "classicstalta",
    5,
    1,
    stt,
    4,
    sta=0.5,
    lta=10,
    trace_ids=trace_ids,
    event_templates=event_templates,
    similarity_threshold=similarity_thresholds
)
pprint(trig)