WIP: migrate Wenxing's DedxSvc.

Examples/options/tut_detsim_SDT.py

@@ -126,10 +126,19 @@ from Configurables import DriftChamberSensDetTool
 
 calo_sensdettool = CalorimeterSensDetTool("CalorimeterSensDetTool")
 driftchamber_sensdettool = DriftChamberSensDetTool("DriftChamberSensDetTool")
-driftchamber_sensdettool.DedxSimTool = "DummyDedxSimTool"
+
+# dedxoption = "DummyDedxSimTool"
+dedxoption = "BetheBlochEquationDedxSimTool"
+
+driftchamber_sensdettool.DedxSimTool = dedxoption
 
 from Configurables import DummyDedxSimTool
-dedx_simtool = DummyDedxSimTool("DummyDedxSimTool")
+from Configurables import BetheBlochEquationDedxSimTool
+
+if dedxoption == "DummyDedxSimTool":
+    dedx_simtool = DummyDedxSimTool("DummyDedxSimTool")
+elif dedxoption == "BetheBlochEquationDedxSimTool":
+    dedx_simtool = BetheBlochEquationDedxSimTool("BetheBlochEquationDedxSimTool")
 
 ##############################################################################
 # POD I/O

Service/DedxSvc/CMakeLists.txt

-gaudi_subdir(DedxSvc v0r0)
-
-set(DedxSvc_srcs
-    src/*.cpp
-)
-find_package(Geant4 REQUIRED ui_all vis_all)
-include(${Geant4_USE_FILE})
-gaudi_install_headers(DedxSvc)
-
-gaudi_add_module(DedxSvc ${DedxSvc_srcs}
-    INCLUDE_DIRS GaudiKernel
-    LINK_LIBRARIES GaudiKernel
-)

Service/DedxSvc/DedxSvc/IDedxSvc.h

-#ifndef I_Dedx_SVC_H
-#define I_Dedx_SVC_H
-
-#include "GaudiKernel/IService.h"
-
-#include "G4Step.hh"
-
-class IDedxSvc: virtual public IService {
-public:
-    DeclareInterfaceID(IDedxSvc, 0, 1); // major/minor version
-    
-    virtual ~IDedxSvc() = default;
-    virtual float pred(const G4Step* aStep)=0 ;
-
-};
-
-
-#endif

Simulation/DetSimDedx/CMakeLists.txt

@@ -11,6 +11,7 @@ find_package(DD4hep COMPONENTS DDG4 REQUIRED)
 
 set(DetSimDedx_srcs
     src/DummyDedxSimTool.cpp
+    src/BetheBlochEquationDedxSimTool.cpp
 )
 
 gaudi_add_module(DetSimDedx ${DetSimDedx_srcs}

Service/DedxSvc/src/DedxSvc.cpp → Simulation/DetSimDedx/src/BetheBlochEquationDedxSimTool.cpp

-#include "DedxSvc.h"
+#include "BetheBlochEquationDedxSimTool.h"
+#include "G4Step.hh"
 
-//https://folk.uib.no/ruv004/
-DECLARE_COMPONENT(DedxSvc)
+// https://folk.uib.no/ruv004/
+DECLARE_COMPONENT(BetheBlochEquationDedxSimTool)
 
-DedxSvc::DedxSvc(const std::string& name, ISvcLocator* svc)
-    : base_class(name, svc)
-{
-}
-
-DedxSvc::~DedxSvc()
-{
-}
 
-float DedxSvc::pred(const G4Step* aStep)
+double BetheBlochEquationDedxSimTool::dedx(const G4Step* aStep)
 {
     G4Track* gTrack = aStep->GetTrack() ;
     G4int z = gTrack->GetDefinition()->GetPDGCharge();
@@ -30,7 +23,7 @@ float DedxSvc::pred(const G4Step* aStep)
     return dedx*m_material_density; // MeV / cm 
 }
 
-StatusCode DedxSvc::initialize()
+StatusCode BetheBlochEquationDedxSimTool::initialize()
 {
     m_distribution = new std::normal_distribution<double>(0, m_resolution);
     m_me = 0.511*pow(10,6);//0.511 MeV to eV
@@ -39,7 +32,7 @@ StatusCode DedxSvc::initialize()
     return StatusCode::SUCCESS;
 }
 
-StatusCode DedxSvc::finalize()
+StatusCode BetheBlochEquationDedxSimTool::finalize()
 {
     return StatusCode::SUCCESS;
 }

Service/DedxSvc/src/DedxSvc.h → Simulation/DetSimDedx/src/BetheBlochEquationDedxSimTool.h

-#ifndef Dedx_SVC_H
-#define Dedx_SVC_H
+#ifndef BetheBlochEquationDedxSimTool_h
+#define BetheBlochEquationDedxSimTool_h
 
-#include "DedxSvc/IDedxSvc.h"
-#include <GaudiKernel/Service.h>
-#include "G4Step.hh"
+#include "DetSimInterface/IDedxSimTool.h"
+#include <GaudiKernel/AlgTool.h>
 #include <random>
 
-class DedxSvc : public extends<Service, IDedxSvc>
-{
+class BetheBlochEquationDedxSimTool: public extends<AlgTool, IDedxSimTool> {
     public:
-        DedxSvc(const std::string& name, ISvcLocator* svc);
-        ~DedxSvc();
-
+        using extends::extends;
 
         StatusCode initialize() override;
         StatusCode finalize() override;
-        float pred(const G4Step* aStep) override;
+        double dedx(const G4Step* aStep) override;
 
     private:
 
@@ -27,6 +23,7 @@ class DedxSvc : public extends<Service, IDedxSvc>
         float m_me;// Here me is the electron rest mass
         float m_K; // K was set as a constant.
         float m_I; // Mean excitation energy
+
         std::default_random_engine m_generator;
         std::normal_distribution<double>* m_distribution;
 };