#!/bin/env python
#
# postsim.py - skim the results of a triplet polarimeter beam
#              simulation for events that pass a minimal trigger
#              requiring at least one triplet polarimeter hit
#              and a coincidence in the pair spectrometer.
#
# author: richard.t.jones at uconn.edu
# version: may 5, 2018

import hddm_s
import sys

count = 0
fout = hddm_s.ostream(sys.argv[2])
for r in hddm_s.istream(sys.argv[1]):
   if len(r.getTpolTruthHits()) == 0:
      continue
   pscarm = [0,0]
   for p in r.getPsTiles():
      if p.arm == 0:
         pscarm[0] += 1
      elif p.arm == 1:
         pscarm[1] += 1
   if pscarm[0] == 0 or pscarm[1] == 0:
      continue
   fout.write(r)
   count += 1
print count