Source code for discrete_optimization.generic_tasks_tools.solvers.cpsat.objectives.calendar_preempted

#  Copyright (c) 2026 AIRBUS and its affiliates.
#  This source code is licensed under the MIT license found in the
#  LICENSE file in the root directory of this source tree.
from __future__ import annotations

from typing import TYPE_CHECKING

from ortools.sat.python.cp_model import LinearExpr

if TYPE_CHECKING:
    from discrete_optimization.generic_tasks_tools.solvers.cpsat.auto import (
        GenericSchedulingAutoCpSatSolver,
    )
from discrete_optimization.generic_tasks_tools.objectives.calendar_preempted import (
    CalendarPreemptedComputer,
)
from discrete_optimization.generic_tasks_tools.solvers.cpsat.objectives.objective_modeler import (
    ObjectiveModelerCpSat,
)



[docs]
class CalendarPreemptedModelerCpSat(ObjectiveModelerCpSat):
    objective_computer: CalendarPreemptedComputer
    solver: GenericSchedulingAutoCpSatSolver


[docs]
    def get_objective_expr(self) -> LinearExpr:
        return self.solver.compute_nb_preempted_tasks()