equalizedHisto.h

Go to the documentation of this file.

/* This software and supporting documentation are distributed by
 *     Institut Federatif de Recherche 49
 *     CEA/NeuroSpin, Batiment 145,
 *     91191 Gif-sur-Yvette cedex
 *     France
 *
 * This software is governed by the CeCILL-B license under
 * French law and abiding by the rules of distribution of free software.
 * You can  use, modify and/or redistribute the software under the
 * terms of the CeCILL-B license as circulated by CEA, CNRS
 * and INRIA at the following URL "http://www.cecill.info".
 *
 * As a counterpart to the access to the source code and  rights to copy,
 * modify and redistribute granted by the license, users are provided only
 * with a limited warranty  and the software's author,  the holder of the
 * economic rights,  and the successive licensors  have only  limited
 * liability.
 *
 * In this respect, the user's attention is drawn to the risks associated
 * with loading,  using,  modifying and/or developing or reproducing the
 * software by the user in light of its specific status of free software,
 * that may mean  that it is complicated to manipulate,  and  that  also
 * therefore means  that it is reserved for developers  and  experienced
 * professionals having in-depth computer knowledge. Users are therefore
 * encouraged to load and test the software's suitability as regards their
 * requirements in conditions enabling the security of their systems and/or
 * data to be ensured and,  more generally, to use and operate it in the
 * same conditions as regards security.
 *
 * The fact that you are presently reading this means that you have had
 * knowledge of the CeCILL-B license and that you accept its terms.
 */
 
 
#ifndef AIMS_HISTOGRAM_EQUALIZEDHISTO_H
#define AIMS_HISTOGRAM_EQUALIZEDHISTO_H
 
#include <aims/histogram/simpleHisto.h>

template< class T>
class Equalizer
{ 
 public:
  

  Equalizer() {};
  virtual ~Equalizer() { }
  
  //
  //
  // Histogram equalization from Volume
  // return the equalized image
  carto::VolumeRef<T> doit(const carto::rc_ptr<carto::Volume<T> > & thing);
};
 
 
template <class T> inline
carto::VolumeRef<T> Equalizer<T>::doit(
  const carto::rc_ptr<carto::Volume<T> > & thing)
{
 
  // 
  // Data 
  // 
  int min = int(thing->min()), max = int(thing->max()),
    nb_tot = thing->getSizeX() * thing->getSizeY() * thing->getSizeZ();
  float fraction;
  int tmp;
  
  carto::VolumeRef<T> res = carto::VolumeRef<T>( thing ).deepcopy();
  typename carto::Volume<T>::const_iterator it;
  
  //
  // Histogram computation
  // 
  max = max - min;
  carto::VolumeRef<int32_t> myHisto( max + 1, 1, 1, 1,
                                     carto::AllocatorContext::fast() );
  
  for(it = thing->begin(); it != thing->end(); ++it)
  {
    tmp = ( int32_t ) *it;
    ++myHisto( tmp - min );
  }
  
  carto::VolumeRef<float> cumul( max + 1, 1, 1, 1,
                                 carto::AllocatorContext::fast() );
  cumul( 0 ) = myHisto( 0 ); 
  
  carto::Volume<int32_t>::iterator it1;
  float * it2 = &*cumul.begin() + 1;
  
  for( it1 = myHisto.begin(), ++it1; it1 != myHisto.end(); ++it1, ++it2 )
      *it2 = *it1 + *( it2 - 1 );
  
  carto::Volume<float>::iterator it4;
  for (it4 = cumul.begin(); it4 != cumul.end(); ++it4 )
      *it4 /= (float) nb_tot;
 
  //
  // Rounded cumulated histogram
  //
  carto::VolumeRef<int32_t> roundcumul( max + 1, 1, 1, 1,
                                        carto::AllocatorContext::fast() );
 
  it2  = &*cumul.begin();
  for (it1  = roundcumul.begin(); it1  != roundcumul.end(); ++it1 , ++it2)
    {
      fraction = *it2  * max;
      *it1 += int32_t(fraction + 0.5);            // round by implicit cast
    }
 
  //
  // Equalized data computation
  // 
  typename carto::Volume<T>::iterator it3;
 
  for ( it3 = res.begin(); it3 != res.end(); ++it3, ++it)
      *it3 = static_cast<T> ( roundcumul( int32_t(*it3) ) - roundcumul(0) );
  
  return (res);  
}
 
#endif