model.h

Go to the documentation of this file.

#ifndef __MODEL_H__
#define __MODEL_H__

#include <mpi.h>
#include "init_models.h"

#include <fstream>
#include <sstream>


namespace CTF_int {
  class Model {
    public:
      virtual void update(MPI_Comm cm){};
      virtual void print(){};
      virtual void print_uo(){};
      virtual void load_coeff(std::string file_name){};
      virtual void write_coeff(std::string file_name){};
      virtual void dump_data(std::string path){};
  };

  void update_all_models(MPI_Comm cm);
  void print_all_models();
  void load_all_models(std::string file_name);
  void write_all_models(std::string file_name);
  void dump_all_models(std::string path);

  template <int nparam>
  class LinModel : Model {
    private:
      int64_t nobs;
      int mat_lda;
      bool is_tuned;
      double tot_time;
      double over_time;
      double under_time;
      double avg_tot_time;
      double avg_over_time;
      double avg_under_time;
      bool is_active;

    public:
      int hist_size;
      double * time_param_mat;
      double coeff_guess[nparam];

      //double regularization[nparam];

      char * name;

      LinModel(double const * init_guess, char const * name, int hist_size=32768);

      LinModel();
      ~LinModel();

      void update(MPI_Comm cm);

      void observe(double const * time_param);

      bool should_observe(double const * time_param);

      double est_time(double const * param);

      void print();

      void print_uo();

      double* get_coeff();

      void load_coeff(std::string file_name);

      void write_coeff(std::string file_name);

      void dump_data(std::string path);
  };

  template <int nparam>
  class CubicModel : Model {
    private:
      LinModel<nparam*(nparam+1)*(nparam+2)/6+nparam*(nparam+1)/2+nparam> lmdl;

    public:
      CubicModel(double const * init_guess, char const * name, int hist_size=8192);

      ~CubicModel();

      void update(MPI_Comm cm);

      void observe(double const * time_param);

      bool should_observe(double const * time_param);

      double est_time(double const * param);

      void print();

      void print_uo();

      double* get_coeff();

      void load_coeff(std::string file_name);

      void write_coeff(std::string file_name);

      void dump_data(std::string path);

  };

}

#endif