hfwt.cpp

Go to the documentation of this file.

// hfwt.cpp
//////////////////////////////////////////////////////////////////////
//          (c) Copyright 1994-2002 Brown Deer Technology, LLC.
//                        All rights reserved.
//////////////////////////////////////////////////////////////////////

#ifdef STORM_USEF
#include "hfwt_.h"
#include "dhfwt_.h"
#include "ihfwt_.h"
#include "hfwtws_.h"
#include "dhfwtws_.h"
#include "ihfwtws_.h"
#endif

#include "stormdef.h"
#define INVSQRT2 STORMDEF_MATHCONST_INVSQRT2

int hfwt( 
          const int& Sizef,
          const int& ns,
          const int& size,
          const float* sdata,
                float* wdata
) { 

#ifdef STORM_USEF

   int iflag;
   hfwt_( &Sizef, &ns, &size, sdata, wdata, &iflag );
   return iflag;

#else

   int j,size2;
   int size1 = size;
   float *xx = new float[Sizef];

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 < ns ) return 1;
#endif

   for( int i = 0; i < size1; i++ ) { wdata[i] = sdata[i]; }
   if ( size1 == ns ) return 0;
   while ( size1 > ns ) {
      size2 = size1 / 2;
      for( int i = 0; i < size2; i++ ) {
         j = 2 * i;
         xx[i] = ( wdata[j] + wdata[j+1] ) * INVSQRT2;
         xx[size2+i] = ( wdata[j] - wdata[j+1] ) * INVSQRT2;
      }
      for( int i = 0; i < size1; i++ ) { wdata[i] = xx[i]; }
      size1 = size2;
   }
   delete [] xx;

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 != ns ) return 2;
#endif

   return 0;

#endif

} 


int hfwt( 
          const int& Sizef,
          const int& ns,
          const int& size,
          const double* sdata,
                double* wdata
) { 

#ifdef STORM_USEF

   int iflag;
   dhfwt_( &Sizef, &ns, &size, sdata, wdata, &iflag );
   return iflag;

#else

   int j,size2;
   int size1 = size;
   double *xx = new double[Sizef];

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 < ns ) return 1;
#endif

   for( int i = 0; i < size1; i++ ) { wdata[i] = sdata[i]; }
   if ( size1 == ns ) return 0;
   while ( size1 > ns ) {
      size2 = size1 / 2;
      for( int i = 0; i < size2; i++ ) {
         j = 2 * i;
         xx[i] = ( wdata[j] + wdata[j+1] ) * INVSQRT2;
         xx[size2+i] = ( wdata[j] - wdata[j+1] ) * INVSQRT2;
      }
      for( int i = 0; i < size1; i++ ) { wdata[i] = xx[i]; }
      size1 = size2;
   }
   delete [] xx;

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 != ns ) return 2;
#endif

   return 0;

#endif

} 


int hfwt( 
          const int& Sizef,
          const int& ns,
          const int& size,
          const int* sdata,
                int* wdata
) { 

#ifdef STORM_USEF

   int iflag;
   ihfwt_( &Sizef, &ns, &size, sdata, wdata, &iflag );
   return iflag;

#else

   int j,size2;
   int size1 = size;
   int *xx = new int[Sizef];

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 < ns ) return 1;
#endif

   for( int i = 0; i < size1; i++ ) { wdata[i] = sdata[i]; }
   if ( size1 == ns ) return 0;
   while ( size1 > ns ) {
      size2 = size1 / 2;
      for( int i = 0; i < size2; i++ ) {
         j = 2 * i;
         xx[i] = ( wdata[j] + wdata[j+1] ) / 2;
         xx[size2+i] = wdata[j] - wdata[j+1];
      }
      for( int i = 0; i < size1; i++ ) { wdata[i] = xx[i]; }
      size1 = size2;
   }
   delete [] xx;

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 != ns ) return 2;
#endif

   return 0;

#endif

} 


int hfwt( 
          const int& Sizef,
          const int& ns,
          const int& size,
          const float* sdata,
                float* wdata,
                float* ws
) { 

#ifdef STORM_USEF

   int iflag;
   hfwtws_( &Sizef, &ns, &size, sdata, wdata, ws, &iflag );
   return iflag;

#else

   int j,size2;
   int size1 = size;
   if ( size1 < ns ) return 1;
   for( int i = 0; i < size1; i++ ) { wdata[i] = sdata[i]; }

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 == ns ) return 0;
#endif

   while ( size1 > ns ) {
      size2 = size1 / 2;
      for( int i = 0; i < size2; i++ ) {
         j = 2 * i;
         ws[i] = ( wdata[j] + wdata[j+1] ) * INVSQRT2;
         ws[size2+i] = ( wdata[j] - wdata[j+1] ) * INVSQRT2;
      }
      for( int i = 0; i < size1; i++ ) { wdata[i] = ws[i]; }
      size1 = size2;
   }

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 != ns ) return 2;
#endif

   return 0;

#endif

} 


int hfwt( 
          const int& Sizef,
          const int& ns,
          const int& size,
          const double* sdata,
                double* wdata,
                double* ws
) { 

#ifdef STORM_USEF

   int iflag;
   dhfwtws_( &Sizef, &ns, &size, sdata, wdata, ws, &iflag );
   return iflag;

#else

   int j,size2;
   int size1 = size;

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 < ns ) return 1;
#endif

   for( int i = 0; i < size1; i++ ) { wdata[i] = sdata[i]; }
   if ( size1 == ns ) return 0;
   while ( size1 > ns ) {
      size2 = size1 / 2;
      for( int i = 0; i < size2; i++ ) {
         j = 2 * i;
         ws[i] = ( wdata[j] + wdata[j+1] ) * INVSQRT2;
         ws[size2+i] = ( wdata[j] - wdata[j+1] ) * INVSQRT2;
      }
      for( int i = 0; i < size1; i++ ) { wdata[i] = ws[i]; }
      size1 = size2;
   }

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 != ns ) return 2;
#endif

   return 0;

#endif

} 


int hfwt( 
          const int& Sizef,
          const int& ns,
          const int& size,
          const int* sdata,
                int* wdata,
                int* ws
) { 

#ifdef STORM_USEF

   int iflag;
   ihfwtws_( &Sizef, &ns, &size, sdata, wdata, ws, &iflag );
   return iflag;

#else

   int j,size2;
   int size1 = size;

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 < ns ) return 1;
#endif

   for( int i = 0; i < size1; i++ ) { wdata[i] = sdata[i]; }
   if ( size1 == ns ) return 0;
   while ( size1 > ns ) {
      size2 = size1 / 2;
      for( int i = 0; i < size2; i++ ) {
         j = 2 * i;
         ws[i] = ( wdata[j] + wdata[j+1] ) / 2;
         ws[size2+i] = wdata[j] - wdata[j+1];
      }
      for( int i = 0; i < size1; i++ ) { wdata[i] = ws[i]; }
      size1 = size2;
   }

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 != ns ) return 2;
#endif

   return 0;

#endif

} 

#undef INVSQRT2

---