1 /*=========================================================================
2 Program: vv http://www.creatis.insa-lyon.fr/rio/vv
5 - University of LYON http://www.universite-lyon.fr/
6 - Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr
7 - CREATIS CNRS laboratory http://www.creatis.insa-lyon.fr
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.
13 It is distributed under dual licence
15 - BSD See included LICENSE.txt file
16 - CeCILL-B http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
17 ======================================================================-====*/
19 #include "vvImageContour.h"
21 #include <vtkImageActor.h>
22 #include <vtkCamera.h>
23 #include <vtkRenderer.h>
24 #include <vtkMarchingSquares.h>
25 #include <vtkImageClip.h>
26 #include <vtkImageData.h>
27 #include <vtkPolyDataMapper.h>
28 #include <vtkProperty.h>
30 //------------------------------------------------------------------------------
31 vvImageContour::vvImageContour()
35 mHiddenImageIsUsed = false;
36 mDisplayModeIsPreserveMemory = true;
37 SetPreserveMemoryModeEnabled(true);
39 //------------------------------------------------------------------------------
42 //------------------------------------------------------------------------------
43 vvImageContour::~vvImageContour()
45 for (unsigned int i = 0; i < mSlicer->GetImage()->GetVTKImages().size(); i++) {
46 mSlicer->GetRenderer()->RemoveActor(mSquaresActorList[i]);
48 mSquaresActorList.clear();
52 //------------------------------------------------------------------------------
55 //------------------------------------------------------------------------------
56 void vvImageContour::SetSlicer(vvSlicer * slicer) {
58 // Create an actor for each time slice
59 for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
60 vtkImageClip * mClipper;// = vtkImageClip::New();
61 vtkMarchingSquares * mSquares;// = vtkMarchingSquares::New();
62 vtkActor * mSquaresActor;// = vtkActor::New();
63 CreateNewActor(&mSquaresActor, &mSquares, &mClipper, numImage);
64 mSquaresActorList.push_back(mSquaresActor);
65 mSquaresList.push_back(mSquares);
66 mClipperList.push_back(mClipper);
69 //------------------------------------------------------------------------------
72 //------------------------------------------------------------------------------
73 void vvImageContour::SetImage(vvImage * image) {
74 for (unsigned int numImage = 0; numImage < image->GetVTKImages().size(); numImage++) {
75 mClipperList[numImage]->SetInput(image->GetVTKImages()[numImage]);
77 mHiddenImageIsUsed = true;
80 //------------------------------------------------------------------------------
83 //------------------------------------------------------------------------------
84 void vvImageContour::SetPreserveMemoryModeEnabled(bool b) {
85 // FastCache mode work only if threshold is always the same
86 if (mDisplayModeIsPreserveMemory == b) return;
87 mDisplayModeIsPreserveMemory = b;
89 clitkExceptionMacro("TODO : not implemented yet");
91 InitializeCacheMode();
94 for(unsigned int d=0; d<mListOfCachedContourActors.size(); d++)
95 mListOfCachedContourActors[d].clear();
96 mListOfCachedContourActors.clear();
100 //------------------------------------------------------------------------------
103 //------------------------------------------------------------------------------
104 void vvImageContour::SetColor(double r, double g, double b) {
105 for(unsigned int i=0; i<mSquaresActorList.size(); i++) {
106 mSquaresActorList[i]->GetProperty()->SetColor(r,g,b);
109 //------------------------------------------------------------------------------
112 //------------------------------------------------------------------------------
113 void vvImageContour::SetLineWidth(double w)
115 for(unsigned int i=0; i<mSquaresActorList.size(); i++) {
116 mSquaresActorList[i]->GetProperty()->SetLineWidth(w);
119 //------------------------------------------------------------------------------
122 //------------------------------------------------------------------------------
123 void vvImageContour::HideActors() {
124 if (!mSlicer) return;
125 mSlice = mSlicer->GetSlice();
126 for(unsigned int i=0; i<mSquaresActorList.size(); i++) {
127 mSquaresActorList[i]->VisibilityOff();
130 //------------------------------------------------------------------------------
133 //------------------------------------------------------------------------------
134 void vvImageContour::ShowActors() {
135 if (!mSlicer) return;
136 mSlice = mSlicer->GetSlice();
137 mTSlice = mSlicer->GetTSlice();
138 mSquaresActorList[mTSlice]->VisibilityOn();
141 //------------------------------------------------------------------------------
144 //------------------------------------------------------------------------------
145 void vvImageContour::Update(double value) {
146 if (!mSlicer) return;
147 if (mPreviousValue == value) {
148 if (mPreviousSlice == mSlicer->GetSlice()) {
149 if (mPreviousTSlice == mSlicer->GetTSlice()) {
150 return; // Nothing to do
155 // Get current threshold value
159 mSlice = mSlicer->GetSlice();
161 if (mDisplayModeIsPreserveMemory) {
162 UpdateWithPreserveMemoryMode();
165 UpdateWithFastCacheMode();
168 // mSlicer->Render(); //DS ---> REMOVE ??
170 mPreviousTSlice = mSlicer->GetTSlice();
171 mPreviousSlice = mSlicer->GetSlice();
172 mPreviousValue = value;
174 //------------------------------------------------------------------------------
177 //------------------------------------------------------------------------------
178 void vvImageContour::UpdateWithPreserveMemoryMode() {
179 // Only change actor visibility if tslice change
180 int mPreviousTslice = mTSlice;
181 mTSlice = mSlicer->GetTSlice();
183 vtkMarchingSquares * mSquares = mSquaresList[mTSlice];
184 vtkImageClip * mClipper = mClipperList[mTSlice];
185 vtkActor * mSquaresActor = mSquaresActorList[mTSlice];
186 int orientation = ComputeCurrentOrientation();
188 UpdateActor(mSquaresActor, mSquares, mClipper, mValue, orientation, mSlice);
189 mSquaresActorList[mTSlice]->VisibilityOn();
191 if (mPreviousTslice != mTSlice) {
192 if (mPreviousTslice != -1) mSquaresActorList[mPreviousTslice]->VisibilityOff();
195 //------------------------------------------------------------------------------
198 //------------------------------------------------------------------------------
199 void vvImageContour::InitializeCacheMode() {
200 clitkExceptionMacro("TODO : not implemented yet");
201 mPreviousSlice = mPreviousOrientation = 0;
202 int dim = mSlicer->GetImage()->GetNumberOfDimensions();
204 mListOfCachedContourActors.resize(dim);
205 for(int d=0; d<dim; d++) {
206 int size = mSlicer->GetImage()->GetSize()[d];
207 mListOfCachedContourActors[d].resize(size);
208 for(int j=0; j<size; j++) {
209 mListOfCachedContourActors[d][j] = NULL;
213 //------------------------------------------------------------------------------
216 //------------------------------------------------------------------------------
217 int vvImageContour::ComputeCurrentOrientation() {
218 // Get extent of image in the slicer
219 int* extent = mSlicer->GetImageActor()->GetDisplayExtent();
221 // Compute orientation
223 for (orientation = 0; orientation < 6; orientation = orientation+2) {
224 if (extent[orientation] == extent[orientation+1]) {
228 orientation = orientation/2;
231 //------------------------------------------------------------------------------
234 //------------------------------------------------------------------------------
235 void vvImageContour::UpdateWithFastCacheMode() {
236 clitkExceptionMacro("TODO : not implemented yet");
238 // Compute orientation
239 int orientation = ComputeCurrentOrientation();
241 if ((mPreviousSlice == mSlice) && (mPreviousOrientation == orientation)) return;
243 vtkActor * actor = mListOfCachedContourActors[orientation][mSlice];
245 mListOfCachedContourActors[orientation][mSlice]->VisibilityOn();
247 vtkImageClip * mClipper;
248 vtkMarchingSquares * mSquares;
249 vtkActor * mSquaresActor;
250 CreateNewActor(&mSquaresActor, &mSquares, &mClipper, 0);
251 UpdateActor(mSquaresActor, mSquares, mClipper, mValue, orientation, mSlice);
252 mListOfCachedContourActors[orientation][mSlice] = mSquaresActor;
253 mSquaresActor->VisibilityOn();
256 if (mListOfCachedContourActors[mPreviousOrientation][mPreviousSlice] != NULL)
257 mListOfCachedContourActors[mPreviousOrientation][mPreviousSlice]->VisibilityOff();
258 mPreviousSlice = mSlice;
259 mPreviousOrientation = orientation;
261 //------------------------------------------------------------------------------
264 //------------------------------------------------------------------------------
265 void vvImageContour::CreateNewActor(vtkActor ** actor,
266 vtkMarchingSquares ** squares,
267 vtkImageClip ** clipper,
269 vtkSmartPointer<vtkActor> mSquaresActor = (*actor = vtkSmartPointer<vtkActor>::New());
270 vtkSmartPointer<vtkImageClip> mClipper = (*clipper = vtkSmartPointer<vtkImageClip>::New());
271 vtkSmartPointer<vtkMarchingSquares> mSquares = (*squares = vtkSmartPointer<vtkMarchingSquares>::New());
272 vtkSmartPointer<vtkPolyDataMapper> mSquaresMapper = vtkSmartPointer<vtkPolyDataMapper>::New();
274 if (mHiddenImageIsUsed)
275 mClipper->SetInput(mHiddenImage->GetVTKImages()[0]);
277 mClipper->SetInput(mSlicer->GetImage()->GetVTKImages()[numImage]);
278 mSquares->SetInput(mClipper->GetOutput());
279 mSquaresMapper->SetInput(mSquares->GetOutput());
280 mSquaresMapper->ScalarVisibilityOff();
281 mSquaresActor->SetMapper(mSquaresMapper);
282 mSquaresActor->GetProperty()->SetColor(1.0,0,0);
283 mSquaresActor->SetPickable(0);
284 mSquaresActor->VisibilityOff();
285 mSlicer->GetRenderer()->AddActor(mSquaresActor);
287 //------------------------------------------------------------------------------
290 //------------------------------------------------------------------------------
291 void vvImageContour::UpdateActor(vtkActor * actor,
292 vtkMarchingSquares * squares,
293 vtkImageClip * clipper,
294 double threshold, int orientation, int slice) {
295 // Set parameter for the MarchigSquare
296 squares->SetValue(0, threshold);
299 int* extent = mSlicer->GetImageActor()->GetDisplayExtent();
301 // Change extent if needed
303 if (mHiddenImageIsUsed) {
304 extent2 = new int[6];
306 extent3 = mHiddenImage->GetFirstVTKImageData()->GetExtent();
308 for(int i=0; i<6; i++) extent2[i] = extent3[i];
310 double s = (double)extent[orientation*2]*(double)mSlicer->GetImage()->GetSpacing()[orientation]; // in mm
311 s = s+mSlicer->GetImage()->GetOrigin()[orientation]; // from origin
312 s = s-mHiddenImage->GetFirstVTKImageData()->GetOrigin()[orientation]; // from corner second image
313 s = s/mHiddenImage->GetFirstVTKImageData()->GetSpacing()[orientation]; // in voxel
316 extent2[orientation*2] = extent2[orientation*2+1] = (int)floor(s);
318 extent2[orientation*2] = (int)floor(s);
319 extent2[orientation*2+1] = extent2[orientation*2];
322 // Do not display a contour if there is no contour on this slice
323 if (extent2[orientation*2+1] > extent3[orientation*2+1]) {
324 actor->VisibilityOff();
327 else actor->VisibilityOn();
333 clipper->SetOutputWholeExtent(extent2[0],extent2[1],extent2[2],
334 extent2[3],extent2[4],extent2[5]);
336 if (mHiddenImage) delete extent2;
338 // Move the actor to be visible
339 switch (orientation) {
341 actor->SetPosition(-1,0,0);
343 // DD(mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[0]);
344 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[0] > slice) {
345 actor->SetPosition(1,0,0);
347 actor->SetPosition(-1,0,0);
351 actor->SetPosition(0,-1,0);
353 // DD(mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[1]);
354 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[1] > slice) {
355 actor->SetPosition(0,1,0);
357 actor->SetPosition(0,-1,0);
362 actor->SetPosition(0,0,-1);
364 DD(mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[2]);
365 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[2] > slice) {
367 actor->SetPosition(0,0,1);
370 actor->SetPosition(0,0,-1);
378 //------------------------------------------------------------------------------