/*++
Copyright (c) 2013 Microsoft Corporation

Module Name:

    opt_context.h

Abstract:
    Facility for running optimization problem.

Author:

    Anh-Dung Phan (t-anphan) 2013-10-16

Notes:

    TODO:

    - type check objective term and assertions. It should pass basic sanity being
      integer, real (, bit-vector) or other supported objective function type.

    - add appropriate statistics tracking to opt::context

--*/
#ifndef _OPT_CONTEXT_H_
#define _OPT_CONTEXT_H_

#include "ast.h"
#include "solver.h"
#include "optimize_objectives.h"
#include "opt_maxsmt.h"

namespace opt {

    class opt_solver;

    class context {
        ast_manager& m;
        expr_ref_vector  m_hard_constraints;
        expr_ref_vector  m_soft_constraints;
        vector<rational> m_weights;
        app_ref_vector   m_objectives;
        svector<bool>    m_is_max;
        ref<solver>      m_solver;
        params_ref       m_params;
        optimize_objectives m_opt_objectives;
        maxsmt           m_maxsmt;
    public:
        context(ast_manager& m);

        void add_soft_constraint(expr* f, rational const& w) {
            m_soft_constraints.push_back(f);
            m_weights.push_back(w);
        }

        void add_objective(app* t, bool is_max);

        void add_hard_constraint(expr* f) {
            m_hard_constraints.push_back(f);
        }

        void set_solver(solver* s) {
            m_solver = s;
        }

        void optimize();

        void cancel();
        void reset_cancel();

        void collect_statistics(statistics& stats);

        static void collect_param_descrs(param_descrs & r);

        void updt_params(params_ref& p);

    private:
        bool is_maxsat_problem() const;

        opt_solver& get_opt_solver(solver& s); 

    };

}

#endif