WarpTwinPyConfigWriter.h

Go to the documentation of this file.

/******************************************************************************
 * Copyright (c) ATTX INC 2025. All Rights Reserved.
 *
 * This software and associated documentation (the "Software") are the
 * proprietary and confidential information of ATTX INC. 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 INC.
 ******************************************************************************/
#ifndef CPP_APP_MODELSPACE_PY_CONFIGURATION_WRITER_H
#define CPP_APP_MODELSPACE_PY_CONFIGURATION_WRITER_H
 
#include "ConfigurationWriter.h"
 
namespace warptwin {
// Define some handy constants to make life easy, and to make updating this for other stuff easy
namespace ms_py_writer {
const std::string LINE_ENDING = "\n";
const std::string CODE_ENDING = "";
const std::string TAB = "    ";
 
// Set the sequencing priority
const std::vector<node_types_e> SEQUENCE_PRIORITY = {FRAME, STANDARD, MODEL, LOGGER, VIZKIT, CESIUM_VIZ};
} // namespace ms_py_writer

  class WarpTwinPyConfigWriter : public ConfigurationWriter {
  public:
    WarpTwinPyConfigWriter() {}
    virtual ~WarpTwinPyConfigWriter() {}
        
    std::future<int> build();

    std::future<int> run();

    std::vector<int> stop(std::vector<std::string> &pids);

    int post();

    void resetRunStatus() {_is_run = false; }
 
  protected:
    int _runPythonSim(const std::string &cmd);

    std::vector<std::string> _getPythonScriptPIDs();

    std::vector<std::string> _getStreamlitPIDs();

    void _killStreamlitPIDs();

    virtual void _setNodeConnectionSequence() override;

    virtual void _decomposeNodeInformation() override;

    virtual void _writeOutputStart() override;

    virtual void _writeDataValue(ImNode *node) override;

    virtual void _writeDataConnection(Connection *upstream, Connection *downstream) override;

    virtual void _writeLogData() override;

    virtual void _writeOutputEnd() override;

    std::string _getSignalName(Connection *signal);

    ImNode *_getNodeByUid(int uid);

    Connection *_getConnectionByUid(int uid);

    std::vector<Connection *> _getConnectionsByName(ImNode *node, const std::string &name);

    std::vector<Connection *> _getConnectionsByType(ImNode *node, int type);

    bool _is_run;

    bool _is_build;
};
 
} // namespace warptwin
 
#endif