aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--voronoisplat.cpp101
-rw-r--r--voronoisplat.h1
2 files changed, 53 insertions, 49 deletions
diff --git a/voronoisplat.cpp b/voronoisplat.cpp
index 173dcc5..dc37a55 100644
--- a/voronoisplat.cpp
+++ b/voronoisplat.cpp
@@ -17,24 +17,7 @@ VoronoiSplatRenderer::VoronoiSplatRenderer(const QSize &size)
, m_size(size)
{
setupShaders();
-
- // sitesVAO: VBOs 0 & 1 are for sites & their values (init'd later)
- m_sitesVAO.create();
- gl.glGenBuffers(3, m_VBOs);
-
- // 2ndPassVAO: VBO 2 is a quad mapping the final texture to the framebuffer
- m_2ndPassVAO.create();
- m_2ndPassVAO.bind();
- GLfloat verts[] = {-1.0f, -1.0f, -1.0f, 1.0f,
- 1.0f, -1.0f, 1.0f, 1.0f};
- gl.glBindBuffer(GL_ARRAY_BUFFER, m_VBOs[2]);
- gl.glBufferData(GL_ARRAY_BUFFER, sizeof(verts), verts, GL_STATIC_DRAW);
-
- int vertAttrib = m_program2->attributeLocation("vert");
- gl.glVertexAttribPointer(vertAttrib, 2, GL_FLOAT, GL_FALSE, 0, 0);
- gl.glEnableVertexAttribArray(vertAttrib);
- m_2ndPassVAO.release();
-
+ setupVAOs();
setupTextures();
}
@@ -46,7 +29,7 @@ R"EOF(#version 440
uniform float rad_blur;
uniform float rad_max;
-uniform float fbSize;
+uniform float fb_size;
in vec2 vert;
in float scalar;
@@ -55,16 +38,16 @@ out float value;
void main() {
gl_PointSize = (rad_max + rad_blur) * 2.0;
- gl_Position = vec4(2.0 * (vert / fbSize) - 1.0, 0, 1);
+ gl_Position = vec4(2.0 * vert / fb_size - 1.0, 0, 1.0);
value = scalar;
}
)EOF");
m_program1->addShaderFromSourceCode(QOpenGLShader::Fragment,
R"EOF(#version 440
-uniform sampler2D siteDT;
uniform float rad_blur;
uniform float rad_max;
+uniform sampler2D siteDT;
in float value;
@@ -72,6 +55,8 @@ layout (location = 0) out vec4 fragColor;
void main() {
float dt = texelFetch(siteDT, ivec2(gl_FragCoord.xy), 0).r;
+ fragColor = vec4(0.25, 0.5, 0, 0);
+ return;
if (dt > rad_max)
discard;
else {
@@ -100,7 +85,6 @@ void main() {
gl_Position = vec4(vert, 0, 1);
}
)EOF");
-
m_program2->addShaderFromSourceCode(QOpenGLShader::Fragment,
R"EOF(
#version 440
@@ -113,7 +97,7 @@ uniform sampler2D colormap;
uniform float rad_max;
vec3 getRGB(float value) {
- return texture2D(colormap, vec2(0, 1)).rgb;
+ return texture2D(colormap, vec2(mix(0.005, 0.995, value), 1)).rgb;
}
void main() {
@@ -126,8 +110,6 @@ void main() {
float value = 0.0f;
if (accum.g > 1.0)
value = (accum.r - 1.0) / (accum.g - 1.0);
- else
- value = accum.r / accum.g;
fragColor.rgb = getRGB(value);
fragColor.a = 1.0 - dt / rad_max;
}
@@ -136,6 +118,38 @@ void main() {
m_program2->link();
}
+void VoronoiSplatRenderer::setupVAOs()
+{
+ gl.glGenBuffers(3, m_VBOs);
+
+ // sitesVAO: VBOs 0 & 1 are for sites & their values (init'd later)
+ m_sitesVAO.create();
+ m_sitesVAO.bind();
+ gl.glBindBuffer(GL_ARRAY_BUFFER, m_VBOs[0]);
+ int vertAttrib = m_program1->attributeLocation("vert");
+ gl.glVertexAttribPointer(vertAttrib, 2, GL_FLOAT, GL_FALSE, 0, 0);
+ gl.glEnableVertexAttribArray(vertAttrib);
+
+ gl.glBindBuffer(GL_ARRAY_BUFFER, m_VBOs[1]);
+ int valueAttrib = m_program1->attributeLocation("scalar");
+ gl.glVertexAttribPointer(valueAttrib, 1, GL_FLOAT, GL_FALSE, 0, 0);
+ gl.glEnableVertexAttribArray(valueAttrib);
+ m_sitesVAO.release();
+
+ // 2ndPassVAO: VBO 2 is a quad mapping the final texture to the framebuffer
+ m_2ndPassVAO.create();
+ m_2ndPassVAO.bind();
+ gl.glBindBuffer(GL_ARRAY_BUFFER, m_VBOs[2]);
+ GLfloat verts[] = {-1.0f, -1.0f, -1.0f, 1.0f,
+ 1.0f, -1.0f, 1.0f, 1.0f};
+ gl.glBufferData(GL_ARRAY_BUFFER, sizeof(verts), verts, GL_STATIC_DRAW);
+
+ vertAttrib = m_program2->attributeLocation("vert");
+ gl.glVertexAttribPointer(vertAttrib, 2, GL_FLOAT, GL_FALSE, 0, 0);
+ gl.glEnableVertexAttribArray(vertAttrib);
+ m_2ndPassVAO.release();
+}
+
void VoronoiSplatRenderer::setupTextures()
{
gl.glGenTextures(2, m_textures);
@@ -184,16 +198,10 @@ void VoronoiSplatRenderer::setSites(const arma::mat &points)
copyPoints(points);
// Update VBO with the new data
- m_sitesVAO.bind();
gl.glBindBuffer(GL_ARRAY_BUFFER, m_VBOs[0]);
gl.glBufferData(GL_ARRAY_BUFFER, m_sites.size() * sizeof(float),
m_sites.data(), GL_STATIC_DRAW);
- int vertAttrib = m_program1->attributeLocation("vert");
- gl.glVertexAttribPointer(vertAttrib, 2, GL_FLOAT, GL_FALSE, 0, 0);
- gl.glEnableVertexAttribArray(vertAttrib);
- m_sitesVAO.release();
-
// Compute DT values for the new positions
computeDT();
}
@@ -206,7 +214,7 @@ void VoronoiSplatRenderer::copyPoints(const arma::mat &points)
double maxY = points.col(1).max();
// Coords are tighly packed into 'm_sites' as [ (x1, y1), (x2, y2), ... ]
- m_sites.resize(points.n_rows * 2);
+ m_sites.resize(2*points.n_rows);
LinearScale<float> sx(minX, maxX, 1, m_size.width() - 1);
const double *col = points.colptr(0);
for (unsigned i = 0; i < points.n_rows; i++) {
@@ -224,9 +232,9 @@ void VoronoiSplatRenderer::computeDT()
{
int w = m_size.width(), h = m_size.height();
- std::vector<float> buf(w * h, 0.0f);
+ std::vector<float> buf(w*h, 0.0f);
for (unsigned i = 0; i < m_sites.size(); i += 2) {
- buf[int(m_sites[i + 1]) * h + int(m_sites[i])] = (float) i / 2 + 1;
+ buf[int(m_sites[i + 1])*h + int(m_sites[i])] = (float) i/2 + 1;
}
// Compute FT of the sites
@@ -234,15 +242,15 @@ void VoronoiSplatRenderer::computeDT()
// Compute DT of the sites (from the resident FT)
skelft2DDT(buf.data(), 0, 0, w, h);
- std::vector<GLfloat> dtTexData(w * h * 2, 0.0f);
+ std::vector<GLfloat> dtTexData(2*w*h, 0.0f);
for (unsigned i = 0; i < buf.size(); i++) {
dtTexData[2*i] = buf[i];
}
// Upload result to lookup texture
gl.glBindTexture(GL_TEXTURE_2D, m_textures[0]);
- gl.glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, m_size.width(), m_size.height(),
- GL_RG, GL_FLOAT, dtTexData.data());
+ gl.glTexSubImage2D(GL_TEXTURE_2D, 0, 0, 0, w, h, GL_RG, GL_FLOAT,
+ dtTexData.data());
}
void VoronoiSplatRenderer::setValues(const arma::vec &values)
@@ -253,18 +261,13 @@ void VoronoiSplatRenderer::setValues(const arma::vec &values)
}
m_values.resize(values.n_elem);
- std::copy(values.begin(), values.end(), m_values.begin());
+ LinearScale<float> scale(values.min(), values.max(), 0, 1.0f);
+ std::transform(values.begin(), values.end(), m_values.begin(), scale);
// Update VBO with the new data
- m_sitesVAO.bind();
gl.glBindBuffer(GL_ARRAY_BUFFER, m_VBOs[1]);
gl.glBufferData(GL_ARRAY_BUFFER, m_values.size() * sizeof(float),
m_values.data(), GL_DYNAMIC_DRAW);
-
- int valueAttrib = m_program1->attributeLocation("scalar");
- gl.glVertexAttribPointer(valueAttrib, 1, GL_FLOAT, GL_TRUE, 0, 0);
- gl.glEnableVertexAttribArray(valueAttrib);
- m_sitesVAO.release();
}
void VoronoiSplatRenderer::setColorMap(const ColorScale *scale)
@@ -276,12 +279,12 @@ void VoronoiSplatRenderer::setColorMap(const ColorScale *scale)
float t = scale->min();
float step = (scale->max() - scale->min()) / COLORMAP_SAMPLES;
qreal r, g, b;
- std::vector<float> cmap(COLORMAP_SAMPLES * 3); // R,G,B
- for (int i = 0; i < COLORMAP_SAMPLES * 3; i += 3) {
+ std::vector<float> cmap(3*COLORMAP_SAMPLES); // R,G,B
+ for (int i = 0; i < 3*COLORMAP_SAMPLES; i += 3) {
scale->color(t).getRgbF(&r, &g, &b);
- cmap[i + 0] = b;
+ cmap[i + 0] = r;
cmap[i + 1] = g;
- cmap[i + 2] = r;
+ cmap[i + 2] = b;
t += step;
}
@@ -327,7 +330,7 @@ void VoronoiSplatRenderer::render()
m_program1->bind();
m_program1->setUniformValue("rad_max", 20.0f);
m_program1->setUniformValue("rad_blur", RAD_BLUR);
- m_program1->setUniformValue("fbSize", float(m_size.width()));
+ m_program1->setUniformValue("fb_size", m_size.width());
gl.glActiveTexture(GL_TEXTURE0);
gl.glBindTexture(GL_TEXTURE_2D, m_textures[0]);
diff --git a/voronoisplat.h b/voronoisplat.h
index 4092da3..5913218 100644
--- a/voronoisplat.h
+++ b/voronoisplat.h
@@ -36,6 +36,7 @@ public:
private:
void setupShaders();
+ void setupVAOs();
void setupTextures();
void copyPoints(const arma::mat &points);
void computeDT();