kallaballa
2 years ago
parent
e657896a7c
commit
f4b8f9ec84
12 changed files with 261 additions and 506 deletions
@ -1,46 +0,0 @@ |
||||
TARGET := video-demo2
|
||||
|
||||
SRCS := video-demo2.cpp
|
||||
|
||||
#precompiled headers
|
||||
HEADERS :=
|
||||
OBJS := ${SRCS:.cpp=.o}
|
||||
DEPS := ${SRCS:.cpp=.dep}
|
||||
|
||||
CXXFLAGS += -fpic
|
||||
LDFLAGS +=
|
||||
LIBS += -lm
|
||||
.PHONY: all release debug clean distclean |
||||
|
||||
all: release |
||||
release: ${TARGET} |
||||
debug: ${TARGET} |
||||
info: ${TARGET} |
||||
profile: ${TARGET} |
||||
unsafe: ${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 |
||||
|
||||
@ -1,159 +0,0 @@ |
||||
#define CL_TARGET_OPENCL_VERSION 120 |
||||
|
||||
#include "../common/subsystems.hpp" |
||||
#include <string> |
||||
|
||||
constexpr long unsigned int WIDTH = 1920; |
||||
constexpr long unsigned int HEIGHT = 1080; |
||||
constexpr const int VA_HW_DEVICE_INDEX = 0; |
||||
constexpr bool OFFSCREEN = true; |
||||
constexpr const char* OUTPUT_FILENAME = "video-demo.mkv"; |
||||
constexpr unsigned long DIAG = hypot(double(WIDTH), double(HEIGHT)); |
||||
|
||||
constexpr int GLOW_KERNEL_SIZE = std::max(int(DIAG / 138 % 2 == 0 ? DIAG / 138 + 1 : DIAG / 138), 1); |
||||
|
||||
using std::cerr; |
||||
using std::endl; |
||||
using std::string; |
||||
|
||||
void init_scene(unsigned long w, unsigned long h) { |
||||
//Initialize the OpenGL scene
|
||||
glViewport(0, 0, w, h); |
||||
glColor3f(1.0, 1.0, 1.0); |
||||
|
||||
glEnable(GL_CULL_FACE); |
||||
glCullFace(GL_BACK); |
||||
|
||||
glMatrixMode(GL_PROJECTION); |
||||
glLoadIdentity(); |
||||
glFrustum(-2, 2, -1.5, 1.5, 1, 40); |
||||
|
||||
glMatrixMode(GL_MODELVIEW); |
||||
glLoadIdentity(); |
||||
glTranslatef(0, 0, -3); |
||||
glRotatef(50, 1, 0, 0); |
||||
glRotatef(70, 0, 1, 0); |
||||
} |
||||
|
||||
void render_scene(unsigned long w, unsigned long h) { |
||||
//Render a tetrahedron using immediate mode because the code is more concise for a demo
|
||||
glViewport(0, 0, w, h); |
||||
glRotatef(1, 0, 1, 0); |
||||
glClearColor(0.0f, 0.0f, 1.0f, 1.0f); |
||||
|
||||
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(); |
||||
} |
||||
|
||||
void glow_effect(const cv::UMat &src, cv::UMat &dst, const int ksize) { |
||||
static cv::UMat resize; |
||||
static cv::UMat blur; |
||||
static cv::UMat dst16; |
||||
|
||||
cv::bitwise_not(src, dst); |
||||
|
||||
//Resize for some extra performance
|
||||
cv::resize(dst, resize, cv::Size(), 0.5, 0.5); |
||||
//Cheap blur
|
||||
cv::boxFilter(resize, resize, -1, cv::Size(ksize, ksize), cv::Point(-1,-1), true, cv::BORDER_REPLICATE); |
||||
//Back to original size
|
||||
cv::resize(resize, blur, src.size()); |
||||
|
||||
//Multiply the src image with a blurred version of itself
|
||||
cv::multiply(dst, blur, dst16, 1, CV_16U); |
||||
//Normalize and convert back to CV_8U
|
||||
cv::divide(dst16, cv::Scalar::all(255.0), dst, 1, CV_8U); |
||||
|
||||
cv::bitwise_not(dst, dst); |
||||
} |
||||
|
||||
int main(int argc, char **argv) { |
||||
using namespace kb; |
||||
|
||||
if(argc != 2) { |
||||
cerr << "Usage: video-demo <video-file>" << endl; |
||||
exit(1); |
||||
} |
||||
|
||||
//Initialize the application
|
||||
app::init("Video Demo", WIDTH, HEIGHT, OFFSCREEN); |
||||
//Print system information
|
||||
app::print_system_info(); |
||||
|
||||
//Initialize MJPEG HW decoding using VAAPI
|
||||
cv::VideoCapture capture(argv[1], cv::CAP_FFMPEG, { |
||||
cv::CAP_PROP_HW_DEVICE, VA_HW_DEVICE_INDEX, |
||||
cv::CAP_PROP_HW_ACCELERATION, cv::VIDEO_ACCELERATION_VAAPI, |
||||
cv::CAP_PROP_HW_ACCELERATION_USE_OPENCL, 1 |
||||
}); |
||||
|
||||
//Copy OpenCL Context for VAAPI. Must be called right after first VideoWriter/VideoCapture initialization.
|
||||
va::copy(); |
||||
|
||||
if (!capture.isOpened()) { |
||||
cerr << "ERROR! Unable to open video input" << endl; |
||||
return -1; |
||||
} |
||||
|
||||
double fps = capture.get(cv::CAP_PROP_FPS); |
||||
|
||||
//Initialize VP9 HW encoding using VAAPI. We don't need to specify the hardware device twice. only generates a warning.
|
||||
cv::VideoWriter writer(OUTPUT_FILENAME, cv::CAP_FFMPEG, cv::VideoWriter::fourcc('V', 'P', '9', '0'), fps, cv::Size(WIDTH, HEIGHT), { |
||||
cv::VIDEOWRITER_PROP_HW_ACCELERATION, cv::VIDEO_ACCELERATION_VAAPI, |
||||
cv::VIDEOWRITER_PROP_HW_ACCELERATION_USE_OPENCL, 1 |
||||
}); |
||||
|
||||
gl::render([WIDTH, HEIGHT]() { |
||||
init_scene(WIDTH, HEIGHT); |
||||
}); |
||||
|
||||
while (true) { |
||||
va::read([&capture](cv::UMat& videoFrame){ |
||||
//videoFrame will be converted to BGRA and stored in the frameBuffer.
|
||||
capture >> videoFrame; |
||||
assert(!videoFrame.empty()); |
||||
}); |
||||
|
||||
cl::work([](cv::UMat& frameBuffer){ |
||||
//Resize the frame if necessary. (OpenCL)
|
||||
cv::resize(frameBuffer, frameBuffer, cv::Size(WIDTH, HEIGHT)); |
||||
}); |
||||
|
||||
gl::render([&WIDTH, &HEIGHT]() { |
||||
//Render using OpenGL
|
||||
render_scene(WIDTH, HEIGHT); |
||||
}); |
||||
|
||||
cl::work([&GLOW_KERNEL_SIZE](cv::UMat& frameBuffer){ |
||||
//Glow effect (OpenCL)
|
||||
glow_effect(frameBuffer, frameBuffer, GLOW_KERNEL_SIZE); |
||||
}); |
||||
|
||||
//If onscreen rendering is enabled it displays the framebuffer in the native window. Returns false if the window was closed.
|
||||
if(!app::display()) |
||||
break; |
||||
|
||||
va::write([&writer](const cv::UMat& videoFrame){ |
||||
//videoFrame is the frameBuffer converted to BGR. Ready to be written.
|
||||
writer << videoFrame; |
||||
}); |
||||
|
||||
app::print_fps(); |
||||
} |
||||
|
||||
app::terminate(); |
||||
|
||||
return 0; |
||||
} |
||||
Loading…
Reference in new issue