CTK/Libs/DICOM/Core/ctkDICOMRetrieve.cpp

/*=========================================================================

  Library:   CTK
 
  Copyright (c) 2010  Kitware Inc.

  Licensed under the Apache License, Version 2.0 (the "License");
  you may not use this file except in compliance with the License.
  You may obtain a copy of the License at

      http://www.commontk.org/LICENSE

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
 
=========================================================================*/

// Qt includes
#include <QSqlQuery>
#include <QSqlRecord>
#include <QVariant>
#include <QDate>
#include <QStringList>
#include <QSet>
#include <QFile>
#include <QDirIterator>
#include <QFileInfo>
#include <QDebug>

// ctkDICOM includes
#include "ctkDICOMRetrieve.h"
#include "ctkLogger.h"

// DCMTK includes
#ifndef WIN32
  #define HAVE_CONFIG_H 
#endif
#include "dcmtk/dcmnet/dimse.h"
#include "dcmtk/dcmnet/diutil.h"

#include <dcmtk/dcmdata/dcfilefo.h>
#include <dcmtk/dcmdata/dcfilefo.h>
#include <dcmtk/dcmdata/dcdeftag.h>
#include <dcmtk/dcmdata/dcdatset.h>
#include <dcmtk/ofstd/ofcond.h>
#include <dcmtk/ofstd/ofstring.h>
#include <dcmtk/ofstd/ofstd.h>        /* for class OFStandard */
#include <dcmtk/dcmdata/dcddirif.h>   /* for class DicomDirInterface */

#include <dcmtk/dcmnet/scu.h>

static ctkLogger logger ( "org.commontk.dicom.DICOMRetrieve" );

//------------------------------------------------------------------------------
class ctkDICOMRetrievePrivate: public ctkPrivate<ctkDICOMRetrieve>
{
public:
  ctkDICOMRetrievePrivate();
  ~ctkDICOMRetrievePrivate();
  QString CallingAETitle;
  QString CalledAETitle;
  QString Host;
  int CallingPort;
  int CalledPort;
  DcmSCU SCU;
  DcmDataset* parameters;
};

//------------------------------------------------------------------------------
// ctkDICOMRetrievePrivate methods

//------------------------------------------------------------------------------
ctkDICOMRetrievePrivate::ctkDICOMRetrievePrivate()
{
  parameters = new DcmDataset();
}

//------------------------------------------------------------------------------
ctkDICOMRetrievePrivate::~ctkDICOMRetrievePrivate()
{
  delete parameters;
}



//------------------------------------------------------------------------------
// ctkDICOMRetrieve methods

//------------------------------------------------------------------------------
ctkDICOMRetrieve::ctkDICOMRetrieve()
{
}

//------------------------------------------------------------------------------
ctkDICOMRetrieve::~ctkDICOMRetrieve()
{
}

/// Set methods for connectivity
void ctkDICOMRetrieve::setCallingAETitle ( QString callingAETitle )
{
  CTK_D(ctkDICOMRetrieve);
  d->CallingAETitle = callingAETitle;
}
const QString& ctkDICOMRetrieve::callingAETitle() 
{
  CTK_D(ctkDICOMRetrieve);
  return d->CallingAETitle;
}
void ctkDICOMRetrieve::setCalledAETitle ( QString calledAETitle )
{
  CTK_D(ctkDICOMRetrieve);
  d->CalledAETitle = calledAETitle;
}
const QString& ctkDICOMRetrieve::calledAETitle()
{
  CTK_D(ctkDICOMRetrieve);
  return d->CalledAETitle;
}
void ctkDICOMRetrieve::setHost ( QString host )
{
  CTK_D(ctkDICOMRetrieve);
  d->Host = host;
}
const QString& ctkDICOMRetrieve::host()
{
  CTK_D(ctkDICOMRetrieve);
  return d->Host;
}
void ctkDICOMRetrieve::setCallingPort ( int port ) 
{
  CTK_D(ctkDICOMRetrieve);
  d->CallingPort = port;
}
int ctkDICOMRetrieve::callingPort()
{
  CTK_D(ctkDICOMRetrieve);
  return d->CallingPort;
}

void ctkDICOMRetrieve::setCalledPort ( int port ) 
{
  CTK_D(ctkDICOMRetrieve);
  d->CalledPort = port;
}
int ctkDICOMRetrieve::calledPort()
{
  CTK_D(ctkDICOMRetrieve);
  return d->CalledPort;
}



//------------------------------------------------------------------------------
void ctkDICOMRetrieve::retrieveSeries ( QString seriesInstanceUID, QDir directory ) {
  CTK_D(ctkDICOMRetrieve);
  logger.info ( "Starting retrieveSeries" );
  DcmSCU scu;
  scu.setAETitle ( this->callingAETitle().toStdString() );
  scu.setPeerAETitle ( this->calledAETitle().toStdString() );
  scu.setPeerHostName ( this->host().toStdString() );
  scu.setPeerPort ( this->calledPort() );

  logger.error ( "Setting Transfer Syntaxes" );
  OFList<OFString> transferSyntaxes;
  transferSyntaxes.push_back ( UID_LittleEndianExplicitTransferSyntax );
  transferSyntaxes.push_back ( UID_BigEndianExplicitTransferSyntax );
  transferSyntaxes.push_back ( UID_LittleEndianImplicitTransferSyntax );
  scu.addPresentationContext ( UID_MOVEStudyRootQueryRetrieveInformationModel, transferSyntaxes );


  if ( !scu.initNetwork().good() ) 
    {
    logger.error ( "Error initializing the network" );
    return;
    }
  logger.debug ( "Negotiating Association" );
  if ( !scu.negotiateAssociation().good() )
    {
    logger.error ( "Error negotiating association" );
    return;
    }
  // Clear the query
  unsigned long elements = d->parameters->card();
  // Clean it out
  for ( unsigned long i = 0; i < elements; i++ ) 
    {
    d->parameters->remove ( (unsigned long) 0 );
    }
  d->parameters->putAndInsertString ( DCM_SeriesInstanceUID, seriesInstanceUID.toStdString().c_str() );

  MOVEResponses *responses = new MOVEResponses();
  OFCondition status = scu.sendMOVERequest ( 0, d->parameters, responses );
  if ( status.good() )
    {
    logger.debug ( "Find succeded" );
    }
  else
    {
    logger.error ( "Find failed" );
    }
  return;
  
}
void ctkDICOMRetrieve::retrieveStudy ( QString studyInstanceUID, QDir directory ) {
}