#! /usr/bin/env python

"""\
%prog yodafile [flatfile]

Convert a YODA data file to a 'flat', more human-readable data format.
"""

import yoda, os, sys, optparse
from yoda.script_helpers import parse_x2y_args, filter_aos

parser = optparse.OptionParser(usage=__doc__)
parser.add_option("-m", "--match", dest="MATCH", metavar="PATT", default=None,
                  help="only write out histograms whose path matches this regex")
parser.add_option("-M", "--unmatch", dest="UNMATCH", metavar="PATT", default=None,
                  help="exclude histograms whose path matches this regex")

opts, args = parser.parse_args()
in_out = parse_x2y_args(args, ".yoda", ".dat")
if not in_out:
    sys.stderr.write("You must specify the YODA and FLAT file names\n")
    sys.exit(1)

for i, o in in_out:
    analysisobjects = yoda.readYODA(i)
    filter_aos(analysisobjects, opts.MATCH, opts.UNMATCH)
    yoda.writeFLAT(analysisobjects, o)
