mirror of
https://git.intern.spaceteamaachen.de/ALPAKA/SPATZ.git
synced 2025-09-28 21:17:33 +00:00
SPATZ migration + proper directory structure
This commit is contained in:
22
spatz/sensors/compound.py
Normal file
22
spatz/sensors/compound.py
Normal file
@@ -0,0 +1,22 @@
|
||||
import numpy as np
|
||||
|
||||
from typing import List
|
||||
from numpy.typing import ArrayLike
|
||||
|
||||
from sensor import Sensor
|
||||
from spatz.dataset import Dataset, List
|
||||
from spatz.logger import Logger
|
||||
from spatz.transforms import Transform
|
||||
|
||||
|
||||
class CompoundSensor(Sensor):
|
||||
def __init__(self, dataset: Dataset, logger: Logger, sensors: List[Sensor], transforms: List[Transform] = []):
|
||||
super().__init__(dataset, logger, transforms)
|
||||
|
||||
self.__sensors = sensors
|
||||
|
||||
def _get_data(self) -> ArrayLike:
|
||||
x = np.stack([sensor() for sensor in self.__sensors])
|
||||
x = self._sensor_specific_effects(x)
|
||||
|
||||
return x
|
Reference in New Issue
Block a user