diff options
Diffstat (limited to 'src/mjpeg-fallback.cpp')
| -rw-r--r-- | src/mjpeg-fallback.cpp | 223 |
1 files changed, 223 insertions, 0 deletions
diff --git a/src/mjpeg-fallback.cpp b/src/mjpeg-fallback.cpp new file mode 100644 index 0000000..f41e68a --- /dev/null +++ b/src/mjpeg-fallback.cpp @@ -0,0 +1,223 @@ +/* Plugin implementation for Mjpeg + * + * \copyright + * Copyright 2017 Red Hat Inc. All rights reserved. + */ + +#include <config.h> +#include <cstring> +#include <exception> +#include <stdexcept> +#include <sstream> +#include <memory> +#include <syslog.h> +#include <X11/Xlib.h> + +#include <spice-streaming-agent/plugin.hpp> +#include <spice-streaming-agent/frame-capture.hpp> + +#include "static-plugin.hpp" +#include "jpeg.hpp" + +using namespace std; +using namespace SpiceStreamingAgent; + +#define ERROR(args) do { \ + std::ostringstream _s; \ + _s << args; \ + throw std::runtime_error(_s.str()); \ +} while(0) + +#define FBC_ERROR(function) \ + ERROR(function " failed(" << fbcStatus << "): " << pFn.nvFBCGetLastErrorStr(fbcHandle)) + +static inline uint64_t get_time() +{ + timespec now; + + clock_gettime(CLOCK_MONOTONIC, &now); + + return (uint64_t)now.tv_sec * 10000000000u + (uint64_t)now.tv_nsec; +} + +namespace { +struct MjpegSettings +{ + int fps; + int quality; +}; + +class MjpegFrameCapture final: public FrameCapture +{ +public: + MjpegFrameCapture(const MjpegSettings &settings); + ~MjpegFrameCapture(); + FrameInfo CaptureFrame() override; + void Reset() override; + SpiceVideoCodecType VideoCodecType() const { + return SPICE_VIDEO_CODEC_TYPE_MJPEG; + } +private: + MjpegSettings settings; + Display *dpy; + + vector<uint8_t> frame; + + // last frame sizes + uint32_t last_width = ~0u, last_height = ~0u; + // last time before capture + uint64_t last_time = 0; +}; + +class MjpegPlugin final: public Plugin +{ +public: + FrameCapture *CreateCapture() override; + unsigned Rank() override; + void ParseOptions(const ConfigureOption *options); + SpiceVideoCodecType VideoCodecType() const { + return SPICE_VIDEO_CODEC_TYPE_MJPEG; + } +private: + MjpegSettings settings = { 10, 80 }; +}; +} + +MjpegFrameCapture::MjpegFrameCapture(const MjpegSettings& settings): + settings(settings) +{ + dpy = XOpenDisplay(NULL); + if (!dpy) + ERROR("Unable to initialize X11"); +} + +MjpegFrameCapture::~MjpegFrameCapture() +{ + XCloseDisplay(dpy); +} + +void MjpegFrameCapture::Reset() +{ + frame.clear(); + last_width = last_height = ~0u; +} + +FrameInfo MjpegFrameCapture::CaptureFrame() +{ + FrameInfo info; + + // reduce speed considering FPS + auto now = get_time(); + if (last_time == 0) { + last_time = now; + } else { + const uint64_t delta = 1000000000u / settings.fps; + if (now >= last_time + delta) { + last_time = now; + } else { + uint64_t wait_time = last_time + delta - now; + // mathematically wait_time must be less than a second as + // delta would be 1 seconds only for FPS == 1 but in this + // side of the if now < last_time + delta + // but is also true that now > last_time so + // last_time + delta > now > last_time so + // 1s >= delta > now - last_time > 0 so + // wait_time = delta - (now - last_time) < delta <= 1s + timespec delay = { 0, (long) wait_time }; + nanosleep(&delay, NULL); + last_time += delta; + } + } + + int screen = XDefaultScreen(dpy); + + Window win = RootWindow(dpy, screen); + + XWindowAttributes win_info; + XGetWindowAttributes(dpy, win, &win_info); + + bool is_first = false; + if (win_info.width != last_width || win_info.height != last_height) { + last_width = win_info.width; + last_height = win_info.height; + is_first = true; + } + + info.size.width = win_info.width; + info.size.height = win_info.height; + + int format = ZPixmap; + // TODO handle errors + XImage *image = XGetImage(dpy, win, win_info.x, win_info.y, + win_info.width, win_info.height, AllPlanes, format); + + // TODO handle errors + // TODO multiple formats (only 32 bit) + write_JPEG_file(frame, settings.quality, (uint8_t*) image->data, + image->width, image->height); + + image->f.destroy_image(image); + + info.buffer = &frame[0]; + info.buffer_size = frame.size(); + + info.stream_start = is_first; + + return info; +} + +FrameCapture *MjpegPlugin::CreateCapture() +{ + return new MjpegFrameCapture(settings); +} + +unsigned MjpegPlugin::Rank() +{ + return FallBackMin; +} + +void MjpegPlugin::ParseOptions(const ConfigureOption *options) +{ +#define arg_error(...) syslog(LOG_ERR, ## __VA_ARGS__); + + for (; options->name; ++options) { + const char *name = options->name; + const char *value = options->value; + + if (strcmp(name, "framerate") == 0) { + int val = atoi(value); + if (val > 0) { + settings.fps = val; + } + else { + arg_error("wrong framerate arg %s\n", value); + } + } + if (strcmp(name, "mjpeg.quality") == 0) { + int val = atoi(value); + if (val > 0) { + settings.quality = val; + } + else { + arg_error("wrong mjpeg.quality arg %s\n", value); + } + } + } +} + +static bool +mjpeg_plugin_init(Agent* agent) +{ + if (agent->Version() != PluginVersion) + return false; + + std::unique_ptr<MjpegPlugin> plugin(new MjpegPlugin()); + + plugin->ParseOptions(agent->Options()); + + agent->Register(*plugin.release()); + + return true; +} + +static StaticPlugin mjpeg_plugin(mjpeg_plugin_init); |