ComplexFloatArray.cpp

Go to the documentation of this file.

#include "ComplexFloatArray.h"
#include "basicmaths.h"
#include "message.h"
 
float ComplexFloatArray::mag(const int i){
  float result;
#ifdef ARM_CORTEX
  arm_cmplx_mag_f32((float*)&(data[i]), &result,1);
#else
  result=sqrtf(mag2(i));
#endif
  return result;
}
 
void ComplexFloatArray::getMagnitudeValues(FloatArray destination){
#ifdef ARM_CORTEX
  arm_cmplx_mag_f32((float*)data, (float*)destination, size);
#else
  for(size_t i=0; i<size; i++){
    destination[i]=mag(i);
  }
#endif
}
 
float ComplexFloatArray::mag2(const int i){
  float result;
#ifdef ARM_CORTEX
  arm_cmplx_mag_squared_f32((float*)&(data[i]), &result, 1);
#else
  float re=data[i].re;
  float im=data[i].im;
  result=re*re+im*im;
#endif  
  return result;
}
 
void ComplexFloatArray::getMagnitudeSquaredValues(FloatArray destination){
  ASSERT(destination.getSize()>=size, "Wrong array size");
#ifdef ARM_CORTEX
  arm_cmplx_mag_squared_f32((float*)data, (float*)destination, size);
#else
  for(size_t i=0; i<size; i++){
    destination[i]=mag2(i);
  }
#endif  
}
 
void ComplexFloatArray::getPolar(FloatArray magnitude, FloatArray phase){
  ASSERT(magnitude.getSize()>=size, "Wrong array size");
  ASSERT(phase.getSize()>=size, "Wrong array size");
  for(size_t i=0; i<size; i++){
    magnitude[i] = data[i].getMagnitude();
    phase[i] = data[i].getPhase();
  }
}
 
void ComplexFloatArray::getPhaseValues(FloatArray destination){
  ASSERT(destination.getSize()>=size, "Wrong array size");
  for(size_t i=0; i<size; i++)
    destination[i] = data[i].getPhase();
}
 
void ComplexFloatArray::complexDotProduct(ComplexFloatArray operand2, ComplexFloat& result){
#ifdef ARM_CORTEX
  arm_cmplx_dot_prod_f32 ( (float*)getData(), (float*)operand2.getData(), size, &(result.re), &(result.im) );
#else
  float *pSrcA=(float*)getData();
  float *pSrcB=(float*)operand2.getData();
  float realResult=0;    
  float imagResult=0;    
  for(size_t n=0; n<size; n++) {    
      realResult += pSrcA[(2*n)+0]*pSrcB[(2*n)+0] - pSrcA[(2*n)+1]*pSrcB[(2*n)+1];    
      imagResult += pSrcA[(2*n)+0]*pSrcB[(2*n)+1] + pSrcA[(2*n)+1]*pSrcB[(2*n)+0];    
  }
  result.re=realResult;
  result.im=imagResult;
#endif  
}
 
void ComplexFloatArray::complexByComplexMultiplication(ComplexFloatArray operand2, ComplexFloatArray result){
  ASSERT(operand2.size == size && result.size >= size, "Arrays size mismatch");
#ifdef ARM_CORTEX
  arm_cmplx_mult_cmplx_f32( (float*)getData(), (float*)operand2.getData(), (float*)result.getData(), size );  
#else
  float *pSrcA=(float*)getData();
  float *pSrcB=(float*)operand2.getData();
  float *pDst=(float*)result.getData();
  for(size_t n=0; n<size; n++) {
    pDst[(2*n)+0] = pSrcA[(2*n)+0] * pSrcB[(2*n)+0] - pSrcA[(2*n)+1] * pSrcB[(2*n)+1];        
    pDst[(2*n)+1] = pSrcA[(2*n)+0] * pSrcB[(2*n)+1] + pSrcA[(2*n)+1] * pSrcB[(2*n)+0];        
  }        
#endif  
}
 
void ComplexFloatArray::add(ComplexFloatArray operand2, ComplexFloatArray destination){
  ASSERT(operand2.size == size && destination.size >= size, "Arrays size mismatch");
#ifdef ARM_CORTEX
  arm_add_f32((float*)getData(), (float*)operand2.getData(), (float*)destination.getData(), size*2);
#else
  for(size_t n=0; n<size; n++){
    destination[n].re = data[n].re + operand2[n].re;
    destination[n].im = data[n].im + operand2[n].im;
  }
#endif /* ARM_CORTEX */  
}
 
void ComplexFloatArray::add(ComplexFloatArray operand2){
  add(operand2, *this);
}
 
void ComplexFloatArray::subtract(ComplexFloatArray operand2, ComplexFloatArray destination){
  ASSERT(operand2.size == size && destination.size >= size, "Arrays size mismatch");
#ifdef ARM_CORTEX
  arm_sub_f32((float*)data, (float*)operand2.data, (float*)destination.getData(), size*2);
#else
  for(size_t n=0; n<size; n++){
    destination[n].re = data[n].re - operand2[n].re;
    destination[n].im = data[n].im - operand2[n].im;
  }
#endif /* ARM_CORTEX */  
}
 
void ComplexFloatArray::subtract(ComplexFloatArray operand2){
  subtract(operand2, *this);
}
 
void ComplexFloatArray::getComplexConjugateValues(ComplexFloatArray destination){
  ASSERT(size==destination.getSize(), "Wrong array size");
#ifdef ARM_CORTEX
  arm_cmplx_conj_f32( (float*)getData(), (float*)destination.getData(), size );  
#else
  float *pSrc=(float*)getData();
  float *pDst=(float *)destination.getData();
  for(size_t n=0; n<size; n++) {        
    pDst[(2*n)+0] = pSrc[(2*n)+0];     // real part        
    pDst[(2*n)+1] = -pSrc[(2*n)+1];    // imag part        
}   
#endif  
}
 
void ComplexFloatArray::complexByRealMultiplication(FloatArray operand2, ComplexFloatArray result){
  ASSERT(size==operand2.getSize(), "Wrong size"); 
#ifdef ARM_CORTEX
  arm_cmplx_mult_real_f32 ( (float*)getData(), (float*)operand2.getData(), (float*)result.getData(), size );  
#else
  float *pSrcCmplx=(float*)getData();
  float *pSrcReal=(float*)operand2.getData();
  float *pCmplxDst=(float*)result.getData();
  for(size_t n=0; n<size; n++) {
      pCmplxDst[(2*n)+0] = pSrcCmplx[(2*n)+0] * pSrcReal[n];        
      pCmplxDst[(2*n)+1] = pSrcCmplx[(2*n)+1] * pSrcReal[n];        
  }        
#endif
}
 
int ComplexFloatArray::getMaxMagnitudeIndex(){ //this is probably slower than getMagnitudeSquaredValues() and getMaxIndex() on it
  float maxMag=-1;
  int maxInd=-1;
  for(size_t n=0; n<size; n++){
    float magnitude=mag2(n); //uses mag2 which is faster
    if(magnitude>maxMag){
      maxMag=magnitude;
      maxInd=n;
    }
  }
  return maxInd;
}
 
ComplexFloatArray ComplexFloatArray::subArray(int offset, size_t length){
  ASSERT(size >= offset+length, "Array too small");
  return ComplexFloatArray(data+offset, length);
}
 
float ComplexFloatArray::getMaxMagnitudeValue(){ //this is probably slower than getMagnitudeSquaredValues() and getMaxValue() on it
  float maxMag=-1;
  for(size_t n=0; n<size; n++){
    float mag=this->mag2(n);
    if(mag>maxMag){
      maxMag=mag;
    }
  }
  maxMag=sqrtf(maxMag);
  return maxMag;
}
 
void ComplexFloatArray::getRealValues(FloatArray buf){
  for(size_t n=0; n<size; n++){
    buf[n]=data[n].re;
  }
}
 
void ComplexFloatArray::getImaginaryValues(FloatArray buf){
  for(size_t n=0; n<size; n++){
    buf[n]=data[n].im;
  }
}
 
void ComplexFloatArray::scale(float factor){
#ifdef ARM_CORTEX  
  arm_scale_f32((float*)data, factor, (float*)data, size*2); //the *2 accounts for the fact that both real and imaginary parts are scaled
#else
  for(size_t n=0; n<size; n++){
    data[n].re *= factor;
    data[n].im *= factor;
  }
#endif
}
 
void ComplexFloatArray::fromFloat(FloatArray source){
  ASSERT(source.getSize() >= size, "Array too small");
  for(size_t n=0; n<size; n++){
    data[n].re = source[n];
    data[n].im = 0;
  }
}
 
/* Copies real part to a FloatArray */
void ComplexFloatArray::toFloat(FloatArray destination){
  ASSERT(destination.getSize() >= size, "Array too small");
  for(size_t n=0; n<size; n++){
    destination[n] = data[n].re;
  }
}
 
 
void ComplexFloatArray::setAll(float value){
#ifdef ARM_CORTEX
  arm_fill_f32(value, (float *)data, size *2 ); //note the *2 multiplier which accounts for real and imaginary parts
#else
  ComplexFloat val;
  val.re=value;
  val.im=value;
  setAll(val);
#endif /* ARM_CORTEX */
}
 
void ComplexFloatArray::setAll(ComplexFloat value){
  for(size_t n=0; n<size; n++){
    data[n].re=value.re;
    data[n].im=value.im;
  }
}
 
void ComplexFloatArray::setAll(float valueRe, float valueIm){
  ComplexFloat value;
  value.re=valueRe;
  value.im=valueIm;
  setAll(value);
}
 
void ComplexFloatArray::setPolar(FloatArray magnitude, FloatArray phase){
  setPolar(magnitude, phase, 0, size);
}
void ComplexFloatArray::setPolar(FloatArray magnitude, FloatArray phase, int offset, size_t count){
  for(size_t n=offset; n<count+offset; n++){
    data[n].setPolar(magnitude[n], phase[n]);
  }
}
 
void ComplexFloatArray::setPhase(FloatArray phase){
  setPhase(phase, 0, size);
}
void ComplexFloatArray::setPhase(FloatArray phase, int offset, size_t count){
  for(size_t n=offset; n<count+offset; n++){
    data[n].setPhase(phase[n]);
  }
}
void ComplexFloatArray::setPhase(FloatArray phase, ComplexFloatArray destination){
  setPhase(phase, 0, size, destination);
}
void ComplexFloatArray::setPhase(FloatArray phase, int offset, size_t count, ComplexFloatArray destination){
  ASSERT(destination.getSize()>=count+offset, "Wrong size");
  for(size_t n=offset; n<count+offset; n++){
    destination.getData()[n].setPolar(getData()[n].getMagnitude(), phase[n]);
  }
}
void ComplexFloatArray::setMagnitude(FloatArray magnitude){
  setMagnitude(magnitude, 0, size);
}
void ComplexFloatArray::setMagnitude(FloatArray magnitude, int offset, size_t count){
  setMagnitude(magnitude, offset, count, *this);
}
void ComplexFloatArray::setMagnitude(FloatArray magnitude, ComplexFloatArray destination){
  setMagnitude(magnitude, 0, size, destination);
}
void ComplexFloatArray::setMagnitude(FloatArray magnitude, int offset, size_t count, ComplexFloatArray destination){
  ASSERT(getSize()==magnitude.getSize(),"wrong size0");
  ASSERT(getSize()==destination.getSize(),"wrong size1");
  ASSERT(offset+count<=destination.getSize(), "Wrong size2");
  ASSERT(offset+count<=getSize(), "Wrong size3");
  for(size_t n=offset; n<count+offset; n++){
    destination.getData()[n].setPolar(magnitude[n], getData()[n].getPhase());
  }
}
 
ComplexFloatArray ComplexFloatArray::create(size_t size){
  ComplexFloatArray obj(new ComplexFloat[size], size);
  obj.clear();
  return obj;
}
 
void ComplexFloatArray::destroy(ComplexFloatArray array){
  delete[] array.data;
}
 
void ComplexFloatArray::copyFrom(FloatArray real, FloatArray imag) {
  for (size_t i = 0; i < getSize(); i++) {
    data[i] = ComplexFloat(real[i], imag[i]);
  }
}
 
void ComplexFloatArray::copyTo(FloatArray real, FloatArray imag) {
  for (size_t i = 0; i < getSize(); i++) {
    real[i] = data[i].re;
    imag[i] = data[i].im;
  }
}