noisegenerator.py

This class provides a 'as-simple-as' possible component, which just creates one source field, that is filled with random values in each timestep.

#!/usr/bin/env python3
 
# Copyright (c) 2024 The YAC Authors
#
# SPDX-License-Identifier: BSD-3-Clause
 
from yac import *
import numpy as np
 
class NoiseGenerator:
    def __init__(self, timestep, yac = None):
        self.yac = yac or YAC.default_instance
        self.comp = self.yac.predef_comp("noisegenerator")
        self.timestep = timestep
 
    def setup(self):
        global grid, noise_field
        x = np.linspace(0,2*np.pi,360)
        y = np.linspace(-0.5*np.pi,0.5*np.pi,180)
        grid = Reg2dGrid(f"noise_grid", x, y)
        points = grid.def_points(Location.CORNER, x, y)
 
        noise_field = Field.create("noise", self.comp, points, 1,
                                   self.timestep, TimeUnit.ISO_FORMAT)
 
    def def_couples(self):
        pass
 
    def step(self):
        print(f"NoiseGenerator: Lets make some noise!!!")
        noise_field.put(np.random.rand(grid.nbr_corners, 1))
        return noise_field.datetime