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         ///Clears the extent
  33     void Clear() { mExtent[0] = mExtent[1] = mExtent[2] = mExtent[3] = 1; }
  34
  35         ///Returns true if the two extents are compatible
  36     bool IsCompatible( const ImageExtent& );
  37
  38         ///Adds the extent passed as a parameter to the current extent
  39     void Add ( const ImageExtent& );
  40
  41         ///Returns the ieth position of the extent
  42     int Get(int i) { return mExtent[i]; }
  43
  44         ///Returns the dimension of the current image
  45     void SetDimension(int dim) { mDim=dim; }
  46
  47         ///Returns the dimension of the current image
  48     int GetDimension() { return mDim; }
  49
  50   private:
  51     int mExtent[4];
  52     int mDim;
  53   };
  54
  55   //======================================================================
  56
  57   //======================================================================
  58   // CTor
  59   GimmickView::GimmickView(boost::shared_ptr<Gimmick> gimmick, int threads)
  60     : mGimmick(gimmick),
  61           mReader(threads)
  62   {
  63     GimmickDebugMessage(1,"GimmickView::GimmickView"
  64                         <<std::endl);
  65         // Anciently started the threads ...
  66     // Threads now automatically start at first image request
  67     //mReader.Start();
  68   }
  69   //======================================================================
  70
  71   //======================================================================
  72   /// Destructor
  73   GimmickView::~GimmickView()
  74   {
  75     GimmickDebugMessage(1,"GimmickView::~GimmickView"
  76                         <<std::endl);
  77   }
  78    //======================================================================
  79
  80   //======================================================================
  81   /// Initializes the view :
  82   /// Creates the TreeViews for all the TreeHandler of the Controller
  83   ///
  84   void GimmickView::Initialize()
  85   {
  86         mReaderStarted=false;
  87   }
  88   //======================================================================
  89
  90   //======================================================================
  91   /// Finalize
  92   void GimmickView::Finalize()
  93   {
  94   }
  95
  96   //======================================================================
  97
  98   //======================================================================
  99   /// Create the tree views
 100   void GimmickView::CreateTreeViews()
 101   {
 102     GimmickMessage(2,"Creating the tree views"<<std::endl);
 103     Gimmick::TreeHandlerMapType::const_iterator i;
 104     for (i = mGimmick->GetTreeHandlerMap().begin();
 105          i!= mGimmick->GetTreeHandlerMap().end();
 106          ++i)
 107       {
 108         this->CreateTreeView(i->second);
 109       }
 110   }
 111
 112   /// Create a tree view with a given name
 113   void GimmickView::CreateSingleTreeView(std::string &i_name)
 114   {
 115         this->CreateTreeView(mGimmick->GetTreeHandlerMap()[i_name]);
 116
 117   }
 118
 119   //======================================================================
 120
 121   //======================================================================
 122   /// Updates the TreeView of given name from level l to bottom
 123   /// (calls the virtual method TreeView::Update())
 124   void GimmickView::UpdateTreeViewLevel(const std::string& t, int l)
 125   {
 126     TreeViewMapType::iterator i;
 127     i = GetTreeViewMap().find(t);
 128     if ( i == GetTreeViewMap().end() )
 129       {
 130         GimmickError("INTERNAL ERROR : GimmickView::UpdateTreeView : '"
 131                      <<t<<"' is not in TreeViewMap");
 132       }
 133     i->second->UpdateLevel(l);
 134   }
 135
 136   //======================================================================
 137   /// Clears the status and begins a new selection process
 138   void GimmickView::ResetExtent()
 139   {
 140           if(mImageExtent!=0)
 141           {
 142                   mImageExtent.reset();
 143           }
 144           valid=true;
 145   }
 146
 147   //======================================================================
 148
 149   //======================================================================
 150   bool ImageExtent::IsCompatible(const ImageExtent& ie)
 151   {
 152           bool compatible=true;
 153           ImageExtent * extent= (ImageExtent*)&ie;
 154           if((*extent).Get(0)!=Get(0)
 155                  || (*extent).Get(1)!=Get(1))
 156           {
 157                   compatible=false;
 158           }
 159           return compatible;
 160   }
 161
 162   //======================================================================
 163
 164   //======================================================================
 165   void ImageExtent::Add(const ImageExtent& ie)
 166   {
 167           ImageExtent * extent= (ImageExtent*)&ie;
 168           mExtent[2]+=(*extent).Get(2);
 169           if(mExtent[2]>1)
 170           {
 171           SetDimension(3);
 172           }
 173   }
 174
 175   //======================================================================
 176   /// No selected image
 177   bool GimmickView::NoValidateSelected ()
 178   {
 179         GimmickDebugMessage(2,"Validating selected"<<std::endl);
 180         std::string mMessage;
 181         mMessage="Cannot have 0 images selected!";
 182         valid=false;
 183         modifyValidationSignal(valid);
 184         SetMessage(mMessage);
 185         return valid;
 186   }
 187
 188   //======================================================================
 189   ///Validates the dimension compliance of the images with the maximum and
 190   ///minimum given, and between their sizes
 191   bool GimmickView::ValidateSelected (tree::Node* sel, int min_dim, int max_dim)
 192   {
 193         GimmickDebugMessage(2,"Validating selected"<<std::endl);
 194         std::string mMessage;
 195
 196         if(sel==0)
 197         {
 198                 mMessage="Cannot have 0 images selected!";
 199                 valid=false;
 200         }
 201         else
 202         {
 203                 boost::shared_ptr<ImageExtent> ie=boost::shared_ptr<ImageExtent>(new ImageExtent((*sel).GetAttribute("D0028_0010"),
 204                                                                         (*sel).GetAttribute("D0028_0011"),
 205                                                                         (*sel).GetAttribute("D0028_0012"),
 206                                                                         ""));
 207         if(mImageExtent==0)
 208         {
 209                 mImageExtent=ie;
 210                 if((mImageExtent->Get(min_dim-1)<2)||(mImageExtent->Get(max_dim)>1))
 211                 {
 212                         valid=false;
 213                 }
 214                 else
 215                 {
 216                         std::stringstream out;
 217                         out << mImageExtent->GetDimension() << "D image " << mImageExtent->Get(0) << "x"<< mImageExtent->Get(1) << "x"<< mImageExtent->Get(2) <<" selected";
 218                         mMessage = out.str();
 219                         mImageExtent->SetDimension(2);
 220                         valid=true;
 221                 }
 222         }
 223         else
 224         {
 225                 if(mImageExtent->IsCompatible(*ie))
 226                 {
 227                         if(mImageExtent->GetDimension()==max_dim && mImageExtent->Get(max_dim)>2)
 228                         {
 229                                 std::stringstream out;
 230                                 out<<"Cannot add this image to selection : would result in a "<<mImageExtent->GetDimension()+1<<"D image!";
 231                                 mMessage=out.str();
 232                                 valid=false;
 233                         }
 234                         else if(max_dim<3)
 235                         {
 236                                 std::stringstream out;
 237                                 out<<"Selecting "<<mImageExtent->GetDimension()<<"D images is not allowed !";
 238                                 mMessage=out.str();
 239                                 valid=false;
 240                         }
 241                         else if(min_dim==3 && (ie->Get(2)+mImageExtent->Get(2))<2)
 242                         {
 243                                 std::stringstream out;
 244                                 out << "Cannot build the selection as it would result in a ";
 245                                 out << mImageExtent->GetDimension();
 246                                 out << "D image, and the minimum is ";
 247                                 out << min_dim;
 248                                 out << "D!";
 249                                 mMessage=out.str();
 250                                 valid=false;
 251                         }
 252                         else
 253                         {
 254                                 mImageExtent->Add(*ie);
 255                                 std::stringstream out;
 256                                 out << mImageExtent->GetDimension() << "D image " << mImageExtent->Get(0) << "x"<< mImageExtent->Get(1) << "x"<< mImageExtent->Get(2) <<" selected";
 257                                 mMessage = out.str();
 258                         }
 259                 }
 260                 else
 261                 {
 262                         mMessage="The selected images are not compatible.";
 263                         valid=false;
 264                 }
 265           }
 266         }
 267
 268         modifyValidationSignal(valid);
 269         SetMessage(mMessage);
 270         return valid;
 271   }
 272
 273   //======================================================================
 274   void GimmickView::modifyValidationSignal(bool ivalid)
 275   {
 276           mValidationSignal(ivalid);
 277   }
 278
 279    //======================================================================
 280   ///Reads Images (Non Threaded)
 281 void GimmickView::ReadImagesNotThreaded(std::vector<vtkImageData*>& s, std::vector<std::string> im, int dimension)
 282 {
 283
 284 /* remember!
 285
 286 #define GIMMICK_NO_IMAGE_SELECTION 0
 287 #define GIMMICK_2D_IMAGE_SELECTION 2
 288 #define GIMMICK_3D_IMAGE_SELECTION 3
 289 #define GIMMICK_4D_IMAGE_SELECTION 4
 290
 291 #define NATIVE 0
 292 #define _2D    2
 293 #define _3D    3
 294
 295 */
 296         // Create the output data
 297         if (im.size()==1)
 298         {
 299                 // Only one image : give it
 300                 vtkImageData* out = vtkImageData::New();
 301                 GimmickDebugMessage(3, "State Check: Full Filename: "
 302                                         <<im.front()
 303                                         <<std::endl);
 304                 out->ShallowCopy(mReader.GetImage(im.front()));
 305                 s.push_back( out );
 306         }
 307         else if (im.size()>1) // Test inutile ? JPR
 308         {
 309                 // Read the FIRST image to get info ... about all the other ones. ?!? // JPR
 310                 vtkImageData* first = mReader.GetImage( im.front());
 311
 312                 if (dimension == 2)
 313                 {
 314                  // n3D
 315                     std::vector<std::string>::iterator it;
 316                         for (it=im.begin(); it!=im.end(); ++it)
 317                         {
 318                                 vtkImageData* out = vtkImageData::New();
 319                                 out->ShallowCopy(mReader.GetImage(*it));
 320                                 s.push_back(out);
 321                         }
 322                 }
 323                 else if (dimension == 3)
 324                 {
 325                         // n*2D to 3D  :: --> NON ! Ca va tres bien pour du 2D+T (le ZSpacing, on s'en tape)
 326                         //                      // ne convient pas pour du n*2D a  lire comme un 3D
 327                         //                      // Il faut un gdcmSerieHelper !
 328                         vtkImageData* out = vtkImageData::New();
 329                         out->CopyStructure(first);
 330                         out->SetScalarType(first->GetScalarType());
 331                         int ext[6];
 332                         //first->GetExtent(ext);     // JPR
 333                         first->GetWholeExtent(ext);  // renvoie egalement 0,0 en Z // JPR
 334                                                      // Normal :
 335
 336 std::cout <<"in GimmickView::ReadImagesNotThreaded GetWholeExtent ext =";
 337 for (int jjj=0;jjj<6;jjj++)
 338         std:cout << "   [" << jjj << "]=" << ext[jjj];
 339 std::cout << std::endl;
 340
 341                         if(ext[5] == 0)
 342                         {
 343                                 ext[5] = im.size()-1;
 344                         }
 345                         else
 346                         {
 347                                 ext[5] = ext[5] * im.size()-1; // to deal with multiframes - JPR
 348                         }
 349                         out->SetExtent(ext);
 350
 351                         // LG : TODO : Z Spacing  ?  // Voir + loin // JPR
 352
 353                         out->AllocateScalars();
 354
 355                         int dim[3];
 356                         first->GetDimensions(dim);
 357                         unsigned long imsize = dim[0] * dim[1];
 358                         imsize = imsize * dim[2] ;  // deal with multiframes // JPR
 359 //EED 03-11-2009
 360                         imsize = imsize * first->GetScalarSize();
 361
 362                         int slice = 0;
 363                         std::vector<std::string>::iterator it;
 364                         for (it=im.begin(); it!=im.end(); ++it)
 365                         {
 366                                 vtkImageData* cur = mReader.GetImage( (*it));
 367                         //      void* src = cur->GetScalarPointer(0,0,0);
 368                         //      void* dst = out->GetScalarPointer(0,0,slice);
 369                                 memcpy(out->GetScalarPointer(0,0,slice), cur->GetScalarPointer(0,0,0), imsize);
 370
 371                                 slice++;
 372                         }
 373                         s.push_back(out);
 374                 }
 375         }
 376 }
 377   //======================================================================
 378
 379 void GimmickView::ReadImagesNotThreadedInVector(std::vector<vtkImageData*>& s, std::vector<std::string> im, int dimension)
 380 {
 381         // Create the output data
 382         if (im.size()==1)
 383         {
 384                 // Only one image : give it
 385                 vtkImageData* out = vtkImageData::New();
 386                 GimmickDebugMessage(3, "State Check: Full Filename: "
 387                                                 <<im.front()
 388                                                 <<std::endl);
 389                 out->ShallowCopy(mReader.GetImage(im.front()));
 390                 s.push_back( out );
 391         }
 392         else if (im.size()>1) // Test inutile ? JPR
 393         {
 394                 vtkImageData* first = mReader.GetImage( im.front());
 395                 if (dimension == 2)
 396                 {
 397                  // n3D
 398                     std::vector<std::string>::iterator it;
 399                         for (it=im.begin(); it!=im.end(); ++it)
 400                         {
 401                         std::cout << "----------------------------------------n3D" << std::endl;
 402                                 vtkImageData* out = vtkImageData::New();
 403                                 out->ShallowCopy(mReader.GetImage(*it));
 404                                 s.push_back(out);
 405                         }
 406                 }
 407                 else
 408                 {
 409                         // n2D to 3D // NO!
 410                         // n *2D + T in a vector :
 411                         std::cout << "---------------------------------------- n *2D + T" << std::endl;
 412                         std::vector<std::string>::iterator it;
 413                         for (it=im.begin(); it!=im.end(); ++it)
 414                         {
 415                         std::cout << "---------------------------------------- [" << *it << "]" << std::endl;
 416                                 //vtkImageData* out = mReader.GetImage( (*it));
 417                                 //s.push_back(out);
 418                                 vtkImageData* out = vtkImageData::New();
 419                                 out->ShallowCopy(mReader.GetImage(*it));
 420                                 s.push_back(out);
 421                         }
 422                 }
 423         }
 424 }
 425   //======================================================================
 426
 427   //======================================================================
 428   ///Requests the reading of an image
 429   void GimmickView::RequestReading(tree::Node* n,
 430           int prio, int selection_index, boost::shared_ptr<ImagePointerHolder> p)
 431   {
 432      if(!mReaderStarted)
 433      {
 434         mReader.Start();
 435         mReaderStarted=true;
 436       }
 437       ImageEventType t(n,selection_index);
 438       t.pointerHolder = p;
 439       mImageEventMap[n->GetAttribute("FullFileName")] = t;
 440       mReader.Request(this,n->GetAttribute("FullFileName"),prio);
 441   }
 442   //======================================================================
 443
 444   //======================================================================
 445   void GimmickView::
 446   OnMultiThreadImageReaderEvent(const std::string& filename,
 447                                 MultiThreadImageReaderUser::EventType e,
 448                                 vtkImageData* image)
 449   {
 450     GimmickDebugMessage(7,
 451                         "MultiThreadImageReader event : "<<e<<std::endl);
 452         if (e==ImageLoaded)
 453         {
 454                 if (filename.size()==0)
 455                 {
 456                   //What to do in this case?
 457                   /*
 458                         GimmickDebugMessage(5,
 459                                         "Pushing unknown image in queue"
 460                                         <<std::endl);
 461                         mImageEventQueue.push_back(ImageEventType(image));*/
 462                         return;
 463                 }
 464                 ImageEventTypeMap::iterator i;
 465 //JCP 22-06-2009, test mImageEventMap.size() > 0
 466                 if(mImageEventMap.size()>0){
 467                         i = mImageEventMap.find(filename);
 468                         if (i!=mImageEventMap.end())
 469                         {
 470                                 GimmickDebugMessage(5,
 471                                                 "Putting image of file '"<<filename<<"' on pointer"
 472                                                 <<std::endl);
 473                                 ImageEventType ie(i->second);
 474                                 ie.image = image;
 475                                 ie.pointerHolder->Set(ie.image);
 476                                 //mImageEventMap.erase(i);
 477                         }
 478                 }
 479         }
 480         else if (e==Error)
 481         {
 482                 std::string mess="ERROR: MultiThreadImageReader: Cannot read image in file ";
 483                 mess+=filename;
 484                 mess+="\n";
 485                 GimmickMessage(1,mess);
 486                 ImageEventTypeMap::iterator i;
 487                 i = mImageEventMap.find(filename);
 488                 if (i!=mImageEventMap.end())
 489                 {
 490                 ImageEventType ie(i->second);
 491                 ie.image = image;
 492                 ie.pointerHolder->Set(GetDefaultImage());
 493                 //mImageEventMap.erase(i);
 494                 }
 495         }
 496
 497         else if (e==ImageUnloaded)
 498         {
 499                 std::string mess="Unloaded image in file ";
 500                 mess+=filename;
 501                 mess+="\n";
 502                 GimmickMessage(1,mess);
 503                                 ImageEventTypeMap::iterator i;
 504                 i = mImageEventMap.find(filename);
 505                 if (i!=mImageEventMap.end())
 506                 {
 507                 ImageEventType ie(i->second);
 508                 ie.image = image;
 509                 ie.pointerHolder->Set(GetDefaultImage());
 510                 //mImageEventMap.erase(i);
 511                 }
 512         }
 513   }
 514
 515   //====================================================================
 516
 517   //====================================================================
 518   void GimmickView::ConnectValidationObserver(ValidationCallbackType callback)
 519   {
 520     mValidationSignal.connect(callback);
 521   }
 522
 523 } // EO namespace creaImageIO
 524