/*
 * CorrelatorFitter.hpp, part of LatAnalyze 3
 *
 * Copyright (C) 2013 - 2020 Antonin Portelli
 *
 * LatAnalyze 3 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.
 *
 * LatAnalyze 3 is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with LatAnalyze 3.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef Latan_CorrelatorFitter_hpp_
#define Latan_CorrelatorFitter_hpp_

#include <LatAnalyze/Global.hpp>
#include <LatAnalyze/Functional/Model.hpp>
#include <LatAnalyze/Statistics/XYSampleData.hpp>

BEGIN_LATAN_NAMESPACE

enum class CorrelatorType {undefined, exp, cosh, sinh, linear, cst};

namespace CorrelatorModels
{
    struct ModelPar
    {
        CorrelatorType type;
        Index          nState;
    };

    DoubleModel makeExpModel(const Index nState);
    DoubleModel makeCoshModel(const Index nState, const Index nt);
    DoubleModel makeSinhModel(const Index nState, const Index nt);
    DoubleModel makeConstModel(void);
    DoubleModel makeLinearModel(void);
    ModelPar    parseModel(const std::string s);
    DoubleModel makeModel(const ModelPar par, const Index nt);
    DVec        parameterGuess(const DMatSample &corr, const ModelPar par);
};

class CorrelatorFitter
{
public:
    CorrelatorFitter(const DMatSample &corr);
    CorrelatorFitter(const std::vector<DMatSample> &corr);
    void setCorrelator(const DMatSample &corr);
    void setCorrelators(const std::vector<DMatSample> &corr);
    const DMatSample & getCorrelator(const Index i = 0) const;
    const std::vector<DMatSample> & getCorrelators(void) const;
    void setModel(const DoubleModel &model, const Index i = 0);
    const DoubleModel & getModel(const Index i = 0) const;
    void setFitRange(const Index tMin, const Index tMax, const Index i = 0);
    void setCorrelation(const bool isCorrelated, const Index i = 0, 
                        const Index j = 0);
    DMat getVarianceMatrix(void);
    void setThinning(const Index thinning, const Index i = 0);
    SampleFitResult fit(Minimizer &minimizer, const DVec &init);
    SampleFitResult fit(std::vector<Minimizer *> &minimizer, const DVec &init);
private:
    void refreshRanges(void);
private:
    Index                                nt_{0};
    std::unique_ptr<XYSampleData>        data_;
    std::vector<DoubleModel>             model_;
    std::vector<std::pair<Index, Index>> range_;
    std::vector<Index>                   thinning_;
};

END_LATAN_NAMESPACE

#endif // Latan_CorrelatorFitter_hpp_