#
# ISC License
#
# Copyright (c) 2016, Autonomous Vehicle Systems Lab, University of Colorado at Boulder
#
# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
#
r"""
This script is a basic demonstration of a script that can be used to rerun a set or subset of Monte Carlo simulations.
.. important::
This script can only be run once there exists data produced by the ``scenarioBskSimAttFeedbackMC.py`` script.
"""
import inspect
import os
import sys
import importlib
from Basilisk.utilities.MonteCarlo.Controller import Controller
from Basilisk.utilities.MonteCarlo.RetentionPolicy import RetentionPolicy
filename = inspect.getframeinfo(inspect.currentframe()).filename
fileNameString = os.path.basename(os.path.splitext(__file__)[0])
path = os.path.dirname(os.path.abspath(filename))
from Basilisk import __path__
bskPath = __path__[0]
sys.path.append(path+"/../BskSim/scenarios/")
[docs]
def run(time=None):
"""
Instructions:
1) Change the scenario name
2) Provide the number of processes to spawn
3) Provide the run numbers you wish to rerun
4) Add any new retention policies to the bottom
"""
# Step 1-3: Change to the relevant scenario
scenarioName = "scenario_AttFeedback" # This is the actual scenario module name
mcName = "scenarioBskSimAttFeedbackMC" # This is the MC script name
monteCarlo = Controller()
monteCarlo.numProcess = 3
runsList = [1]
# Generic initialization
icName = path + "/" + mcName # Use MC script name for directory
newDataDir = path + "/" + mcName + "/rerun"
# Use importlib to import the scenario module
scenarioModule = importlib.import_module(scenarioName)
# Access the class and methods dynamically using getattr()
simulationModule = getattr(scenarioModule, scenarioName)
if time is not None:
simulationModule.simBaseTime = time
executionModule = getattr(scenarioModule, "runScenario")
monteCarlo.setSimulationFunction(simulationModule)
monteCarlo.setExecutionFunction(executionModule)
monteCarlo.setICDir(icName)
monteCarlo.setICRunFlag(True)
monteCarlo.setArchiveDir(newDataDir)
monteCarlo.setExecutionCount(len(runsList))
monteCarlo.setShouldDisperseSeeds(False)
monteCarlo.shouldArchiveParameters = False
# Step 4: Add any additional retention policies desired
retentionPolicy = RetentionPolicy()
retentionPolicy.logRate = int(2E9)
retentionPolicy.addMessageLog("attGuidMsg", ["sigma_BR"])
monteCarlo.addRetentionPolicy(retentionPolicy)
failed = monteCarlo.runInitialConditions(runsList)
assert len(failed) == 0, "Should run ICs successfully"
if __name__ == "__main__":
run()