PyDistribution.cpp

Go to the documentation of this file.

//
// Python API for Distribution
//
// Copyright (c) 2023, Chris Rogers, STFC Rutherford Appleton Laboratory, Didcot, UK
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL.  If not, see <https://www.gnu.org/licenses/>.
//
 
#include "Distribution/Distribution.h"
#include "PyOpal/PyCore/ExceptionTranslation.h"
#include "PyOpal/PyCore/Globals.h"
#include "PyOpal/PyCore/PyOpalObject.h"
 
namespace PyOpal {
    template <>
    std::vector<PyOpalObjectNS::AttributeDef>
        PyOpalObjectNS::PyOpalObject<Distribution>::attributes = {
            {"TYPE", "type", "", PyOpalObjectNS::PREDEFINED_STRING},
            {"FNAME", "filename", "", PyOpalObjectNS::STRING},
            {"INPUTMOUNITS", "momentum_units", "", PyOpalObjectNS::PREDEFINED_STRING},
    };
 
    namespace PyDistributionNS {
 
        void registerDistribution(PyOpalObjectNS::PyOpalObject<Distribution>& dist) {
            Object* obj = &(*dist.getOpalShared());
            obj->update();
            OpalData::getInstance()->define(obj);
        }
 
        BOOST_PYTHON_MODULE(distribution) {
            PyOpal::Globals::Initialise();
            ExceptionTranslation::registerExceptions();
            PyOpalObjectNS::PyOpalObject<Distribution> distributionObject;
            auto distributionClass = distributionObject.make_class("Distribution");
            distributionObject.addExecute(distributionClass);
            distributionClass.def("register", &registerDistribution);
        }
 
    }  // namespace PyDistributionNS
}  // namespace PyOpal