/home/oan/prosjekt/gotools/segmentation/gpl_distro/lsseg_1.0_gpl/include/ParzenDistributionForce.h

Go to the documentation of this file.

//===========================================================================
// The Level-Set Segmentation Library (LSSEG)
//
//
// Copyright (C) 2000-2005 SINTEF ICT, Applied Mathematics, Norway.
//
// This program is free software; you can redistribute it and/or          
// modify it under the terms of the GNU General Public License            
// as published by the Free Software Foundation version 2 of the License. 
//
// This program is distributed in the hope that it will be useful,        
// but WITHOUT ANY WARRANTY; without even the implied warranty of         
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the          
// GNU General Public License for more details.                           
//
// You should have received a copy of the GNU General Public License      
// along with this program; if not, write to the Free Software            
// Foundation, Inc.,                                                      
// 59 Temple Place - Suite 330,                                           
// Boston, MA  02111-1307, USA.                                           
//
// Contact information: e-mail: tor.dokken@sintef.no                      
// SINTEF ICT, Department of Applied Mathematics,                         
// P.O. Box 124 Blindern,                                                 
// 0314 Oslo, Norway.                                                     
// 
//
// Other licenses are also available for this software, notably licenses
// for:
// - Building commercial software.                                        
// - Building software whose source code you wish to keep private.        
//
//===========================================================================
//===========================================================================
//                                                                           
// File: ParzenDistributionForce.h                                          
//                                                                           
// Created: Wed Feb 22 21:10:10 2006                                         
//                                                                           
// Author: Odd A. Andersen <Odd.Andersen@sintef.no>
//                                                                           
// Revision: $Id: ParzenDistributionForce.h,v 1.5 2006/11/25 20:08:23 oan Exp $
//                                                                           
// Description:
//                                                                           
//===========================================================================

#ifndef _PARZENDISTRIBUTIONFORCE_H
#define _PARZENDISTRIBUTIONFORCE_H

#include "ForceGenerator.h" // base class
#include "Image.h"
#include "Mask.h"
#include "LevelSetFunction.h"
#include "Histogram.h"
#include <vector>
#include <iostream>

namespace lsseg {

//===========================================================================
class ParzenDistributionForce : public ForceGenerator
//===========================================================================
{
public:
    ParzenDistributionForce(bool multireg_mode = false) : 
        multi_region_mode_(multireg_mode), 
        num_force_bins_(256),
        img_(0), 
        mask_(0)
    {} // must be initialized before use

    explicit ParzenDistributionForce(const Image<double>* img, 
                                      Mask* mask = 0, 
                                      bool multireg_mode = false) ;

    virtual ~ParzenDistributionForce() {} 

    virtual void init(const Image<double>* img, const Mask* mask = 0);

    virtual void update(const LevelSetFunction& phi);
    virtual double force2D(int x, int y) const;
    virtual double force3D(int x, int y, int z) const ;
    virtual double force(size_t ix) const;
    virtual void force(LevelSetFunction&, const Mask* mask = 0) const;

    void saveChannelDistribution(int channel, std::ostream& os, bool inside) const;

    void fixChannelDistributionTo(int channel, std::istream& is, bool inside);

    void fixDistributionToUniform(int channel, bool inside);

    void unfixChannelDistribution(int channel, bool inside);

    const Image<double>* const baseImage() const {return img_;}

private:

    const bool multi_region_mode_;

    const int num_force_bins_; // number of bins for the precalculated force, must not
                               // be confused with the invidivual number of bins for the
                               // histograms.

    const Image<double>* img_;

    const Mask* mask_;

    std::vector<std::pair<bool, bool> > is_fixed_;

    std::vector<std::pair<Histogram, Histogram> > hist_; // histogram inside / outside

    std::vector<std::pair<double, double> > channel_ranges_;

    std::vector<double> force_bin_factor_;

    std::vector<std::vector<double> > precalc_force_;


    inline void report() const;

    inline int getForceBin(double val, int channel) const; 

    void precompute_force(int channel,
                          const Histogram& ihist,
                          const Histogram& ohist,
                          std::vector<double>& forcevec);
    
    double getCenterBinValue(int channel,int ix);

    inline void get_histogram(int channel, 
                              const LevelSetFunction& phi,
                              Histogram& inside_hist,
                              Histogram& outside_hist) const;
    
    Histogram makeDefaultHistogram(int channel) const ;
};

}; // end namespace lsseg

#endif // _PARZENDISTRIBUTIONFORCE_H

---