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 "vvBinaryImageOverlayActor.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>
29 #include <vtkImageMapToRGBA.h>
30 #include <vtkLookupTable.h>
32 //------------------------------------------------------------------------------
33 vvBinaryImageOverlayActor::vvBinaryImageOverlayActor()
41 mColorLUT = vtkSmartPointer<vtkLookupTable>::New();
46 //------------------------------------------------------------------------------
49 //------------------------------------------------------------------------------
50 vvBinaryImageOverlayActor::~vvBinaryImageOverlayActor()
52 for (unsigned int i = 0; i < mSlicer->GetImage()->GetVTKImages().size(); i++) {
53 mSlicer->GetRenderer()->RemoveActor(mImageActorList[i]);
56 mImageActorList.clear();
58 //------------------------------------------------------------------------------
61 //------------------------------------------------------------------------------
62 void vvBinaryImageOverlayActor::SetColor(double r, double g, double b)
68 //------------------------------------------------------------------------------
71 //------------------------------------------------------------------------------
72 void vvBinaryImageOverlayActor::SetSlicer(vvSlicer * slicer)
76 //------------------------------------------------------------------------------
79 //------------------------------------------------------------------------------
80 void vvBinaryImageOverlayActor::Initialize(bool IsVisible)
83 std::cerr << "ERROR. Please use setSlicer before setSlicer in vvBinaryImageOverlayActor." << std::endl;
88 std::cerr << "ERROR. Please use setImage before setSlicer in vvBinaryImageOverlayActor." << std::endl;
91 // Create an actor for each time slice
92 for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
93 // how many intensity ?
94 vtkSmartPointer<vtkImageMapToRGBA> mOverlayMapper = vtkSmartPointer<vtkImageMapToRGBA>::New();
95 mOverlayMapper->SetInput(mImage->GetVTKImages()[0]); // DS TODO : to change if it is 4D !!!
98 mImage->GetVTKImages()[0]->GetScalarRange(range);
99 int n = range[1]-range[0]+1;
100 mColorLUT->SetRange(range[0],range[1]);
101 mColorLUT->SetNumberOfTableValues(n);
103 // Mode BG -> all is color except BG
105 for(int i=0; i<n; i++) {
106 mColorLUT->SetTableValue(i, mColor[0], mColor[1], mColor[2], mAlpha);
108 mColorLUT->SetTableValue(mBackgroundValue, 0, 0, 0, 0.0);
111 // Mode FG -> all is BG, except FG which is color
112 for(int i=0; i<n; i++) {
113 mColorLUT->SetTableValue(i, 0, 0, 0, 0.0);
115 mColorLUT->SetTableValue(mForegroundValue, mColor[0], mColor[1], mColor[2], mAlpha);
118 mOverlayMapper->SetLookupTable(mColorLUT);
120 vtkSmartPointer<vtkImageActor> mOverlayActor = vtkSmartPointer<vtkImageActor>::New();
121 mOverlayActor->SetInput(mOverlayMapper->GetOutput());
122 mOverlayActor->SetPickable(0);
123 mOverlayActor->SetVisibility(IsVisible);
124 //mOverlayActor->SetOpacity(1.0);
126 mMapperList.push_back(mOverlayMapper);
127 mImageActorList.push_back(mOverlayActor);
128 mSlicer->GetRenderer()->AddActor(mOverlayActor);
131 //------------------------------------------------------------------------------
134 //------------------------------------------------------------------------------
135 void vvBinaryImageOverlayActor::SetOpacity(double d)
139 //------------------------------------------------------------------------------
142 //------------------------------------------------------------------------------
143 // void vvBinaryImageOverlayActor::UpdateOpacity(double d) {
145 // mColorLUT->SetTableValue(1, mColor[0], mColor[1], mColor[2], mAlpha); // FG
146 // for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
147 // // how many intensity ?
148 // vtkImageMapToRGBA * mOverlayMapper = mMapperList[numImage];
149 // mOverlayMapper->SetLookupTable(mColorLUT);
151 // vtkImageActor * mOverlayActor = mImageActorList[numImage];
152 // mOverlayActor->SetInput(mOverlayMapper->GetOutput());
155 //------------------------------------------------------------------------------
158 //------------------------------------------------------------------------------
159 // void vvBinaryImageOverlayActor::SetColor(double r, double v, double b) {
163 // // mColorLUT->SetTableValue(1, mColor[0], mColor[1], mColor[2], mAlpha); // FG
164 // // for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
165 // // // how many intensity ?
166 // // vtkImageMapToRGBA * mOverlayMapper = mMapperList[numImage];
167 // // mOverlayMapper->SetLookupTable(mColorLUT);
169 // // vtkImageActor * mOverlayActor = mImageActorList[numImage];
170 // // mOverlayActor->SetInput(mOverlayMapper->GetOutput());
173 //------------------------------------------------------------------------------
176 //------------------------------------------------------------------------------
177 void vvBinaryImageOverlayActor::SetImage(vvImage * image, double bg, bool modeBG)
181 mBackgroundValue = bg;
185 mForegroundValue = bg;
189 //------------------------------------------------------------------------------
192 //------------------------------------------------------------------------------
193 void vvBinaryImageOverlayActor::HideActors()
195 if (!mSlicer) return;
196 mSlice = mSlicer->GetSlice();
197 for(unsigned int i=0; i<mImageActorList.size(); i++) {
198 mImageActorList[i]->VisibilityOff();
200 // Caller MUST call Render
201 // mSlicer->Render();
203 //------------------------------------------------------------------------------
206 //------------------------------------------------------------------------------
207 void vvBinaryImageOverlayActor::ShowActors()
209 if (!mSlicer) return;
210 mSlice = mSlicer->GetSlice();
211 mTSlice = mSlicer->GetTSlice();
212 // for(unsigned int i=0; i<mSquaresActorList.size(); i++) {
213 mImageActorList[mTSlice]->VisibilityOn();
214 UpdateSlice(0, mSlice);
216 // Caller MUST call Render
219 //------------------------------------------------------------------------------
222 //------------------------------------------------------------------------------
223 void vvBinaryImageOverlayActor::UpdateColor()
225 mColorLUT->SetTableValue(1, mColor[0], mColor[1], mColor[2], mAlpha); // FG
226 for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
227 // how many intensity ?
228 vtkImageMapToRGBA * mOverlayMapper = mMapperList[numImage];
229 mOverlayMapper->SetLookupTable(mColorLUT);
231 vtkImageActor * mOverlayActor = mImageActorList[numImage];
232 mOverlayActor->SetInput(mOverlayMapper->GetOutput());
235 //------------------------------------------------------------------------------
237 //------------------------------------------------------------------------------
238 void vvBinaryImageOverlayActor::UpdateSlice(int slicer, int slice)
240 if (!mSlicer) return;
242 if (mPreviousSlice == mSlicer->GetSlice()) {
243 if (mPreviousTSlice == mSlicer->GetTSlice()) {
244 //DD("=========== NOTHING");
245 return; // Nothing to do
250 mSlice = mSlicer->GetSlice();
251 mTSlice = mSlicer->GetTSlice();
254 int * imageExtent = mSlicer->GetExtent();
255 int orientation = mSlicer->GetOrientation();
257 ComputeExtent(orientation, mSlice, imageExtent, maskExtent);
258 //ComputeExtent(imageExtent, maskExtent, mSlicer->GetImage()->GetFirstVTKImageData(), mImage->GetFirstVTKImageData());
259 ComputeExtent(maskExtent, maskExtent, mSlicer->GetImage()->GetFirstVTKImageData(), mImage->GetFirstVTKImageData());
260 // std::cout << "maskExtent = " << maskExtent[0] << " " << maskExtent[1] << " " << maskExtent[2] << " "
261 // << maskExtent[3] << " " << maskExtent[4] << " " << maskExtent[5] << std::endl;
262 mSlicer->ClipDisplayedExtent(maskExtent, mMapperList[mTSlice]->GetInput()->GetWholeExtent());
263 // std::cout << "maskExtent = " << maskExtent[0] << " " << maskExtent[1] << " " << maskExtent[2] << " "
264 // << maskExtent[3] << " " << maskExtent[4] << " " << maskExtent[5] << std::endl;
265 SetDisplayExtentAndCameraPosition(orientation, mSlice, maskExtent, mImageActorList[mTSlice], 0.0);
267 // set previous slice
268 mPreviousTSlice = mSlicer->GetTSlice();
269 mPreviousSlice = mSlicer->GetSlice();
271 //------------------------------------------------------------------------------
274 //------------------------------------------------------------------------------
275 void vvBinaryImageOverlayActor::ComputeExtent(int orientation,
280 switch (orientation) {
281 case vtkImageViewer2::SLICE_ORIENTATION_XY:
282 for(int i=0; i<4; i++) outExtent[i] = inExtent[i];
283 outExtent[4] = outExtent[5] = slice;
285 case vtkImageViewer2::SLICE_ORIENTATION_XZ:
286 for(int i=0; i<2; i++) outExtent[i] = inExtent[i];
287 for(int i=4; i<6; i++) outExtent[i] = inExtent[i];
288 outExtent[2] = outExtent[3] = slice;
290 case vtkImageViewer2::SLICE_ORIENTATION_YZ:
291 for(int i=2; i<6; i++) outExtent[i] = inExtent[i];
292 outExtent[0] = outExtent[1] = slice;
296 //------------------------------------------------------------------------------
299 //----------------------------------------------------------------------------
300 void vvBinaryImageOverlayActor::ComputeExtent(int * inExtent, int * outExtent, vtkImageData * image, vtkImageData * overlay)
302 outExtent[0] = (( image->GetOrigin()[0] + inExtent[0]*image->GetSpacing()[0] ) - overlay->GetOrigin()[0]) /
303 overlay->GetSpacing()[0];
304 outExtent[1] = (( image->GetOrigin()[0] + inExtent[1]*image->GetSpacing()[0] ) - overlay->GetOrigin()[0]) /
305 overlay->GetSpacing()[0];
306 outExtent[2] = (( image->GetOrigin()[1] + inExtent[2]*image->GetSpacing()[1] ) - overlay->GetOrigin()[1]) /
307 overlay->GetSpacing()[1];
308 outExtent[3] = (( image->GetOrigin()[1] + inExtent[3]*image->GetSpacing()[1] ) - overlay->GetOrigin()[1]) /
309 overlay->GetSpacing()[1];
310 outExtent[4] = (( image->GetOrigin()[2] + inExtent[4]*image->GetSpacing()[2] ) - overlay->GetOrigin()[2]) /
311 overlay->GetSpacing()[2];
312 outExtent[5] = (( image->GetOrigin()[2] + inExtent[5]*image->GetSpacing()[2] ) - overlay->GetOrigin()[2]) /
313 overlay->GetSpacing()[2];
315 //----------------------------------------------------------------------------
318 //------------------------------------------------------------------------------
319 void vvBinaryImageOverlayActor::SetDisplayExtentAndCameraPosition(int orientation,
322 vtkImageActor * actor,
326 if (orientation == vtkImageViewer2::SLICE_ORIENTATION_XY) {
327 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[2] > slice)
328 actor->SetPosition(0,0, position);
330 actor->SetPosition(0,0, -position);
332 if (orientation == vtkImageViewer2::SLICE_ORIENTATION_XZ) {
333 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[1] > slice)
334 actor->SetPosition(0,position,0);
336 actor->SetPosition(0,-position,0);
338 if (orientation == vtkImageViewer2::SLICE_ORIENTATION_YZ) {
339 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[0] > slice)
340 actor->SetPosition(position,0, 0);
342 actor->SetPosition(-position,0, 0);
344 actor->SetDisplayExtent(extent);
346 //------------------------------------------------------------------------------