Source code for scenario_BasicOrbitFormation

#
#  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"""
Overview
--------

This script sets up two 3-DOF spacecraft orbiting the Earth. The goal of the scenario is to

#. highlight how the BSK_Sim structure of a formation flying tutorial is different from the basic orbit scenario,
#. demonstrate how to create a formation flying scenario, and
#. how to customize the :ref:`BSK_FormationDynamics.py <BSK_FormationDynamics>` and :ref:`BSK_FormationFSW.py <BSK_FormationFSW>` files.

The script is found in the folder ``basilisk/examples/BskSim/scenarios`` and executed by using::

      python3 scenario_BasicOrbitFormation.py

The simulation mimics the basic simulation in the earlier tutorial in :ref:`scenario_BasicOrbit`.

The flight software mode is set to inertial3D. The goal of this mode is to align the body axes of the spacecraft
with an inertial 3D point guidance coordinate system defined in :ref:`BSK_FormationFSW`.
However the flight software mode can also be set to "standby".

The simulation layout is shown in the following illustrations.

.. image:: /_images/static/test_scenario_BasicOrbitFormation.svg
   :align: center

.. image:: /_images/static/Simulation_container_layout_BasicOrbitFormation.svg
   :align: center


Configuring the scenario file
-----------------------------
The simulation layout is very similar to the one used for the :ref:`scenario_BasicOrbit` file.
Two simulation processes are created: one which contains dynamics modules, and one that contains
the Flight Software (FSW) modules. First of all, it can be observed that the Dynamics- and FSW files used are
the :ref:`BSK_FormationDynamics` and :ref:`BSK_FormationFSW` files.
These two files have been created for this specific formation flying implementation into Basilisk.

After initializing the interfaces and making sure that the :ref:`scenario_BasicOrbitFormation`
class inherits from the BSKSim class,
it is time to configure the initial conditions using the ``configure_initial_conditions`` method.
It can be observed that two sets of
orbital elements are created. Each set corresponding to one spacecraft. After that the initial
conditions are set for each spacecraft.

After that the function that logs the outputs can be observed. Again this looks very similar to
the log_outputs method
in the :ref:`scenario_BasicOrbit` file, however one discrepancy can be noticed. Looking at
the code below it can be observed that
two instances of the ``simpleNavObject`` are logged (``simpleNavObject`` and ``simpleNavObject2`` respectively).
Each object corresponds
two one of the spacecraft. The same is true for the FSW objects. More on this will be discussed later.

The same is true for the ``pull_outputs`` method. Also in this function, it can be observed
that the outputs of two instances
of a specific object are pulled.



BSK_FormationDynamics file description
--------------------------------------
Looking at the :ref:`BSK_FormationDynamics` file, it can be observed that the dynamics
process consists of two tasks named DynamicsTask
and ``DynamicsTask2`` respectively. These tasks are added to the dynamics process and to each task,
an instance of a specific object is added.

The gravity body (Earth in this case) is created using the ``gravBodyFactory`` and is attached
as a separate object to each spacecraft.


After that each object is added to the corresponding task. Something that is very important is the message names.
In case multiple spacecraft are implemented in Basilisk it is necessary to manually connect an output message of
one module to the input of a different module. This can be seen in the module-initialization methods
in the :ref:`BSK_FormationDynamics.py <BSK_FormationDynamics>` file.

BSK_FormationFsw file description
---------------------------------
The setup of the FSW file (:ref:`BSK_FormationFSW.py <BSK_FormationFSW>`) in case of
formation flying is very similar to the setup of the dynamics file.
Also in this case, an instance of each task is initialized that corresponds to
one of the two spacecraft. Furthermore, it is
necessary to manually set the input- and output message names for the FSW
modules. In order to make this tutorial work properly its
is very important to set the ``self.mrpFeedbackRWs.Ki`` and ``self.mrpFeedbackRWs2.Ki``
variables in :ref:`BSK_FormationFsw` to -1. Otherwise
the orientation and rates of both spacecraft will not converge!


Illustration of Simulation Results
----------------------------------

::

    showPlots = True

.. image:: /_images/Scenarios/scenario_BasicOrbitFormation_attitude_error_chief.svg
   :align: center

.. image:: /_images/Scenarios/scenario_BasicOrbitFormation_rate_error_chief.svg
   :align: center

.. image:: /_images/Scenarios/scenario_BasicOrbitFormation_attitude_error_deputy.svg
   :align: center

.. image:: /_images/Scenarios/scenario_BasicOrbitFormation_rate_error_deputy.svg
   :align: center

.. image:: /_images/Scenarios/scenario_BasicOrbitFormation_orbits.svg
   :align: center

"""

# Import utilities
from Basilisk.utilities import orbitalMotion, macros, vizSupport

# Get current file path
import sys, os, inspect
filename = inspect.getframeinfo(inspect.currentframe()).filename
path = os.path.dirname(os.path.abspath(filename))

# Import master classes: simulation base class and scenario base class
sys.path.append(path + '/..')
from BSK_masters import BSKSim, BSKScenario
import BSK_FormationDynamics, BSK_FormationFsw

# Import plotting files for your scenario
sys.path.append(path + '/../plotting')
import BSK_Plotting as BSK_plt

sys.path.append(path + '/../../scenarios')



# Create your own scenario child class
[docs]class scenario_BasicOrbitFormation(BSKSim, BSKScenario): def __init__(self): super(scenario_BasicOrbitFormation, self).__init__() self.name = 'scenario_BasicOrbitFormation' # declare empty class variables self.sNavTransRec = None self.sNavTrans2Rec = None self.attErrRec = None self.attErr2Rec = None self.scStateRec = None self.scState2Rec = None self.set_DynModel(BSK_FormationDynamics) self.set_FswModel(BSK_FormationFsw) self.configure_initial_conditions() self.log_outputs() # if this scenario is to interface with the BSK Viz, uncomment the following line if vizSupport.vizFound: viz = vizSupport.enableUnityVisualization(self, self.DynModels.taskName , [self.get_DynModel().scObject, self.get_DynModel().scObject2] , rwEffectorList=[self.DynModels.rwStateEffector, self.DynModels.rwStateEffector2] # , saveFile=__file__ )
[docs] def configure_initial_conditions(self): self.mu = self.get_DynModel().gravFactory.gravBodies['earth'].mu # Configure Dynamics initial conditions self.oe = orbitalMotion.ClassicElements() self.oe.a = 10000000.0 # meters self.oe.e = 0.01 self.oe.i = 33.3 * macros.D2R self.oe.Omega = 48.2 * macros.D2R self.oe.omega = 347.8 * macros.D2R self.oe.f = 85.3 * macros.D2R rN, vN = orbitalMotion.elem2rv(self.mu, self.oe) orbitalMotion.rv2elem(self.mu, rN, vN) self.get_DynModel().scObject.hub.r_CN_NInit = rN # m - r_CN_N self.get_DynModel().scObject.hub.v_CN_NInit = vN # m/s - v_CN_N self.get_DynModel().scObject.hub.sigma_BNInit = [[0.1], [0.2], [-0.3]] # sigma_BN_B self.get_DynModel().scObject.hub.omega_BN_BInit = [[0.001], [-0.01], [0.03]] # rad/s - omega_BN_B # Configure Dynamics initial conditions self.oe2 = orbitalMotion.ClassicElements() self.oe2.a = 12000000.0 # meters self.oe2.e = 0.1 self.oe2.i = 33.3 * macros.D2R self.oe2.Omega = 48.2 * macros.D2R self.oe2.omega = 347.8 * macros.D2R self.oe2.f = 85.3 * macros.D2R rN2, vN2 = orbitalMotion.elem2rv(self.mu, self.oe2) orbitalMotion.rv2elem(self.mu, rN2, vN2) self.get_DynModel().scObject2.hub.r_CN_NInit = rN2 # m - r_CN_N self.get_DynModel().scObject2.hub.v_CN_NInit = vN2 # m/s - v_CN_N self.get_DynModel().scObject2.hub.sigma_BNInit = [[-0.3], [0.0], [0.5]] # sigma_BN_B self.get_DynModel().scObject2.hub.omega_BN_BInit = [[0.003], [-0.02], [0.01]] # rad/s - omega_BN_B
[docs] def log_outputs(self): samplingTime = self.get_DynModel().processTasksTimeStep DynModels = self.get_DynModel() FswModel = self.get_FswModel() self.sNavTransRec = DynModels.simpleNavObject.transOutMsg.recorder(samplingTime) self.sNavTrans2Rec = DynModels.simpleNavObject2.transOutMsg.recorder(samplingTime) self.attErrRec = FswModel.attGuidMsg.recorder(samplingTime) self.attErr2Rec = FswModel.attGuid2Msg.recorder(samplingTime) self.scStateRec = DynModels.scObject.scStateOutMsg.recorder(samplingTime) self.scState2Rec = DynModels.scObject2.scStateOutMsg.recorder(samplingTime) self.AddModelToTask(DynModels.taskName, self.sNavTransRec) self.AddModelToTask(DynModels.taskName2, self.sNavTrans2Rec) self.AddModelToTask(DynModels.taskName, self.attErrRec) self.AddModelToTask(DynModels.taskName2, self.attErr2Rec) self.AddModelToTask(DynModels.taskName, self.scStateRec) self.AddModelToTask(DynModels.taskName2, self.scState2Rec)
[docs] def pull_outputs(self, showPlots): # Dynamics process outputs r_BN_N_chief = self.sNavTransRec.r_BN_N r_BN_N_deputy = self.sNavTrans2Rec.r_BN_N v_BN_N_chief = self.sNavTransRec.v_BN_N v_BN_N_deputy = self.sNavTrans2Rec.v_BN_N # FSW process outputs omega_BR_B_chief = self.attErrRec.omega_BR_B omega_BR_B_deputy = self.attErr2Rec.omega_BR_B sigma_BR_chief = self.attErrRec.sigma_BR sigma_BR_deputy = self.attErr2Rec.sigma_BR # Plot results BSK_plt.clear_all_plots() timeData = self.sNavTransRec.times() * macros.NANO2MIN BSK_plt.plot_attitude_error(timeData, sigma_BR_chief) BSK_plt.plot_rate_error(timeData, omega_BR_B_chief) BSK_plt.plot_attitude_error(timeData, sigma_BR_deputy) BSK_plt.plot_rate_error(timeData, omega_BR_B_deputy) BSK_plt.plot_peri_and_orbit(self.oe, self.mu, r_BN_N_chief, v_BN_N_chief) BSK_plt.plot_peri_and_orbit(self.oe2, self.mu, r_BN_N_deputy, v_BN_N_deputy) figureList = {} if showPlots: BSK_plt.show_all_plots() else: fileName = os.path.basename(os.path.splitext(__file__)[0]) figureNames = ["attitude_error_chief", "rate_error_chief", "attitude_error_deputy", "rate_error_deputy", "orbits"] figureList = BSK_plt.save_all_plots(fileName, figureNames) return figureList
def runScenario(scenario): scenario.InitializeSimulation() # Configure FSW mode scenario.modeRequest = 'inertial3D' # Configure run time and execute simulation simulationTime = macros.min2nano(10.) scenario.ConfigureStopTime(simulationTime) scenario.ExecuteSimulation()
[docs]def run(showPlots): """ The scenarios can be run with the followings setups parameters: Args: showPlots (bool): Determines if the script should display plots """ TheScenario = scenario_BasicOrbitFormation() runScenario(TheScenario) figureList = TheScenario.pull_outputs(showPlots) return figureList
if __name__ == "__main__": run(True)