vv/vvReadState.cxx

   1 #include "vvReadState.h"
   2 #include "vvMainWindow.h"
   3 #include "vvSlicerManager.h"
   4
   5 #include <qtreewidget.h>
   6
   7 #include <QFile>
   8 #include <QXmlStreamReader>
   9
  10 #include <cassert>
  11 #include <string>
  12
  13 vvReadState::vvReadState() : m_XmlReader(new QXmlStreamReader), m_File(new QFile)
  14 {
  15   m_NumImages = 0;
  16 }
  17
  18 vvReadState::~vvReadState()
  19 {
  20 }
  21
  22 void vvReadState::Run(vvMainWindow* vvWindow, const std::string& file)
  23 {
  24   assert(vvWindow);
  25
  26   m_NumImages = 0;
  27   m_File->setFileName(file.c_str());
  28   m_File->open(QIODevice::ReadOnly);
  29   m_XmlReader->setDevice(m_File.get());
  30   m_Window = vvWindow;
  31   QTreeWidget* tree = m_Window->GetTree();
  32   m_TreeItemCount = tree->topLevelItemCount();
  33
  34   ReadGUI();
  35   ReadTree();
  36 }
  37
  38 void vvReadState::ReadTree()
  39 {
  40   std::string value;
  41
  42   while (!m_XmlReader->atEnd()) {
  43     m_XmlReader->readNext();
  44     value = m_XmlReader->qualifiedName().toString().toStdString();
  45     if (m_XmlReader->isStartElement()) {
  46       if (value == "Image")
  47         value = ReadImage();
  48     }
  49   }
  50
  51   if (m_XmlReader->hasError())
  52     std::cout << "Error " << m_XmlReader->error() << " XML " << std::endl;
  53 }
  54
  55 std::string  vvReadState::ReadImage()
  56 {
  57   std::string value;
  58   int current_index = -1;
  59   std::vector<std::string> files(1);
  60
  61   QXmlStreamAttributes attributes = m_XmlReader->attributes();
  62   if (!m_XmlReader->hasError())
  63     current_index = attributes.value("Index").toString().toInt();
  64
  65   current_index += m_TreeItemCount;
  66
  67   while (!m_XmlReader->isEndElement() || value != "Image") {
  68     m_XmlReader->readNext();
  69     value = m_XmlReader->qualifiedName().toString().toStdString();
  70     //std::cout << "Value = " << value << std::endl;
  71     if (m_XmlReader->isStartElement()) {
  72       if (value == "FileName") {
  73         files[0] = m_XmlReader->readElementText().toStdString();
  74         if (!m_XmlReader->hasError()) {
  75           m_Window->LoadImages(files, vvImageReader::IMAGE);
  76         }
  77       }
  78       else if (current_index >= 0) {
  79         if (value == "Fusion")
  80           value = ReadFusion(current_index);
  81         else if (value == "Overlay")
  82           value = ReadOverlay(current_index);
  83         else if (value == "Vector")
  84           value = ReadVector(current_index);
  85       }
  86     }
  87   }
  88
  89   if (!m_XmlReader->hasError())
  90     m_NumImages++;
  91
  92   return value;
  93 }
  94
  95 std::string vvReadState::ReadFusion(int index)
  96 {
  97   std::string file, value;
  98   int vali;
  99   double vald;
 100   vvSlicerManager* slicerManager = m_Window->GetSlicerManagers()[index];
 101   while (!m_XmlReader->isEndElement() || value != "Fusion") {
 102     m_XmlReader->readNext();
 103     value = m_XmlReader->qualifiedName().toString().toStdString();
 104     if (m_XmlReader->isStartElement()) {
 105       if (value == "FileName") {
 106         file = m_XmlReader->readElementText().toStdString();
 107         if (!m_XmlReader->hasError())
 108           m_Window->AddFusionImage(index, file.c_str());
 109       }
 110       if (value == "FusionOpacity") {
 111         vali = m_XmlReader->readElementText().toInt();
 112         if (!m_XmlReader->hasError())
 113           slicerManager->SetFusionOpacity(vali);
 114       }
 115       if (value == "FusionThresholdOpacity") {
 116         vali = m_XmlReader->readElementText().toInt();
 117         if (!m_XmlReader->hasError())
 118           slicerManager->SetFusionThresholdOpacity(vali);
 119       }
 120       if (value == "FusionColorMap") {
 121         vali = m_XmlReader->readElementText().toInt();
 122         if (!m_XmlReader->hasError())
 123           slicerManager->SetFusionColorMap(vali);
 124       }
 125       if (value == "FusionWindow") {
 126         vald = m_XmlReader->readElementText().toDouble();
 127         if (!m_XmlReader->hasError())
 128           slicerManager->SetFusionWindow(vald);
 129       }
 130       if (value == "FusionLevel") {
 131         vald = m_XmlReader->readElementText().toDouble();
 132         if (!m_XmlReader->hasError())
 133           slicerManager->SetFusionLevel(vald);
 134       }
 135     }
 136   }
 137   m_Window->ImageInfoChanged();
 138   return value;
 139 }
 140
 141 std::string vvReadState::ReadOverlay(int index)
 142 {
 143   std::string file, value;
 144   while (!m_XmlReader->isEndElement() || value != "Overlay") {
 145     m_XmlReader->readNext();
 146     value = m_XmlReader->qualifiedName().toString().toStdString();
 147     if (m_XmlReader->isStartElement()) {
 148       if (value == "FileName") {
 149         file = m_XmlReader->readElementText().toStdString();
 150         if (!m_XmlReader->hasError())
 151           m_Window->AddOverlayImage(index, file.c_str());
 152       }
 153     }
 154   }
 155   return value;
 156 }
 157
 158 std::string vvReadState::ReadVector(int index)
 159 {
 160   std::string file, value;
 161   while (!m_XmlReader->isEndElement() || value != "Vector") {
 162     m_XmlReader->readNext();
 163     value = m_XmlReader->qualifiedName().toString().toStdString();
 164     if (m_XmlReader->isStartElement()) {
 165       if (value == "FileName") {
 166         file = m_XmlReader->readElementText().toStdString();
 167         if (!m_XmlReader->hasError())
 168           m_Window->AddField(file.c_str(), index);
 169       }
 170     }
 171   }
 172   return value;
 173 }
 174
 175 void vvReadState::ReadGUI()
 176 {
 177
 178 }
 179