send_merge_ntuples.py

Go to the documentation of this file.

#!/usr/bin/env python

###### Script to send jobs to merge ntuples
import os, sys, subprocess
import string
from utilities import *

# Setting folders
folder    = '/net/cms2/cms2r0/babymaker/babies/2016_06_14/mc/'
skims = ['standard', 'baseline', 'nl2nb2nj3', 'qcd', 'dy_ht300']


os.system("JobSetup.csh")
for skim in skims:
  # Checking if input folder exists and creating output folder
  infolder = folder+"/skim_"+skim+"/"
  if not os.path.exists(infolder):
    print "Skim "+infolder+" does not exist. Skipping"
    continue
  outfolder = folder+"/merged_"+skim+"/"
  if not os.path.exists(outfolder):
    os.system("mkdir -p "+outfolder)

  ## Finding tags for each dataset
  infiles = findBaseSampleNames(infolder)

  # Sending jobs for each tag
  runfolder = outfolder+"run/" 
  if not os.path.exists(runfolder):
    os.system("mkdir -p "+runfolder)
  ijob = 0
  for ifile in infiles:
    ijob += 1
    exename = runfolder+"/merge_ntuples_"+str(ijob)+".sh"
    fexe = open(exename,"w")
    os.system("chmod u+x "+exename)
    fexe.write("#!/bin/bash\n\n")
    fexe.write("./run/merge_ntuples.exe "+infolder+' '+outfolder+' '+ifile+' '+skim+'\n')
    fexe.close()
    cmd = "JobSubmit.csh ./run/wrapper.sh "+exename
    os.system(cmd)
    
  print "\nSubmitted "+str(len(infiles))+" merging jobs. Output goes to "+outfolder+"\n"

sys.exit(0)