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 = vtkLookupTable::New();
43 //------------------------------------------------------------------------------
46 //------------------------------------------------------------------------------
47 vvBinaryImageOverlayActor::~vvBinaryImageOverlayActor()
49 for (unsigned int i = 0; i < mSlicer->GetImage()->GetVTKImages().size(); i++) {
50 mSlicer->GetRenderer()->RemoveActor(mImageActorList[i]);
52 mImageActorList.clear();
55 //------------------------------------------------------------------------------
58 //------------------------------------------------------------------------------
59 void vvBinaryImageOverlayActor::SetColor(double r, double g, double b)
65 //------------------------------------------------------------------------------
68 //------------------------------------------------------------------------------
69 void vvBinaryImageOverlayActor::SetSlicer(vvSlicer * slicer)
73 //------------------------------------------------------------------------------
76 //------------------------------------------------------------------------------
77 void vvBinaryImageOverlayActor::Initialize()
80 std::cerr << "ERROR. Please use setSlicer before setSlicer in vvBinaryImageOverlayActor." << std::endl;
85 std::cerr << "ERROR. Please use setImage before setSlicer in vvBinaryImageOverlayActor." << std::endl;
88 // Create an actor for each time slice
89 for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
90 // how many intensity ?
91 vtkImageMapToRGBA * mOverlayMapper = vtkImageMapToRGBA::New();
92 mOverlayMapper->SetInput(mImage->GetVTKImages()[0]); // DS TODO : to change if it is 4D !!!
93 mColorLUT->SetRange(0,1);
94 mColorLUT->SetNumberOfTableValues(2);
95 mColorLUT->SetTableValue(mBackgroundValue, 0, 0, 0, 0.0); // BG
96 mColorLUT->SetTableValue(1, mColor[0], mColor[1], mColor[2], mAlpha); // FG
97 mOverlayMapper->SetLookupTable(mColorLUT);
99 vtkImageActor * mOverlayActor = vtkImageActor::New();
100 mOverlayActor->SetInput(mOverlayMapper->GetOutput());
101 mOverlayActor->SetPickable(0);
102 mOverlayActor->SetVisibility(true);
103 //mOverlayActor->SetOpacity(1.0);
105 mMapperList.push_back(mOverlayMapper);
106 mImageActorList.push_back(mOverlayActor);
107 mSlicer->GetRenderer()->AddActor(mOverlayActor);
110 //------------------------------------------------------------------------------
113 //------------------------------------------------------------------------------
114 void vvBinaryImageOverlayActor::SetOpacity(double d)
118 //------------------------------------------------------------------------------
121 //------------------------------------------------------------------------------
122 // void vvBinaryImageOverlayActor::UpdateOpacity(double d) {
124 // mColorLUT->SetTableValue(1, mColor[0], mColor[1], mColor[2], mAlpha); // FG
125 // for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
126 // // how many intensity ?
127 // vtkImageMapToRGBA * mOverlayMapper = mMapperList[numImage];
128 // mOverlayMapper->SetLookupTable(mColorLUT);
130 // vtkImageActor * mOverlayActor = mImageActorList[numImage];
131 // mOverlayActor->SetInput(mOverlayMapper->GetOutput());
134 //------------------------------------------------------------------------------
137 //------------------------------------------------------------------------------
138 // void vvBinaryImageOverlayActor::SetColor(double r, double v, double b) {
142 // // mColorLUT->SetTableValue(1, mColor[0], mColor[1], mColor[2], mAlpha); // FG
143 // // for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
144 // // // how many intensity ?
145 // // vtkImageMapToRGBA * mOverlayMapper = mMapperList[numImage];
146 // // mOverlayMapper->SetLookupTable(mColorLUT);
148 // // vtkImageActor * mOverlayActor = mImageActorList[numImage];
149 // // mOverlayActor->SetInput(mOverlayMapper->GetOutput());
152 //------------------------------------------------------------------------------
155 //------------------------------------------------------------------------------
156 void vvBinaryImageOverlayActor::SetImage(vvImage::Pointer image, double bg)
159 mBackgroundValue = bg;
161 //------------------------------------------------------------------------------
164 //------------------------------------------------------------------------------
165 void vvBinaryImageOverlayActor::HideActors()
167 if (!mSlicer) return;
168 mSlice = mSlicer->GetSlice();
169 for(unsigned int i=0; i<mImageActorList.size(); i++) {
170 mImageActorList[i]->VisibilityOff();
172 // Caller MUST call Render
173 // mSlicer->Render();
175 //------------------------------------------------------------------------------
178 //------------------------------------------------------------------------------
179 void vvBinaryImageOverlayActor::ShowActors()
181 if (!mSlicer) return;
182 mSlice = mSlicer->GetSlice();
183 mTSlice = mSlicer->GetTSlice();
184 // for(unsigned int i=0; i<mSquaresActorList.size(); i++) {
185 mImageActorList[mTSlice]->VisibilityOn();
186 UpdateSlice(0, mSlice);
188 // Caller MUST call Render
191 //------------------------------------------------------------------------------
194 //------------------------------------------------------------------------------
195 void vvBinaryImageOverlayActor::UpdateColor()
197 mColorLUT->SetTableValue(1, mColor[0], mColor[1], mColor[2], mAlpha); // FG
198 for (unsigned int numImage = 0; numImage < mSlicer->GetImage()->GetVTKImages().size(); numImage++) {
199 // how many intensity ?
200 vtkImageMapToRGBA * mOverlayMapper = mMapperList[numImage];
201 mOverlayMapper->SetLookupTable(mColorLUT);
203 vtkImageActor * mOverlayActor = mImageActorList[numImage];
204 mOverlayActor->SetInput(mOverlayMapper->GetOutput());
207 //------------------------------------------------------------------------------
209 //------------------------------------------------------------------------------
210 void vvBinaryImageOverlayActor::UpdateSlice(int slicer, int slice)
212 if (!mSlicer) return;
214 if (mPreviousSlice == mSlicer->GetSlice()) {
215 if (mPreviousTSlice == mSlicer->GetTSlice()) {
216 //DD("=========== NOTHING");
217 return; // Nothing to do
222 mSlice = mSlicer->GetSlice();
223 mTSlice = mSlicer->GetTSlice();
226 int * imageExtent = mSlicer->GetExtent();
227 int orientation = mSlicer->GetOrientation();
229 ComputeExtent(orientation, mSlice, imageExtent, maskExtent);
230 //ComputeExtent(imageExtent, maskExtent, mSlicer->GetImage()->GetFirstVTKImageData(), mImage->GetFirstVTKImageData());
231 ComputeExtent(maskExtent, maskExtent, mSlicer->GetImage()->GetFirstVTKImageData(), mImage->GetFirstVTKImageData());
232 // std::cout << "maskExtent = " << maskExtent[0] << " " << maskExtent[1] << " " << maskExtent[2] << " "
233 // << maskExtent[3] << " " << maskExtent[4] << " " << maskExtent[5] << std::endl;
234 mSlicer->ClipDisplayedExtent(maskExtent, mMapperList[mTSlice]->GetInput()->GetWholeExtent());
235 // std::cout << "maskExtent = " << maskExtent[0] << " " << maskExtent[1] << " " << maskExtent[2] << " "
236 // << maskExtent[3] << " " << maskExtent[4] << " " << maskExtent[5] << std::endl;
237 SetDisplayExtentAndCameraPosition(orientation, mSlice, maskExtent, mImageActorList[mTSlice], 0.0);
239 // set previous slice
240 mPreviousTSlice = mSlicer->GetTSlice();
241 mPreviousSlice = mSlicer->GetSlice();
243 //------------------------------------------------------------------------------
246 //------------------------------------------------------------------------------
247 void vvBinaryImageOverlayActor::ComputeExtent(int orientation,
252 // std::cout << "InExtent = " << inExtent[0] << " " << inExtent[1] << " " << inExtent[2] << " "
253 // << inExtent[3] << " " << inExtent[4] << " " << inExtent[5] << std::endl;
255 switch (orientation) {
256 case vtkImageViewer2::SLICE_ORIENTATION_XY:
257 for(int i=0; i<4; i++) outExtent[i] = inExtent[i];
258 outExtent[4] = outExtent[5] = slice;
260 case vtkImageViewer2::SLICE_ORIENTATION_XZ:
261 for(int i=0; i<2; i++) outExtent[i] = inExtent[i];
262 for(int i=4; i<6; i++) outExtent[i] = inExtent[i];
263 outExtent[2] = outExtent[3] = slice;
265 case vtkImageViewer2::SLICE_ORIENTATION_YZ:
266 for(int i=2; i<6; i++) outExtent[i] = inExtent[i];
267 outExtent[0] = outExtent[1] = slice;
270 // std::cout << "OutExtent = " << outExtent[0] << " " << outExtent[1] << " " << outExtent[2] << " "
271 // << outExtent[3] << " " << outExtent[4] << " " << outExtent[5] << std::endl;
273 //------------------------------------------------------------------------------
276 //----------------------------------------------------------------------------
277 void vvBinaryImageOverlayActor::ComputeExtent(int * inExtent, int * outExtent, vtkImageData * image, vtkImageData * overlay)
279 outExtent[0] = (( image->GetOrigin()[0] + inExtent[0]*image->GetSpacing()[0] ) - overlay->GetOrigin()[0]) /
280 overlay->GetSpacing()[0];
281 outExtent[1] = (( image->GetOrigin()[0] + inExtent[1]*image->GetSpacing()[0] ) - overlay->GetOrigin()[0]) /
282 overlay->GetSpacing()[0];
283 outExtent[2] = (( image->GetOrigin()[1] + inExtent[2]*image->GetSpacing()[1] ) - overlay->GetOrigin()[1]) /
284 overlay->GetSpacing()[1];
285 outExtent[3] = (( image->GetOrigin()[1] + inExtent[3]*image->GetSpacing()[1] ) - overlay->GetOrigin()[1]) /
286 overlay->GetSpacing()[1];
287 outExtent[4] = (( image->GetOrigin()[2] + inExtent[4]*image->GetSpacing()[2] ) - overlay->GetOrigin()[2]) /
288 overlay->GetSpacing()[2];
289 outExtent[5] = (( image->GetOrigin()[2] + inExtent[5]*image->GetSpacing()[2] ) - overlay->GetOrigin()[2]) /
290 overlay->GetSpacing()[2];
292 //----------------------------------------------------------------------------
295 //------------------------------------------------------------------------------
296 void vvBinaryImageOverlayActor::SetDisplayExtentAndCameraPosition(int orientation,
299 vtkImageActor * actor,
303 if (orientation == vtkImageViewer2::SLICE_ORIENTATION_XY) {
304 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[2] > slice)
305 actor->SetPosition(0,0, position);
307 actor->SetPosition(0,0, -position);
309 if (orientation == vtkImageViewer2::SLICE_ORIENTATION_XZ) {
310 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[1] > slice)
311 actor->SetPosition(0,position,0);
313 actor->SetPosition(0,-position,0);
315 if (orientation == vtkImageViewer2::SLICE_ORIENTATION_YZ) {
316 if (mSlicer->GetRenderer()->GetActiveCamera()->GetPosition()[0] > slice)
317 actor->SetPosition(position,0, 0);
319 actor->SetPosition(-position,0, 0);
321 actor->SetDisplayExtent(extent);
323 //------------------------------------------------------------------------------