import sys
import os
import re
import time
from optparse import OptionParser
from ProdSample import ProdSample
from ProdSample import aod_selector
from ProdSample import esd_selector
from ProdSample import rdo_selector
from ProdSample import hit_selector

clparser = OptionParser()
clparser.add_option("-d", "--debug", dest="debug",
                    help="debug", default=0)
clparser.add_option("-b", "--block_exceptions", action="store_true", dest="block_exceptions",
                    help="debug", default=0)
clparser.add_option("-f", "--formats", action="store_true", dest="formats", 
                    help="debug", default=0)
clparser.add_option("-o", "--show_overlaps", action="store_true", dest="show_overlaps",
                    help="show overlaps", default=0)
(clopts, clargs) = clparser.parse_args()
intask=clargs[0]
dump = len(clargs)>1
print "Checking sample " + intask

print
sam = ProdSample(intask)
sam.dbg = clopts.debug
if clopts.block_exceptions:
  sam.check_duplicates_are_aborted = False
formats = clopts.formats
sam.fetch_status()
if dump: sam.dump()
sam.show_status()

if clopts.show_overlaps:
  sam.show_task_overlaps()

if formats:
  for sel in [ hit_selector, rdo_selector, esd_selector, aod_selector ]:
    selsam = sam.select(sel)
    print
    if selsam.exists() and selsam.ntask:
      if dump: selsam.dump()
      selsam.show_status()
    elif selsam.ntask == 0:
      print "No " + sel.label() + " tasks found"
    else:
      print "Task not found: " + sel.label()