import logging
import sys

from obspy import read_inventory, read_events


def count_picks_from_stations(quakeml_file, inventory):

    # Read the QuakeML file
    try:
        events = read_events(quakeml_file)
    except Exception as e:
        logging.error(f"Error reading QuakeML file: {e}")
        sys.exit(1)

    # Extract station network and station codes from the inventory
    stations = set()
    for network in inventory:
        for station in network:
            stations.add((network.code, station.code))

    # Count the number of picks in the QuakeML file that come from the stations in the inventory
    pick_count = 0
    for event in events:
        for pick in event.picks:
            net_sta = (pick.waveform_id.network_code, pick.waveform_id.station_code)
            if net_sta in stations:
                pick_count += 1

    return pick_count