1 /*# ---------------------------------------------------------------------
3 # Copyright (c) CREATIS (Centre de Recherche en Acquisition et Traitement de l'Image
5 # Authors : Eduardo Davila, Frederic Cervenansky, Claire Mouton
6 # Previous Authors : Laurent Guigues, Jean-Pierre Roux
7 # CreaTools website : www.creatis.insa-lyon.fr/site/fr/creatools_accueil
9 # This software is governed by the CeCILL-B license under French law and
10 # abiding by the rules of distribution of free software. You can use,
11 # modify and/ or redistribute the software under the terms of the CeCILL-B
12 # license as circulated by CEA, CNRS and INRIA at the following URL
13 # http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
14 # or in the file LICENSE.txt.
16 # As a counterpart to the access to the source code and rights to copy,
17 # modify and redistribute granted by the license, users are provided only
18 # with a limited warranty and the software's author, the holder of the
19 # economic rights, and the successive licensors have only limited
22 # The fact that you are presently reading this means that you have had
23 # knowledge of the CeCILL-B license and that you accept its terms.
24 # ------------------------------------------------------------------------ */
28 * \author Eduardo Davila and Claire Mouton.
29 * \brief Class bbtk::ThresholdImageView.
30 * \date September 2012
33 #include "ColorLayerImageView.h"
35 #include <vtkImageReslice.h>
36 #include <vtkLookupTable.h>
37 #include <vtkImageData.h>
42 //=========================================================================
43 //=========================================================================
44 //=========================================================================
45 //=========================================================================
46 ColorLayerImageView::ColorLayerImageView( )
53 //=========================================================================
54 ColorLayerImageView::~ColorLayerImageView()
57 //=========================================================================
60 //----------------------------------------------------------------------------
61 void ColorLayerImageView::SetSliceFixDynamic(bool fix_dynamic)
63 _fix_dynamic = fix_dynamic;
66 //----------------------------------------------------------------------------
67 void ColorLayerImageView::SetX2(int x2)
72 //----------------------------------------------------------------------------
73 void ColorLayerImageView::SetY2(int y2)
79 //----------------------------------------------------------------------------
80 void ColorLayerImageView::SetZ2(int z2)
85 //----------------------------------------------------------------------------
86 int ColorLayerImageView::GetX() // virtual
89 if (_fix_dynamic==false)
95 result = LayerImageBase::GetX();
101 //----------------------------------------------------------------------------
102 int ColorLayerImageView::GetY() // virtual
105 if (_fix_dynamic==false)
111 result = LayerImageBase::GetY();
117 //----------------------------------------------------------------------------
118 int ColorLayerImageView::GetZ() // virtual
121 if (_fix_dynamic==false)
127 result = LayerImageBase::GetZ();
133 //----------------------------------------------------------------------------
134 void ColorLayerImageView::SetBaseColors(std::vector<double> & base_color)
136 // The base color vector must be of a size multiple of 3, not null.
137 if (base_color.size() != 0 && base_color.size() % 3 == 0)
138 _base_color = base_color;
139 // Otherwise, an exception should be thrown.
140 else if (base_color.size() != 0)
141 std::cout << "CM ColorLayerImageView::SetBaseColors : ERROR!!! The base color vector has an inconsistent size. It must be of a size multiple of 3, different from 0, but its size is: "<< base_color.size() << ". Therefore, the default values will be used as base colors." << std::endl;
145 //----------------------------------------------------------------------------
146 void ColorLayerImageView::SetBaseTransparence(std::vector<double> & base_transparence)
148 _transparence_level_boundary = base_transparence;
152 //----------------------------------------------------------------------------
153 double ColorLayerImageView::GetBaseColors(unsigned int index)
155 if (_base_color.size() > index)
156 return _base_color.at(index);
157 // An exception should be thrown if the index does not exist in the color vector.
160 std::cout << "CM ColorLayerImageView::GetBaseColors : ERROR!!! The index " << index << "is out of the base color vector range." << std ::endl;
165 //----------------------------------------------------------------------------
166 void ColorLayerImageView::SetGreyLevelBoundaries(std::vector<double> & grey_level_boundary)
168 // The size must be greater than or equal to 2 (at least min and max must exist).
169 if ( grey_level_boundary.size() >= 2)
171 sort ( grey_level_boundary.begin(), grey_level_boundary.end() );
172 _grey_level_boundary = grey_level_boundary;
174 // Otherwise, an exception should be thrown.
175 else if (grey_level_boundary.size() != 0)
176 std::cout << "CM ColorLayerImageView::SetGreyLevelBoundaries : ERROR!!! The grey level boundaries vector has an inconsistent size. It must be of a size greater than or equal to 2 (at least min and max must exist), but its size is: " << grey_level_boundary.size() << ". Therefore, the histogram will be equally split." << std::endl;
179 //----------------------------------------------------------------------------
180 double ColorLayerImageView::GetGreyLevelBoundaries(unsigned int index)
182 if (_grey_level_boundary.size() > index)
183 return _grey_level_boundary.at(index);
184 // An exception should be thrown if the index does not exist in the grey level boundary vector.
187 std::cout << "CM ColorLayerImageView::GetGreyLevelBoundaries : ERROR!!! The index " << index << "is out of the grey level boundaries vector range." << std ::endl;
192 //----------------------------------------------------------------------------
193 void ColorLayerImageView::SetPlainOrGradientColor(bool color_type)
195 _color_type = color_type;
198 //----------------------------------------------------------------------------
199 int ColorLayerImageView::GetBaseColorNb()
201 return _base_color.size()/3;
204 //----------------------------------------------------------------------------
205 void ColorLayerImageView::SetDefaultGreyLevelBoundary()
207 // CM Sets the default behaviour concerning the lookup table keeping the base colors already set.
208 _grey_level_boundary.clear();
210 // This avoids a potential division by 0 through delta during the lookup table configuration.
218 // Grey level extrema are set by default to the image grey level extrema.
219 double minTot = _range[0];
220 double maxTot = _range[1];
224 delta = (maxTot - minTot)/GetBaseColorNb();
225 // By default, the histogram is split into BaseColorNb areas of equal width.
226 for (i = 0; i <= GetBaseColorNb() ; i ++)
228 _grey_level_boundary.push_back(minTot + i * delta);
231 delta = (maxTot - minTot)/(GetBaseColorNb()-1);
232 for (i = 0; i < GetBaseColorNb() ; i ++)
234 _grey_level_boundary.push_back(minTot + i * delta);
240 //----------------------------------------------------------------------------
241 void ColorLayerImageView::SetDefaultBaseColorAndGreyLevelBoundary()
243 // CM Sets the default behaviour concerning the lookup table.
244 _grey_level_boundary.clear();
246 // Base colors are set to blue, yellow and red.
248 _base_color.push_back(0.0);
249 _base_color.push_back(0.0);
250 _base_color.push_back(0.5);
252 _base_color.push_back(0.0);
253 _base_color.push_back(0.0);
254 _base_color.push_back(1.0);
256 _base_color.push_back(0.0);
257 _base_color.push_back(1.0);
258 _base_color.push_back(1.0);
260 _base_color.push_back(1.0);
261 _base_color.push_back(1.0);
262 _base_color.push_back(0.0);
264 _base_color.push_back(1.0);
265 _base_color.push_back(0.0);
266 _base_color.push_back(0.0);
268 _base_color.push_back(0.5);
269 _base_color.push_back(0.0);
270 _base_color.push_back(0.0);
273 // This avoids a potential division by 0 through delta during the lookup table configuration.
279 // double minTot = floor (_range[0]);
280 // double maxTot = ceil (_range[1]);
281 double minTot = _range[0];
282 double maxTot = _range[1];
284 double delta = (maxTot - minTot)/8.0;
285 // By default, the histogram is split into three areas of equal width.
287 _grey_level_boundary.push_back(minTot + delta*0);
288 _grey_level_boundary.push_back(minTot + delta*1);
289 // _grey_level_boundary.push_back(minTot + delta*2);
290 _grey_level_boundary.push_back(minTot + delta*3);
291 // _grey_level_boundary.push_back(minTot + delta*4);
292 _grey_level_boundary.push_back(minTot + delta*5);
293 // _grey_level_boundary.push_back(minTot + delta*6);
294 _grey_level_boundary.push_back(minTot + delta*7);
295 _grey_level_boundary.push_back(minTot + delta*8);
298 //----------------------------------------------------------------------------
299 void ColorLayerImageView::ConfigLookupTable() // virtual
304 // Grey level extrema retrieved from the image grey level extrema.
306 GetImage()->GetScalarRange(_range);
307 double minRange = _range[0];
308 double maxRange = _range[1];
310 // ------------------ Setting Default Values
311 // Checks the size consistency of vectors _base_color and _grey_level_boundary.
312 // In case of inconsistency, an exception should be thrown. Instead, the default values are set.
313 if (GetBaseColorNb() == 0)
315 SetDefaultBaseColorAndGreyLevelBoundary();
317 else { // If at least one color has been set, set the grey level boundaries to build an equipartition of the image grey levels, keeping the base colors defined.
318 if (GetBaseColorNb() != (int)_grey_level_boundary.size() )
320 std::cout << "CM ColorLayerImageView::ConfigLookupTable : ERROR!!! Inconsistency between the sizes of vectors _base_color and _grey_level_boundary. _base_color.size()/3 (=" <<_base_color.size()/3 << ") should be equal to _grey_level_boundary.size() - 1 (=" << _grey_level_boundary.size() - 1 << ")." << std::endl;
321 std::cout << "CM The default values for the base colors and the grey level boundaries will be set." << std::endl;
322 SetDefaultGreyLevelBoundary();
325 // ------------------ End Of Setting Default Values
327 int iTransparence,transparenceSize = _transparence_level_boundary.size();
328 for (iTransparence=transparenceSize ; iTransparence < GetBaseColorNb(); iTransparence++ )
330 _transparence_level_boundary.push_back(1);
333 // ------------------ Defining the Threshold Table
334 vtkLookupTable* thresholdTable = GetThresholdTable();
335 double thresholdTable_range[2];
336 double r1,r2,g1,g2,b1,b2,t1,t2;
339 // CM Number of table values consistent with the effective number of values (see loop below).
340 thresholdTable_range[1]= std::max( GetGreyLevelBoundaries( _grey_level_boundary.size()-1 ) , ceil(maxRange) );
341 thresholdTable_range[0]= std::min( GetGreyLevelBoundaries(0), floor(minRange) );
343 thresholdTable->SetTableRange(thresholdTable_range);
344 thresholdTable->SetAlphaRange( 0,1 );
345 thresholdTable->SetValueRange( 0,1 );
347 thresholdTable->SetSaturationRange( 0,0 );
348 thresholdTable->SetRampToLinear( );
350 maxColorsThresholdTable = 1000;
351 thresholdTable->SetNumberOfTableValues( maxColorsThresholdTable );
354 // Defines the lookup table.
355 // If the image has a degradation in one (or both) extremum (lower or higher),
356 // sets the corresponding grey levels as transparent white in the lookup table.
358 // _color_type true, i.e. plain colors.
359 // Sets the plain color for each grey level interval.
360 if (_color_type) // Plain
362 if ( minRange<GetGreyLevelBoundaries(0) )
365 end = GrayLevel_TO_colorTableIndex( GetGreyLevelBoundaries(0) );
366 FillColorTable(start,end, 0,0,0, 0,0,0, 0,0);
369 if ( maxRange>GetGreyLevelBoundaries( _grey_level_boundary.size()-1 ) )
371 start = GrayLevel_TO_colorTableIndex( GetGreyLevelBoundaries( _grey_level_boundary.size()-1 ) );
372 end = GrayLevel_TO_colorTableIndex( maxRange );
373 FillColorTable(start,end, 0,0,0, 0,0,0, 0,0);
375 for (int iColor = 0; iColor < GetBaseColorNb(); iColor++)
377 r1 = GetBaseColors(iColor*3+0);
378 g1 = GetBaseColors(iColor*3+1);
379 b1 = GetBaseColors(iColor*3+2);
380 start = GrayLevel_TO_colorTableIndex( GetGreyLevelBoundaries(iColor) );
381 end = GrayLevel_TO_colorTableIndex( GetGreyLevelBoundaries(iColor+1) );
382 t1 = _transparence_level_boundary[iColor];
383 // t2 = _transparence_level_boundary[iColor+1];
384 FillColorTable(start,end, r1,g1,b1,r1,g1,b1,t1,t1);
386 } else { // _color_type false, i.e. gradient color
387 if (GetBaseColorNb() > 1)
389 for (int iColor = 0; iColor < GetBaseColorNb()-1; iColor++)
391 r1 = GetBaseColors( (iColor+0)*3+0 );
392 g1 = GetBaseColors( (iColor+0)*3+1 );
393 b1 = GetBaseColors( (iColor+0)*3+2 );
394 r2 = GetBaseColors( (iColor+1)*3+0);
395 g2 = GetBaseColors( (iColor+1)*3+1);
396 b2 = GetBaseColors( (iColor+1)*3+2);
397 start = GrayLevel_TO_colorTableIndex( GetGreyLevelBoundaries(iColor) );
398 end = GrayLevel_TO_colorTableIndex( GetGreyLevelBoundaries(iColor+1) );
399 t1 = _transparence_level_boundary[iColor];
400 t2 = _transparence_level_boundary[iColor+1];
401 FillColorTable( start,end, r1,g1,b1, r2,g2,b2, t1,t2 );
404 } // End Of if (!_color_type)
406 double minRangeForColorBar, maxRangeForColorBar;
407 GetRangeForColorBar( minRangeForColorBar, maxRangeForColorBar );
408 thresholdTable->SetRange( minRangeForColorBar, maxRangeForColorBar );
410 // thresholdTable->SetTableRange( minRange, maxRange );
411 // thresholdTable->SetValueRange( 0.0, 1.0 );
412 // thresholdTable->SetAboveRangeColor(0,0,0,0);
413 // thresholdTable->SetBelowRangeColor(0,0,0,0);
415 thresholdTable->Build( );
420 //----------------------------------------------------------------------------
421 int ColorLayerImageView::GrayLevel_TO_colorTableIndex( double VALUE )
423 GetImage()->Update();
424 GetImage()->GetScalarRange(_range);
425 double minRange = _range[0];
426 double maxRange = _range[1];
427 return maxColorsThresholdTable * (VALUE-minRange) / (maxRange-minRange);
430 //----------------------------------------------------------------------------
431 void ColorLayerImageView::FillColorTable(int start, int end, double r1, double g1, double b1, double r2, double g2, double b2, double t1, double t2)
433 vtkLookupTable* thresholdTable = GetThresholdTable();
435 double delta = end-start;
436 double dr = (r2-r1)/delta;
437 double dg = (g2-g1)/delta;
438 double db = (b2-b1)/delta;
439 double dt = (t2-t1)/delta;
440 for (iTable=0; iTable<=delta; iTable++)
442 thresholdTable->SetTableValue( iTable+start , r1+dr*iTable, g1+dg*iTable, b1+db*iTable,t1+dt*iTable);