vv/vvLandmarksPanel.cxx

   1 /*=========================================================================
   2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
   3
   4   Authors belong to:
   5   - University of LYON              http://www.universite-lyon.fr/
   6   - Léon Bérard cancer center       http://www.centreleonberard.fr
   7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
   8
   9   This software is distributed WITHOUT ANY WARRANTY; without even
  10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  11   PURPOSE.  See the copyright notices for more information.
  12
  13   It is distributed under dual licence
  14
  15   - BSD        See included LICENSE.txt file
  16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  17 ===========================================================================**/
  18 #ifndef _vvLandmarksPanel_CXX
  19 #define _vvLandmarksPanel_CXX
  20 #include "vvLandmarksPanel.h"
  21
  22 #include <QtGui>
  23 #include <Qt>
  24 #include "QTreePushButton.h"
  25 #include "vvLandmarks.h"
  26
  27 #include <vtksys/SystemTools.hxx>
  28 #include <clitkDD.h>
  29
  30 //====================================================================
  31 vvLandmarksPanel::vvLandmarksPanel(QWidget * parent):QWidget(parent)
  32 {
  33   setupUi(this);
  34
  35   tableWidget->verticalHeader()->hide();
  36   tableWidget->setSelectionBehavior(QAbstractItemView::SelectRows);
  37   tableWidget->setSelectionMode(QAbstractItemView::SingleSelection);
  38   loadButton->setEnabled(0);
  39   saveButton->setEnabled(0);
  40   removeButton->setEnabled(0);
  41   connect(loadButton, SIGNAL(clicked()),this,SLOT(Load()));
  42   connect(saveButton, SIGNAL(clicked()),this,SLOT(Save()));
  43   connect(removeButton, SIGNAL(clicked()),this,SLOT(RemoveSelectedPoints()));
  44   connect(removeAllButton, SIGNAL(clicked()),this,SLOT(RemoveAllPoints()));
  45   connect(tableWidget,SIGNAL(cellChanged(int,int)),this,SLOT(CommentsChanged(int,int)));
  46 }
  47
  48 void vvLandmarksPanel::Load()
  49 {
  50   QString file = QFileDialog::getOpenFileName(this,tr("Load Landmarks"),
  51                  mCurrentPath.c_str(),tr("Landmarks ( *.txt *.pts)"));
  52   if (!file.isEmpty()) {
  53     std::vector<std::string> files(1, file.toStdString());
  54     LoadFromFile(files);
  55   }
  56 }
  57
  58 bool vvLandmarksPanel::LoadFromFile(std::vector<std::string> files)
  59 {
  60   if (!mCurrentLandmarks->LoadFile(files))
  61     return false;
  62
  63   SetCurrentLandmarks(mCurrentLandmarks,2);
  64   emit UpdateRenderWindows();
  65   return true;
  66 }
  67
  68 void vvLandmarksPanel::Save()
  69 {
  70   QString file = QFileDialog::getSaveFileName(this,
  71                  tr("Save Landmarks"),
  72                  mCurrentPath.c_str(),tr("Landmarks ( *.txt)"));
  73   if (!file.isEmpty()) {
  74     std::string filename = vtksys::SystemTools::GetFilenamePath(file.toStdString());
  75     filename += "/" + vtksys::SystemTools::GetFilenameWithoutLastExtension(file.toStdString());
  76     filename += ".txt";
  77     mCurrentLandmarks->SaveFile(filename.c_str());
  78   }
  79 }
  80
  81 void vvLandmarksPanel::RemoveSelectedPoints()
  82 {
  83   if (tableWidget->rowCount() > 0) {
  84     QList<QTableWidgetItem *> items = tableWidget->selectedItems();
  85     if (items.empty()) {
  86       tableWidget->removeRow(tableWidget->rowCount()-1);
  87       mCurrentLandmarks->RemoveLastLandmark();
  88     }
  89     else {
  90       // we're using single-selection mode
  91       int row = items[0]->row();
  92       mCurrentLandmarks->RemoveLandmark(row);
  93       tableWidget->removeRow(row);
  94
  95       for (int i = row; i < tableWidget->rowCount(); i++) {
  96         QTableWidgetItem* iItem = tableWidget->item(i, 0);
  97         iItem->setText(QString::number(i));
  98       }
  99
 100     }
 101     emit UpdateRenderWindows();
 102   }
 103 }
 104
 105 void vvLandmarksPanel::RemoveAllPoints()
 106 {
 107   mCurrentLandmarks->RemoveAll();
 108   tableWidget->clearContents();
 109   tableWidget->setRowCount(0);
 110   emit UpdateRenderWindows();
 111 }
 112
 113 void vvLandmarksPanel::AddPoint()
 114 {
 115   AddPoint(mCurrentLandmarks->GetNumberOfPoints()-1);
 116 }
 117
 118 void vvLandmarksPanel::AddPoint(int landmarksIndex)
 119 {
 120   int rowIndex = tableWidget->rowCount();
 121 //   DD(rowIndex);
 122   tableWidget->setRowCount(rowIndex+1);
 123   tableWidget->setRowHeight(rowIndex,20);
 124   QTableWidgetItem* iItem = new QTableWidgetItem(QString::number(rowIndex));
 125   iItem->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
 126   tableWidget->setItem(rowIndex,0,iItem);
 127
 128   QTableWidgetItem* xItem = new QTableWidgetItem(
 129     QString::number(mCurrentLandmarks->GetCoordinates(landmarksIndex)[0],'f',1));
 130   xItem->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
 131   tableWidget->setItem(rowIndex,1,xItem);
 132
 133   QTableWidgetItem* yItem = new QTableWidgetItem(
 134     QString::number(mCurrentLandmarks->GetCoordinates(landmarksIndex)[1],'f',1));
 135   yItem->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
 136   tableWidget->setItem(rowIndex,2,yItem);
 137
 138   QTableWidgetItem* zItem = new QTableWidgetItem(
 139     QString::number(mCurrentLandmarks->GetCoordinates(landmarksIndex)[2],'f',1));
 140   zItem->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
 141   tableWidget->setItem(rowIndex,3,zItem);
 142
 143   QTableWidgetItem* tItem = new QTableWidgetItem(
 144     QString::number(mCurrentLandmarks->GetCoordinates(landmarksIndex)[3],'f',1));
 145   tItem->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
 146   tableWidget->setItem(rowIndex,4,tItem);
 147
 148
 149   QTableWidgetItem* vItem = new QTableWidgetItem(
 150     QString::number(mCurrentLandmarks->GetPixelValue(landmarksIndex),'f',1));
 151   vItem->setFlags(Qt::ItemIsEnabled | Qt::ItemIsSelectable);
 152   tableWidget->setItem(rowIndex,5,vItem);
 153
 154   tableWidget->setItem(rowIndex,6, new QTableWidgetItem(mCurrentLandmarks->GetComments(landmarksIndex).c_str()));
 155 }
 156
 157 void vvLandmarksPanel::SetCurrentLandmarks(vvLandmarks* lm,int time)
 158 {
 159   if (time != lm->GetTime())
 160     return;
 161
 162   loadButton->setEnabled(1);
 163   saveButton->setEnabled(1);
 164   removeButton->setEnabled(1);
 165   mCurrentLandmarks = lm;
 166   tableWidget->clearContents();
 167   tableWidget->setRowCount(0);
 168   for (unsigned int i = 0; i < mCurrentLandmarks->GetNumberOfPoints(); i++) {
 169       AddPoint(i);
 170   }
 171   //if (time > 1)
 172   //tableWidget->setColumnHidden(4,1);
 173   //else
 174   //tableWidget->setColumnHidden(4,0);
 175   tableWidget->resizeColumnsToContents();
 176 }
 177
 178 void vvLandmarksPanel::SetCurrentImage(std::string filename)
 179 {
 180   QString image = "<b>CurrentImage : </b>";
 181   image += vtksys::SystemTools::GetFilenameWithoutLastExtension(filename).c_str();
 182   nameLabel->setText(image);
 183 }
 184
 185 void vvLandmarksPanel::CommentsChanged(int row, int column)
 186 {
 187   if (column == 6) {
 188     mCurrentLandmarks->ChangeComments(row,std::string(tableWidget->item(row,column)->text().toStdString()));
 189     tableWidget->resizeColumnsToContents();
 190   }
 191 }
 192
 193 #endif /* end #define _vvLandmarksPanel_CXX */
 194