TRGCDCT3DConverterModule.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.                  *
 **************************************************************************/
//---------------------------------------------------------------
// $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>
#include "boost/multi_array.hpp"
 
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(boost::multi_array<double, 4>& tsfInfo);
 
      void filterTSData(boost::multi_array<double, 4>& tsfFirmwareInfo,
                        boost::multi_array<double, 3>& tsfInfo);
 
      void addTSDatastore(boost::multi_array<double, 3>& tsfInfo, int isSt = 1);
 
      void store2DFirmwareData(boost::multi_array<double, 3>& t2DFirmwareInfo,
                               boost::multi_array<double, 4>& axTsfFirmwareInfo);
 
      //void filter2DData(boost::multi_array<double, 3>  & t2DFirmwareInfo, boost::multi_array<double, 4> & t2DTsfFirmwareInfo, boost::multi_array<double, 2> & t2DInfo, boost::multi_array<double, 3>  & t2DTsfInfo, std::vector<int> & t2DClks, std::vector<int> & t2DId);
 
      void filter2DData(boost::multi_array<double, 3>& t2DFirmwareInfo,
                        boost::multi_array<double, 4>& t2DTsfFirmwareInfo, boost::multi_array<double, 2>& t2DInfo,
                        boost::multi_array<double, 3>& t2DTsfInfo);
 
      void add2DDatastore(boost::multi_array<double, 2>& t2DInfo, boost::multi_array<double, 3>& t2DTsfInfo);
 
      void store3DFirmwareData(boost::multi_array<double, 3>& t3DFirmwareInfo);
 
      void store3DFastSimData(boost::multi_array<double, 3>& t3DFirmwareInfo);
 
      void store3DFirmSimData(boost::multi_array<double, 3>& t3DFirmwareInfo);
 
      void filter3DData(boost::multi_array<double, 3>& t3DFirmwareInfo, boost::multi_array<double, 2>& t3DInfo);
 
      void add3DDatastore(boost::multi_array<double, 2>& 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