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