DICOMIGTLApp.cpp

#include <iostream>
#include <QtGui> 

#include "DICOMIGTLApp.h"
 

#include "igtlOSUtil.h"
#include "igtlImageMessage.h"
#include "igtlServerSocket.h"
#include "igtlMultiThreader.h"
#include "igtlOSUtil.h"
#include "igtlTCPConnectorServerOIGTL.h"
#include "igtlTCPConnectorDICOMDirectoryMonitor.h"

const int DICOMIGTLApp::StatusColorTable[][3] = {
  {100, 100, 100},  // STOP
  {100, 100, 200},  // WAITING
  {100, 200, 100},  // CONNECTED
  {200, 100, 100}   // ERROR
};

const int DICOMIGTLApp::DataIOColorTable[][3] = {
  {50, 50, 50},        // None
  {100, 200, 100},  // Received
  {200, 100, 100}   // ERROR
};

 
DICOMIGTLApp::DICOMIGTLApp(QWidget *parent)
{
  setupUi(this); // this sets up GUI

  fDirectoryMonitorActive = false;
  fClientActive  = false;

  // Signals and Slots
  connect(pbQuit, SIGNAL( clicked() ), this, SLOT( quit() ));
  connect(pbAbout, SIGNAL( clicked() ), this, SLOT( about() ));
  connect(pbDirectoryMonitorActivate, SIGNAL( clicked() ),
          this, SLOT( directoryMonitorActivateClicked() ));
  connect(pbClientActivate, SIGNAL( clicked() ), this, SLOT( clientActivateClicked() ));
  connect(pbInputDirectory, SIGNAL( clicked() ), this, SLOT( pbInputDirectoryClicked() ));
  connect(pbProcessedDirectory, SIGNAL( clicked() ), this, SLOT( pbProcessedDirectoryClicked() ));

  connect(leInputDirectory, SIGNAL( textChanged( const QString &  ) ),
          this, SLOT( inputDirectoryChanged( const QString &  ) ));
  connect(leProcessedDirectory, SIGNAL( textChanged( const QString & ) ),
          this, SLOT( processedDirectoryChanged( const QString & ) ));
  connect(leOpenIGTLinkPort, SIGNAL( textChanged( const QString & ) ),
          this, SLOT( igtlPortChanged( const QString &  ) ));

  // Default values
  QString qs;
  leInputDirectory->setText(tr("~/"));
  leProcessedDirectory->setText(tr("~/"));
  leOpenIGTLinkPort->setText(qs.setNum(DEFAULT_OIGTL_PORT));

  // Time for GUI update (every 200 ms)
  timer = new QTimer(this);
  connect(timer, SIGNAL(timeout()), this, SLOT(updateStatus()));
  timer->start(200); 
  
  //// OpenIGTLink Server Socket
  oigtlConnector = igtl::TCPConnectorServerOIGTL::New();
  oigtlConnector->SetPort(18944);
  
  dicomConnector = igtl::TCPConnectorDICOMDirectoryMonitor::New();
  
  // Connect connectors
  //oigtlConnector->SetOutputConnector(rmpConnector);
  dicomConnector->SetOutputConnector(oigtlConnector);

  this->Threader = igtl::MultiThreader::New();
  this->Threader->SpawnThread((igtl::ThreadFunctionType)
			      &igtl::TCPConnectorServerOIGTL::MonitorThreadFunction,
                              oigtlConnector);
  this->Threader->SpawnThread((igtl::ThreadFunctionType)
			      &igtl::TCPConnectorDICOMDirectoryMonitor::MonitorThreadFunction,
                              dicomConnector);
}
 
 
void DICOMIGTLApp::getPath()
{
  // QString path;
  
  //path = QFileDialog::getOpenFileName(
  //    this,
  //    "Choose a file to open",
  //    QString::null,
  //    QString::null);
  //
  //lineEdit->setText( path );
}
 
 
void DICOMIGTLApp::about() 
{
  QMessageBox::about(this,"About DICOMIGTL",
                     "DICOM OpenIGTLink Server\n\n"
                     "Copyright (C) 2015\n"
                     "Brigham and Women's Hospital.\n\n"
                     );
}

void DICOMIGTLApp::directoryMonitorActivateClicked()
{
  if (fDirectoryMonitorActive)
    {
    pbDirectoryMonitorActivate->setText("Activate");
    fDirectoryMonitorActive = false;
    if (dicomConnector.IsNotNull())
      {
      dicomConnector->Deactivate();
      }
    }
  else
    {
    pbDirectoryMonitorActivate->setText("Deactivate");
    fDirectoryMonitorActive = true;
    if (dicomConnector.IsNotNull())
      {
      dicomConnector->Activate();
      }
    }
}


void DICOMIGTLApp::clientActivateClicked()
{
  if (fClientActive)
    {
    pbClientActivate->setText("Activate");
    fClientActive = false;
    if (oigtlConnector.IsNotNull())
      {
      oigtlConnector->Deactivate();
      
      }
    }
  else
    {
    pbClientActivate->setText("Deactivate");
    fClientActive = true;
    if (oigtlConnector.IsNotNull())
      {
      std::cerr << "Activating OpenIGTLink connector with:" << std::endl;
      std::cerr << "    Port: " << igtlPort.toInt() << std::endl;
      oigtlConnector->SetPort(igtlPort.toInt());
      oigtlConnector->Activate();
      }
    }
}


void DICOMIGTLApp::pbInputDirectoryClicked()
{
  QString dir = QFileDialog::getExistingDirectory(
    this, 
    tr("Open Input Directory"),
    "~/",
    QFileDialog::ShowDirsOnly|QFileDialog::DontResolveSymlinks);
  leInputDirectory->setText(dir);
}

void DICOMIGTLApp::pbProcessedDirectoryClicked()
{
  QString dir = QFileDialog::getExistingDirectory(
    this, 
    tr("Open Processed Directory"),
    "~/",
    QFileDialog::ShowDirsOnly|QFileDialog::DontResolveSymlinks);
  leProcessedDirectory->setText(dir);
}

void DICOMIGTLApp::inputDirectoryChanged( const QString & text )
{
  scannerAddress = text;
}

void DICOMIGTLApp::processedDirectoryChanged( const QString & text )
{
  controlPort = text;
}

void DICOMIGTLApp::igtlPortChanged( const QString & text )
{
  igtlPort = text;
}

void DICOMIGTLApp::changeStatusTextColor(QLineEdit* le, int status)
{
  if (status < igtl::TCPConnectorBase::STATUS_NUM && status >= 0)
    {
    QPalette p = le->palette();
    p.setColor( le->backgroundRole(),
                QColor(StatusColorTable[status][0],
                       StatusColorTable[status][1],
                       StatusColorTable[status][2]) );
    le->setPalette(p);
    }
}

void DICOMIGTLApp::changeDataIOTextColor(QLineEdit* le, int status)
{
  if (status < 3 && status >= 0)
    {
    QPalette p = le->palette();
    p.setColor( le->backgroundRole(),
                QColor(DataIOColorTable[status][0],
                       DataIOColorTable[status][1],
                       DataIOColorTable[status][2]) );
    le->setPalette(p);
    }
}

void DICOMIGTLApp::updateStatus() 
{
  if (oigtlConnector.IsNotNull())
    {
    leStatusClient->setText(oigtlConnector->GetStatusString());
    changeStatusTextColor(leStatusClient, oigtlConnector->GetStatus());
    leOpenIGTLinkPort->setEnabled(oigtlConnector->GetStatus() == igtl::TCPConnectorBase::STATUS_STOP);
    }
  if (dicomConnector.IsNotNull())
    {
    leStatusDirectoryMonitor->setText(dicomConnector->GetStatusString());
    changeStatusTextColor(leStatusDirectoryMonitor, dicomConnector->GetStatus());
    bool editFlag = dicomConnector->GetStatus() == igtl::TCPConnectorBase::STATUS_STOP;
    leInputDirectory->setEnabled(editFlag);
    leProcessedDirectory->setEnabled(editFlag);
    }
}

void DICOMIGTLApp::quit() 
{

  if (oigtlConnector.IsNotNull())
    {
    oigtlConnector->CloseThread();
    }
  
  std::cerr << "CLOSING..." << std::endl;
  timer->stop(); 
  close();
}