196 lines
7.8 KiB
C++
196 lines
7.8 KiB
C++
#include "test_vertex_shader.h"
|
|
#include "device/dvc_buffer_accessor.h"
|
|
#include "math/dvc_vector2.h"
|
|
#include "math/dvc_vector3.h"
|
|
#include "math/dvc_matrix4.h"
|
|
#include "pipe/dvc_render_pipe.h"
|
|
#include "device/dvc_uniform_buffer.h"
|
|
|
|
TestVertexShader::TestVertexShader()
|
|
{
|
|
//output point v_position = point(0, 0, 0) [[ string attribute="POSITION"]],
|
|
m_outputVertexDesc.addElement("v_position", davinci::VET_POSITION, davinci::CT_Float32, davinci::DT_Vec3);
|
|
//output vector v_normal = vector(0,1,0) [[ string attribute="NORMAL"]],
|
|
m_outputVertexDesc.addElement("v_normal", davinci::VET_NORMAL, davinci::CT_Float32, davinci::DT_Vec3);
|
|
//output float v_uv[2] = {0, 0} [[ string attribute="TEXCOORD_0"]],
|
|
m_outputVertexDesc.addElement("v_uv", davinci::VET_TEXCOORD_0, davinci::CT_Float32, davinci::DT_Vec2);
|
|
//output vector v_toCamera = vector(0, 0, 0),
|
|
m_outputVertexDesc.addElement("v_toCamera", davinci::VET_UNKNOWN, davinci::CT_Float32, davinci::DT_Vec3);
|
|
//output vector v_toLight = vector(0,0,0)
|
|
m_outputVertexDesc.addElement("v_toLight", davinci::VET_UNKNOWN, davinci::CT_Float32, davinci::DT_Vec3);
|
|
|
|
m_outputVertexDesc.addElement("v_tangent", davinci::VET_TANGENT, davinci::CT_Float32, davinci::DT_Vec3);
|
|
m_outputVertexDesc.addElement("v_binormal", davinci::VET_BINORMAL, davinci::CT_Float32, davinci::DT_Vec3);
|
|
m_outputVertexDesc.addElement("v_wposition", davinci::VET_UNKNOWN, davinci::CT_Float32, davinci::DT_Vec3);
|
|
}
|
|
|
|
TestVertexShader::~TestVertexShader()
|
|
{
|
|
|
|
}
|
|
|
|
const char* TestVertexShader::getName(void) const
|
|
{
|
|
return "test";
|
|
}
|
|
|
|
size_t TestVertexShader::getOutputVertexByteSize(void) const
|
|
{
|
|
return m_outputVertexDesc.vertexByteSize();
|
|
}
|
|
|
|
const davinci::VertexDesc& TestVertexShader::getOutputVertexDesc() const
|
|
{
|
|
return m_outputVertexDesc;
|
|
}
|
|
|
|
void TestVertexShader::prepare(const davinci::RenderPipe* pipe, SetInputAttributeFunc setInputAttributeFunc)
|
|
{
|
|
m_pipe = pipe;
|
|
|
|
//point a_position = vector(0, 0, 0) [[ string attribute="POSITION"]],
|
|
m_POSITION = setInputAttributeFunc(davinci::VET_POSITION);
|
|
//vector a_normal = vector(0, 1, 0) [[ string attribute="NORMAL"]],
|
|
m_NORMAL = setInputAttributeFunc(davinci::VET_NORMAL);
|
|
//float a_uv[2] = {0,0} [[ string attribute="TEXCOORD_0"]],
|
|
m_TEXCOORD_0 = setInputAttributeFunc(davinci::VET_TEXCOORD_0);
|
|
//
|
|
m_TANGENT = setInputAttributeFunc(davinci::VET_TANGENT);
|
|
}
|
|
|
|
void TestVertexShader::process(size_t index, void* outputVertex, davinci::fVector3& vertexWorldPos)
|
|
{
|
|
static const float32_t DEFAULT_POSITION[] = {0.f, 0.f, 0.f};
|
|
static const float32_t DEFAULT_NORMAL[] = { 0.f, 0.f, 0.f };
|
|
static const float32_t DEFAULT_TEXCOORD0[] = { 0.f, 0.f};
|
|
static const float32_t DEFAULT_TANGENT[] = { 1.f, 0.f, 0.f };
|
|
|
|
const davinci::UniformBuffer& uniformBuffer = m_pipe->getUniformBuffer();
|
|
|
|
davinci::fVector3 a_position(m_POSITION ? (const float32_t*)(m_POSITION->getElement(index)) : DEFAULT_POSITION);
|
|
|
|
davinci::fVector3 a_normal(m_NORMAL ? (const float32_t*)(m_NORMAL->getElement(index)) : DEFAULT_NORMAL);
|
|
|
|
davinci::fVector2 a_uv(m_TEXCOORD_0 ? (const float32_t*)(m_TEXCOORD_0->getElement(index)) : DEFAULT_TEXCOORD0);
|
|
|
|
davinci::fVector3 a_tangent(m_TANGENT ? (const float32_t*)(m_TANGENT->getElement(index)) : DEFAULT_TANGENT);
|
|
|
|
davinci::fMatrix4 mat_world = davinci::fMatrix4::IDENTITY;
|
|
if (davinci::fMatrix4* p_mat_world = (davinci::fMatrix4*)(uniformBuffer.getAttribute("self:world_transform")))
|
|
{
|
|
mat_world = *p_mat_world;
|
|
}
|
|
|
|
davinci::fMatrix4 mat_camera_view_proj = davinci::fMatrix4::IDENTITY;
|
|
if (davinci::fMatrix4* p_mat_camera_view_proj = (davinci::fMatrix4*)(uniformBuffer.getAttribute("camera:view_proj_mat")))
|
|
{
|
|
mat_camera_view_proj = *p_mat_camera_view_proj;
|
|
}
|
|
|
|
davinci::fVector3 camera_position = davinci::fVector3::ZERO;
|
|
if (davinci::fVector3* p_camera_position = (davinci::fVector3*)(uniformBuffer.getAttribute("camera:eye_pos")))
|
|
{
|
|
camera_position = *p_camera_position;
|
|
}
|
|
|
|
davinci::fVector3 light0_position = davinci::fVector3::ZERO;
|
|
if (davinci::fVector3* p_light0_position = (davinci::fVector3*)(uniformBuffer.getAttribute("light0:position")))
|
|
{
|
|
light0_position = *p_light0_position;
|
|
}
|
|
|
|
davinci::fVector3 temp1 = mat_world * a_position;
|
|
davinci::fVector3 v_position = mat_camera_view_proj * temp1;
|
|
|
|
davinci::fMatrix4 matNormal = davinci::fMatrix4::IDENTITY;
|
|
if (davinci::fMatrix4* p_mat_normal = (davinci::fMatrix4*)(uniformBuffer.getAttribute("self:world_transform_inverse_trans")))
|
|
{
|
|
matNormal = *p_mat_normal;
|
|
}
|
|
davinci::fVector3 v_normal = (matNormal * davinci::fVector4(a_normal, 0)).xyz().normalise();
|
|
davinci::fVector3 v_tangent = (matNormal * davinci::fVector4(a_tangent, 0)).xyz().normalise();
|
|
davinci::fVector3 v_binormal = v_normal.crossProduct(v_tangent).normalise();
|
|
|
|
davinci::fMatrix3 tbnMatrix = davinci::fMatrix3(
|
|
v_tangent.x, v_binormal.x, v_normal.x,
|
|
v_tangent.y, v_binormal.y, v_normal.y,
|
|
v_tangent.z, v_binormal.z, v_normal.z);
|
|
|
|
davinci::fVector3 v_to_camera = (camera_position - temp1).normalise() * tbnMatrix;
|
|
davinci::fVector3 v_to_light = (light0_position - temp1).normalise() * tbnMatrix;
|
|
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(0)->offset, &v_position, sizeof(davinci::fVector3));
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(1)->offset, &v_normal, sizeof(davinci::fVector3));
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(2)->offset, &a_uv, sizeof(float32_t) * 2);
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(3)->offset, &v_to_camera, sizeof(davinci::fVector3));
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(4)->offset, &v_to_light, sizeof(davinci::fVector3));
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(5)->offset, &v_tangent, sizeof(davinci::fVector3));
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(6)->offset, &v_binormal, sizeof(davinci::fVector3));
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(7)->offset, &temp1, sizeof(davinci::fVector3));
|
|
|
|
vertexWorldPos = mat_world * a_position;
|
|
}
|
|
|
|
|
|
StandardVertexShader::StandardVertexShader()
|
|
{
|
|
//output point v_position = point(0, 0, 0) [[ string attribute="POSITION"]],
|
|
m_outputVertexDesc.addElement("v_position", davinci::VET_POSITION, davinci::CT_Float32, davinci::DT_Vec3);
|
|
}
|
|
|
|
StandardVertexShader::~StandardVertexShader()
|
|
{
|
|
|
|
}
|
|
|
|
const char* StandardVertexShader::getName(void) const
|
|
{
|
|
return "standard";
|
|
}
|
|
|
|
size_t StandardVertexShader::getOutputVertexByteSize(void) const
|
|
{
|
|
return m_outputVertexDesc.vertexByteSize();
|
|
}
|
|
|
|
const davinci::VertexDesc& StandardVertexShader::getOutputVertexDesc() const
|
|
{
|
|
return m_outputVertexDesc;
|
|
}
|
|
|
|
void StandardVertexShader::prepare(const davinci::RenderPipe* pipe, SetInputAttributeFunc setInputAttributeFunc)
|
|
{
|
|
m_pipe = pipe;
|
|
|
|
//point a_position = vector(0, 0, 0) [[ string attribute="POSITION"]],
|
|
m_POSITION = setInputAttributeFunc(davinci::VET_POSITION);
|
|
}
|
|
|
|
void StandardVertexShader::process(size_t index, void* outputVertex, davinci::fVector3& vertexWorldPos)
|
|
{
|
|
static const float DEFAULT_POSITION[] = {0.f, 0.f, 0.f};
|
|
|
|
const davinci::UniformBuffer& uniformBuffer = m_pipe->getUniformBuffer();
|
|
|
|
davinci::fVector3 a_position(m_POSITION ? (const float32_t*)(m_POSITION->getElement(index)) : DEFAULT_POSITION);
|
|
|
|
davinci::fMatrix4 mat_world = davinci::fMatrix4::IDENTITY;
|
|
if (davinci::fMatrix4* p_mat_world = (davinci::fMatrix4*)(uniformBuffer.getAttribute("self:world_transform")))
|
|
{
|
|
mat_world = *p_mat_world;
|
|
}
|
|
|
|
davinci::fMatrix4 mat_camera_view_proj = davinci::fMatrix4::IDENTITY;
|
|
if (davinci::fMatrix4* p_mat_camera_view_proj = (davinci::fMatrix4*)(uniformBuffer.getAttribute("camera:view_proj_mat")))
|
|
{
|
|
mat_camera_view_proj = *p_mat_camera_view_proj;
|
|
}
|
|
|
|
davinci::fVector3 temp1 = mat_world * a_position;
|
|
davinci::fVector3 v_position = mat_camera_view_proj * temp1;
|
|
|
|
memcpy((char*)outputVertex + m_outputVertexDesc.getElement(0)->offset, &v_position, sizeof(davinci::fVector3));
|
|
|
|
vertexWorldPos = mat_world * a_position;
|
|
}
|