BlockDynamicSystem.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "BlockAstro.hpp"
#include "AstCore/BlockSystem.hpp"
#include "AstCore/BlockDerivative.hpp"
#include "AstCore/SimTime.hpp"
#include <unordered_map>
 
AST_NAMESPACE_BEGIN
 
class BlockDerivative;
 
class AST_CORE_API BlockDynamicSystem : public BlockSystem
{
public:
    BlockDynamicSystem() = default;
    ~BlockDynamicSystem() = default;
 
    size_t size() const { return state_.size(); }
 
    int getDimension() const { return static_cast<int>(this->size()); }
 
    const double* getStateData() const { return state_.data(); }
 
    const double* getDerivativeData() const { return derivative_.data(); }
 
    size_t getStateIndex(Identifier* id) const;
    size_t getStateIndex(StringView name) const;
 
    void setStateData(const double* y) { state_.assign(y, y + this->size()); }
 
    #ifdef _MSC_VER
    #pragma warning(push, 0)
    #endif
 
    void getDerivativeData(double* dydt) const { std::copy_n(derivative_.begin(), derivative_.size(), dydt); }
 
    void fillDerivativeData(double value) { std::fill_n(derivative_.begin(), derivative_.size(), value); }
 
    void fillAccumulateData(double value) { std::fill_n(accumulate_.begin(), accumulate_.size(), value); }
    
    #ifdef _MSC_VER
    #pragma warning(pop)
    #endif
 
    errc_t initialize();
 
    errc_t sortBlocks();
 
    errc_t createStateMap();
 
    errc_t connectSignalsByNames();
 
    void addBlock(BlockDerivative* block);
 
    void addDerivativeBlock(BlockDerivative* block);
 
    using BlockSystem::addBlock;
 
    void clearBlocks();
 
    void reset();
 
    errc_t run(const SimTime& simTime) final;
 
protected:
    A_DISABLE_COPY(BlockDynamicSystem);
    using StateMap = std::unordered_map<Identifier*, double*>;
 
    std::vector<BlockDerivative*>   derivativeBlocks_{}; // 状态量导数函数块
    std::vector<double>             state_{};            // 状态量
    std::vector<double>             accumulate_{};       // 累加状态量
    std::vector<double>             derivative_{};       // 状态量导数
    StateMap                        stateMap_{};         // 状态量映射表
    StateMap                        derivativeMap_{};    // 状态量导数映射表
};
 
 
AST_NAMESPACE_END