140 lines
3.9 KiB
C++
140 lines
3.9 KiB
C++
#include "instanced_test.h"
|
||
#include <assets/assets_module.h>
|
||
#include <event/events.h>
|
||
#include <module/module_registry.h>
|
||
#include <renderer/renderer_module.h>
|
||
#include <utils/logger.h>
|
||
#include <cmath>
|
||
|
||
namespace extra2d {
|
||
|
||
InstancedTestNode::InstancedTestNode() {
|
||
setName("InstancedTest");
|
||
}
|
||
|
||
InstancedTestNode::~InstancedTestNode() {
|
||
instanceBuffer_.shutdown();
|
||
}
|
||
|
||
bool InstancedTestNode::initialize(uint32_t instanceCount) {
|
||
instanceCount_ = instanceCount;
|
||
|
||
// 获取资源模块
|
||
auto* assets = getModule<AssetsModule>();
|
||
if (!assets) {
|
||
E2D_LOG_ERROR("InstancedTestNode: AssetsModule not available");
|
||
return false;
|
||
}
|
||
|
||
// 使用默认网格（四边形）
|
||
mesh_ = assets->getDefaultQuad();
|
||
if (!mesh_.isValid()) {
|
||
E2D_LOG_ERROR("InstancedTestNode: Failed to get default quad mesh");
|
||
return false;
|
||
}
|
||
|
||
// 使用默认纹理
|
||
texture_ = assets->getDefaultTexture();
|
||
|
||
// 获取实例化渲染材质
|
||
material_ = assets->getInstancedMaterial();
|
||
if (!material_.isValid()) {
|
||
E2D_LOG_ERROR("InstancedTestNode: Failed to get instanced material");
|
||
return false;
|
||
}
|
||
|
||
// 初始化实例缓冲区
|
||
if (!instanceBuffer_.initialize(instanceCount)) {
|
||
E2D_LOG_ERROR("InstancedTestNode: Failed to initialize instance buffer");
|
||
return false;
|
||
}
|
||
|
||
// 预分配实例数据
|
||
instanceData_.resize(instanceCount);
|
||
|
||
// 初始化实例数据
|
||
updateInstances();
|
||
|
||
E2D_LOG_INFO("InstancedTestNode initialized with {} instances", instanceCount);
|
||
return true;
|
||
}
|
||
|
||
void InstancedTestNode::update(float dt) {
|
||
time_ += dt;
|
||
|
||
// 更新实例变换
|
||
updateInstances();
|
||
|
||
// 更新GPU缓冲区
|
||
if (!instanceData_.empty()) {
|
||
instanceBuffer_.updateInstances(instanceData_.data(), instanceCount_);
|
||
}
|
||
}
|
||
|
||
void InstancedTestNode::updateInstances() {
|
||
// 创建螺旋分布的实例
|
||
float radius = 200.0f;
|
||
float centerX = 640.0f;
|
||
float centerY = 360.0f;
|
||
|
||
for (uint32_t i = 0; i < instanceCount_; ++i) {
|
||
float t = static_cast<float>(i) / instanceCount_;
|
||
float angle = t * 6.28318f * 3.0f + time_; // 3圈螺旋
|
||
float r = radius * (0.2f + 0.8f * t);
|
||
|
||
// 位置
|
||
instanceData_[i].position.x = centerX + r * std::cos(angle);
|
||
instanceData_[i].position.y = centerY + r * std::sin(angle);
|
||
|
||
// 旋转（朝向中心）
|
||
instanceData_[i].rotation = angle + 1.5708f;
|
||
|
||
// 缩放（随距离变化）
|
||
float scale = 0.5f + 0.5f * t;
|
||
instanceData_[i].scale.x = scale * 32.0f;
|
||
instanceData_[i].scale.y = scale * 32.0f;
|
||
|
||
// 颜色（彩虹色）
|
||
float hue = t + time_ * 0.1f;
|
||
float r_color = std::abs(std::fmod(hue * 6.0f, 2.0f) - 1.0f);
|
||
float g_color = std::abs(std::fmod(hue * 6.0f + 2.0f, 2.0f) - 1.0f);
|
||
float b_color = std::abs(std::fmod(hue * 6.0f + 4.0f, 2.0f) - 1.0f);
|
||
|
||
instanceData_[i].color.r = r_color;
|
||
instanceData_[i].color.g = g_color;
|
||
instanceData_[i].color.b = b_color;
|
||
instanceData_[i].color.a = 1.0f;
|
||
|
||
// UV坐标（使用完整纹理）
|
||
instanceData_[i].uvX = 0.0f;
|
||
instanceData_[i].uvY = 0.0f;
|
||
instanceData_[i].uvWidth = 1.0f;
|
||
instanceData_[i].uvHeight = 1.0f;
|
||
}
|
||
}
|
||
|
||
void InstancedTestNode::render() {
|
||
if (!isVisible() || instanceCount_ == 0) {
|
||
return;
|
||
}
|
||
|
||
// 检查资源有效性
|
||
if (!mesh_.isValid() || !material_.isValid() || !instanceBuffer_.isValid()) {
|
||
return;
|
||
}
|
||
|
||
// 提交实例化渲染命令
|
||
RenderCommand cmd;
|
||
cmd.type = RenderCommandType::DrawMeshInstanced;
|
||
cmd.sortKey = 0;
|
||
cmd.drawInstanced.mesh = mesh_;
|
||
cmd.drawInstanced.material = material_;
|
||
cmd.drawInstanced.instanceBuffer = &instanceBuffer_;
|
||
cmd.drawInstanced.instanceCount = instanceCount_;
|
||
cmd.drawInstanced.instanceOffset = 0;
|
||
|
||
events::OnRenderSubmit::emit(cmd);
|
||
}
|
||
|
||
} // namespace extra2d
|