gmagneto.h

Go to the documentation of this file.

#pragma once
 
// gemc
#include "gfield.h"
#include "gfield_options.h"
#include "gfieldConventions.h"
 
// geant4
#include "G4ThreeVector.hh"
 
// c++
#include <memory>
#include <set>
#include <string>
 
class GLogger;
 
class GMagneto : public GBase<GMagneto> {
public:
    explicit GMagneto(const std::shared_ptr<GOptions>& gopts,
                      const std::set<std::string>&     required_fields = {});
 
    static std::shared_ptr<GField> initialize_magnetic_field(
        const std::shared_ptr<GOptions>& gopts, double& field_polarity,
        std::shared_ptr<GLogger> caller_log = nullptr);
 
    static double magnetic_field_magnitude_tesla(
        const std::shared_ptr<GField>& magnetic_field, const G4ThreeVector& position);
 
private:
    using gFieldMap    = std::unordered_map<std::string, std::shared_ptr<GField>>;
    using gFieldMgrMap = std::unordered_map<std::string, std::shared_ptr<G4FieldManager>>;
 
    std::shared_ptr<gFieldMap> fields_map;
 
    std::shared_ptr<gFieldMgrMap> fields_manager;
 
public:
    bool isField(const std::string& name) const { return fields_map->find(name) != fields_map->end(); }
 
    std::vector<std::string> getFieldNames() const;
 
    std::shared_ptr<GField> getField(std::string name) {
        bool not_found = (fields_map->find(name) == fields_map->end());
 
        if (not_found) { log->error(ERR_WRONG_FIELD_NOT_FOUND, "GField >", name, "< not found. Exiting."); }
        return fields_map->at(name);
    }
 
    std::shared_ptr<G4FieldManager> getFieldMgr(std::string name) {
 
        // Debug visibility: print out the full map of registered field managers.
        for (const auto& [key, value] : *fields_manager) {
            log->info(2, "GFieldManager >", key, "< >", value);
        }
 
        if (fields_manager->find(name) == fields_manager->end()) {
            log->error(ERR_WRONG_FIELD_NOT_FOUND, "GField >", name, "< not found. Exiting.");
        }
 
        return fields_manager->at(name);
    }
};