imaging/vtkImageAnisotropicDiffusion2D.h Source File

00001 /*=========================================================================
00002 
00003   Program:   Visualization Toolkit
00004   Module:    $RCSfile: vtkImageAnisotropicDiffusion2D.h,v $
00005   Language:  C++
00006 
00007 Copyright (c) 1993-2001 Ken Martin, Will Schroeder, Bill Lorensen 
00008 All rights reserved.
00009 
00010 Redistribution and use in source and binary forms, with or without
00011 modification, are permitted provided that the following conditions are met:
00012 
00013  * Redistributions of source code must retain the above copyright notice,
00014    this list of conditions and the following disclaimer.
00015 
00016  * Redistributions in binary form must reproduce the above copyright notice,
00017    this list of conditions and the following disclaimer in the documentation
00018    and/or other materials provided with the distribution.
00019 
00020  * Neither name of Ken Martin, Will Schroeder, or Bill Lorensen nor the names
00021    of any contributors may be used to endorse or promote products derived
00022    from this software without specific prior written permission.
00023 
00024  * Modified source versions must be plainly marked as such, and must not be
00025    misrepresented as being the original software.
00026 
00027 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
00028 AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00029 IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
00030 ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE FOR
00031 ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
00032 DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
00033 SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
00034 CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
00035 OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
00036 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00037 
00038 =========================================================================*/
00070 #ifndef __vtkImageAnisotropicDiffusion2D_h
00071 #define __vtkImageAnisotropicDiffusion2D_h
00072 
00073 
00074 #include "vtkImageSpatialFilter.h"
00075 class VTK_EXPORT vtkImageAnisotropicDiffusion2D : public vtkImageSpatialFilter
00076 {
00077 public:
00078   static vtkImageAnisotropicDiffusion2D *New();
00079   vtkTypeMacro(vtkImageAnisotropicDiffusion2D,vtkImageSpatialFilter);
00080   void PrintSelf(ostream& os, vtkIndent indent);
00081 
00084   void SetNumberOfIterations(int num);
00085 
00087   vtkGetMacro(NumberOfIterations,int);
00088 
00090   vtkSetMacro(DiffusionThreshold,float);
00091   vtkGetMacro(DiffusionThreshold,float);
00092   
00094   vtkSetMacro(DiffusionFactor,float);
00095   vtkGetMacro(DiffusionFactor,float);
00096 
00098   vtkSetMacro(Faces,int);
00099   vtkGetMacro(Faces,int);
00100   vtkBooleanMacro(Faces,int);
00101   vtkSetMacro(Edges,int);
00102   vtkGetMacro(Edges,int);
00103   vtkBooleanMacro(Edges,int);
00104   vtkSetMacro(Corners,int);
00105   vtkGetMacro(Corners,int);
00106   vtkBooleanMacro(Corners,int);
00107 
00110   vtkSetMacro(GradientMagnitudeThreshold,int);
00111   vtkGetMacro(GradientMagnitudeThreshold,int);
00112   vtkBooleanMacro(GradientMagnitudeThreshold,int);
00113   
00114 protected:
00115   vtkImageAnisotropicDiffusion2D();
00116   ~vtkImageAnisotropicDiffusion2D() {};
00117   vtkImageAnisotropicDiffusion2D(const vtkImageAnisotropicDiffusion2D&) {};
00118   void operator=(const vtkImageAnisotropicDiffusion2D&) {};
00119 
00120   int NumberOfIterations;
00121   float DiffusionThreshold;
00122   float DiffusionFactor;  
00123   // to determine which neighbors to diffuse
00124   int Faces;
00125   int Edges;
00126   int Corners;
00127   // What threshold to use
00128   int GradientMagnitudeThreshold;
00129   
00130   void ThreadedExecute(vtkImageData *inData, vtkImageData *outData, 
00131              int extent[6], int id);
00132   void Iterate(vtkImageData *in, vtkImageData *out, 
00133           float ar0, float ar1, int *coreExtent, int count);
00134 };
00135 
00136 #endif
00137 
00138 
00139