make_template_knob_config.py

Go to the documentation of this file.

#!/usr/bin/python

# This script outputs a template table of all tunable GENIE knobs pertaining to the tag release indicated by the SAM definition name used for producing the files
# One has to setup_nova before using this script since this script utilizes ROOT
# Usage:
# $ setup_nova
# $ python make_template_knob_config.py -d <sam_defname>

from ROOT import *
import argparse
import collections
import json
import subprocess, os, sys

def setknob(knobname, value):
  if knobname in kconf:
    kconf[knobname] = value

# use production 3 ND MC as the default dataset
default_dataset = 'prod_caf_R17-03-01-prod3reco.d_nd_genie_nonswap_fhc_nova_v08_full_v1'

# command line parser
parser = argparse.ArgumentParser(description='script to make a template configuration file of all tunable GENIE knobs pertaining to the tag release indicated by the SAM definition name used for producing the files')
parser.add_argument('-d','--defname',type=str,default=default_dataset)
parser.add_argument('--printout', dest='printout', action='store_true')
parser.add_argument('--no-printout', dest='printout', action='store_false')
parser.set_defaults(printout=True)
args = parser.parse_args()

# remind of dataset used
if default_dataset == args.defname:
  print '********** NOTE **********'
print 'SAM definition being used for generating the config file is'
print args.defname
if default_dataset == args.defname:
  print 'It can be changed with -d command line option.'

# check if the given SAM definition name contains a valid tag release
tag_found = False
found_tag = ''
(stdout, stderr) = subprocess.Popen('svn list svn+ssh://p-novaart@cdcvs.fnal.gov/cvs/projects/novaart/pkgs.svn/tags', shell=True, stdout=subprocess.PIPE).communicate()
for tag in stdout.split('\n'):
  if tag.rstrip('/') in args.defname:
    tag_found = True
    found_tag = tag.rstrip('/')
    break
if not tag_found:
  print 'The tag version indicated by the input SAM definition name is not found.'
  sys.exit()


# open the source file containing the GENIE knobs and store the knob names in a list
(stdout, stderr) = subprocess.Popen('source /grid/fermiapp/nova/novaart/novasvn/setup/setup_nova.sh -r '+found_tag+' &>/dev/null;printenv', shell=True, stdout=subprocess.PIPE).communicate()

# fill environment variable dictionary
envdict = dict()
for line in stdout.split('\n'):
  if len(line.split('=')) >= 2:
    envdict[line.split('=')[0]] = line.split('=')[1].rstrip('\n')

knoblist = []
with open(os.path.join(envdict['NUGEN_DIR'],'source/NuReweight/ReweightLabels.h')) as fsrc:
  for line in fsrc:
    if 'genie::rew::' in line:
      if len(line.split('=')) >= 2:
        knoblist.append(line.split('=')[0].strip().replace('fReweight','').replace('kReweight',''))

if(args.printout):
  print 'all available knobs:\n', '='*80
  for knobname in knoblist: sys.stdout.write(knobname+' ')
print '\n'+'='*80

# It can happen that the number of knobs used in CAF files is different from
# that in nugen. What follows is to get the number used in CAF.
(stdout, stderr) = subprocess.Popen('source /grid/fermiapp/nova/novaart/novasvn/setup/setup_nova.sh -r '+found_tag+' &>/dev/null; samweb2xrootd `samweb list-files "defname: '+args.defname+' with limit 1"`', shell=True, stdout=subprocess.PIPE).communicate()
tfilepn = stdout.rstrip('\n')
if args.printout:
  print 'Opening CAF file:', tfilepn

# setup CAF dictionary
gSystem.Load("libCintex.so")
Cintex.Enable()
gSystem.Load(os.path.join(envdict['SRT_PUBLIC_CONTEXT'],'lib/Linux2.6-GCC-debug/libStandardRecord_dict.so'))
gInterpreter.AddIncludePath(envdict['SRT_PUBLIC_CONTEXT'])

tfile = TFile.Open(tfilepn)
recTree = tfile.Get('recTree')
nknobs = 0
for entry in recTree:
  if entry.rec.mc.nnu <= 0: continue
  nknobs = len(entry.rec.mc.nu[0].rwgt.genie)
  if nknobs: break

# knob configuration
kconf = collections.OrderedDict()
for i in range(0, nknobs):
  kconf[knoblist[i]] = 1

## knobs that need extra care
setknob('Null', 0)
setknob('NormCCQEenu', 0)
setknob('MaCCQE', 0)
# by default, disable z expansion knobs
setknob('ZNormCCQE', 0)
setknob('ZExpA1CCQE', 0)
setknob('ZExpA2CCQE', 0)
setknob('ZExpA3CCQE', 0)
setknob('ZExpA4CCQE', 0)
setknob('AxFFCCQEshape', 0)

# output configuration to file
#~ with open('knob_config.json', 'w') as outfile:
  #~ json.dump(kconf, outfile, indent=2)

# output configuration to file
# Due to a bug in the C++ boost library, I will use plain text format instead.
# Boost property tree v1_57_0a when compiled with c++11 results in a
# compile time error needing a patch.
# See https://svn.boost.org/trac/boost/ticket/6785 for more information.
outfilename = 'knob_config.txt'
print 'Writing output file to '+outfilename
with open(outfilename, 'w') as outfile:
  for k,v in kconf.items():
    outfile.write(k+' '+str(v)+'\n')
print 'Done'