julienmalard/Tikon

from tikon.central import Módulo, SimulMódulo, Modelo, Exper, Parcela, Coso
from tikon.central.res import Resultado
from tikon.ecs import ÁrbolEcs, CategEc, Ecuación, SubcategEc, EcuaciónVacía

valor = 6.02214e23


class EcuaciónObtVal(Ecuación):
    nombre = 'obt val'
    eje_cosos = 'coso'

    def eval(símismo, paso, sim):
        val = símismo.obt_valor_extern(sim, 'res 2', 'otro módulo')
        símismo.poner_valor_mód(sim, 'res', val)


class SubCategObtVal(SubcategEc):
    nombre = 'subcateg'
    cls_ramas = [EcuaciónObtVal, EcuaciónVacía]
    eje_cosos = 'coso'


class CategObtVal(CategEc):
    nombre = 'categ'
    cls_ramas = [SubCategObtVal]
    eje_cosos = 'coso'


class EcsObtVal(ÁrbolEcs):
    nombre = 'Ecs obt val'
    cls_ramas = [CategObtVal]


class CosoObtValor(Coso):
    def __init__(símismo, nombre):
        super().__init__(nombre, EcsObtVal)


class ResEjeCoso(Resultado):
    unids = None

    def __init__(símismo, sim, coords, vars_interés):
        coords = {'coso': sim.ecs.cosos, **coords}
        super().__init__(sim=sim, coords=coords, vars_interés=vars_interés)

    @property
    def nombre(símismo):
        raise NotImplementedError


class Res(ResEjeCoso):
    nombre = 'res'


class Res2(Resultado):
    nombre = 'res 2'
    unids = None


class SimulMóduloObtVal(SimulMódulo):
    resultados = [Res]


coso = CosoObtValor('hola')


class MóduloObtVal(Módulo):
    nombre = 'módulo'
    cls_simul = SimulMóduloObtVal

    def __init__(símismo, cosos=None):
        símismo.l_cosos = cosos
        super().__init__(cosos)

    def gen_ecs(símismo, modelo, mód, exper, n_reps):
        return EcsObtVal(modelo, mód, exper, cosos=símismo.l_cosos, n_reps=n_reps)


class SimulOtroMódulo(SimulMódulo):
    resultados = [Res2]

    def incrementar(símismo, paso, f):
        super().incrementar(paso, f)
        símismo.poner_valor('res 2', valor)


class OtroMódulo(Módulo):
    nombre = 'otro módulo'
    cls_simul = SimulOtroMódulo


mi_exper = Exper('exper', Parcela('parcela'))
mi_modelo = Modelo([MóduloObtVal([coso]), OtroMódulo])