Source code for pyomo.solvers.plugins.solvers.IPOPT

#  ___________________________________________________________________________
#
#  Pyomo: Python Optimization Modeling Objects
#  Copyright (c) 2008-2025
#  National Technology and Engineering Solutions of Sandia, LLC
#  Under the terms of Contract DE-NA0003525 with National Technology and
#  Engineering Solutions of Sandia, LLC, the U.S. Government retains certain
#  rights in this software.
#  This software is distributed under the 3-clause BSD License.
#  ___________________________________________________________________________

import os
import subprocess

from pyomo.common import Executable
from pyomo.common.collections import Bunch
from pyomo.common.errors import ApplicationError
from pyomo.common.tee import capture_output
from pyomo.common.tempfiles import TempfileManager

from pyomo.opt.base import ProblemFormat, ResultsFormat
from pyomo.opt.base.solvers import _extract_version, SolverFactory
from pyomo.opt.results import SolverStatus, SolverResults, TerminationCondition
from pyomo.opt.solver import SystemCallSolver

from pyomo.solvers.amplfunc_merge import amplfunc_merge

import logging

logger = logging.getLogger('pyomo.solvers')


[docs] @SolverFactory.register('ipopt', doc='The Ipopt NLP solver') class IPOPT(SystemCallSolver): """ An interface to the Ipopt optimizer that uses the AMPL Solver Library. """
[docs] def __init__(self, **kwds): # # Call base constructor # kwds["type"] = "ipopt" super(IPOPT, self).__init__(**kwds) # # Setup valid problem formats, and valid results for each problem format # Also set the default problem and results formats. # self._valid_problem_formats = [ProblemFormat.nl] self._valid_result_formats = {} self._valid_result_formats[ProblemFormat.nl] = [ResultsFormat.sol] self.set_problem_format(ProblemFormat.nl) # Note: Undefined capabilities default to 'None' self._capabilities = Bunch() self._capabilities.linear = True self._capabilities.integer = False self._capabilities.quadratic_objective = True self._capabilities.quadratic_constraint = True self._capabilities.sos1 = False self._capabilities.sos2 = False
def _default_results_format(self, prob_format): return ResultsFormat.sol def _default_executable(self): executable = Executable("ipopt") if not executable: logger.warning( "Could not locate the 'ipopt' executable, " "which is required for solver %s" % self.name ) self.enable = False return None return executable.path() def _get_version(self): """ Returns a tuple describing the solver executable version. """ solver_exec = self.executable() if solver_exec is None: return _extract_version('') results = subprocess.run( [solver_exec, "-v"], timeout=self._version_timeout, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True, ) return _extract_version(results.stdout)
[docs] def create_command_line(self, executable, problem_files): assert self._problem_format == ProblemFormat.nl assert self._results_format == ResultsFormat.sol # # Define log file # if self._log_file is None: self._log_file = TempfileManager.create_tempfile(suffix="_ipopt.log") fname = problem_files[0] if '.' in fname: tmp = fname.split('.') if len(tmp) > 2: fname = '.'.join(tmp[:-1]) else: fname = tmp[0] self._soln_file = fname + ".sol" # # Define results file (since an external parser is used) # self._results_file = self._soln_file # # Define command line # env = os.environ.copy() # # Merge the PYOMO_AMPLFUNC (externals defined within # Pyomo/Pyomo) with any user-specified external function # libraries # amplfunc = amplfunc_merge(env) if amplfunc: env['AMPLFUNC'] = amplfunc cmd = [executable, problem_files[0], '-AMPL'] if self._timer: cmd.insert(0, self._timer) env_opt = [] of_opt = [] ofn_option_used = False for key in self.options: if key == 'solver': continue elif key.startswith("OF_"): assert len(key) > 3 of_opt.append((key[3:], self.options[key])) else: if key == "option_file_name": ofn_option_used = True if isinstance(self.options[key], str) and ' ' in self.options[key]: env_opt.append(key + "=\"" + str(self.options[key]) + "\"") cmd.append(str(key) + "=" + str(self.options[key])) else: env_opt.append(key + "=" + str(self.options[key])) cmd.append(str(key) + "=" + str(self.options[key])) if len(of_opt) > 0: # If the 'option_file_name' command-line option # was used, we don't know if we should overwrite, # merge it, or it is was a mistake, so raise an # exception. Maybe this can be changed. if ofn_option_used: raise ValueError( "The 'option_file_name' command-line " "option for Ipopt can not be used " "when specifying options for the " "options file (i.e., options that " "start with 'OF_'" ) # Now check if an 'ipopt.opt' file exists in the # current working directory. If so, we need to # make it clear that this file will be ignored. default_of_name = os.path.join(os.getcwd(), 'ipopt.opt') if os.path.exists(default_of_name): logger.warning( "A file named '%s' exists in " "the current working directory, but " "Ipopt options file options (i.e., " "options that start with 'OF_') were " "provided. The options file '%s' will " "be ignored." % (default_of_name, default_of_name) ) # Now write the new options file options_filename = TempfileManager.create_tempfile(suffix="_ipopt.opt") with open(options_filename, "w") as f: for key, val in of_opt: f.write(key + " " + str(val) + "\n") # Now set the command-line option telling Ipopt # to use this file env_opt.append('option_file_name="' + str(options_filename) + '"') cmd.append('option_file_name=' + str(options_filename)) envstr = "%s_options" % self.options.solver # Merge with any options coming in through the environment env[envstr] = " ".join(env_opt) return Bunch(cmd=cmd, log_file=self._log_file, env=env)
[docs] def process_output(self, rc): if os.path.exists(self._results_file): return super(IPOPT, self).process_output(rc) else: res = SolverResults() res.solver.status = SolverStatus.warning res.solver.termination_condition = TerminationCondition.other if os.path.exists(self._log_file): with open(self._log_file) as f: for line in f: if "TOO_FEW_DEGREES_OF_FREEDOM" in line: res.solver.message = line.split(':')[2].strip() assert "degrees of freedom" in res.solver.message return res
def has_linear_solver(self, linear_solver): import pyomo.core as AML m = AML.ConcreteModel() m.x = AML.Var() m.o = AML.Objective(expr=(m.x - 2) ** 2) try: with capture_output() as OUT: self.solve(m, tee=True, options={'linear_solver': linear_solver}) except ApplicationError: return False return 'running with linear solver' in OUT.getvalue()