Bert Peters
7fdce48259
MultiImageLayer and FlatLayer now use the same code for arranging nodes, rather than then same logic twice.
60 lines
1.9 KiB
C++
60 lines
1.9 KiB
C++
#include <glog/logging.h>
|
|
#include "MultiImageVisualisation.hpp"
|
|
#include "glutils.hpp"
|
|
#include "Range.hpp"
|
|
|
|
using namespace fmri;
|
|
using namespace std;
|
|
|
|
MultiImageVisualisation::MultiImageVisualisation(const fmri::LayerData &layer)
|
|
{
|
|
auto dimensions = layer.shape();
|
|
CHECK_EQ(4, dimensions.size()) << "Should be image-like layer";
|
|
|
|
const auto images = dimensions[0],
|
|
channels = dimensions[1],
|
|
width = dimensions[2],
|
|
height = dimensions[3];
|
|
|
|
CHECK_EQ(1, images) << "Only single input image is supported" << endl;
|
|
|
|
texture = loadTexture(layer.data(), width, channels * height, channels);
|
|
initNodePositions<Ordering::SQUARE>(channels, 3);
|
|
|
|
vertexBuffer = std::make_unique<float[]>(channels * BASE_VERTICES.size());
|
|
texCoordBuffer = std::make_unique<float[]>(channels * 2u * BASE_VERTICES.size() / 3);
|
|
|
|
auto v = 0;
|
|
|
|
for (auto i : Range(channels)) {
|
|
const auto& nodePos = &nodePositions_[3 * i];
|
|
for (auto j : Range(BASE_VERTICES.size())) {
|
|
vertexBuffer[v++] = nodePos[j % 3] + BASE_VERTICES[j];
|
|
}
|
|
|
|
const float textureCoords[] = {
|
|
1, (i + 1) / (float) channels,
|
|
1, i / (float) channels,
|
|
0, i / (float) channels,
|
|
0, (i + 1) / (float) channels,
|
|
};
|
|
|
|
memcpy(texCoordBuffer.get() + 8 * i, textureCoords, sizeof(textureCoords));
|
|
}
|
|
}
|
|
|
|
void MultiImageVisualisation::render()
|
|
{
|
|
glEnableClientState(GL_TEXTURE_COORD_ARRAY);
|
|
glEnableClientState(GL_VERTEX_ARRAY);
|
|
glEnable(GL_TEXTURE_2D);
|
|
glTexEnvf(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_REPLACE);
|
|
texture.bind(GL_TEXTURE_2D);
|
|
glTexCoordPointer(2, GL_FLOAT, 0, texCoordBuffer.get());
|
|
glVertexPointer(3, GL_FLOAT, 0, vertexBuffer.get());
|
|
glDrawArrays(GL_QUADS, 0, nodePositions_.size() / 3 * 4);
|
|
glDisable(GL_TEXTURE_2D);
|
|
glDisableClientState(GL_VERTEX_ARRAY);
|
|
glDisableClientState(GL_TEXTURE_COORD_ARRAY);
|
|
}
|