ECLMatchingPerformanceExpertModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
/* ECL headers. */
#include <ecl/dataobjects/ECLCalDigit.h>
#include <ecl/geometry/ECLNeighbours.h>
 
/* Basf2 headers. */
#include <framework/core/Module.h>
#include <framework/dataobjects/EventMetaData.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <mdst/dataobjects/ECLCluster.h>
#include <mdst/dataobjects/Track.h>
#include <mdst/dataobjects/TrackFitResult.h>
#include <tracking/dataobjects/ExtHit.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <tracking/modules/standardTrackingPerformance/ParticleProperties.h>
 
// forward declarations
class TTree;
class TFile;
 
namespace Belle2 {
  template< class T >
  class StoreArray;
 
 
  class ECLMatchingPerformanceExpertModule : public Module {
  public:
    ECLMatchingPerformanceExpertModule();
 
    void initialize() override;
 
    void event() override;
 
    void terminate() override;
 
  private:
    std::string m_outputFileName; 
    double m_minCalDigitEnergy; 
    double m_innerDistanceEnergy; 
    StoreArray<ECLCalDigit> m_eclCalDigits; 
    StoreArray<ECLCluster> m_eclClusters; 
    StoreArray<ExtHit> m_extHits; 
    StoreArray<RecoTrack> m_recoTracks; 
    StoreArray<Track> m_tracks; 
    StoreArray<TrackFitResult> m_trackFitResults; 
    StoreObjPtr<EventMetaData> m_EventMetaData;
 
    ECL::ECLNeighbours* m_eclNeighbours1x1{nullptr}; 
    ECL::ECLNeighbours* m_eclNeighbours3x3{nullptr}; 
    ECL::ECLNeighbours* m_eclNeighbours5x5{nullptr}; 
    TFile* m_outputFile{nullptr}; 
    TTree* m_dataTree{nullptr}; 
    ParticleProperties m_trackProperties;
 
    int m_iExperiment = 0;
 
    int m_iRun = 0;
 
    int m_iEvent = 0;
 
    int m_trackMultiplicity = 0;
 
    double m_pValue = 0.0;
 
    int m_charge = 0;
 
    double m_d0 = 0.0;
 
    double m_z0 = 0.0;
 
    int m_matchedToECLCluster = 0;
 
    int m_hypothesisOfMatchedECLCluster = 0;
 
    double m_distance = 0.0;
 
    double m_trackLength = 0.0;
 
    double m_deposited_energy = 0.0;
 
    double m_innerdistance = 0.0;
 
    int m_matchedTo1x1Neighbours = 0;
 
    int m_matchedTo3x3Neighbours = 0;
 
    int m_matchedTo5x5Neighbours = 0;
 
    int m_matchedToDecreasedPhi = 0;
 
    int m_matchedToIncreasedPhi = 0;
 
    int m_matchedToDecreasedTheta = 0;
 
    int m_matchedToIncreasedTheta = 0;
 
    int m_matchedToDecreasedPhiDecreasedTheta = 0;
 
    int m_matchedToIncreasedPhiDecreasedTheta = 0;
 
    int m_matchedToDecreasedPhiIncreasedTheta = 0;
 
    int m_matchedToIncreasedPhiIncreasedTheta = 0;
 
    int m_enter = 0;
 
    int m_exit = 0;
 
    int m_enteringcellid = 0;
 
    double m_enteringcelltheta = 0.0;
 
    int m_ndf = 0;
 
    void setVariablesToDefaultValue();
 
    void setupTree();
 
    void writeData();
 
    void addVariableToTree(const std::string& varName, double& varReference);
 
    void addVariableToTree(const std::string& varName, int& varReference);
 
    void findECLCalDigitMatchInNeighbouringCell(ECL::ECLNeighbours* eclneighbours, int& matchedToNeighbours, const int& cell);
 
    void findECLCalDigitMatch(const int& cell, int& matched);
  };
 
 
} // end of namespace