import numpy as np
from pom.constants import vertical_layers


class BfmStateVariableData:
    def __init__(self, current=np.zeros(vertical_layers), forward=np.zeros(vertical_layers), backward=np.zeros(vertical_layers),
                 surface_value=0, surface_flux=0, bottom_flux=0):
        self.current = current
        self.forward = forward
        self.backward = backward
        self.surface_value = surface_value
        self.surface_flux = surface_flux
        self.bottom_flux = bottom_flux


class NutrientData:
    def __init__(self,NO3surf,NH4surf,PO4surf,SIO4surf,O2bott,NO3bott,PO4bott,PONbott_grad):
        self.NO3surf = NO3surf
        self.NH4surf = NH4surf
        self.PO4surf = PO4surf
        self.SIO4surf = SIO4surf
        self.O2bott = O2bott
        self.NO3bott = NO3bott
        self.PO4bott = PO4bott
        self.PONbott_grad = PONbott_grad
