CpSolver.java

Go to the documentation of this file.

// Copyright 2010-2018 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.

package com.google.ortools.sat;

import com.google.ortools.sat.CpSolverResponse;
import com.google.ortools.sat.CpSolverStatus;
import com.google.ortools.sat.SatParameters;

public final class CpSolver {
  public CpSolver() {
    this.solveParameters = SatParameters.newBuilder();
  }

  public CpSolverStatus solve(CpModel model) {
    solveResponse = SatHelper.solveWithParameters(model.model(), solveParameters.build());
    return solveResponse.getStatus();
  }

  public CpSolverStatus solveWithSolutionCallback(CpModel model, CpSolverSolutionCallback cb) {
    solveResponse = SatHelper.solveWithParametersAndSolutionCallback(
        model.model(), solveParameters.build(), cb);
    return solveResponse.getStatus();
  }

  public CpSolverStatus searchAllSolutions(CpModel model, CpSolverSolutionCallback cb) {
    solveParameters.setEnumerateAllSolutions(true);
    solveResponse = SatHelper.solveWithParametersAndSolutionCallback(
        model.model(), solveParameters.build(), cb);
    solveParameters.setEnumerateAllSolutions(true);
    return solveResponse.getStatus();
  }

  public double objectiveValue() {
    return solveResponse.getObjectiveValue();
  }

  public double bestObjectiveBound() {
    return solveResponse.getBestObjectiveBound();
  }

  public long value(IntVar var) {
    return solveResponse.getSolution(var.getIndex());
  }

  public Boolean booleanValue(Literal var) {
    int index = var.getIndex();
    if (index >= 0) {
      return solveResponse.getSolution(index) != 0;
    } else {
      return solveResponse.getSolution(-index - 1) == 0;
    }
  }

  public CpSolverResponse response() {
    return solveResponse;
  }

  public long numBranches() {
    return solveResponse.getNumBranches();
  }

  public long numConflicts() {
    return solveResponse.getNumConflicts();
  }

  public double wallTime() {
    return solveResponse.getWallTime();
  }

  public double userTime() {
    return solveResponse.getUserTime();
  }

  public SatParameters.Builder getParameters() {
    return solveParameters;
  }

  public String responseStats() {
    return SatHelper.solverResponseStats(solveResponse);
  }

  private CpSolverResponse solveResponse;
  private final SatParameters.Builder solveParameters;
}