Event.h

/*
 Get * Event.h
 *
 *  Created on: 04-05-2022
 *      Author: Daniel Wielanek
 *      E-mail: daniel.wielanek@gmail.com
 *      Warsaw University of Technology, Faculty of Physics
 */
 
#ifndef HALEVENT_H_
#define HALEVENT_H_
 
#include "EventInterface.h"
#include "Track.h"
 
#include <TClonesArray.h>
#include <TNamed.h>
#include <TObjArray.h>
#include <TString.h>
 
#include "Std.h"
 
class TDatabasePDG;
namespace Hal {
  class CompressionMap;
  class Package;
  class McEvent;
  class ComplexEvent;
  class SmearedEvent;
 
  class Event : public TNamed {
  private:
    friend class SmearedEvent;
    friend class ComplexEvent;
    friend class Track;
    Int_t fTotalV0s;
    Int_t fTotalXis;
 
  protected:
    TDatabasePDG* fPDG;  
    TClonesArray* fTracks;
    TClonesArray* fV0sHiddenInfo;
    TClonesArray* fXisHiddenInfo;
    TLorentzVector* fVertex;
    Double_t fPhi, fPhiError;
    Int_t fEventId;
    Int_t fTotalTracksNo;
    Int_t fMultiplicity;
    Double_t CalculateCharge(Int_t pdg) const;
    void ResetHiddenInfoCounter() { fTotalV0s = 0; };
    void CopyData(Event* event);
    void CopyCompress(Event* event, const CompressionMap& map);
    void Compress(TClonesArray* array, const CompressionMap& map);
    virtual void ShallowCopyTracks(Event* event);
    virtual void ShallowCopyCompressTracks(Event* event, const CompressionMap& map);
    virtual void ShallowCopyEvent(Event* event);
    [[deprecated("Use CheckBranches({std::initializer_list}) instead.")]] Bool_t CheckBranches(Int_t n...) const;
    Bool_t CheckBranches(std::initializer_list<TString> list) const;
 
    Event(TString track_class, TString v0_class = "Hal::V0Track", TString xi_class = "Hal::XiTrack");
 
  public:
    Event();
    Event(const Event& other);
    void Build(Event* event, const CompressionMap& map);
    void Build(Event* event);
    void Compress(const CompressionMap& map);
    virtual void Update(EventInterface* interface);
    virtual void Clear(Option_t* opt = " ");
    inline void SetEventID(Int_t newID) { fEventId = newID; }
    inline void SetPhi(Double_t phi, Double_t phi_error = 0) {
      fPhi      = phi;
      fPhiError = phi_error;
    };
    virtual void RotateZ(Double_t phi);
    virtual void Print(Option_t* opt = "") const;
    inline Track* GetTrack(Int_t i) const { return (Track*) fTracks->UncheckedAt(i); };
    Track* AddTrack();
    virtual Bool_t ExistInTree() const { return kFALSE; };
    virtual Bool_t IsCompatible(const Event* non_buffered) const;
    inline Int_t GetEventID() const { return fEventId; };
    inline Int_t GetTotalTrackNo() const { return fTotalTracksNo; };
    inline Int_t GetTotalV0No() const { return fTotalV0s; };
    inline Int_t GetMutliplicity() const { return fMultiplicity; }
    inline Double_t GetPhi() const { return fPhi; };
    inline Double_t GetPhiError() const { return fPhiError; };
    inline TLorentzVector* GetVertex() const { return fVertex; };
    virtual EFormatType GetFormatType() const { return EFormatType::kReco; };
    virtual TString GetFormatName() const;
    Track* GetNewTrack() const;
    virtual EventInterface* CreateInterface() const = 0;
    virtual Float_t GetFieldVal(Int_t fieldID) const;
    virtual TString GetFieldName(Int_t fieldID) const;
    inline TClonesArray* GetV0HiddenInfo() const { return fV0sHiddenInfo; };
    virtual ~Event();
    /*******************************************************************************************************************************
     * functions below that are rarely used and used rather should not overwrite them
     ******************************************************************************************************************************/
    virtual void CopyHiddenSettings(const Event* /*event*/) {};
    virtual void Boost(Double_t vx, Double_t vy, Double_t vz);
    virtual Event* GetNewEvent() const;
    virtual Int_t GetMaxExpectedLinks() const { return 10; }
    virtual Bool_t HasHiddenSettings() const { return kTRUE; };
    virtual Package* Report() const;
    ClassDef(Event, 2)
  };
}  // namespace Hal
#endif /* HALEVENT_H_ */