/* -*- mode: C++; c-basic-offset: 2; indent-tabs-mode: nil -*- */

/*
 *  Main authors:
 *     Guido Tack <guido.tack@monash.edu>
 */

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include <minizinc/eval_par.hh>
#include <minizinc/solver_instance_base.hh>

#ifdef _MSC_VER
#define _CRT_SECURE_NO_WARNINGS
#undef ERROR  // MICROsoft.
#undef min
#undef max
#endif

namespace MiniZinc {

SolverInstanceBase::Status SolverInstanceBase::solve(void) { return SolverInstance__ERROR; }

void SolverInstanceBase::reset(void) { assert(false); }

void SolverInstanceBase::resetWithConstraints(Model::iterator begin, Model::iterator end) {
  assert(false);
}

void SolverInstanceBase::processPermanentConstraints(Model::iterator begin, Model::iterator end) {
  assert(false);
}

void Registry::add(const std::string& name, poster p) { _registry.insert(std::make_pair(name, p)); }
void Registry::post(std::string name, Constraint* c) {
  std::unordered_map<std::string, poster>::iterator it = _registry.find(name);
  if (it == _registry.end()) {
    GCLock lock;
    throw InternalError("Error: solver backend cannot handle constraint: " + name + "\n");
  }
  it->second(_base, c);
}

void SolverInstanceBase::printSolution() {
  std::ostringstream oss;

  if (_options->printStatistics) printStatistics(1);  // Insert stats before sol separator
  if (0 == pS2Out) {
    getEnv()->evalOutput(std::cout);  // deprecated
    std::cout << oss.str();
    if (oss.str().size() && '\n' != oss.str().back()) std::cout << '\n';
    std::cout << "----------" << std::endl;
  } else
    getSolns2Out()->evalOutput(oss.str());
}

void SolverInstanceBase2::printSolution() {
  GCLock lock;
  assignSolutionToOutput();
  SolverInstanceBase::printSolution();
}

//   void
//   SolverInstanceBase::assignSolutionToOutput(void) {
//     for (VarDeclIterator it = getEnv()->output()->begin_vardecls(); it !=
//     getEnv()->output()->end_vardecls(); ++it) {
//       if (it->e()->e() == NULL) {
//         it->e()->e(getSolutionValue(it->e()->id()));
//       }
//     }
//   }

void SolverInstanceBase2::assignSolutionToOutput() {
  GCLock lock;

  MZN_ASSERT_HARD_MSG(
      0 != pS2Out, "Setup a Solns2Out object to use default solution extraction/reporting procs");

  if (_varsWithOutput.empty()) {
    for (VarDeclIterator it = getEnv()->flat()->begin_vardecls();
         it != getEnv()->flat()->end_vardecls(); ++it) {
      if (!it->removed()) {
        VarDecl* vd = it->e();
        if (!vd->ann().isEmpty()) {
          if (vd->ann().containsCall(constants().ann.output_array.aststr()) ||
              vd->ann().contains(constants().ann.output_var)) {
            _varsWithOutput.push_back(vd);
          }
        }
      }
    }
  }

  pS2Out->declNewOutput();  // Even for empty output decl

  // iterate over set of ids that have an output annotation && obtain their right hand side from the
  // flat model
  for (unsigned int i = 0; i < _varsWithOutput.size(); i++) {
    VarDecl* vd = _varsWithOutput[i];
    // std::cout << "DEBUG: Looking at var-decl with output-annotation: " << *vd << std::endl;
    if (Call* output_array_ann = Expression::dyn_cast<Call>(
            getAnnotation(vd->ann(), constants().ann.output_array.aststr()))) {
      assert(vd->e());

      if (ArrayLit* al = vd->e()->dyn_cast<ArrayLit>()) {
        std::vector<Expression*> array_elems;
        ArrayLit& array = *al;
        for (unsigned int j = 0; j < array.size(); j++) {
          if (Id* id = array[j]->dyn_cast<Id>()) {
            // std::cout << "DEBUG: getting solution value from " << *id  << " : " << id->v() <<
            // std::endl;
            array_elems.push_back(getSolutionValue(id));
          } else if (FloatLit* floatLit = array[j]->dyn_cast<FloatLit>()) {
            array_elems.push_back(floatLit);
          } else if (IntLit* intLit = array[j]->dyn_cast<IntLit>()) {
            array_elems.push_back(intLit);
          } else if (BoolLit* boolLit = array[j]->dyn_cast<BoolLit>()) {
            array_elems.push_back(boolLit);
          } else if (SetLit* setLit = array[j]->dyn_cast<SetLit>()) {
            array_elems.push_back(setLit);
          } else if (StringLit* strLit = array[j]->dyn_cast<StringLit>()) {
            array_elems.push_back(strLit);
          } else {
            std::ostringstream oss;
            oss << "Error: array element " << *array[j] << " is not an id nor a literal";
            throw InternalError(oss.str());
          }
        }
        GCLock lock;
        ArrayLit* dims;
        Expression* e = output_array_ann->arg(0);
        if (ArrayLit* al = e->dyn_cast<ArrayLit>()) {
          dims = al;
        } else if (Id* id = e->dyn_cast<Id>()) {
          dims = id->decl()->e()->cast<ArrayLit>();
        } else {
          throw -1;
        }
        std::vector<std::pair<int, int> > dims_v;
        for (int i = 0; i < dims->length(); i++) {
          IntSetVal* isv = eval_intset(getEnv()->envi(), (*dims)[i]);
          if (isv->size() == 0) {
            dims_v.push_back(std::pair<int, int>(1, 0));
          } else {
            dims_v.push_back(std::pair<int, int>(static_cast<int>(isv->min().toInt()),
                                                 static_cast<int>(isv->max().toInt())));
          }
        }
        ArrayLit* array_solution = new ArrayLit(Location(), array_elems, dims_v);
        KeepAlive ka(array_solution);
        auto& de = getSolns2Out()->findOutputVar(vd->id()->str().str());
        de.first->e(array_solution);
      }
    } else if (vd->ann().contains(constants().ann.output_var)) {
      Expression* sol = getSolutionValue(vd->id());
      vd->e(sol);
      auto& de = getSolns2Out()->findOutputVar(vd->id()->str().str());
      de.first->e(sol);
    }
  }
}

void SolverInstanceBase::flattenSearchAnnotations(const Annotation& ann,
                                                  std::vector<Expression*>& out) {
  for (ExpressionSetIter i = ann.begin(); i != ann.end(); ++i) {
    Expression* e = *i;
    if (e->isa<Call>() && (e->cast<Call>()->id().str() == "seq_search" ||
                           e->cast<Call>()->id().str() == "warm_start_array")) {
      Call* c = e->cast<Call>();
      auto* anns = c->arg(0)->cast<ArrayLit>();
      for (unsigned int i = 0; i < anns->size(); i++) {
        Annotation subann;
        subann.add((*anns)[i]);
        flattenSearchAnnotations(subann, out);
      }
    } else {
      out.push_back(*i);
    }
  }
}

}  // namespace MiniZinc