mirror/OpenSpace
1
0
Fork 0
mirror of https://github.com/OpenSpace/OpenSpace.git synced 2026-01-31 08:29:04 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add test code to ffmpegtileprovider update function

Browse Source
This commit is contained in:
Ylva Selling
2022-09-07 03:49:01 -04:00
parent c1c35e0f2a
commit 66b32c2b3d
2 changed files with 94 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

87
modules/globebrowsing/src/tileprovider/ffmpegtileprovider.cpp
View File

@@ -143,6 +143,93 @@ void FfmpegTileProvider::update() {
_lastFrameTime = now;
// @TODO Do something with it!
_glFrame = av_frame_alloc();
int sz = av_image_get_buffer_size(
AV_PIX_FMT_RGB24,
_codecContext->width,
_codecContext->height,
1
);
uint8_t* internalBuffer = reinterpret_cast<uint8_t*>(av_malloc(sz * sizeof(uint8_t)));
av_image_fill_arrays(
_glFrame->data,
_glFrame->linesize,
internalBuffer,
AV_PIX_FMT_RGB24,
_codecContext->width,
_codecContext->height,
1
);
_packet = av_packet_alloc();
glGenTextures(1, &_frameTexture);
glBindTexture(GL_TEXTURE_2D, _frameTexture);
glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_REPEAT);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_REPEAT);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
glTexImage2D(
GL_TEXTURE_2D,
0,
GL_RGB,
_codecContext->width,
_codecContext->height,
0,
GL_RGB,
GL_UNSIGNED_BYTE,
nullptr
);
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
glPushAttrib(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
glEnable(GL_TEXTURE_2D);
glTexEnvi(GL_TEXTURE_ENV, GL_TEXTURE_ENV_MODE, GL_DECAL);
glBindTexture(GL_TEXTURE_2D, _frameTexture);
glFlush();
glDisable(GL_TEXTURE_2D);
///FBO
glGenFramebuffers(1, &_FBO);
glBindFramebuffer(GL_FRAMEBUFFER, _FBO);
//Attach 2D texture to this FBO
glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, _frameTexture, 0);
// Print to file
FILE* output_image;
int output_width, output_height;
output_width = _codecContext->width,
output_height = _codecContext->height;
/// READ THE PIXELS VALUES from FBO AND SAVE TO A .PPM FILE
int i, j, k;
unsigned char* pixels = (unsigned char*)malloc(output_width * output_height * 3);
/// READ THE CONTENT FROM THE FBO
glReadBuffer(GL_COLOR_ATTACHMENT0);
glReadPixels(0, 0, output_width, output_height, GL_RGB, GL_UNSIGNED_BYTE, pixels);
output_image = fopen("C:\\Users\\ylvaselling\\Documents\\Work\\Dataset\\output.ppm", "wt");
fprintf(output_image, "P3\n");
fprintf(output_image, "# Created by Ricao\n");
fprintf(output_image, "%d %d\n", output_width, output_height);
fprintf(output_image, "255\n");
k = 0;
for (i = 0; i < output_width; i++)
{
for (j = 0; j < output_height; j++)
{
fprintf(output_image, "%u %u %u ", (unsigned int)pixels[k], (unsigned int)pixels[k + 1],
(unsigned int)pixels[k + 2]);
k = k + 3;
}
fprintf(output_image, "\n");
}
free(pixels);
break;

10
modules/globebrowsing/src/tileprovider/ffmpegtileprovider.h
View File

@@ -30,9 +30,9 @@
#include <ghoul/glm.h>
// FFMPEG
#include <libavcodec/avcodec.h> // avcodec_alloc_context3
#include <libavformat/avformat.h> // avformat_open_input, AVFormatContext
#include <libavutil/imgutils.h> // av_image_get_buffer_size
#include <avcodec> // avcodec_alloc_context3
#include <avformat> // avformat_open_input, AVFormatContext
#include <imgutils.h> // av_image_get_buffer_size
namespace openspace { struct Documentation; }
@@ -60,6 +60,10 @@ private:
double _lastFrameTime; // The in gmae time of the last frame in J2000 seconds
std::string _startTime;
GLuint _frameTexture;
GLuint _FBO = 1; ///Frame-buffer Object
AVFormatContext* _formatContext = nullptr;
int _streamIndex = -1;
AVStream* _videoStream = nullptr;