from __future__ import annotations
from typing import TYPE_CHECKING
import numpy as np
import xarray as xr
from glotaran.model import DatasetModel
from glotaran.model import Megacomplex
from glotaran.model import megacomplex
if TYPE_CHECKING:
from glotaran.typing.types import ArrayLike
[docs]
@megacomplex(unique=True)
class BaselineMegacomplex(Megacomplex):
type: str = "baseline"
[docs]
def calculate_matrix(
self,
dataset_model: DatasetModel,
global_axis: ArrayLike,
model_axis: ArrayLike,
**kwargs,
):
clp_label = [f"{dataset_model.label}_baseline"]
matrix = np.ones((model_axis.size, 1), dtype=np.float64)
return clp_label, matrix
[docs]
def finalize_data(
self,
dataset_model: DatasetModel,
dataset: xr.Dataset,
is_full_model: bool = False,
as_global: bool = False,
):
if not is_full_model:
dataset["baseline"] = dataset.clp.sel(clp_label=f"{dataset_model.label}_baseline")