#pragma once
/**
 * @file PolyfitStat.h
 * @brief
 * @author Cat (null.null.null@qq.com)
 * @version 0.1
 * @date 2021-04-07
 *
 * Company: Baosight Co. Ltd.
 * DO NOT COPY/USE WITHOUT PERMISSION
 *
 * @ChangeLog:
 */
#include <boost/optional.hpp>
#include <mix_cc/algorithm/polyfit.h>
#include <mix_cc/utility/db2/db_com.h>
#include <log4cplus/LOG.h>

class PolyfitStat {
private:
  /* data */
  mix_cc::DBCom db_com_;
  int level_count_;
  double data_full_level_;
  long max_fit_count_;

  std::string rule_id_;

  std::vector<std::vector<double>> tmp_sampled_data_;

  struct SampleDataPolyfit {
    bool calculated_;
    std::vector<double> coeff_;
    std::vector<double> x_;
    std::vector<double> y_;
  };

  std::vector<SampleDataPolyfit>
      loaded_sample_data_; // index is level, data is sample data

public:
  PolyfitStat(std::string ruleid, int max_fit_count);
  ~PolyfitStat();

  int SetLevelInfo(int level_count, double data_full_level);

  int SaveXY(double x, double y, double level_param);

  int LoadXY();

  int Fit();

  boost::optional<double> GetY(double x, double level_param);
};