QuaternionFT.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2005 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/Util/fft/QuaternionFT.h
 
#ifndef __QuaternionFT_H
#define __QuaternionFT_H
 
#include "Quaternion.h"
#include "Matrix.h"
#include <vector>
 
class QuaternionFT {
 
public:
 
  enum Direction { Forward, Backward };

  static bool verbose;

  QuaternionFT ();

  QuaternionFT (double theta, double phi);

  void set_axis (double theta, double phi);

  void set_axis (Vector<3,float> axis);

  void set_power (double power);
  void set_extent (double extent);

  Vector<3,float> get_axis () const;

  void set_direction (Direction direction);

  void fft (const std::vector<Quaternion<float> >& in,
            std::vector<Quaternion<float> >& out);
 
  void fft (unsigned npt, const Quaternion<float>* in, Quaternion<float>* out);
 
  void fft (unsigned npt, const float* input, float* output);
 

  void ft (const std::vector<Quaternion<float> >& in,
           std::vector<Quaternion<float> >& out);
 
  void ft (unsigned npt, const Quaternion<float>* in, Quaternion<float>* out);
 
  void ft (unsigned npt, const float* input, float* output);
 
protected:

  Quaternion<float> mu[3];

  Matrix<3,3,float> xform;

  float sign;

  void set_xform ();

  double power, extent;
 
};
 
#endif