Erdnaxela3/bioptim_gui

from typing import Optional, Union

from bioptim import QuadratureRule, Node
from pydantic import BaseModel

from bioptim_gui_api.penalty.misc.enums import ObjectiveType
from bioptim_gui_api.variables.misc.enums import Dynamics, InterpolationType


class NbPhasesRequest(BaseModel):
    nb_phases: int


class NbShootingPointsRequest(BaseModel):
    nb_shooting_points: int


class DynamicsRequest(BaseModel):
    dynamics: Dynamics


class PhaseDurationRequest(BaseModel):
    duration: float


class NodesRequest(BaseModel):
    nodes: Node


class QuadraticRequest(BaseModel):
    quadratic: bool


class ExpandRequest(BaseModel):
    expand: bool


class TargetRequest(BaseModel):
    target: Optional[list] = None


class DerivativeRequest(BaseModel):
    derivative: bool


class IntegrationRuleRequest(BaseModel):
    integration_rule: QuadratureRule


class MultiThreadRequest(BaseModel):
    multi_thread: bool


class WeightRequest(BaseModel):
    weight: float


class ObjectiveTypeRequest(BaseModel):
    objective_type: ObjectiveType


class PenaltyTypeRequest(BaseModel):
    penalty_type: str


class ConstraintFcnRequest(BaseModel):
    penalty_type: str


class VariableUpdateRequest(BaseModel):
    x: int
    y: int
    value: float


class ObjectiveFcnRequest(BaseModel):
    penalty_type: str


class BoundsInterpolationTypeRequest(BaseModel):
    bounds_interpolation_type: InterpolationType


class InitialGuessInterpolationTypeRequest(BaseModel):
    initial_guess_interpolation_type: InterpolationType


class DimensionRequest(BaseModel):
    dimension: int


class ArgumentRequest(BaseModel):
    type: str
    value: Union[float | int | str | list | None]