obs-StreamFX/source/gs-mipmapper.cpp

160 lines
4.5 KiB
C++
Raw Normal View History

/*
* Modern effects for a modern Streamer
* Copyright (C) 2017 Michael Fabian Dirks
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include "gs-mipmapper.h"
extern "C" {
#pragma warning (push)
#pragma warning (disable: 4201)
#include "libobs/obs.h"
#include "libobs/graphics/graphics.h"
#pragma warning (pop)
}
#define MAX_LAYERS 16
GS::MipMapper::MipMapper() : m_rt(GS_RGBA, GS_ZS_NONE), m_vb(65535) {
// Sub layers start here now.
double_t size = 0.5, offset = 0.0;
for (size_t layer = 0; layer < MAX_LAYERS; layer++) {
// Large Block
if (layer == 0) {
{
GS::Vertex& v = m_vb.at(layer * 12);
vec3_set(&v.position, offset, offset, 0);
vec4_set(&v.uv[0], 0, 0, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 1);
vec3_set(&v.position, offset + size, offset, 0);
vec4_set(&v.uv[0], 1, 0, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 2);
vec3_set(&v.position, offset + size, offset + size, 0);
vec4_set(&v.uv[0], 1, 1, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 3);
vec3_set(&v.position, offset, offset + size, 0);
vec4_set(&v.uv[0], 0, 1, 0, 0);
}
} else {
{
GS::Vertex& v = m_vb.at(layer * 12);
vec3_set(&v.position, offset, offset, 0);
vec4_set(&v.uv[0], offset - size * 2, offset - size * 2, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 1);
vec3_set(&v.position, offset + size, offset, 0);
vec4_set(&v.uv[0], offset, offset - size * 2, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 2);
vec3_set(&v.position, offset + size, offset + size, 0);
vec4_set(&v.uv[0], offset, offset, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 3);
vec3_set(&v.position, offset, offset + size, 0);
vec4_set(&v.uv[0], offset - size * 2, offset, 0, 0);
}
}
if (layer != 0) {
// Horizontal Slice
{
GS::Vertex& v = m_vb.at(layer * 12 + 4);
vec3_set(&v.position, 0, offset, 0);
vec4_set(&v.uv[0], 0, offset - size, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 5);
vec3_set(&v.position, size, offset, 0);
vec4_set(&v.uv[0], .5, offset - size, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 6);
vec3_set(&v.position, size, offset + size, 0);
vec4_set(&v.uv[0], .5, offset, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 7);
vec3_set(&v.position, 0, offset + size, 0);
vec4_set(&v.uv[0], 0, offset, 0, 0);
}
// Vertical Slice
{
GS::Vertex& v = m_vb.at(layer * 12 + 8);
vec3_set(&v.position, offset, 0, 0);
vec4_set(&v.uv[0], offset - size, 0, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 9);
vec3_set(&v.position, offset + size, 0, 0);
vec4_set(&v.uv[0], offset, 0, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 10);
vec3_set(&v.position, offset + size, size, 0);
vec4_set(&v.uv[0], offset, 0.5, 0, 0);
}
{
GS::Vertex& v = m_vb.at(layer * 12 + 11);
vec3_set(&v.position, offset, size, 0);
vec4_set(&v.uv[0], offset - size, 0.5, 0, 0);
}
}
offset += size;
size /= 2.0;
}
}
GS::MipMapper::~MipMapper() {}
gs_texture_t* GS::MipMapper::Render(gs_texture_t* intex) {
// Get Texture Size
uint32_t width = gs_texture_get_width(intex),
height = gs_texture_get_height(intex);
// Calculate total mipmap layers.
uint32_t layers = 1, size = height;
if (width > height) size = width;
layers = static_cast<uint32_t>(ceil(log(double_t(size)) / log(2.0)));
size = pow(2, layers);
// Render
m_buf = GS::Texture(size, size, GS_RGBA, 1, nullptr, GS_DYNAMIC);
for (uint32_t layer = 0; layer < layers; layer++) {
try {
GS::RenderTargetOp op = m_rt.Render(size * 2, size * 2);
gs_load_vertexbuffer(m_vb.get());
gs_load_indexbuffer(nullptr);
gs_draw(gs_draw_mode::GS_TRIS, layer * 12, 12);
} catch (...) {
return intex;
}
gs_copy_texture(m_buf.GetObject(), m_rt.GetTextureObject());
}
return m_buf.GetObject();
}