MVAConfigBuilder.py

Go to the documentation of this file.

#!/usr/bin/env python

"""
MVAConfigBuilder.py

Author: Evan K. Friis, UC Davis friis@physics.ucdavis.edu

Usage: Any objects with file type [name].mvac will be turned into [name].xml with the correct format

A configuration script to automate changes in available MVA input variables (for the PhysicsTools/MVAComputer framework)
across numerous MVA configurations.

This is total kludge and will be deprecated when 
        a. I figure out how you use <xi:include... success full (xml sucks)
        b. Proposed changes to the MVA objects allow classification steering in a single xml file
"""
from __future__ import print_function

import os
import glob
import string

filesToConvert = glob.glob("./*.mvac")

RemoveNumEventsSpecifiers = True

for aFile in filesToConvert:
   xmlFileName = string.replace(aFile, "mvac", "xml")
   if os.path.exists(xmlFileName):
      os.system("cp %s %s.bak" % (xmlFileName, xmlFileName))
      os.remove(xmlFileName)

   print("Building %s..." % xmlFileName)
   os.system("cat Preamble.xml.fragment >  %s" %         xmlFileName)  
   os.system("cat Inputs.xml.fragment   >> %s" %         xmlFileName)  
   os.system("cat Helpers.xml.fragment  >> %s" %         xmlFileName)  
   if RemoveNumEventsSpecifiers:
      os.system("cat %s | sed 's|NSigTest=[0-9]*|NSigTest=0|' | sed 's|NBkgTest=[0-9]*|NBkgTest=0|' | sed 's|NSigTrain=[0-9]*|NSigTrain=0|' | sed 's|NBkgTrain=[0-9]*|NBkgTrain=0|' >> %s" % (aFile, xmlFileName)) 
   else:
      os.system("cat %s                    >> %s" % (aFile, xmlFileName)) 
   os.system("cat Finale.xml.fragment   >> %s" %         xmlFileName)