MillepedeCollectorModule Class Reference

Calibration data collector for Millepede Algorithm. More...

#include <MillepedeCollectorModule.h>

Inheritance diagram for MillepedeCollectorModule:

Public Types
enum	EModulePropFlags {   c_Input = 1 ,   c_Output = 2 ,   c_ParallelProcessingCertified = 4 ,   c_HistogramManager = 8 ,   c_InternalSerializer = 16 ,   c_TerminateInAllProcesses = 32 ,   c_DontCollectStatistics = 64 }
	Each module can be tagged with property flags, which indicate certain features of the module. More...
typedef ModuleCondition::EAfterConditionPath	EAfterConditionPath
	Forward the EAfterConditionPath definition from the ModuleCondition.

Public Member Functions
	MillepedeCollectorModule ()
	Constructor: Sets the description, the properties and the parameters of the module.
virtual void	prepare () override
	Prepration.
virtual void	collect () override
	Data collection.
virtual void	closeRun () override
	Only for closing mille binaries after each run.
virtual void	finish () override
	Register mille binaries in file catalog.
std::string	getUniqueMilleName ()
	Make a name for mille binary (encodes module name + starting exp, run and event + process id)
std::vector< genfit::Track * >	getParticlesTracks (std::vector< Particle * > particles, bool addVertexPoint=true)
	Get all usable tracks for particles.
bool	fitRecoTrack (RecoTrack &recoTrack, Particle *particle=nullptr)
	Fit given RecoTrack with GBL.
TMatrixD	getGlobalToLocalTransform (const genfit::MeasuredStateOnPlane &msop)
	Compute the transformation matrix d(q/p,u',v',u,v)/d(x,y,z,px,py,pz) from state at first track point (vertex)
TMatrixD	getLocalToGlobalTransform (const genfit::MeasuredStateOnPlane &msop)
	Compute the transformation matrix d(x,y,z,px,py,pz)/d(q/p,u',v',u,v) from state at first track point (vertex)
std::pair< TMatrixD, TMatrixD >	getTwoBodyToLocalTransform (Particle &mother, double motherMass)
	Compute the transformation matrices d(q/p,u'v',u,v)/d(vx,vy,vz,px,py,pz,theta,phi,M) = dq/d(v,z) for both particles in pair.
void	storeTrajectory (gbl::GblTrajectory &trajectory)
	Write down a GBL trajectory (to TTree or binary file)
std::tuple< B2Vector3D, TMatrixDSym >	getPrimaryVertexAndCov () const
	Get the primary vertex position estimation and its size from BeamSpot.
void	initialize () final
	Set up a default RunRange object in datastore and call prepare()
void	event () final
	Check current experiment and run and update if needed, fill into RunRange and collect()
void	beginRun () final
	Reset the m_runCollectOnRun flag, if necessary, to begin collection again.
void	endRun () final
	Write the current collector objects to a file and clear their memory.
void	terminate () final
	Write the final objects to the file.
void	defineHisto () final
	Runs due to HistoManager, allows us to discover the correct file.
template<class T >
void	registerObject (std::string name, T *obj)
	Register object with a name, takes ownership, do not access the pointer beyond prepare()
template<class T >
T *	getObjectPtr (std::string name)
	Calls the CalibObjManager to get the requested stored collector data.
virtual std::vector< std::string >	getFileNames (bool outputFiles)
	Return a list of output filenames for this modules.
const std::string &	getName () const
	Returns the name of the module.
const std::string &	getType () const
	Returns the type of the module (i.e.
const std::string &	getPackage () const
	Returns the package this module is in.
const std::string &	getDescription () const
	Returns the description of the module.
void	setName (const std::string &name)
	Set the name of the module.
void	setPropertyFlags (unsigned int propertyFlags)
	Sets the flags for the module properties.
LogConfig &	getLogConfig ()
	Returns the log system configuration.
void	setLogConfig (const LogConfig &logConfig)
	Set the log system configuration.
void	setLogLevel (int logLevel)
	Configure the log level.
void	setDebugLevel (int debugLevel)
	Configure the debug messaging level.
void	setAbortLevel (int abortLevel)
	Configure the abort log level.
void	setLogInfo (int logLevel, unsigned int logInfo)
	Configure the printed log information for the given level.
void	if_value (const std::string &expression, const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
	Add a condition to the module.
void	if_false (const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
	A simplified version to add a condition to the module.
void	if_true (const std::shared_ptr< Path > &path, EAfterConditionPath afterConditionPath=EAfterConditionPath::c_End)
	A simplified version to set the condition of the module.
bool	hasCondition () const
	Returns true if at least one condition was set for the module.
const ModuleCondition *	getCondition () const
	Return a pointer to the first condition (or nullptr, if none was set)
const std::vector< ModuleCondition > &	getAllConditions () const
	Return all set conditions for this module.
bool	evalCondition () const
	If at least one condition was set, it is evaluated and true returned if at least one condition returns true.
std::shared_ptr< Path >	getConditionPath () const
	Returns the path of the last true condition (if there is at least one, else reaturn a null pointer).
Module::EAfterConditionPath	getAfterConditionPath () const
	What to do after the conditional path is finished.
std::vector< std::shared_ptr< Path > >	getAllConditionPaths () const
	Return all condition paths currently set (no matter if the condition is true or not).
bool	hasProperties (unsigned int propertyFlags) const
	Returns true if all specified property flags are available in this module.
bool	hasUnsetForcedParams () const
	Returns true and prints error message if the module has unset parameters which the user has to set in the steering file.
const ModuleParamList &	getParamList () const
	Return module param list.
template<typename T >
ModuleParam< T > &	getParam (const std::string &name) const
	Returns a reference to a parameter.
bool	hasReturnValue () const
	Return true if this module has a valid return value set.
int	getReturnValue () const
	Return the return value set by this module.
std::shared_ptr< PathElement >	clone () const override
	Create an independent copy of this module.
std::shared_ptr< boost::python::list >	getParamInfoListPython () const
	Returns a python list of all parameters.

Static Public Member Functions
static void	exposePythonAPI ()
	Exposes methods of the Module class to Python.

Protected Member Functions
virtual void	startRun ()
	Replacement for beginRun(). Do anything you would normally do in beginRun here.
virtual void	inDefineHisto ()
	Replacement for defineHisto(). Do anything you would normally do in defineHisto here.
virtual void	def_initialize ()
	Wrappers to make the methods without "def_" prefix callable from Python.
virtual void	def_beginRun ()
	Wrapper method for the virtual function beginRun() that has the implementation to be used in a call from Python.
virtual void	def_event ()
	Wrapper method for the virtual function event() that has the implementation to be used in a call from Python.
virtual void	def_endRun ()
	This method can receive that the current run ends as a call from the Python side.
virtual void	def_terminate ()
	Wrapper method for the virtual function terminate() that has the implementation to be used in a call from Python.
void	setDescription (const std::string &description)
	Sets the description of the module.
void	setType (const std::string &type)
	Set the module type.
template<typename T >
void	addParam (const std::string &name, T &paramVariable, const std::string &description, const T &defaultValue)
	Adds a new parameter to the module.
template<typename T >
void	addParam (const std::string &name, T &paramVariable, const std::string &description)
	Adds a new enforced parameter to the module.
void	setReturnValue (int value)
	Sets the return value for this module as integer.
void	setReturnValue (bool value)
	Sets the return value for this module as bool.
void	setParamList (const ModuleParamList &params)
	Replace existing parameter list.

Protected Attributes
TDirectory *	m_dir
	The top TDirectory that collector objects for this collector will be stored beneath.
CalibObjManager	m_manager
	Controls the creation, collection and access to calibration objects.
RunRange *	m_runRange
	Overall list of runs processed.
Calibration::ExpRun	m_expRun
	Current ExpRun for object retrieval (becomes -1,-1 for granularity=all)
StoreObjPtr< EventMetaData >	m_emd
	Current EventMetaData.

Private Member Functions
void	updateMassWidthIfSet (std::string listName, double &mass, double &width)
	Update mass and width of the particle (mother in list) with user custom-defined values.
bool	getPreScaleChoice ()
	I'm a little worried about floating point precision when comparing to 0.0 and 1.0 as special values.
std::list< ModulePtr >	getModules () const override
	no submodules, return empty list
std::string	getPathString () const override
	return the module name.
void	setParamPython (const std::string &name, const boost::python::object &pyObj)
	Implements a method for setting boost::python objects.
void	setParamPythonDict (const boost::python::dict &dictionary)
	Implements a method for reading the parameter values from a boost::python dictionary.

Private Attributes
std::vector< std::string >	m_tracks
	Names of arrays with single RecoTracks fitted by GBL.
std::vector< std::string >	m_particles
	Names of particle list with single particles.
std::vector< std::string >	m_vertices
	Name of particle list with mothers of daughters to be used with vertex constraint in calibration.
std::vector< std::string >	m_primaryVertices
	Name of particle list with mothers of daughters to be used with vertex + IP profile (+ optional calibration) constraint in calibration.
std::vector< std::string >	m_twoBodyDecays
	Name of particle list with mothers of daughters to be used with vertex + mass constraint in calibration.
std::vector< std::string >	m_primaryTwoBodyDecays
	Name of particle list with mothers of daughters to be used with vertex + IP profile (+ optional calibration) + IP kinematics (+ optional calibration) constraint in calibration.
std::vector< std::string >	m_primaryMassTwoBodyDecays
	Name of particle list with mothers of daughters to be used with vertex + IP profile + mass constraint in calibration.
std::vector< std::string >	m_primaryMassVertexTwoBodyDecays
	Name of particle list with mothers of daughters to be used with vertex + IP profile + mass constraint in calibration.
double	m_stableParticleWidth
	Width (in GeV/c/c) to use for invariant mass constraint for 'stable' particles (like K short).
bool	m_doublePrecision
	Use double (instead of single/float) precision for binary files.
bool	m_calibrateVertex
	Add derivatives for beam spot vertex calibration for primary vertices.
bool	m_calibrateKinematics = true
	Add derivatives for beam spot kinematics calibration for primary vertices.
double	m_minPValue
	Minimum p.value for output.
bool	m_useGblTree
	Whether to use TTree to accumulate GBL data instead of binary files.
bool	m_absFilePaths
	Use absolute path to locate binary files in MilleData.
std::vector< std::string >	m_components {}
	Whether to use VXD alignment hierarchy.
int	m_externalIterations
	Number of external iterations of GBL fitter.
std::string	m_internalIterations
	String defining internal GBL iterations for outlier down-weighting.
int	m_recalcJacobians
	Up to which external iteration propagation Jacobians should be re-calculated.
bool	m_fitTrackT0
	Add local parameter for track T0 fit in GBL (local derivative)
bool	m_updateCDCWeights
	Update L/R weights from previous DAF fit result?
double	m_minCDCHitWeight
	Minimum CDC hit weight.
double	m_minUsedCDCHitFraction
	Minimum CDC used hit fraction.
int	m_hierarchyType
	Type of alignment hierarchy (for VXD only for now): 0 = None, 1 = Flat (only constraints, no new global parameters/derivatives), 2 = Half-Shells + sensors (no ladders), 3 = Full.
bool	m_enablePXDHierarchy
	enable PXD hierarchy
bool	m_enableSVDHierarchy
	enable SVD hierarchy
bool	m_enableWireByWireAlignment
	Enable global derivatives for wire-by-wire alignment.
bool	m_enableWireSagging
	Enable global derivatives for wire sagging.
std::vector< std::tuple< int, int, int > >	m_eventNumbers {}
	List of event meta data entries at which payloads can change for timedep calibration.
std::vector< std::tuple< std::vector< int >, std::vector< std::tuple< int, int, int > > > >	m_timedepConfig
	Config for time dependence: list( tuple( list( param1, param2, ... ), list( (ev, run, exp), ... )), ...
std::map< std::string, std::tuple< double, double > >	m_customMassConfig
	Map of list_name -> (mass, width) for custom mass and width setting.
std::vector< gbl::GblData >	m_currentGblData {}
	Current vector of GBL data from trajectory to be stored in a tree.
StoreObjPtr< EventT0 >	m_eventT0
	Optional input for EventT0.
StoreObjPtr< EventMetaData >	m_evtMetaData
	Required object pointer to EventMetaData.
std::string	m_granularity
	Granularity of data collection = run|all(= no granularity, exp,run=-1,-1)
int	m_maxEventsPerRun
	Maximum number of events to be collected at the start of each run (-1 = no maximum)
float	m_preScale
	Prescale module parameter, this fraction of events will have collect() run on them [0.0 -> 1.0].
bool	m_runCollectOnRun = true
	Whether or not we will run the collect() at all this run, basically skips the event() function if false.
std::map< Calibration::ExpRun, int >	m_expRunEvents
	How many events processed for each ExpRun so far, stops counting up once max is hit Only used/incremented if m_maxEventsPerRun > -1.
int *	m_eventsCollectedInRun
	Will point at correct value in m_expRunEvents.
std::string	m_name
	The name of the module, saved as a string (user-modifiable)
std::string	m_type
	The type of the module, saved as a string.
std::string	m_package
	Package this module is found in (may be empty).
std::string	m_description
	The description of the module.
unsigned int	m_propertyFlags
	The properties of the module as bitwise or (with |) of EModulePropFlags.
LogConfig	m_logConfig
	The log system configuration of the module.
ModuleParamList	m_moduleParamList
	List storing and managing all parameter of the module.
bool	m_hasReturnValue
	True, if the return value is set.
int	m_returnValue
	The return value.
std::vector< ModuleCondition >	m_conditions
	Module condition, only non-null if set.

Detailed Description

Calibration data collector for Millepede Algorithm.

Collects data from GBL-fitted tracks and produces binary files for Millepede

Definition at line 34 of file MillepedeCollectorModule.h.

Member Typedef Documentation

EAfterConditionPath

typedef ModuleCondition::EAfterConditionPath EAfterConditionPath

inherited

Forward the EAfterConditionPath definition from the ModuleCondition.

Definition at line 88 of file Module.h.

Member Enumeration Documentation

EModulePropFlags

enum EModulePropFlags

inherited

Each module can be tagged with property flags, which indicate certain features of the module.

Enumerator
c_Input	This module is an input module (reads data).
c_Output	This module is an output module (writes data).
c_ParallelProcessingCertified	This module can be run in parallel processing mode safely (All I/O must be done through the data store, in particular, the module must not write any files.)
c_HistogramManager	This module is used to manage histograms accumulated by other modules.
c_InternalSerializer	This module is an internal serializer/deserializer for parallel processing.
c_TerminateInAllProcesses	When using parallel processing, call this module's terminate() function in all processes(). This will also ensure that there is exactly one process (single-core if no parallel modules found) or at least one input, one main and one output process.
c_DontCollectStatistics	No statistics is collected for this module.

Definition at line 77 of file Module.h.

                          {
      c_Input                       = 1,  
      c_Output                      = 2,  
      c_ParallelProcessingCertified = 4,  
      c_HistogramManager            = 8, 
      c_InternalSerializer          = 16,  
      c_TerminateInAllProcesses     = 32,  
      c_DontCollectStatistics       = 64,  
    };

Constructor & Destructor Documentation

MillepedeCollectorModule()

MillepedeCollectorModule

(

)

Constructor: Sets the description, the properties and the parameters of the module.

Definition at line 62 of file MillepedeCollectorModule.cc.

                                                   : CalibrationCollectorModule()
{
  setPropertyFlags(c_ParallelProcessingCertified);
  setDescription("Calibration data collector for Millepede Algorithm");
 
  // Configure input sample types
  addParam("tracks", m_tracks, "Names of collections of RecoTracks (already fitted with DAF) for calibration", vector<string>({""}));
  addParam("particles", m_particles, "Names of particle list of single particles", vector<string>());
  addParam("vertices", m_vertices,
           "Name of particle list of (mother) particles with daughters for calibration using vertex constraint", vector<string>());
  addParam("primaryVertices", m_primaryVertices,
           "Name of particle list of (mother) particles with daughters for calibration using vertex + IP profile constraint",
           vector<string>());
  addParam("twoBodyDecays", m_twoBodyDecays,
           "Name of particle list of (mother) particles with daughters for calibration using vertex + mass constraint",
           vector<string>());
  addParam("primaryTwoBodyDecays", m_primaryTwoBodyDecays,
           "Name of particle list of (mother) particles with daughters for calibration using vertex + IP profile + kinematics constraint",
           vector<string>());
  addParam("primaryMassTwoBodyDecays", m_primaryMassTwoBodyDecays,
           "Name of particle list of (mother) particles with daughters for calibration using vertex + mass constraint",
           vector<string>());
  addParam("primaryMassVertexTwoBodyDecays", m_primaryMassVertexTwoBodyDecays,
           "Name of particle list of (mother) particles with daughters for calibration using vertex + IP profile + mass constraint",
           vector<string>());
 
  addParam("stableParticleWidth", m_stableParticleWidth,
           "Width (in GeV/c/c) to use for invariant mass constraint for 'stable' particles (like K short). Temporary until proper solution is found.",
           double(0.002));
  // Configure output
  addParam("doublePrecision", m_doublePrecision, "Use double (=true) or single/float (=false) precision for writing binary files",
           bool(false));
  addParam("useGblTree", m_useGblTree, "Store GBL trajectories in a tree instead of output to binary files",
           bool(true));
  addParam("absFilePaths", m_absFilePaths, "Use absolute paths to remember binary files. Only applies if useGblTree=False",
           bool(false));
 
  // Configure global parameters
  addParam("components", m_components,
           "Specify which DB objects are calibrated, like ['BeamSpot', 'CDCTimeWalks'] or leave empty to use all components available.",
           m_components);
  addParam("calibrateVertex", m_calibrateVertex,
           "For primary vertices / two body decays, beam spot vertex calibration derivatives are added",
           bool(true));
  addParam("calibrateKinematics", m_calibrateKinematics,
           "For primary two body decays, beam spot kinematics calibration derivatives are added",
           bool(true));
 
  //Configure GBL fit of individual tracks
  addParam("externalIterations", m_externalIterations, "Number of external iterations of GBL fitter",
           int(0));
  addParam("internalIterations", m_internalIterations, "String defining internal GBL iterations for outlier down-weighting",
           string(""));
  addParam("recalcJacobians", m_recalcJacobians, "Up to which external iteration propagation Jacobians should be re-calculated",
           int(0));
 
  addParam("minPValue", m_minPValue, "Minimum p-value to write out a (combined) trajectory. Set <0 to write out all.",
           double(-1.));
 
  // Configure CDC specific options
  addParam("fitTrackT0", m_fitTrackT0, "Add local parameter for track T0 fit in GBL",
           bool(true));
  addParam("updateCDCWeights", m_updateCDCWeights, "Update L/R weights from previous DAF fit result",
           bool(true));
  addParam("minCDCHitWeight", m_minCDCHitWeight, "Minimum (DAF) CDC hit weight for usage by GBL",
           double(1.0E-6));
  addParam("minUsedCDCHitFraction", m_minUsedCDCHitFraction, "Minimum used CDC hit fraction to write out a trajectory",
           double(0.85));
 
  addParam("hierarchyType", m_hierarchyType, "Type of (VXD only now) hierarchy: 0 = None, 1 = Flat, 2 = Half-Shells, 3 = Full",
           int(3));
  addParam("enablePXDHierarchy", m_enablePXDHierarchy, "Enable PXD in hierarchy (flat or full)",
           bool(true));
  addParam("enableSVDHierarchy", m_enableSVDHierarchy, "Enable SVD in hierarchy (flat or full)",
           bool(true));
 
  addParam("enableWireByWireAlignment", m_enableWireByWireAlignment, "Enable global derivatives for wire-by-wire alignment",
           bool(false));
  addParam("enableWireSagging", m_enableWireSagging, "Enable global derivatives for wire sagging",
           bool(false));
 
  // Time dependence
  addParam("events", m_eventNumbers,
           "List of (event, run, exp) with event numbers at which payloads can change for timedep calibration.",
           m_eventNumbers);
  // Time dependence config
  addParam("timedepConfig", m_timedepConfig,
           "list{ {list{param1, param2, ...}, list{(ev1, run1, exp1), ...}}, ... }.",
           m_timedepConfig);
 
  // Custom mass+width config
  addParam("customMassConfig", m_customMassConfig,
           "dict{ list_name: (mass, width), ... } with custom mass and width to use as external measurement.",
           m_customMassConfig);
}

Member Function Documentation

beginRun()

void beginRun ( void  )

finalvirtualinherited

Reset the m_runCollectOnRun flag, if necessary, to begin collection again.

It seems that the beginRun() function is called in each basf2 subprocess when the run changes in each process. This is nice because it allows us to write the new (exp,run) object creation in the beginRun function as though the other processes don't exist.

Reimplemented from HistoModule.

Definition at line 77 of file CalibrationCollectorModule.cc.

{
  // Current (Exp,Run)
  ExpRun expRun = make_pair(m_emd->getExperiment(), m_emd->getRun());
  m_runRange->add(expRun.first, expRun.second);
 
  // Do we care about the number of events collected in each (input data) ExpRun?
  // If so, we want to create values for the events collected map
  if (m_maxEventsPerRun > -1) {
    // Do we have a count for this ExpRun yet? If not create one
    auto i_eventsInExpRun = m_expRunEvents.find(expRun);
    if (i_eventsInExpRun == m_expRunEvents.end()) {
      m_expRunEvents[expRun] = 0;
    }
 
    // Set our pointer to the correct location for this ExpRun
    m_eventsCollectedInRun = &m_expRunEvents[expRun];
    // Want to reset our flag to start collection if necessary
    if ((*m_eventsCollectedInRun) < m_maxEventsPerRun) {
      B2INFO("New run has had less events than the maximum collected so far ("
             << (*m_eventsCollectedInRun)
             << " < "
             << m_maxEventsPerRun
             << "). Turning on collection.");
      m_runCollectOnRun = true;
    } else {
      B2INFO("New run has had more events than the maximum collected so far ("
             << (*m_eventsCollectedInRun)
             << " >= "
             << m_maxEventsPerRun
             << "). Turning off collection.");
      m_runCollectOnRun = false;
    }
  }
  // Granularity=all removes data splitting by runs by setting
  // always the same exp, run for calibration data objects
  if (m_granularity == "all") {
    m_expRun = { -1, -1};
  } else {
    m_expRun = expRun;
  }
  m_manager.createExpRunDirectories(m_expRun);
  // Run the user's startRun() implementation if there is one
  startRun();
}

clone()

std::shared_ptr< PathElement > clone ( ) const

overridevirtualinherited

Create an independent copy of this module.

Note that parameters are shared, so changing them on a cloned module will also affect the original module.

Implements PathElement.

Definition at line 179 of file Module.cc.

{
  ModulePtr newModule = ModuleManager::Instance().registerModule(getType());
  newModule->m_moduleParamList.setParameters(getParamList());
  newModule->setName(getName());
  newModule->m_package = m_package;
  newModule->m_propertyFlags = m_propertyFlags;
  newModule->m_logConfig = m_logConfig;
  newModule->m_conditions = m_conditions;
 
  return newModule;
}

closeRun()

void closeRun ( )

overridevirtual

Only for closing mille binaries after each run.

Reimplemented from CalibrationCollectorModule.

Definition at line 985 of file MillepedeCollectorModule.cc.

{
  // We close the file at end of run, producing
  // one file per run (and process id) which is more
  // convenient than one large binary block.
  auto mille = getObjectPtr<MilleData>("mille");
  if (mille->isOpen())
    mille->close();
}

collect()

void collect ( )

overridevirtual

Data collection.

Reimplemented from CalibrationCollectorModule.

Definition at line 254 of file MillepedeCollectorModule.cc.

{
  alignment::GlobalCalibrationManager::getInstance().preCollect(*m_evtMetaData);
 
  if (!m_useGblTree) {
    // Open new file on request (at start or after being closed)
    auto mille = getObjectPtr<MilleData>("mille");
    if (!mille->isOpen())
      mille->open(getUniqueMilleName());
  }
 
  std::shared_ptr<genfit::GblFitter> gbl(new genfit::GblFitter());
  double chi2 = -1.;
  double lostWeight = -1.;
  int ndf = -1;
  float evt0 = -9999.;
 
  for (auto arrayName : m_tracks) {
    StoreArray<RecoTrack> recoTracks(arrayName);
    if (!recoTracks.isValid())
      continue;
 
    for (auto& recoTrack : recoTracks) {
 
      if (!fitRecoTrack(recoTrack))
        continue;
 
      auto& track = RecoTrackGenfitAccess::getGenfitTrack(recoTrack);
      if (!track.hasFitStatus())
        continue;
      genfit::GblFitStatus* fs = dynamic_cast<genfit::GblFitStatus*>(track.getFitStatus());
      if (!fs)
        continue;
 
      if (!fs->isFittedWithReferenceTrack())
        continue;
 
      using namespace gbl;
      GblTrajectory trajectory(gbl->collectGblPoints(&track, track.getCardinalRep()), fs->hasCurvature());
 
      trajectory.fit(chi2, ndf, lostWeight);
      getObjectPtr<TH1I>("ndf")->Fill(ndf);
      getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
      getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
      if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
        evt0 = m_eventT0->getEventT0();
        getObjectPtr<TH1F>("evt0")->Fill(evt0);
      }
 
      if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(trajectory);
 
    }
 
  }
 
  for (auto listName : m_particles) {
    StoreObjPtr<ParticleList> list(listName);
    if (!list.isValid())
      continue;
 
    for (unsigned int iParticle = 0; iParticle < list->getListSize(); ++iParticle) {
      for (auto& track : getParticlesTracks({list->getParticle(iParticle)}, false)) {
        auto gblfs = dynamic_cast<genfit::GblFitStatus*>(track->getFitStatus());
 
        gbl::GblTrajectory trajectory(gbl->collectGblPoints(track, track->getCardinalRep()), gblfs->hasCurvature());
 
        trajectory.fit(chi2, ndf, lostWeight);
        getObjectPtr<TH1I>("ndf")->Fill(ndf);
        getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
        getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
        if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
          evt0 = m_eventT0->getEventT0();
          getObjectPtr<TH1F>("evt0")->Fill(evt0);
        }
 
        if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(trajectory);
 
      }
    }
  }
 
  for (auto listName : m_vertices) {
    StoreObjPtr<ParticleList> list(listName);
    if (!list.isValid())
      continue;
 
    for (unsigned int iParticle = 0; iParticle < list->getListSize(); ++iParticle) {
      auto mother = list->getParticle(iParticle);
      std::vector<std::pair<std::vector<gbl::GblPoint>, TMatrixD> > daughters;
 
      for (auto& track : getParticlesTracks(mother->getDaughters()))
        daughters.push_back({
        gbl->collectGblPoints(track, track->getCardinalRep()),
        getGlobalToLocalTransform(track->getFittedState()).GetSub(0, 4, 0, 2)
      });
 
      if (daughters.size() > 1) {
        gbl::GblTrajectory combined(daughters);
 
        combined.fit(chi2, ndf, lostWeight);
        getObjectPtr<TH1I>("ndf")->Fill(ndf);
        getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
        getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
        if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
          evt0 = m_eventT0->getEventT0();
          getObjectPtr<TH1F>("evt0")->Fill(evt0);
        }
 
 
        if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(combined);
 
        B2RESULT("Vertex-constrained fit NDF = " << ndf << " Chi2/NDF = " << chi2 / double(ndf));
 
      }
    }
  }
 
  for (auto listName : m_primaryVertices) {
    StoreObjPtr<ParticleList> list(listName);
    if (!list.isValid())
      continue;
 
    for (unsigned int iParticle = 0; iParticle < list->getListSize(); ++iParticle) {
      auto mother = list->getParticle(iParticle);
      std::vector<std::pair<std::vector<gbl::GblPoint>, TMatrixD> > daughters;
 
      TMatrixD extProjection(5, 3);
      TMatrixD locProjection(3, 5);
 
      // geometric constaint: 3 common (position) parameters + 3 external (curv., directions) per daughter
      TMatrixD innerTrafo(5, 3 + 3 * mother->getDaughters().size());
      unsigned int iCol(3);
 
      bool first(true);
      for (auto& track : getParticlesTracks(mother->getDaughters())) {
        if (first) {
          // For first trajectory only
          extProjection = getGlobalToLocalTransform(track->getFittedState()).GetSub(0, 4, 0, 2);
          locProjection = getLocalToGlobalTransform(track->getFittedState()).GetSub(0, 2, 0, 4);
          first = false;
        }
 
        innerTrafo.Zero();
        innerTrafo.SetSub(3, 0, getGlobalToLocalTransform(track->getFittedState()).GetSub(3, 4, 0, 2));
        innerTrafo[0][iCol++] = 1.;
        innerTrafo[1][iCol++] = 1.;
        innerTrafo[2][iCol++] = 1.;
 
 
        daughters.push_back({
          gbl->collectGblPoints(track, track->getCardinalRep()),
          innerTrafo
        });
      }
 
      if (daughters.size() > 1) {
        auto beam = getPrimaryVertexAndCov();
 
        TMatrixDSym vertexCov(get<TMatrixDSym>(beam));
        TMatrixDSym vertexPrec(get<TMatrixDSym>(beam).Invert());
        B2Vector3D vertexResidual = - (B2Vector3D(mother->getVertex()) - get<B2Vector3D>(beam));
 
        TVectorD extMeasurements(3);
        extMeasurements[0] = vertexResidual[0];
        extMeasurements[1] = vertexResidual[1];
        extMeasurements[2] = vertexResidual[2];
 
        TMatrixD extDeriv(3, 9);
        extDeriv.Zero();
        // beam vertex constraint
        extDeriv(0, 0) = 1.;
        extDeriv(1, 1) = 1.;
        extDeriv(2, 2) = 1.;
 
        if (m_calibrateVertex) {
          TMatrixD derivatives(3, 3);
          derivatives.Zero();
          derivatives(0, 0) = 1.;
          derivatives(1, 1) = 1.;
          derivatives(2, 2) = 1.;
 
          std::vector<int> labels;
          GlobalLabel label = GlobalLabel::construct<BeamSpot>(0, 0);
          labels.push_back(label.setParameterId(1));
          labels.push_back(label.setParameterId(2));
          labels.push_back(label.setParameterId(3));
 
          // Allow to disable BeamSpot externally
          alignment::GlobalDerivatives globals(labels, derivatives);
          // Add derivatives for vertex calibration to first point of first trajectory
          // NOTE: use GlobalDerivatives operators vector<int> and TMatrixD which filter
          // the derivatives to not pass those with zero labels (useful to get rid of some params)
          std::vector<int> lab(globals); TMatrixD der(globals);
 
          // Transformation from local system at (vertex) point to global (vx,vy,vz)
          // of the (decay) vertex
          //
          // d(q/p,u',v',u,v)/d(vy,vy,vz) = dLocal_dExt
          //
          //
          // Note its transpose is its "inverse" in the sense that
          //
          // dloc/dext * (dloc/dext)^T = diag(0, 0, 0, 0, 1, 1)
          //
          //
          // N.B. typical dLocal_dExt matrix (5x3):
          //
          //      |      0    |      1    |      2    |
          // --------------------------------------------
          //    0 |          0           0           0
          //    1 |          0           0           0
          //    2 |          0           0           0
          //    3 |   -0.02614     -0.9997           0
          //    4 |          0           0           1
          //
          // Therefore one can simplify things by only taking the last two rows/columns in vectors/matrices
          // and vertex measurement can be expressed as standard 2D measurement in GBL.
          //
          TMatrixD dLocal_dExt = extProjection;
          TMatrixD dExt_dLocal = locProjection;
 
          TVectorD locRes = dLocal_dExt * extMeasurements;
          // Do not use inverted covariance - seems to have issues with numeric precision
          TMatrixD locCov = dLocal_dExt * vertexCov * dExt_dLocal;
          // Invert here only the 2D sub-matrix (rest is zero due to the foŕm of dLocal_dExt)
          TMatrixD locPrec = locCov.GetSub(3, 4, 3, 4).Invert();
          TMatrixDSym locPrec2D(2); locPrec2D.Zero();
          for (int i = 0; i < 2; ++i)
            for (int j = 0; j < 2; ++j)
              locPrec2D(i, j) = locPrec(i, j);
 
          // Take the 2 last components also for residuals and global derivatives
          // (in local system of vertex point - defined during fitRecoTrack(..., particle) and using
          // the (hopefully) updated momentum and position seed after vertex fit by modularAnalysis
          TVectorD locRes2D = locRes.GetSub(3, 4);
          TMatrixD locDerivs2D = (extProjection * der).GetSub(3, 4, 0, 2);
 
          // Attach the primary beamspot vertex position as a measurement at 1st point
          // of first trajectory (and optionally also the global derivatives for beamspot alignment
          daughters[0].first[0].addMeasurement(locRes2D, locPrec2D);
          if (!lab.empty()) {
            daughters[0].first[0].addGlobals(lab, locDerivs2D);
          }
 
          gbl::GblTrajectory combined(daughters);
          //combined.printTrajectory(100);
          //combined.printPoints(100);
 
          combined.fit(chi2, ndf, lostWeight);
          getObjectPtr<TH1I>("ndf")->Fill(ndf);
          getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
          getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
          if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
            evt0 = m_eventT0->getEventT0();
            getObjectPtr<TH1F>("evt0")->Fill(evt0);
          }
 
          if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(combined);
          B2RESULT("Beam vertex constrained fit results NDF = " << ndf << " Chi2/NDF = " << chi2 / double(ndf));
 
        } else {
 
          gbl::GblTrajectory combined(daughters, extDeriv, extMeasurements, vertexPrec);
 
          combined.fit(chi2, ndf, lostWeight);
          getObjectPtr<TH1I>("ndf")->Fill(ndf);
          getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
          getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
          if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
            evt0 = m_eventT0->getEventT0();
            getObjectPtr<TH1F>("evt0")->Fill(evt0);
          }
 
          if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(combined);
 
          B2RESULT("Beam vertex constrained fit results NDF = " << ndf << " Chi2/NDF = " << chi2 / double(ndf));
 
        }
      }
    }
  }
 
  for (auto listName : m_twoBodyDecays) {
    StoreObjPtr<ParticleList> list(listName);
    if (!list.isValid())
      continue;
 
    for (unsigned int iParticle = 0; iParticle < list->getListSize(); ++iParticle) {
 
      auto mother = list->getParticle(iParticle);
      auto track12 = getParticlesTracks(mother->getDaughters());
      if (track12.size() != 2) {
        B2ERROR("Did not get 2 fitted tracks. Skipping this mother.");
        continue;
      }
 
      auto pdgdb = EvtGenDatabasePDG::Instance();
      double motherMass = mother->getPDGMass();
      double motherWidth = pdgdb->GetParticle(mother->getPDGCode())->Width();
 
      updateMassWidthIfSet(listName, motherMass, motherWidth);
 
      //TODO: what to take as width for "real" particles? -> make a param for default detector mass resolution??
      if (motherWidth == 0.) {
        motherWidth = m_stableParticleWidth * Unit::GeV;
        B2WARNING("Using artificial width for " << pdgdb->GetParticle(mother->getPDGCode())->GetName() << " : " << motherWidth << " GeV");
      }
 
      auto dfdextPlusMinus = getTwoBodyToLocalTransform(*mother, motherMass);
      std::vector<std::pair<std::vector<gbl::GblPoint>, TMatrixD> > daughters;
 
      daughters.push_back({gbl->collectGblPoints(track12[0], track12[0]->getCardinalRep()), dfdextPlusMinus.first});
      daughters.push_back({gbl->collectGblPoints(track12[1], track12[1]->getCardinalRep()), dfdextPlusMinus.second});
 
      TMatrixDSym massPrec(1); massPrec(0, 0) = 1. / motherWidth / motherWidth;
      TVectorD massResidual(1); massResidual = - (mother->getMass() - motherMass);
 
      TVectorD extMeasurements(1);
      extMeasurements[0] = massResidual[0];
 
      TMatrixD extDeriv(1, 9);
      extDeriv.Zero();
      extDeriv(0, 8) = 1.;
 
      gbl::GblTrajectory combined(daughters, extDeriv, extMeasurements, massPrec);
 
      combined.fit(chi2, ndf, lostWeight);
      //combined.printTrajectory(1000);
      //combined.printPoints(1000);
      getObjectPtr<TH1I>("ndf")->Fill(ndf);
      getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
      getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
      if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
        evt0 = m_eventT0->getEventT0();
        getObjectPtr<TH1F>("evt0")->Fill(evt0);
      }
 
 
      B2RESULT("Mass(PDG) + vertex constrained fit results NDF = " << ndf << " Chi2/NDF = " << chi2 / double(ndf));
 
      if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(combined);
 
    }
  }
 
  for (auto listName : m_primaryMassTwoBodyDecays) {
    StoreObjPtr<ParticleList> list(listName);
    if (!list.isValid())
      continue;
 
    DBObjPtr<BeamParameters> beam;
 
    double motherMass = beam->getMass();
    double motherWidth = sqrt((beam->getCovHER() + beam->getCovLER())(0, 0));
 
    updateMassWidthIfSet(listName, motherMass, motherWidth);
 
    for (unsigned int iParticle = 0; iParticle < list->getListSize(); ++iParticle) {
 
      auto mother = list->getParticle(iParticle);
      auto track12 = getParticlesTracks(mother->getDaughters());
      if (track12.size() != 2) {
        B2ERROR("Did not get 2 fitted tracks. Skipping this mother.");
        continue;
      }
 
      auto dfdextPlusMinus = getTwoBodyToLocalTransform(*mother, motherMass);
      std::vector<std::pair<std::vector<gbl::GblPoint>, TMatrixD> > daughters;
 
      daughters.push_back({gbl->collectGblPoints(track12[0], track12[0]->getCardinalRep()), dfdextPlusMinus.first});
      daughters.push_back({gbl->collectGblPoints(track12[1], track12[1]->getCardinalRep()), dfdextPlusMinus.second});
 
      TMatrixDSym massPrec(1); massPrec(0, 0) = 1. / motherWidth / motherWidth;
      TVectorD massResidual(1); massResidual = - (mother->getMass() - motherMass);
 
      TVectorD extMeasurements(1);
      extMeasurements[0] = massResidual[0];
 
      TMatrixD extDeriv(1, 9);
      extDeriv.Zero();
      extDeriv(0, 8) = 1.;
 
      gbl::GblTrajectory combined(daughters, extDeriv, extMeasurements, massPrec);
 
      combined.fit(chi2, ndf, lostWeight);
      getObjectPtr<TH1I>("ndf")->Fill(ndf);
      getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
      getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
      if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
        evt0 = m_eventT0->getEventT0();
        getObjectPtr<TH1F>("evt0")->Fill(evt0);
      }
 
 
      B2RESULT("Mass constrained fit results NDF = " << ndf << " Chi2/NDF = " << chi2 / double(ndf));
 
      if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(combined);
 
    }
  }
 
  for (auto listName : m_primaryMassVertexTwoBodyDecays) {
    StoreObjPtr<ParticleList> list(listName);
    if (!list.isValid())
      continue;
 
    DBObjPtr<BeamParameters> beam;
 
    double motherMass = beam->getMass();
    double motherWidth = sqrt((beam->getCovHER() + beam->getCovLER())(0, 0));
 
    updateMassWidthIfSet(listName, motherMass, motherWidth);
 
    for (unsigned int iParticle = 0; iParticle < list->getListSize(); ++iParticle) {
 
      auto mother = list->getParticle(iParticle);
      auto track12 = getParticlesTracks(mother->getDaughters());
      if (track12.size() != 2) {
        B2ERROR("Did not get 2 fitted tracks. Skipping this mother.");
        continue;
      }
 
      auto dfdextPlusMinus = getTwoBodyToLocalTransform(*mother, motherMass);
      std::vector<std::pair<std::vector<gbl::GblPoint>, TMatrixD> > daughters;
 
      daughters.push_back({gbl->collectGblPoints(track12[0], track12[0]->getCardinalRep()), dfdextPlusMinus.first});
      daughters.push_back({gbl->collectGblPoints(track12[1], track12[1]->getCardinalRep()), dfdextPlusMinus.second});
 
      TMatrixDSym vertexPrec(get<TMatrixDSym>(getPrimaryVertexAndCov()).Invert());
      B2Vector3D vertexResidual = - (B2Vector3D(mother->getVertex()) - get<B2Vector3D>(getPrimaryVertexAndCov()));
 
      TMatrixDSym massPrec(1); massPrec(0, 0) = 1. / motherWidth / motherWidth;
      TVectorD massResidual(1); massResidual = - (mother->getMass() - motherMass);
 
      TMatrixDSym extPrec(4); extPrec.Zero();
      extPrec.SetSub(0, 0, vertexPrec);
      extPrec(3, 3) = massPrec(0, 0);
 
      TVectorD extMeasurements(4);
      extMeasurements[0] = vertexResidual[0];
      extMeasurements[1] = vertexResidual[1];
      extMeasurements[2] = vertexResidual[2];
      extMeasurements[3] = massResidual[0];
 
      TMatrixD extDeriv(4, 9);
      extDeriv.Zero();
      extDeriv(0, 0) = 1.;
      extDeriv(1, 1) = 1.;
      extDeriv(2, 2) = 1.;
      extDeriv(3, 8) = 1.;
 
      gbl::GblTrajectory combined(daughters, extDeriv, extMeasurements, extPrec);
 
      combined.fit(chi2, ndf, lostWeight);
      getObjectPtr<TH1I>("ndf")->Fill(ndf);
      getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
      getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
      if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
        evt0 = m_eventT0->getEventT0();
        getObjectPtr<TH1F>("evt0")->Fill(evt0);
      }
 
 
      if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(combined);
 
      B2RESULT("Mass + vertex constrained fit results NDF = " << ndf << " Chi2/NDF = " << chi2 / double(ndf));
 
    }
  }
 
  for (auto listName : m_primaryTwoBodyDecays) {
    B2WARNING("This should NOT be used for production of calibration constants for the real detector (yet)!");
 
    StoreObjPtr<ParticleList> list(listName);
    if (!list.isValid())
      continue;
 
    DBObjPtr<BeamParameters> beam;
 
    // For the error of invariant mass M = 2 * sqrt(E_HER * E_LER) (for m_e ~ 0)
    double M = beam->getMass();
    double E_HER = beam->getHER().E();
    double E_LER = beam->getLER().E();
 
    double pz = beam->getHER().Pz() + beam->getLER().Pz();
    double E  = (beam->getHER() + beam->getLER()).E();
 
    double motherMass = beam->getMass();
    double motherWidth = sqrt((E_HER / M) * (E_HER / M) * beam->getCovLER()(0, 0) + (E_LER / M) * (E_LER / M) * beam->getCovHER()(0,
                              0));
 
    updateMassWidthIfSet(listName, motherMass, motherWidth);
 
    for (unsigned int iParticle = 0; iParticle < list->getListSize(); ++iParticle) {
 
      B2WARNING("Two body decays with full kinematic constraint not yet correct - need to resolve strange covariance provided by BeamParameters!");
 
      auto mother = list->getParticle(iParticle);
 
      auto track12 = getParticlesTracks(mother->getDaughters());
      if (track12.size() != 2) {
        B2ERROR("Did not get exactly 2 fitted tracks. Skipping this mother in list " << listName);
        continue;
      }
 
      auto dfdextPlusMinus = getTwoBodyToLocalTransform(*mother, motherMass);
      std::vector<std::pair<std::vector<gbl::GblPoint>, TMatrixD> > daughters;
 
      daughters.push_back({gbl->collectGblPoints(track12[0], track12[0]->getCardinalRep()), dfdextPlusMinus.first});
      daughters.push_back({gbl->collectGblPoints(track12[1], track12[1]->getCardinalRep()), dfdextPlusMinus.second});
 
      TMatrixDSym extCov(7); extCov.Zero();
 
      // 3x3 IP vertex covariance
      extCov.SetSub(0, 0, get<TMatrixDSym>(getPrimaryVertexAndCov()));
 
      // 3x3 boost vector covariance
      //NOTE: BeamSpot return covarince in variables (E, theta_x, theta_y)
      // We need to transform it to our variables (px, py, pz)
 
      TMatrixD dBoost_dVect(3, 3);
      dBoost_dVect(0, 0) = 0.;     dBoost_dVect(0, 1) = 1. / pz; dBoost_dVect(0, 2) = 0.;
      dBoost_dVect(1, 0) = 0.;     dBoost_dVect(1, 1) = 0.;      dBoost_dVect(1, 2) = 1. / pz;
      dBoost_dVect(2, 0) = pz / E; dBoost_dVect(2, 1) = 0.;      dBoost_dVect(2, 2) = 0.;
 
      TMatrixD dVect_dBoost(3, 3);
      dVect_dBoost(0, 0) = 0.;     dVect_dBoost(0, 1) = 0.;      dVect_dBoost(0, 2) = E / pz;
      dVect_dBoost(1, 0) = pz;     dVect_dBoost(1, 1) = 0.;      dVect_dBoost(1, 2) = 0.;
      dVect_dBoost(2, 0) = 0.;     dVect_dBoost(2, 1) = pz;      dVect_dBoost(2, 2) = 0.;
 
      TMatrixD covBoost(3, 3);
      for (int i = 0; i < 3; ++i) {
        for (int j = i; j < 3; ++j) {
          covBoost(j, i) = covBoost(i, j) = (beam->getCovHER() + beam->getCovLER())(i, j);
        }
      }
      //TODO: Temporary fix: if theta_x, theta_y covariance is zero, use arbitrary 10mrad^2
//       if (covBoost(1, 1) == 0.) covBoost(1, 1) = 1.;
//       if (covBoost(2, 2) == 0.) covBoost(2, 2) = 1.;
      if (covBoost(1, 1) == 0.) covBoost(1, 1) = 1.e-4;
      if (covBoost(2, 2) == 0.) covBoost(2, 2) = 1.e-4;
 
      TMatrixD covVect =  dBoost_dVect * covBoost * dVect_dBoost;
 
      extCov.SetSub(3, 3, covVect);
 
      extCov(6, 6) = motherWidth * motherWidth;
      auto extPrec = extCov; extPrec.Invert();
 
      TVectorD extMeasurements(7);
      extMeasurements[0] = - (B2Vector3D(mother->getVertex()) - get<B2Vector3D>(getPrimaryVertexAndCov()))[0];
      extMeasurements[1] = - (B2Vector3D(mother->getVertex()) - get<B2Vector3D>(getPrimaryVertexAndCov()))[1];
      extMeasurements[2] = - (B2Vector3D(mother->getVertex()) - get<B2Vector3D>(getPrimaryVertexAndCov()))[2];
      extMeasurements[3] = - (B2Vector3D(mother->getMomentum()) - (beam->getHER().Vect() + beam->getLER().Vect()))[0];
      extMeasurements[4] = - (B2Vector3D(mother->getMomentum()) - (beam->getHER().Vect() + beam->getLER().Vect()))[1];
      extMeasurements[5] = - (B2Vector3D(mother->getMomentum()) - (beam->getHER().Vect() + beam->getLER().Vect()))[2];
      extMeasurements[6] = - (mother->getMass() - motherMass);
 
      B2INFO("mother mass = " << mother->getMass() << "  and beam mass = " << beam->getMass());
 
      TMatrixD extDeriv(7, 9);
      extDeriv.Zero();
      // beam vertex constraint
      extDeriv(0, 0) = 1.;
      extDeriv(1, 1) = 1.;
      extDeriv(2, 2) = 1.;
      // beam kinematics constraint
      extDeriv(3, 3) = 1.;
      extDeriv(4, 4) = 1.;
      extDeriv(5, 5) = 1.;
      // beam inv. mass constraint
      extDeriv(6, 8) = 1;
 
      if (m_calibrateVertex || m_calibrateKinematics) {
        B2WARNING("Primary vertex+kinematics calibration not (yet?) fully implemented!");
        B2WARNING("This code is highly experimental and has (un)known issues!");
 
        // up to d(x,y,z,px,py,pz,theta,phi,M)/d(vx,vy,vz,theta_x,theta_y,E)
        TMatrixD derivatives(9, 6);
        std::vector<int> labels;
        derivatives.Zero();
 
        if (m_calibrateVertex) {
          derivatives(0, 0) = 1.;
          derivatives(1, 1) = 1.;
          derivatives(2, 2) = 1.;
          GlobalLabel label = GlobalLabel::construct<BeamSpot>(0, 0);
          labels.push_back(label.setParameterId(1));
          labels.push_back(label.setParameterId(2));
          labels.push_back(label.setParameterId(3));
        } else {
          labels.push_back(0);
          labels.push_back(0);
          labels.push_back(0);
        }
 
        if (m_calibrateKinematics) {
          derivatives(3, 3) = mother->getMomentumMagnitude();
          derivatives(4, 4) = mother->getMomentumMagnitude();
          derivatives(8, 5) = (beam->getLER().E() + beam->getHER().E()) / beam->getMass();
 
          GlobalLabel label = GlobalLabel::construct<BeamSpot>(0, 0);
          labels.push_back(label.setParameterId(4)); //theta_x
          labels.push_back(label.setParameterId(5)); //theta_y
          labels.push_back(label.setParameterId(6)); //E
 
        } else {
          labels.push_back(0);
          labels.push_back(0);
          labels.push_back(0);
        }
 
        // Allow to disable BeamSpot externally
        alignment::GlobalDerivatives globals(labels, derivatives);
 
        // Add derivatives for vertex calibration to first point of first trajectory
        // NOTE: use GlobalDerivatives operators vector<int> and TMatrixD which filter
        // the derivatives to not pass those with zero labels (useful to get rid of some params)
        std::vector<int> lab(globals); TMatrixD der(globals);
 
        // I want: dlocal/dext = dlocal/dtwobody * dtwobody/dext = dfdextPlusMinus * dtwobody/dext
        TMatrixD dTwoBody_dExt(9, 7);
        dTwoBody_dExt.Zero();
        // beam vertex constraint
        dTwoBody_dExt(0, 0) = 1.;
        dTwoBody_dExt(1, 1) = 1.;
        dTwoBody_dExt(2, 2) = 1.;
        // beam kinematics constraint
        dTwoBody_dExt(3, 3) = 1.;
        dTwoBody_dExt(4, 4) = 1.;
        dTwoBody_dExt(5, 5) = 1.;
        // beam inv. mass constraint
        dTwoBody_dExt(8, 6) = 1.;
 
        const TMatrixD dLocal_dExt = dfdextPlusMinus.first * dTwoBody_dExt;
        TMatrixD dLocal_dExt_T = dLocal_dExt; dLocal_dExt_T.T();
 
        // The 5x7 transformation matrix d(q/p,u',v',u,v)/d(vx,vy,vz,px,py,pz,M) needs to be "inverted"
        // to transform the covariance of the beamspot and boost vector of SuperKEKB into the local system
        // of one GBL point - such that Millepede can align the beamspot (or even beam kinematics) if requested.
        //
        // I tested also other methods, but only the Singular Value Decomposition gives nice-enough results,
        // with almost no code:
        //
        TDecompSVD svd(dLocal_dExt_T);
        TMatrixD dExt_dLocal  = svd.Invert().T();
        //
        // (dLocal_dExt * dExt_dLocal).Print(); // Check how close we are to unit matrix
        //
        // 5x5 matrix is as follows
        //
        //      |      0    |      1    |      2    |      3    |      4    |
        // ----------------------------------------------------------------------
        //    0 |          1   -2.58e-17   6.939e-18   1.571e-17  -1.649e-19
        //    1 |  1.787e-14           1   5.135e-16  -3.689e-16  -2.316e-18
        //    2 | -1.776e-15  -7.806e-17           1   5.636e-17   6.193e-18
        //    3 | -2.453e-15    7.26e-18   2.009e-16           1   -1.14e-16
        //    4 | -1.689e-14  -9.593e-17  -2.317e-15  -3.396e-17           1
        //
        // It took me half a day to find out how to do this with 2 lines of code (3 with the include).
        // Source: ROOT macro example - actually found at:
        // <https://root.cern.ch/root/html/tutorials/matrix/solveLinear.C.html>
        for (int i = 0; i < 7; ++i) {
          for (int j = 0; j < 5; ++j) {
            if (fabs(dExt_dLocal(i, j)) < 1.e-6)
              dExt_dLocal(i, j) = 0.;
          }
        }
        const TVectorD locRes = dLocal_dExt * extMeasurements;
        const TMatrixD locPrec =  dLocal_dExt * extPrec * dExt_dLocal;
 
        TMatrixDSym locPrecSym(5); locPrecSym.Zero();
        for (int i = 0; i < 5; ++i) {
          for (int j = i; j < 5; ++j) {
            //locPrecSym(j, i) = locPrecSym(i, j) = locPrec(i, j);
            locPrecSym(j, i) = locPrecSym(i, j) = (fabs(locPrec(i, j)) > 1.e-6) ? locPrec(i, j) : 0.;
          }
        }
 
        daughters[0].first[0].addMeasurement(locRes, locPrecSym);
        if (!lab.empty())
          daughters[0].first[0].addGlobals(lab, dfdextPlusMinus.first * der);
 
        //TODO: Understand this: either find a bug somewhere or improve the parametrization or .... ?
        // This should be enough, but the parametrization seems to fail for nearly horizontal pairs...
        //gbl::GblTrajectory combined(daughters);
        // This should not be needed, it actually seems to make worse Chi2/NDF, but GBL does not fail.
        // The measurement added just to be able to add the global derivatives (done just above) is redundant
        // to the external measurement added here:
        gbl::GblTrajectory combined(daughters, extDeriv, extMeasurements, extPrec);
        //combined.printTrajectory(1000);
        //combined.printPoints(1000);
 
        combined.fit(chi2, ndf, lostWeight);
        getObjectPtr<TH1I>("ndf")->Fill(ndf);
        getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
        getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
        if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
          evt0 = m_eventT0->getEventT0();
          getObjectPtr<TH1F>("evt0")->Fill(evt0);
        }
 
 
        B2RESULT("Full kinematic-constrained fit (calibration version) results NDF = " << ndf << " Chi2/NDF = " << chi2 / double(ndf));
 
        if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(combined);
 
      } else {
 
        gbl::GblTrajectory combined(daughters, extDeriv, extMeasurements, extPrec);
        //combined.printTrajectory(1000);
        //combined.printPoints(1000);
 
        combined.fit(chi2, ndf, lostWeight);
        getObjectPtr<TH1I>("ndf")->Fill(ndf);
        getObjectPtr<TH1F>("chi2_per_ndf")->Fill(chi2 / double(ndf));
        getObjectPtr<TH1F>("pval")->Fill(TMath::Prob(chi2, ndf));
        if (m_eventT0.isValid() && m_eventT0->hasEventT0()) {
          evt0 = m_eventT0->getEventT0();
          getObjectPtr<TH1F>("evt0")->Fill(evt0);
        }
 
 
        B2RESULT("Full kinematic-constrained fit results NDF = " << ndf << " Chi2/NDF = " << chi2 / double(ndf));
 
        if (TMath::Prob(chi2, ndf) > m_minPValue) storeTrajectory(combined);
      }
    }
  }
}

def_beginRun()

virtual void def_beginRun ( )

inlineprotectedvirtualinherited

Wrapper method for the virtual function beginRun() that has the implementation to be used in a call from Python.

Reimplemented in PyModule.

Definition at line 425 of file Module.h.

{ beginRun(); }

def_endRun()

virtual void def_endRun ( )

inlineprotectedvirtualinherited

This method can receive that the current run ends as a call from the Python side.

For regular C++-Modules that forwards the call to the regular endRun() method.

Reimplemented in PyModule.

Definition at line 438 of file Module.h.

{ endRun(); }

def_event()

virtual void def_event ( )

inlineprotectedvirtualinherited

Wrapper method for the virtual function event() that has the implementation to be used in a call from Python.

Reimplemented in PyModule.

Definition at line 431 of file Module.h.

{ event(); }

def_initialize()

virtual void def_initialize ( )

inlineprotectedvirtualinherited

Wrappers to make the methods without "def_" prefix callable from Python.

Overridden in PyModule. Wrapper method for the virtual function initialize() that has the implementation to be used in a call from Python.

Reimplemented in PyModule.

Definition at line 419 of file Module.h.

{ initialize(); }

def_terminate()

virtual void def_terminate ( )

inlineprotectedvirtualinherited

Wrapper method for the virtual function terminate() that has the implementation to be used in a call from Python.

Reimplemented in PyModule.

Definition at line 444 of file Module.h.

{ terminate(); }

defineHisto()

void defineHisto ( )

finalvirtualinherited

Runs due to HistoManager, allows us to discover the correct file.

Reimplemented from HistoModule.

Definition at line 127 of file CalibrationCollectorModule.cc.

{
  if (!ProcHandler::parallelProcessingUsed() || ProcHandler::isWorkerProcess()) {
    m_dir = gDirectory->mkdir(getName().c_str(), "", true);
    m_manager.setDirectory(m_dir);
    B2INFO("Saving output to TDirectory " << m_dir->GetPath());
    B2DEBUG(100, "Creating directories for individual collector objects.");
    m_manager.createDirectories();
    m_runRange = new RunRange();
    m_runRange->setGranularity(m_granularity);
    m_runRange->SetName(Calibration::RUN_RANGE_OBJ_NAME.c_str());
    m_dir->Add(m_runRange);
  }
  inDefineHisto();
}

endRun()

void endRun ( void  )

finalvirtualinherited

Write the current collector objects to a file and clear their memory.

Reimplemented from HistoModule.

Definition at line 143 of file CalibrationCollectorModule.cc.

{
  closeRun();
  // Moving between runs possibly creates new objects if getObjectPtr is called and granularity is run
  // So we should write and clear the current memory objects.
  if (m_granularity == "run") {
    ExpRun expRun = make_pair(m_emd->getExperiment(), m_emd->getRun());
    m_manager.writeCurrentObjects(expRun);
    m_manager.clearCurrentObjects(expRun);
  }
}

evalCondition()

bool evalCondition ( ) const

inherited

If at least one condition was set, it is evaluated and true returned if at least one condition returns true.

If no condition or result value was defined, the method returns false. Otherwise, the condition is evaluated and true returned, if at least one condition returns true. To speed up the evaluation, the condition strings were already parsed in the method if_value().

Returns

True if at least one condition and return value exists and at least one condition expression was evaluated to true.

Definition at line 96 of file Module.cc.

{
  if (m_conditions.empty()) return false;
 
  //okay, a condition was set for this Module...
  if (!m_hasReturnValue) {
    B2FATAL("A condition was set for '" << getName() << "', but the module did not set a return value!");
  }
 
  for (const auto& condition : m_conditions) {
    if (condition.evaluate(m_returnValue)) {
      return true;
    }
  }
  return false;
}

event()

void event ( void  )

finalvirtualinherited

Check current experiment and run and update if needed, fill into RunRange and collect()

Reimplemented from HistoModule.

Definition at line 52 of file CalibrationCollectorModule.cc.

{
  // Should we collect data this event based on the number collected in the run?
  if (m_runCollectOnRun) {
    // If yes, does our preScale return true?
    if (getPreScaleChoice()) {
      collect();
      // Since we collected, do we care about incrementing the number of events collected?
      if (m_maxEventsPerRun > -1) {
        (*m_eventsCollectedInRun) += 1;
        // Now that we incremented, have we exceeded our maximum collected events in this run?
        if ((*m_eventsCollectedInRun) >= m_maxEventsPerRun) {
          // If we have, we should skip collection until further notice
          B2INFO("Reached maximum number of events processed by collector for this run ("
                 << (*m_eventsCollectedInRun)
                 << " >= "
                 << m_maxEventsPerRun
                 << "). Turning off collection.");
          m_runCollectOnRun = false;
        }
      }
    }
  }
}

exposePythonAPI()

void exposePythonAPI ( )

staticinherited

Exposes methods of the Module class to Python.

Definition at line 325 of file Module.cc.

{
  // to avoid confusion between std::arg and boost::python::arg we want a shorthand namespace as well
  namespace bp = boost::python;
 
  docstring_options options(true, true, false); //userdef, py sigs, c++ sigs
 
  void (Module::*setReturnValueInt)(int) = &Module::setReturnValue;
 
  enum_<Module::EAfterConditionPath>("AfterConditionPath",
                                     R"(Determines execution behaviour after a conditional path has been executed:
 
.. attribute:: END
 
  End processing of this path after the conditional path. (this is the default for if_value() etc.)
 
.. attribute:: CONTINUE
 
  After the conditional path, resume execution after this module.)")
  .value("END", Module::EAfterConditionPath::c_End)
  .value("CONTINUE", Module::EAfterConditionPath::c_Continue)
  ;
 
  /* Do not change the names of >, <, ... we use them to serialize conditional paths */
  enum_<Belle2::ModuleCondition::EConditionOperators>("ConditionOperator")
  .value(">", Belle2::ModuleCondition::EConditionOperators::c_GT)
  .value("<", Belle2::ModuleCondition::EConditionOperators::c_ST)
  .value(">=", Belle2::ModuleCondition::EConditionOperators::c_GE)
  .value("<=", Belle2::ModuleCondition::EConditionOperators::c_SE)
  .value("==", Belle2::ModuleCondition::EConditionOperators::c_EQ)
  .value("!=", Belle2::ModuleCondition::EConditionOperators::c_NE)
  ;
 
  enum_<Module::EModulePropFlags>("ModulePropFlags",
                                  R"(Flags to indicate certain low-level features of modules, see :func:`Module.set_property_flags()`, :func:`Module.has_properties()`. Most useful flags are:
 
.. attribute:: PARALLELPROCESSINGCERTIFIED
 
    This module can be run in parallel processing mode safely (All I/O must be done through the data store, in particular, the module must not write any files.)
 
.. attribute:: HISTOGRAMMANAGER
 
  This module is used to manage histograms accumulated by other modules
 
.. attribute:: TERMINATEINALLPROCESSES
 
  When using parallel processing, call this module's terminate() function in all processes. This will also ensure that there is exactly one process (single-core if no parallel modules found) or at least one input, one main and one output process.
)")
  .value("INPUT", Module::EModulePropFlags::c_Input)
  .value("OUTPUT", Module::EModulePropFlags::c_Output)
  .value("PARALLELPROCESSINGCERTIFIED", Module::EModulePropFlags::c_ParallelProcessingCertified)
  .value("HISTOGRAMMANAGER", Module::EModulePropFlags::c_HistogramManager)
  .value("INTERNALSERIALIZER", Module::EModulePropFlags::c_InternalSerializer)
  .value("TERMINATEINALLPROCESSES", Module::EModulePropFlags::c_TerminateInAllProcesses)
  ;
 
  //Python class definition
  class_<Module, PyModule> module("Module", R"(
Base class for Modules.
 
A module is the smallest building block of the framework.
A typical event processing chain consists of a Path containing
modules. By inheriting from this base class, various types of
modules can be created. To use a module, please refer to
:func:`Path.add_module()`.  A list of modules is available by running
``basf2 -m`` or ``basf2 -m package``, detailed information on parameters is
given by e.g. ``basf2 -m RootInput``.
 
The 'Module Development' section in the manual provides detailed information
on how to create modules, setting parameters, or using return values/conditions:
https://xwiki.desy.de/xwiki/rest/p/f4fa4/#HModuleDevelopment
 
)");
  module
  .def("__str__", &Module::getPathString)
  .def("name", &Module::getName, return_value_policy<copy_const_reference>(),
       "Returns the name of the module. Can be changed via :func:`set_name() <Module.set_name()>`, use :func:`type() <Module.type()>` for identifying a particular module class.")
  .def("type", &Module::getType, return_value_policy<copy_const_reference>(),
       "Returns the type of the module (i.e. class name minus 'Module')")
  .def("set_name", &Module::setName, args("name"), R"(
Set custom name, e.g. to distinguish multiple modules of the same type.
 
>>> path.add_module('EventInfoSetter')
>>> ro = path.add_module('RootOutput', branchNames=['EventMetaData'])
>>> ro.set_name('RootOutput_metadata_only')
>>> print(path)
[EventInfoSetter -> RootOutput_metadata_only]
 
)")
  .def("description", &Module::getDescription, return_value_policy<copy_const_reference>(),
       "Returns the description of this module.")
  .def("package", &Module::getPackage, return_value_policy<copy_const_reference>(),
       "Returns the package this module belongs to.")
  .def("available_params", &_getParamInfoListPython,
       "Return list of all module parameters as `ModuleParamInfo` instances")
  .def("has_properties", &Module::hasProperties, (bp::arg("properties")),
       R"DOCSTRING(Allows to check if the module has the given properties out of `ModulePropFlags` set.
 
>>> if module.has_properties(ModulePropFlags.PARALLELPROCESSINGCERTIFIED):
>>>     ...
 
Parameters:
  properties (int): bitmask of `ModulePropFlags` to check for.
)DOCSTRING")
  .def("set_property_flags", &Module::setPropertyFlags, args("property_mask"),
       "Set module properties in the form of an OR combination of `ModulePropFlags`.");
  {
    // python signature is too crowded, make ourselves
    docstring_options subOptions(true, false, false); //userdef, py sigs, c++ sigs
    module
    .def("if_value", &Module::if_value,
         (bp::arg("expression"), bp::arg("condition_path"), bp::arg("after_condition_path")= Module::EAfterConditionPath::c_End),
         R"DOCSTRING(if_value(expression, condition_path, after_condition_path=AfterConditionPath.END)
 
Sets a conditional sub path which will be executed after this
module if the return value set in the module passes the given ``expression``.
 
Modules can define a return value (int or bool) using ``setReturnValue()``,
which can be used in the steering file to split the Path based on this value, for example
 
>>> module_with_condition.if_value("<1", another_path)
 
In case the return value of the ``module_with_condition`` for a given event is
less than 1, the execution will be diverted into ``another_path`` for this event.
 
You could for example set a special return value if an error occurs, and divert
the execution into a path containing :b2:mod:`RootOutput` if it is found;
saving only the data producing/produced by the error.
 
After a conditional path has executed, basf2 will by default stop processing
the path for this event. This behaviour can be changed by setting the
``after_condition_path`` argument.
 
Parameters:
  expression (str): Expression to determine if the conditional path should be executed.
      This should be one of the comparison operators ``<``, ``>``, ``<=``,
      ``>=``, ``==``, or ``!=`` followed by a numerical value for the return value
  condition_path (Path): path to execute in case the expression is fulfilled
  after_condition_path (AfterConditionPath): What to do once the ``condition_path`` has been executed.
)DOCSTRING")
    .def("if_false", &Module::if_false,
         (bp::arg("condition_path"), bp::arg("after_condition_path")= Module::EAfterConditionPath::c_End),
         R"DOC(if_false(condition_path, after_condition_path=AfterConditionPath.END)
 
Sets a conditional sub path which will be executed after this module if
the return value of the module evaluates to False. This is equivalent to
calling `if_value` with ``expression=\"<1\"``)DOC")
    .def("if_true", &Module::if_true,
         (bp::arg("condition_path"), bp::arg("after_condition_path")= Module::EAfterConditionPath::c_End),
         R"DOC(if_true(condition_path, after_condition_path=AfterConditionPath.END)
 
Sets a conditional sub path which will be executed after this module if
the return value of the module evaluates to True. It is equivalent to
calling `if_value` with ``expression=\">=1\"``)DOC");
  }
  module
  .def("has_condition", &Module::hasCondition,
       "Return true if a conditional path has been set for this module "
       "using `if_value`, `if_true` or `if_false`")
  .def("get_all_condition_paths", &_getAllConditionPathsPython,
       "Return a list of all conditional paths set for this module using "
       "`if_value`, `if_true` or `if_false`")
  .def("get_all_conditions", &_getAllConditionsPython,
       "Return a list of all conditional path expressions set for this module using "
       "`if_value`, `if_true` or `if_false`")
  .add_property("logging", make_function(&Module::getLogConfig, return_value_policy<reference_existing_object>()),
                &Module::setLogConfig)

finish()

void finish ( )

overridevirtual

Register mille binaries in file catalog.

Reimplemented from CalibrationCollectorModule.

Definition at line 995 of file MillepedeCollectorModule.cc.

{
  Belle2::alignment::GlobalCalibrationManager::getInstance().writeConstraints("constraints.txt");
 
  StoreObjPtr<FileMetaData> fileMetaData("", DataStore::c_Persistent);
  if (!fileMetaData.isValid()) {
    B2ERROR("Cannot register binaries in FileCatalog.");
    return;
  }
 
 
  const std::vector<string> parents = {fileMetaData->getLfn()};
  for (auto binary : getObjectPtr<MilleData>("mille")->getFiles()) {
    FileMetaData milleMetaData(*fileMetaData);
    // We reset filename to be set directly by the registerFile procedure
    milleMetaData.setLfn("");
    milleMetaData.setParents(parents);
    FileCatalog::Instance().registerFile(binary, milleMetaData);
  }
 
}

fitRecoTrack()

bool fitRecoTrack	(	RecoTrack &	recoTrack,
		Particle *	particle = nullptr
	)

Fit given RecoTrack with GBL.

Parameters

recoTrack	A RecoTrack object to be fitted
particle	Pointer to reconstructed daughter particle updated by vertex fit OR nullptr for single track

Returns

true for success, false when some problems occurred (or track too much down-weighted by previous DAF fit)

Definition at line 1048 of file MillepedeCollectorModule.cc.

{
  try {
    // For already fitted tracks, try to get fitted (DAF) weights for CDC
    if (m_updateCDCWeights && recoTrack.getNumberOfCDCHits() && recoTrack.getTrackFitStatus()
        && recoTrack.getTrackFitStatus()->isFitted()) {
      double sumCDCWeights = recoTrack.getNumberOfCDCHits(); // start with full weights
      // Do the hits synchronisation
      auto relatedRecoHitInformation =
        recoTrack.getRelationsTo<RecoHitInformation>(recoTrack.getStoreArrayNameOfRecoHitInformation());
 
      for (RecoHitInformation& recoHitInformation : relatedRecoHitInformation) {
 
        if (recoHitInformation.getFlag() == RecoHitInformation::c_pruned) {
          B2FATAL("Found pruned point in RecoTrack. Pruned tracks cannot be used in MillepedeCollector.");
        }
 
        if (recoHitInformation.getTrackingDetector() != RecoHitInformation::c_CDC) continue;
 
        const genfit::TrackPoint* trackPoint = recoTrack.getCreatedTrackPoint(&recoHitInformation);
        if (trackPoint) {
          if (not trackPoint->hasFitterInfo(recoTrack.getCardinalRepresentation()))
            continue;
          auto kalmanFitterInfo = dynamic_cast<genfit::KalmanFitterInfo*>(trackPoint->getFitterInfo());
          if (not kalmanFitterInfo) {
            continue;
          } else {
            std::vector<double> weights = kalmanFitterInfo->getWeights();
            if (weights.size() == 2) {
              if (weights.at(0) > weights.at(1))
                recoHitInformation.setRightLeftInformation(RecoHitInformation::c_left);
              else if (weights.at(0) < weights.at(1))
                recoHitInformation.setRightLeftInformation(RecoHitInformation::c_right);
 
              double weightLR = weights.at(0) + weights.at(1);
              if (weightLR < m_minCDCHitWeight)  recoHitInformation.setUseInFit(false);
              sumCDCWeights += weightLR - 1.; // reduce weight sum if weightLR<1
            }
          }
        }
      }
 
      double usedCDCHitFraction = sumCDCWeights / double(recoTrack.getNumberOfCDCHits());
      getObjectPtr<TH1F>("cdc_hit_fraction")->Fill(usedCDCHitFraction);
      if (usedCDCHitFraction < m_minUsedCDCHitFraction)
        return false;
    }
  } catch (...) {
    B2ERROR("Error in checking DAF weights from previous fit to resolve hit ambiguity. Why? Failed fit points in DAF? Skip track to be sure.");
    return false;
  }
 
  std::shared_ptr<genfit::GblFitter> gbl(new genfit::GblFitter());
  gbl->setOptions(m_internalIterations, true, true, m_externalIterations, m_recalcJacobians);
  gbl->setTrackSegmentController(new GblMultipleScatteringController);
 
  MeasurementAdder factory("", "", "", "", "");
 
  // We need the store arrays
  StoreArray<RecoHitInformation::UsedCDCHit> cdcHits("");
  StoreArray<RecoHitInformation::UsedPXDHit> pxdHits("");
  StoreArray<RecoHitInformation::UsedSVDHit> svdHits("");
  StoreArray<RecoHitInformation::UsedBKLMHit> bklmHits("");
  StoreArray<RecoHitInformation::UsedEKLMHit> eklmHits("");
 
  // Create the genfit::MeasurementFactory
  genfit::MeasurementFactory<genfit::AbsMeasurement> genfitMeasurementFactory;
 
  // Add producer for alignable RecoHits to factory
  if (pxdHits.isOptional()) {
    genfit::MeasurementProducer <RecoHitInformation::UsedPXDHit, AlignablePXDRecoHit>* PXDProducer =  new genfit::MeasurementProducer
    <RecoHitInformation::UsedPXDHit, AlignablePXDRecoHit> (pxdHits.getPtr());
    genfitMeasurementFactory.addProducer(Const::PXD, PXDProducer);
  }
 
  if (svdHits.isOptional())  {
    genfit::MeasurementProducer <RecoHitInformation::UsedSVDHit, AlignableSVDRecoHit>* SVDProducer =  new genfit::MeasurementProducer
    <RecoHitInformation::UsedSVDHit, AlignableSVDRecoHit> (svdHits.getPtr());
    genfitMeasurementFactory.addProducer(Const::SVD, SVDProducer);
  }
 
  if (cdcHits.isOptional()) {
    genfit::MeasurementProducer <RecoHitInformation::UsedCDCHit, AlignableCDCRecoHit>* CDCProducer =  new genfit::MeasurementProducer
    <RecoHitInformation::UsedCDCHit, AlignableCDCRecoHit> (cdcHits.getPtr());
    genfitMeasurementFactory.addProducer(Const::CDC, CDCProducer);
  }
 
  if (bklmHits.isOptional()) {
    genfit::MeasurementProducer <RecoHitInformation::UsedBKLMHit, AlignableBKLMRecoHit>* BKLMProducer =  new genfit::MeasurementProducer
    <RecoHitInformation::UsedBKLMHit, AlignableBKLMRecoHit> (bklmHits.getPtr());
    genfitMeasurementFactory.addProducer(Const::BKLM, BKLMProducer);
  }
 
  if (eklmHits.isOptional()) {
    genfit::MeasurementProducer <RecoHitInformation::UsedEKLMHit, AlignableEKLMRecoHit>* EKLMProducer =  new genfit::MeasurementProducer
    <RecoHitInformation::UsedEKLMHit, AlignableEKLMRecoHit> (eklmHits.getPtr());
    genfitMeasurementFactory.addProducer(Const::EKLM, EKLMProducer);
  }
 
 
  // Create the measurement creators
  std::vector<std::shared_ptr<PXDBaseMeasurementCreator>> pxdMeasurementCreators = { std::shared_ptr<PXDBaseMeasurementCreator>(new PXDCoordinateMeasurementCreator(genfitMeasurementFactory)) };
  std::vector<std::shared_ptr<SVDBaseMeasurementCreator>> svdMeasurementCreators = { std::shared_ptr<SVDBaseMeasurementCreator>(new SVDCoordinateMeasurementCreator(genfitMeasurementFactory)) };
  // TODO: Create a new MeasurementCreator based on SVDBaseMeasurementCreator (or on SVDCoordinateMeasurementCreator), which does the combination on the fly.
 
  std::vector<std::shared_ptr<CDCBaseMeasurementCreator>> cdcMeasurementCreators = { std::shared_ptr<CDCBaseMeasurementCreator>(new CDCCoordinateMeasurementCreator(genfitMeasurementFactory)) };
  std::vector<std::shared_ptr<BKLMBaseMeasurementCreator>> bklmMeasurementCreators = { std::shared_ptr<BKLMBaseMeasurementCreator>(new BKLMCoordinateMeasurementCreator(genfitMeasurementFactory)) };
  std::vector<std::shared_ptr<EKLMBaseMeasurementCreator>> eklmMeasurementCreators = { std::shared_ptr<EKLMBaseMeasurementCreator>(new EKLMCoordinateMeasurementCreator(genfitMeasurementFactory)) };
 
  // TODO: Or put it in here and leave the svdMeasurementCreators empty.
  std::vector<std::shared_ptr<BaseMeasurementCreator>> additionalMeasurementCreators = {};
  factory.resetMeasurementCreators(pxdMeasurementCreators, svdMeasurementCreators, cdcMeasurementCreators, bklmMeasurementCreators,
                                   eklmMeasurementCreators, additionalMeasurementCreators);
  factory.addMeasurements(recoTrack);
 
  auto& gfTrack = RecoTrackGenfitAccess::getGenfitTrack(recoTrack);
 
  int currentPdgCode = TrackFitter::createCorrectPDGCodeForChargedStable(Const::muon, recoTrack);
  if (particle)
    currentPdgCode = particle->getPDGCode();
 
  genfit::AbsTrackRep* trackRep = RecoTrackGenfitAccess::createOrReturnRKTrackRep(recoTrack, currentPdgCode);
  gfTrack.setCardinalRep(gfTrack.getIdForRep(trackRep));
 
  if (particle) {
    B2Vector3D vertexPos = particle->getVertex();
    B2Vector3D vertexMom = particle->getMomentum();
    gfTrack.setStateSeed(vertexPos, vertexMom);
 
    genfit::StateOnPlane vertexSOP(gfTrack.getCardinalRep());
    B2Vector3D vertexRPhiDir(vertexPos[0], vertexPos[1], 0);
    B2Vector3D vertexZDir(0, 0, vertexPos[2]);
    //FIXME: This causes problem to current GBL version in genfit -> needs update of GBL to re-enable
    // genfit::SharedPlanePtr vertexPlane(new genfit::DetPlane(vertexPos, vertexRPhiDir, vertexZDir));
    //This works instead fine:
    genfit::SharedPlanePtr vertexPlane(new genfit::DetPlane(vertexPos, vertexMom));
 
    vertexSOP.setPlane(vertexPlane);
    vertexSOP.setPosMom(vertexPos, vertexMom);
    TMatrixDSym vertexCov(5);
    vertexCov.UnitMatrix();
    // By using negative covariance no measurement is added to GBL. But this first point
    // is then used as additional point in trajectory at the assumed point of its fitted vertex
    vertexCov *= -1.;
    genfit::MeasuredStateOnPlane mop(vertexSOP, vertexCov);
    genfit::FullMeasurement* vertex = new genfit::FullMeasurement(mop, Const::IR);
    gfTrack.insertMeasurement(vertex, 0);
  }
 
  try {
    for (unsigned int i = 0; i < gfTrack.getNumPoints() - 1; ++i) {
      //if (gfTrack.getPointWithMeasurement(i)->getNumRawMeasurements() != 1)
      //  continue;
      genfit::PlanarMeasurement* planarMeas1 = dynamic_cast<genfit::PlanarMeasurement*>(gfTrack.getPointWithMeasurement(
                                                 i)->getRawMeasurement(0));
      genfit::PlanarMeasurement* planarMeas2 = dynamic_cast<genfit::PlanarMeasurement*>(gfTrack.getPointWithMeasurement(
                                                 i + 1)->getRawMeasurement(0));
 
      if (planarMeas1 != NULL && planarMeas2 != NULL &&
          planarMeas1->getDetId() == planarMeas2->getDetId() &&
          planarMeas1->getPlaneId() != -1 &&   // -1 is default plane id
          planarMeas1->getPlaneId() == planarMeas2->getPlaneId()) {
        Belle2::AlignableSVDRecoHit* hit1 = dynamic_cast<Belle2::AlignableSVDRecoHit*>(planarMeas1);
        Belle2::AlignableSVDRecoHit* hit2 = dynamic_cast<Belle2::AlignableSVDRecoHit*>(planarMeas2);
        if (hit1 && hit2) {
          Belle2::AlignableSVDRecoHit* hitU(NULL);
          Belle2::AlignableSVDRecoHit* hitV(NULL);
          // We have to decide U/V now (else AlignableSVDRecoHit2D could throw FATAL)
          if (hit1->isU() && !hit2->isU()) {
            hitU = hit1;
            hitV = hit2;
          } else if (!hit1->isU() && hit2->isU()) {
            hitU = hit2;
            hitV = hit1;
          } else {
            continue;
          }
          Belle2::AlignableSVDRecoHit2D* hit = new Belle2::AlignableSVDRecoHit2D(*hitU, *hitV);
          // insert measurement before point i (increases number of correct point to i+1)
          gfTrack.insertMeasurement(hit, i);
          // now delete current point (at its original place, we have the new 2D recohit)
          gfTrack.deletePoint(i + 1);
          gfTrack.deletePoint(i + 1);
        }
      }
    }
  } catch (std::exception& e) {
    B2ERROR(e.what());
    B2ERROR("SVD Cluster combination failed. This is symptomatic of pruned tracks. MillepedeCollector cannot process pruned tracks.");
    return false;
  }
 
  try {
    gbl->processTrackWithRep(&gfTrack, gfTrack.getCardinalRep(), true);
  } catch (genfit::Exception& e) {
    B2ERROR(e.what());
    return false;
  } catch (...) {
    B2ERROR("GBL fit failed.");
    return false;
  }
 
  return true;
}

getAfterConditionPath()

Module::EAfterConditionPath getAfterConditionPath ( ) const

inherited

What to do after the conditional path is finished.

(defaults to c_End if no condition is set)

Definition at line 133 of file Module.cc.

{
  if (m_conditions.empty()) return EAfterConditionPath::c_End;
 
  //okay, a condition was set for this Module...
  if (!m_hasReturnValue) {
    B2FATAL("A condition was set for '" << getName() << "', but the module did not set a return value!");
  }
 
  for (const auto& condition : m_conditions) {
    if (condition.evaluate(m_returnValue)) {
      return condition.getAfterConditionPath();
    }
  }
 
  return EAfterConditionPath::c_End;
}

getAllConditionPaths()

std::vector< std::shared_ptr< Path > > getAllConditionPaths ( ) const

inherited

Return all condition paths currently set (no matter if the condition is true or not).

Definition at line 150 of file Module.cc.

{
  std::vector<std::shared_ptr<Path>> allConditionPaths;
  for (const auto& condition : m_conditions) {
    allConditionPaths.push_back(condition.getPath());
  }
 
  return allConditionPaths;
}

getAllConditions()

const std::vector< ModuleCondition > & getAllConditions ( ) const

inlineinherited

Return all set conditions for this module.

Definition at line 323 of file Module.h.

    {
      return m_conditions;
    }

getCondition()

const ModuleCondition * getCondition ( ) const

inlineinherited

Return a pointer to the first condition (or nullptr, if none was set)

Definition at line 313 of file Module.h.

    {
      if (m_conditions.empty()) {
        return nullptr;
      } else {
        return &m_conditions.front();
      }
    }

getConditionPath()

std::shared_ptr< Path > getConditionPath ( ) const

inherited

Returns the path of the last true condition (if there is at least one, else reaturn a null pointer).

Definition at line 113 of file Module.cc.

{
  PathPtr p;
  if (m_conditions.empty()) return p;
 
  //okay, a condition was set for this Module...
  if (!m_hasReturnValue) {
    B2FATAL("A condition was set for '" << getName() << "', but the module did not set a return value!");
  }
 
  for (const auto& condition : m_conditions) {
    if (condition.evaluate(m_returnValue)) {
      return condition.getPath();
    }
  }
 
  // if none of the conditions were true, return a null pointer.
  return p;
}

getDescription()

const std::string & getDescription ( ) const

inlineinherited

Returns the description of the module.

Definition at line 201 of file Module.h.

{return m_description;}

getFileNames()

virtual std::vector< std::string > getFileNames ( bool  outputFiles )

inlinevirtualinherited

Return a list of output filenames for this modules.

This will be called when basf2 is run with "--dry-run" if the module has set either the c_Input or c_Output properties.

If the parameter outputFiles is false (for modules with c_Input) the list of input filenames should be returned (if any). If outputFiles is true (for modules with c_Output) the list of output files should be returned (if any).

If a module has sat both properties this member is called twice, once for each property.

The module should return the actual list of requested input or produced output filenames (including handling of input/output overrides) so that the grid system can handle input/output files correctly.

This function should return the same value when called multiple times. This is especially important when taking the input/output overrides from Environment as they get consumed when obtained so the finalized list of output files should be stored for subsequent calls.

Reimplemented in RootInputModule, StorageRootOutputModule, and RootOutputModule.

Definition at line 133 of file Module.h.

    {
      return std::vector<std::string>();
    }

getGlobalToLocalTransform()

TMatrixD getGlobalToLocalTransform

(

const genfit::MeasuredStateOnPlane &

msop

)

Compute the transformation matrix d(q/p,u',v',u,v)/d(x,y,z,px,py,pz) from state at first track point (vertex)

Parameters

msop	MeasuredStateOnPlane - linearization point (track state @ plane) at which the transformation should be computed

Definition at line 1447 of file MillepedeCollectorModule.cc.

{
  auto state = msop;
  const B2Vector3D& U(state.getPlane()->getU());
  const B2Vector3D& V(state.getPlane()->getV());
  const B2Vector3D& O(state.getPlane()->getO());
  const B2Vector3D& W(state.getPlane()->getNormal());
 
  const double* state5 = state.getState().GetMatrixArray();
 
  double spu = 1.;
 
  const TVectorD& auxInfo = state.getAuxInfo();
  if (auxInfo.GetNrows() == 2
      || auxInfo.GetNrows() == 1) // backwards compatibility with old RKTrackRep
    spu = state.getAuxInfo()(0);
 
  TVectorD state7(7);
 
  state7[0] = O.X() + state5[3] * U.X() + state5[4] * V.X(); // x
  state7[1] = O.Y() + state5[3] * U.Y() + state5[4] * V.Y(); // y
  state7[2] = O.Z() + state5[3] * U.Z() + state5[4] * V.Z(); // z
 
  state7[3] = spu * (W.X() + state5[1] * U.X() + state5[2] * V.X()); // a_x
  state7[4] = spu * (W.Y() + state5[1] * U.Y() + state5[2] * V.Y()); // a_y
  state7[5] = spu * (W.Z() + state5[1] * U.Z() + state5[2] * V.Z()); // a_z
 
  // normalize dir
  double norm = 1. / sqrt(state7[3] * state7[3] + state7[4] * state7[4] + state7[5] * state7[5]);
  for (unsigned int i = 3; i < 6; ++i) state7[i] *= norm;
 
  state7[6] = state5[0]; // q/p
 
  const double AtU = state7[3] * U.X() + state7[4] * U.Y() + state7[5] * U.Z();
  const double AtV = state7[3] * V.X() + state7[4] * V.Y() + state7[5] * V.Z();
  const double AtW = state7[3] * W.X() + state7[4] * W.Y() + state7[5] * W.Z();
 
  // J_Mp matrix is d(q/p,u',v',u,v) / d(x,y,z,px,py,pz)       (in is 6x6)
 
  const double qop = state7[6];
  const double p = state.getCharge() / qop; // momentum
 
  TMatrixD J_Mp_6x5(6, 5);
  J_Mp_6x5.Zero();
 
  //d(u)/d(x,y,z)
  J_Mp_6x5(0, 3)  = U.X(); // [0][3]
  J_Mp_6x5(1, 3)  = U.Y(); // [1][3]
  J_Mp_6x5(2, 3) = U.Z(); // [2][3]
  //d(v)/d(x,y,z)
  J_Mp_6x5(0, 4)  = V.X(); // [0][4]
  J_Mp_6x5(1, 4)  = V.Y(); // [1][4]
  J_Mp_6x5(2, 4) = V.Z(); // [2][4]
 
  // d(q/p)/d(px,py,pz)
  double fact = (-1.) * qop / p;
  J_Mp_6x5(3, 0) = fact * state7[3]; // [3][0]
  J_Mp_6x5(4, 0) = fact * state7[4]; // [4][0]
  J_Mp_6x5(5, 0) = fact * state7[5]; // [5][0]
  // d(u')/d(px,py,pz)
  fact = 1. / (p * AtW * AtW);
  J_Mp_6x5(3, 1) = fact * (U.X() * AtW - W.X() * AtU); // [3][1]
  J_Mp_6x5(4, 1) = fact * (U.Y() * AtW - W.Y() * AtU); // [4][1]
  J_Mp_6x5(5, 1) = fact * (U.Z() * AtW - W.Z() * AtU); // [5][1]
  // d(v')/d(px,py,pz)
  J_Mp_6x5(3, 2) = fact * (V.X() * AtW - W.X() * AtV); // [3][2]
  J_Mp_6x5(4, 2) = fact * (V.Y() * AtW - W.Y() * AtV); // [4][2]
  J_Mp_6x5(5, 2) = fact * (V.Z() * AtW - W.Z() * AtV); // [5][2]
 
  return J_Mp_6x5.T();
}

getLocalToGlobalTransform()

TMatrixD getLocalToGlobalTransform

(

const genfit::MeasuredStateOnPlane &

msop

)

Compute the transformation matrix d(x,y,z,px,py,pz)/d(q/p,u',v',u,v) from state at first track point (vertex)

Parameters

msop	MeasuredStateOnPlane - linearization point (track state @ plane) at which the transformation should be computed

Definition at line 1519 of file MillepedeCollectorModule.cc.

{
  auto state = msop;
  // get vectors and aux variables
  const B2Vector3D& U(state.getPlane()->getU());
  const B2Vector3D& V(state.getPlane()->getV());
  const B2Vector3D& W(state.getPlane()->getNormal());
 
  const TVectorD& state5(state.getState());
  double spu = 1.;
 
  const TVectorD& auxInfo = state.getAuxInfo();
  if (auxInfo.GetNrows() == 2
      || auxInfo.GetNrows() == 1) // backwards compatibility with old RKTrackRep
    spu = state.getAuxInfo()(0);
 
  TVectorD pTilde(3);
  pTilde[0] = spu * (W.X() + state5(1) * U.X() + state5(2) * V.X()); // a_x
  pTilde[1] = spu * (W.Y() + state5(1) * U.Y() + state5(2) * V.Y()); // a_y
  pTilde[2] = spu * (W.Z() + state5(1) * U.Z() + state5(2) * V.Z()); // a_z
 
  const double pTildeMag = sqrt(pTilde[0] * pTilde[0] + pTilde[1] * pTilde[1] + pTilde[2] * pTilde[2]);
  const double pTildeMag2 = pTildeMag * pTildeMag;
 
  const double utpTildeOverpTildeMag2 = (U.X() * pTilde[0] + U.Y() * pTilde[1] + U.Z() * pTilde[2]) / pTildeMag2;
  const double vtpTildeOverpTildeMag2 = (V.X() * pTilde[0] + V.Y() * pTilde[1] + V.Z() * pTilde[2]) / pTildeMag2;
 
  //J_pM matrix is d(x,y,z,px,py,pz) / d(q/p,u',v',u,v)       (out is 6x6)
 
  const double qop = state5(0);
  const double p = state.getCharge() / qop; // momentum
 
  TMatrixD J_pM_5x6(5, 6);
  J_pM_5x6.Zero();
 
  // d(px,py,pz)/d(q/p)
  double fact = -1. * p / (pTildeMag * qop);
  J_pM_5x6(0, 3) = fact * pTilde[0]; // [0][3]
  J_pM_5x6(0, 4) = fact * pTilde[1]; // [0][4]
  J_pM_5x6(0, 5) = fact * pTilde[2]; // [0][5]
  // d(px,py,pz)/d(u')
  fact = p * spu / pTildeMag;
  J_pM_5x6(1, 3)  = fact * (U.X() - pTilde[0] * utpTildeOverpTildeMag2); // [1][3]
  J_pM_5x6(1, 4) = fact * (U.Y() - pTilde[1] * utpTildeOverpTildeMag2); // [1][4]
  J_pM_5x6(1, 5) = fact * (U.Z() - pTilde[2] * utpTildeOverpTildeMag2); // [1][5]
  // d(px,py,pz)/d(v')
  J_pM_5x6(2, 3) = fact * (V.X() - pTilde[0] * vtpTildeOverpTildeMag2); // [2][3]
  J_pM_5x6(2, 4) = fact * (V.Y() - pTilde[1] * vtpTildeOverpTildeMag2); // [2][4]
  J_pM_5x6(2, 5) = fact * (V.Z() - pTilde[2] * vtpTildeOverpTildeMag2); // [2][5]
  // d(x,y,z)/d(u)
  J_pM_5x6(3, 0) = U.X(); // [3][0]
  J_pM_5x6(3, 1) = U.Y(); // [3][1]
  J_pM_5x6(3, 2) = U.Z(); // [3][2]
  // d(x,y,z)/d(v)
  J_pM_5x6(4, 0) = V.X(); // [4][0]
  J_pM_5x6(4, 1) = V.Y(); // [4][1]
  J_pM_5x6(4, 2) = V.Z(); // [4][2]
 
  return J_pM_5x6.T();
 
}

getLogConfig()

LogConfig & getLogConfig ( )

inlineinherited

Returns the log system configuration.

Definition at line 224 of file Module.h.

{return m_logConfig;}

getModules()

std::list< ModulePtr > getModules ( ) const

inlineoverrideprivatevirtualinherited

no submodules, return empty list

Implements PathElement.

Definition at line 505 of file Module.h.

{ return std::list<ModulePtr>(); }

getName()

const std::string & getName ( ) const

inlineinherited

Returns the name of the module.

This can be changed via e.g. set_name() in the steering file to give more useful names if there is more than one module of the same type.

For identifying the type of a module, using getType() (or type() in Python) is recommended.

Definition at line 186 of file Module.h.

{return m_name;}

getObjectPtr()

T * getObjectPtr ( std::string  name )

inlineinherited

Calls the CalibObjManager to get the requested stored collector data.

Definition at line 64 of file CalibrationCollectorModule.h.

    {
      return m_manager.getObject<T>(name, m_expRun);
    }

getPackage()

const std::string & getPackage ( ) const

inlineinherited

Returns the package this module is in.

Definition at line 196 of file Module.h.

{return m_package;}

getParamInfoListPython()

std::shared_ptr< boost::python::list > getParamInfoListPython ( ) const

inherited

Returns a python list of all parameters.

Each item in the list consists of the name of the parameter, a string describing its type, a python list of all default values and the description of the parameter.

Returns

A python list containing the parameters of this parameter list.

Definition at line 279 of file Module.cc.

{
  return m_moduleParamList.getParamInfoListPython();
}

getParamList()

const ModuleParamList & getParamList ( ) const

inlineinherited

Return module param list.

Definition at line 362 of file Module.h.

{ return m_moduleParamList; }

getParticlesTracks()

std::vector< genfit::Track * > getParticlesTracks	(	std::vector< Particle * >	particles,
		bool	addVertexPoint = true
	)

Get all usable tracks for particles.

Parameters

particles	vector of Belle2::Particles to be changed in vector of genfit::Tracks
addVertexPoint	flag for adding the vertex point

Definition at line 1253 of file MillepedeCollectorModule.cc.

{
  std::vector< genfit::Track* > tracks;
  for (auto particle : particles) {
    auto belle2Track = particle->getTrack();
    if (!belle2Track) {
      B2WARNING("No Belle2::Track for particle (particle->X");
      continue;
    }
//     auto trackFitResult = belle2Track->getTrackFitResult(Const::chargedStableSet.find(abs(particle->getPDGCode())));
//     if (!trackFitResult) {
//       B2INFO("No track fit result for track");
//       continue;
//     }
//    auto recoTrack = trackFitResult->getRelatedFrom<RecoTrack>();
    auto recoTrack = belle2Track->getRelatedTo<RecoTrack>();
 
    if (!recoTrack) {
      B2WARNING("No related RecoTrack for Belle2::Track (particle->Track->X)");
      continue;
    }
 
    // If any track fails, fail completely
    if (!fitRecoTrack(*recoTrack, (addVertexPoint) ? particle : nullptr))
      return {};
 
    auto& track = RecoTrackGenfitAccess::getGenfitTrack(*recoTrack);
 
    if (!track.hasFitStatus()) {
      B2WARNING("Track has no fit status");
      continue;
    }
    genfit::GblFitStatus* fs = dynamic_cast<genfit::GblFitStatus*>(track.getFitStatus());
    if (!fs) {
      B2WARNING("Track FitStatus is not GblFitStatus.");
      continue;
    }
    if (!fs->isFittedWithReferenceTrack()) {
      B2WARNING("Track is not fitted with reference track.");
      continue;
    }
 
    tracks.push_back(&track);
  }
 
  return tracks;
}

getPathString()

std::string getPathString ( ) const

overrideprivatevirtualinherited

return the module name.

Implements PathElement.

Definition at line 192 of file Module.cc.

{
 
  std::string output = getName();
 
  for (const auto& condition : m_conditions) {
    output += condition.getString();
  }
 
  return output;
}

getPreScaleChoice()

bool getPreScaleChoice ( )

inlineprivateinherited

I'm a little worried about floating point precision when comparing to 0.0 and 1.0 as special values.

But since a user will have set them (or left them as default) as exactly equal to 0.0 or 1.0 rather than calculating them in almost every case, I think we can assume that the equalities hold.

Definition at line 122 of file CalibrationCollectorModule.h.

    {
      if (m_preScale == 1.) {
        return true;
      } else if (m_preScale == 0.) {
        return false;
      } else {
        const double randomNumber = gRandom->Uniform();
        return randomNumber < m_preScale;
      }
    }

getPrimaryVertexAndCov()

tuple< B2Vector3D, TMatrixDSym > getPrimaryVertexAndCov

(

)

const

Get the primary vertex position estimation and its size from BeamSpot.

Returns

tuple<B2Vector3D, TMatrixDSym> tuple with position and size as covariance matrix

Definition at line 1581 of file MillepedeCollectorModule.cc.

{
  DBObjPtr<BeamSpot> beam;
  return {beam->getIPPosition(), beam->getSizeCovMatrix()};
}

getReturnValue()

int getReturnValue ( ) const

inlineinherited

Return the return value set by this module.

This value is only meaningful if hasReturnValue() is true

Definition at line 380 of file Module.h.

{ return m_returnValue; }

getTwoBodyToLocalTransform()

std::pair< TMatrixD, TMatrixD > getTwoBodyToLocalTransform	(	Particle &	mother,
		double	motherMass
	)

Compute the transformation matrices d(q/p,u'v',u,v)/d(vx,vy,vz,px,py,pz,theta,phi,M) = dq/d(v,z) for both particles in pair.

Only for decays of type V0(*)->f+f- (same mass for f)

Parameters

mother	The mother Belle2::Particle with two daughters, its 4-momenta should already be updated by a previous vertex fit done by modularAnalysis.
motherMass	This function expect the assumed invariant mass of the pair. This is to allow to set artificial values (e.g. for e+e- -> mu+mu-)

Returns

a pair of 5x9 matrices {dq+/d(v,z), dq-/d(v,z)}. One for each particle in list (in list order). NOTE: The signs DO NOT refer to charges of the particles! If you want to know: (+) particle is that one which goes along the mother momentum in CM system

The transformation is from local measurement system at 1st (GBL) point of each track in pair (virtual measurement (see fitRecoTrack(..., particle) and addVertexPoint parameter of getParticlesTracks(...)) to the common parameters which staticaly and kinematicaly describe the two-body decay:

• Position of the common vertex (vy,vy,vz)
• Total momentum of the pair (particles are back-to-back in their CM) (px,py,pz) and the invariant mass (M) of the decay
• 2 angles describing the orientation of the decay particles in the system of the mother (CM)

Reference: Widl, Edmund ; Frühwirth R; "Representation and Estimation of Trajectories from Two-body Decays", CMS-NOTE-2007-032, http://cds.cern.ch/record/1073690

Definition at line 1301 of file MillepedeCollectorModule.cc.

{
  std::vector<TMatrixD> result;
 
  double px = mother.getPx();
  double py = mother.getPy();
  double pz = mother.getPz();
  double pt = sqrt(px * px + py * py);
  double p  = mother.getMomentumMagnitude();
  double M  = motherMass;
  double m  = mother.getDaughter(0)->getPDGMass();
 
  if (mother.getNDaughters() != 2
      || m != mother.getDaughter(1)->getPDGMass()) B2FATAL("Only two same-mass daughters (V0->f+f- decays) allowed.");
 
  // Rotation matrix from mother reference system to lab system
  TMatrixD mother2lab(3, 3);
  mother2lab(0, 0) = px * pz / pt / p; mother2lab(0, 1) = - py / pt; mother2lab(0, 2) = px / p;
  mother2lab(1, 0) = py * pz / pt / p; mother2lab(1, 1) =   px / pt; mother2lab(1, 2) = py / p;
  mother2lab(2, 0) = - pt / p;         mother2lab(2, 1) =   0;       mother2lab(2, 2) = pz / p;
  ROOT::Math::Rotation3D lab2mother;
  lab2mother.SetRotationMatrix(mother2lab); lab2mother.Invert();
 
  // Need to rotate and boost daughters' momenta to know which goes forward (+sign in decay model)
  // and to get the angles theta, phi of the decaying daughter system in mothers' reference frame
  RestFrame boostedFrame(&mother);
  ROOT::Math::PxPyPzEVector fourVector1 = mother.getDaughter(0)->get4Vector();
  ROOT::Math::PxPyPzEVector fourVector2 = mother.getDaughter(1)->get4Vector();
 
  auto mom1 = lab2mother * boostedFrame.getMomentum(fourVector1).Vect();
  auto mom2 = lab2mother * boostedFrame.getMomentum(fourVector2).Vect();
  // One momentum has opposite direction (otherwise should be same in CMS of mother), but which?
  double sign = 1.;
  auto avgMom = 0.5 * (mom1 - mom2);
  if (avgMom.Z() < 0.) {
    avgMom *= -1.;
    // switch meaning of plus/minus trajectories
    sign = -1.;
  }
 
  double theta = atan2(avgMom.rho(), avgMom.Z());
  double phi = atan2(avgMom.Y(), avgMom.X());
  if (phi < 0.) phi += 2. * TMath::Pi();
 
  double alpha = M / 2. / m;
  double c1 = m * sqrt(alpha * alpha - 1.);
  double c2 = 0.5 * sqrt((alpha * alpha - 1.) / alpha / alpha * (p * p + M * M));
 
  double p3 = p * p * p;
  double pt3 = pt * pt * pt;
 
 
  for (auto& track : getParticlesTracks(mother.getDaughters())) {
 
 
    TMatrixD R = mother2lab;
    B2Vector3D P(sign * c1 * sin(theta) * cos(phi),
                 sign * c1 * sin(theta) * sin(phi),
                 p / 2. + sign * c2 * cos(theta));
 
    TMatrixD dRdpx(3, 3);
    dRdpx(0, 0) = - pz * (pow(px, 4.) - pow(py, 4.) - py * py * pz * pz) / pt3 / p3;
    dRdpx(0, 1) = px * py / pt3;
    dRdpx(0, 2) = (py * py + pz * pz) / p3;
 
    dRdpx(1, 0) = - px * py * pz * (2. * px * px + 2. * py * py + pz * pz) / pt3 / p3;
    dRdpx(1, 1) = - py * py / pt3;
    dRdpx(1, 2) = px * py / p3;
 
    dRdpx(2, 0) = - px * pz * pz / pt / p3;
    dRdpx(2, 1) = 0.;
    dRdpx(2, 2) = - px * pz / p3;
 
    TMatrixD dRdpy(3, 3);
    dRdpy(0, 0) = - px * py * pz * (2. * px * px + 2. * py * py + pz * pz) / pt3 / p3;
    dRdpy(0, 1) = - px * px / pt3;
    dRdpy(0, 2) = px * pz / p3;
 
    dRdpy(1, 0) = - pz * (- pow(px, 4.) - px * px * pz * pz + pow(py, 4.)) / pt3 / p3;
    dRdpy(1, 1) = px * py / pt3;
    dRdpy(1, 2) = (px * px + pz * pz) / p3;
 
    dRdpy(2, 0) = - py * pz * pz / pt / p3;
    dRdpy(2, 1) = 0.;
    dRdpy(2, 2) = - py * pz / p3;
 
    TMatrixD dRdpz(3, 3);
    dRdpz(0, 0) = px * pt / p3;
    dRdpz(0, 1) = 0.;
    dRdpz(0, 2) = - px * pz / p3;
 
    dRdpz(1, 0) = py * pt / p3;
    dRdpz(1, 1) = 0.;
    dRdpz(1, 2) = py * pz / p3;
 
    dRdpz(2, 0) = pz * pt / p3;
    dRdpz(2, 1) = 0.;
    dRdpz(2, 2) = (px * px + py * py) / p3;
 
    auto K = 1. / 2. / p + sign * cos(theta) * m * m * (M * M / 4. / m / m - 1.) / M / M / sqrt(m * m * (M * M / 4. / m / m - 1.) *
             (M * M + p * p) / M / M);
 
    B2Vector3D dpdpx = dRdpx * P + R * K * px * B2Vector3D(0., 0., 1.);
    B2Vector3D dpdpy = dRdpy * P + R * K * py * B2Vector3D(0., 0., 1.);
    B2Vector3D dpdpz = dRdpz * P + R * K * pz * B2Vector3D(0., 0., 1.);
 
    B2Vector3D dpdtheta = R * B2Vector3D(sign * c1 * cos(theta) * cos(phi),
                                         sign * c1 * cos(theta) * sin(phi),
                                         sign * c2 * (- sin(theta)));
 
 
    B2Vector3D dpdphi = R * B2Vector3D(sign * c1 * sin(theta) * (- sin(phi)),
                                       sign * c1 * sin(theta) * cos(phi),
                                       0.);
 
    double dc1dM = m * M / (2. * sqrt(M * M - 4. * m * m));
    double dc2dM = M * (4. * m * m * p * p + pow(M, 4)) / (2 * M * M * M * sqrt((M * M - 4. * m * m) * (p * p + M * M)));
 
    B2Vector3D dpdM = R * B2Vector3D(sign * sin(theta) * cos(phi) * dc1dM,
                                     sign * sin(theta) * sin(phi) * dc1dM,
                                     sign * cos(theta) * dc2dM);
 
    TMatrixD dpdz(3, 6);
    dpdz(0, 0) = dpdpx(0); dpdz(0, 1) = dpdpy(0); dpdz(0, 2) = dpdpz(0); dpdz(0, 3) = dpdtheta(0); dpdz(0, 4) = dpdphi(0);
    dpdz(0, 5) = dpdM(0);
    dpdz(1, 0) = dpdpx(1); dpdz(1, 1) = dpdpy(1); dpdz(1, 2) = dpdpz(1); dpdz(1, 3) = dpdtheta(1); dpdz(1, 4) = dpdphi(1);
    dpdz(1, 5) = dpdM(1);
    dpdz(2, 0) = dpdpx(2); dpdz(2, 1) = dpdpy(2); dpdz(2, 2) = dpdpz(2); dpdz(2, 3) = dpdtheta(2); dpdz(2, 4) = dpdphi(2);
    dpdz(2, 5) = dpdM(2);
 
    TMatrixD dqdv = getGlobalToLocalTransform(track->getFittedState()).GetSub(0, 4, 0, 2);
    TMatrixD dqdp = getGlobalToLocalTransform(track->getFittedState()).GetSub(0, 4, 3, 5);
    TMatrixD dfdvz(5, 9);
    dfdvz.SetSub(0, 0, dqdv);
    dfdvz.SetSub(0, 3, dqdp * dpdz);
 
    result.push_back(dfdvz);
 
    // switch sign for second trajectory
    sign *= -1.;
  }
 
  return {result[0], result[1]};
}

getType()

const std::string & getType ( ) const

inherited

Returns the type of the module (i.e.

class name minus 'Module')

Definition at line 41 of file Module.cc.

{
  if (m_type.empty())
    B2FATAL("Module type not set for " << getName());
  return m_type;
}

getUniqueMilleName()

std::string getUniqueMilleName

(

)

Make a name for mille binary (encodes module name + starting exp, run and event + process id)

Definition at line 1032 of file MillepedeCollectorModule.cc.

{
  string name = getName();
 
  name += "-e"   + to_string(m_evtMetaData->getExperiment());
  name += "-r"   + to_string(m_evtMetaData->getRun());
  name += "-ev"  + to_string(m_evtMetaData->getEvent());
 
  if (ProcHandler::parallelProcessingUsed())
    name += "-pid" + to_string(ProcHandler::EvtProcID());
 
  name += ".mille";
 
  return name;
}

hasCondition()

bool hasCondition ( ) const

inlineinherited

Returns true if at least one condition was set for the module.

Definition at line 310 of file Module.h.

{ return not m_conditions.empty(); };

hasProperties()

bool hasProperties ( unsigned int  propertyFlags ) const

inherited

Returns true if all specified property flags are available in this module.

Parameters

propertyFlags

Ored EModulePropFlags which should be compared with the module flags.

Definition at line 160 of file Module.cc.

{
  return (propertyFlags & m_propertyFlags) == propertyFlags;
}

hasReturnValue()

bool hasReturnValue ( ) const

inlineinherited

Return true if this module has a valid return value set.

Definition at line 377 of file Module.h.

{ return m_hasReturnValue; }

hasUnsetForcedParams()

bool hasUnsetForcedParams ( ) const

inherited

Returns true and prints error message if the module has unset parameters which the user has to set in the steering file.

Definition at line 166 of file Module.cc.

{
  auto missing = m_moduleParamList.getUnsetForcedParams();
  std::string allMissing = "";
  for (const auto& s : missing)
    allMissing += s + " ";
  if (!missing.empty())
    B2ERROR("The following required parameters of Module '" << getName() << "' were not specified: " << allMissing <<
            "\nPlease add them to your steering file.");
  return !missing.empty();
}

if_false()

void if_false ( const std::shared_ptr< Path > &  path, EAfterConditionPath  afterConditionPath = EAfterConditionPath::c_End  )

inherited

A simplified version to add a condition to the module.

Please note that successive calls of this function will add more than one condition to the module. If more than one condition results in true, only the last of them will be used.

Please be careful: Avoid creating cyclic paths, e.g. by linking a condition to a path which is processed before the path where this module is located in.

It is equivalent to the if_value() method, using the expression "<1". This method is meant to be used together with the setReturnValue(bool value) method.

Parameters

path	Shared pointer to the Path which will be executed if the return value is false.
afterConditionPath	What to do after executing 'path'.

Definition at line 85 of file Module.cc.

{
  if_value("<1", path, afterConditionPath);
}

if_true()

void if_true ( const std::shared_ptr< Path > &  path, EAfterConditionPath  afterConditionPath = EAfterConditionPath::c_End  )

inherited

A simplified version to set the condition of the module.

Please note that successive calls of this function will add more than one condition to the module. If more than one condition results in true, only the last of them will be used.

Please be careful: Avoid creating cyclic paths, e.g. by linking a condition to a path which is processed before the path where this module is located in.

It is equivalent to the if_value() method, using the expression ">=1". This method is meant to be used together with the setReturnValue(bool value) method.

Parameters

path	Shared pointer to the Path which will be executed if the return value is true.
afterConditionPath	What to do after executing 'path'.

Definition at line 90 of file Module.cc.

{
  if_value(">=1", path, afterConditionPath);
}

if_value()

void if_value ( const std::string &  expression, const std::shared_ptr< Path > &  path, EAfterConditionPath  afterConditionPath = EAfterConditionPath::c_End  )

inherited

Add a condition to the module.

Please note that successive calls of this function will add more than one condition to the module. If more than one condition results in true, only the last of them will be used.

See https://xwiki.desy.de/xwiki/rest/p/a94f2 or ModuleCondition for a description of the syntax.

Please be careful: Avoid creating cyclic paths, e.g. by linking a condition to a path which is processed before the path where this module is located in.

Parameters

expression	The expression of the condition.
path	Shared pointer to the Path which will be executed if the condition is evaluated to true.
afterConditionPath	What to do after executing 'path'.

Definition at line 79 of file Module.cc.

{
  m_conditions.emplace_back(expression, path, afterConditionPath);
}

inDefineHisto()

virtual void inDefineHisto ( )

inlineprotectedvirtualinherited

Replacement for defineHisto(). Do anything you would normally do in defineHisto here.

Reimplemented in CaTestModule, ECLBhabhaTCollectorModule, eclBhabhaTimeCalibrationValidationCollectorModule, eclHadronTimeCalibrationValidationCollectorModule, and eclTimeShiftsPlottingCollectorModule.

Definition at line 81 of file CalibrationCollectorModule.h.

{}

initialize()

void initialize ( void  )

finalvirtualinherited

Set up a default RunRange object in datastore and call prepare()

Reimplemented from HistoModule.

Definition at line 44 of file CalibrationCollectorModule.cc.

{
  m_evtMetaData.isRequired();
  REG_HISTOGRAM
  prepare();
}

prepare()

void prepare ( )

overridevirtual

Prepration.

Reimplemented from CalibrationCollectorModule.

Definition at line 158 of file MillepedeCollectorModule.cc.

{
  m_eventT0.isOptional();
 
  if (m_tracks.empty() &&
      m_particles.empty() &&
      m_vertices.empty() &&
      m_primaryVertices.empty() &&
      m_twoBodyDecays.empty() &&
      m_primaryTwoBodyDecays.empty() &&
      m_primaryMassTwoBodyDecays.empty() &&
      m_primaryMassVertexTwoBodyDecays.empty())
    B2ERROR("You have to specify either arrays of single tracks or particle lists of single single particles or mothers with vertex constrained daughters.");
 
  if (!m_tracks.empty()) {
    for (auto arrayName : m_tracks)
      continue;
    // StoreArray<RecoTrack>::required(arrayName);
  }
 
  if (!m_particles.empty() || !m_vertices.empty() || !m_primaryVertices.empty()) {
    // StoreArray<RecoTrack> recoTracks;
    // StoreArray<Track> tracks;
    // StoreArray<TrackFitResult> trackFitResults;
 
    //recoTracks.isRequired();
    //tracks.isRequired();
    //trackFitResults.isRequired();
  }
 
  for (auto listName : m_particles) {
    StoreObjPtr<ParticleList> list(listName);
    //list.isRequired();
  }
 
  for (auto listName : m_vertices) {
    StoreObjPtr<ParticleList> list(listName);
    //list.isRequired();
  }
 
  for (auto listName : m_primaryVertices) {
    StoreObjPtr<ParticleList> list(listName);
    //list.isRequired();
  }
 
  // Register Mille output
  registerObject<MilleData>("mille", new MilleData(m_doublePrecision, m_absFilePaths));
 
  auto gblDataTree = new TTree("GblDataTree", "GblDataTree");
  gblDataTree->Branch<std::vector<gbl::GblData>>("GblData", &m_currentGblData, 32000, 99);
  registerObject<TTree>("GblDataTree", gblDataTree);
 
  registerObject<TH1I>("ndf", new TH1I("ndf", "ndf", 200, 0, 200));
  registerObject<TH1F>("chi2_per_ndf", new TH1F("chi2_per_ndf", "chi2 divided by ndf", 200, 0., 50.));
  registerObject<TH1F>("pval", new TH1F("pval", "pval", 100, 0., 1.));
 
  registerObject<TH1F>("cdc_hit_fraction", new TH1F("cdc_hit_fraction", "cdc_hit_fraction", 100, 0., 1.));
  registerObject<TH1F>("evt0", new TH1F("evt0", "evt0", 400, -100., 100.));
 
  // Configure the (VXD) hierarchy before being built
  if (m_hierarchyType == 0)
    Belle2::alignment::VXDGlobalParamInterface::s_hierarchyType = VXDGlobalParamInterface::c_None;
  else if (m_hierarchyType == 1)
    Belle2::alignment::VXDGlobalParamInterface::s_hierarchyType = VXDGlobalParamInterface::c_Flat;
  else if (m_hierarchyType == 2)
    Belle2::alignment::VXDGlobalParamInterface::s_hierarchyType = VXDGlobalParamInterface::c_HalfShells;
  else if (m_hierarchyType == 3)
    Belle2::alignment::VXDGlobalParamInterface::s_hierarchyType = VXDGlobalParamInterface::c_Full;
 
  Belle2::alignment::VXDGlobalParamInterface::s_enablePXD = m_enablePXDHierarchy;
  Belle2::alignment::VXDGlobalParamInterface::s_enableSVD = m_enableSVDHierarchy;
 
  std::vector<EventMetaData> events;
  for (auto& ev_run_exp : m_eventNumbers) {
    events.push_back(EventMetaData(std::get<0>(ev_run_exp), std::get<1>(ev_run_exp), std::get<2>(ev_run_exp)));
  }
 
  // This will also build the hierarchy for the first time:
  if (!m_timedepConfig.empty() && m_eventNumbers.empty()) {
    auto autoEvents = Belle2::alignment::timeline::setupTimedepGlobalLabels(m_timedepConfig);
    Belle2::alignment::GlobalCalibrationManager::getInstance().initialize(m_components, autoEvents);
  } else if (m_timedepConfig.empty() && !m_eventNumbers.empty()) {
    Belle2::alignment::GlobalCalibrationManager::getInstance().initialize(m_components, events);
  } else if (m_timedepConfig.empty() && m_eventNumbers.empty()) {
    Belle2::alignment::GlobalCalibrationManager::getInstance().initialize(m_components);
  } else {
    B2ERROR("Cannot set both, event list and timedep config.");
  }
 
//   Belle2::alignment::GlobalCalibrationManager::getInstance().writeConstraints("constraints.txt");
 
  AlignableCDCRecoHit::s_enableTrackT0LocalDerivative = m_fitTrackT0;
  AlignableCDCRecoHit::s_enableWireSaggingGlobalDerivative = m_enableWireSagging;
  AlignableCDCRecoHit::s_enableWireByWireAlignmentGlobalDerivatives = m_enableWireByWireAlignment;
}

registerObject()

void registerObject ( std::string  name, T *  obj  )

inlineinherited

Register object with a name, takes ownership, do not access the pointer beyond prepare()

Definition at line 55 of file CalibrationCollectorModule.h.

    {
      std::shared_ptr<T> calObj(obj);
      calObj->SetName(name.c_str());
      m_manager.addObject(name, calObj);
    }

setAbortLevel()

void setAbortLevel ( int  abortLevel )

inherited

Configure the abort log level.

Definition at line 67 of file Module.cc.

{
  m_logConfig.setAbortLevel(static_cast<LogConfig::ELogLevel>(abortLevel));
}

setDebugLevel()

void setDebugLevel ( int  debugLevel )

inherited

Configure the debug messaging level.

Definition at line 61 of file Module.cc.

{
  m_logConfig.setDebugLevel(debugLevel);
}

setDescription()

void setDescription ( const std::string &  description )

protectedinherited

Sets the description of the module.

Parameters

description

A description of the module.

Definition at line 214 of file Module.cc.

{
  m_description = description;
}

setLogConfig()

void setLogConfig ( const LogConfig &  logConfig )

inlineinherited

Set the log system configuration.

Definition at line 229 of file Module.h.

{m_logConfig = logConfig;}

setLogInfo()

void setLogInfo ( int  logLevel, unsigned int  logInfo  )

inherited

Configure the printed log information for the given level.

Parameters

logLevel	The log level (one of LogConfig::ELogLevel)
logInfo	What kind of info should be printed? ORed combination of LogConfig::ELogInfo flags.

Definition at line 73 of file Module.cc.

{
  m_logConfig.setLogInfo(static_cast<LogConfig::ELogLevel>(logLevel), logInfo);
}

setLogLevel()

void setLogLevel ( int  logLevel )

inherited

Configure the log level.

Definition at line 55 of file Module.cc.

{
  m_logConfig.setLogLevel(static_cast<LogConfig::ELogLevel>(logLevel));
}

setName()

void setName ( const std::string &  name )

inlineinherited

Set the name of the module.

Note

The module name is set when using the REG_MODULE macro, but the module can be renamed before calling process() using the set_name() function in your steering file.

Parameters

name	The name of the module

Definition at line 213 of file Module.h.

{ m_name = name; };

setParamList()

void setParamList ( const ModuleParamList &  params )

inlineprotectedinherited

Replace existing parameter list.

Definition at line 500 of file Module.h.

{ m_moduleParamList = params; }

setParamPython()

void setParamPython ( const std::string &  name, const boost::python::object &  pyObj  )

privateinherited

Implements a method for setting boost::python objects.

The method supports the following types: list, dict, int, double, string, bool The conversion of the python object to the C++ type and the final storage of the parameter value is done in the ModuleParam class.

Parameters

name	The unique name of the parameter.
pyObj	The object which should be converted and stored as the parameter value.

Definition at line 234 of file Module.cc.

{
  LogSystem& logSystem = LogSystem::Instance();
  logSystem.updateModule(&(getLogConfig()), getName());
  try {
    m_moduleParamList.setParamPython(name, pyObj);
  } catch (std::runtime_error& e) {
    throw std::runtime_error("Cannot set parameter '" + name + "' for module '"
                             + m_name + "': " + e.what());
  }
 
  logSystem.updateModule(nullptr);
}

setParamPythonDict()

void setParamPythonDict ( const boost::python::dict &  dictionary )

privateinherited

Implements a method for reading the parameter values from a boost::python dictionary.

The key of the dictionary has to be the name of the parameter and the value has to be of one of the supported parameter types.

Parameters

dictionary

The python dictionary from which the parameter values are read.

Definition at line 249 of file Module.cc.

{
 
  LogSystem& logSystem = LogSystem::Instance();
  logSystem.updateModule(&(getLogConfig()), getName());
 
  boost::python::list dictKeys = dictionary.keys();
  int nKey = boost::python::len(dictKeys);
 
  //Loop over all keys in the dictionary
  for (int iKey = 0; iKey < nKey; ++iKey) {
    boost::python::object currKey = dictKeys[iKey];
    boost::python::extract<std::string> keyProxy(currKey);
 
    if (keyProxy.check()) {
      const boost::python::object& currValue = dictionary[currKey];
      setParamPython(keyProxy, currValue);
    } else {
      B2ERROR("Setting the module parameters from a python dictionary: invalid key in dictionary!");
    }
  }
 
  logSystem.updateModule(nullptr);
}

setPropertyFlags()

void setPropertyFlags ( unsigned int  propertyFlags )

inherited

Sets the flags for the module properties.

Parameters

propertyFlags

bitwise OR of EModulePropFlags

Definition at line 208 of file Module.cc.

{
  m_propertyFlags = propertyFlags;
}

setReturnValue() [1/2]

void setReturnValue ( bool  value )

protectedinherited

Sets the return value for this module as bool.

The bool value is saved as an integer with the convention 1 meaning true and 0 meaning false. The value can be used in the steering file to divide the analysis chain into several paths.

Parameters

value

The value of the return value.

Definition at line 227 of file Module.cc.

{
  m_hasReturnValue = true;
  m_returnValue = value;
}

setReturnValue() [2/2]

void setReturnValue ( int  value )

protectedinherited

Sets the return value for this module as integer.

The value can be used in the steering file to divide the analysis chain into several paths.

Parameters

value

The value of the return value.

Definition at line 220 of file Module.cc.

{
  m_hasReturnValue = true;
  m_returnValue = value;
}

setType()

void setType ( const std::string &  type )

protectedinherited

Set the module type.

Only for use by internal modules (which don't use the normal REG_MODULE mechanism).

Definition at line 48 of file Module.cc.

{
  if (!m_type.empty())
    B2FATAL("Trying to change module type from " << m_type << " is not allowed, the value is assumed to be fixed.");
  m_type = type;
}

startRun()

virtual void startRun ( )

inlineprotectedvirtualinherited

Replacement for beginRun(). Do anything you would normally do in beginRun here.

Reimplemented in TOPValidationCollectorModule, CaTestModule, CDCCrudeT0CollectorModule, eclAutocovarianceCalibrationC1CollectorModule, eclAutocovarianceCalibrationC3CollectorModule, eclAutocovarianceCalibrationC4CollectorModule, eclWaveformTemplateCalibrationC1CollectorModule, eclWaveformTemplateCalibrationC2CollectorModule, KLMStripEfficiencyCollectorModule, PXDClusterChargeCollectorModule, PXDPerformanceCollectorModule, PXDPerformanceVariablesCollectorModule, SVDCrossTalkCalibrationsCollectorModule, SVDOccupancyCalibrationsCollectorModule, SVDClusterTimeShifterCollectorModule, SVDTimeCalibrationCollectorModule, and SVDTimeValidationCollectorModule.

Definition at line 75 of file CalibrationCollectorModule.h.

{}

storeTrajectory()

void storeTrajectory

(

gbl::GblTrajectory &

trajectory

)

Write down a GBL trajectory (to TTree or binary file)

Definition at line 1017 of file MillepedeCollectorModule.cc.

{
  if (m_useGblTree) {
    if (trajectory.isValid())
      m_currentGblData = trajectory.getData();
    else
      m_currentGblData.clear();
 
    if (!m_currentGblData.empty())
      getObjectPtr<TTree>("GblDataTree")->Fill();
  } else {
    getObjectPtr<MilleData>("mille")->fill(trajectory);
  }
}

terminate()

void terminate ( void  )

finalvirtualinherited

Write the final objects to the file.

Reimplemented from HistoModule.

Definition at line 155 of file CalibrationCollectorModule.cc.

{
  finish();
  // actually this should be done by the write() called by HistoManager....
 
  // Haven't written objects yet if collecting with granularity == all
  // Write them now that everything is done.
//  if (m_granularity == "all") {
//    m_manager.writeCurrentObjects(m_expRun);
//    m_manager.clearCurrentObjects(m_expRun);
//  }
  m_manager.deleteHeldObjects();
}

updateMassWidthIfSet()

void updateMassWidthIfSet ( std::string  listName, double &  mass, double &  width  )

private

Update mass and width of the particle (mother in list) with user custom-defined values.

Definition at line 1587 of file MillepedeCollectorModule.cc.

{
  if (m_customMassConfig.find(listName) != m_customMassConfig.end()) {
    auto massWidth = m_customMassConfig.at(listName);
    mass = std::get<0>(massWidth);
    width = std::get<1>(massWidth);
  }
}

Member Data Documentation

m_absFilePaths

bool m_absFilePaths

private

Use absolute path to locate binary files in MilleData.

Definition at line 148 of file MillepedeCollectorModule.h.

m_calibrateKinematics

bool m_calibrateKinematics = true

private

Add derivatives for beam spot kinematics calibration for primary vertices.

Definition at line 142 of file MillepedeCollectorModule.h.

m_calibrateVertex

bool m_calibrateVertex

private

Add derivatives for beam spot vertex calibration for primary vertices.

Definition at line 140 of file MillepedeCollectorModule.h.

m_components

std::vector<std::string> m_components {}

private

Whether to use VXD alignment hierarchy.

Definition at line 150 of file MillepedeCollectorModule.h.

m_conditions

std::vector<ModuleCondition> m_conditions

privateinherited

Module condition, only non-null if set.

Definition at line 520 of file Module.h.

m_currentGblData

std::vector<gbl::GblData> m_currentGblData {}

private

Current vector of GBL data from trajectory to be stored in a tree.

Definition at line 187 of file MillepedeCollectorModule.h.

m_customMassConfig

std::map<std::string, std::tuple<double, double> > m_customMassConfig

private

Map of list_name -> (mass, width) for custom mass and width setting.

Definition at line 184 of file MillepedeCollectorModule.h.

m_description

std::string m_description

privateinherited

The description of the module.

Definition at line 510 of file Module.h.

m_dir

TDirectory* m_dir

protectedinherited

The top TDirectory that collector objects for this collector will be stored beneath.

Definition at line 84 of file CalibrationCollectorModule.h.

m_doublePrecision

bool m_doublePrecision

private

Use double (instead of single/float) precision for binary files.

Definition at line 138 of file MillepedeCollectorModule.h.

m_emd

StoreObjPtr<EventMetaData> m_emd

protectedinherited

Current EventMetaData.

Definition at line 96 of file CalibrationCollectorModule.h.

m_enablePXDHierarchy

bool m_enablePXDHierarchy

private

enable PXD hierarchy

Definition at line 169 of file MillepedeCollectorModule.h.

m_enableSVDHierarchy

bool m_enableSVDHierarchy

private

enable SVD hierarchy

Definition at line 171 of file MillepedeCollectorModule.h.

m_enableWireByWireAlignment

bool m_enableWireByWireAlignment

private

Enable global derivatives for wire-by-wire alignment.

Definition at line 173 of file MillepedeCollectorModule.h.

m_enableWireSagging

bool m_enableWireSagging

private

Enable global derivatives for wire sagging.

Definition at line 175 of file MillepedeCollectorModule.h.

m_eventNumbers

std::vector<std::tuple<int, int, int> > m_eventNumbers {}

private

List of event meta data entries at which payloads can change for timedep calibration.

Definition at line 178 of file MillepedeCollectorModule.h.

m_eventsCollectedInRun

int* m_eventsCollectedInRun

privateinherited

Will point at correct value in m_expRunEvents.

Definition at line 117 of file CalibrationCollectorModule.h.

m_eventT0

StoreObjPtr<EventT0> m_eventT0

private

Optional input for EventT0.

Definition at line 190 of file MillepedeCollectorModule.h.

m_evtMetaData

StoreObjPtr<EventMetaData> m_evtMetaData

private

Required object pointer to EventMetaData.

Definition at line 193 of file MillepedeCollectorModule.h.

m_expRun

Calibration::ExpRun m_expRun

protectedinherited

Current ExpRun for object retrieval (becomes -1,-1 for granularity=all)

Definition at line 93 of file CalibrationCollectorModule.h.

m_expRunEvents

std::map<Calibration::ExpRun, int> m_expRunEvents

privateinherited

How many events processed for each ExpRun so far, stops counting up once max is hit Only used/incremented if m_maxEventsPerRun > -1.

Definition at line 115 of file CalibrationCollectorModule.h.

m_externalIterations

int m_externalIterations

private

Number of external iterations of GBL fitter.

Definition at line 152 of file MillepedeCollectorModule.h.

m_fitTrackT0

bool m_fitTrackT0

private

Add local parameter for track T0 fit in GBL (local derivative)

Definition at line 158 of file MillepedeCollectorModule.h.

m_granularity

std::string m_granularity

privateinherited

Granularity of data collection = run|all(= no granularity, exp,run=-1,-1)

Definition at line 101 of file CalibrationCollectorModule.h.

m_hasReturnValue

bool m_hasReturnValue

privateinherited

True, if the return value is set.

Definition at line 517 of file Module.h.

m_hierarchyType

int m_hierarchyType

private

Type of alignment hierarchy (for VXD only for now): 0 = None, 1 = Flat (only constraints, no new global parameters/derivatives), 2 = Half-Shells + sensors (no ladders), 3 = Full.

Definition at line 167 of file MillepedeCollectorModule.h.

m_internalIterations

std::string m_internalIterations

private

String defining internal GBL iterations for outlier down-weighting.

Definition at line 154 of file MillepedeCollectorModule.h.

m_logConfig

LogConfig m_logConfig

privateinherited

The log system configuration of the module.

Definition at line 513 of file Module.h.

m_manager

CalibObjManager m_manager

protectedinherited

Controls the creation, collection and access to calibration objects.

Definition at line 87 of file CalibrationCollectorModule.h.

m_maxEventsPerRun

int m_maxEventsPerRun

privateinherited

Maximum number of events to be collected at the start of each run (-1 = no maximum)

Definition at line 103 of file CalibrationCollectorModule.h.

m_minCDCHitWeight

double m_minCDCHitWeight

private

Minimum CDC hit weight.

Definition at line 162 of file MillepedeCollectorModule.h.

m_minPValue

double m_minPValue

private

Minimum p.value for output.

Definition at line 144 of file MillepedeCollectorModule.h.

m_minUsedCDCHitFraction

double m_minUsedCDCHitFraction

private

Minimum CDC used hit fraction.

Definition at line 164 of file MillepedeCollectorModule.h.

m_moduleParamList

ModuleParamList m_moduleParamList

privateinherited

List storing and managing all parameter of the module.

Definition at line 515 of file Module.h.

m_name

std::string m_name

privateinherited

The name of the module, saved as a string (user-modifiable)

Definition at line 507 of file Module.h.

m_package

std::string m_package

privateinherited

Package this module is found in (may be empty).

Definition at line 509 of file Module.h.

m_particles

std::vector<std::string> m_particles

private

Names of particle list with single particles.

Definition at line 121 of file MillepedeCollectorModule.h.

m_preScale

float m_preScale

privateinherited

Prescale module parameter, this fraction of events will have collect() run on them [0.0 -> 1.0].

Definition at line 105 of file CalibrationCollectorModule.h.

m_primaryMassTwoBodyDecays

std::vector<std::string> m_primaryMassTwoBodyDecays

private

Name of particle list with mothers of daughters to be used with vertex + IP profile + mass constraint in calibration.

Definition at line 131 of file MillepedeCollectorModule.h.

m_primaryMassVertexTwoBodyDecays

std::vector<std::string> m_primaryMassVertexTwoBodyDecays

private

Name of particle list with mothers of daughters to be used with vertex + IP profile + mass constraint in calibration.

Definition at line 133 of file MillepedeCollectorModule.h.

m_primaryTwoBodyDecays

std::vector<std::string> m_primaryTwoBodyDecays

private

Name of particle list with mothers of daughters to be used with vertex + IP profile (+ optional calibration) + IP kinematics (+ optional calibration) constraint in calibration.

Definition at line 129 of file MillepedeCollectorModule.h.

m_primaryVertices

std::vector<std::string> m_primaryVertices

private

Name of particle list with mothers of daughters to be used with vertex + IP profile (+ optional calibration) constraint in calibration.

Definition at line 125 of file MillepedeCollectorModule.h.

m_propertyFlags

unsigned int m_propertyFlags

privateinherited

The properties of the module as bitwise or (with |) of EModulePropFlags.

Definition at line 511 of file Module.h.

m_recalcJacobians

int m_recalcJacobians

private

Up to which external iteration propagation Jacobians should be re-calculated.

Definition at line 156 of file MillepedeCollectorModule.h.

m_returnValue

int m_returnValue

privateinherited

The return value.

Definition at line 518 of file Module.h.

m_runCollectOnRun

bool m_runCollectOnRun = true

privateinherited

Whether or not we will run the collect() at all this run, basically skips the event() function if false.

Definition at line 111 of file CalibrationCollectorModule.h.

m_runRange

RunRange* m_runRange

protectedinherited

Overall list of runs processed.

Definition at line 90 of file CalibrationCollectorModule.h.

m_stableParticleWidth

double m_stableParticleWidth

private

Width (in GeV/c/c) to use for invariant mass constraint for 'stable' particles (like K short).

Temporary until proper solution is found

Definition at line 136 of file MillepedeCollectorModule.h.

m_timedepConfig

std::vector< std::tuple< std::vector< int >, std::vector< std::tuple< int, int, int > > > > m_timedepConfig

private

Config for time dependence: list( tuple( list( param1, param2, ... ), list( (ev, run, exp), ... )), ...

Definition at line 181 of file MillepedeCollectorModule.h.

m_tracks

std::vector<std::string> m_tracks

private

Names of arrays with single RecoTracks fitted by GBL.

Definition at line 119 of file MillepedeCollectorModule.h.

m_twoBodyDecays

std::vector<std::string> m_twoBodyDecays

private

Name of particle list with mothers of daughters to be used with vertex + mass constraint in calibration.

Definition at line 127 of file MillepedeCollectorModule.h.

m_type

std::string m_type

privateinherited

The type of the module, saved as a string.

Definition at line 508 of file Module.h.

m_updateCDCWeights

bool m_updateCDCWeights

private

Update L/R weights from previous DAF fit result?

Definition at line 160 of file MillepedeCollectorModule.h.

m_useGblTree

bool m_useGblTree

private

Whether to use TTree to accumulate GBL data instead of binary files.

Definition at line 146 of file MillepedeCollectorModule.h.

m_vertices

std::vector<std::string> m_vertices

private

Name of particle list with mothers of daughters to be used with vertex constraint in calibration.

Definition at line 123 of file MillepedeCollectorModule.h.

---

The documentation for this class was generated from the following files:

• alignment/modules/MillepedeCollector/include/MillepedeCollectorModule.h
• alignment/modules/MillepedeCollector/src/MillepedeCollectorModule.cc