"""The counted flares the primary analysis lists, with their extents, for localize.py.

python counted.py <results dir> <run dir> <out.csv>
"""
import json
import os
import sys

import pandas as pd

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from analyze import table  # noqa: E402

res, run, out = sys.argv[1], sys.argv[2], sys.argv[3]
flares = pd.DataFrame(json.load(open(os.path.join(res, 'primary.json')))['flares'])
cands = table(run, 'cands')
cands = cands[cands.kind == 'flare'][['file', 'tpk', 't0', 't1']]
if len(flares):
    flares = flares[['file', 'tpk']].merge(cands, on=['file', 'tpk'])
else:
    flares = pd.DataFrame(columns=['file', 'tpk', 't0', 't1'])
flares.to_csv(out, index=False)
print(len(flares), 'counted flares')