CDCTrigger3DFitterModule.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.                  *
 **************************************************************************/
#ifndef CDCTRIGGER3DFITTERModule_H
#define CDCTRIGGER3DFITTERModule_H
 
#include "framework/core/Module.h"
#include <string>
 
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <trg/cdc/dataobjects/CDCTriggerSegmentHit.h>
#include <trg/cdc/dataobjects/CDCTriggerTrack.h>
#include <framework/dataobjects/BinnedEventT0.h>
 
namespace Belle2 {
  class TRGCDCJSignal;
  class TRGCDCJLUT;
  class TRGCDCJSignalData;
 
  class CDCTrigger3DFitterModule : public Module {
 
  public:
 
    CDCTrigger3DFitterModule();
 
    virtual ~CDCTrigger3DFitterModule();
 
    virtual void initialize() override;
 
    virtual void event() override;
 
  protected:
    void finder(int charge, double rho, double phi,
                std::vector<int>& bestTSIndex, std::vector<double>& bestTSPhi);
    void fitter(const std::vector<int>& bestTSIndex, std::vector<double>& bestTSPhi,
                int charge, double rho, double phi,
                double& z0, double& cot, double& chi2);
 
    std::string m_hitCollectionName;
    std::string m_EventTimeName;
    std::string m_inputCollectionName;
    std::string m_outputCollectionName;
    unsigned m_fitterMode;
    unsigned m_minHits;
    bool m_xtSimple;
    bool m_isVerbose;
 
  private:
    std::vector<double> nWires = {};
    std::vector<double> rr = {};
    std::vector<double> zToStraw = {};
    std::vector<int> nShift = {};
    std::vector<double> angleSt = {};
    std::vector<std::vector<double>> xtTables = {};
 
    std::map<std::string, std::vector<double> > m_stGeometry;
    std::vector<std::vector<double> > m_stXts;
 
    Belle2::TRGCDCJSignalData* m_commonData = nullptr;
    std::map<std::string, Belle2::TRGCDCJSignal> m_mSignalStorage;
    std::map<std::string, Belle2::TRGCDCJLUT*> m_mLutStorage;
    std::map<std::string, double> m_mConstD;
    std::map<std::string, std::vector<double> > m_mConstV;
 
    StoreArray<CDCTriggerTrack> m_tracks2D;
    StoreArray<CDCTriggerTrack> m_tracks3D;
    StoreArray<CDCTriggerSegmentHit> m_hits;
    StoreObjPtr<BinnedEventT0> m_eventTime;
  };
 
} // namespace Belle2
 
#endif // CDCTrigger3DFitterModule_H