123 lines
5.6 KiB
C++
123 lines
5.6 KiB
C++
|
|
// QT includes
|
|
#include <QCoreApplication>
|
|
#include <QImage>
|
|
|
|
// getoptPlusPLus includes
|
|
#include <getoptPlusPlus/getoptpp.h>
|
|
|
|
#include "protoserver/ProtoConnectionWrapper.h"
|
|
#include "X11Wrapper.h"
|
|
#include "HyperionConfig.h"
|
|
|
|
using namespace vlofgren;
|
|
|
|
// save the image as screenshot
|
|
void saveScreenshot(const char * filename, const Image<ColorRgb> & image)
|
|
{
|
|
// store as PNG
|
|
QImage pngImage((const uint8_t *) image.memptr(), image.width(), image.height(), 3*image.width(), QImage::Format_RGB888);
|
|
pngImage.save(filename);
|
|
}
|
|
|
|
int main(int argc, char ** argv)
|
|
{
|
|
std::cout
|
|
<< "hyperion-x11:" << std::endl
|
|
<< "\tversion : " << HYPERION_VERSION_ID << std::endl
|
|
<< "\tbuild time: " << __DATE__ << " " << __TIME__ << std::endl;
|
|
|
|
QCoreApplication app(argc, argv);
|
|
|
|
try
|
|
{
|
|
// create the option parser and initialize all parameters
|
|
OptionsParser optionParser("X11 capture application for Hyperion");
|
|
ParameterSet & parameters = optionParser.getParameters();
|
|
|
|
IntParameter & argFps = parameters.add<IntParameter> ('f', "framerate", "Capture frame rate [default: 10]");
|
|
SwitchParameter<> & argXGetImage = parameters.add<SwitchParameter<>> ('x', "xgetimage", "Use XGetImage instead of XRender");
|
|
IntParameter & argCropWidth = parameters.add<IntParameter> (0x0, "crop-width", "Number of pixels to crop from the left and right sides of the picture before decimation [default: 0]");
|
|
IntParameter & argCropHeight = parameters.add<IntParameter> (0x0, "crop-height", "Number of pixels to crop from the top and the bottom of the picture before decimation [default: 0]");
|
|
IntParameter & argCropLeft = parameters.add<IntParameter> (0x0, "crop-left", "Number of pixels to crop from the left of the picture before decimation (overrides --crop-width)");
|
|
IntParameter & argCropRight = parameters.add<IntParameter> (0x0, "crop-right", "Number of pixels to crop from the right of the picture before decimation (overrides --crop-width)");
|
|
IntParameter & argCropTop = parameters.add<IntParameter> (0x0, "crop-top", "Number of pixels to crop from the top of the picture before decimation (overrides --crop-height)");
|
|
IntParameter & argCropBottom = parameters.add<IntParameter> (0x0, "crop-bottom", "Number of pixels to crop from the bottom of the picture before decimation (overrides --crop-height)");
|
|
IntParameter & argSizeDecimation = parameters.add<IntParameter> ('s', "size-decimator", "Decimation factor for the output size [default=8]");
|
|
SwitchParameter<> & argScreenshot = parameters.add<SwitchParameter<>> (0x0, "screenshot", "Take a single screenshot, save it to file and quit");
|
|
StringParameter & argAddress = parameters.add<StringParameter> ('a', "address", "Set the address of the hyperion server [default: 127.0.0.1:19445]");
|
|
IntParameter & argPriority = parameters.add<IntParameter> ('p', "priority", "Use the provided priority channel (the lower the number, the higher the priority) [default: 800]");
|
|
SwitchParameter<> & argSkipReply = parameters.add<SwitchParameter<>> (0x0, "skip-reply", "Do not receive and check reply messages from Hyperion");
|
|
SwitchParameter<> & argHelp = parameters.add<SwitchParameter<>> ('h', "help", "Show this help message and exit");
|
|
|
|
// set defaults
|
|
argFps.setDefault(10);
|
|
argCropWidth.setDefault(0);
|
|
argCropHeight.setDefault(0);
|
|
argSizeDecimation.setDefault(8);
|
|
argAddress.setDefault("127.0.0.1:19445");
|
|
argPriority.setDefault(800);
|
|
|
|
// parse all options
|
|
optionParser.parse(argc, const_cast<const char **>(argv));
|
|
|
|
// check if we need to display the usage. exit if we do.
|
|
if (argHelp.isSet())
|
|
{
|
|
optionParser.usage();
|
|
return 0;
|
|
}
|
|
|
|
// cropping values if not defined
|
|
if (!argCropLeft.isSet()) argCropLeft.setDefault(argCropWidth.getValue());
|
|
if (!argCropRight.isSet()) argCropRight.setDefault(argCropWidth.getValue());
|
|
if (!argCropTop.isSet()) argCropTop.setDefault(argCropHeight.getValue());
|
|
if (!argCropBottom.isSet()) argCropBottom.setDefault(argCropHeight.getValue());
|
|
|
|
// Create the X11 grabbing stuff
|
|
int grabInterval = 1000 / argFps.getValue();
|
|
bool useXGetImage = argXGetImage.isSet();
|
|
X11Wrapper x11Wrapper(
|
|
grabInterval,
|
|
useXGetImage,
|
|
argCropLeft.getValue(),
|
|
argCropRight.getValue(),
|
|
argCropTop.getValue(),
|
|
argCropBottom.getValue(),
|
|
argSizeDecimation.getValue(), // horizontal decimation
|
|
argSizeDecimation.getValue()); // vertical decimation
|
|
|
|
if (!x11Wrapper.displayInit())
|
|
return -1;
|
|
|
|
if (argScreenshot.isSet())
|
|
{
|
|
// Capture a single screenshot and finish
|
|
const Image<ColorRgb> & screenshot = x11Wrapper.getScreenshot();
|
|
saveScreenshot("screenshot.png", screenshot);
|
|
}
|
|
else
|
|
{
|
|
// Create the Proto-connection with hyperiond
|
|
ProtoConnectionWrapper protoWrapper(argAddress.getValue(), argPriority.getValue(), 1000, argSkipReply.isSet());
|
|
|
|
// Connect the screen capturing to the proto processing
|
|
QObject::connect(&x11Wrapper, SIGNAL(sig_screenshot(const Image<ColorRgb> &)), &protoWrapper, SLOT(receiveImage(Image<ColorRgb>)));
|
|
|
|
// Start the capturing
|
|
x11Wrapper.start();
|
|
|
|
// Start the application
|
|
app.exec();
|
|
}
|
|
}
|
|
catch (const std::runtime_error & e)
|
|
{
|
|
// An error occured. Display error and quit
|
|
std::cerr << e.what() << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|