494 lines
12 KiB
C++
494 lines
12 KiB
C++
/*
|
|
* Copyright 2011-2017 Branimir Karadzic. All rights reserved.
|
|
* License: https://github.com/bkaradzic/bgfx#license-bsd-2-clause
|
|
*/
|
|
|
|
#include "common.h"
|
|
#include "bgfx_utils.h"
|
|
|
|
#include <bx/allocator.h>
|
|
#include <bx/string.h>
|
|
#include <bx/crtimpl.h>
|
|
|
|
#include "aviwriter.h"
|
|
|
|
#include <inttypes.h>
|
|
|
|
struct PosColorVertex
|
|
{
|
|
float m_x;
|
|
float m_y;
|
|
float m_z;
|
|
uint32_t m_abgr;
|
|
|
|
static void init()
|
|
{
|
|
ms_decl
|
|
.begin()
|
|
.add(bgfx::Attrib::Position, 3, bgfx::AttribType::Float)
|
|
.add(bgfx::Attrib::Color0, 4, bgfx::AttribType::Uint8, true)
|
|
.end();
|
|
};
|
|
|
|
static bgfx::VertexDecl ms_decl;
|
|
};
|
|
|
|
bgfx::VertexDecl PosColorVertex::ms_decl;
|
|
|
|
static PosColorVertex s_cubeVertices[8] =
|
|
{
|
|
{-1.0f, 1.0f, 1.0f, 0xff000000 },
|
|
{ 1.0f, 1.0f, 1.0f, 0xff0000ff },
|
|
{-1.0f, -1.0f, 1.0f, 0xff00ff00 },
|
|
{ 1.0f, -1.0f, 1.0f, 0xff00ffff },
|
|
{-1.0f, 1.0f, -1.0f, 0xffff0000 },
|
|
{ 1.0f, 1.0f, -1.0f, 0xffff00ff },
|
|
{-1.0f, -1.0f, -1.0f, 0xffffff00 },
|
|
{ 1.0f, -1.0f, -1.0f, 0xffffffff },
|
|
};
|
|
|
|
static const uint16_t s_cubeIndices[36] =
|
|
{
|
|
0, 1, 2, // 0
|
|
1, 3, 2,
|
|
4, 6, 5, // 2
|
|
5, 6, 7,
|
|
0, 2, 4, // 4
|
|
4, 2, 6,
|
|
1, 5, 3, // 6
|
|
5, 7, 3,
|
|
0, 4, 1, // 8
|
|
4, 5, 1,
|
|
2, 3, 6, // 10
|
|
6, 3, 7,
|
|
};
|
|
|
|
void imageWriteTga(bx::WriterI* _writer, uint32_t _width, uint32_t _height, uint32_t _pitch, const void* _src, bool _grayscale, bool _yflip, bx::Error* _err)
|
|
{
|
|
BX_ERROR_SCOPE(_err);
|
|
|
|
uint8_t type = _grayscale ? 3 : 2;
|
|
uint8_t bpp = _grayscale ? 8 : 32;
|
|
|
|
uint8_t header[18] = {};
|
|
header[ 2] = type;
|
|
header[12] = _width &0xff;
|
|
header[13] = (_width >>8)&0xff;
|
|
header[14] = _height &0xff;
|
|
header[15] = (_height>>8)&0xff;
|
|
header[16] = bpp;
|
|
header[17] = 32;
|
|
|
|
bx::write(_writer, header, sizeof(header), _err);
|
|
|
|
uint32_t dstPitch = _width*bpp/8;
|
|
if (_yflip)
|
|
{
|
|
uint8_t* data = (uint8_t*)_src + _pitch*_height - _pitch;
|
|
for (uint32_t yy = 0; yy < _height; ++yy)
|
|
{
|
|
bx::write(_writer, data, dstPitch, _err);
|
|
data -= _pitch;
|
|
}
|
|
}
|
|
else if (_pitch == dstPitch)
|
|
{
|
|
bx::write(_writer, _src, _height*_pitch, _err);
|
|
}
|
|
else
|
|
{
|
|
uint8_t* data = (uint8_t*)_src;
|
|
for (uint32_t yy = 0; yy < _height; ++yy)
|
|
{
|
|
bx::write(_writer, data, dstPitch, _err);
|
|
data += _pitch;
|
|
}
|
|
}
|
|
}
|
|
|
|
void saveTga(const char* _filePath, uint32_t _width, uint32_t _height, uint32_t _srcPitch, const void* _src, bool _grayscale, bool _yflip)
|
|
{
|
|
bx::FileWriter writer;
|
|
bx::Error err;
|
|
if (bx::open(&writer, _filePath, false, &err) )
|
|
{
|
|
imageWriteTga(&writer, _width, _height, _srcPitch, _src, _grayscale, _yflip, &err);
|
|
bx::close(&writer);
|
|
}
|
|
}
|
|
|
|
struct BgfxCallback : public bgfx::CallbackI
|
|
{
|
|
virtual ~BgfxCallback()
|
|
{
|
|
}
|
|
|
|
virtual void fatal(bgfx::Fatal::Enum _code, const char* _str) BX_OVERRIDE
|
|
{
|
|
// Something unexpected happened, inform user and bail out.
|
|
bx::debugPrintf("Fatal error: 0x%08x: %s", _code, _str);
|
|
|
|
// Must terminate, continuing will cause crash anyway.
|
|
abort();
|
|
}
|
|
|
|
virtual void traceVargs(const char* _filePath, uint16_t _line, const char* _format, va_list _argList) BX_OVERRIDE
|
|
{
|
|
bx::debugPrintf("%s (%d): ", _filePath, _line);
|
|
bx::debugPrintfVargs(_format, _argList);
|
|
}
|
|
|
|
virtual uint32_t cacheReadSize(uint64_t _id) BX_OVERRIDE
|
|
{
|
|
char filePath[256];
|
|
bx::snprintf(filePath, sizeof(filePath), "temp/%016" PRIx64, _id);
|
|
|
|
// Use cache id as filename.
|
|
bx::FileReaderI* reader = entry::getFileReader();
|
|
bx::Error err;
|
|
if (bx::open(reader, filePath, &err) )
|
|
{
|
|
uint32_t size = (uint32_t)bx::getSize(reader);
|
|
bx::close(reader);
|
|
// Return size of shader file.
|
|
return size;
|
|
}
|
|
|
|
// Return 0 if shader is not found.
|
|
return 0;
|
|
}
|
|
|
|
virtual bool cacheRead(uint64_t _id, void* _data, uint32_t _size) BX_OVERRIDE
|
|
{
|
|
char filePath[256];
|
|
bx::snprintf(filePath, sizeof(filePath), "temp/%016" PRIx64, _id);
|
|
|
|
// Use cache id as filename.
|
|
bx::FileReaderI* reader = entry::getFileReader();
|
|
bx::Error err;
|
|
if (bx::open(reader, filePath, &err) )
|
|
{
|
|
// Read shader.
|
|
uint32_t result = bx::read(reader, _data, _size, &err);
|
|
bx::close(reader);
|
|
|
|
// Make sure that read size matches requested size.
|
|
return result == _size;
|
|
}
|
|
|
|
// Shader is not found in cache, needs to be rebuilt.
|
|
return false;
|
|
}
|
|
|
|
virtual void cacheWrite(uint64_t _id, const void* _data, uint32_t _size) BX_OVERRIDE
|
|
{
|
|
char filePath[256];
|
|
bx::snprintf(filePath, sizeof(filePath), "temp/%016" PRIx64, _id);
|
|
|
|
// Use cache id as filename.
|
|
bx::FileWriterI* writer = entry::getFileWriter();
|
|
bx::Error err;
|
|
if (bx::open(writer, filePath, false, &err) )
|
|
{
|
|
// Write shader to cache location.
|
|
bx::write(writer, _data, _size, &err);
|
|
bx::close(writer);
|
|
}
|
|
}
|
|
|
|
virtual void screenShot(const char* _filePath, uint32_t _width, uint32_t _height, uint32_t _pitch, const void* _data, uint32_t /*_size*/, bool _yflip) BX_OVERRIDE
|
|
{
|
|
char temp[1024];
|
|
|
|
// Save screen shot as TGA.
|
|
bx::snprintf(temp, BX_COUNTOF(temp), "%s.tga", _filePath);
|
|
saveTga(temp, _width, _height, _pitch, _data, false, _yflip);
|
|
}
|
|
|
|
virtual void captureBegin(uint32_t _width, uint32_t _height, uint32_t /*_pitch*/, bgfx::TextureFormat::Enum /*_format*/, bool _yflip) BX_OVERRIDE
|
|
{
|
|
m_writer = BX_NEW(entry::getAllocator(), AviWriter)(entry::getFileWriter() );
|
|
if (!m_writer->open("temp/capture.avi", _width, _height, 60, _yflip) )
|
|
{
|
|
BX_DELETE(entry::getAllocator(), m_writer);
|
|
m_writer = NULL;
|
|
}
|
|
}
|
|
|
|
virtual void captureEnd() BX_OVERRIDE
|
|
{
|
|
if (NULL != m_writer)
|
|
{
|
|
m_writer->close();
|
|
BX_DELETE(entry::getAllocator(), m_writer);
|
|
m_writer = NULL;
|
|
}
|
|
}
|
|
|
|
virtual void captureFrame(const void* _data, uint32_t /*_size*/) BX_OVERRIDE
|
|
{
|
|
if (NULL != m_writer)
|
|
{
|
|
m_writer->frame(_data);
|
|
}
|
|
}
|
|
|
|
AviWriter* m_writer;
|
|
};
|
|
|
|
class BgfxAllocator : public bx::AllocatorI
|
|
{
|
|
public:
|
|
BgfxAllocator()
|
|
: m_numBlocks(0)
|
|
, m_maxBlocks(0)
|
|
{
|
|
}
|
|
|
|
virtual ~BgfxAllocator()
|
|
{
|
|
}
|
|
|
|
virtual void* realloc(void* _ptr, size_t _size, size_t _align, const char* _file, uint32_t _line) BX_OVERRIDE
|
|
{
|
|
if (0 == _size)
|
|
{
|
|
if (NULL != _ptr)
|
|
{
|
|
if (BX_CONFIG_ALLOCATOR_NATURAL_ALIGNMENT >= _align)
|
|
{
|
|
bx::debugPrintf("%s(%d): FREE %p\n", _file, _line, _ptr);
|
|
::free(_ptr);
|
|
--m_numBlocks;
|
|
}
|
|
else
|
|
{
|
|
bx::alignedFree(this, _ptr, _align, _file, _line);
|
|
}
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
else if (NULL == _ptr)
|
|
{
|
|
if (BX_CONFIG_ALLOCATOR_NATURAL_ALIGNMENT >= _align)
|
|
{
|
|
void* ptr = ::malloc(_size);
|
|
bx::debugPrintf("%s(%d): ALLOC %p of %d byte(s)\n", _file, _line, ptr, _size);
|
|
++m_numBlocks;
|
|
m_maxBlocks = bx::uint32_max(m_maxBlocks, m_numBlocks);
|
|
return ptr;
|
|
}
|
|
|
|
return bx::alignedAlloc(this, _size, _align, _file, _line);
|
|
}
|
|
|
|
if (BX_CONFIG_ALLOCATOR_NATURAL_ALIGNMENT >= _align)
|
|
{
|
|
void* ptr = ::realloc(_ptr, _size);
|
|
bx::debugPrintf("%s(%d): REALLOC %p (old %p) of %d byte(s)\n", _file, _line, ptr, _ptr, _size);
|
|
|
|
if (NULL == _ptr)
|
|
{
|
|
++m_numBlocks;
|
|
m_maxBlocks = bx::uint32_max(m_maxBlocks, m_numBlocks);
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
|
|
return bx::alignedRealloc(this, _ptr, _size, _align, _file, _line);
|
|
}
|
|
|
|
void dumpStats() const
|
|
{
|
|
bx::debugPrintf("Allocator stats: num blocks %d (peak: %d)\n", m_numBlocks, m_maxBlocks);
|
|
}
|
|
|
|
private:
|
|
uint32_t m_numBlocks;
|
|
uint32_t m_maxBlocks;
|
|
};
|
|
|
|
class ExampleCallback : public entry::AppI
|
|
{
|
|
void init(int _argc, char** _argv) BX_OVERRIDE
|
|
{
|
|
Args args(_argc, _argv);
|
|
|
|
m_width = 1280;
|
|
m_height = 720;
|
|
|
|
// Enumerate supported backend renderers.
|
|
m_numRenderers = bgfx::getSupportedRenderers(BX_COUNTOF(m_renderers), m_renderers);
|
|
|
|
bgfx::init(bgfx::RendererType::Count == args.m_type
|
|
? m_renderers[bx::getHPCounter() % m_numRenderers] /* randomize renderer */
|
|
: args.m_type
|
|
, args.m_pciId
|
|
, 0
|
|
, &m_callback // custom callback handler
|
|
, &m_allocator // custom allocator
|
|
);
|
|
bgfx::reset(m_width, m_height, BGFX_RESET_CAPTURE|BGFX_RESET_MSAA_X16);
|
|
|
|
// Enable debug text.
|
|
bgfx::setDebug(BGFX_DEBUG_TEXT);
|
|
|
|
// Set view 0 default viewport.
|
|
bgfx::setViewRect(0, 0, 0, 1280, 720);
|
|
|
|
// Set view 0 clear state.
|
|
bgfx::setViewClear(0
|
|
, BGFX_CLEAR_COLOR|BGFX_CLEAR_DEPTH
|
|
, 0x303030ff
|
|
, 1.0f
|
|
, 0
|
|
);
|
|
|
|
// Create vertex stream declaration.
|
|
PosColorVertex::init();
|
|
|
|
// Create static vertex buffer.
|
|
m_vbh = bgfx::createVertexBuffer(
|
|
bgfx::makeRef(s_cubeVertices, sizeof(s_cubeVertices) )
|
|
, PosColorVertex::ms_decl
|
|
);
|
|
|
|
// Create static index buffer.
|
|
m_ibh = bgfx::createIndexBuffer(bgfx::makeRef(s_cubeIndices, sizeof(s_cubeIndices) ) );
|
|
|
|
// Create program from shaders.
|
|
m_program = loadProgram("vs_callback", "fs_callback");
|
|
|
|
m_time = 0.0f;
|
|
m_frame = 0;
|
|
}
|
|
|
|
virtual int shutdown() BX_OVERRIDE
|
|
{
|
|
// Cleanup.
|
|
bgfx::destroyIndexBuffer(m_ibh);
|
|
bgfx::destroyVertexBuffer(m_vbh);
|
|
bgfx::destroyProgram(m_program);
|
|
|
|
// Shutdown bgfx.
|
|
bgfx::shutdown();
|
|
|
|
m_allocator.dumpStats();
|
|
|
|
return 0;
|
|
}
|
|
|
|
bool update() BX_OVERRIDE
|
|
{
|
|
|
|
|
|
// 5 second 60Hz video
|
|
if ( m_frame < 300)
|
|
{
|
|
++m_frame;
|
|
const bgfx::RendererType::Enum rendererType = bgfx::getRendererType();
|
|
|
|
// This dummy draw call is here to make sure that view 0 is cleared
|
|
// if no other draw calls are submitted to view 0.
|
|
bgfx::touch(0);
|
|
|
|
int64_t now = bx::getHPCounter();
|
|
static int64_t last = now;
|
|
const int64_t frameTime = now - last;
|
|
last = now;
|
|
const double freq = double(bx::getHPFrequency() );
|
|
const double toMs = 1000.0/freq;
|
|
|
|
// Use debug font to print information about this example.
|
|
bgfx::dbgTextClear();
|
|
bgfx::dbgTextPrintf( 0, 1, 0x4f, "bgfx/examples/07-callback");
|
|
bgfx::dbgTextPrintf( 0, 2, 0x6f, "Description: Implementing application specific callbacks for taking screen shots,");
|
|
bgfx::dbgTextPrintf(13, 3, 0x6f, "caching OpenGL binary shaders, and video capture.");
|
|
bgfx::dbgTextPrintf( 0, 4, 0x0f, "Frame: % 7.3f[ms]", double(frameTime)*toMs);
|
|
|
|
bgfx::dbgTextPrintf( 2, 6, 0x0e, "Supported renderers:");
|
|
for (uint8_t ii = 0; ii < m_numRenderers; ++ii)
|
|
{
|
|
bgfx::dbgTextPrintf( 2, 7+ii, 0x0c, "[%c] %s"
|
|
, m_renderers[ii] == rendererType ? '\xfe' : ' '
|
|
, bgfx::getRendererName(m_renderers[ii])
|
|
);
|
|
}
|
|
|
|
float at[3] = { 0.0f, 0.0f, 0.0f };
|
|
float eye[3] = { 0.0f, 0.0f, -35.0f };
|
|
|
|
float view[16];
|
|
float proj[16];
|
|
bx::mtxLookAt(view, eye, at);
|
|
bx::mtxProj(proj, 60.0f, float(m_width)/float(m_height), 0.1f, 100.0f, bgfx::getCaps()->homogeneousDepth);
|
|
|
|
// Set view and projection matrix for view 0.
|
|
bgfx::setViewTransform(0, view, proj);
|
|
|
|
m_time += 1.0f/60.0f;
|
|
|
|
// Submit 11x11 cubes.
|
|
for (uint32_t yy = 0; yy < 11; ++yy)
|
|
{
|
|
for (uint32_t xx = 0; xx < 11-yy; ++xx)
|
|
{
|
|
float mtx[16];
|
|
bx::mtxRotateXY(mtx, m_time + xx*0.21f, m_time + yy*0.37f);
|
|
mtx[12] = -15.0f + float(xx)*3.0f;
|
|
mtx[13] = -15.0f + float(yy)*3.0f;
|
|
mtx[14] = 0.0f;
|
|
|
|
// Set model matrix for rendering.
|
|
bgfx::setTransform(mtx);
|
|
|
|
// Set vertex and index buffer.
|
|
bgfx::setVertexBuffer(0, m_vbh);
|
|
bgfx::setIndexBuffer(m_ibh);
|
|
|
|
// Set render states.
|
|
bgfx::setState(BGFX_STATE_DEFAULT);
|
|
|
|
// Submit primitive for rendering to view 0.
|
|
bgfx::submit(0, m_program);
|
|
}
|
|
}
|
|
|
|
// Take screen shot at frame 150.
|
|
if (150 == m_frame)
|
|
{
|
|
bgfx::FrameBufferHandle fbh = BGFX_INVALID_HANDLE;
|
|
bgfx::requestScreenShot(fbh, "temp/frame150");
|
|
}
|
|
|
|
// Advance to next frame. Rendering thread will be kicked to
|
|
// process submitted rendering primitives.
|
|
bgfx::frame();
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
BgfxCallback m_callback;
|
|
BgfxAllocator m_allocator;
|
|
|
|
uint32_t m_width;
|
|
uint32_t m_height;
|
|
|
|
bgfx::RendererType::Enum m_renderers[bgfx::RendererType::Count];
|
|
uint8_t m_numRenderers;
|
|
|
|
bgfx::VertexBufferHandle m_vbh;
|
|
bgfx::IndexBufferHandle m_ibh;
|
|
bgfx::ProgramHandle m_program;
|
|
float m_time;
|
|
uint32_t m_frame;
|
|
};
|
|
|
|
ENTRY_IMPLEMENT_MAIN(ExampleCallback);
|
|
|