bgfx/examples/24-nbody/nbody.cpp

484 lines
13 KiB
C++
Raw Normal View History

2014-12-20 08:09:58 +03:00
/*
2014-12-23 07:14:48 +03:00
* Copyright 2014 Stanlo Slasinski. All rights reserved.
2016-01-01 11:11:04 +03:00
* License: https://github.com/bkaradzic/bgfx#license-bsd-2-clause
2014-12-23 07:14:48 +03:00
*/
2014-12-20 08:09:58 +03:00
#include "common.h"
#include "bgfx_utils.h"
#include "imgui/imgui.h"
#include "camera.h"
#include <bgfx/bgfx.h>
2014-12-20 08:09:58 +03:00
2017-06-26 07:44:04 +03:00
namespace
{
2017-06-18 07:01:05 +03:00
static const char* s_shapeNames[] =
{
"Point",
"Sphere",
"Box",
"Donut"
};
2017-06-17 07:33:18 +03:00
struct ParamsData
2014-12-23 03:36:18 +03:00
{
float timeStep;
int32_t dispatchSize;
float gravity;
float damping;
float particleIntensity;
float particleSize;
int32_t baseSeed;
float particlePower;
float initialSpeed;
int32_t initialShape;
float maxAccel;
};
2017-06-17 07:33:18 +03:00
void initializeParams(int32_t _mode, ParamsData* _params)
2014-12-23 03:36:18 +03:00
{
2014-12-23 07:14:48 +03:00
switch(_mode)
{
case 0:
_params->timeStep = 0.0067f;
_params->dispatchSize = 32;
_params->gravity = 0.069f;
_params->damping = 0.0f;
_params->particleIntensity = 0.35f;
_params->particleSize = 0.925f;
_params->baseSeed = 0;
_params->particlePower = 5.0f;
_params->initialSpeed = 122.6f;
_params->initialShape = 0;
_params->maxAccel = 30.0;
break;
case 1:
_params->timeStep = 0.0157f;
_params->dispatchSize = 32;
_params->gravity = 0.109f;
_params->damping = 0.25f;
_params->particleIntensity = 0.64f;
_params->particleSize = 0.279f;
_params->baseSeed = 57;
_params->particlePower = 3.5f;
_params->initialSpeed = 3.2f;
_params->initialShape = 1;
_params->maxAccel = 100.0;
break;
case 2:
_params->timeStep = 0.02f;
_params->dispatchSize = 32;
_params->gravity = 0.24f;
_params->damping = 0.12f;
_params->particleIntensity = 1.0f;
_params->particleSize = 1.0f;
_params->baseSeed = 23;
_params->particlePower = 4.0f;
_params->initialSpeed = 31.1f;
_params->initialShape = 2;
_params->maxAccel = 39.29f;
break;
case 3:
_params->timeStep = 0.0118f;
_params->dispatchSize = 32;
_params->gravity = 0.141f;
_params->damping = 1.0f;
_params->particleIntensity = 0.64f;
_params->particleSize = 0.28f;
_params->baseSeed = 60;
_params->particlePower = 1.97f;
_params->initialSpeed = 69.7f;
_params->initialShape = 3;
_params->maxAccel = 3.21f;
break;
}
2014-12-23 03:36:18 +03:00
}
static const float s_quadVertices[] =
2014-12-23 21:57:48 +03:00
{
2014-12-20 08:09:58 +03:00
1.0f, 1.0f,
-1.0f, 1.0f,
-1.0f, -1.0f,
1.0f, -1.0f,
};
static const uint16_t s_quadIndices[] = { 0, 1, 2, 2, 3, 0, };
2017-06-17 07:33:18 +03:00
const uint32_t kThreadGroupUpdateSize = 512;
const uint32_t kMaxParticleCount = 32 * 1024;
class ExampleNbody : public entry::AppI
2014-12-20 08:09:58 +03:00
{
2017-06-26 07:44:04 +03:00
public:
ExampleNbody(const char* _name, const char* _description)
: entry::AppI(_name, _description)
{
}
2017-07-15 10:17:29 +03:00
void init(int32_t _argc, const char* const* _argv, uint32_t _width, uint32_t _height) override
2017-06-17 07:33:18 +03:00
{
Args args(_argc, _argv);
2015-10-24 06:52:22 +03:00
2017-06-30 08:23:18 +03:00
m_width = _width;
m_height = _height;
2017-06-26 07:44:04 +03:00
m_debug = BGFX_DEBUG_NONE;
2017-06-17 07:33:18 +03:00
m_reset = BGFX_RESET_VSYNC;
2014-12-20 08:09:58 +03:00
2017-06-17 07:33:18 +03:00
bgfx::init(args.m_type, args.m_pciId);
bgfx::reset(m_width, m_height, m_reset);
2014-12-20 08:09:58 +03:00
2017-06-17 07:33:18 +03:00
// Enable debug text.
bgfx::setDebug(m_debug);
2014-12-20 08:09:58 +03:00
2017-06-17 07:33:18 +03:00
// Set view 0 clear state.
bgfx::setViewClear(0
, BGFX_CLEAR_COLOR|BGFX_CLEAR_DEPTH
, 0x303030ff
, 1.0f
, 0
);
2014-12-20 08:09:58 +03:00
2017-06-17 07:33:18 +03:00
const bgfx::Caps* caps = bgfx::getCaps();
const bool computeSupported = !!(caps->supported & BGFX_CAPS_COMPUTE);
const bool indirectSupported = !!(caps->supported & BGFX_CAPS_DRAW_INDIRECT);
2014-12-20 08:09:58 +03:00
2017-06-27 08:51:56 +03:00
imguiCreate();
2017-06-17 07:33:18 +03:00
if (computeSupported)
{
bgfx::VertexDecl quadVertexDecl;
quadVertexDecl.begin()
.add(bgfx::Attrib::Position, 2, bgfx::AttribType::Float)
.end();
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
// Create static vertex buffer.
m_vbh = bgfx::createVertexBuffer(
2015-03-31 06:36:31 +03:00
// Static data can be passed with bgfx::makeRef
bgfx::makeRef(s_quadVertices, sizeof(s_quadVertices) )
, quadVertexDecl
);
2017-06-17 07:33:18 +03:00
// Create static index buffer.
m_ibh = bgfx::createIndexBuffer(
2015-03-31 06:36:31 +03:00
// Static data can be passed with bgfx::makeRef
bgfx::makeRef(s_quadIndices, sizeof(s_quadIndices) )
);
2017-06-17 07:33:18 +03:00
// Create particle program from shaders.
m_particleProgram = loadProgram("vs_particle", "fs_particle");
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
// Setup compute buffers
bgfx::VertexDecl computeVertexDecl;
computeVertexDecl.begin()
.add(bgfx::Attrib::TexCoord0, 4, bgfx::AttribType::Float)
.end();
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
m_currPositionBuffer0 = bgfx::createDynamicVertexBuffer(1 << 15, computeVertexDecl, BGFX_BUFFER_COMPUTE_READ_WRITE);
m_currPositionBuffer1 = bgfx::createDynamicVertexBuffer(1 << 15, computeVertexDecl, BGFX_BUFFER_COMPUTE_READ_WRITE);
m_prevPositionBuffer0 = bgfx::createDynamicVertexBuffer(1 << 15, computeVertexDecl, BGFX_BUFFER_COMPUTE_READ_WRITE);
m_prevPositionBuffer1 = bgfx::createDynamicVertexBuffer(1 << 15, computeVertexDecl, BGFX_BUFFER_COMPUTE_READ_WRITE);
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
u_params = bgfx::createUniform("u_params", bgfx::UniformType::Vec4, 3);
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
m_initInstancesProgram = bgfx::createProgram(loadShader("cs_init_instances"), true);
m_updateInstancesProgram = bgfx::createProgram(loadShader("cs_update_instances"), true);
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
m_indirectProgram = BGFX_INVALID_HANDLE;
m_indirectBuffer = BGFX_INVALID_HANDLE;
2017-06-17 07:33:18 +03:00
if (indirectSupported)
{
m_indirectProgram = bgfx::createProgram(loadShader("cs_indirect"), true);
m_indirectBuffer = bgfx::createIndirectBuffer(2);
}
2017-06-17 07:33:18 +03:00
initializeParams(0, &m_paramsData);
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
bgfx::setUniform(u_params, &m_paramsData, 3);
bgfx::setBuffer(0, m_prevPositionBuffer0, bgfx::Access::Write);
bgfx::setBuffer(1, m_currPositionBuffer0, bgfx::Access::Write);
bgfx::dispatch(0, m_initInstancesProgram, kMaxParticleCount / kThreadGroupUpdateSize, 1, 1);
2014-12-23 03:36:18 +03:00
2017-06-17 07:33:18 +03:00
float initialPos[3] = { 0.0f, 0.0f, -45.0f };
cameraCreate();
cameraSetPosition(initialPos);
cameraSetVerticalAngle(0.0f);
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
m_useIndirect = false;
2014-12-20 08:09:58 +03:00
2017-06-17 07:33:18 +03:00
m_timeOffset = bx::getHPCounter();
}
}
2014-12-20 08:09:58 +03:00
2017-07-15 10:17:29 +03:00
virtual int shutdown() override
2017-06-17 07:33:18 +03:00
{
// Cleanup.
cameraDestroy();
imguiDestroy();
if (bgfx::isValid(m_indirectProgram) )
{
bgfx::destroy(m_indirectProgram);
bgfx::destroy(m_indirectBuffer);
2017-06-17 07:33:18 +03:00
}
bgfx::destroy(u_params);
bgfx::destroy(m_currPositionBuffer0);
bgfx::destroy(m_currPositionBuffer1);
bgfx::destroy(m_prevPositionBuffer0);
bgfx::destroy(m_prevPositionBuffer1);
bgfx::destroy(m_updateInstancesProgram);
bgfx::destroy(m_initInstancesProgram);
bgfx::destroy(m_ibh);
bgfx::destroy(m_vbh);
bgfx::destroy(m_particleProgram);
2017-06-17 07:33:18 +03:00
// Shutdown bgfx.
bgfx::shutdown();
return 0;
}
2017-07-15 10:17:29 +03:00
bool update() override
2017-06-17 07:33:18 +03:00
{
2017-06-18 07:01:05 +03:00
if (!entry::processEvents(m_width, m_height, m_debug, m_reset, &m_mouseState) )
2014-12-20 08:09:58 +03:00
{
2017-06-26 07:44:04 +03:00
imguiBeginFrame(
m_mouseState.m_mx
, m_mouseState.m_my
, (m_mouseState.m_buttons[entry::MouseButton::Left ] ? IMGUI_MBUT_LEFT : 0)
| (m_mouseState.m_buttons[entry::MouseButton::Right ] ? IMGUI_MBUT_RIGHT : 0)
| (m_mouseState.m_buttons[entry::MouseButton::Middle] ? IMGUI_MBUT_MIDDLE : 0)
, m_mouseState.m_mz
, uint16_t(m_width)
, uint16_t(m_height)
);
2017-06-30 09:19:20 +03:00
showExampleDialog(this);
2017-06-26 07:44:04 +03:00
2017-06-17 07:33:18 +03:00
const bgfx::Caps* caps = bgfx::getCaps();
const bool computeSupported = !!(caps->supported & BGFX_CAPS_COMPUTE);
const bool indirectSupported = !!(caps->supported & BGFX_CAPS_DRAW_INDIRECT);
2015-03-31 06:36:31 +03:00
int64_t now = bx::getHPCounter();
2017-06-17 07:33:18 +03:00
float time = (float)( (now - m_timeOffset)/double(bx::getHPFrequency() ) );
2015-03-31 06:36:31 +03:00
static int64_t last = now;
const int64_t frameTime = now - last;
last = now;
const double freq = double(bx::getHPFrequency() );
const float deltaTime = float(frameTime/freq);
2014-12-20 08:09:58 +03:00
// Set view 0 default viewport.
2017-06-17 07:33:18 +03:00
bgfx::setViewRect(0, 0, 0, uint16_t(m_width), uint16_t(m_height) );
2015-03-31 06:36:31 +03:00
2017-06-17 07:33:18 +03:00
if (computeSupported)
{
2017-06-27 08:51:56 +03:00
ImGui::SetNextWindowPos(
ImVec2(m_width - m_width / 5.0f - 10.0f, 10.0f)
, ImGuiSetCond_FirstUseEver
);
ImGui::Begin("Settings"
2017-06-18 07:01:05 +03:00
, NULL
, ImVec2(m_width / 5.0f, m_height / 1.5f)
, ImGuiWindowFlags_AlwaysAutoResize
);
bool reset = false;
int32_t shape = m_paramsData.initialShape;
if (ImGui::Combo("Initial shape", &shape, s_shapeNames, BX_COUNTOF(s_shapeNames) ) )
2017-06-17 07:33:18 +03:00
{
2017-06-18 07:01:05 +03:00
// Modify parameters and reset if shape is changed
initializeParams(shape, &m_paramsData);
reset = true;
2017-06-17 07:33:18 +03:00
}
2017-06-18 07:01:05 +03:00
ImGui::SliderInt("Random seed", &m_paramsData.baseSeed, 0, 100);
if (ImGui::Button("Reset") )
2017-06-17 07:33:18 +03:00
{
2017-06-18 07:01:05 +03:00
reset = true;
}
ImGui::Separator();
ImGui::SliderInt("Particle count (x512)", &m_paramsData.dispatchSize, 1, 64);
ImGui::SliderFloat("Gravity", &m_paramsData.gravity, 0.0f, 0.3f);
ImGui::SliderFloat("Damping", &m_paramsData.damping, 0.0f, 1.0f);
ImGui::SliderFloat("Max acceleration", &m_paramsData.maxAccel, 0.0f, 100.0f);
ImGui::SliderFloat("Time step", &m_paramsData.timeStep, 0.0f, 0.02f);
ImGui::Separator();
ImGui::SliderFloat("Particle intensity", &m_paramsData.particleIntensity, 0.0f, 1.0f);
ImGui::SliderFloat("Particle size", &m_paramsData.particleSize, 0.0f, 1.0f);
ImGui::SliderFloat("Particle power", &m_paramsData.particlePower, 0.001f, 16.0f);
ImGui::Separator();
if (indirectSupported)
{
ImGui::Checkbox("Use draw/dispatch indirect", &m_useIndirect);
2017-06-17 07:33:18 +03:00
}
2017-06-18 07:01:05 +03:00
ImGui::End();
if (reset)
2017-06-17 07:33:18 +03:00
{
bgfx::setBuffer(0, m_prevPositionBuffer0, bgfx::Access::Write);
bgfx::setBuffer(1, m_currPositionBuffer0, bgfx::Access::Write);
bgfx::setUniform(u_params, &m_paramsData, 3);
bgfx::dispatch(0, m_initInstancesProgram, kMaxParticleCount / kThreadGroupUpdateSize, 1, 1);
}
if (m_useIndirect)
{
bgfx::setUniform(u_params, &m_paramsData, 3);
bgfx::setBuffer(0, m_indirectBuffer, bgfx::Access::Write);
bgfx::dispatch(0, m_indirectProgram);
}
bgfx::setBuffer(0, m_prevPositionBuffer0, bgfx::Access::Read);
bgfx::setBuffer(1, m_currPositionBuffer0, bgfx::Access::Read);
bgfx::setBuffer(2, m_prevPositionBuffer1, bgfx::Access::Write);
bgfx::setBuffer(3, m_currPositionBuffer1, bgfx::Access::Write);
bgfx::setUniform(u_params, &m_paramsData, 3);
if (m_useIndirect)
{
bgfx::dispatch(0, m_updateInstancesProgram, m_indirectBuffer, 1);
}
else
{
bgfx::dispatch(0, m_updateInstancesProgram, uint16_t(m_paramsData.dispatchSize), 1, 1);
}
bx::xchg(m_currPositionBuffer0, m_currPositionBuffer1);
bx::xchg(m_prevPositionBuffer0, m_prevPositionBuffer1);
// Update camera.
2017-06-18 07:01:05 +03:00
cameraUpdate(deltaTime, m_mouseState);
2017-06-17 07:33:18 +03:00
float view[16];
cameraGetViewMtx(view);
// Set view and projection matrix for view 0.
const bgfx::HMD* hmd = bgfx::getHMD();
if (NULL != hmd && 0 != (hmd->flags & BGFX_HMD_RENDERING) )
{
float viewHead[16];
float eye[3] = {};
bx::mtxQuatTranslationHMD(viewHead, hmd->eye[0].rotation, eye);
float tmp[16];
bx::mtxMul(tmp, view, viewHead);
bgfx::setViewTransform(
0
, tmp
, hmd->eye[0].projection
, BGFX_VIEW_STEREO
, hmd->eye[1].projection
);
// Set view 0 default viewport.
//
// Use HMD's width/height since HMD's internal frame buffer size
// might be much larger than window size.
bgfx::setViewRect(0, 0, 0, hmd->width, hmd->height);
}
else
{
float proj[16];
bx::mtxProj(
proj
, 90.0f
, float(m_width)/float(m_height)
, 0.1f
, 10000.0f
, bgfx::getCaps()->homogeneousDepth
);
bgfx::setViewTransform(0, view, proj);
// Set view 0 default viewport.
bgfx::setViewRect(0, 0, 0, uint16_t(m_width), uint16_t(m_height) );
}
// Set vertex and index buffer.
bgfx::setVertexBuffer(0, m_vbh);
bgfx::setIndexBuffer(m_ibh);
bgfx::setInstanceDataBuffer(m_currPositionBuffer0
, 0
, m_paramsData.dispatchSize * kThreadGroupUpdateSize
2015-03-31 06:36:31 +03:00
);
2017-06-17 07:33:18 +03:00
// Set render states.
bgfx::setState(0
2015-03-31 06:36:31 +03:00
| BGFX_STATE_RGB_WRITE
| BGFX_STATE_BLEND_ADD
| BGFX_STATE_DEPTH_TEST_ALWAYS
);
2017-06-17 07:33:18 +03:00
// Submit primitive for rendering to view 0.
if (m_useIndirect)
{
bgfx::submit(0, m_particleProgram, m_indirectBuffer, 0);
}
else
{
bgfx::submit(0, m_particleProgram);
}
}
else
{
2017-06-17 07:33:18 +03:00
bgfx::setViewRect(0, 0, 0, uint16_t(m_width), uint16_t(m_height) );
bool blink = uint32_t(time*3.0f)&1;
2017-06-26 07:44:04 +03:00
bgfx::dbgTextPrintf(0, 0, blink ? 0x1f : 0x01, " Compute is not supported by GPU. ");
2017-06-17 07:33:18 +03:00
bgfx::touch(0);
}
2015-03-31 06:36:31 +03:00
2017-06-26 07:44:04 +03:00
imguiEndFrame();
2015-03-31 06:36:31 +03:00
// Advance to next frame. Rendering thread will be kicked to
// process submitted rendering primitives.
bgfx::frame();
2017-06-30 09:19:20 +03:00
return true;
}
2017-06-17 07:33:18 +03:00
return false;
2015-03-31 06:36:31 +03:00
}
2014-12-20 08:09:58 +03:00
2017-06-26 07:44:04 +03:00
entry::MouseState m_mouseState;
2017-06-17 07:33:18 +03:00
uint32_t m_width;
uint32_t m_height;
uint32_t m_debug;
uint32_t m_reset;
bool m_useIndirect;
ParamsData m_paramsData;
bgfx::VertexBufferHandle m_vbh;
bgfx::IndexBufferHandle m_ibh;
bgfx::ProgramHandle m_particleProgram;
bgfx::ProgramHandle m_indirectProgram;
bgfx::ProgramHandle m_initInstancesProgram;
bgfx::ProgramHandle m_updateInstancesProgram;
bgfx::IndirectBufferHandle m_indirectBuffer;
bgfx::DynamicVertexBufferHandle m_currPositionBuffer0;
bgfx::DynamicVertexBufferHandle m_currPositionBuffer1;
bgfx::DynamicVertexBufferHandle m_prevPositionBuffer0;
bgfx::DynamicVertexBufferHandle m_prevPositionBuffer1;
bgfx::UniformHandle u_params;
int64_t m_timeOffset;
};
2014-12-20 08:09:58 +03:00
2017-06-26 07:44:04 +03:00
} // namespace
ENTRY_IMPLEMENT_MAIN(ExampleNbody, "24-nbody", "N-body simulation with compute shaders using buffers.");