ERBeamDetTrackFinder.h

/********************************************************************************
 *              Copyright (C) Joint Institute for Nuclear Research              *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/

#ifndef ERBeamDetTrackFinder_H
#define ERBeamDetTrackFinder_H

#include "TClonesArray.h"
#include "TRandom3.h"

#include "ERTask.h"
#include "ERBeamDetTrack.h"
#include "ERBeamDetSetup.h"

class ERBeamDetTrackFinder : public ERTask {
public:
  ERBeamDetTrackFinder();
  ERBeamDetTrackFinder(Int_t verbose);
  ~ERBeamDetTrackFinder() = default;
  /* Modifiers */
  void SetInteractionPosition(const TString& volume_name, float depth_ratio = 0.5) {
    interaction_volume_name_ = volume_name; depth_ratio_ = depth_ratio;
  }
  void SetTargetVolume(const TString& volume_name) { interaction_volume_name_ = volume_name; }
public:
  virtual InitStatus Init();
  virtual void Exec(Option_t* opt);
  virtual void Reset();
protected:
  //Paramaters
  ERBeamDetSetup* fBeamDetSetup = nullptr; 
  //Input arrays
  TClonesArray* fBeamDetMWPCDigiX1 = nullptr; 
  TClonesArray* fBeamDetMWPCDigiX2 = nullptr;
  TClonesArray* fBeamDetMWPCDigiY1 = nullptr;
  TClonesArray* fBeamDetMWPCDigiY2 = nullptr; 
  //Output arrays
  TClonesArray* fBeamDetTrack = nullptr;     
  Int_t fMultipicityMWPCX1 = -1;
  Int_t fMultipicityMWPCY1 = -1;
  Int_t fMultipicityMWPCX2 = -1;
  Int_t fMultipicityMWPCY2 = -1;
  TRandom3* fRand = nullptr;
  TString interaction_volume_name_ = "target";
  float depth_ratio_ = 0.5;
private:
  ERBeamDetTrack* AddTrack(Double_t xt, Double_t yt, Double_t zt, TVector3 v);
  Bool_t   IsCluster(TClonesArray* digiArray);
  Double_t CalcCoordinateAvg(const TString& digi_branch_name, const TClonesArray* digiArray,
                             char coordType);
  ClassDef(ERBeamDetTrackFinder,1)
};
#endif