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;

      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