Classes
class	ConstraintsGenerator

Functions
def	save_config (constraint_sets, timedep_config=None, global_tags=None, init_event=None)

def	read_config (filename='constraint_config.pickled')

def	gen_constraints (constraint_sets, timedep_config=None, global_tags=None, init_event=None)

def	gen_constraints_from_config (filename='constraint_config.pickled')

Variables
	filename = sys.argv[1]
	File name.

Detailed Description

Created on 28 May 2018

@author: kleinwrt

Function Documentation

◆ gen_constraints()

def alignment.constraints_generator.gen_constraints	(	constraint_sets,
		timedep_config = `None`,
		global_tags = `None`,
		init_event = `None`
	)

Generate "event files" from timedep config and run over them to collect constraint data
and write it to files.

This is a bit tricky. I did not found a way to run basf2 over just a specified list of events
other than generating the files with metadata for event and then running over the files.
This uses unning basf2 multiple times in one script - seems to work despite the warnings - but only
for the generation of "event files"

Parameters
----------
constraint_sets : list (alignment.Constraints)
  List of sets of constraints
timedep_config : list(tuple(list(int), list(tuple(int, int, int))))
    Time-depence configuration.
    Each list item is 2-tuple with list of parameter numbers (use alignment.parameters to get them) and
    the (event, run, exp) numbers at which values of these parameters can change.
global_tags : list (str)
    List of global tag names and/or (absolute) file paths to local databases
init_event : tuple( int, int, int)
    Event (event, run, exp) at which to initialize time-INdependent constraints

Definition at line 110 of file constraints_generator.py.

 def gen_constraints(constraint_sets, timedep_config=None, global_tags=None, init_event=None):
     """
     Generate "event files" from timedep config and run over them to collect constraint data
     and write it to files.
  
     This is a bit tricky. I did not found a way to run basf2 over just a specified list of events
     other than generating the files with metadata for event and then running over the files.
     This uses unning basf2 multiple times in one script - seems to work despite the warnings - but only
     for the generation of "event files"
  
     Parameters
     ----------
     constraint_sets : list (alignment.Constraints)
       List of sets of constraints
     timedep_config : list(tuple(list(int), list(tuple(int, int, int))))
         Time-depence configuration.
         Each list item is 2-tuple with list of parameter numbers (use alignment.parameters to get them) and
         the (event, run, exp) numbers at which values of these parameters can change.
     global_tags : list (str)
         List of global tag names and/or (absolute) file paths to local databases
     init_event : tuple( int, int, int)
         Event (event, run, exp) at which to initialize time-INdependent constraints
  
     """
     if timedep_config is None:
         timedep_config = []
  
     if global_tags is None:
         global_tags = [tag for tag in b2.conditions.default_globaltags]
  
     events = []
     for (labels, events_) in timedep_config:
         events += [event for event in events_]
  
     if not len(timedep_config):
         if init_event is None:
             init_event = (0, 0, 0)
         events = [init_event]
  
     events = [(exp, run, ev) for (ev, run, exp) in events]
     events = sorted(list(set(events)))
     events = [(ev_, run_, exp_) for (exp_, run_, ev_) in events]
  
     fileName = 'TimedepConfigEvent_exp{}run{}ev{}.root'
     files = []
  
     print('Global tags:')
     print(global_tags)
     print('Global tags reversed (this will be used for b2.conditions.override_globaltags(...)):')
     print([tag for tag in reversed(global_tags)])
  
     for tag in [tag for tag in reversed(global_tags)]:
         if os.path.exists(tag):
             b2.conditions.append_testing_payloads(os.path.abspath(tag))
         else:
             b2.conditions.append_globaltag(tag)
  
     for index, event in enumerate(events):
         #  conditions.reset()
  
         ev, run, exp = event
         path = b2.create_path()
         path.add_module("EventInfoSetter",
                         skipNEvents=ev,
                         evtNumList=[ev + 1],
                         runList=[run],
                         expList=[exp])
         path.add_module('Progress')
         this_filename = fileName.format(exp, run, ev)
         path.add_module('RootOutput', outputFileName=this_filename, ignoreCommandLineOverride=True)
         files.append(this_filename)
         b2.process(path)
         print(b2.statistics)
  
     print(files)
  
     # conditions.override_globaltags(global_tags)
  
     path = b2.create_path()
     path.add_module("RootInput", inputFileNames=files, ignoreCommandLineOverride=True)
     path.add_module('HistoManager')
     path.add_module('Progress')
     path.add_module('Gearbox')
     path.add_module('Geometry')
  
     collector = path.add_module('MillepedeCollector',
                                 timedepConfig=timedep_config)
  
     constraint_files = []
     for constraint_set in constraint_sets:
         constraint_set.configure_collector(collector)
         constraint_files.append(constraint_set.filename)
         path.add_module(ConstraintsGenerator(constraint_set))
  
     b2.process(path)
     print(b2.statistics)
  
     return [os.path.abspath(file) for file in constraint_files]
  
  

◆ gen_constraints_from_config()

def alignment.constraints_generator.gen_constraints_from_config ( filename = 'constraint_config.pickled' )

Generate constraints from pickled configuration

Definition at line 210 of file constraints_generator.py.

◆ read_config()

def alignment.constraints_generator.read_config ( filename = 'constraint_config.pickled' )

Read the pickled constraint configuration from a file

filename : str
  File with pickled constraints' configuration

Definition at line 97 of file constraints_generator.py.

◆ save_config()

def alignment.constraints_generator.save_config	(	constraint_sets,
		timedep_config = `None`,
		global_tags = `None`,
		init_event = `None`
	)

Save constraints configuration to a file (using pickle)

Parameters
----------
constraint_sets : list (alignment.Constraints)
  List of sets of constraints
timedep_config : list(tuple(list(int), list(tuple(int, int, int))))
    Time-depence configuration.
    Each list item is 2-tuple with list of parameter numbers (use alignment.parameters to get them) and
    the (event, run, exp) numbers at which values of these parameters can change.
global_tags : list (str)
    List of global tag names and/or (absolute) file paths to local databases
init_event : tuple( int, int, int)
    Event (event, run, exp) at which to initialize time-INdependent constraints

Definition at line 65 of file constraints_generator.py.

Classes