Reference Manual

Functions
def	get_all_ids_names

def	get_his_data

def	get_used_binaries

def	main

Function Documentation

def mps_parse_pedechi2hist.get_all_ids_names ( mps_db )

Returns two lists containing the mille job IDs and the associated names.

Arguments:
- `mps_db`: path to the MPS database file

Definition at line 75 of file mps_parse_pedechi2hist.py.

Referenced by main().

 
 def get_all_ids_names(mps_db):
     """Returns two lists containing the mille job IDs and the associated names.
     
     Arguments:
     - `mps_db`: path to the MPS database file
     """
 
     lib = mpslib.jobdatabase()
     lib.read_db(mps_db)
 
     ids = lib.JOBNUMBER[:lib.nJobs]
     names = lib.JOBSP3[:lib.nJobs]
 
     return ids, names
 

mps_parse_pedechi2hist.get_all_ids_names

def get_all_ids_names

Definition: mps_parse_pedechi2hist.py:75

def mps_parse_pedechi2hist.get_his_data ( his_file )

Parse the pede histogram file.

Arguments:
- `his_file`: pede histogram file

Definition at line 121 of file mps_parse_pedechi2hist.py.

Referenced by main().

 
 def get_his_data(his_file):
     """Parse the pede histogram file.
     
     Arguments:
     - `his_file`: pede histogram file
     """
     
     his_data = []
     with open(his_file, "r") as his:
         found_chi2_start = False;
         
         for line in his:
             if r"final <Chi^2/Ndf> from accepted local fits vs file number" in line:
                 found_chi2_start = True
             if not found_chi2_start:
                 continue
             else:
                 if r"end of xy-data" in line: break
                 if not re.search("\d", line): continue
                 if re.search(r"[a-z]", line): continue
                 splitted = line.split()
                 his_data.append(splitted[-1])
 
     return his_data
 

mps_parse_pedechi2hist.get_his_data

def get_his_data

Definition: mps_parse_pedechi2hist.py:121

def mps_parse_pedechi2hist.get_used_binaries	(	cfg,
		no_binary_check
	)

Returns list of used binary IDs.

Arguments:
- `cfg`: python config used to run the pede job
- `no_binary_check`: if 'True' a check for file existence is skipped

Definition at line 91 of file mps_parse_pedechi2hist.py.

Referenced by main().

 
 def get_used_binaries(cfg, no_binary_check):
     """Returns list of used binary IDs.
     
     Arguments:
     - `cfg`: python config used to run the pede job
     - `no_binary_check`: if 'True' a check for file existence is skipped
     """
 
     sys.path.append(os.path.dirname(cfg))
 
     cache_stdout = sys.stdout
     sys.stdout = open(os.devnull, "w") # suppress unwanted output
     from alignment_merge import process.AlignmentProducer.algoConfig
     sys.stdout = cache_stdout
 
     binaries = algoConfig.mergeBinaryFiles
     if no_binary_check:
         used_binaries = binaries
     else:
         # following check works only if 'args.cfg' was run from the same directory:
         used_binaries = [b for b in binaries
                          if os.path.exists(os.path.join(os.path.dirname(cfg), b))]
 
     used_binaries = [int(re.sub(r"milleBinary(\d+)\.dat", r"\1", b))
                      for b in used_binaries]
     del sys.path[-1]            # remove the temporary search path extension
 
     return used_binaries
 

mps_parse_pedechi2hist.get_used_binaries

def get_used_binaries

Definition: mps_parse_pedechi2hist.py:91

def mps_parse_pedechi2hist.main ( argv = None )

Main routine of the script.

Arguments:
- `argv`: arguments passed to the main routine

Definition at line 27 of file mps_parse_pedechi2hist.py.

References get_all_ids_names(), get_his_data(), get_used_binaries(), and join().

 
 def main(argv = None):
     """Main routine of the script.
 
     Arguments:
     - `argv`: arguments passed to the main routine
     """
 
     if argv == None:
         argv = sys.argv[1:]
 
     parser = argparse.ArgumentParser(description="Analysis pede histogram file")
     parser.add_argument("-d", "--mps-db", dest="mps_db", required=True,
                         metavar="PATH", help="MPS database file ('mps.db')")
     parser.add_argument("--his", dest="his_file", required=True,
                         metavar="PATH", help="pede histogram file")
     parser.add_argument("-c", "--cfg", dest="cfg", metavar="PATH", required=True,
                         help="python configuration file of pede job")
     parser.add_argument("-b", "--no-binary-check", dest="no_binary_check",
                         default=False, action="store_true",
                         help=("skip check for existing binaries "
                               "(possibly needed if used interactively)"))
     args = parser.parse_args(argv)
 
 
     for input_file in (args.mps_db, args.his_file, args.cfg):
         if not os.path.exists(input_file):
             print "Could not find input file:", input_file
             sys.exit(1)
 
     ids, names = get_all_ids_names(args.mps_db)
     used_binaries = get_used_binaries(args.cfg, args.no_binary_check)
     his_data = get_his_data(args.his_file)
 
     if len(his_data) != len(used_binaries):
         print "The number of used binaries is", len(used_binaries),
         print "whereas in contrast, however, the <chi2/ndf> histogram in Pede has",
         print len(his_data), "bins (Pede version >= rev92 might help if #bins < #binaries).",
         print "Exiting."
         sys.exit(1)
 
     with open("chi2pedehis.txt", "w") as f:
         for i, b in enumerate(used_binaries):
             index = ids.index(b)
             name = names[index]
             f.write(" ".join([name, "{:03d}".format(b), his_data[i]])+"\n")
 