import numpy as np

from experiment.meteo_france_SCM_study.abstract_variable import AbstractVariable


class CrocusVariable(AbstractVariable):

    def __init__(self, variable_array):
        super().__init__(variable_array)

    @property
    def daily_time_serie_array(self) -> np.ndarray:
        return self.variable_array


class CrocusSweVariable(CrocusVariable):
    NAME = 'Snow Water Equivalent'
    UNIT = 'kg per m2 or mm'

    @classmethod
    def keyword(cls):
        return 'SWE_1DY_ISBA'


class CrocusDepthVariable(CrocusVariable):
    NAME = 'Snow Depth'
    UNIT = 'm'

    @classmethod
    def keyword(cls):
        return "SD_1DY_ISBA"