2016-02-24 14:34:19 +01:00
// QT includes
# include <QCoreApplication>
# include <QImage>
2018-12-30 22:07:53 +01:00
# include <flatbufserver/FlatBufferConnection.h>
2016-02-24 14:34:19 +01:00
# include "DispmanxWrapper.h"
2016-03-10 12:01:10 +01:00
# include "HyperionConfig.h"
2016-08-28 15:10:43 +02:00
# include <commandline/Parser.h>
2016-03-10 12:01:10 +01:00
2018-12-30 22:07:53 +01:00
// ssdp discover
# include <ssdp/SSDPDiscover.h>
2016-08-28 15:10:43 +02:00
using namespace commandline ;
2016-02-24 14:34:19 +01:00
// save the image as screenshot
2016-08-28 15:10:43 +02:00
void saveScreenshot ( QString filename , const Image < ColorRgb > & image )
2016-02-24 14:34:19 +01:00
{
// 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 )
{
2016-03-10 12:01:10 +01:00
std : : cout
< < " hyperion-dispmanx: " < < std : : endl
2016-06-24 23:22:31 +02:00
< < " \t Version : " < < HYPERION_VERSION < < " ( " < < HYPERION_BUILD_ID < < " ) " < < std : : endl
2016-03-10 12:01:10 +01:00
< < " \t build time: " < < __DATE__ < < " " < < __TIME__ < < std : : endl ;
2016-02-24 14:34:19 +01:00
QCoreApplication app ( argc , argv ) ;
try
{
// create the option parser and initialize all parameters
2018-12-30 22:07:53 +01:00
Parser parser ( " Dispmanx capture application for Hyperion. Will automatically search a Hyperion server if -a option isn't used. Please note that if you have more than one server running it's more or less random which one will be used. " ) ;
2016-08-28 15:10:43 +02:00
2018-12-31 15:48:29 +01:00
IntOption & argFps = parser . add < IntOption > ( ' f ' , " framerate " , " Capture frame rate [default: %1] " , " 10 " , 1 , 25 ) ;
IntOption & argWidth = parser . add < IntOption > ( 0x0 , " width " , " Width of the captured image [default: %1] " , " 64 " , 64 ) ;
IntOption & argHeight = parser . add < IntOption > ( 0x0 , " height " , " Height of the captured image [default: %1] " , " 64 " , 64 ) ;
2016-09-17 00:40:29 +02:00
BooleanOption & argScreenshot = parser . add < BooleanOption > ( 0x0 , " screenshot " , " Take a single screenshot, save it to file and quit " ) ;
2019-02-05 19:55:48 +01:00
Option & argAddress = parser . add < Option > ( ' a ' , " address " , " Set the address of the hyperion server [default: %1] " , " 127.0.0.1:19400 " ) ;
2017-01-17 21:53:35 +01:00
IntOption & argPriority = parser . add < IntOption > ( ' p ' , " priority " , " Use the provided priority channel (suggested 100-199) [default: %1] " , " 150 " ) ;
2016-09-17 00:40:29 +02:00
BooleanOption & argSkipReply = parser . add < BooleanOption > ( 0x0 , " skip-reply " , " Do not receive and check reply messages from Hyperion " ) ;
BooleanOption & argHelp = parser . add < BooleanOption > ( ' h ' , " help " , " Show this help message and exit " ) ;
IntOption & argCropLeft = parser . add < IntOption > ( 0x0 , " crop-left " , " pixels to remove on left after grabbing " ) ;
IntOption & argCropRight = parser . add < IntOption > ( 0x0 , " crop-right " , " pixels to remove on right after grabbing " ) ;
IntOption & argCropTop = parser . add < IntOption > ( 0x0 , " crop-top " , " pixels to remove on top after grabbing " ) ;
IntOption & argCropBottom = parser . add < IntOption > ( 0x0 , " crop-bottom " , " pixels to remove on bottom after grabbing " ) ;
BooleanOption & arg3DSBS = parser . add < BooleanOption > ( 0x0 , " 3DSBS " , " Interpret the incoming video stream as 3D side-by-side " ) ;
BooleanOption & arg3DTAB = parser . add < BooleanOption > ( 0x0 , " 3DTAB " , " Interpret the incoming video stream as 3D top-and-bottom " ) ;
2016-05-18 11:26:25 +02:00
2016-02-24 14:34:19 +01:00
// parse all options
2016-08-28 15:10:43 +02:00
parser . process ( app ) ;
2016-02-24 14:34:19 +01:00
2016-05-18 11:26:25 +02:00
VideoMode videoMode = VIDEO_2D ;
2016-08-28 15:10:43 +02:00
if ( parser . isSet ( arg3DSBS ) )
2016-05-18 11:26:25 +02:00
{
videoMode = VIDEO_3DSBS ;
}
2016-08-28 15:10:43 +02:00
else if ( parser . isSet ( arg3DTAB ) )
2016-05-18 11:26:25 +02:00
{
videoMode = VIDEO_3DTAB ;
}
2016-02-24 14:34:19 +01:00
// check if we need to display the usage. exit if we do.
2016-08-28 15:10:43 +02:00
if ( parser . isSet ( argHelp ) )
2016-02-24 14:34:19 +01:00
{
2016-08-28 15:10:43 +02:00
parser . showHelp ( 1 ) ;
2016-02-24 14:34:19 +01:00
}
// Create the dispmanx grabbing stuff
2016-08-28 15:10:43 +02:00
DispmanxWrapper dispmanxWrapper (
argWidth . getInt ( parser ) ,
argHeight . getInt ( parser ) ,
2016-05-18 11:26:25 +02:00
videoMode ,
2016-08-28 15:10:43 +02:00
argCropLeft . getInt ( parser ) ,
argCropRight . getInt ( parser ) ,
argCropTop . getInt ( parser ) ,
argCropBottom . getInt ( parser ) ,
1000 / argFps . getInt ( parser ) ) ;
2016-02-24 14:34:19 +01:00
2016-08-28 15:10:43 +02:00
if ( parser . isSet ( argScreenshot ) )
2016-02-24 14:34:19 +01:00
{
// Capture a single screenshot and finish
const Image < ColorRgb > & screenshot = dispmanxWrapper . getScreenshot ( ) ;
saveScreenshot ( " screenshot.png " , screenshot ) ;
}
else
{
2018-12-30 22:07:53 +01:00
// server searching by ssdp
2019-08-17 09:44:57 +02:00
QString address = argAddress . value ( parser ) ;
if ( argAddress . value ( parser ) = = " 127.0.0.1:19400 " )
2018-12-30 22:07:53 +01:00
{
SSDPDiscover discover ;
address = discover . getFirstService ( STY_FLATBUFSERVER ) ;
if ( address . isEmpty ( ) )
{
address = argAddress . value ( parser ) ;
}
}
// Create the Flabuf-connection
FlatBufferConnection flatbuf ( " Dispmanx Standalone " , address , argPriority . getInt ( parser ) , parser . isSet ( argSkipReply ) ) ;
// Connect the screen capturing to flatbuf connection processing
QObject : : connect ( & dispmanxWrapper , SIGNAL ( sig_screenshot ( const Image < ColorRgb > & ) ) , & flatbuf , SLOT ( setImage ( Image < ColorRgb > ) ) ) ;
2016-02-24 14:34:19 +01:00
// Start the capturing
dispmanxWrapper . start ( ) ;
// Start the application
app . exec ( ) ;
}
}
2016-02-24 16:03:21 +01:00
catch ( const std : : runtime_error & e )
{
// An error occured. Display error and quit
2016-07-12 15:13:06 +02:00
Error ( Logger : : getInstance ( " DISPMANXGRABBER " ) , " %s " , e . what ( ) ) ;
2016-02-24 16:03:21 +01:00
return - 1 ;
}
2016-02-24 14:34:19 +01:00
2016-02-24 16:03:21 +01:00
return 0 ;
2016-02-24 14:34:19 +01:00
}