---

packtree_base.cpp

Go to the documentation of this file.

#include "queue.h"
#include "packnode.h"
#include "packcontainer.h"
#include "packtree_base.h"


void packtree_base::newLevel( packnode<double>* top, 
                              bool freqCalc, 
                              bool reverse )
{
  if (top != 0) {
    const size_t len = top->length();
    if (len > 1) {
      // Create a new wavelet packet container for use in
      // calculating the wavelet transform.  Note that the
      // container is only used locally.
      packcontainer container( top );

      if (reverse) {
        // Calculate the reverse foward wavelet transform step, 
        // where the high pass result is stored in the upper half
        // of the container and the low pass result is stored
        // in the lower half of the container.
        waveObj->forwardStepRev( container, len );
      }
      else {
        // Calculate the foward wavelet transform step, where
        // the high pass result is stored in the upper half
        // of the container and the low pass result is stored
        // in the lower half of the container.
        waveObj->forwardStep( container, len );
      }

      packnode<double> *lhs = new packnode<double>(container.lhsData(), 
                                                   len/2,
                                                   packnode<double>::LowPass );
      packnode<double> *rhs = new packnode<double>(container.rhsData(), 
                                                   len/2,
                                                   packnode<double>::HighPass );

      // set the "mark" in the top node to false and
      // mark the two children to true.
      top->mark( false );
      lhs->mark( true );
      rhs->mark( true );

      top->lhsChild( lhs );
      top->rhsChild( rhs );

      // The transform on the left hand side always uses
      // the standard order (e.g., low pass filter result
      // goes in the lower half, high pass goes in the
      // upper half of the container).
      newLevel( lhs, freqCalc, false );

      if (freqCalc) {
        // wavelet packet frequency analysis reverses the
        // storage locations for the filter results in the
        // right hand child
        newLevel( rhs, freqCalc, true );
      }
      else { // freq == false
        // use standard filter location
        newLevel( rhs, freqCalc, false );
      }
    }
  }
} // newLevel



void packtree_base::breadthFirstPrint(const printKind kind)
{
  queue<double> Q;

  Q.addQueue( root, 0 );
  while (! Q.queueEmpty() ) {
    packnode<double> *node = Q.queueStart()->node;
    size_t indent = Q.queueStart()->indent;
    Q.deleteStart();
    
    if (indent > 0) {
      // print 'indent' spaces
      printf("%*c", indent, ' ');
    }

    switch (kind) {
    case printData: node->pr();
      break;
    case printCost: node->prCost();
      break;
    case printBestBasis: node->prBestBasis();
      break;
    default:
      assert( false );
      break;
    } // switch
    
    packnode<double> *lhs = node->lhsChild();
    packnode<double> *rhs = node->rhsChild();

    if (lhs != 0) {
      Q.addQueue( lhs, indent + 2 );
    }
    if (rhs != 0) {
      Q.addQueue( rhs, indent + 2 );
    }
  }
} // packtree_base::breadthFirstPrint




void packtree_base::pr()
{
  if (root != 0) {
    breadthFirstPrint(printData);
  }
} // pr

---