GpuAgentSimulation.h

浏览该文件的文档.

#pragma once
#include "common/Export.h"
 
#include "common/Result.h"
#include "gpuagents/AgentState.h"
#include "gpuagents/EffectProfile.h"
#include "gpuagents/EnvironmentSnapshot.h"
#include "gpuagents/IAgentSolver.h"
 
#include <cstdint>
#include <functional>
#include <memory>
#include <vector>
 
namespace eve::gpuagents {
 
class EVENGINE_API_DOMAINS GpuAgentSimulation {
public:
    GpuAgentSimulation() = default;
 
    [[nodiscard("check simulation configure")]] Result<void> configure(const EffectProfile& profile);
 
    [[nodiscard("check solver bind")]] Result<void> setSolver(std::unique_ptr<IAgentSolver> solver);
 
    [[nodiscard("check initialize")]] Result<void> initialize(int count,
                                                              const std::function<AgentState(int)>& spawnFn);
 
    void reset();
 
    [[nodiscard("check simulation step")]] Result<void> step(float dt, const EnvironmentSnapshot& env);
 
    int capacity() const { return capacity_; }
    int aliveCount() const;
    const std::vector<AgentState>& states() const { return buffers_[front_]; }
    std::vector<AgentState>& states() { return buffers_[front_]; }
    const EffectProfile& profile() const { return profile_; }
    bool isConfigured() const { return configured_; }
    float simTime() const { return simTime_; }
    std::uint64_t stepCount() const { return stepCount_; }
 
private:
    EffectProfile                   profile_{};
    std::unique_ptr<IAgentSolver>   solver_;
    std::vector<AgentState>         buffers_[2]{};
    int                             front_      = 0;
    int                             capacity_   = 0;
    float                           accumulator_ = 0.f;
    float                           simTime_     = 0.f;
    std::uint64_t                   stepCount_   = 0;
    bool                            configured_  = false;
 
    [[nodiscard]] Result<void> makeDefaultSolver();
};
 
}  // namespace eve::gpuagents