linearintegralmodel.h

#ifndef LSST_GAUSS2D_FIT_LINEARINTEGRALMODEL_H
#define LSST_GAUSS2D_FIT_LINEARINTEGRALMODEL_H
 
#include <map>
#include <memory>
 
#include "channel.h"
#include "parameters.h"
#include "integralmodel.h"
 
namespace lsst::gauss2d::fit {

class LinearIntegralModel : public IntegralModel {
public:
    typedef std::pair<std::reference_wrapper<const Channel>, std::shared_ptr<IntegralParameterD>>
            ChannelIntegralParameterD;
    typedef std::vector<ChannelIntegralParameterD> Data;

    explicit LinearIntegralModel(const Data *data_in);
    ~LinearIntegralModel();

    std::shared_ptr<IntegralParameterD> at(const Channel &channel);
    std::shared_ptr<const IntegralParameterD> at(const Channel &channel) const;
 
    typename Data::iterator begin() noexcept;
    typename Data::const_iterator cbegin() const noexcept;
 
    typename Data::iterator end() noexcept;
    typename Data::const_iterator cend() const noexcept;
 
    std::vector<std::reference_wrapper<const Channel>> get_channels() const override;
    double get_integral(const Channel &channel) const override;
    std::vector<std::pair<ParamBaseCRef, ExtraParamFactorValues>> get_integral_derivative_factors(
            const Channel &channel) const override;
 
    ParamRefs &get_parameters(ParamRefs &params, ParamFilter *filter = nullptr) const override;
    ParamCRefs &get_parameters_const(ParamCRefs &params, ParamFilter *filter = nullptr) const override;

    size_t size() const;
 
    std::string repr(bool name_keywords = false,
                     std::string_view namespace_separator = Object::CC_NAMESPACE_SEPARATOR) const override;
    std::string str() const override;
 
    const bool operator<(const IntegralModel &m) const { return &(*this) < &m; };
    const bool operator==(const IntegralModel &m) const { return &(*this) == &m; };
    const bool operator!=(const IntegralModel &m) const { return &(*this) != &m; };
 
private:
    Data _data = {};
    // This could be unordered, but std::hash<std::string> won't take const strings
    // ... and it doesn't seem to be worth the effort to work around
    std::map<std::reference_wrapper<const Channel>, std::shared_ptr<IntegralParameterD>> _map = {};
    struct Shared_enabler;
};
 
}  // namespace lsst::gauss2d::fit
 
#endif