// Copyright 2010-2025 Google LLC
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// IWYU pragma: private, include "ortools/math_opt/cpp/math_opt.h"
// IWYU pragma: friend "ortools/math_opt/cpp/.*"

#ifndef ORTOOLS_MATH_OPT_CPP_MODEL_SOLVE_PARAMETERS_H_
#define ORTOOLS_MATH_OPT_CPP_MODEL_SOLVE_PARAMETERS_H_

#include <sys/types.h>

#include <cstdint>
#include <initializer_list>
#include <optional>
#include <vector>

#include "absl/container/flat_hash_set.h"
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/time/time.h"
#include "ortools/math_opt/cpp/linear_constraint.h"
#include "ortools/math_opt/cpp/map_filter.h"  // IWYU pragma: export
#include "ortools/math_opt/cpp/model.h"
#include "ortools/math_opt/cpp/solution.h"
#include "ortools/math_opt/cpp/variable_and_expressions.h"
#include "ortools/math_opt/model_parameters.pb.h"
#include "ortools/math_opt/storage/model_storage.h"

namespace operations_research {
namespace math_opt {

// Parameters to control a single solve that are specific to the input
// model (see SolveParametersProto for model independent parameters).
struct ModelSolveParameters {
  // Returns the parameters that empty DualSolution and DualRay, only keep the
  // values of all variables in PrimalSolution and PrimalRay.
  //
  // This is a shortcut method that is equivalent to setting the dual filters
  // with MakeSkipAllFilter().
  static ModelSolveParameters OnlyPrimalVariables();

  // Returns the parameters that empty DualSolution and DualRay, only keep the
  // values of the specified variables in PrimalSolution and PrimalRay.
  //
  // The input Collection must be usable in a for-range loop with Variable
  // values. This will be typically a std::vector<Variable> or and
  // std::initializer_list<Variable> (see the other overload).
  //
  // This is a shortcut method that is equivalent to setting the dual filters
  // with MakeSkipAllFilter() and the variable_values_filter with
  // MakeKeepKeysFilter(variables).
  //
  // Example:
  //   std::vector<Variable> decision_vars = ...;
  //   const auto params =
  //      ModelSolveParameters::OnlySomePrimalVariables(decision_vars);
  template <typename Collection>
  static ModelSolveParameters OnlySomePrimalVariables(
      const Collection& variables);

  // Returns the parameters that empty DualSolution and DualRay, only keeping
  // the values of the specified variables in PrimalSolution and PrimalRay.
  //
  // See the other overload's documentation for details. This overload is needed
  // since C++ can't guess the type when using an initializer list expression.
  //
  // Example:
  //   const Variable a = ...;
  //   const Variable b = ...;
  //   const auto params =
  //      ModelSolveParameters::OnlySomePrimalVariables({a, b});
  static ModelSolveParameters OnlySomePrimalVariables(
      std::initializer_list<Variable> variables);

  // The filter that is applied to variable_values of both PrimalSolution and
  // PrimalRay.
  MapFilter<Variable> variable_values_filter;

  // The filter that is applied to dual_values of DualSolution and DualRay.
  MapFilter<LinearConstraint> dual_values_filter;

  // The filter that is applied to quadratic_dual_values of DualSolution and
  // DualRay.
  MapFilter<QuadraticConstraint> quadratic_dual_values_filter;

  // The filter that is applied to reduced_costs of DualSolution and DualRay.
  MapFilter<Variable> reduced_costs_filter;

  // Optional initial basis for warm starting simplex LP solvers. If set, it is
  // expected to be valid.
  std::optional<Basis> initial_basis;

  // A suggested starting solution for the solver.
  //
  // MIP solvers generally only want primal information (`variable_values`),
  // while LP solvers want both primal and dual information (`dual_values`).
  //
  // Many MIP solvers can work with: (1) partial solutions that do not specify
  // all variables or (2) infeasible solutions. In these cases, solvers
  // typically solve a sub-MIP to complete/correct the hint.
  //
  // How the hint is used by the solver, if at all, is highly dependent on the
  // solver, the problem type, and the algorithm used. The most reliable way to
  // ensure your hint has an effect is to read the underlying solvers logs with
  // and without the hint.
  //
  // Simplex-based LP solvers typically prefer an initial basis to a solution
  // hint (they need to crossover to convert the hint to a basic feasible
  // solution otherwise).
  struct SolutionHint {
    // Values should be finite and not NaN (otherwise, `Solve()` will return a
    // `Status` error).
    VariableMap<double> variable_values;

    // Values should be finite and not NaN (otherwise, `Solve()` will return a
    // `Status` error).
    LinearConstraintMap<double> dual_values;

    // Returns a failure if the referenced variables and constraints don't
    // belong to the input expected_storage (which must not be nullptr).
    absl::Status CheckModelStorage(ModelStorageCPtr expected_storage) const;

    // Returns the proto equivalent of this object.
    //
    // The caller should use CheckModelStorage() as this function does not check
    // internal consistency of the referenced variables and constraints.
    SolutionHintProto Proto() const;

    // Returns the hint corresponding to this proto and the given model.
    //
    // This can be useful when loading a hint from another format, e.g. with
    // MPModelProtoSolutionHintToMathOptHint().
    static absl::StatusOr<SolutionHint> FromProto(
        const Model& model, const SolutionHintProto& hint_proto);
  };

  // Optional solution hints. If the underlying solver only accepts a single
  // hint, the first hint is used.
  std::vector<SolutionHint> solution_hints;

  // Optional branching priorities. Variables with higher values will be
  // branched on first. Variables for which priorities are not set get the
  // solver's default priority (usually zero).
  VariableMap<int32_t> branching_priorities;

  // Parameters for an individual objective in a multi-objective model.
  struct ObjectiveParameters {
    // Optional objective degradation absolute tolerance. For a hierarchical
    // multi-objective solver, each objective fⁱ is processed in priority order:
    // the solver determines the optimal objective value Γⁱ, if it exists,
    // subject to all constraints in the model and the additional constraints
    // that fᵏ(x) = Γᵏ (within tolerances) for each k < i. If set, a solution is
    // considered to be "within tolerances" for this objective fᵏ if
    // |fᵏ(x) - Γᵏ| ≤ `objective_degradation_absolute_tolerance`.
    //
    // See also `objective_degradation_relative_tolerance`; if both parameters
    // are set for a given objective, the solver need only satisfy one to be
    // considered "within tolerances".
    //
    //  If set, must be nonnegative.
    std::optional<double> objective_degradation_absolute_tolerance;

    // Optional objective degradation relative tolerance. For a hierarchical
    // multi-objective solver, each objective fⁱ is processed in priority order:
    // the solver determines the optimal objective value Γⁱ, if it exists,
    // subject to all constraints in the model and the additional constraints
    // that fᵏ(x) = Γᵏ (within tolerances) for each k < i. If set, a solution is
    // considered to be "within tolerances" for this objective fᵏ if
    // |fᵏ(x) - Γᵏ| ≤ `objective_degradation_relative_tolerance` * |Γᵏ|.
    //
    // See also `objective_degradation_absolute_tolerance`; if both parameters
    // are set for a given objective, the solver need only satisfy one to be
    // considered "within tolerances".
    //
    //  If set, must be nonnegative.
    std::optional<double> objective_degradation_relative_tolerance;

    // Maximum time a solver should spend on optimizing this particular
    // objective (or infinite if not set).
    //
    // Note that this does not supersede the global time limit in
    // SolveParametersProto.time_limit; both will be enforced when set.
    //
    // This value is not a hard limit, solve time may slightly exceed this
    // value.
    absl::Duration time_limit = absl::InfiniteDuration();

    // Returns the proto equivalent of this object.
    absl::StatusOr<ObjectiveParametersProto> Proto() const;

    static absl::StatusOr<ObjectiveParameters> FromProto(
        const ObjectiveParametersProto& proto);
  };
  // Parameters for individual objectives in a multi-objective model.
  ObjectiveMap<ObjectiveParameters> objective_parameters;

  // Optional lazy constraint annotations. Included linear constraints will be
  // marked as "lazy" with supporting solvers, meaning that they will only be
  // added to the working model as-needed as the solver runs.
  //
  // Note that this an algorithmic hint that does not affect the model's
  // feasible region; solvers not supporting these annotations will simply
  // ignore it.
  absl::flat_hash_set<LinearConstraint> lazy_linear_constraints;

  // Returns a failure if the referenced variables and constraints do not belong
  // to the input expected_storage (which must not be nullptr).
  absl::Status CheckModelStorage(ModelStorageCPtr expected_storage) const;

  // Returns the proto equivalent of this object.
  //
  // The caller should use CheckModelStorage() as this function does not check
  // internal consistency of the referenced variables and constraints.
  absl::StatusOr<ModelSolveParametersProto> Proto() const;

  // Returns the ModelSolveParameters corresponding to this proto and the given
  // model.
  static absl::StatusOr<ModelSolveParameters> FromProto(
      const Model& model, const ModelSolveParametersProto& proto);
};

////////////////////////////////////////////////////////////////////////////////
// Inline functions implementations.
////////////////////////////////////////////////////////////////////////////////

template <typename Collection>
ModelSolveParameters ModelSolveParameters::OnlySomePrimalVariables(
    const Collection& variables) {
  ModelSolveParameters parameters = OnlyPrimalVariables();
  parameters.variable_values_filter = MakeKeepKeysFilter(variables);
  return parameters;
}

}  // namespace math_opt
}  // namespace operations_research

#endif  // ORTOOLS_MATH_OPT_CPP_MODEL_SOLVE_PARAMETERS_H_