mlRTObjectSave.h

Go to the documentation of this file.

// Copyright (c) Fraunhofer MEVIS, Germany. All rights reserved.
// **InsertLicense** code
//----------------------------------------------------------------------------------
 
//----------------------------------------------------------------------------------
 
 
#ifndef __mlRTObjectSave_H
#define __mlRTObjectSave_H
 
// Local includes
#include "MLDcmtkMLConverterModulesSystem.h"
 
// ML includes
#include <mlModuleIncludes.h>
#include <CSOBase/CSOList.h>
#include "DCMTree_Tree.h"
#include <ThirdPartyWarningsDisable.h>
#include <mlBackgroundTaskBaseModule.h>
#include <ThirdPartyWarningsRestore.h>
 
ML_START_NAMESPACE
 
class RTBase;
 
class MLDCMTKMLCONVERTERMODULES_EXPORT RTObjectSaveBackgroundTask
    : public ModuleBackgroundTask
{
public:
  RTObjectSaveBackgroundTask(Module* module, FloatField* progressField, StringField* statusField);
 
  virtual ~RTObjectSaveBackgroundTask()
  {
    _dicomTree.reset();
  }
 
  void setDicomTree(DCMTree::TreePtr dicomTree)
  {
    if (isSynchronous())
    {
      _dicomTree = dicomTree;
    }
    else
    {
      _dicomTree = dicomTree->copy();
    }
  }
 
  void setFilename(const std::string& filename) { _filename = filename; }
 
  bool hasSuccessfullySaved() const { return _savingWasSuccessful; }
 
  virtual void run() override;
 
private:
  std::string _filename;
  bool _savingWasSuccessful;
  DCMTree::TreePtr _dicomTree;
};
 
 
class MLDCMTKMLCONVERTERMODULES_EXPORT RTObjectSave : public BackgroundTaskBaseModule
{
public:
 
  RTObjectSave ();
 
  virtual void handleNotification (Field *field) override;
 
private:
 
  bool _isInputRTObjectValid();
 
  bool _getInputDicomTreeFromBaseField();
 
  virtual void taskFinished(BackgroundTask* task) override;
 
  virtual ModuleBackgroundTask* createTask() override;
 
 
  BoolField* _autoSaveFld;
  BaseField* _rtObjectFld;
  BoolField* _isBaseInputValidFld;
  StringField* _statusFld;
  StringField* _dcmFilenameFld;
  DCMTree::TreePtr _dicomTree;
  RTBase* _rtObject;
 
  ML_MODULE_CLASS_HEADER(RTObjectSave);
};
 
 
ML_END_NAMESPACE
 
#endif // __mlRTObjectSave_H