g4objectsFactory.cc

Go to the documentation of this file.

 
#include "g4objectsFactory.h"
 
#include "gutilities.h"
#include "gsystemConventions.h"
#include "g4systemConventions.h"
 
// Geant4
#include "G4LogicalVolumeStore.hh"
#include "G4Material.hh"
#include "G4NistManager.hh"
#include "G4PVPlacement.hh"
 
// c++
#include <algorithm>
#include <cctype>
#include <iostream>
#include <string_view>
#include <vector>
 
namespace {
    std::vector<std::string> tokenizeRotation(std::string rotation) {
        std::replace(rotation.begin(), rotation.end(), ',', ' ');
        return gutilities::getStringVectorFromString(rotation);
    }
 
    std::string lowercase(std::string value) {
        std::transform(value.begin(), value.end(), value.begin(),
                       [](unsigned char c) { return static_cast<char>(std::tolower(c)); });
        return value;
    }
 
    void requireRotationTokenCount(const std::vector<std::string> &tokens,
                                   size_t expected,
                                   const GVolume *s) {
        if (tokens.size() == expected) { return; }
        std::cerr << "     >> ERROR: rotation <" << s->getRot() << "> for " << s->getName()
                  << " has wrong number of entries. Exiting." << std::endl;
        exit(1);
    }
 
    void applyOrderedRotation(G4RotationMatrix *rot,
                              const std::string &order,
                              const std::vector<double> &angles,
                              const GVolume *s) {
        if (order == "xzy") {
            rot->rotateX(angles[0]);
            rot->rotateZ(angles[1]);
            rot->rotateY(angles[2]);
        }
        else if (order == "yxz") {
            rot->rotateY(angles[0]);
            rot->rotateX(angles[1]);
            rot->rotateZ(angles[2]);
        }
        else if (order == "yzx") {
            rot->rotateY(angles[0]);
            rot->rotateZ(angles[1]);
            rot->rotateX(angles[2]);
        }
        else if (order == "zxy") {
            rot->rotateZ(angles[0]);
            rot->rotateX(angles[1]);
            rot->rotateY(angles[2]);
        }
        else if (order == "zyx") {
            rot->rotateZ(angles[0]);
            rot->rotateY(angles[1]);
            rot->rotateX(angles[2]);
        }
        else {
            std::cerr << "     >> ERROR: Ordered rotation <" << order << "> for " << s->getName()
                      << " is wrong, it's none of the following: xzy, yxz, yzx, zxy or zyx. Exiting."
                      << std::endl;
            exit(1);
        }
    }
}
 
// Configure factory behavior (overlap checking + backup material).
void G4ObjectsFactory::initialize_context(int check_overlaps,
                                          const std::string &backup_mat) {
    checkOverlaps = check_overlaps;
    backupMaterial = backup_mat;
}
 
// Convenience: retrieve the solid pointer for a named volume from the wrapper map.
G4VSolid *G4ObjectsFactory::getSolidFromMap(const std::string &vname,
                                            std::unordered_map<std::string, G4Volume *> *g4s) {
    auto it = g4s->find(vname);
    return (it != g4s->end()) ? it->second->getSolid() : nullptr;
}
 
// Convenience: retrieve the logical pointer for a named volume from the wrapper map.
G4LogicalVolume *G4ObjectsFactory::getLogicalFromMap(const std::string &volume_name,
                                                     std::unordered_map<std::string, G4Volume *> *g4s) {
    auto it = g4s->find(volume_name);
    return (it != g4s->end()) ? it->second->getLogical() : nullptr;
}
 
// Convenience: retrieve the physical pointer for a named volume from the wrapper map.
G4VPhysicalVolume *G4ObjectsFactory::getPhysicalFromMap(const std::string &vname,
                                                        std::unordered_map<std::string, G4Volume *> *g4s) {
    auto it = g4s->find(vname);
    return (it != g4s->end()) ? it->second->getPhysical() : nullptr;
}
 
G4VisAttributes G4ObjectsFactory::createVisualAttributes(const GVolume *s) {
    // Convert GEMC color + opacity into a Geant4 color. The helper returns a \c G4Colour.
    std::string_view color = s->getColor();
    double opacity = s->getOpacity();
 
    const auto g4color = gutilities::makeG4Colour(color, opacity);
 
    log->info(2, className(), " createVisualAttributes for color ", color,
              " resulted in RGB = (", g4color.GetRed(), ", ", g4color.GetGreen(), ", ", g4color.GetBlue(),
              ", opacity: ", opacity, ")");
 
    G4VisAttributes attr(g4color);
 
    // Visibility and style flags are stored on the volume definition.
    s->isVisible() ? attr.SetVisibility(true) : attr.SetVisibility(false);
 
    switch (s->getStyle()) {
        case 0:
            attr.SetForceWireframe(true);
            break;
        case 1:
            attr.SetForceSolid(true);
            break;
        case 2:
            attr.SetForceCloud(true);
            break;
        default:
            break;
    }
 
 
    return attr;
}
 
G4RotationMatrix *G4ObjectsFactory::getRotation(const GVolume *s) {
    auto rot = new G4RotationMatrix();
 
    const auto tokens = tokenizeRotation(s->getRot());
    if (tokens.empty()) { return rot; }
 
    const auto rotationType = tokens[0];
    if (rotationType != "ordered:" && rotationType != "doubleRotation:") {
        requireRotationTokenCount(tokens, 3, s);
        const auto pars = gutilities::getG4NumbersFromStringVector(tokens);
        rot->rotateX(pars[0]);
        rot->rotateY(pars[1]);
        rot->rotateZ(pars[2]);
    }
    else if (rotationType == "doubleRotation:") {
        requireRotationTokenCount(tokens, 7, s);
        const auto pars = gutilities::getG4NumbersFromStringVector(
            {tokens[1], tokens[2], tokens[3], tokens[4], tokens[5], tokens[6]});
        rot->rotateX(pars[0]);
        rot->rotateY(pars[1]);
        rot->rotateZ(pars[2]);
        rot->rotateX(pars[3]);
        rot->rotateY(pars[4]);
        rot->rotateZ(pars[5]);
    }
    else if (rotationType == "ordered:") {
        requireRotationTokenCount(tokens, 5, s);
        const auto order = lowercase(tokens[1]);
        const auto pars = gutilities::getG4NumbersFromStringVector({tokens[2], tokens[3], tokens[4]});
        applyOrderedRotation(rot, order, pars, s);
    }
    else {
        std::cerr << "     >> ERROR: rotation type <" << rotationType << "> unknown. Exiting."
                  << std::endl;
        exit(1);
    }
    return rot;
}
 
G4ThreeVector G4ObjectsFactory::getPosition(const GVolume *s) {
    // Base placement position.
    G4ThreeVector pos(0., 0., 0.);
    const auto vec = gutilities::getG4NumbersFromString(s->getPos());
    if (vec.size() == 3) pos.set(vec[0], vec[1], vec[2]);
 
    // Optional shift modifier (applied after parsing the base position).
    if (s->getShift() != GSYSTEMNOMODIFIER) {
        const auto shift = gutilities::getG4NumbersFromString(s->getShift());
        if (shift.size() == 3) pos += G4ThreeVector(shift[0], shift[1], shift[2]);
    }
    return pos;
}
 
G4LogicalVolume *G4ObjectsFactory::buildLogical(const GVolume *s,
                                                std::unordered_map<std::string, G4Volume *> *g4s) {
    const std::string g4name = s->getG4Name();
    auto thisG4Volume = getOrCreateG4Volume(g4name, g4s);
 
    // Logical exists, return it.
    if (thisG4Volume->getLogical()) return thisG4Volume->getLogical();
 
    // If this volume is a "copy of" another, reuse the logical volume if it already exists.
    std::string copyOf = s->getCopyOf();
    if (copyOf != "" && copyOf != UNINITIALIZEDSTRINGQUANTITY) {
        auto gsystem = s->getSystem();
        auto volume_copy = gsystem + "/" + copyOf;
        auto copyG4Volume = getOrCreateG4Volume(volume_copy, g4s);
        if (copyG4Volume->getLogical() != nullptr) {
            return copyG4Volume->getLogical();
        }
    }
 
    // Material lookup:
    // - first try requested material
    // - if missing and a backup material was configured, fall back to it
    auto *nist = G4NistManager::Instance();
    auto *mat = nist->FindOrBuildMaterial(s->getMaterial());
    if (!mat && !backupMaterial.empty()) {
        mat = nist->FindOrBuildMaterial(backupMaterial);
        log->warning("Material <", s->getMaterial(), "> not found. Using backup material <", backupMaterial, ">.");
    }
 
    if (!mat) {
        log->error(ERR_G4MATERIALNOTFOUND,
                   "Material <", s->getMaterial(), "> not found.");
    }
 
    // Create the logical volume with the already-created solid.
    auto *logical = new G4LogicalVolume(thisG4Volume->getSolid(),
                                        mat, g4name);
 
    // Apply visualization attributes (color/opacity/visibility/style).
    logical->SetVisAttributes(createVisualAttributes(s));
    thisG4Volume->setLogical(logical, log);
    return logical;
}
 
G4VPhysicalVolume *G4ObjectsFactory::buildPhysical(const GVolume *s,
                                                   std::unordered_map<std::string, G4Volume *> *g4s) {
    // Nonexistent volumes are ignored by design.
    if (!s->getExistence()) return nullptr;
 
    // Mother/logical prerequisites must exist; otherwise caller will retry later.
    if (!checkPhysicalDependencies(s, g4s)) return nullptr;
 
    const std::string g4name = s->getG4Name();
    auto thisG4Volume = getOrCreateG4Volume(g4name, g4s);
    auto logicalVolume = thisG4Volume->getLogical();
 
    // If this is a copy, reuse the source logical volume when available.
    std::string copyOf = s->getCopyOf();
    if (copyOf != "" && copyOf != UNINITIALIZEDSTRINGQUANTITY) {
        auto gsystem = s->getSystem();
        auto volume_copy = gsystem + "/" + copyOf;
        auto copyG4Volume = getOrCreateG4Volume(volume_copy, g4s);
        if (copyG4Volume->getLogical() != nullptr) {
            logicalVolume = copyG4Volume->getLogical();
        }
    }
 
    // Create the placement only once; subsequent calls return the cached physical volume.
    if (!thisG4Volume->getPhysical()) {
        auto *rotation = getRotation(s);
        G4ThreeVector translation = getPosition(s);
        const auto placementType = s->getG4PlacementType();
 
        if (placementType == "active") {
            G4RotationMatrix rotation_instance = *rotation;
            delete rotation;
            thisG4Volume->setPhysical(new G4PVPlacement(G4Transform3D(rotation_instance, translation),
                                                        logicalVolume,
                                                        g4name,
                                                        getLogicalFromMap(s->getG4MotherName(), g4s),
                                                        false,
                                                        s->getPCopyNo(),
                                                        checkOverlaps > 0),
                                      log);
        }
        else if (placementType == "passive") {
            thisG4Volume->setPhysical(new G4PVPlacement(rotation,
                                                        translation,
                                                        logicalVolume,
                                                        g4name,
                                                        getLogicalFromMap(s->getG4MotherName(), g4s),
                                                        false,
                                                        s->getPCopyNo(),
                                                        checkOverlaps > 0),
                                      log);
        }
        else {
            log->error(ERR_G4PLACEMENTTYPE,
                       "GVolume <", s->getName(), "> has unsupported g4placement_type <",
                       placementType, ">. Use 'active' or 'passive'.");
        }
    }
    return thisG4Volume->getPhysical();
}
 
bool G4ObjectsFactory::build_g4volume(const GVolume *s,
                                      std::unordered_map<std::string, G4Volume *> *g4s) {
    const auto name = s->getG4Name();
 
    // Build steps can fail due to missing dependencies; each returns nullptr in that case.
    auto sbuild = buildSolid(s, g4s);
    auto lbuild = buildLogical(s, g4s);
    auto pbuild = buildPhysical(s, g4s);
 
    const bool okSolid = (sbuild != nullptr);
    const bool okLogical = (lbuild != nullptr);
    const bool okPhysical = (pbuild != nullptr);
    const bool okAll = okSolid && okLogical && okPhysical;
 
    log->info(2, className(), " result for ",
              name,
              ": solid: ", okSolid,
              " logical: ", okLogical,
              " physical: ", okPhysical);
 
    return okAll;
}