PndMdtPointsToWaveform.h

Go to the documentation of this file.

//----------------------------------------------------------------------
// File and Version Information:
//      $Id: Exp $
//
// Description:
//      Class PndMdtPointsToWaveform. Module to take the point list for the
//      mdt induced current.
//
//
// Author List:
//  Jifeng Hu, hu@to.infn.it, Torino University
//----------------------------------------------------------------------
//#pragma once
#ifndef PndMdtPointsToWaveform_H
#define PndMdtPointsToWaveform_H

#include <PndPersistencyTask.h>
#include "PndMdtWaveform.h"
#include "TVector3.h"
#include <TFile.h>
#include <TTree.h>

class TClonesArray;
// class PndMdtDigiPar;
// class PndMdtGeoPar;
class PndMdtWaveformWriteoutBuffer;
class PndMdtParamDigi;
class PndMdtIGeometry;

class PndMdtPointsToWaveform : public PndPersistencyTask {

 public:
  // Constructors

  PndMdtPointsToWaveform(Int_t verbose = 0, Bool_t storewaves = kTRUE);

  // Destructor

  virtual ~PndMdtPointsToWaveform();

  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  void SetStorageOfData(Bool_t val); // Method to specify whether waveforms are stored or not.

  void RunTimeBased() { fTimeOrderedWaveform = kTRUE; }

  void FinishTask();

 private:
  TClonesArray *fMcTrackArray;

  TClonesArray *fPointArray;

  /* output array for general sim */
  TClonesArray *fWaveformArray;

  /* output array for time-based sim */
  Bool_t fTimeOrderedWaveform;
  PndMdtWaveformWriteoutBuffer *fDataBuffer;

  PndMdtParamDigi *fParamDigiModel;
  PndMdtIGeometry *fGeoIF;

  //    PndMdtDigiPar*    fDigiPar;      /** Digitisation parameter container **/
  //    PndMdtGeoPar*     fGeoPar;       /** Geometry parameter container **/
  virtual void SetParContainers();
  virtual void exec_t();
  virtual void exec_e();
  Int_t PdgToIndex(Int_t pdg);

  Int_t fVerbose;
  // counters for task
  Int_t HowManyPoint;
  Int_t nWaveformProduced;

  PndMdtPointsToWaveform(const PndMdtPointsToWaveform &L);
  PndMdtPointsToWaveform &operator=(const PndMdtPointsToWaveform &);

  Bool_t Digitize(PndMdtWaveform *theWf, Double_t &time, Double_t &amp, Bool_t isWire);

  struct key {
    key() {}
    key(Int_t _tid, Int_t _detid) : TrkID(_tid), DetID(_detid) {}
    bool operator<(const key &rhs) const
    {
      if (TrkID < rhs.TrkID)
        return true;
      if (TrkID == rhs.TrkID)
        return DetID < rhs.DetID;
      return false;
    }
    bool operator==(const key &rhs) const { return TrkID == rhs.TrkID && DetID == rhs.DetID; }
    Int_t TrkID; // track id
    Int_t DetID; // detector id
  };

  TFile *fFile;
  TTree *tTree;
  Double_t fWirpT;
  Double_t fStripT;
  Double_t fEvtT;
  Double_t fLength;
  Double_t fDis;
  Int_t fMod;
  Int_t fPid;

  ClassDef(PndMdtPointsToWaveform, 2);
};

#endif