VariableShapeFilter.h

Go to the documentation of this file.

//     ____   ______ __
//    / __ \ / ____// /
//   / /_/ // /    / /
//  / ____// /___ / /___   PixInsight Class Library
// /_/     \____//_____/   PCL 2.7.0
// ----------------------------------------------------------------------------
// pcl/VariableShapeFilter.h - Released 2024-06-18T15:48:54Z
// ----------------------------------------------------------------------------
// This file is part of the PixInsight Class Library (PCL).
// PCL is a multiplatform C++ framework for development of PixInsight modules.
//
// Copyright (c) 2003-2024 Pleiades Astrophoto S.L. All Rights Reserved.
//
// Redistribution and use in both source and binary forms, with or without
// modification, is permitted provided that the following conditions are met:
//
// 1. All redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
//
// 2. All redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
//
// 3. Neither the names "PixInsight" and "Pleiades Astrophoto", nor the names
//    of their contributors, may be used to endorse or promote products derived
//    from this software without specific prior written permission. For written
//    permission, please contact info@pixinsight.com.
//
// 4. All products derived from this software, in any form whatsoever, must
//    reproduce the following acknowledgment in the end-user documentation
//    and/or other materials provided with the product:
//
//    "This product is based on software from the PixInsight project, developed
//    by Pleiades Astrophoto and its contributors (https://pixinsight.com/)."
//
//    Alternatively, if that is where third-party acknowledgments normally
//    appear, this acknowledgment must be reproduced in the product itself.
//
// THIS SOFTWARE IS PROVIDED BY PLEIADES ASTROPHOTO AND ITS CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
// TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL PLEIADES ASTROPHOTO OR ITS
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, BUSINESS
// INTERRUPTION; PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; AND LOSS OF USE,
// DATA OR PROFITS) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
// ----------------------------------------------------------------------------
 
#ifndef __PCL_VariableShapeFilter_h
#define __PCL_VariableShapeFilter_h
 
 
#include <pcl/Defs.h>
 
#include <pcl/AutoPointer.h>
#include <pcl/KernelFilter.h>
#include <pcl/Math.h>
#include <pcl/PixelInterpolation.h>
 
namespace pcl
{
 
// ----------------------------------------------------------------------------
 
class PCL_CLASS VariableShapeFilter : public KernelFilter
{
public:
 
   VariableShapeFilter() = default;
 
   VariableShapeFilter( float sigma, float shape = 2, float epsilon = 0.01, const String& name = String() )
   {
      Initialize( sigma, shape, epsilon, 1, 0 );
      Rename( name );
   }
 
   VariableShapeFilter( float sigma, float shape, float epsilon, float rho, float theta = 0, const String& name = String() )
   {
      Initialize( sigma, shape, epsilon, rho, theta );
      Rename( name );
   }
 
   VariableShapeFilter( int n, float shape = 2, float epsilon = 0.01, const String& name = String() )
   {
      Initialize( n, shape, epsilon, 1, 0 );
      Rename( name );
   }
 
   VariableShapeFilter( int n, float shape, float epsilon, float rho, float theta = 0, const String& name = String() )
   {
      Initialize( n, shape, epsilon, rho, theta );
      Rename( name );
   }
 
   VariableShapeFilter( const VariableShapeFilter& ) = default;
 
   VariableShapeFilter( VariableShapeFilter&& ) = default;
 
   KernelFilter* Clone() const override
   {
      return new VariableShapeFilter( *this );
   }
 
   SeparableFilter AsSeparableFilter( float tolerance = __PCL_DEFAULT_FILTER_SEPARABILITY_TOLERANCE ) const override
   {
      if ( m_shape == 2 && m_rho == 1 )
      {
         FVector v = coefficients.RowVector( Size()>>1 );
         return SeparableFilter( v, v );
      }
      return SeparableFilter();
   }
 
   bool IsSeparable() const override
   {
      return m_shape == 2 && m_rho == 1;
   }
 
   VariableShapeFilter& operator =( const VariableShapeFilter& ) = default;
 
   VariableShapeFilter& operator =( VariableShapeFilter&& ) = default;
 
   float SigmaX() const
   {
      return m_sigma;
   }
 
   float SigmaY() const
   {
      return m_rho*m_sigma;
   }
 
   float Sigma() const
   {
      return SigmaX();
   }
 
   float Shape() const
   {
      return m_shape;
   }
 
   float Truncation() const
   {
      return m_epsilon;
   }
 
   float AspectRatio() const
   {
      return m_rho;
   }
 
   float RotationAngle() const
   {
      return m_theta;
   }
 
   double FWHMx() const
   {
      return 2 * m_sigma * Pow( m_shape*0.6931471805599453, 1.0/m_shape );
 
   }
 
   double FWHMy() const
   {
      return m_rho * FWHMx();
 
   }
 
   double FWHM() const
   {
      return FWHMx();
   }
 
   void Set( float sigma, float shape, float epsilon, float rho, float theta )
   {
      Initialize( sigma, shape, epsilon, rho, theta );
   }
 
   void Set( float sigma, float shape, float epsilon, float rho )
   {
      Initialize( sigma, shape, epsilon, rho, m_theta );
   }
 
   void Set( float sigma, float shape, float epsilon )
   {
      Initialize( sigma, shape, epsilon, m_rho, m_theta );
   }
 
   void Set( float sigma, float shape )
   {
      Initialize( sigma, shape, m_epsilon, m_rho, m_theta );
   }
 
   void Set( float sigma )
   {
      Initialize( sigma, m_shape, m_epsilon, m_rho, m_theta );
   }
 
   void SetSigma( float sigma )
   {
      Set( sigma );
   }
 
   void SetShape( float shape )
   {
      Set( m_sigma, shape );
   }
 
   void SetTruncation( float epsilon )
   {
      Set( m_sigma, m_shape, epsilon );
   }
 
   void SetAspectRatio( float rho )
   {
      Set( m_sigma, m_shape, m_epsilon, rho );
   }
 
   void SetRotationAngle( float theta )
   {
      Set( m_sigma, m_shape, m_epsilon, m_rho, theta );
   }
 
   void Resize( int n ) override
   {
      Initialize( n, m_shape, m_epsilon, m_rho, m_theta );
   }
 
private:
 
   float m_sigma = 2.0F;    // standard deviation, horizontal axis
   float m_shape = 2.0F;    // controls the kurtosis of the filter distribution (2=normal)
   float m_rho = 1.0F;      // vertical:horizontal axes ratio
   float m_theta = 0.0F;    // rotation angle in radians, [0,+pi]
   float m_epsilon = 0.01F; // maximum truncation error in sigma units
 
   void Initialize( float s, float k, float e, float r, float a )
   {
      PCL_PRECONDITION( s > 0 )
      PCL_PRECONDITION( k > 0 )
      PCL_PRECONDITION( e > 0 )
      PCL_PRECONDITION( r >= 0 && r <= 1 )
      PCL_PRECONDITION( a >= 0 && a <= Const<float>::pi() )
      m_sigma = s;
      m_shape = k;
      m_epsilon = Abs( e );
      m_rho = Range( r, 0.0F, 1.0F );
      m_theta = Range( a, 0.0F, Const<float>::pi() );
      KernelFilter::Resize( 1 + (Max( 1, RoundInt( Pow( -m_shape*Pow( m_sigma, m_shape )*Ln( m_epsilon ), 1/m_shape ) ) ) << 1) );
      Rebuild();
   }
 
   void Initialize( int n, float k, float e, float r, float a )
   {
      PCL_PRECONDITION( n == 0 || n >= 3 && (n & 1) != 0 )
      PCL_PRECONDITION( k > 0 )
      PCL_PRECONDITION( e > 0 )
      PCL_PRECONDITION( r >= 0 && r <= 1 )
      PCL_PRECONDITION( a >= 0 && a <= Const<float>::pi() )
      KernelFilter::Resize( n );
      m_shape = k;
      m_epsilon = Abs( e );
      m_sigma = (Size() >> 1)/Pow( -m_shape*Ln( m_epsilon ), 1/m_shape );
      m_rho = Range( r, 0.0F, 1.0F );
      m_theta = Range( a, 0.0F, Const<float>::pi() );
      Rebuild();
   }
 
   void Rebuild()
   {
      int size = Size();
      if ( size == 0 )
         return;
 
      float* h = *coefficients;
      float rk = m_shape * Pow( m_sigma, m_shape );
 
      for ( int n2 = size >> 1, y = -n2; y <= n2; ++y )
         if ( y > 0 )
            for ( int x = 0; x < size; ++x, ++h )
               *h = *(h - ((y+y)*size));
         else
            for ( int x = -n2; x <= n2; ++x, ++h )
               *h = (x > 0) ? *(h - (x+x)) : float( Exp( -Pow( Sqrt( float( x*x + y*y ) ), m_shape )/rk ) );
 
      if ( m_rho != 1 )
      {
         BicubicPixelInterpolation B;
 
         {
            coefficient_matrix ctemp( coefficients );
            ctemp.EnsureUnique(); // because we need invariant *ctemp for interpolation
            AutoPointer<PixelInterpolation::Interpolator<FloatPixelTraits> >
               interpolator( B.NewInterpolator<FloatPixelTraits>( *ctemp, size, size ) );
            h = *coefficients;
            float r = Max( 0.5F/size, m_rho );
            for ( int n2 = size >> 1, i = -n2; i <= n2; ++i )
            {
               float y = float( i )/r + n2;
               for ( int x = 0; x < size; ++x )
                  *h++ = (y < 0 || y >= size) ? 0.0F : (*interpolator)( float( x ), y );
            }
         }
 
         if ( m_theta != 0 )
         {
            coefficient_matrix ctemp( coefficients );
            ctemp.EnsureUnique(); // because we need invariant *ctemp for interpolation
            AutoPointer<PixelInterpolation::Interpolator<FloatPixelTraits> >
               interpolator( B.NewInterpolator<FloatPixelTraits>( *ctemp, size, size ) );
            h = *coefficients;
            float sa, ca;
            SinCos( -m_theta, sa, ca );
            for ( int n2 = size >> 1, i = -n2; i <= n2; ++i )
               for ( int j = -n2; j <= n2; ++j )
               {
                  float x =  ca*j + sa*i + n2;
                  float y = -sa*j + ca*i + n2;
                  *h++ = (x < 0 || y < 0 || x >= size || y >= size) ? 0.0F : (*interpolator)( x, y );
               }
         }
      }
   }
};
 
// ----------------------------------------------------------------------------
 
} // pcl
 
#endif   // __PCL_VariableShapeFilter_h
 
// ----------------------------------------------------------------------------
// EOF pcl/VariableShapeFilter.h - Released 2024-06-18T15:48:54Z