mageec-ml.h

Go to the documentation of this file.

/*  MAGEEC Machine Learning Interface
    Copyright (C) 2013, 2014 Embecosm Limited and University of Bristol

    This file is part of MAGEEC.

    This program 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.

    This program 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 this program.  If not, see <http://www.gnu.org/licenses/>. */

#ifndef __MAGEEC_ML_H__
#define __MAGEEC_ML_H__

#include <string>
#include <vector>
#include <stdint.h>
#include "mageec-decision.h"
#include "mageec-features.h"
#include "mageec-passes.h"
#include "mageec-result.h"
#include "mageec-db.h"

namespace mageec
{
class mageec_ml
{
  database *db;
  std::vector<mageec_pass*> known_passes;
public:
  virtual ~mageec_ml() {}

  virtual int init (std::string dbfilename);

  virtual int init (std::string compiler_version,
                    std::string compiler_target);
  void new_file (void);

  void end_file (void);

  void finish (void);

  void provide_features (std::vector<mageec_feature*> features);

  std::vector<mageec_pass> get_passes();

  std::vector<mageec_pass*> all_passes();

  virtual decision make_decision (mageec_pass* pass,
                                  std::vector<mageec_feature*> features);

  void process_results();

  void add_result (std::vector<mageec_feature*> features,
                   std::vector<mageec_pass*> passes,
                   int64_t metric,
                   bool good);

  void add_result (result res);

  void raw_result (std::vector<mageec_feature*> features,
                   std::vector<mageec_pass*> passes,
                   std::vector<int64_t> metrics,
                   bool good);
};

class file_ml : public mageec_ml
{
  std::vector<std::string> passlist;
public:
  virtual int init (std::string dbfilename);
  virtual int init (std::string compiler_version,
                    std::string compiler_target);
  virtual decision make_decision (mageec_pass* pass,
                                  std::vector<mageec_feature*> features);
};

} // End namespace mageec

#endif