merge_histograms Namespace Reference

Functions
def	is_good_histo_file (myfile)
def	merge_histos (filevec, outfile)
def	merge_globes (filevec, outfile)

Variables
	scriptdir = os.path.abspath(sys.argv[0]+"/../..")
	Determine default g4lbne directory (the directory this script is in) More...
string	usage = "usage: %prog [options]"
	Setup parser that reads in options. More...
	parser = OptionParser(usage=usage)
	dest
	help
	default
	options
	args
	g4lbne_dir
	input_dir
	output_dir
	physics_list = options.physics_list
	version = options.version
	Determine G4LBNE Version. More...
string	file_prefix = "histos_g4lbne_"
	Print options. More...
string	input_flux_dir = options.input_dir+"/"
string	output_flux_dir = options.output_dir+"/"
list	plot_files = []
list	fastmc_files = []
list	globes_files = []

Function Documentation

def merge_histograms.is_good_histo_file

(

myfile

)

Definition at line 15 of file merge_histograms.py.

def is_good_histo_file(myfile):
    # Check that file is not a zombie
    if myfile.IsZombie():
        return False
    # Check that numu flux is not empty
    keys = myfile.GetListOfKeys()
    if "numu_flux" in keys:
        if myfile.Get("numu_flux").Integral()>0:
            return True
    if "numu_flux_forplots" in keys:
        if myfile.Get("numu_flux_forplots").Integral()>0:
            return True
    return False

def merge_histograms.merge_globes	(		filevec,
			outfile
	)

Definition at line 90 of file merge_histograms.py.

def merge_globes(filevec,outfile):

    if len(filevec)==0:
        print "merge_histograms.py, merge_globes(): No files to merge."
        return;

    n_files = 0
            
    total_flux = []
    for i in range(501):
        total_flux.append([0,0,0,0,0,0,0])

    #temppath = "/scratch/dune/"+os.getenv("USER")
    #if not os.path.exists(temppath):
    #    os.makedirs(temppath)

    print "merging",len(filevec),"globes files"
    last_time = datetime.datetime.now() 
    for i in range(0,len(filevec)):
        if i%50==0: 
            now = datetime.datetime.now()
            print "Merging file number",i,"; time since last print:",now-last_time
            last_time = now

        #tempfile = temppath+"/"+os.path.basename(filevec[i])
        #shutil.copyfile(filevec[i],tempfile)
        next_file = open(filevec[i])
        lines = next_file.readlines()
        next_file.close()
        #os.remove(tempfile)

        flux = []
        problem_found = (len(lines)!=501)
        for line in lines:

            if len(line.split())!=7: 
                problem_found = True
            elif float(line.split()[0])>1 and float(line.split()[0])<5 and (float(line.split()[1])+float(line.split()[2])+float(line.split()[3])+float(line.split()[4])+float(line.split()[5])+float(line.split()[6]))==0.0:
                # make sure fluxes aren't all zero (by looking in peak)
                    problem_found = True
            else:
                flux.append([float(line.split()[0]),float(line.split()[1]),float(line.split()[2]),float(line.split()[3]),float(line.split()[4]),float(line.split()[5]),float(line.split()[6])])
    
        if not problem_found:
            n_files = n_files+1       
            for j in range(501):
                total_flux[j][0] = flux[j][0]
                for k in range(1,7):
                    total_flux[j][k] += flux[j][k]
            
    print "Number of good globes files found:",n_files
    for i in range(501):
        for j in range(1,7):
            total_flux[i][j] = total_flux[i][j]/n_files

    out_file = open(outfile,"w");
    print "writign to "+outfile
    for i in range(501):
        for j in range(7):
            out_file.write(str(total_flux[i][j])+" ")
        out_file.write("\n")

    return

def merge_histograms.merge_histos	(		filevec,
			outfile
	)

Definition at line 29 of file merge_histograms.py.

def merge_histos(filevec,outfile):

    if len(filevec)==0:
        print "merge_hitsograms.py, mergehistos(): No files to merge."
        return;

    # Temporary staging area
    #temppath = "/dune/data2/users/"+os.getenv("USER")+"/scratch/"
    #if not os.path.exists(temppath):
    #    os.makedirs(temppath)

    #for i in range (0,len(filevec)):
    #    tempfile = temppath + "/" +os.path.basename(filevec[i])
    #    shutil.copyfile(filevec[i],tempfile)

    # find first good file
    first_file_index = -1
    
    for i in range(0,len(filevec)):
        #tempfile = temppath + "/" +os.path.basename(filevec[i])
        in_tfile = ROOT.TFile(filevec[i]);
        if is_good_histo_file(in_tfile):
            first_file_index = i
            break

    n_files = 1

    print "First good file: "+str(first_file_index)

    last_time = datetime.datetime.now() 
    # Look through first file to find list of histograms that we should merge
    output_histos = []
    for key in in_tfile.GetListOfKeys():
        if key.ReadObj().IsA().InheritsFrom("TH1"):
            output_histos.append(in_tfile.Get(key.GetName()).Clone())

    for i in range(first_file_index+1,len(filevec)):
        if i%50 == 0: 
            now = datetime.datetime.now()
            print "Merging file number",i,"; time to merge last 50 files:",now-last_time
            last_time = now
            
        #tempfile = temppath + "/" +os.path.basename(filevec[i])
        next_tfile = ROOT.TFile(filevec[i])
        if is_good_histo_file(next_tfile):
            n_files = n_files + 1
            for hist in output_histos:
                temphist = next_tfile.Get(hist.GetName())
                hist.Add(temphist)
        next_tfile.Close()
        #os.remove(tempfile)
    out_tfile = ROOT.TFile(outfile,"RECREATE");
    print "writing to "+outfile
    scale_factor = 1.0/n_files;
    for hist in output_histos:
        hist.Scale(scale_factor)
        hist.Write()
        
    return

Variable Documentation

merge_histograms.args

Definition at line 191 of file merge_histograms.py.

merge_histograms.default

Definition at line 175 of file merge_histograms.py.

merge_histograms.dest

Definition at line 174 of file merge_histograms.py.

list merge_histograms.fastmc_files = []

Definition at line 236 of file merge_histograms.py.

string merge_histograms.file_prefix = "histos_g4lbne_"

Print options.

Find files

Definition at line 228 of file merge_histograms.py.

merge_histograms.g4lbne_dir

Definition at line 195 of file merge_histograms.py.

list merge_histograms.globes_files = []

Definition at line 237 of file merge_histograms.py.

merge_histograms.help

Definition at line 175 of file merge_histograms.py.

merge_histograms.input_dir

Definition at line 197 of file merge_histograms.py.

string merge_histograms.input_flux_dir = options.input_dir+"/"

Definition at line 229 of file merge_histograms.py.

merge_histograms.options

Definition at line 191 of file merge_histograms.py.

merge_histograms.output_dir

Definition at line 199 of file merge_histograms.py.

string merge_histograms.output_flux_dir = options.output_dir+"/"

Definition at line 230 of file merge_histograms.py.

merge_histograms.parser = OptionParser(usage=usage)

Definition at line 172 of file merge_histograms.py.

merge_histograms.physics_list = options.physics_list

Definition at line 201 of file merge_histograms.py.

list merge_histograms.plot_files = []

Definition at line 235 of file merge_histograms.py.

merge_histograms.scriptdir = os.path.abspath(sys.argv[0]+"/../..")

Determine default g4lbne directory (the directory this script is in)

Definition at line 162 of file merge_histograms.py.

string merge_histograms.usage = "usage: %prog [options]"

Setup parser that reads in options.

Definition at line 171 of file merge_histograms.py.

merge_histograms.version = options.version

Determine G4LBNE Version.

Definition at line 209 of file merge_histograms.py.