add geant4 scene and gcompose app
This commit is contained in:
@@ -23,52 +23,127 @@
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#include <Geant4/G4Box.hh>
|
||||
#include <Geant4/G4LogicalVolume.hh>
|
||||
#include <Geant4/G4Material.hh>
|
||||
#include <Geant4/G4NistManager.hh>
|
||||
#include <Geant4/G4LogicalVolume.hh>
|
||||
#include <Geant4/G4PVPlacement.hh>
|
||||
#include <Geant4/G4RunManager.hh>
|
||||
#include <Geant4/G4RunManagerFactory.hh>
|
||||
#include <Geant4/G4SystemOfUnits.hh>
|
||||
#include <Geant4/G4VPhysicalVolume.hh>
|
||||
|
||||
#include "Core/Vector.h"
|
||||
#include "Matter.h"
|
||||
#include "HEP/Geant/DetectorConstruction.hh"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Solid.h"
|
||||
|
||||
#include "Scene.h"
|
||||
|
||||
#include "PhysicsList.hh"
|
||||
#include "ActionInitialization.hh"
|
||||
|
||||
namespace uLib {
|
||||
|
||||
|
||||
class DetectorsScenePimpl {
|
||||
|
||||
class SceneDetectorConstruction : public DetectorConstruction {
|
||||
public:
|
||||
|
||||
|
||||
// members //
|
||||
//Vector<Solid> m_Solids;
|
||||
SceneDetectorConstruction(class SceneImpl *owner);
|
||||
G4VPhysicalVolume *Construct() override;
|
||||
private:
|
||||
class SceneImpl *m_Owner;
|
||||
};
|
||||
|
||||
class SceneImpl {
|
||||
public:
|
||||
// constructor //
|
||||
SceneImpl() : m_RunManager(G4RunManagerFactory::CreateRunManager()) {}
|
||||
|
||||
// destructor //
|
||||
~SceneImpl() {
|
||||
if (m_RunManager) delete m_RunManager;
|
||||
if (m_World) delete m_World;
|
||||
}
|
||||
|
||||
void Initialize() {
|
||||
// Set mandatory initialization classes for Geant4
|
||||
m_RunManager->SetUserInitialization(new SceneDetectorConstruction(this));
|
||||
m_RunManager->SetUserInitialization(new PhysicsList);
|
||||
m_RunManager->SetUserInitialization(new ActionInitialization);
|
||||
|
||||
DetectorsScene::DetectorsScene() :
|
||||
d(new DetectorsScenePimpl())
|
||||
{}
|
||||
// Initialize Geant4
|
||||
m_RunManager->Initialize();
|
||||
}
|
||||
|
||||
DetectorsScene::~DetectorsScene()
|
||||
{
|
||||
delete d;
|
||||
}
|
||||
// members //
|
||||
Vector<Solid *> m_Solids;
|
||||
Solid *m_World = nullptr;
|
||||
G4RunManager *m_RunManager;
|
||||
};
|
||||
|
||||
void DetectorsScene::AddSolid(const Solid &solid)
|
||||
{
|
||||
// d->m_Solids.push_back(solid);
|
||||
SceneDetectorConstruction::SceneDetectorConstruction(SceneImpl *owner)
|
||||
: DetectorConstruction("Scene"), m_Owner(owner) {}
|
||||
|
||||
G4VPhysicalVolume *SceneDetectorConstruction::Construct() {
|
||||
return m_Owner->m_World->GetPhysical();
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Scene::Scene() : d(new SceneImpl()) {}
|
||||
Scene::~Scene() { delete d; }
|
||||
|
||||
void Scene::AddSolid(Solid *solid, Solid *parent) {
|
||||
d->m_Solids.push_back(solid);
|
||||
if (!d->m_World) {
|
||||
d->m_World = solid;
|
||||
} else {
|
||||
solid->SetParent(parent ? parent : d->m_World);
|
||||
}
|
||||
}
|
||||
|
||||
void Scene::ConstructWorldBox(const ContainerBox *box, const char *material) {
|
||||
// Get nist material manager
|
||||
if (!d->m_World) {
|
||||
d->m_World = new Solid("World");
|
||||
d->m_World->SetNistMaterial(material);
|
||||
AddSolid(d->m_World);
|
||||
}
|
||||
|
||||
Vector3f size = box->GetSize();
|
||||
|
||||
G4Box *solidWorld = new G4Box("World",
|
||||
0.5 * size(0),
|
||||
0.5 * size(1),
|
||||
0.5 * size(2));
|
||||
|
||||
G4LogicalVolume *logicWorld = new G4LogicalVolume(solidWorld,
|
||||
d->m_World->GetMaterial(),
|
||||
d->m_World->GetName());
|
||||
|
||||
d->m_World->SetLogical(logicWorld);
|
||||
|
||||
G4PVPlacement *physWorld = new G4PVPlacement(
|
||||
nullptr,
|
||||
G4ThreeVector(0, 0, 0),
|
||||
logicWorld,
|
||||
d->m_World->GetName(),
|
||||
0,
|
||||
false,
|
||||
0,
|
||||
true);
|
||||
|
||||
d->m_World->SetPhysical(physWorld);
|
||||
|
||||
Matrix4f transform = box->GetMatrix();
|
||||
d->m_World->SetTransform(transform);
|
||||
}
|
||||
|
||||
void Scene::Initialize() {
|
||||
d->Initialize();
|
||||
}
|
||||
|
||||
} // namespace uLib
|
||||
|
||||
Reference in New Issue
Block a user