src2/creaImageIOGimmickView.cpp

   1 #include <creaImageIOGimmickView.h>
   2 #include <creaImageIOSystem.h>
   3 #include "boost/filesystem.hpp"
   4
   5 namespace fs = boost::filesystem;
   6
   7 namespace creaImageIO
   8 {
   9
  10         ///Class used to represent the actual state of the image selected and to perform comparisons on its values
  11   class ImageExtent
  12   {
  13   public:
  14     ImageExtent(const std::string& x, const std::string& y, const std::string& z, const std::string& t)
  15         {
  16                  sscanf(x.c_str(),"%d",&mExtent[0]);
  17                  sscanf(y.c_str(),"%d",&mExtent[1]);
  18                  sscanf(z.c_str(),"%d",&mExtent[2]);
  19                  sscanf(t.c_str(),"%d",&mExtent[3]);
  20                  if(x==""){mExtent[0]=1;}
  21              if(y==""){mExtent[1]=1;}
  22              if(z==""){mExtent[2]=1;}
  23                  if(t==""){mExtent[3]=1;}
  24
  25                  if (mExtent[3]>1) mDim=4;
  26                  else if (mExtent[2]>1) mDim=3;
  27              else if (mExtent[1]>1) mDim=2;
  28              else if (mExtent[0]>1) mDim=1;
  29                  else mDim=0;
  30         }
  31
  32
  33         ///Clears the extent
  34     void Clear() { mExtent[0] = mExtent[1] = mExtent[2] = mExtent[3] = 1; }
  35
  36         ///Returns true if the two extents are compatible
  37     bool IsCompatible( const ImageExtent& );
  38
  39         ///Adds the extent passed as a parameter to the current extent
  40     void Add ( const ImageExtent& );
  41
  42         ///Returns the ieth position of the extent
  43     int Get(int i) { return mExtent[i]; }
  44
  45         ///Returns the dimension of the current image
  46     void SetDimension(int dim) { mDim=dim; }
  47
  48         ///Returns the dimension of the current image
  49     int GetDimension() { return mDim; }
  50
  51   private:
  52     int mExtent[4];
  53     int mDim;
  54   };
  55
  56   //======================================================================
  57
  58   //======================================================================
  59   // CTor
  60   GimmickView::GimmickView(Gimmick* gimmick, int threads)
  61     : mGimmick(gimmick),
  62           mReader(threads)
  63   {
  64     GimmickDebugMessage(1,"GimmickView::GimmickView"
  65                         <<std::endl);
  66         // Anciently started the threads ...
  67     // Threads now automatically start at first image request
  68     //mReader.Start();
  69
  70   }
  71   //======================================================================
  72
  73   //======================================================================
  74   /// Destructor
  75   GimmickView::~GimmickView()
  76   {
  77     GimmickDebugMessage(1,"GimmickView::~GimmickView"
  78                         <<std::endl);
  79   }
  80    //======================================================================
  81
  82   //======================================================================
  83   /// Initializes the view :
  84   /// Creates the TreeViews for all the TreeHandler of the Controller
  85   ///
  86   void GimmickView::Initialize()
  87   {
  88         mImageExtent=0;
  89         mReaderStarted=false;
  90   }
  91   //======================================================================
  92
  93   //======================================================================
  94   /// Finalize
  95   void GimmickView::Finalize()
  96   {
  97   }
  98
  99   //======================================================================
 100
 101   //======================================================================
 102   /// Create the tree views
 103   void GimmickView::CreateTreeViews()
 104   {
 105     GimmickMessage(2,"Creating the tree views"<<std::endl);
 106     Gimmick::TreeHandlerMapType::const_iterator i;
 107     for (i = mGimmick->GetTreeHandlerMap().begin();
 108          i!= mGimmick->GetTreeHandlerMap().end();
 109          ++i)
 110       {
 111         this->CreateTreeView(i->second);
 112       }
 113   }
 114
 115   /// Create a tree view with a given name
 116   void GimmickView::CreateSingleTreeView(std::string &i_name)
 117   {
 118                 this->CreateTreeView(mGimmick->GetTreeHandlerMap()[i_name]);
 119
 120   }
 121
 122
 123   //======================================================================
 124
 125   //======================================================================
 126   /// Updates the TreeView of given name from level l to bottom
 127   /// (calls the virtual method TreeView::Update())
 128   void GimmickView::UpdateTreeViewLevel(const std::string& t, int l)
 129   {
 130     TreeViewMapType::iterator i;
 131     i = GetTreeViewMap().find(t);
 132     if ( i == GetTreeViewMap().end() )
 133       {
 134         GimmickError("INTERNAL ERROR : GimmickView::UpdateTreeView : '"
 135                      <<t<<"' is not in TreeViewMap");
 136       }
 137     i->second->UpdateLevel(l);
 138   }
 139   //======================================================================
 140   /// Clears the status and begins a new selection process
 141   void GimmickView::ResetExtent()
 142   {
 143           if(mImageExtent!=0)
 144           {
 145           mImageExtent=0;
 146           }
 147           valid=true;
 148   }
 149
 150
 151   //======================================================================
 152
 153   //======================================================================
 154   bool ImageExtent::IsCompatible(const ImageExtent& ie)
 155   {
 156           bool compatible=true;
 157           ImageExtent * extent= (ImageExtent*)&ie;
 158           if((*extent).Get(0)!=Get(0)
 159                  || (*extent).Get(1)!=Get(1))
 160           {
 161                   compatible=false;
 162           }
 163           return compatible;
 164   }
 165
 166   //======================================================================
 167
 168   //======================================================================
 169   void ImageExtent::Add(const ImageExtent& ie)
 170   {
 171           ImageExtent * extent= (ImageExtent*)&ie;
 172           mExtent[2]+=(*extent).Get(2);
 173           if(mExtent[2]>1)
 174           {
 175           SetDimension(3);
 176           }
 177   }
 178
 179   //======================================================================
 180   ///Validates the dimension compliance of the images with the maximum and
 181   ///minimum given, and between their sizes
 182   bool GimmickView::ValidateSelected (tree::Node* sel, int min_dim, int max_dim)
 183   {
 184         GimmickDebugMessage(2,"Validating selected"<<std::endl);
 185         std::string mMessage;
 186
 187         if(sel==0)
 188         {
 189                 mMessage="Cannot have 0 images selected!";
 190                 valid=false;
 191         }
 192         else
 193         {
 194         ImageExtent* ie=new ImageExtent((*sel).GetAttribute("D0028_0010"),
 195                                                  (*sel).GetAttribute("D0028_0011"),
 196                                                                          (*sel).GetAttribute("D0028_0012"),
 197                                                                          "");
 198         if(mImageExtent==0)
 199         {
 200                 mImageExtent=ie;
 201                 if((mImageExtent->Get(min_dim-1)<2)||(mImageExtent->Get(max_dim)>1))
 202                 {
 203                         valid=false;
 204                 }
 205                 else
 206                 {
 207                         std::stringstream out;
 208                         out << mImageExtent->GetDimension() << "D image " << mImageExtent->Get(0) << "x"<< mImageExtent->Get(1) << "x"<< mImageExtent->Get(2) <<" selected";
 209                 mMessage = out.str();
 210                         mImageExtent->SetDimension(2);
 211                         valid=true;
 212                 }
 213         }
 214         else
 215         {
 216                 if(mImageExtent->IsCompatible(*ie))
 217                 {
 218                         if(mImageExtent->GetDimension()==max_dim && mImageExtent->Get(max_dim)>2)
 219                         {
 220                                 std::stringstream out;
 221                                 out<<"Cannot add this image to selection : would result in a "<<mImageExtent->GetDimension()+1<<"D image!";
 222                                 mMessage=out.str();
 223                                 valid=false;
 224                         }
 225                         else if(max_dim<3)
 226                         {
 227                                 std::stringstream out;
 228                                 out<<"Selecting "<<mImageExtent->GetDimension()<<"D images is not allowed !";
 229                                 mMessage=out.str();
 230                                 valid=false;
 231                         }
 232                         else if(min_dim==3 && (ie->Get(2)+mImageExtent->Get(2))<2)
 233                         {
 234                                 std::stringstream out;
 235                                 out << "Cannot build the selection as it would result in a ";
 236                                 out << mImageExtent->GetDimension();
 237                                 out << "D image, and the minimum is ";
 238                                 out << min_dim;
 239                                 out << "D!";
 240                                 mMessage=out.str();
 241                                 valid=false;
 242                         }
 243                         else
 244                         {
 245             mImageExtent->Add(*ie);
 246                         std::stringstream out;
 247                         out << mImageExtent->GetDimension() << "D image " << mImageExtent->Get(0) << "x"<< mImageExtent->Get(1) << "x"<< mImageExtent->Get(2) <<" selected";
 248                 mMessage = out.str();
 249                         }
 250
 251                 }
 252                 else
 253                 {
 254                         mMessage="The selected images are not compatible.";
 255                         valid=false;
 256                 }
 257         }
 258         }
 259
 260
 261         modifyValidationSignal(valid);
 262         SetMessage(mMessage);
 263         return valid;
 264   }
 265
 266   //======================================================================
 267   void GimmickView::modifyValidationSignal(bool ivalid)
 268   {
 269           mValidationSignal(ivalid);
 270   }
 271
 272
 273    //======================================================================
 274   ///Reads Images (Non Threaded)
 275 void GimmickView::ReadImagesNotThreaded(std::vector<vtkImageData*>& s,std::vector<std::string> im, int dimension)
 276 {
 277
 278         // Create the output data
 279         if (im.size()==1)
 280         {
 281                 // Only one image : give it
 282                 vtkImageData* out = vtkImageData::New();
 283                 GimmickDebugMessage(3, "State Check: Full Filename: "
 284                                                 <<im.front()
 285                                                 <<std::endl);
 286                 out->ShallowCopy(mReader.GetImage(im.front()));
 287                 s.push_back( out );
 288         }
 289     else if (im.size()>1)
 290         {
 291                 vtkImageData* first = mReader.GetImage( im.front());
 292                 if (dimension == 2)
 293                 {
 294                  // n3D
 295                   std::vector<std::string>::iterator it;
 296                         for (it=im.begin(); it!=im.end(); ++it)
 297                         {
 298                                 vtkImageData* out = vtkImageData::New();
 299                                 out->ShallowCopy(mReader.GetImage(*it));
 300                                 s.push_back(out);
 301                         }
 302                 }
 303                 else
 304                 {
 305                         // n2D to 3D
 306                         vtkImageData* out = vtkImageData::New();
 307                         out->CopyStructure(first);
 308                         out->SetScalarType(first->GetScalarType());
 309                         int ext[6];
 310                         first->GetExtent(ext);
 311                         ext[5] = im.size()-1;
 312                         out->SetExtent(ext);
 313                         // LG : TODO : Z Spacing  ?
 314
 315                         out->AllocateScalars();
 316
 317                         //first->Print(std::cout);
 318                         //          out->Print(std::cout);
 319
 320                         int dim[3];
 321                         first->GetDimensions(dim);
 322                         unsigned long imsize = dim[0] * dim[1];/*
 323                                 ( (unsigned long)first->GetScalarPointer(0,1,0)
 324                         - (unsigned long)first->GetScalarPointer(0,0,0))
 325                                 *dim[1];*/
 326
 327                         int slice = 0;
 328                         std::vector<std::string>::iterator it;
 329                         for (it=im.begin(); it!=im.end(); ++it)
 330                                 {
 331                         //std::cout << "copying slice "<<slice <<std::endl;
 332                                         vtkImageData* cur = mReader.GetImage( (*it));
 333
 334                         void* src = cur->GetScalarPointer(0,0,0);
 335                         void* dst = out->GetScalarPointer(0,0,slice);
 336                         //              std::cout << "src="<<src<<std::endl;
 337                         //              std::cout << "dst="<<dst<<std::endl;
 338                         //              std::cout << "siz="<<imsize<<std::endl;
 339                         std::cout << slice;
 340                         memcpy(dst,src,imsize);
 341
 342                         slice++;
 343                                 }
 344                         s.push_back(out);
 345
 346                 }
 347         }
 348 }
 349   //======================================================================
 350
 351   //======================================================================
 352   ///Requests the reading of an image
 353   void GimmickView::RequestReading(tree::Node* n,
 354           int prio, int selection_index, boost::shared_ptr<ImagePointerHolder> p)
 355   {
 356           if(!mReaderStarted)
 357           {
 358                 mReader.Start();
 359                 mReaderStarted=true;
 360           }
 361     ImageEventType t(n,selection_index);
 362         t.pointerHolder = p;
 363     mImageEventMap[n->GetAttribute("FullFileName")] = t;
 364     mReader.Request(this,n->GetAttribute("FullFileName"),prio);
 365   }
 366   //======================================================================
 367
 368   //======================================================================
 369   void GimmickView::
 370   OnMultiThreadImageReaderEvent(const std::string& filename,
 371                                 MultiThreadImageReaderUser::EventType e,
 372                                 vtkImageData* image)
 373   {
 374     GimmickDebugMessage(7,
 375                         "MultiThreadImageReader event : "<<e<<std::endl);
 376         if (e==ImageLoaded)
 377         {
 378                 if (filename.size()==0)
 379                 {
 380                   //What to do in this case?
 381                   /*
 382                 GimmickDebugMessage(5,
 383                                         "Pushing unknown image in queue"
 384                                         <<std::endl);
 385                 mImageEventQueue.push_back(ImageEventType(image));*/
 386                         return;
 387                 }
 388                 ImageEventTypeMap::iterator i;
 389 //JCP 22-06-2009, test mImageEventMap.size() > 0
 390                 if(mImageEventMap.size()>0){
 391                         i = mImageEventMap.find(filename);
 392                         if (i!=mImageEventMap.end())
 393                         {
 394                                 GimmickDebugMessage(5,
 395                                                         "Putting image of file '"<<filename<<"' on pointer"
 396                                                         <<std::endl);
 397                                 ImageEventType ie(i->second);
 398                                 ie.image = image;
 399                                 ie.pointerHolder->Set(ie.image);
 400                                 //mImageEventMap.erase(i);
 401                         }
 402                 }
 403         }
 404         else if (e==Error)
 405         {
 406                 std::string mess="ERROR: MultiThreadImageReader: Cannot read image in file ";
 407                 mess+=filename;
 408                 mess+="\n";
 409                 GimmickMessage(1,mess);
 410                 ImageEventTypeMap::iterator i;
 411                 i = mImageEventMap.find(filename);
 412                 if (i!=mImageEventMap.end())
 413                 {
 414                 ImageEventType ie(i->second);
 415                 ie.image = image;
 416                 ie.pointerHolder->Set(GetDefaultImage());
 417                 //mImageEventMap.erase(i);
 418                 }
 419         }
 420
 421         else if (e==ImageUnloaded)
 422         {
 423                 std::string mess="Unloaded image in file ";
 424                 mess+=filename;
 425                 mess+="\n";
 426                 GimmickMessage(1,mess);
 427                                 ImageEventTypeMap::iterator i;
 428                 i = mImageEventMap.find(filename);
 429                 if (i!=mImageEventMap.end())
 430                 {
 431                 ImageEventType ie(i->second);
 432                 ie.image = image;
 433                 ie.pointerHolder->Set(GetDefaultImage());
 434                 //mImageEventMap.erase(i);
 435                 }
 436
 437         }
 438   }
 439
 440   //====================================================================
 441
 442   //====================================================================
 443   void GimmickView::ConnectValidationObserver(ValidationCallbackType callback)
 444   {
 445     mValidationSignal.connect(callback);
 446   }
 447
 448
 449 } // EO namespace creaImageIO
 450
 451