MixerAlg.h

Go to the documentation of this file.

#ifndef MixerAlg_h
#define MixerAlg_h
 
#include "GaudiKernel/Algorithm.h"
#include "GaudiKernel/MsgStream.h"
#include "GaudiKernel/NTuple.h"
#include "GaudiKernel/SmartDataPtr.h"
 
#include "BesRndmGenSvc/IBesRndmGenSvc.h"
#include "BesTimerSvc/IBesTimerSvc.h"
#include "DataInfoSvc/IDataInfoSvc.h"
#include "EmcRawEvent/EmcDigi.h"
#include "MdcRawEvent/MdcDigi.h"
#include "MucRawEvent/MucDigi.h"
#include "RawDataCnv/EventManagement/EmcBuilder.h"
#include "RawDataCnv/EventManagement/MdcBuilder.h"
#include "RawDataCnv/EventManagement/MucBuilder.h"
#include "RawDataCnv/EventManagement/RAWEVENT.h"
#include "RawDataCnv/EventManagement/TofBuilder.h"
#include "RawDataCnv/Util/EmcConverter.h"
#include "RawDataCnv/Util/MdcConverter.h"
#include "RawDataCnv/Util/MucConverter.h"
#include "RawDataCnv/Util/TofConverter.h"
#include "RawEvent/RawDataUtil.h"
#include "RawFile/RawFileReader.h"
#include "RealizationSvc/IRealizationSvc.h"
#include "TofRawEvent/TofDigi.h"
//
// An algorithm to combine MC signal and background from raw data file at digi level
//
// A.Zhemchugov (zhemchugov@jinr.ru), 09/08/2006
// A.Zhemchugov: modified to new data format, 29.07.08
 
#define MAX_LOOP_TIMES 1000000
 
template <class T1, class T2>
void combineDigits( SmartDataPtr<T1>& mcDigits, T1& bgDigits, int verbosity );
void combineMdcDigits( SmartDataPtr<MdcDigiCol>& mcDigits, MdcDigiCol& bgDigits,
                       int verbosity );
void combineTofDigits( SmartDataPtr<TofDigiCol>& mcDigits, TofDigiCol& bgDigits,
                       int verbosity );
 
class MixerAlg : public Algorithm {
public:
  MixerAlg( const std::string& name, ISvcLocator* pSvcLocator );
  StatusCode initialize();
  StatusCode execute();
  StatusCode finalize();
  ~MixerAlg() {}
 
protected:
  bool nextEvent( int nskip = 0, int evtbyte = 0, int eventsToEnd = 0 );
 
  void mixDigi( SmartDataPtr<MdcDigiCol>& mdcMcDigits, SmartDataPtr<EmcDigiCol>& emcMcDigits,
                SmartDataPtr<MucDigiCol>& mucMcDigits, SmartDataPtr<TofDigiCol>& tofMcDigits );
 
  void decodeMdc( MdcDigiCol* digiCol );
  void decodeMuc( MucDigiCol* digiCol );
  void decodeEmc( EmcDigiCol* digiCol );
  void decodeTof( TofDigiCol* digiCol );
 
  int         getTiming();
  std::string eventType();
 
  std::string prepareDbQuery();
 
  bool file_sort( std::vector<std::string>& files, std::vector<int>& ranEvtNums );
 
private:
  MsgStream* log;
 
  RAWEVENT*      m_raw_event;
  RawFileReader* m_fr;
 
  // IDatabaseSvc* m_db;
 
  // Converters to create Digi objects
  MdcConverter* m_mdcCnv;
  MucConverter* m_mucCnv;
  EmcConverter* m_emcCnv;
  TofConverter* m_tofCnv;
 
  // Algorithm properties to be set in data card
  std::string              m_run;
  std::vector<std::string> m_runs;
  std::vector<std::string> m_dates;
 
  // Flags to choose subdetectors for which background should be mixed in
  bool b_mdc;
  bool b_emc;
  bool b_muc;
  bool b_tof;
 
  // Number of background events to be mixed in one signal event
  // bool b_loop;
  bool m_skip;
  bool m_dbUserRequest;
  int  m_NSkip;
  int  m_skipCount;
  int  nskipped;
  int  m_mixingMethod;
 
  int m_nevent;
  int m_runNo;
  int m_totRanEvtNum;
  int m_totalEvent;
  int m_totEvtNumInCurFile;
  int m_nEventsToEnd;
 
  std::vector<int> m_ranTrgEvents;
  std::vector<int> m_numSets; // No. of sets of each bg file, for example:
                              // run_**_RandomTrg_file001_SF0-1.raw, 001 is set No.
  std::vector<int> m_vRanEvtNumInSubSet;
  std::vector<int> m_vStreamNumInSubSet;
  std::vector<int> m_vStepLength;
  std::vector<int> m_ranStepLenInCurrentFile;
 
  std::map<int, std::vector<int>> map_stepLength;
 
  int  m_maxLoop;
  bool m_ifSmearT0;
 
  int m_readBGMethod;
 
  bool m_usingFilter;
 
  // Current bg file
  std::string currentBGFile;
 
  // Current MC file
  std::string currentMCFile;
 
  // Input file name of background data (raw data format)
  std::vector<std::string> m_bgfiles;
  std::vector<std::string> m_bgfilesIndex;
  // std::vector<std::string>::iterator m_bgfile;
 
  // Replace  /besfs/offline/data/randomtrg in the file name from the database to the user
  // defined value
  std::string m_pattern;
  // Use the user defined value of directory which contains random trigger files
  std::string m_newdatadir;
 
  // Reference to RealizationSvc
  IRealizationSvc* m_RealizationSvc;
  IDataInfoSvc*    m_jobInfoSvc;
 
  NTuple::Tuple*      m_tuple1;
  NTuple::Item<float> m_time1;
  NTuple::Item<float> m_time2;
  NTuple::Item<float> m_time3;
  NTuple::Item<float> m_time4;
  NTuple::Item<float> m_time5;
 
  NTuple::Tuple*     m_tuple2;
  NTuple::Tuple*     m_tuple3;
  NTuple::Item<long> m_tdc;
 
  IBesTimerSvc* m_timersvc;
  BesTimer*     m_timer;
  BesTimer*     m_timer1;
  bool          m_ifOutPut;
  bool          m_dump;
 
  // caogf for random seed
  IBesRndmGenSvc* p_BesRndmGenSvc;
 
  // configuration file for raw data revise
  std::string m_rawReviseConfig;
};
 
#endif