ComplexShortArray.h

Go to the documentation of this file.

#ifndef __ComplexShortArray_h__
#define __ComplexShortArray_h__
 
#include "ShortArray.h"
 
struct ComplexShort {
  int16_t re;
  
  int16_t im;
  
  int16_t getMagnitude();
  
  int16_t getPhase();
  
  void setPhase(int16_t phase){
    int16_t magnitude = getMagnitude();
    setPolar(magnitude, phase);
  }
  
  void setMagnitude(int16_t magnitude){
    int16_t phase = getPhase();
    setPolar(magnitude, phase);
  }
  
  void setPolar(int16_t magnitude, int16_t phase);
 
  bool operator==(const ComplexShort& other) const {
    return re == other.re && im == other.im;
  }
  
  bool operator!=(const ComplexShort& other) const {
    return re != other.re || im != other.im;
  }
};
 
class ComplexShortArray : public SimpleArray<ComplexShort> {
public:
  ComplexShortArray(){}
  ComplexShortArray(ComplexShort* data, size_t size) :
    SimpleArray(data, size) {}
 
  int16_t re(const int i){
    return data[i].re;
  }
  
  int16_t im(const int i){
    return data[i].im;
  }
  
  void clear(){
    setAll(0);
  }
 
  int16_t mag(const int i);
  
  void getMagnitudeValues(ShortArray destination);
  
  int16_t mag2(const int i);
  
  void getMagnitudeSquaredValues(ShortArray destination);
  
  void getComplexConjugateValues(ComplexShortArray destination);
  
  void complexDotProduct(ComplexShortArray operand2, ComplexShort& result);
 
  void complexByComplexMultiplication(ComplexShortArray operand2, ComplexShortArray result);
  
  void complexByRealMultiplication(ShortArray operand2, ComplexShortArray result);
 
  void add(ComplexShortArray operand2, ComplexShortArray destination);
 
  void add(ComplexShortArray operand2);
 
  void subtract(ComplexShortArray operand2, ComplexShortArray destination);
 
  void subtract(ComplexShortArray operand2);
  
  int16_t getMaxMagnitudeValue();
  
  unsigned int getMaxMagnitudeIndex();
  
  ComplexShortArray subArray(unsigned int offset, unsigned int length);
  
  void getRealValues(ShortArray buf);
  
  void getImaginaryValues(ShortArray buf);
  
  void scale(int16_t factor);
  
  static ComplexShortArray create(unsigned int size);
 
  static void destroy(ComplexShortArray);
 
  void setAll(ComplexShort value);
  
  void setAll(int16_t value);
  
  void setAll(int16_t valueRe, int16_t valueIm);
 
  void setPolar(ShortArray magnitude, ShortArray phase);
  
  void setPolar(ShortArray magnitude, ShortArray phase, int offset, int count);
 
  void setPhase(ShortArray phase);
 
  void setPhase(ShortArray phase, int offset, int count);
  
  void setPhase(ShortArray phase, ComplexShortArray destination);
  
  void setPhase(ShortArray phase, int offset, int count, ComplexShortArray destination);
 
 
  void setMagnitude(ShortArray magnitude);
 
  void setMagnitude(ShortArray magnitude, int offset, int count);
  
  void setMagnitude(ShortArray magnitude, ComplexShortArray destination);
  
  void setMagnitude(ShortArray magnitude, int offset, int count, ComplexShortArray destination);
};
#endif // __ComplexShortArray_h__