VisualsModel.h

Go to the documentation of this file.

/******************************************************************************
* Copyright (c) ATTX LLC 2024. All Rights Reserved.
*
* This software and associated documentation (the "Software") are the 
* proprietary and confidential information of ATTX, LLC. The Software is 
* furnished under a license agreement between ATTX and the user organization 
* and may be used or copied only in accordance with the terms of the agreement.
* Refer to 'license/attx_license.adoc' for standard license terms.
*
* EXPORT CONTROL NOTICE: THIS SOFTWARE MAY INCLUDE CONTENT CONTROLLED UNDER THE
* INTERNATIONAL TRAFFIC IN ARMS REGULATIONS (ITAR) OR THE EXPORT ADMINISTRATION 
* REGULATIONS (EAR99). No part of the Software may be used, reproduced, or 
* transmitted in any form or by any means, for any purpose, without the express 
* written permission of ATTX, LLC.
******************************************************************************/
/*
Visuals model header file
 
Author: Alex Reynolds
*/
 
#ifndef SIMULATION_VISUALS_MODEL_H
#define SIMULATION_VISUALS_MODEL_H
 
#include <string>
#include <map>
#include <set>
#include <fstream>
 
#include "simulation/SimulationSteps.h"
#include "frames/Frame.h"
#include "dynamics/Quaternion.h"
#include "locations.h"
#include "flight/App.h"
#include "core/mathmacros.h"
 
namespace modelspace {
 
    class RateMonitor;
    class SimulationExecutive;
    class GroundStationModel;
 
    class VisualsModel : public cfspp::App { 
    public:
        // Model params
        //         NAME                     TYPE                    DEFAULT VALUE
        START_PARAMS
            SIGNAL(planet_frame,            Frame*,                 nullptr)
            SIGNAL(sc_rate,                 clockwerk::Time,        clockwerk::Time(100, 0))
            SIGNAL(gs_rate,                 clockwerk::Time,        clockwerk::Time(1, 0))
            SIGNAL(lead_time,               clockwerk::Time,        clockwerk::Time(5400, 0))
            SIGNAL(trail_time,              clockwerk::Time,        clockwerk::Time(5400, 0))
            SIGNAL(run_cesium_app,          int,                    1)
            SIGNAL(max_range,               double,                 10000.0)
            SIGNAL(max_range_rate,          double,                 1000.0)
        END_PARAMS
 
        // Model inputs
        //         NAME                     TYPE                    DEFAULT VALUE
        START_INPUTS
 
        END_INPUTS
 
        // Model outputs
        //         NAME                     TYPE                    DEFAULT VALUE
        START_OUTPUTS 
 
        END_OUTPUTS
 
        int addFrame(Frame* new_frame_ptr, const std::string& visual_model_dir = "");
 
        int addGroundStation(clockwerk::GraphTreeObject* new_gs_ptr);
 
        int addLink(const std::string& link_name, Frame* primary_sc_ptr, Frame* secondary_sc_ptr);
 
        int addSecondarySpacecraft(const std::string& link_name, Frame* secondary_sc_ptr);
        
        // Model-specific implementations of startup and derivative
        VisualsModel(cfspp::FlightExecutive &e, const std::string &m_name="visuals_model");
        virtual ~VisualsModel();
 
        void forceWrite() {_writeCzmlFile();}
    protected:
        int16 start() override;
        int16 execute() override;
 
        struct RelMotion {
            CartesianVector3 rel_position;
            CartesianVector3 rel_velocity;
            double range;
            double range_rate;
        };
 
        RelMotion computeRelativeMotion(Frame* primary_ptr, Frame* secondary_ptr);
 
        bool _data_written = false;
 
        void _runAppProcess();
 
        std::string _slurpFile(std::ifstream& in);
 
        void _writeCzmlFile();
 
        SimulationExecutive* _smex;
 
        std::ofstream _fout;
 
        std::string _sim_start_time;
 
        // Set of vectors to log tracked spacecraft frames, spacecraft models, spacecraft times, and spacecraft states at those times
        std::vector<Frame*> _sc_frames;
        std::vector<std::string> _sc_frame_models;
        std::vector<double> _sc_times;
        std::vector<std::vector<CartesianVector3>> _sc_positions;
        std::vector<std::vector<clockwerk::Quaternion>> _sc_attitudes;
        CartesianVector3 _pos_f_planet_frame;
        RateMonitor* _sc_rate_monitor;                           // Controls the rate of SC logging
 
        // Set of vectors to log ground station states
        std::vector<GroundStationModel*> _gs_models;
        std::vector<std::vector<std::string>> _gs_sign_times_et; // Vector of times when GS visibility changed. Must include zero time
        std::vector<std::vector<bool>> _gs_sign_after_time;      // Vector of GS visibility signs after corresponding entry in _gs_sign_times_et
                                                    // with true = visible and false = not visible
        RateMonitor* _gs_rate_monitor;               // Controls the rate of GS logging
 
        // Set of vectors to log linkages between spacecraft
        std::vector<std::string> _link_names;
        std::vector<Frame*> _primary_frames;
        std::map<std::string, std::vector<Frame*>> _secondary_frames;
        std::map<std::string, std::map<Frame*, std::vector<std::string>>> _sc_link_times_et_secondary;
        std::map<std::string, std::map<Frame*, std::vector<bool>>> _sc_link_after_time_secondary;
    };
}
 
#endif