mlMultiFileVolumeListDcmtkTools.h

Go to the documentation of this file.

// Copyright (c) Fraunhofer MEVIS, Germany. All rights reserved.
// **InsertLicense** code
//----------------------------------------------------------------------------------
 
//----------------------------------------------------------------------------------
#pragma once
 
// ML-includes
// Local includes
#include "MLMultiFileVolumeListDcmtkOutputsSystem.h"
#include "mlTypeDefs.h"
 
#include <dcmtk/dcmdata/dcfilefo.h>
#include <dcmtk/dcmdata/dcdeftag.h>
 
typedef boost::shared_ptr<DcmFileFormat> DcmFileFormatPtr;
 
ML_START_NAMESPACE
 
//--------------------------------------------------------------------------------------------------
//--------------------------------------------------------------------------------------------------
inline bool loadDcmtkFile(const std::string& fName, DcmFileFormat& dcmfile, std::string* errVal)
{
  OFCondition cond = dcmfile.loadFile(fName.c_str());
  if (cond.bad()) {
    if (errVal) { *errVal = std::string("loadDcmFile:") + std::string(cond.text()); }
    return false;
  }
  cond = dcmfile.loadAllDataIntoMemory();
  if (cond.bad()) {
    if (errVal) { *errVal = std::string("loadDcmFile:") + std::string(cond.text()); }
    return false;
  }
  return true;
}
 
//--------------------------------------------------------------------------------------------------
//--------------------------------------------------------------------------------------------------
inline OFString getDcmtkModality(DcmFileFormat& dcmfile, std::string* errVal)
{
  DcmDataset *dataset = dcmfile.getDataset();
  OFString modality;
  OFCondition cond = dataset->findAndGetOFString(DCM_Modality, modality);
  if (cond.bad()) {
    if (errVal){ *errVal = std::string("getModality:") + std::string(cond.text()); }
  }
  return modality;
}
 
//--------------------------------------------------------------------------------------------------
//--------------------------------------------------------------------------------------------------
struct DefaultDcmtkReaderFunctor
{
  template <typename DCMObjType>
    OFCondition read(DCMObjType &dcmObj, DcmItem &item)
    {
       return dcmObj->read(item);
    }
};
 
//--------------------------------------------------------------------------------------------------
//--------------------------------------------------------------------------------------------------
template <typename DCMObjType, typename DCMObjBoostPtrType, class DCMReaderFunctor=DefaultDcmtkReaderFunctor>
  DCMObjBoostPtrType getDCMTKObjFromFile(DcmFileFormat     &dcmFile,
                                         std::string       *errVal,
                                         const std::string &modalityStr,
                                         const std::string &typeStr,
                                         DCMReaderFunctor  *readerFunctor=NULL)
{
  OFString modality = getDcmtkModality(dcmFile, errVal);
  if (modality.compare(modalityStr.c_str()) == 0) {
    DcmDataset *dataset = dcmFile.getDataset();
    DcmItem &item = dynamic_cast<DcmItem&>(*dataset);
    DCMObjBoostPtrType dcmObj = DCMObjBoostPtrType(new DCMObjType());
    DCMReaderFunctor reader;
    OFCondition cond = readerFunctor ? readerFunctor->read(dcmObj, item) : reader.read(dcmObj, item);
    if (cond.bad()) {
      if (errVal) { *errVal = std::string("get") + typeStr + ": " + std::string(cond.text()); }
    } else {
      return dcmObj;
    }
  } else {
    if (errVal) { *errVal = std::string("get") + typeStr + ": Data set contains no " + typeStr + " object"; }
  }
  return DCMObjBoostPtrType();
}
 
//--------------------------------------------------------------------------------------------------
//--------------------------------------------------------------------------------------------------
template <typename DCMObjType, typename DCMObjBoostPtrType, class DCMReaderFunctor=DefaultDcmtkReaderFunctor>
  DCMObjBoostPtrType getDCMTKObjFromString(const std::string  &fName,
                                           std::string        *errVal,
                                           DcmFileFormatPtr   *dcmFilePtr,
                                           const std::string  &modalityStr,
                                           const std::string  &typeStr,
                                           DCMReaderFunctor   *readerFunctor=NULL)
{
  // Allow all three combinations:
  // dcmFilePtr == NULL                        : A temporary DcmFileFormat is used for loading and a boost pointer
  //                                             to the loaded object is returned.
  // dcmFilePtr != NULL and *dcmFilePtr == NULL: Create a new DcmFileFormat object and set *dcmFilePtr to it and a
  //                                             boost pointer to the loaded object is returned.
  //                                             The file pointer is set even if loading the fName fails.
  // dcmFilePtr != NULL and *dcmFilePtr != NULL: Load file from filename with already existing *dcmFilePtr
  //                                             so no new DcmFileFormat is created. A boost pointer to the
  //                                             loaded object is returned.
  bool success = true;
  if (dcmFilePtr && !(*dcmFilePtr)) {
    // No DcmFileFormat is available. Create a new one and return it in *dcmFilePtr.
    *dcmFilePtr = DcmFileFormatPtr(new DcmFileFormat());
    success = (*dcmFilePtr != NULL);
  }
 
  // Temporary loader used in case of NULL dcmFilePtr.
  DcmFileFormat dcmfile;
  if (success) {
    // Either dcmFilePtr && *dcmFilePtr should be valid or dcmFilePtr is NULL.
    success &= loadDcmtkFile(fName, dcmFilePtr ? **dcmFilePtr : dcmfile, errVal);
  }
  if (success) {
    // DcmFile could be loaded either with **dcmFilePtr or with dcmfile loader.
    // Either dcmFilePtr && *dcmFilePtr should be valid or dcmFilePtr is NULL.
    return getDCMTKObjFromFile<DCMObjType, DCMObjBoostPtrType, DCMReaderFunctor>(dcmFilePtr ? **dcmFilePtr : dcmfile,
                                                                                 errVal,
                                                                                 modalityStr,
                                                                                 typeStr,
                                                                                 readerFunctor);
  }
 
  // If we reached here then loading or DcmFileFormat object creation failed, return a NULL object.
  return DCMObjBoostPtrType();
}
 
ML_END_NAMESPACE