// STL includes
#include <csignal>
#include <iomanip>
#include <clocale>

// QT includes
#include <QCoreApplication>

// getoptPlusPLus includes
#include <getoptPlusPlus/getoptpp.h>

// blackborder includes
#include <blackborder/BlackBorderProcessor.h>

// grabber includes
#include "grabber/V4L2Grabber.h"

// proto includes
#include "protoserver/ProtoConnection.h"
#include "protoserver/ProtoConnectionWrapper.h"

// hyperion-v4l2 includes
#include "VideoStandardParameter.h"
#include "PixelFormatParameter.h"
#include "ScreenshotHandler.h"

using namespace vlofgren;

// save the image as screenshot
void saveScreenshot(void *, 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("screenshot.png");
}

int main(int argc, char** argv)
{
    QCoreApplication app(argc, argv);

    // force the locale
    setlocale(LC_ALL, "C");
    QLocale::setDefault(QLocale::c());

    // register the image type to use in signals
    qRegisterMetaType<Image<ColorRgb>>("Image<ColorRgb>");

    try
    {
        // create the option parser and initialize all parameters
        OptionsParser optionParser("V4L capture application for Hyperion");
        ParameterSet & parameters = optionParser.getParameters();

        StringParameter        & argDevice          = parameters.add<StringParameter>       ('d', "device",           "The device to use [default=/dev/video0]");
        VideoStandardParameter & argVideoStandard   = parameters.add<VideoStandardParameter>('v', "video-standard",   "The used video standard. Valid values are PAL or NTSC (optional)");
        PixelFormatParameter   & argPixelFormat     = parameters.add<PixelFormatParameter>  (0x0, "pixel-format",     "The use pixel format. Valid values are YUYV, UYVY, and RGB32 (optional)");
        IntParameter           & argInput           = parameters.add<IntParameter>          (0x0, "input",            "Input channel (optional)");
        IntParameter           & argWidth           = parameters.add<IntParameter>          (0x0, "width",            "Try to set the width of the video input (optional)");
        IntParameter           & argHeight          = parameters.add<IntParameter>          (0x0, "height",           "Try to set the height of the video input (optional)");
        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=1]");
        IntParameter           & argFrameDecimation = parameters.add<IntParameter>          ('f', "frame-decimator",  "Decimation factor for the video frames [default=1]");
        SwitchParameter<>      & argScreenshot      = parameters.add<SwitchParameter<>>     (0x0, "screenshot",       "Take a single screenshot, save it to file and quit");
        DoubleParameter        & argSignalThreshold = parameters.add<DoubleParameter>       ('t', "signal-threshold", "The signal threshold for detecting the presence of a signal. Value should be between 0.0 and 1.0.");
        DoubleParameter        & argRedSignalThreshold = parameters.add<DoubleParameter>    (0x0, "red-threshold",    "The red signal threshold. Value should be between 0.0 and 1.0. (overrides --signal-threshold)");
        DoubleParameter        & argGreenSignalThreshold = parameters.add<DoubleParameter>  (0x0, "green-threshold",  "The green signal threshold. Value should be between 0.0 and 1.0. (overrides --signal-threshold)");
        DoubleParameter        & argBlueSignalThreshold = parameters.add<DoubleParameter>   (0x0, "blue-threshold",   "The blue signal threshold. Value should be between 0.0 and 1.0. (overrides --signal-threshold)");
        SwitchParameter<>      & arg3DSBS           = parameters.add<SwitchParameter<>>     (0x0, "3DSBS",            "Interpret the incoming video stream as 3D side-by-side");
        SwitchParameter<>      & arg3DTAB           = parameters.add<SwitchParameter<>>     (0x0, "3DTAB",            "Interpret the incoming video stream as 3D top-and-bottom");
        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
        argDevice.setDefault("/dev/video0");
        argVideoStandard.setDefault(VIDEOSTANDARD_NO_CHANGE);
        argPixelFormat.setDefault(PIXELFORMAT_NO_CHANGE);
        argInput.setDefault(-1);
        argWidth.setDefault(-1);
        argHeight.setDefault(-1);
        argCropWidth.setDefault(0);
        argCropHeight.setDefault(0);
        argSizeDecimation.setDefault(1);
        argFrameDecimation.setDefault(1);
        argAddress.setDefault("127.0.0.1:19445");
        argPriority.setDefault(800);
        argSignalThreshold.setDefault(-1);

        // 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;
        }

        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());

        // initialize the grabber
        V4L2Grabber grabber(
                    argDevice.getValue(),
                    argInput.getValue(),
                    argVideoStandard.getValue(),
                    argPixelFormat.getValue(),
                    argWidth.getValue(),
                    argHeight.getValue(),
                    std::max(1, argFrameDecimation.getValue()),
                    std::max(1, argSizeDecimation.getValue()),
                    std::max(1, argSizeDecimation.getValue()));

        // set signal detection
        grabber.setSignalThreshold(
                    std::min(1.0, std::max(0.0, argRedSignalThreshold.isSet() ? argRedSignalThreshold.getValue() : argSignalThreshold.getValue())),
                    std::min(1.0, std::max(0.0, argGreenSignalThreshold.isSet() ? argGreenSignalThreshold.getValue() : argSignalThreshold.getValue())),
                    std::min(1.0, std::max(0.0, argBlueSignalThreshold.isSet() ? argBlueSignalThreshold.getValue() : argSignalThreshold.getValue())),
                    50);

        // set cropping values
        grabber.setCropping(
                    std::max(0, argCropLeft.getValue()),
                    std::max(0, argCropRight.getValue()),
                    std::max(0, argCropTop.getValue()),
                    std::max(0, argCropBottom.getValue()));

        // set 3D mode if applicable
        if (arg3DSBS.isSet())
        {
            grabber.set3D(VIDEO_3DSBS);
        }
        else if (arg3DTAB.isSet())
        {
            grabber.set3D(VIDEO_3DTAB);
        }

        // run the grabber
        if (argScreenshot.isSet())
        {
            ScreenshotHandler handler("screenshot.png");
            QObject::connect(&grabber, SIGNAL(newFrame(Image<ColorRgb>)), &handler, SLOT(receiveImage(Image<ColorRgb>)));
            grabber.start();
            QCoreApplication::exec();
            grabber.stop();
        }
        else
        {
            ProtoConnectionWrapper handler(argAddress.getValue(), argPriority.getValue(), 1000, argSkipReply.isSet());
            QObject::connect(&grabber, SIGNAL(newFrame(Image<ColorRgb>)), &handler, SLOT(receiveImage(Image<ColorRgb>)));
            grabber.start();
            QCoreApplication::exec();
            grabber.stop();
        }
    }
    catch (const std::runtime_error & e)
    {
        // An error occured. Display error and quit
        std::cerr << e.what() << std::endl;
        return 1;
    }

    return 0;
}