cxImageServer.cpp

Go to the documentation of this file.

/*=========================================================================
This file is part of CustusX, an Image Guided Therapy Application.
                 
Copyright (c) SINTEF Department of Medical Technology.
All rights reserved.
                 
CustusX is released under a BSD 3-Clause license.
                 
See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.
=========================================================================*/

#include "cxImageServer.h"

#include "cxLogger.h"
#include "cxTypeConversions.h"
#include <iostream>
#include <QCoreApplication>
#include <QHostAddress>
#include <QNetworkInterface>
#include <QTcpSocket>
#include "cxCommandlineImageStreamerFactory.h"
//#include "cxSender.h"
#include "cxGrabberSenderQTcpSocket.h"

namespace cx
{

ImageServer::ImageServer(QObject* parent) :
        QTcpServer(parent)
{}

bool ImageServer::initialize()
{
        bool ok = false;

        StringMap args = cx::extractCommandlineOptions(QCoreApplication::arguments());
        mImageSender = CommandlineImageStreamerFactory().getFromArguments(args);
        if(!mImageSender)
                return false;

        ok = true;

        return ok;
}

bool ImageServer::startListen(int port)
{
        bool started = this->listen(QHostAddress::Any, port);

        if (started)
        {
                //Find IP adresses
                std::cout <<  "Server IP adresses: " << std::endl;
                foreach(QNetworkInterface interface, QNetworkInterface::allInterfaces())
                {
                        if (interface.flags().testFlag(QNetworkInterface::IsRunning))
                        foreach (QNetworkAddressEntry entry, interface.addressEntries())
                        {
                                if ( interface.hardwareAddress() != "00:00:00:00:00:00" && entry.ip().toString() != "127.0.0.1" && entry.ip().toString().contains(".") )
                                std::cout << string_cast(interface.name()) << " " << entry.ip().toString() << std::endl;
                        }
                }

                std::cout << QString("Server is listening to port %2").arg(this->serverPort()).toStdString() << std::endl;
                return true;
        }
        else
        {
                std::cout << "Server failed to start. Error: " << this->errorString().toStdString() << std::endl;
                return false;
        }
}

ImageServer::~ImageServer()
{
}

void ImageServer::incomingConnection(qintptr socketDescriptor)
{
        std::cout << "Server: Incoming connection..." << std::endl;

        if (mSocket != 0)
        {
                reportError("The image server can only handle a single connection.");
                return;
        }

        mSocket = new QTcpSocket();
        connect(mSocket, SIGNAL(disconnected()), this, SLOT(socketDisconnectedSlot()));
        mSocket->setSocketDescriptor(socketDescriptor);
        QString clientName = mSocket->localAddress().toString();
        report("Connected to "+clientName+". Session started.");
        SenderPtr sender(new GrabberSenderQTcpSocket(mSocket));

        mImageSender->startStreaming(sender);
}

void ImageServer::socketDisconnectedSlot()
{
        if (mImageSender)
                mImageSender->stopStreaming();

        if (mSocket)
        {
                QString clientName = mSocket->localAddress().toString();
                report("Disconnected from "+clientName+". Session ended.");
                mSocket->deleteLater();
        }
}

void ImageServer::printHelpText()
{
        std::cout << getArgumentHelpText(qApp->applicationName());
        std::cout << std::endl;
        std::cout << std::endl;
        std::cout << "Press Ctrl + C to close the server."<< std::endl;
        std::cout << std::endl;
}

QString ImageServer::getArgumentHelpText(QString applicationName)
{
        std::stringstream ss;
        cx::CommandlineImageStreamerFactory factory;

        ss << "Usage: " << applicationName << " (--arg <argval>)*" << std::endl;
        ss << "    --port   : Tcp/IP port # (default=18333)" << std::endl;
        ss << "    --type   : Grabber type  (default=" << factory.getDefaultSenderType().toStdString() << ")"
                << std::endl;
        ss << std::endl;
        ss << "    Select one of the types below:" << std::endl;

        QStringList types = factory.getSenderTypes();
        for (int i = 0; i < types.size(); ++i)
        {
                QStringList args = factory.getArgumentDescription(types[i]);
                ss << std::endl;
                ss << "      type = " << types[i].toStdString() << std::endl;
                for (int j = 0; j < args.size(); ++j)
                        ss << "        " << args[j].toStdString() << std::endl;
        }
        return qstring_cast(ss.str());
}

}