PyFmt2.cpp

Go to the documentation of this file.

//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      Sim/Export/PyFmt2.cpp
//! @brief     Implements functions from namespace pyfmt2.
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************
 
#include "Sim/Export/PyFmt2.h"
#include "Base/Axis/FixedBinAxis.h"
#include "Base/Axis/PointwiseAxis.h"
#include "Base/Const/Units.h"
#include "Base/Math/Constants.h"
#include "Base/Py/PyFmt.h"
#include "Base/Util/Algorithms.h"
#include "Base/Util/StringUtils.h"
#include "Device/Mask/Ellipse.h"
#include "Device/Mask/InfinitePlane.h"
#include "Device/Mask/Line.h"
#include "Device/Mask/Polygon.h"
#include "Device/Mask/Rectangle.h"
#include "Param/Distrib/Distributions.h"
#include "Param/Distrib/ParameterDistribution.h"
#include "Param/Distrib/RangedDistributions.h"
#include "Sim/Export/PyFmtLimits.h"
#include <iomanip>
 
namespace Py::Fmt2 {
 
//! Returns fixed Python code snippet that defines the function "simulate".
 
std::string representShape2D(const std::string& indent, const IShape2D* ishape, bool mask_value,
                             std::function<std::string(double)> printValueFunc)
{
    std::ostringstream result;
    result << std::setprecision(12);
 
    if (const auto* shape = dynamic_cast<const Polygon*>(ishape)) {
        std::vector<double> xpos, ypos;
        shape->getPoints(xpos, ypos);
        result << indent << "points = [";
        for (size_t i = 0; i < xpos.size(); ++i) {
            result << "[" << printValueFunc(xpos[i]) << ", " << printValueFunc(ypos[i]) << "]";
            if (i != xpos.size() - 1)
                result << ", ";
        }
        result << "]\n";
        result << indent << "simulation.addMask("
               << "ba.Polygon(points), " << Py::Fmt::printBool(mask_value) << ")\n";
    } else if (dynamic_cast<const InfinitePlane*>(ishape))
        result << indent << "simulation.maskAll()\n";
    else if (const auto* shape = dynamic_cast<const Ellipse*>(ishape)) {
        result << indent << "simulation.addMask(";
        result << "ba.Ellipse(" << printValueFunc(shape->getCenterX()) << ", "
               << printValueFunc(shape->getCenterY()) << ", " << printValueFunc(shape->radiusX())
               << ", " << printValueFunc(shape->radiusY());
        if (shape->getTheta() != 0.0)
            result << ", " << Py::Fmt::printDegrees(shape->getTheta());
        result << "), " << Py::Fmt::printBool(mask_value) << ")\n";
    }
 
    else if (const auto* shape = dynamic_cast<const Rectangle*>(ishape)) {
        result << indent << "simulation.addMask(";
        result << "ba.Rectangle(" << printValueFunc(shape->getXlow()) << ", "
               << printValueFunc(shape->getYlow()) << ", " << printValueFunc(shape->getXup())
               << ", " << printValueFunc(shape->getYup()) << "), " << Py::Fmt::printBool(mask_value)
               << ")\n";
    }
 
    else if (const auto* shape = dynamic_cast<const VerticalLine*>(ishape)) {
        result << indent << "simulation.addMask(";
        result << "ba.VerticalLine(" << printValueFunc(shape->getXpos()) << "), "
               << Py::Fmt::printBool(mask_value) << ")\n";
    }
 
    else if (const auto* shape = dynamic_cast<const HorizontalLine*>(ishape)) {
        result << indent << "simulation.addMask(";
        result << "ba.HorizontalLine(" << printValueFunc(shape->getYpos()) << "), "
               << Py::Fmt::printBool(mask_value) << ")\n";
    } else
        throw std::runtime_error("representShape2D(const IShape2D*) -> Error. Unknown shape");
 
    return result.str();
}
 
//! Prints an axis.
std::string printAxis(const IAxis* axis, const std::string& unit)
{
    std::ostringstream result;
    if (const auto* a = dynamic_cast<const FixedBinAxis*>(axis); a)
        result << "ba.FixedBinAxis(" << Py::Fmt::printString(a->axisName()) << ", " << a->size()
               << ", " << Py::Fmt::printValue(a->min(), unit) << ", "
               << Py::Fmt::printValue(a->max(), unit) << ")";
    else if (const auto* a = dynamic_cast<const PointwiseAxis*>(axis); a) {
        result << "numpy.asarray([";
        const std::vector<double>& points = a->binCenters();
        for (auto iter = points.begin(); iter != points.end() - 1; ++iter)
            result << Py::Fmt::printValue(*iter, unit) << ",";
        result << Py::Fmt::printValue(points.back(), unit) << "])\n";
    } else
        throw std::runtime_error("printAxis not implemented for current axis type");
    return result.str();
}
 
std::string printParameterDistribution(const ParameterDistribution& par_distr,
                                       const std::string& distVarName, const std::string& units)
{
    std::ostringstream result;
 
    result << "ba.ParameterDistribution(ba." << par_distr.whichParameterAsPyEnum() << ", "
           << distVarName << ", " << par_distr.nDraws() << ", "
           << Py::Fmt::printDouble(par_distr.sigmaFactor())
           << Py::Fmt::printRealLimitsArg(par_distr.getLimits(), units) << ")";
 
    return result.str();
}
 
std::string printRangedDistribution(const IRangedDistribution& distr)
{
    std::ostringstream result;
    result << Py::Fmt::indent() << "distribution = ba.";
    result << distr.name();
    result << "(" << distr.nSamples() << ", " << Py::Fmt::printDouble(distr.sigmaFactor());
    if (!distr.limits().isLimitless())
        result << Py::Fmt::printRealLimitsArg(distr.limits());
    result << ")";
    return result.str();
}
 
} // namespace Py::Fmt2