
    tKgD                         d Z ddlZddlZddlmZmZ ddlmZ ddl	m
Z
 ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$ ddl%m&Z&m'Z'm(Z( d	 Z)d
 Z*d Z+	 	 	 	 	 	 	 	 ddZ,y)a^  HiGHS Linear Optimization Methods

Interface to HiGHS linear optimization software.
https://highs.dev/

.. versionadded:: 1.5.0

References
----------
.. [1] Q. Huangfu and J.A.J. Hall. "Parallelizing the dual revised simplex
           method." Mathematical Programming Computation, 10 (1), 119-142,
           2018. DOI: 10.1007/s12532-017-0130-5

    N   )OptimizeWarningOptimizeResult)warn)_highs_wrapper)	CONST_INFMESSAGE_LEVEL_NONEHIGHS_OBJECTIVE_SENSE_MINIMIZEMODEL_STATUS_NOTSETMODEL_STATUS_LOAD_ERRORMODEL_STATUS_MODEL_ERRORMODEL_STATUS_PRESOLVE_ERRORMODEL_STATUS_SOLVE_ERRORMODEL_STATUS_POSTSOLVE_ERRORMODEL_STATUS_MODEL_EMPTYMODEL_STATUS_OPTIMALMODEL_STATUS_INFEASIBLE$MODEL_STATUS_UNBOUNDED_OR_INFEASIBLEMODEL_STATUS_UNBOUNDED5MODEL_STATUS_REACHED_DUAL_OBJECTIVE_VALUE_UPPER_BOUND%MODEL_STATUS_REACHED_OBJECTIVE_TARGETMODEL_STATUS_REACHED_TIME_LIMIT$MODEL_STATUS_REACHED_ITERATION_LIMITHIGHS_SIMPLEX_STRATEGY_DUAL HIGHS_SIMPLEX_CRASH_STRATEGY_OFF)HIGHS_SIMPLEX_EDGE_WEIGHT_STRATEGY_CHOOSE*HIGHS_SIMPLEX_EDGE_WEIGHT_STRATEGY_DANTZIG(HIGHS_SIMPLEX_EDGE_WEIGHT_STRATEGY_DEVEX0HIGHS_SIMPLEX_EDGE_WEIGHT_STRATEGY_STEEPEST_EDGE)
csc_matrixvstackissparsec                 *   i ddt         dt        dt        dt        dt        dt
        dt        dt        dt        dt        dt        dt        dt        dt        d	t        d
}d}|j                  | |      \  }}| d|  d| d}||fS )zCConverts HiGHS status number/message to SciPy status number/messageN)   z%HiGHS did not provide a status code. )r$    )   r%   )r   z&Optimization terminated successfully. )r   zTime limit reached. )r   zIteration limit reached. )r&   zThe problem is infeasible. )   zThe problem is unbounded. )r$   z(The problem is unbounded or infeasible. )r$   z*The HiGHS status code was not recognized. z(HiGHS Status z: ))r   r   r   r   r   r   r   MODEL_STATUS_RDOVUBr   r   r   r   r   r   r   get)highs_statushighs_messagescipy_statuses_messagesunrecognizedscipy_statusscipy_messages         a/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/scipy/optimize/_linprog_highs.py_highs_to_scipy_status_messager2   7   s)   C:CWC 	 C 	!'	C
 	$WC 	!'C 	%gC 	!'C 	WC 	.wC 	KC 	()DC 	-.NC 	 !CC 	 AC  	- /B!C$ EL##L,?  L-%%l^2m_AGM&&    c                     t        j                  |       }t        j                  d      5  t        j                  | |         t        z  | |<   d d d        | S # 1 sw Y   | S xY w)Nignoreinvalid)npisinferrstatesignr   )xinfss     r1   _replace_infr>   T   sK    88A;D	X	&''!D'"9,$ 
'H 
'Hs   #AA#c                 @   	 || j                            S # t        $ r ||    cY S t        $ rp t        j                  t
              }|j                  |   j                  }t        d| d|  dt        |j                                d| d	t        d       ||   cY S w xY w)NzOption z is z, but only values in z are allowed. Using default: .r'   
stacklevel)lowerAttributeErrorKeyErrorinspect	signature_linprog_highs
parametersdefaultr   setkeysr   )option
option_strchoicessigdefault_strs        r1   _convert_to_highs_enumrR   \   s    $v||~&& v $/nnZ088wzl$vh.CGLLN#$$A}A 	, {##$s    BA5BBc                    |rd| d}t        |t        d       t        |	dt        t        t
        t        dd      }| \  }}}}}}}}|j                  j                         \  }}t        j                  d	
      5  t        j                  |       t        j                  z  }ddd       |}|}|}t        j                  |f      }t        j                  ||f      }t        |      st        |      rt        ||f      }nt        j                  ||f      }t!        |      }i d|dt"        d|d|dt$        d|d|d|d|d|d|d|dt&        dt(        d|d|d|
} | j+                  |       t-        |      }t-        |      }t-        |      }t-        |      }|t        j.                  |      dk(  rt        j0                  d      }nt        j2                  |      }t5        ||j6                  |j8                  |j:                  |||||j=                  t        j>                        | 
      }!d|!v rH|!d   }"t        j2                  |"tA        |      d       }#t        j2                  |"dtA        |             }"nd\  }"}#d|!v r|!d   }$t        j2                  |$dtA        |             }%t        j2                  |$tA        |      d       }&t        j2                  |!d   d ddf         }'t        j2                  |!d   dddf         }(n
d\  }%}&d\  }'}(|!jC                  d!d      })|!jC                  d"d      }*tE        |)|*      \  }+}d#|!v rt        j2                  |!d#         nd},|,|"|#tG        |"|%d$      tG        |#|&d$      tG        |,dn|,|z
  |(d$      tG        |,dn||,z
  |'d$      |!jC                  d%      |+|!d!   tH        k(  ||!jC                  d&d      xs |!jC                  d'd      |!jC                  d(      d)}-t        jJ                  |,      rG|E|-j+                  |!jC                  d*d      |!jC                  d+d,      |!jC                  d-d,      d.       |-S # 1 sw Y   xY w)/a  
    Solve the following linear programming problem using one of the HiGHS
    solvers:

    User-facing documentation is in _linprog_doc.py.

    Parameters
    ----------
    lp :  _LPProblem
        A ``scipy.optimize._linprog_util._LPProblem`` ``namedtuple``.
    solver : "ipm" or "simplex" or None
        Which HiGHS solver to use.  If ``None``, "simplex" will be used.

    Options
    -------
    maxiter : int
        The maximum number of iterations to perform in either phase. For
        ``solver='ipm'``, this does not include the number of crossover
        iterations.  Default is the largest possible value for an ``int``
        on the platform.
    disp : bool
        Set to ``True`` if indicators of optimization status are to be printed
        to the console each iteration; default ``False``.
    time_limit : float
        The maximum time in seconds allotted to solve the problem; default is
        the largest possible value for a ``double`` on the platform.
    presolve : bool
        Presolve attempts to identify trivial infeasibilities,
        identify trivial unboundedness, and simplify the problem before
        sending it to the main solver. It is generally recommended
        to keep the default setting ``True``; set to ``False`` if presolve is
        to be disabled.
    dual_feasibility_tolerance : double
        Dual feasibility tolerance.  Default is 1e-07.
        The minimum of this and ``primal_feasibility_tolerance``
        is used for the feasibility tolerance when ``solver='ipm'``.
    primal_feasibility_tolerance : double
        Primal feasibility tolerance.  Default is 1e-07.
        The minimum of this and ``dual_feasibility_tolerance``
        is used for the feasibility tolerance when ``solver='ipm'``.
    ipm_optimality_tolerance : double
        Optimality tolerance for ``solver='ipm'``.  Default is 1e-08.
        Minimum possible value is 1e-12 and must be smaller than the largest
        possible value for a ``double`` on the platform.
    simplex_dual_edge_weight_strategy : str (default: None)
        Strategy for simplex dual edge weights. The default, ``None``,
        automatically selects one of the following.

        ``'dantzig'`` uses Dantzig's original strategy of choosing the most
        negative reduced cost.

        ``'devex'`` uses the strategy described in [15]_.

        ``steepest`` uses the exact steepest edge strategy as described in
        [16]_.

        ``'steepest-devex'`` begins with the exact steepest edge strategy
        until the computation is too costly or inexact and then switches to
        the devex method.

        Currently, using ``None`` always selects ``'steepest-devex'``, but this
        may change as new options become available.

    mip_max_nodes : int
        The maximum number of nodes allotted to solve the problem; default is
        the largest possible value for a ``HighsInt`` on the platform.
        Ignored if not using the MIP solver.
    unknown_options : dict
        Optional arguments not used by this particular solver. If
        ``unknown_options`` is non-empty, a warning is issued listing all
        unused options.

    Returns
    -------
    sol : dict
        A dictionary consisting of the fields:

            x : 1D array
                The values of the decision variables that minimizes the
                objective function while satisfying the constraints.
            fun : float
                The optimal value of the objective function ``c @ x``.
            slack : 1D array
                The (nominally positive) values of the slack,
                ``b_ub - A_ub @ x``.
            con : 1D array
                The (nominally zero) residuals of the equality constraints,
                ``b_eq - A_eq @ x``.
            success : bool
                ``True`` when the algorithm succeeds in finding an optimal
                solution.
            status : int
                An integer representing the exit status of the algorithm.

                ``0`` : Optimization terminated successfully.

                ``1`` : Iteration or time limit reached.

                ``2`` : Problem appears to be infeasible.

                ``3`` : Problem appears to be unbounded.

                ``4`` : The HiGHS solver ran into a problem.

            message : str
                A string descriptor of the exit status of the algorithm.
            nit : int
                The total number of iterations performed.
                For ``solver='simplex'``, this includes iterations in all
                phases. For ``solver='ipm'``, this does not include
                crossover iterations.
            crossover_nit : int
                The number of primal/dual pushes performed during the
                crossover routine for ``solver='ipm'``.  This is ``0``
                for ``solver='simplex'``.
            ineqlin : OptimizeResult
                Solution and sensitivity information corresponding to the
                inequality constraints, `b_ub`. A dictionary consisting of the
                fields:

                residual : np.ndnarray
                    The (nominally positive) values of the slack variables,
                    ``b_ub - A_ub @ x``.  This quantity is also commonly
                    referred to as "slack".

                marginals : np.ndarray
                    The sensitivity (partial derivative) of the objective
                    function with respect to the right-hand side of the
                    inequality constraints, `b_ub`.

            eqlin : OptimizeResult
                Solution and sensitivity information corresponding to the
                equality constraints, `b_eq`.  A dictionary consisting of the
                fields:

                residual : np.ndarray
                    The (nominally zero) residuals of the equality constraints,
                    ``b_eq - A_eq @ x``.

                marginals : np.ndarray
                    The sensitivity (partial derivative) of the objective
                    function with respect to the right-hand side of the
                    equality constraints, `b_eq`.

            lower, upper : OptimizeResult
                Solution and sensitivity information corresponding to the
                lower and upper bounds on decision variables, `bounds`.

                residual : np.ndarray
                    The (nominally positive) values of the quantity
                    ``x - lb`` (lower) or ``ub - x`` (upper).

                marginals : np.ndarray
                    The sensitivity (partial derivative) of the objective
                    function with respect to the lower and upper
                    `bounds`.

            mip_node_count : int
                The number of subproblems or "nodes" solved by the MILP
                solver. Only present when `integrality` is not `None`.

            mip_dual_bound : float
                The MILP solver's final estimate of the lower bound on the
                optimal solution. Only present when `integrality` is not
                `None`.

            mip_gap : float
                The difference between the final objective function value
                and the final dual bound, scaled by the final objective
                function value. Only present when `integrality` is not
                `None`.

    Notes
    -----
    The result fields `ineqlin`, `eqlin`, `lower`, and `upper` all contain
    `marginals`, or partial derivatives of the objective function with respect
    to the right-hand side of each constraint. These partial derivatives are
    also referred to as "Lagrange multipliers", "dual values", and
    "shadow prices". The sign convention of `marginals` is opposite that
    of Lagrange multipliers produced by many nonlinear solvers.

    References
    ----------
    .. [15] Harris, Paula MJ. "Pivot selection methods of the Devex LP code."
            Mathematical programming 5.1 (1973): 1-28.
    .. [16] Goldfarb, Donald, and John Ker Reid. "A practicable steepest-edge
            simplex algorithm." Mathematical Programming 12.1 (1977): 361-371.
    zUnrecognized options detected: z). These will be passed to HiGHS verbatim.r'   rA   !simplex_dual_edge_weight_strategyN)dantzigdevexzsteepest-devexsteepestN)rO   r5   r6   presolvesensesolver
time_limithighs_debug_leveldual_feasibility_toleranceipm_optimality_tolerancelog_to_consolemip_max_nodesoutput_flagprimal_feasibility_tolerancesimplex_strategysimplex_crash_strategyipm_iteration_limitsimplex_iteration_limitmip_rel_gapr   slack)NNlambda	marg_bndsr   statusmessager<   )residual	marginalsfunsimplex_nitipm_nitcrossover_nit)r<   rh   conineqlineqlinrC   upperro   rk   successrl   nitrr   mip_node_countmip_dual_boundg        mip_gap)ry   rz   r{   )&r   r   rR   r   r   r   r   Tcopyr8   r:   	ones_likeinfconcatenater"   r!   r    r
   r	   r   r   updater>   sumemptyarrayr   indptrindicesdataastypeuint8lenr*   r2   r   r   any).lprZ   r[   rX   dispmaxiterr]   rb   r^   rT   rg   r`   unknown_optionsrl   &simplex_dual_edge_weight_strategy_enumcA_ubb_ubA_eqb_eqboundsx0integralitylbublhs_ubrhs_ublhs_eqrhs_eqlhsrhsAoptionsresrh   rs   lamdamarg_ineqlin
marg_eqlin
marg_upper
marg_lowerr+   r,   rk   r<   sols.                                                 r1   rH   rH   m   s   J 4_4E F= =Wo!4 .D)+FB#LA.* :<6AtT4vr;XX]]_FB	X	&,,t$$RVV+ 
'FFF
..&&)
*C
..&&)
*C~$D$< IItTl#1AH/ 	& 	j	
 	/ 	%&@ 	#$< 	$ 	 	t 	'(D 	,2 	7 	!"B  	w!" 	"7#$ 	{%G( NN?# s
C
s
C	b	B	b	Bbff[1Q6hhqkhh{+
AHHaiicR!3!3BHH!=wHC #~GhhuSYZ()zD	*+
s 3Hxxjs4y 12XXeCIJ/0
XXc+.q!t45
XXc+.q!t45
#- j!+
J
 778T*LGGIt,M4\5BDOFG "SjSdA$ (&  #&$  ##$94!b&&$  ##$94"q&&$  ''%.(m';;''-+Dswwy!/DGGO41C6 
vvay[,

!gg&6:!gg&6<wwy#.
 	 Je 
'	&s   =(Q++Q5)
NTFNNNNNNN)-__doc__rF   numpyr8   	_optimizer   r   warningsr   _highs._highs_wrapperr   _highs._highs_constantsr   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r)   r   r   r   r   r   r   r   r   r   scipy.sparser    r!   r"   r2   r>   rR   rH    r3   r1   <module>r      su      6  1      > 6 5':$" :>'+.204,059#!%Kr3   