Source code for discrete_optimization.generic_tools.hub_solver.optal.model_collections

#  Copyright (c) 2025 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.
import os
from enum import Enum




class DoProblemEnum(Enum):
    JSP = 0
    FJSP = 1
    RCPSP = 2
    MRCPSP = 3
    MSRCPSP = 4
    SINGLEMACHINE = 5
    TSP = 6
    TSPTW = 7
    WORKFORCE_SCHEDULING = 8



this_folder = os.path.abspath(os.path.dirname(__file__))
models_folder = os.path.join(this_folder, "models/")
# This will be updated when new model enrich our collection
problem_to_script_path = {
    DoProblemEnum.JSP: os.path.join(models_folder, "jsp/jobshop.mts"),
    DoProblemEnum.FJSP: os.path.join(models_folder, "fjsp/flexible-jobshop.mts"),
    DoProblemEnum.RCPSP: os.path.join(models_folder, "rcpsp/rcpsp.mts"),
    DoProblemEnum.MRCPSP: os.path.join(models_folder, "rcpsp/mrcpsp.mts"),
    DoProblemEnum.MSRCPSP: os.path.join(models_folder, "msrcpsp/msrcpsp.mts"),
    DoProblemEnum.SINGLEMACHINE: os.path.join(
        models_folder, "single_machine/single_machine.mts"
    ),
    DoProblemEnum.TSP: os.path.join(models_folder, "tsp/tsp.mts"),
    DoProblemEnum.TSPTW: os.path.join(models_folder, "tsptw/tsptw.mts"),
    DoProblemEnum.WORKFORCE_SCHEDULING: os.path.join(
        models_folder, "workforce/workforce_scheduling.mts"
    ),
}