gmagneto.h

Go to the documentation of this file.

#pragma once
 
// gemc
#include "gfield.h"
#include "gfieldOptions.h"
#include "gStateMessage.h"
 
class GMagneto : public GStateMessage {
public:
    GMagneto(GOptions *gopts);
 
    ~GMagneto() {
        delete gFieldMap;
        delete gFieldMgrMap;
    }
 
private:
 
    std::map<std::string, GField *> *gFieldMap; 
    std::map<std::string, G4FieldManager *> *gFieldMgrMap; 
 
public:
 
    bool isField(std::string name) {
        return gFieldMap->find(name) != gFieldMap->end();
    }
 
 
    GField *getField(std::string name) {
        bool not_found = (gFieldMap->find(name) == gFieldMap->end());
 
        if (not_found) {
            logError("GField " + name + " not found.", EC__WRONG_FIELD_NOT_FOUND);
        }
        return gFieldMap->at(name);
    }
 
    G4FieldManager *getFieldMgr(std::string name) {
        if (gFieldMgrMap->find(name) == gFieldMgrMap->end()) {
            logError("G4FieldManager " + name + " not found.", EC__WRONG_FIELD_NOT_FOUND);
        }
 
        return gFieldMgrMap->at(name);
    }
 
};