TOPAlignerModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Marko Staric                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <top/reconstruction/TOPalign.h>
 
#include <framework/datastore/StoreArray.h>
#include <mdst/dataobjects/Track.h>
#include <tracking/dataobjects/ExtHit.h>
#include <top/dataobjects/TOPDigit.h>
#include <top/dataobjects/TOPRecBunch.h>
#include <framework/gearbox/Const.h>
 
#include <string>
 
#include "TFile.h"
#include "TTree.h"
 
namespace Belle2 {
  class TOPAlignerModule : public Module {
 
  public:
 
    TOPAlignerModule();
 
    virtual ~TOPAlignerModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  private:
 
    bool selectTrack(const TOP::TOPtrack& trk);
 
    // module paramenets
    double m_minBkgPerBar; 
    double m_scaleN0; 
    int m_targetMid; 
    int m_maxFails; 
    std::string m_sample; 
    double m_minMomentum; 
    double m_deltaEcms; 
    double m_dr; 
    double m_dz; 
    double m_minZ; 
    double m_maxZ; 
    std::string m_outFileName; 
    double m_stepPosition; 
    double m_stepAngle; 
    double m_stepTime; 
    double m_stepRefind; 
    std::vector<int> m_gridSize; 
    std::vector<double> m_parInit; 
    std::vector<std::string> m_parFixed; 
    // alignment procedure
    TOP::TOPalign m_align;     
    Const::ChargedStable m_chargedStable = Const::muon; 
    int m_countFails = 0;      
    // output
    TFile* m_file = 0;                 
    TTree* m_alignTree = 0;            
    // tree variables
    int m_iter = 0;  
    int m_ntrk = 0;                      
    int m_errorCode = 0;                 
    std::vector<float> m_vAlignPars;     
    std::vector<float> m_vAlignParsErr;  
    bool m_valid = false;  
    float m_x = 0; 
    float m_y = 0; 
    float m_z = 0; 
    float m_p = 0; 
    float m_theta = 0; 
    float m_phi = 0; 
    float m_pocaR = 0; 
    float m_pocaZ = 0; 
    float m_pocaX = 0; 
    float m_pocaY = 0; 
    float m_cmsE = 0; 
    int m_charge = 0; 
    int m_PDG = 0; 
    // datastore objects
    StoreArray<TOPDigit> m_digits; 
    StoreArray<Track> m_tracks;    
    StoreArray<ExtHit> m_extHits;  
    StoreObjPtr<TOPRecBunch> m_recBunch; 
  };
 
} // Belle2 namespace