"""

biogeme.segmentation
====================

Examples of use of several functions.

This is designed for programmers who need examples of use of the
functions of the module. The examples are designed to illustrate the
syntax. They do not correspond to any meaningful model.

:author: Michel Bierlaire
:date: Thu Dec  7 15:18:45 2023
"""

from biogeme.expressions import Beta, Variable
from biogeme.segmentation import DiscreteSegmentationTuple, Segmentation
from biogeme.version import get_text

# %%
# Version of Biogeme.
print(get_text())

# %%
socio_eco_1 = Variable('socio_eco_1')

# %%
segmentation_1 = DiscreteSegmentationTuple(
    variable=socio_eco_1,
    mapping={
        1: 'high',
        2: 'medium',
        3: 'low',
    },
)

# %%
socio_eco_2 = Variable('socio_eco_2')

# %%
segmentation_2 = DiscreteSegmentationTuple(
    variable=socio_eco_2,
    mapping={
        1: 'level_1',
        2: 'level_2',
        3: 'level_3',
        4: 'level_4',
    },
)

# %%
beta_x = Beta('beta_x', 0, None, None, 0)

# %%
segmented_parameter = Segmentation(beta_x, (segmentation_1,))

# %%
# The expressions for the segmented parameter is obtained as follows:
segmented_parameter.segmented_beta()

# %%
# The corresponding Python code can be obtained as follows.
print(segmented_parameter.segmented_code())

# %%
# The code of the original Beta is also available:
print(segmented_parameter.beta_ref_code())

# %%
# Same with the second segmentation
segmented_parameter = Segmentation(beta_x, (segmentation_2,))

# %%
segmented_parameter.segmented_beta()

# %%
print(segmented_parameter.segmented_code())

# %%
# The two segmentations can be combined.
segmented_parameter = Segmentation(
    beta_x,
    (
        segmentation_1,
        segmentation_2,
    ),
)

# %%
segmented_parameter.segmented_beta()

# %%
print(segmented_parameter.segmented_code())