kallaballa
3 years ago
parent
0cc99b1309
commit
6e105f15ac
5 changed files with 640 additions and 0 deletions
@ -0,0 +1,9 @@ |
|||||||
|
.cproject |
||||||
|
.project |
||||||
|
.settings/ |
||||||
|
src/*.o |
||||||
|
src/*.dep |
||||||
|
src/*.d |
||||||
|
src/tetra/tetra-demo |
||||||
|
|
||||||
|
|
||||||
@ -0,0 +1,49 @@ |
|||||||
|
CXX := g++
|
||||||
|
CXXFLAGS := -I/usr/local/include/ -std=c++20 -pthread -fno-strict-aliasing -pedantic -Wall -march=native -flto
|
||||||
|
LDFLAGS := -L/usr/local/lib64/ -L/opt/local/lib -flto
|
||||||
|
.PHONY: all release debian-release info debug asan clean debian-clean distclean |
||||||
|
DESTDIR := /
|
||||||
|
PREFIX := /usr/local
|
||||||
|
|
||||||
|
all: release |
||||||
|
|
||||||
|
release: CXXFLAGS += -g0 -O3 |
||||||
|
release: dirs |
||||||
|
|
||||||
|
info: CXXFLAGS += -g3 -O0 |
||||||
|
info: dirs |
||||||
|
|
||||||
|
debug: CXXFLAGS += -g3 -O0 -rdynamic |
||||||
|
debug: dirs |
||||||
|
|
||||||
|
profile: CXXFLAGS += -g3 -O1 |
||||||
|
profile: dirs |
||||||
|
|
||||||
|
unsafe: CXXFLAGS += -g0 -Ofast -DNDEBUG -ffast-math -ftree-vectorizer-verbose=1 -funroll-loops -ftree-vectorize -fno-signed-zeros -fno-trapping-math -frename-registers |
||||||
|
unsafe: dirs |
||||||
|
|
||||||
|
asan: CXXFLAGS += -g3 -O0 -fno-omit-frame-pointer -fsanitize=address |
||||||
|
asan: LDFLAGS += -fsanitize=address |
||||||
|
asan: LIBS+= -lbfd -ldw |
||||||
|
asan: dirs |
||||||
|
|
||||||
|
clean: dirs |
||||||
|
|
||||||
|
export LDFLAGS |
||||||
|
export CXXFLAGS |
||||||
|
|
||||||
|
dirs: |
||||||
|
${MAKE} -C src/tetra/ ${MAKEFLAGS} CXX=${CXX} ${MAKECMDGOALS}
|
||||||
|
|
||||||
|
debian-release: |
||||||
|
${MAKE} -C src/tetra/ ${MAKEFLAGS} CXX=${CXX} release
|
||||||
|
|
||||||
|
debian-clean: |
||||||
|
${MAKE} -C src/tetra/ ${MAKEFLAGS} CXX=${CXX} clean
|
||||||
|
|
||||||
|
install: ${TARGET} |
||||||
|
true
|
||||||
|
|
||||||
|
distclean: |
||||||
|
true
|
||||||
|
|
||||||
@ -0,0 +1,19 @@ |
|||||||
|
export OPENCV_LOG_LEVEL=DEBUG |
||||||
|
export OPENCV_FFMPEG_LOGLEVEL=trace |
||||||
|
export OPENCV_VIDEOIO_DEBUG=1 |
||||||
|
export OPENCV_VIDEOWRITER_DEBUG=1 |
||||||
|
export OPENCV_FFMPEG_DEBUG=1 |
||||||
|
export OPENCV_OPENCL_RAISE_ERROR=1 |
||||||
|
export OPENCV_OPENCL_ABORT_ON_BUILD_ERROR=1 |
||||||
|
#export OPENCV_OPENCL_ENABLE_MEM_USE_HOST_PTR=1 |
||||||
|
#export OPENCV_OPENCL_ALIGNMENT_MEM_USE_HOST_PTR=1 |
||||||
|
#export OPENCV_OPENCL_RUNTIME= |
||||||
|
#export OPENCV_OPENCL_DEVICE= |
||||||
|
#export OPENCV_OPENCL_SVM_DISABLE=1 |
||||||
|
export OPENCV_DUMP_ERRORS=1 |
||||||
|
export OPENCV_DUMP_CONFIG=1 |
||||||
|
#export OPENCV_CPU_DISABLE=1 |
||||||
|
export OPENCV_TRACE=1 |
||||||
|
export OPENCV_TRACE_SYNC_OPENCL=1 |
||||||
|
#export OPENCV_FFMPEG_CAPTURE_OPTIONS= |
||||||
|
#export OPENCV_FFMPEG_WRITER_OPTIONS= |
||||||
@ -0,0 +1,46 @@ |
|||||||
|
TARGET := tetra-demo
|
||||||
|
|
||||||
|
SRCS := tetra-demo.cpp
|
||||||
|
|
||||||
|
#precompiled headers
|
||||||
|
HEADERS :=
|
||||||
|
OBJS := ${SRCS:.cpp=.o}
|
||||||
|
DEPS := ${SRCS:.cpp=.dep}
|
||||||
|
|
||||||
|
CXXFLAGS += -fpic `pkg-config --cflags egl opencv4 libva-drm glew`
|
||||||
|
LDFLAGS +=
|
||||||
|
LIBS += -lm `pkg-config --libs egl opencv4 libva-drm glew`
|
||||||
|
.PHONY: all release debug clean distclean |
||||||
|
|
||||||
|
all: release |
||||||
|
release: ${TARGET} |
||||||
|
debug: ${TARGET} |
||||||
|
info: ${TARGET} |
||||||
|
profile: ${TARGET} |
||||||
|
hardcore: ${TARGET} |
||||||
|
asan: ${TARGET} |
||||||
|
|
||||||
|
${TARGET}: ${OBJS} |
||||||
|
${CXX} ${LDFLAGS} -o $@ $^ ${LIBS}
|
||||||
|
|
||||||
|
${OBJS}: %.o: %.cpp %.dep ${GCH} |
||||||
|
${CXX} ${CXXFLAGS} -o $@ -c $<
|
||||||
|
|
||||||
|
${DEPS}: %.dep: %.cpp Makefile |
||||||
|
${CXX} ${CXXFLAGS} -MM $< > $@
|
||||||
|
|
||||||
|
${GCH}: %.gch: ${HEADERS} |
||||||
|
${CXX} ${CXXFLAGS} -o $@ -c ${@:.gch=.hpp}
|
||||||
|
|
||||||
|
install: |
||||||
|
mkdir -p ${DESTDIR}/${PREFIX}
|
||||||
|
cp ${TARGET} ${DESTDIR}/${PREFIX}
|
||||||
|
|
||||||
|
uninstall: |
||||||
|
rm ${DESTDIR}/${PREFIX}/${TARGET}
|
||||||
|
|
||||||
|
clean: |
||||||
|
rm -f *~ ${DEPS} ${OBJS} ${CUO} ${GCH} ${TARGET}
|
||||||
|
|
||||||
|
distclean: clean |
||||||
|
|
||||||
@ -0,0 +1,517 @@ |
|||||||
|
#define CL_TARGET_OPENCL_VERSION 300 |
||||||
|
|
||||||
|
#include <iostream> |
||||||
|
#include <filesystem> |
||||||
|
#include <dirent.h> |
||||||
|
#include <fcntl.h> |
||||||
|
#include <stdio.h> |
||||||
|
#include <stdlib.h> |
||||||
|
#include <string.h> |
||||||
|
#include <sys/stat.h> |
||||||
|
#include <sys/types.h> |
||||||
|
#include <unistd.h> |
||||||
|
|
||||||
|
#include <va/va.h> |
||||||
|
#include <va/va_drm.h> |
||||||
|
#include <opencv2/opencv.hpp> |
||||||
|
#include "opencv2/core/va_intel.hpp" |
||||||
|
#include <opencv2/videoio.hpp> |
||||||
|
#include <GL/glew.h> |
||||||
|
#include <GL/gl.h> |
||||||
|
#include <EGL/egl.h> |
||||||
|
#include <CL/cl.h> |
||||||
|
#include <CL/cl_gl.h> |
||||||
|
#include <opencv2/core/ocl.hpp> |
||||||
|
#include <opencv2/core/opengl.hpp> |
||||||
|
|
||||||
|
constexpr off_t WIDTH = 3840; |
||||||
|
constexpr off_t HEIGHT = 2160; |
||||||
|
constexpr double FPS = 30; |
||||||
|
|
||||||
|
using std::cout; |
||||||
|
using std::cerr; |
||||||
|
using std::endl; |
||||||
|
|
||||||
|
namespace kb { |
||||||
|
namespace va { |
||||||
|
//code in the kb::va namespace adapted from https://github.com/opencv/opencv/blob/4.x/samples/va_intel/display.cpp.inc
|
||||||
|
|
||||||
|
bool open_display(); |
||||||
|
void close_display(); |
||||||
|
|
||||||
|
VADisplay display = NULL; |
||||||
|
bool initialized = false; |
||||||
|
|
||||||
|
#define VA_INTEL_PCI_DIR "/sys/bus/pci/devices" |
||||||
|
#define VA_INTEL_DRI_DIR "/dev/dri/" |
||||||
|
#define VA_INTEL_PCI_DISPLAY_CONTROLLER_CLASS 0x03 |
||||||
|
|
||||||
|
static unsigned read_id(const char *devName, const char *idName); |
||||||
|
static int find_adapter(unsigned desiredVendorId); |
||||||
|
|
||||||
|
int drmfd = -1; |
||||||
|
|
||||||
|
class Directory |
||||||
|
{ |
||||||
|
typedef int (*fsort)(const struct dirent**, const struct dirent**); |
||||||
|
public: |
||||||
|
Directory(const char *path) |
||||||
|
{ |
||||||
|
dirEntries_ = 0; |
||||||
|
numEntries_ = scandir(path, &dirEntries_, filterFunc, (fsort) alphasort); |
||||||
|
} |
||||||
|
~Directory() |
||||||
|
{ |
||||||
|
if (numEntries_ && dirEntries_) |
||||||
|
{ |
||||||
|
for (int i = 0; i < numEntries_; ++i) |
||||||
|
free(dirEntries_[i]); |
||||||
|
free(dirEntries_); |
||||||
|
} |
||||||
|
} |
||||||
|
int count() const |
||||||
|
{ |
||||||
|
return numEntries_; |
||||||
|
} |
||||||
|
const struct dirent* operator[](int index) const |
||||||
|
{ |
||||||
|
return ((dirEntries_ != 0) && (index >= 0) && (index < numEntries_)) ? dirEntries_[index] : 0; |
||||||
|
} |
||||||
|
protected: |
||||||
|
static int filterFunc(const struct dirent *dir) |
||||||
|
{ |
||||||
|
if (!dir) |
||||||
|
return 0; |
||||||
|
if (!strcmp(dir->d_name, ".")) |
||||||
|
return 0; |
||||||
|
if (!strcmp(dir->d_name, "..")) |
||||||
|
return 0; |
||||||
|
return 1; |
||||||
|
} |
||||||
|
private: |
||||||
|
int numEntries_; |
||||||
|
struct dirent **dirEntries_; |
||||||
|
}; |
||||||
|
|
||||||
|
static unsigned read_id(const char *devName, const char *idName) |
||||||
|
{ |
||||||
|
long int id = 0; |
||||||
|
|
||||||
|
std::string fileName = cv::format("%s/%s/%s", VA_INTEL_PCI_DIR, devName, idName); |
||||||
|
|
||||||
|
FILE *file = fopen(fileName.c_str(), "r"); |
||||||
|
if (file) |
||||||
|
{ |
||||||
|
char str[16] = ""; |
||||||
|
if (fgets(str, sizeof(str), file)) |
||||||
|
id = strtol(str, NULL, 16); |
||||||
|
fclose(file); |
||||||
|
} |
||||||
|
return (unsigned) id; |
||||||
|
} |
||||||
|
|
||||||
|
static int find_adapter(unsigned desiredVendorId) |
||||||
|
{ |
||||||
|
int adapterIndex = -1; |
||||||
|
|
||||||
|
Directory dir(VA_INTEL_PCI_DIR); |
||||||
|
|
||||||
|
for (int i = 0; i < dir.count(); ++i) |
||||||
|
{ |
||||||
|
const char *name = dir[i]->d_name; |
||||||
|
|
||||||
|
unsigned classId = read_id(name, "class"); |
||||||
|
if ((classId >> 16) == VA_INTEL_PCI_DISPLAY_CONTROLLER_CLASS) |
||||||
|
{ |
||||||
|
unsigned vendorId = read_id(name, "vendor"); |
||||||
|
if (vendorId == desiredVendorId) |
||||||
|
{ |
||||||
|
std::string subdirName = cv::format("%s/%s/%s", VA_INTEL_PCI_DIR, name, "drm"); |
||||||
|
Directory subdir(subdirName.c_str()); |
||||||
|
for (int j = 0; j < subdir.count(); ++j) |
||||||
|
{ |
||||||
|
if (!strncmp(subdir[j]->d_name, "card", 4)) |
||||||
|
{ |
||||||
|
adapterIndex = strtoul(subdir[j]->d_name + 4, NULL, 10); |
||||||
|
} |
||||||
|
} |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return adapterIndex; |
||||||
|
} |
||||||
|
|
||||||
|
class NodeInfo |
||||||
|
{ |
||||||
|
enum { |
||||||
|
NUM_NODES = 2 |
||||||
|
}; |
||||||
|
public: |
||||||
|
NodeInfo(int adapterIndex) |
||||||
|
{ |
||||||
|
const char *names[NUM_NODES] = { "renderD", "card" }; |
||||||
|
int numbers[NUM_NODES]; |
||||||
|
numbers[0] = adapterIndex + 128; |
||||||
|
numbers[1] = adapterIndex; |
||||||
|
for (int i = 0; i < NUM_NODES; ++i) |
||||||
|
{ |
||||||
|
paths_[i] = cv::format("%s%s%d", VA_INTEL_DRI_DIR, names[i], numbers[i]); |
||||||
|
} |
||||||
|
} |
||||||
|
~NodeInfo() |
||||||
|
{ |
||||||
|
// nothing
|
||||||
|
} |
||||||
|
int count() const |
||||||
|
{ |
||||||
|
return NUM_NODES; |
||||||
|
} |
||||||
|
const char* path(int index) const |
||||||
|
{ |
||||||
|
return ((index >= 0) && (index < NUM_NODES)) ? paths_[index].c_str() : 0; |
||||||
|
} |
||||||
|
private: |
||||||
|
std::string paths_[NUM_NODES]; |
||||||
|
}; |
||||||
|
|
||||||
|
static bool open_device_intel(); |
||||||
|
static bool open_device_generic(); |
||||||
|
|
||||||
|
static bool open_device_intel() |
||||||
|
{ |
||||||
|
const unsigned IntelVendorID = 0x8086; |
||||||
|
|
||||||
|
int adapterIndex = find_adapter(IntelVendorID); |
||||||
|
if (adapterIndex >= 0) |
||||||
|
{ |
||||||
|
NodeInfo nodes(adapterIndex); |
||||||
|
|
||||||
|
for (int i = 0; i < nodes.count(); ++i) |
||||||
|
{ |
||||||
|
drmfd = open(nodes.path(i), O_RDWR); |
||||||
|
if (drmfd >= 0) |
||||||
|
{ |
||||||
|
display = vaGetDisplayDRM(drmfd); |
||||||
|
if (display) |
||||||
|
return true; |
||||||
|
close(drmfd); |
||||||
|
drmfd = -1; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
static bool open_device_generic() |
||||||
|
{ |
||||||
|
static const char *device_paths[] = { "/dev/dri/renderD128", "/dev/dri/card0" }; |
||||||
|
static const int num_devices = sizeof(device_paths) / sizeof(device_paths[0]); |
||||||
|
|
||||||
|
for (int i = 0; i < num_devices; ++i) |
||||||
|
{ |
||||||
|
drmfd = open(device_paths[i], O_RDWR); |
||||||
|
if (drmfd >= 0) |
||||||
|
{ |
||||||
|
display = vaGetDisplayDRM(drmfd); |
||||||
|
if (display) |
||||||
|
return true; |
||||||
|
close(drmfd); |
||||||
|
drmfd = -1; |
||||||
|
} |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
bool open_display() |
||||||
|
{ |
||||||
|
if (!initialized) |
||||||
|
{ |
||||||
|
drmfd = -1; |
||||||
|
display = 0; |
||||||
|
|
||||||
|
if (open_device_intel() || open_device_generic()) |
||||||
|
{ |
||||||
|
int majorVersion = 0, minorVersion = 0; |
||||||
|
if (vaInitialize(display, &majorVersion, &minorVersion) == VA_STATUS_SUCCESS) |
||||||
|
{ |
||||||
|
initialized = true; |
||||||
|
return true; |
||||||
|
} |
||||||
|
close(drmfd); |
||||||
|
display = 0; |
||||||
|
drmfd = -1; |
||||||
|
} |
||||||
|
return false; // Can't open VA display
|
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
void close_display() |
||||||
|
{ |
||||||
|
if (initialized) |
||||||
|
{ |
||||||
|
if (display) |
||||||
|
vaTerminate(display); |
||||||
|
if (drmfd >= 0) |
||||||
|
close(drmfd); |
||||||
|
display = 0; |
||||||
|
drmfd = -1; |
||||||
|
initialized = false; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void check_if_YUV420_available() { |
||||||
|
VAEntrypoint entrypoints[5]; |
||||||
|
int num_entrypoints, vld_entrypoint; |
||||||
|
VAConfigAttrib attrib; |
||||||
|
VAStatus status; |
||||||
|
|
||||||
|
status = vaQueryConfigEntrypoints(va::display, VAProfileVP9Profile0, entrypoints, &num_entrypoints); |
||||||
|
assert(status == VA_STATUS_SUCCESS); |
||||||
|
|
||||||
|
for (vld_entrypoint = 0; vld_entrypoint < num_entrypoints; ++vld_entrypoint) |
||||||
|
{ |
||||||
|
if (entrypoints[vld_entrypoint] == VAEntrypointVLD) |
||||||
|
break; |
||||||
|
} |
||||||
|
if (vld_entrypoint == num_entrypoints) |
||||||
|
throw std::runtime_error("Failed to find VLD entry point"); |
||||||
|
|
||||||
|
attrib.type = VAConfigAttribRTFormat; |
||||||
|
vaGetConfigAttributes(va::display, VAProfileVP9Profile0, VAEntrypointVLD, &attrib, 1); |
||||||
|
if ((attrib.value & VA_RT_FORMAT_YUV420) == 0) |
||||||
|
throw std::runtime_error("Desired YUV444 RT format not found"); |
||||||
|
} |
||||||
|
} // namespace va
|
||||||
|
} |
||||||
|
namespace kb { |
||||||
|
namespace egl { |
||||||
|
//code in the kb::egl namespace deals with setting up EGL
|
||||||
|
EGLDisplay display; |
||||||
|
EGLSurface surface; |
||||||
|
EGLContext context; |
||||||
|
|
||||||
|
void eglCheckError(const std::filesystem::path &file, unsigned int line, const char *expression) |
||||||
|
{ |
||||||
|
EGLint errorCode = eglGetError(); |
||||||
|
|
||||||
|
if (errorCode != EGL_SUCCESS) { |
||||||
|
cerr << "EGL failed in " << file.filename() << " (" << line << ") : " |
||||||
|
<< "\nExpression:\n " << expression << "\nError code:\n " << errorCode << "\n " |
||||||
|
<< endl; |
||||||
|
assert(false); |
||||||
|
} |
||||||
|
} |
||||||
|
#define eglCheck(expr) \ |
||||||
|
expr; \
|
||||||
|
egl::eglCheckError(__FILE__, __LINE__, #expr); |
||||||
|
|
||||||
|
void init_egl() { |
||||||
|
eglCheck(eglBindAPI(EGL_OPENGL_API)); |
||||||
|
eglCheck(display = eglGetDisplay(EGL_DEFAULT_DISPLAY)); |
||||||
|
eglCheck(eglInitialize(display, nullptr, nullptr)); |
||||||
|
|
||||||
|
const EGLint attributes[] = |
||||||
|
{ EGL_BUFFER_SIZE, |
||||||
|
static_cast<EGLint>(24), |
||||||
|
EGL_DEPTH_SIZE, |
||||||
|
static_cast<EGLint>(24), |
||||||
|
EGL_STENCIL_SIZE, |
||||||
|
static_cast<EGLint>(0), |
||||||
|
EGL_SAMPLE_BUFFERS, |
||||||
|
EGL_FALSE, |
||||||
|
EGL_SAMPLES, |
||||||
|
0, |
||||||
|
EGL_SURFACE_TYPE, |
||||||
|
EGL_WINDOW_BIT | EGL_PBUFFER_BIT, |
||||||
|
EGL_RENDERABLE_TYPE, |
||||||
|
EGL_OPENGL_BIT, |
||||||
|
EGL_NONE }; |
||||||
|
|
||||||
|
EGLint configCount; |
||||||
|
EGLConfig configs[1]; |
||||||
|
|
||||||
|
eglCheck(eglChooseConfig(display, attributes, configs, 1, &configCount)); |
||||||
|
EGLint attrib_list[] = { EGL_WIDTH, WIDTH, EGL_HEIGHT, HEIGHT, EGL_NONE }; |
||||||
|
|
||||||
|
eglCheck(surface = eglCreatePbufferSurface(display, configs[0], attrib_list)); |
||||||
|
|
||||||
|
const EGLint contextVersion[] = { EGL_CONTEXT_CLIENT_VERSION, 1, EGL_CONTEXT_OPENGL_DEBUG, EGL_FALSE, EGL_NONE }; |
||||||
|
|
||||||
|
eglCheck(context = eglCreateContext(display, configs[0], EGL_NO_CONTEXT, contextVersion)); |
||||||
|
eglCheck(eglMakeCurrent(display, surface, surface, context)); |
||||||
|
|
||||||
|
cerr << "EGL Version: " << eglQueryString(display, EGL_VERSION) << endl; |
||||||
|
cerr << "OpenGL version: " << glGetString(GL_VERSION) << endl; |
||||||
|
} |
||||||
|
|
||||||
|
EGLBoolean swapBuffers() { |
||||||
|
return eglSwapBuffers(display, surface); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
namespace kb { |
||||||
|
namespace gl { |
||||||
|
//code in the kb::gl namespace deals with OpenGL (and OpenCV/GL) internals
|
||||||
|
cv::ogl::Texture2D *frame_buf_tex; |
||||||
|
GLuint frame_buf_tex_name; |
||||||
|
|
||||||
|
void glCheckError(const std::filesystem::path &file, unsigned int line, const char *expression) |
||||||
|
{ |
||||||
|
GLint errorCode = glGetError(); |
||||||
|
|
||||||
|
if (errorCode != GL_NO_ERROR) { |
||||||
|
cerr << "GL failed in " << file.filename() << " (" << line << ") : " |
||||||
|
<< "\nExpression:\n " << expression << "\nError code:\n " << errorCode << "\n " |
||||||
|
<< endl; |
||||||
|
assert(false); |
||||||
|
} |
||||||
|
} |
||||||
|
#define glCheck(expr) \ |
||||||
|
expr; \
|
||||||
|
gl::glCheckError(__FILE__, __LINE__, #expr); |
||||||
|
|
||||||
|
void init_gl() { |
||||||
|
glewInit(); |
||||||
|
|
||||||
|
cv::ogl::ocl::initializeContextFromGL(); |
||||||
|
|
||||||
|
frame_buf_tex_name = 0; |
||||||
|
glCheck(glGenFramebuffers(1, &frame_buf_tex_name)); |
||||||
|
glCheck(glBindFramebuffer(GL_FRAMEBUFFER, frame_buf_tex_name)); |
||||||
|
frame_buf_tex = new cv::ogl::Texture2D(cv::Size(WIDTH, HEIGHT), cv::ogl::Texture2D::RGBA, false); |
||||||
|
frame_buf_tex->bind(); |
||||||
|
|
||||||
|
glCheck(glFramebufferTexture(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, frame_buf_tex->texId(), 0)); |
||||||
|
GLenum drawBuffers[1] = { GL_COLOR_ATTACHMENT0 }; |
||||||
|
glCheck(glDrawBuffers(1, drawBuffers)); |
||||||
|
assert(glCheckFramebufferStatus(GL_FRAMEBUFFER) == GL_FRAMEBUFFER_COMPLETE); |
||||||
|
glCheck(glBindFramebuffer(GL_FRAMEBUFFER, frame_buf_tex_name)); |
||||||
|
|
||||||
|
glCheck(glClearColor(0.1, 0.39, 0.88, 1.0)); |
||||||
|
glCheck(glColor3f(1.0, 1.0, 1.0)); |
||||||
|
|
||||||
|
glCheck(glEnable(GL_CULL_FACE)); |
||||||
|
glCheck(glCullFace(GL_BACK)); |
||||||
|
|
||||||
|
glCheck(glMatrixMode(GL_PROJECTION)); |
||||||
|
glCheck(glLoadIdentity()); |
||||||
|
glCheck(glFrustum(-2, 2, -1.5, 1.5, 1, 40)); |
||||||
|
|
||||||
|
glCheck(glMatrixMode(GL_MODELVIEW)); |
||||||
|
glCheck(glLoadIdentity()); |
||||||
|
glCheck(glTranslatef(0, 0, -3)); |
||||||
|
glCheck(glRotatef(50, 1, 0, 0)); |
||||||
|
glCheck(glRotatef(70, 0, 1, 0)); |
||||||
|
} |
||||||
|
|
||||||
|
void swapBuffers() { |
||||||
|
kb::egl::swapBuffers(); |
||||||
|
} |
||||||
|
|
||||||
|
void fetch_frame_buffer(cv::UMat &m) { |
||||||
|
cv::ogl::convertFromGLTexture2D(*frame_buf_tex, m); |
||||||
|
assert(glGetError() == GL_NO_ERROR); |
||||||
|
} |
||||||
|
|
||||||
|
void return_frame_buffer(cv::UMat &m) { |
||||||
|
cv::ogl::convertToGLTexture2D(m, *frame_buf_tex); |
||||||
|
assert(glGetError() == GL_NO_ERROR); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
int main(int argc, char **argv) { |
||||||
|
using namespace kb; |
||||||
|
|
||||||
|
if (!va::open_display()) |
||||||
|
throw std::runtime_error("Failed to open VA display for CL-VA interoperability"); |
||||||
|
|
||||||
|
va::check_if_YUV420_available(); |
||||||
|
|
||||||
|
cv::va_intel::ocl::initializeContextFromVA(va::display, true); |
||||||
|
cv::ocl::OpenCLExecutionContext vaContext = cv::ocl::OpenCLExecutionContext::getCurrent(); |
||||||
|
|
||||||
|
cv::VideoWriter video("out.mkv", cv::CAP_FFMPEG, cv::VideoWriter::fourcc('V', 'P', '9', '0'), FPS, cv::Size(WIDTH, HEIGHT), { |
||||||
|
cv::VIDEOWRITER_PROP_HW_DEVICE, 0, |
||||||
|
cv::VIDEOWRITER_PROP_HW_ACCELERATION, cv::VIDEO_ACCELERATION_VAAPI, |
||||||
|
cv::VIDEOWRITER_PROP_HW_ACCELERATION_USE_OPENCL, 1 |
||||||
|
}); |
||||||
|
|
||||||
|
egl::init_egl(); |
||||||
|
gl::init_gl(); |
||||||
|
cv::ocl::OpenCLExecutionContext glContext = cv::ocl::OpenCLExecutionContext::getCurrent(); |
||||||
|
|
||||||
|
cv::UMat frameBuffer(HEIGHT, WIDTH, CV_8UC4, cv::Scalar::all(0)); |
||||||
|
cv::UMat mask; |
||||||
|
cv::UMat bgr; |
||||||
|
|
||||||
|
while (true) { |
||||||
|
int64 start = cv::getTickCount(); |
||||||
|
|
||||||
|
//draw a rotating tetrahedron
|
||||||
|
glContext.bind(); |
||||||
|
glCheck(glRotatef(1, 0, 1, 0)); |
||||||
|
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT); |
||||||
|
|
||||||
|
glColor3f(1.0, 1.0, 1.0); |
||||||
|
glBegin(GL_LINES); |
||||||
|
for (GLfloat i = -2.5; i <= 2.5; i += 0.25) { |
||||||
|
glVertex3f(i, 0, 2.5); |
||||||
|
glVertex3f(i, 0, -2.5); |
||||||
|
glVertex3f(2.5, 0, i); |
||||||
|
glVertex3f(-2.5, 0, i); |
||||||
|
} |
||||||
|
glEnd(); |
||||||
|
|
||||||
|
glBegin(GL_TRIANGLE_STRIP); |
||||||
|
glColor3f(1, 1, 1); |
||||||
|
glVertex3f(0, 2, 0); |
||||||
|
glColor3f(1, 0, 0); |
||||||
|
glVertex3f(-1, 0, 1); |
||||||
|
glColor3f(0, 1, 0); |
||||||
|
glVertex3f(1, 0, 1); |
||||||
|
glColor3f(0, 0, 1); |
||||||
|
glVertex3f(0, 0, -1.4); |
||||||
|
glColor3f(1, 1, 1); |
||||||
|
glVertex3f(0, 2, 0); |
||||||
|
glColor3f(1, 0, 0); |
||||||
|
glVertex3f(-1, 0, 1); |
||||||
|
glEnd(); |
||||||
|
|
||||||
|
glFlush(); |
||||||
|
gl::swapBuffers(); |
||||||
|
gl::fetch_frame_buffer(frameBuffer); //hand over the data (GPU 2 GPU) to OpenCV/OpenCL
|
||||||
|
|
||||||
|
//Using OpenCL in the background
|
||||||
|
cv::flip(frameBuffer, frameBuffer, 0); // flip the image in the y-axis
|
||||||
|
|
||||||
|
//do a glow effect using blur
|
||||||
|
{ |
||||||
|
cv::blur(frameBuffer, mask, cv::Size(50,50)); |
||||||
|
cv::bitwise_not(mask, mask); |
||||||
|
cv::bitwise_not(frameBuffer, frameBuffer); |
||||||
|
mask.assignTo(mask, CV_16U); |
||||||
|
frameBuffer.assignTo(frameBuffer, CV_16U); |
||||||
|
multiply(mask,frameBuffer, mask); |
||||||
|
cv::divide(mask, cv::Scalar::all(255.0), mask); |
||||||
|
mask.assignTo(mask, CV_8U); |
||||||
|
cv::bitwise_not(mask, frameBuffer); |
||||||
|
} |
||||||
|
|
||||||
|
cv::cvtColor(frameBuffer, bgr, cv::COLOR_BGRA2BGR); //convert to bgr
|
||||||
|
|
||||||
|
vaContext.bind(); |
||||||
|
//encode the frame using VAAPI on the GPU.
|
||||||
|
video.write(bgr); |
||||||
|
|
||||||
|
int64 tick = cv::getTickCount(); |
||||||
|
if(tick % int64(FPS) == 0) |
||||||
|
cerr << "FPS : " << cv::getTickFrequency() / (cv::getTickCount() - start) << '\r'; |
||||||
|
} |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
Loading…
Reference in new issue