implement label drawable
This commit is contained in:
ohyzha
257
openVulkanoCpp/Scene/Prefabs/LabelDrawable.cpp
Normal file
257
openVulkanoCpp/Scene/Prefabs/LabelDrawable.cpp
Normal file
@@ -0,0 +1,257 @@
|
||||
/*
|
||||
* This Source Code Form is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
||||
*/
|
||||
|
||||
#include "LabelDrawable.hpp"
|
||||
#include "Scene/TextDrawable.hpp"
|
||||
#include "Scene/DrawEncoder.hpp"
|
||||
#include "Scene/IFontAtlasGenerator.hpp"
|
||||
#include "Base/Logger.hpp"
|
||||
#include <optional>
|
||||
|
||||
namespace OpenVulkano::Scene
|
||||
{
|
||||
using namespace Math;
|
||||
|
||||
LabelDrawable::LabelDrawable(const LabelDrawableSettings& settings, bool isBillboard)
|
||||
: Drawable(DrawEncoder::GetDrawEncoder<LabelDrawable>(), DrawPhase::MAIN)
|
||||
{
|
||||
m_isBillboard = isBillboard;
|
||||
SetLabelSettings(settings);
|
||||
SetupShaders();
|
||||
SetupBuffers();
|
||||
m_backgroundGeometry.SetFreeAfterUpload(false);
|
||||
}
|
||||
|
||||
void LabelDrawable::SetLabelSettings(const LabelDrawableSettings& settings)
|
||||
{
|
||||
m_settings = settings;
|
||||
m_labelData.hasRoundedCorners = settings.hasRoundedCorners;
|
||||
m_labelData.hasArrow = settings.hasArrow;
|
||||
m_labelData.cornerRadius = settings.cornerRadius;
|
||||
m_labelData.arrowLength = settings.arrowLength;
|
||||
}
|
||||
|
||||
void LabelDrawable::AddText(TextDrawable* td, const std::string& text)
|
||||
{
|
||||
if (!td || text.empty())
|
||||
{
|
||||
return;
|
||||
}
|
||||
if (!td->GetShader())
|
||||
{
|
||||
Logger::RENDER->error("Text drawable shader is null, cannot add text to label drawable\n");
|
||||
return;
|
||||
}
|
||||
|
||||
TextDrawable& textDrawable = m_texts.emplace_back(td->GetConfig());
|
||||
// do not render glyph's background
|
||||
textDrawable.GetConfig().backgroundColor.a = 0;
|
||||
if (m_isBillboard)
|
||||
{
|
||||
textDrawable.SetShader(GetBillboardTextShader(td->GetShader()));
|
||||
}
|
||||
else
|
||||
{
|
||||
textDrawable.SetShader(td->GetShader());
|
||||
}
|
||||
double lineHeight;
|
||||
if (textDrawable.GetFontAtlasGenerator())
|
||||
{
|
||||
textDrawable.SetFontAtlasGenerator(td->GetFontAtlasGenerator());
|
||||
lineHeight = td->GetFontAtlasGenerator()->GetAtlasMetadata().lineHeight;
|
||||
}
|
||||
else
|
||||
{
|
||||
textDrawable.SetAtlasData(td->GetAtlasData());
|
||||
lineHeight = td->GetAtlasData()->meta.lineHeight;
|
||||
}
|
||||
textDrawable.GenerateText(text, m_position);
|
||||
RecalculateBbox(textDrawable.GetBoundingBox());
|
||||
// update position for next text entry
|
||||
m_position.y = m_bbox.GetMin().y - lineHeight;
|
||||
|
||||
if (!m_settings.hasArrow)
|
||||
{
|
||||
if (m_backgroundGeometry.vertexCount != 4)
|
||||
{
|
||||
m_backgroundGeometry.Init(4, 6);
|
||||
uint32_t indices[6] = { 0, 1, 2, 0, 2, 3 };
|
||||
m_backgroundGeometry.SetIndices(indices, 6);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (m_backgroundGeometry.vertexCount != 7)
|
||||
{
|
||||
m_backgroundGeometry.Init(7, 9);
|
||||
uint32_t indices[9] = { 0, 1, 2, 0, 2, 3, 4, 5, 6 };
|
||||
m_backgroundGeometry.SetIndices(indices, 9);
|
||||
}
|
||||
}
|
||||
|
||||
const auto& min = m_bbox.GetMin();
|
||||
const auto& max = m_bbox.GetMax();
|
||||
Vertex v, v2, v3, v4;
|
||||
v.color = v2.color = v3.color = v4.color = m_settings.backgroundColor;
|
||||
const float offset = 0.001;
|
||||
v.position = Vector3f(min.x - m_settings.horizontalOffset, min.y - m_settings.verticalOffset, min.z - offset);
|
||||
v2.position = Vector3f(max.x + m_settings.horizontalOffset, min.y - m_settings.verticalOffset, min.z - offset);
|
||||
v3.position = Vector3f(max.x + m_settings.horizontalOffset, max.y + m_settings.verticalOffset, min.z - offset);
|
||||
v4.position = Vector3f(min.x - m_settings.horizontalOffset, max.y + m_settings.verticalOffset, min.z - offset);
|
||||
if (!m_settings.hasArrow)
|
||||
{
|
||||
v.textureCoordinates = Vector3f(0, 0, 0);
|
||||
v2.textureCoordinates = Vector3f(1, 0, 0);
|
||||
v3.textureCoordinates = Vector3f(1, 1, 0);
|
||||
v4.textureCoordinates = Vector3f(0, 1, 0);
|
||||
}
|
||||
else
|
||||
{
|
||||
const float w = v2.position.x - v.position.x;
|
||||
const float h = v3.position.y - v2.position.y + m_settings.arrowLength;
|
||||
v.textureCoordinates = Vector3f(0, v.position.y / h, 0);
|
||||
v2.textureCoordinates = Vector3f(1, v2.position.y / h, 0);
|
||||
v3.textureCoordinates = Vector3f(1, 1, 0);
|
||||
v4.textureCoordinates = Vector3f(0, 1, 0);
|
||||
// arrow vertices
|
||||
Vertex a, b, c;
|
||||
a.position = Vector3f(v.position.x + w / 3, v.position.y, v.position.z);
|
||||
b.position = Vector3f(v.position.x + w / 2, v.position.y - m_settings.arrowLength, v.position.z);
|
||||
c.position = Vector3f(v2.position.x - w / 3, v.position.y, v.position.z);
|
||||
a.color = b.color = c.color = m_settings.backgroundColor;
|
||||
a.textureCoordinates = Vector3f(a.position.x / w, a.position.y / h, 0);
|
||||
b.textureCoordinates = Vector3f(b.position.x / w, 0, 0);
|
||||
c.textureCoordinates = Vector3f(c.position.x / w, c.position.y / h, 0);
|
||||
m_backgroundGeometry.vertices[4] = a;
|
||||
m_backgroundGeometry.vertices[5] = b;
|
||||
m_backgroundGeometry.vertices[6] = c;
|
||||
}
|
||||
m_backgroundGeometry.vertices[0] = v;
|
||||
m_backgroundGeometry.vertices[1] = v2;
|
||||
m_backgroundGeometry.vertices[2] = v3;
|
||||
m_backgroundGeometry.vertices[3] = v4;
|
||||
|
||||
if (m_settings.hasRoundedCorners)
|
||||
{
|
||||
m_labelData.textSize.x = v2.position.x - v.position.x;
|
||||
m_labelData.textSize.y = v3.position.y - v.position.y;
|
||||
}
|
||||
}
|
||||
|
||||
void LabelDrawable::SetBillboardSettings(const BillboardControlBlock& settings)
|
||||
{
|
||||
m_billboardSettings = settings;
|
||||
}
|
||||
|
||||
void LabelDrawable::RecalculateBbox(const Math::AABB& other)
|
||||
{
|
||||
if (m_bbox.IsEmpty())
|
||||
{
|
||||
m_bbox = other;
|
||||
}
|
||||
else
|
||||
{
|
||||
auto& currentMin = m_bbox.GetMin();
|
||||
auto& currentMax = m_bbox.GetMax();
|
||||
currentMin.x = std::min(currentMin.x, other.min.x);
|
||||
currentMin.y = std::min(currentMin.y, other.min.y);
|
||||
currentMax.x = std::max(currentMax.x, other.max.x);
|
||||
currentMax.y = std::max(currentMax.y, other.max.y);
|
||||
}
|
||||
}
|
||||
|
||||
// if we have want our label to be billboard, then we'll create new shader and set it's vertex shader to billboard.vert
|
||||
// to avoid modifying input shader
|
||||
Shader* LabelDrawable::GetBillboardTextShader(Shader* src)
|
||||
{
|
||||
auto GetFragmentShaderName = [](const Shader& shaderToCheck) -> std::optional<std::string>
|
||||
{
|
||||
const auto& shaderPrograms = shaderToCheck.shaderPrograms;
|
||||
auto pos = std::find_if(shaderPrograms.begin(), shaderPrograms.end(),
|
||||
[](const ShaderProgram& program)
|
||||
{ return program.type == ShaderProgramType::FRAGMENT; });
|
||||
return pos != shaderPrograms.end() ? std::optional<std::string>(pos->name) : std::nullopt;
|
||||
};
|
||||
|
||||
std::optional<std::string> inputFragmentName = GetFragmentShaderName(*src);
|
||||
if (!inputFragmentName)
|
||||
{
|
||||
throw std::runtime_error("Shader doesn't have fragment program");
|
||||
}
|
||||
|
||||
bool exists = false;
|
||||
int shaderIdx = 0;
|
||||
for (; shaderIdx < m_billboardTextShaders.size(); shaderIdx++)
|
||||
{
|
||||
std::optional<std::string> name = GetFragmentShaderName(m_billboardTextShaders[shaderIdx]);
|
||||
if (name.value() == inputFragmentName.value())
|
||||
{
|
||||
exists = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!exists)
|
||||
{
|
||||
Shader& shader = m_billboardTextShaders.emplace_back();
|
||||
shaderIdx = m_billboardTextShaders.size() - 1;
|
||||
shader.AddShaderProgram(OpenVulkano::ShaderProgramType::VERTEX, "Shader/billboard");
|
||||
shader.AddShaderProgram(OpenVulkano::ShaderProgramType::FRAGMENT, inputFragmentName.value());
|
||||
shader.AddVertexInputDescription(OpenVulkano::Vertex::GetVertexInputDescription());
|
||||
shader.AddDescriptorSetLayoutBinding(Texture::DESCRIPTOR_SET_LAYOUT_BINDING);
|
||||
DescriptorSetLayoutBinding binding = UniformBuffer::DESCRIPTOR_SET_LAYOUT_BINDING;
|
||||
binding.stageFlags = ShaderProgramType::Type::VERTEX;
|
||||
DescriptorSetLayoutBinding binding2 = UniformBuffer::DESCRIPTOR_SET_LAYOUT_BINDING;
|
||||
binding2.stageFlags = ShaderProgramType::FRAGMENT;
|
||||
shader.AddDescriptorSetLayoutBinding(binding2, 3);
|
||||
shader.AddDescriptorSetLayoutBinding(binding, 4);
|
||||
shader.depthWrite = false;
|
||||
shader.depthCompareOp = CompareOp::LESS_OR_EQUAL;
|
||||
shader.alphaBlend = true;
|
||||
shader.cullMode = CullMode::NONE;
|
||||
}
|
||||
return &m_billboardTextShaders[shaderIdx];
|
||||
}
|
||||
|
||||
void LabelDrawable::SetupShaders()
|
||||
{
|
||||
DescriptorSetLayoutBinding binding = UniformBuffer::DESCRIPTOR_SET_LAYOUT_BINDING;
|
||||
if (!m_isBillboard)
|
||||
{
|
||||
m_backgroundShader.AddShaderProgram(OpenVulkano::ShaderProgramType::VERTEX, "Shader/basic");
|
||||
}
|
||||
else
|
||||
{
|
||||
m_backgroundShader.AddShaderProgram(OpenVulkano::ShaderProgramType::VERTEX, "Shader/billboard");
|
||||
// binding for billboard's buffer
|
||||
binding.stageFlags = ShaderProgramType::Type::VERTEX;
|
||||
m_backgroundShader.AddDescriptorSetLayoutBinding(binding, 4);
|
||||
}
|
||||
m_backgroundShader.AddShaderProgram(OpenVulkano::ShaderProgramType::FRAGMENT, "Shader/label");
|
||||
m_backgroundShader.AddVertexInputDescription(OpenVulkano::Vertex::GetVertexInputDescription());
|
||||
m_backgroundShader.AddDescriptorSetLayoutBinding(Texture::DESCRIPTOR_SET_LAYOUT_BINDING, 2);
|
||||
binding.stageFlags = ShaderProgramType::Type::FRAGMENT;
|
||||
m_backgroundShader.AddDescriptorSetLayoutBinding(binding, 5);
|
||||
m_backgroundShader.cullMode = CullMode::NONE;
|
||||
SetShader(&m_backgroundShader);
|
||||
}
|
||||
|
||||
void LabelDrawable::SetupBuffers()
|
||||
{
|
||||
DescriptorSetLayoutBinding binding = UniformBuffer::DESCRIPTOR_SET_LAYOUT_BINDING;
|
||||
binding.stageFlags = ShaderProgramType::Type::VERTEX;
|
||||
m_billboardBuffer.size = sizeof(BillboardControlBlock);
|
||||
m_billboardBuffer.data = &m_billboardSettings;
|
||||
m_billboardBuffer.setId = 4;
|
||||
m_billboardBuffer.binding = binding;
|
||||
|
||||
binding.stageFlags = ShaderProgramType::Type::FRAGMENT;
|
||||
m_labelBuffer.size = sizeof(LabelUniformData);
|
||||
m_labelBuffer.data = &m_labelData;
|
||||
m_labelBuffer.setId = 5;
|
||||
m_labelBuffer.binding = binding;
|
||||
}
|
||||
}
|
||||
83
openVulkanoCpp/Scene/Prefabs/LabelDrawable.hpp
Normal file
83
openVulkanoCpp/Scene/Prefabs/LabelDrawable.hpp
Normal file
@@ -0,0 +1,83 @@
|
||||
/*
|
||||
* This Source Code Form is subject to the terms of the Mozilla Public
|
||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "Base/Wrapper.hpp"
|
||||
#include "Scene/Drawable.hpp"
|
||||
#include "Scene/Shader/Shader.hpp"
|
||||
#include "Scene/Texture.hpp"
|
||||
#include "Scene/UniformBuffer.hpp"
|
||||
#include "Scene/Vertex.hpp"
|
||||
#include "Scene/BillboardControlBlock.hpp"
|
||||
#include "Math/AABB.hpp"
|
||||
#include "Scene/TextDrawable.hpp"
|
||||
#include <list>
|
||||
|
||||
namespace OpenVulkano::Scene
|
||||
{
|
||||
|
||||
struct LabelDrawableSettings
|
||||
{
|
||||
Math::Vector4f backgroundColor = { 1, 0, 0, 1 };
|
||||
float horizontalOffset = 0.05f;
|
||||
float verticalOffset = 0.05f;
|
||||
float cornerRadius = 0.5f;
|
||||
float arrowLength = 0.5f;
|
||||
int32_t hasRoundedCorners = false;
|
||||
int32_t hasArrow = false;
|
||||
};
|
||||
|
||||
namespace
|
||||
{
|
||||
struct LabelUniformData
|
||||
{
|
||||
Math::Vector2f textSize = { 0, 0 };
|
||||
float cornerRadius = 0.f;
|
||||
float arrowLength = 0.f;
|
||||
int32_t hasRoundedCorners = false;
|
||||
int32_t hasArrow = false;
|
||||
};
|
||||
};
|
||||
|
||||
class LabelDrawable final : public Drawable
|
||||
{
|
||||
public:
|
||||
LabelDrawable(const LabelDrawableSettings& settings = LabelDrawableSettings(), bool isBillboard = false);
|
||||
void AddText(TextDrawable* td, const std::string& text);
|
||||
void SetLabelSettings(const LabelDrawableSettings& settings);
|
||||
void SetBillboardSettings(const BillboardControlBlock& settings);
|
||||
void SetPosition(const Math::Vector3f& pos) { m_position = pos; }
|
||||
std::list<TextDrawable>& GetTexts() { return m_texts; }
|
||||
LabelDrawableSettings& GetSettings() { return m_settings; }
|
||||
UniformBuffer* GetBillboardBuffer() { return &m_billboardBuffer; }
|
||||
UniformBuffer* GetLabelBuffer() { return &m_labelBuffer; }
|
||||
Geometry& GetBackgroundGeometry() { return m_backgroundGeometry; }
|
||||
BillboardControlBlock& GetBillboardSettings() { return m_billboardSettings; }
|
||||
Math::Vector3f& GetPosition() { return m_position; }
|
||||
bool IsBillboard() const { return m_isBillboard; }
|
||||
const Math::AABB& GetBoundingBox() const { return m_bbox; }
|
||||
private:
|
||||
void RecalculateBbox(const Math::AABB& other);
|
||||
Shader* GetBillboardTextShader(Shader* src);
|
||||
void SetupShaders();
|
||||
void SetupBuffers();
|
||||
private:
|
||||
Shader m_backgroundShader;
|
||||
Geometry m_backgroundGeometry;
|
||||
UniformBuffer m_billboardBuffer;
|
||||
UniformBuffer m_labelBuffer;
|
||||
std::vector<Shader> m_billboardTextShaders;
|
||||
// list over vector to prevent memory reallocation and crash
|
||||
std::list<TextDrawable> m_texts;
|
||||
LabelDrawableSettings m_settings;
|
||||
LabelUniformData m_labelData;
|
||||
BillboardControlBlock m_billboardSettings;
|
||||
Math::Vector3f m_position = { 0, 0, 0 };
|
||||
Math::AABB m_bbox;
|
||||
bool m_isBillboard;
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user