R3BNeuland.h

Go to the documentation of this file.

/******************************************************************************
 *   Copyright (C) 2019 GSI Helmholtzzentrum für Schwerionenforschung GmbH    *
 *   Copyright (C) 2019-2025 Members of R3B Collaboration                     *
 *                                                                            *
 *             This software is distributed under the terms of the            *
 *                 GNU General Public Licence (GPL) version 3,                *
 *                    copied verbatim in the file "LICENSE".                  *
 *                                                                            *
 * In applying this license GSI does not waive the privileges and immunities  *
 * granted to it by virtue of its status as an Intergovernmental Organization *
 * or submit itself to any jurisdiction.                                      *
 ******************************************************************************/
 
#ifndef R3BNEULAND_H
#define R3BNEULAND_H
 
#include "R3BDetector.h"
#include "R3BNeulandPoint.h"
#include "TLorentzVector.h"
#include <R3BIOConnector.h>
#include <R3BNeulandCommon.h>
#include <R3BNeulandGeoCreator.h>
#include <TClonesArray.h>
#include <memory>
#include <string>
 
class FairVolume;
class TClonesArray;
class R3BNeulandGeoPar;

 
class R3BNeuland : public FairDetector
{
  public:
    R3BNeuland();

    R3BNeuland(const TString& geoFile, const TGeoTranslation& trans, const TGeoRotation& rot = TGeoRotation());

    explicit R3BNeuland(const TString& geoFile, const TGeoCombiTrans& combi = TGeoCombiTrans());

    R3BNeuland(int nDP, const TGeoTranslation& trans, const TGeoRotation& rot = TGeoRotation());

    explicit R3BNeuland(int nDP, const TGeoCombiTrans& combi = TGeoCombiTrans());

    void EnableAutoGeoBuild(bool is_enabled = true) { is_geo_auto_built = is_enabled; }

    void SetLocationDirection(const TGeoCombiTrans& pos) { rot_trans_ = pos; }
 
  private:
    bool is_geo_auto_built = false;                       
    bool is_last_hit_done_ = false;                       
    int num_of_planes_ = R3B::Neuland::MaxNumberOfPlanes; 
 
    // Track information to be stored until the track leaves the active volume.
    int fTrackId = 0;            
    int fPaddleId = 0;           
    int particle_id_ = 0;        
    int parent_particle_id_ = 0; 
    double time_ = 0.;           
    double length_ = 0.;         
    double energy_loss_ = 0.;    
    double light_yield_ = 0.;    
    TLorentzVector pos_in_;      
    TLorentzVector pos_out_;     
    TLorentzVector mom_in_;      
    TLorentzVector mom_out_;     

    std::unique_ptr<TClonesArray> tca_points_buffer_ = std::make_unique<TClonesArray>(R3BNeulandPoint::Class()); 
    R3B::OutputVectorConnector<R3BNeulandPoint> neuland_points_{ "NeulandPoints" }; 
    R3BNeulandGeoPar* neuland_geo_par_ = nullptr; 
    R3B::Neuland::Geometry::Creator geo_creator_; 
    TGeoCombiTrans rot_trans_;
    std::string geo_file_;
    std::map<int, int> track_pid_map_;
 
    // private virtual functions:
 
    auto ProcessHits(FairVolume* /*v*/ = nullptr) -> bool override;
 
    void EndOfEvent() override;
 
    void FinishEvent() override;
 
    void Print(Option_t* /*unused*/ = "") const override;
 
    void Reset() override;
 
    auto CheckIfSensitive(std::string name) -> bool override;
 
    [[nodiscard]] auto GetCollection(int iColl) const -> TClonesArray* override;
 
    void Register() override;
 
    // private non-virtual member functions:
 
    void reset_values();
 
    void write_parameter_file();
 
    void ConstructGeometry() override;
 
    void Initialize() override;
 
    // non-virtual functions:
 
    void create_geo();
 
    void create_geo_from_root_file();
 
    ClassDefOverride(R3BNeuland, 4);
};
 
#endif // R3BNEULAND_H