/home/oan/prosjekt/gotools/segmentation/gpl_distro/lsseg_1.0_gpl/include/LevelSetFunction.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: LevelSetFunction.h                                                  
//                                                                           
// Created: Fri Feb 17 13:29:58 2006                                         
//                                                                           
// Author: Odd A. Andersen <Odd.Andersen@sintef.no>
//                                                                           
// Revision: $Id: LevelSetFunction.h,v 1.12 2006/11/16 06:38:21 oan Exp $
//                                                                           
// Description:
//                                                                           
//===========================================================================

#ifndef _LEVELSETFUNCTION_H
#define _LEVELSETFUNCTION_H

#include "errormacros.h"
#include "Image.h"
#include "Mask.h"

namespace lsseg {

//===========================================================================
class LevelSetFunction : public Image<double>
//===========================================================================
{
public:
    LevelSetFunction() : Image<double>(0, 0, 0, 1) {}
    LevelSetFunction(int x, int y, int z = 1, double val = 0) 
        : Image<double>(x, y, z, 1, val) {}
    virtual  ~LevelSetFunction() {}

    LevelSetFunction(const LevelSetFunction& rhs) : Image<double>(rhs) {};
    
    LevelSetFunction(const LevelSetFunction& rhs, bool copy) 
        : Image<double>(rhs, copy) {}

    virtual Image<double>& operator=(const Image<double>& rhs) {
        single_channel_assert(rhs.numChannels());
        return Image<double>::operator=(rhs);
    }

    virtual LevelSetFunction& operator=(const double& val)
    {
        std::fill(data_.begin(), data_.end(), val);
        return *this;
    }

    void resize(const LevelSetFunction& rhs)
    {
        resize(rhs.dimx(), rhs.dimy(), rhs.dimz(), 1);
    }

    virtual void resize(int x, int y, int z= 1, int channels = 1) {
        single_channel_assert(channels);
        Image<double>::resize(x, y, z, channels);
    }
    
    virtual void swap(Image<double>& rhs) {
        single_channel_assert(rhs.numChannels());
        Image<double>::swap(rhs);
    }
    
    virtual void permute(const int* const perm) {
        ALWAYS_ERROR_IF(perm[3] != 3, 
                        "cannot permute the 'channel' of a LevelSetFunction");
        Image<double>::permute(perm);
    }    

    virtual void read(std::istream& is, bool binary=true) {
        LevelSetFunction tmp;
        tmp.Image<double>::read(is, binary);
        single_channel_assert(tmp.numChannels());
        swap(tmp);
    }

    //======================================================================
    //  Functions purely proper to the LevelSetFunction (not to be found in
    //  the Image<double> object).
    //======================================================================

    inline double gradientNorm2D(int x, int y) const;

    inline double gradientNorm3D(int x, int y, int z) const;

    inline double curvature2D(int x, int y) const;
    
    inline double curvature3D(int x, int y, int z) const;

    inline void curvature2D(Image<double>& target, Mask* mask = 0) const;

    inline void curvature3D(Image<double>& target, Mask* mask = 0) const;

    inline void curvatureTimesGrad2D(Image<double>& target, Mask* mask = 0) const;

    inline double curvatureTimesGrad2D(int x, int y) const;

    inline double curvatureTimesGrad3D(int x, int y, int z) const;

    inline void curvatureTimesGrad3D(Image<double>& target, Mask* mask = 0) const;

    inline void gradientNorm2D(Image<double>& target, Mask* mask = 0) const;

    inline void gradientNorm3D(Image<double>& target, Mask* mask = 0) const;

    void reinitialize2D(const Mask* mask = 0);

    void reinitialize3D(const Mask* mask = 0);
    
private:
    void single_channel_assert(int num_chan) {
        ALWAYS_ERROR_IF(num_chan != 1, "Single channel assert failed.");
    }

    mutable double cached_;
};

}; // end namespace lsseg

#include "LevelSetFunction_implementation.h"

#endif // _LEVELSETFUNCTION_H

---