owl_patch 0.8.0

Rust SDK for Rebel Technology Owl2/3 devices
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

#include "ShortFastFourierTransform.h"
#include "message.h"
#include "ProgramVector.h"
#include "ServiceCall.h"

#ifdef ARM_CORTEX
ShortFastFourierTransform::ShortFastFourierTransform(){}

ShortFastFourierTransform::ShortFastFourierTransform(int len){
  init(len);
}

ShortFastFourierTransform::~ShortFastFourierTransform(){}

void ShortFastFourierTransform::init(int aSize){
  len = aSize;
  ASSERT(len==32 || len ==64 || len==128 || len==256 || len==512 || len==1024 || len==2048 || len==4096, "Unsupported FFT size");
  void* args[] = {(void*)&instance, (void*)&len};
  // getProgramVector()->serviceCall(OWL_SERVICE_ARM_RFFT_FAST_INIT_Q15, args, 2);
  // Supported FFT Lengths are 32, 64, 128, 256, 512, 1024, 2048, 4096.
}

void ShortFastFourierTransform::fft(ShortArray in, ComplexShortArray out){
  ASSERT(in.getSize() >= getSize(), "Input array too small");
  ASSERT(out.getSize() >= getSize(), "Output array too small");
  arm_rfft_init_q15(&instance, len, 0, 1);
  arm_rfft_q15(&instance, (int16_t*)in.getData(), (int16_t*)out.getData());
}

void ShortFastFourierTransform::ifft(ComplexShortArray in, ShortArray out){
  ASSERT(in.getSize() >= getSize(), "Input array too small");
  ASSERT(out.getSize() >= getSize(), "Output array too small");
  arm_rfft_init_q15(&instance, len, 1, 1);
  arm_rfft_q15(&instance, (int16_t*)in.getData(), (int16_t*)out.getData());
}

size_t ShortFastFourierTransform::getSize(){
  return len;
}

#else /* ARM_CORTEX */

ShortFastFourierTransform::ShortFastFourierTransform(){}

ShortFastFourierTransform::ShortFastFourierTransform(int aSize){
  init(aSize);
}

ShortFastFourierTransform::~ShortFastFourierTransform(){
  ASSERT(false, "TODO");
  ComplexShortArray::destroy(temp);
}

void ShortFastFourierTransform::init(int aSize){
  ASSERT(aSize==32 || aSize ==64 || aSize==128 || aSize==256 || aSize==512 || aSize==1024 || aSize==2048 || aSize==4096, "Unsupported FFT size");
  cfgfft = kiss_fft_alloc(aSize, 0 , 0, 0);
  cfgifft = kiss_fft_alloc(aSize, 1,0, 0);
  temp = ComplexShortArray::create(getSize());
}

void ShortFastFourierTransform::fft(ShortArray input, ComplexShortArray output){
  ASSERT(input.getSize() >= getSize(), "Input array too small");
  ASSERT(output.getSize() >= getSize(), "Output array too small");
  for(size_t n=0; n<getSize(); n++){
    temp[n].re=input[n];
    temp[n].im=0;
  }
  kiss_fft(cfgfft, (kiss_fft_cpx*)(int16_t*)temp.getData(), (kiss_fft_cpx*)(int16_t*)output.getData());
}
  
void ShortFastFourierTransform::ifft(ComplexShortArray input, ShortArray output){
  ASSERT(input.getSize() >= getSize(), "Input array too small");
  ASSERT(output.getSize() >= getSize(), "Output array too small");
  kiss_fft(cfgifft, (kiss_fft_cpx*)(int16_t*)input.getData(), (kiss_fft_cpx*)(int16_t*)temp.getData());
  float scale=1.0f/getSize();
  for(size_t n=0; n<getSize(); n++){
    output[n]=temp[n].re*scale;
  }
}
    
size_t ShortFastFourierTransform::getSize(){
  return temp.getSize();
}

#endif /* ifndef ARM_CORTEX */