Issue #85: Set initial storage level

AVSLab · Nov 28, 2023 · ff2f976 · ff2f976
1 parent 43417d3
commit ff2f976
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bsk_rl/envs/general_satellite_tasking/simulation/dynamics.py b/bsk_rl/envs/general_satellite_tasking/simulation/dynamics.py
@@ -841,12 +841,15 @@ def _set_instrument(
             self.task_name, self.instrument, ModelPriority=priority
         )
 
-    @default_args(dataStorageCapacity=20 * 8e6, storageUnitValidCheck=True)
+    @default_args(
+        dataStorageCapacity=20 * 8e6, storageUnitValidCheck=True, storageInit=0
+    )
     def _set_storage_unit(
         self,
         dataStorageCapacity: int,
         priority: int = 699,
         storageUnitValidCheck: bool = True,
+        storageInit: int = 0,
         **kwargs,
     ) -> None:
         """Configure the storage unit and its buffers.
@@ -856,13 +859,15 @@ def _set_storage_unit(
             priority: Model priority.
             storageUnitValidCheck: If True, check that the storage level is below the
                 storage capacity.
+            setStorageInit: Initial storage level [bits]
         """
         self.storageUnit = simpleStorageUnit.SimpleStorageUnit()
         self.storageUnit.ModelTag = "storageUnit" + self.satellite.id
         self.storageUnit.storageCapacity = dataStorageCapacity  # bits
         self.storageUnit.addDataNodeToModel(self.instrument.nodeDataOutMsg)
         self.storageUnit.addDataNodeToModel(self.transmitter.nodeDataOutMsg)
         self.storageUnitValidCheck = storageUnitValidCheck
+        self.storageUnit.setDataBuffer(storageInit)
 
         # Add the storage unit to the transmitter
         self.transmitter.addStorageUnitToTransmitter(