g4world.h

Go to the documentation of this file.

#pragma once
// gemc
#include <gemc/gsystem/gworld.h>
 
// g4system
#include "g4volume.h"
#include "g4objectsFactories/g4objectsFactory.h"
#include <gemc/g4system/g4systemConventions.h>
 
// c++
#include <memory>
#include <string>
#include <unordered_map>
 
// geant4
#include "G4NistManager.hh"
 
class G4OpticalSurface;
 
class G4World : public GBase<G4World>
{
public:
    G4World(const GWorld* gworld, const std::shared_ptr<GOptions>& gopts);
 
    ~G4World() {
        for (auto& [name, vol] : g4volumesMap) { delete vol; }
    }
 
    // G4World now owns the raw G4Volume* wrappers (freed in the destructor). It is only ever held
    // by shared_ptr, so forbid copy/move to avoid a shallow copy double-freeing the wrappers.
    G4World(const G4World&)            = delete;
    G4World& operator=(const G4World&) = delete;
    G4World(G4World&&)                 = delete;
    G4World& operator=(G4World&&)      = delete;
 
    // ────── lookup / mutators ────────────────────────────────────────
 
    [[nodiscard]] const G4Volume* getG4Volume(const std::string& volumeName) const;
 
    void setFieldManagerForVolume(const std::string& volumeName,
                                  G4FieldManager*    fm,
                                  bool               forceToAllDaughters);
 
    [[nodiscard]] bool is_empty() const noexcept { return g4volumesMap.empty(); }
 
    [[nodiscard]] std::size_t number_of_volumes() const noexcept { return g4volumesMap.size(); }
 
    [[nodiscard]] G4ObjectsFactory* get_factory(const std::string& factoryName) {
        auto it = g4systemFactory.find(factoryName);
        if (it != g4systemFactory.end()) return it->second;
        else {
            log->error(ERR_G4SYSTEMFACTORYNOTFOUND,
                       "G4World: factory <", factoryName, "> not found.");
        }
    }
 
    [[nodiscard]] std::unordered_map<std::string, G4Volume*> get_g4volumes_map() const {
        return g4volumesMap;
    }
 
private:
    // ────── helper functions ─────────────────────────────────────────
 
    std::string g4FactoryNameFromSystemFactory(const std::string& factory) const;
 
    bool createG4Material(const std::shared_ptr<GMaterial>& gmaterial);
 
    void buildDefaultMaterialsElementsAndIsotopes();
 
    void buildMaterials(SystemMap* system_map);
 
    void createG4SystemFactory(const std::shared_ptr<GOptions>& gopts,
                               SystemMap*                       gsystemsMap,
                               const std::string&               backup_material,
                               int                              check_overlaps);
 
    bool build_g4volume(const GVolume* s, G4ObjectsFactory* objectFactory);
 
    void buildOpticalSurfaces(SystemMap* system_map);
 
    G4OpticalSurface* getOrCreateOpticalSurface(const std::string& systemName, const GMirror* gmirror);
 
    // ────── dependency helpers ──────────────────────────────────────
    // These pointers are cached so that multiple predefined materials can reuse the same element instances.
    G4Element* Deuterium = nullptr;
    G4Element* Helium3   = nullptr;
    G4Element* Tritium   = nullptr;
 
    // ────── data members ────────────────────────────────────────────
 
    std::unordered_map<std::string, G4ObjectsFactory*> g4systemFactory;
 
    std::unordered_map<std::string, G4Volume*> g4volumesMap;
 
    std::unordered_map<std::string, G4Material*> g4materialsMap;
 
    std::unordered_map<std::string, G4OpticalSurface*> g4opticalSurfacesMap;
};