61 lines
1.0 KiB
C++
61 lines
1.0 KiB
C++
#ifndef FRAMEREADER_HPP
|
|
#define FRAMEREADER_HPP
|
|
|
|
#include <unistd.h>
|
|
#include <vector>
|
|
#include <map>
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <list>
|
|
#include <condition_variable>
|
|
|
|
#include "channel.hpp"
|
|
|
|
// independent of QT, needs ffmpeg
|
|
extern "C" {
|
|
#include <libavcodec/avcodec.h>
|
|
#include <libavformat/avformat.h>
|
|
#include <libswscale/swscale.h>
|
|
}
|
|
|
|
|
|
class FrameReader {
|
|
public:
|
|
FrameReader(const char *fn);
|
|
uint8_t *get(int idx);
|
|
AVFrame *toRGB(AVFrame *);
|
|
void waitForReady() {
|
|
while (!joined) usleep(10*1000);
|
|
}
|
|
int getRGBSize() { return width*height*3; }
|
|
void loaderThread();
|
|
void cacherThread();
|
|
|
|
//TODO: get this from the actual frame
|
|
int width = 1164;
|
|
int height = 874;
|
|
|
|
private:
|
|
AVFormatContext *pFormatCtx = NULL;
|
|
AVCodecContext *pCodecCtx = NULL;
|
|
|
|
struct SwsContext *sws_ctx = NULL;
|
|
|
|
std::vector<AVPacket *> pkts;
|
|
|
|
std::thread *t;
|
|
bool joined = false;
|
|
|
|
std::map<int, uint8_t *> cache;
|
|
std::mutex mcache;
|
|
|
|
void GOPCache(int idx);
|
|
channel<int> to_cache;
|
|
|
|
bool valid = true;
|
|
char url[0x400];
|
|
};
|
|
|
|
#endif
|
|
|