PythonFMU3/examples/arraytypes.py at 5ce50a3d833ff4d40b105bfec96483dc986784f9 · StephenSmith25/PythonFMU3 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

from pythonfmu3 import Fmi3Causality, Fmi3Variability, Dimension, Fmi3Slave, Fmi3Status, Float64, Int32, Int64, UInt64, String, Boolean

import numpy as np

SIZE = 10

TYPES = [UInt64, Float64, Boolean, Int32, Int64]
CAUSALITY = [Fmi3Causality.output, Fmi3Causality.input]

TYPE_MAP = {
    UInt64: np.uint64,
    Float64: np.float64,
    Int32: np.int32,
    Int64: np.int64,
    Boolean: bool
}

def var_names(var_type, causality):
    return f"{var_type.__name__.lower()}_{causality.name.lower()}"


def init_var(var_type, causality, array=True):
    if array:
        return np.zeros(SIZE, dtype=TYPE_MAP[var_type])
    else:
        return TYPE_MAP[var_type]()


def create_vars(self):
    dimensions = [Dimension(start=str(SIZE))]
    for var_type in TYPES:
      for causality in CAUSALITY:
        name = var_names(var_type, causality)
        if var_type == Float64:
            var = var_type(name, causality=causality, variability=Fmi3Variability.continuous, dimensions=dimensions)
        else:
            var = var_type(name, causality=causality, variability=Fmi3Variability.discrete, dimensions=dimensions)
        setattr(self, name, init_var(var_type, causality))
        self.register_variable(var)


def generate_random_data(self):
    for var_type in TYPES:
      causality = Fmi3Causality.output
      name = var_names(var_type, causality)
      var = getattr(self, name)
      if var_type == Boolean:
          setattr(self, name, np.random.choice(a=[False, True], size=SIZE).astype(bool))
      elif var_type == Int32 or var_type == Int64 or var_type == UInt64:
          setattr(self, name, np.random.randint(0, 100, size=SIZE).astype(TYPE_MAP[var_type]))
      else:
          setattr(self, name, np.random.rand(SIZE).astype(TYPE_MAP[var_type]))

class ArrayTypes(Fmi3Slave):

  def __init__(self, **kwargs):
      super().__init__(**kwargs)

      self.author = "Stephen Smith"
      self.description = "All types example"

      self.time = 0.0

      self.register_variable(Float64("time", causality=Fmi3Causality.independent, variability=Fmi3Variability.continuous))

      create_vars(self)


  def do_step(self, current_time: float, step_size: float) -> Fmi3Status:
      generate_random_data(self)
      return True