FitObjective.h

Go to the documentation of this file.

//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      Sim/Fitting/FitObjective.h
//! @brief     Defines class FitObjective.
//!
//! @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)
//
//  ************************************************************************************************
 
#ifndef BORNAGAIN_SIM_FITTING_FITOBJECTIVE_H
#define BORNAGAIN_SIM_FITTING_FITOBJECTIVE_H
 
#ifndef USER_API
#include "Device/Data/ArrayUtils.h"
#include "Sim/Fitting/FitTypes.h"
 
class FitStatus;
class IChiSquaredModule;
class IMetricWrapper;
class IterationInfo;
class ObjectiveMetric;
class PyBuilderCallback;
class PyObserverCallback;
class SimDataPair;
class SimulationResult;
namespace mumufit {
 
class MinimizerResult;
class Parameters;
 
} // namespace mumufit
 
//! Holds vector of SimDataPair%s (experimental data and simulation results) for use in fitting.
//! Thereby supports simultaneous fitting of several data sets and model functions,
//! as demonstrated in example fit52/multiple_datasets.
 
class FitObjective {
public:
    FitObjective();
    virtual ~FitObjective();
 
#ifndef SWIG
    void execAddSimulationAndData(const simulation_builder_t& builder, const Datafield& data,
                                  std::unique_ptr<Datafield>&& stdv, double weight = 1.0);
#endif
    //! Constructs simulation/data pair for later fit.
    //! @param callback: simulation builder capable of producing simulations
    //! @param data: experimental data array
    //! @param weight: weight of dataset in metric calculations
    void addSimulationAndData(const PyBuilderCallback& callback,
                              const std::vector<std::vector<double>>& data, double weight = 1.0);
 
    void addSimulationAndData(const PyBuilderCallback& callback, const std::vector<double>& data,
                              double weight = 1.0);
 
    //! Constructs simulation/data pair for later fit.
    //! @param callback: simulation builder capable of producing simulations
    //! @param data: experimental data array
    //! @param stdv: data uncertainties array
    //! @param weight: weight of dataset in metric calculations
    void addSimulationAndData(const PyBuilderCallback& callback,
                              const std::vector<std::vector<double>>& data,
                              const std::vector<std::vector<double>>& stdv, double weight = 1.0);
 
    void addSimulationAndData(const PyBuilderCallback& callback, const std::vector<double>& data,
                              const std::vector<double>& stdv, double weight = 1.0);
 
    virtual double evaluate(const mumufit::Parameters& params);
 
    virtual std::vector<double> evaluate_residuals(const mumufit::Parameters& params);
 
    SimulationResult simulationResult(size_t i_item = 0) const;
    SimulationResult experimentalData(size_t i_item = 0) const;
    SimulationResult uncertaintyData(size_t i_item = 0) const;
    SimulationResult relativeDifference(size_t i_item = 0) const;
    SimulationResult absoluteDifference(size_t i_item = 0) const;
 
    std::vector<double> experimental_array() const;
    std::vector<double> simulation_array() const;
    std::vector<double> uncertainties() const;
    std::vector<double> weights_array() const;
 
    //! Initializes printing to standard output on every_nth fit iteration.
    void initPrint(int every_nth);
 
    //! Initializes observer callback to be called on every_nth fit iteration.
    void initPlot(int every_nth, PyObserverCallback& callback);
 
    IterationInfo iterationInfo() const;
 
    mumufit::MinimizerResult minimizerResult() const;
 
    //! Should be explicitly called on last iteration to notify all observers.
    void finalize(const mumufit::MinimizerResult& result);
 
    unsigned fitObjectCount() const;
 
    void execSimulations(const mumufit::Parameters& params);
 
    void setChiSquaredModule(const IChiSquaredModule& module);
 
    void setObjectiveMetric(const std::string& metric);
    //! Sets objective metric to the FitObjective.
    //! @param metric: metric name
    //! @param norm: metric norm name (defaults to L2-norm)
    void setObjectiveMetric(const std::string& metric, const std::string& norm);
 
    bool containsUncertainties(size_t i_item) const;
    bool allPairsHaveUncertainties() const;
 
    static std::string availableMetricOptions();
 
#ifndef SWIG
    //! Returns a reference to i-th SimDataPair.
    const SimDataPair& dataPair(size_t i_item = 0) const;
 
    void initPlot(int every_nth, fit_observer_t&& observer);
 
    bool isCompleted() const;
 
    void interruptFitting();
 
    bool isInterrupted() const;
 
    bool isFirstIteration() const;
 
    void setObjectiveMetric(std::unique_ptr<ObjectiveMetric> metric);
#endif // SWIG
 
private:
    using DataPairAccessor = std::vector<double> (SimDataPair::*)() const;
 
    std::vector<double> composeArray(DataPairAccessor getter) const;
 
    std::vector<SimDataPair> m_fit_objects;
    std::unique_ptr<IMetricWrapper> m_metric_module;
    std::unique_ptr<FitStatus> m_fit_status;
};
 
#endif // BORNAGAIN_SIM_FITTING_FITOBJECTIVE_H
#endif // USER_API