From 8290f0783232fe0e91ab134cd6e5b38ec50e5b40 Mon Sep 17 00:00:00 2001 From: John ZuHone Date: Mon, 5 Sep 2022 10:38:24 -0400 Subject: [PATCH] Initial groundwork for calling a model within a model --- xija/component/base.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/xija/component/base.py b/xija/component/base.py index 64ec37b..73000e2 100644 --- a/xija/component/base.py +++ b/xija/component/base.py @@ -330,6 +330,24 @@ def plot_resid__data(self, fig, ax): lines[0].set_ydata(resids) +class ModeledData(ModelComponent): + def __init__(self, model, name, modeled_msid, model_spec): + from xija.model import XijaModel + super().__init__(model) + self.modeled_msid = modeled_msid + self.model2 = XijaModel(name, start=self.model.datestart, + stop=self.model.datestop, + model_spec=model_spec) + self.model2.make() + self.model2.calc() + + def get_dvals_tlm(self): + return self.model2.comp[self.modeled_msid].mvals + + def __str__(self): + return self.modeled_msid + + class Coupling(ModelComponent): """\ First-order coupling between Nodes `node1` and `node2`