KVGeoImport.h

 
#ifndef __KVGEOIMPORT_H
#define __KVGEOIMPORT_H
 
#include "KVGeoNavigator.h"
#include "KVGeoDNTrajectory.h"
#include "KVNucleusEvent.h"
#include <KVSegmentedDetector.h>
class KVMultiDetArray;
class KVIonRangeTable;
class KVIonRangeTableMaterial;
class KVGroup;
 
class KVGeoImport : public KVGeoNavigator {
 
   KVMultiDetArray* fArray;
   KVIonRangeTable* fRangeTable;
   KVDetector* fLastDetector;
   Bool_t fCreateArray;
   TString fDetectorPlugin;
   KVGeoDNTrajectory fCurrentTrajectory;
   std::unique_ptr<TVector3> fOrigin;
 
   KVDetector* GetCurrentDetector();
   KVDetector* BuildDetector(TString det_name, TGeoVolume* det_vol, bool segment = false);
   void AddLayer(KVDetector*, TGeoVolume*);
   KVDetector* AddSubDetector(KVSegmentedDetector*, TGeoVolume*, Int_t node_number);
   KVNameValueList fAcceptedDetectorNames;
   Bool_t fCheckDetVolNames;
 
protected:
   int ndets0;
   int count;
   KVNucleusEvent EVT;
   KVEvent* evt;
   KVNucleus* nuc;
   void ImportGeometry_init_import();
   void ImportGeometry_test_trajectory(double theta, double phi);
   void ImportGeometry_setup_geo();
 
   virtual KVDetector* instantiate_new_detector(const TString& det_name, bool segmented = false);
public:
   KVGeoImport(TGeoManager*, KVIonRangeTable*, KVMultiDetArray*, Bool_t create = kTRUE);
 
   void ParticleEntersNewVolume(KVNucleus*) override;
 
   void ImportGeometry(Double_t dTheta = 0.1/*degrees*/, Double_t dPhi = 1.0/*degrees*/,
                       Double_t ThetaMin = 0.0/*degrees*/, Double_t PhiMin = 0.0/*degrees*/,
                       Double_t ThetaMax = 180.0/*degrees*/, Double_t PhiMax = 360.0/*degrees*/);
 
   void SetLastDetector(KVDetector* d)
   {
      fLastDetector = d;
   }
 
   void SetDetectorPlugin(const TString& name)
   {
      fDetectorPlugin = name;
   }
   void PropagateParticle(KVNucleus*, TVector3* TheOrigin = nullptr) override;
 
   void SetOrigin(double x, double y, double z)
   {
      fOrigin = std::make_unique<TVector3>(x, y, z);
   }
 
   void AddAcceptedDetectorName(const char* name);
 
   ClassDefOverride(KVGeoImport, 0) //Import a ROOT geometry into a KVMultiDetArray object
};
 
#endif