hifwt2ds.cpp

Go to the documentation of this file.

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

#ifdef STORM_USEF
#include "hifwt2ds_.h"
#include "dhifwt2ds_.h"
#include "ihifwt2ds_.h"
#include "hifwt2dsws_.h"
#include "dhifwt2dsws_.h"
#include "ihifwt2dsws_.h"
#endif

#include "stormdef.h"
#define INVSQRT2 STORMDEF_MATHCONST_INVSQRT2

int hifwt2ds( 
          const int& Sizef2d,
          const int& ns,
          const int& size,
          const float* wdata,
                float* sdata
) { 

#ifdef STORM_USEF

   int iflag;
   hifwt2ds_( &Sizef2d, &ns, &size, wdata, sdata, &iflag );
   return iflag;

#else

   int ii,size2;
   int size1 = ns;
   float *u = new float[Sizef2d];
   float *v = new float[Sizef2d];

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

   for( int i = 0; i < size*size; i++ ) {
      sdata[i] = wdata[i];
   }

   if ( size == ns ) return 0;

   while ( size1 < size ) {

      size2 = size1;
      size1 = 2 * size1;

      for( int j = 0; j < size1; j++ ) {        // index 1
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            u[i] = sdata[ii+j];
         }
         for( int i = 0; i < size2; i++ ) {
            v[2*i] = ( u[i] + u[size2+i] ) * INVSQRT2;
            v[2*i+1] = ( u[i] - u[size2+i] ) * INVSQRT2;
         }
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            sdata[ii+j] = v[i];
         }  
      }

      for( int i = 0; i < size1; i++ ) {        // index 2
         ii = i * size;
         for( int j = 0; j < size1; j++ ) {
            u[j] = sdata[ii+j];
         }
            for( int j = 0; j < size2; j++ ) {
               v[2*j] = ( u[j] + u[size2+j] ) * INVSQRT2;
               v[2*j+1] = ( u[j] - u[size2+j] ) * INVSQRT2;
            }
         for( int j = 0; j < size1; j++ ) {
            sdata[ii+j] = v[j];
         }
      }

   }
   
   delete [] u;
   delete [] v;

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 > size ) return 2;
#endif

   return 0;

#endif
} 


int hifwt2ds( 
          const int& Sizef2d,
          const int& ns,
          const int& size,
          const double* wdata,
                double* sdata
) { 

#ifdef STORM_USEF

   int iflag;
   dhifwt2ds_( &Sizef2d, &ns, &size, wdata, sdata, &iflag );
   return iflag;

#else

   int ii,size2;
   int size1 = ns;
   double *u = new double[Sizef2d];
   double *v = new double[Sizef2d];

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

   for( int i = 0; i < size*size; i++ ) {
      sdata[i] = wdata[i];
   }

   if ( size == ns ) return 0;

   while ( size1 < size ) {

      size2 = size1;
      size1 = 2 * size1;

      for( int j = 0; j < size1; j++ ) {        // index 1
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            u[i] = sdata[ii+j];
         }
         for( int i = 0; i < size2; i++ ) {
            v[2*i] = ( u[i] + u[size2+i] ) * INVSQRT2;
            v[2*i+1] = ( u[i] - u[size2+i] ) * INVSQRT2;
         }
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            sdata[ii+j] = v[i];
         }
      }

      for( int i = 0; i < size1; i++ ) {        // index 2
         ii = i * size;
         for( int j = 0; j < size1; j++ ) {
            u[j] = sdata[ii+j];
         }
         for( int j = 0; j < size2; j++ ) {
            v[2*j] = ( u[j] + u[size2+j] ) * INVSQRT2;
            v[2*j+1] = ( u[j] - u[size2+j] ) * INVSQRT2;
         }
         for( int j = 0; j < size1; j++ ) {
            sdata[ii+j] = v[j];
         }
      }

   }

   delete [] u;
   delete [] v;

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 > size ) return 2;
#endif

   return 0;

#endif
} 


int hifwt2ds( 
          const int& Sizef2d,
          const int& ns,
          const int& size,
          const int* wdata,
                int* sdata
) { 

#ifdef STORM_USEF

   int iflag;
   ihifwt2ds_( &Sizef2d, &ns, &size, wdata, sdata, &iflag );
   return iflag;

#else

   int ii,size2;
   int size1 = ns;
   int *u = new int[Sizef2d];
   int *v = new int[Sizef2d];

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

   for( int i = 0; i < size*size; i++ ) {
      sdata[i] = wdata[i];
   }

   if ( size == ns ) return 0;

   while ( size1 < size ) {

      size2 = size1;
      size1 = 2 * size1;

      for( int j = 0; j < size1; j++ ) {        // index 1
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            u[i] = sdata[ii+j];
         }
         for( int i = 0; i < size2; i++ ) {
            v[2*i] = ( u[i] + u[size2+i] ) / 2;
            v[2*i+1] = u[i] - u[size2+i];
         }
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            sdata[ii+j] = v[i];
         }  
      }

      for( int i = 0; i < size1; i++ ) {        // index 2
         ii = i * size;
         for( int j = 0; j < size1; j++ ) {
            u[j] = sdata[ii+j];
         }
            for( int j = 0; j < size2; j++ ) {
               v[2*j] = ( u[j] + u[size2+j] ) / 2;
               v[2*j+1] = u[j] - u[size2+j];
            }
         for( int j = 0; j < size1; j++ ) {
            sdata[ii+j] = v[j];
         }
      }

   }
   
   delete [] u;
   delete [] v;

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 > size ) return 2;
#endif

   return 0;

#endif
} 


int hifwt2ds( 
          const int& Sizef2d,
          const int& ns,
          const int& size,
          const float* wdata,
                float* sdata,
                float* ws
) { 

#ifdef STORM_USEF

   int iflag;
   hifwt2dsws_( &Sizef2d, &ns, &size, wdata, sdata, ws, &iflag );
   return iflag;

#else

   int ii,size2;
   int size1 = ns;
   float *u = ws;
   float *v = &ws[Sizef2d];

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

   for( int i = 0; i < size*size; i++ ) {
      sdata[i] = wdata[i];
   }

   if ( size == ns ) return 0;

   while ( size1 < size ) {

      size2 = size1;
      size1 = 2 * size1;

      for( int j = 0; j < size1; j++ ) {        // index 1
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            u[i] = sdata[ii+j];
         }
         for( int i = 0; i < size2; i++ ) {
            v[2*i] = ( u[i] + u[size2+i] ) * INVSQRT2;
            v[2*i+1] = ( u[i] - u[size2+i] ) * INVSQRT2;
         }
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            sdata[ii+j] = v[i];
         }  
      }

      for( int i = 0; i < size1; i++ ) {        // index 2
         ii = i * size;
         for( int j = 0; j < size1; j++ ) {
            u[j] = sdata[ii+j];
         }
         for( int j = 0; j < size2; j++ ) {
            v[2*j] = ( u[j] + u[size2+j] ) * INVSQRT2;
            v[2*j+1] = ( u[j] - u[size2+j] ) * INVSQRT2;
         }
         for( int j = 0; j < size1; j++ ) {
            sdata[ii+j] = v[j];
         }
      }

   }
   
#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 > size ) return 2;
#endif

   return 0;

#endif
} 


int hifwt2ds( 
          const int& Sizef2d,
          const int& ns,
          const int& size,
          const double* wdata,
                double* sdata,
                double* ws
) { 

#ifdef STORM_USEF

   int iflag;
   dhifwt2dsws_( &Sizef2d, &ns, &size, wdata, sdata, ws, &iflag );
   return iflag;

#else

   int ii,size2;
   int size1 = ns;
   double *u = ws;
   double *v = &ws[Sizef2d];

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

   for( int i = 0; i < size*size; i++ ) {
      sdata[i] = wdata[i];
   }

   if ( size == ns ) return 0;

   while ( size1 < size ) {

      size2 = size1;
      size1 = 2 * size1;

      for( int j = 0; j < size1; j++ ) {        // index 1
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            u[i] = sdata[ii+j];
         }
         for( int i = 0; i < size2; i++ ) {
            v[2*i] = ( u[i] + u[size2+i] ) * INVSQRT2;
            v[2*i+1] = ( u[i] - u[size2+i] ) * INVSQRT2;
         }
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            sdata[ii+j] = v[i];
         }
      }

      for( int i = 0; i < size1; i++ ) {        // index 2
         ii = i * size;
         for( int j = 0; j < size1; j++ ) {
            u[j] = sdata[ii+j];
         }
         for( int j = 0; j < size2; j++ ) {
            v[2*j] = ( u[j] + u[size2+j] ) * INVSQRT2;
            v[2*j+1] = ( u[j] - u[size2+j] ) * INVSQRT2;
         }
         for( int j = 0; j < size1; j++ ) {
            sdata[ii+j] = v[j];
         }
      }

   }

#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 > size ) return 2;
#endif

   return 0;

#endif
} 


int hifwt2ds( 
          const int& Sizef2d,
          const int& ns,
          const int& size,
          const int* wdata,
                int* sdata,
                int* ws
) { 

#ifdef STORM_USEF

   int iflag;
   ihifwt2dsws_( &Sizef2d, &ns, &size, wdata, sdata, ws, &iflag );
   return iflag;

#else

   int ii,size2;
   int size1 = ns;
   int *u = ws;
   int *v = &ws[Sizef2d];

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

   for( int i = 0; i < size*size; i++ ) {
      sdata[i] = wdata[i];
   }

   if ( size == ns ) return 0;

   while ( size1 < size ) {

      size2 = size1;
      size1 = 2 * size1;

      for( int j = 0; j < size1; j++ ) {        // index 1
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            u[i] = sdata[ii+j];
         }
         for( int i = 0; i < size2; i++ ) {
            v[2*i] = ( u[i] + u[size2+i] ) / 2;
            v[2*i+1] = u[i] - u[size2+i];
         }
         for( int i = 0; i < size1; i++ ) {
            ii = i * size;
            sdata[ii+j] = v[i];
         }  
      }

      for( int i = 0; i < size1; i++ ) {        // index 2
         ii = i * size;
         for( int j = 0; j < size1; j++ ) {
            u[j] = sdata[ii+j];
         }
         for( int j = 0; j < size2; j++ ) {
            v[2*j] = ( u[j] + u[size2+j] ) / 2;
            v[2*j+1] = u[j] - u[size2+j];
         }
         for( int j = 0; j < size1; j++ ) {
            sdata[ii+j] = v[j];
         }
      }

   }
  
#ifdef STORM_FWT_CCHECKUSAGE
   if ( size1 > size ) return 2;
#endif

   return 0;

#endif
} 

#undef INVSQRT2

---