TRGCDCT3DConverterModule.h

//---------------------------------------------------------------
// $Id$
//---------------------------------------------------------------
// Filename : TRGCDCT3DConverterModule.h
// Section  :
// Owner    :
// Email    :
//---------------------------------------------------------------
// Description : TRGCDCT3DConverter Module
//---------------------------------------------------------------
// $Log$
//---------------------------------------------------------------
 
#ifndef TRGCDCT3DCONVERTER_H
#define TRGCDCT3DCONVERTER_H
 
#include <string>
#include <vector>
 
#include <framework/datastore/StoreArray.h>
#include <framework/core/Module.h>
#include <framework/datastore/StoreObjPtr.h>
#include "trg/cdc/dataobjects/TRGCDCT3DUnpackerStore.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;
 
  namespace TRGCDCT3DCONVERTERSPACE {
 
    class TRGCDCT3DConverterModule : public Module {
 
    public:
 
      TRGCDCT3DConverterModule();
 
      virtual ~TRGCDCT3DConverterModule();
 
      virtual void initialize() override;
 
      virtual void event() override;
 
      virtual void terminate() override;
 
      virtual void beginRun() override;
 
      virtual void endRun() override;
 
      std::string version() const;
 
      void storeTSFirmwareData(std::vector<std::vector<std::vector<std::vector<double> > > >& tsfInfo);
 
      void filterTSData(std::vector<std::vector<std::vector<std::vector<double> > > >& tsfFirmwareInfo,
                        std::vector<std::vector<std::vector<double> > >& tsfInfo);
 
      void addTSDatastore(std::vector<std::vector<std::vector<double> > >& tsfInfo, int isSt = 1);
 
      void store2DFirmwareData(std::vector<std::vector<std::vector<double> > >& t2DFirmwareInfo,
                               std::vector<std::vector<std::vector<std::vector<double> > > >& axTsfFirmwareInfo);
 
      //void filter2DData(std::vector<std::vector<std::vector<double> > > & t2DFirmwareInfo, std::vector<std::vector<std::vector<std::vector<double> > > > & t2DTsfFirmwareInfo, std::vector<std::vector<double> > & t2DInfo, std::vector<std::vector<std::vector<double> > > & t2DTsfInfo, std::vector<int> & t2DClks, std::vector<int> & t2DId);
 
      void filter2DData(std::vector<std::vector<std::vector<double> > >& t2DFirmwareInfo,
                        std::vector<std::vector<std::vector<std::vector<double> > > >& t2DTsfFirmwareInfo, std::vector<std::vector<double> >& t2DInfo,
                        std::vector<std::vector<std::vector<double> > >& t2DTsfInfo);
 
      void add2DDatastore(std::vector<std::vector<double> >& t2DInfo, std::vector<std::vector<std::vector<double> > >& t2DTsfInfo);
 
      void store3DFirmwareData(std::vector<std::vector<std::vector<double> > >& t3DFirmwareInfo);
 
      void store3DFastSimData(std::vector<std::vector<std::vector<double> > >& t3DFirmwareInfo);
 
      void store3DFirmSimData(std::vector<std::vector<std::vector<double> > >& t3DFirmwareInfo);
 
      void filter3DData(std::vector<std::vector<std::vector<double> > >& t3DFirmwareInfo, std::vector<std::vector<double> >& t3DInfo);
 
      void add3DDatastore(std::vector<std::vector<double> >& t3DInfo, bool doConvert = 1);
 
      void debug3DFirmware();
 
      int toTSID(int iSL, int iWire);
 
      int toSigned(int value, int nBits);
 
      int t2DRhoTot3DRho(int value, bool isSigned = 1);
 
      int t2DPhiTot3DPhi(int phi, int rho);
 
    protected:
      std::string m_firmwareResultCollectionName;
      std::string m_hitCollectionName;
      bool m_addTSToDatastore;
      std::string m_inputCollectionName;
      bool m_add2DFinderToDatastore;
      std::string m_outputCollectionName;
      bool m_add3DToDatastore;
      unsigned m_fit3DWithTSIM;
      std::string m_EventTimeName;
      bool m_addEventTimeToDatastore;
      unsigned m_isVerbose;
 
 
 
    private:
      StoreArray<TRGCDCT3DUnpackerStore> m_firmwareResults;
      StoreArray<CDCTriggerSegmentHit> m_hits;
      StoreArray<CDCTriggerTrack> m_tracks2D;
      StoreArray<CDCTriggerTrack> m_tracks3D;
      StoreObjPtr<BinnedEventT0> m_eventTime;
 
      Belle2::TRGCDCJSignalData* m_commonData;
      std::map<std::string, TRGCDCJSignal> m_mSignalStorage;
      std::map<std::string, TRGCDCJLUT*> m_mLutStorage;
 
    };
  }
} // end namespace Belle2
#endif // TRGCDCT3DUNPACKER_H