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

#ifndef __ComplexFourierTransform_h__
#define __ComplexFourierTransform_h__

#include "ComplexFloatArray.h"

#ifdef ARM_CORTEX
#include "arm_const_structs.h"

class ComplexFourierTransform {
private:
  arm_cfft_instance_f32 instance;
public:
  ComplexFourierTransform(){};
  ComplexFourierTransform(int len){
    init(len);
  }
  void init(int len){
    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_CFFT_INIT_F32, args, 2);
    // Supported FFT Lengths are 32, 64, 128, 256, 512, 1024, 2048, 4096.
  }
  void fft(ComplexFloatArray inout){
    ASSERT(inout.getSize() >= getSize(), "Input array too small");
    arm_cfft_f32(&instance, (float*)inout.getData(), 0, 1); //forward
  }
  void ifft(ComplexFloatArray inout){
    ASSERT(inout.getSize() >= getSize(), "Input array too small");
   arm_cfft_f32(&instance, (float*)inout.getData(), 1, 1); //inverse
  }
  int getSize(){
    return instance.fftLen;
  }
};
#endif /* ARM_CORTEX */

#ifndef ARM_CORTEX
#include "kiss_fft.h"
class ComplexFourierTransform {
private:
  kiss_fft_cfg cfgfft;
  kiss_fft_cfg cfgifft;
  ComplexFloatArray temp;
public:
  ComplexFourierTransform(){}
  ComplexFourierTransform(int len){
    init(len);
  }
  ~ComplexFourierTransform(){
    ComplexFloatArray::destroy(temp);
  }
  void init(int len){
    ASSERT(len==32 || len ==64 || len==128 || len==256 || len==512 || len==1024 || len==2048 || len==4096, "Unsupported FFT size");
    cfgfft = kiss_fft_alloc(len, 0 , 0, 0);
    cfgifft = kiss_fft_alloc(len, 1,0, 0);
    temp = ComplexFloatArray::create(len);
  }
  void fft(ComplexFloatArray inout){
    ASSERT(inout.getSize() >= getSize(), "Input array too small");
    kiss_fft(cfgfft, (kiss_fft_cpx*)(float*)inout.getData(), (kiss_fft_cpx*)(float*)temp.getData());
    inout.copyFrom(temp);
  }
  void ifft(ComplexFloatArray inout){
    ASSERT(inout.getSize() >= getSize(), "Input array too small");
    kiss_fft(cfgifft, (kiss_fft_cpx*)(float*)inout.getData(), (kiss_fft_cpx*)(float*)temp.getData());
    temp.scale(1.0f/getSize());
    inout.copyFrom(temp);
  }
  size_t getSize(){
    return temp.getSize();
  }
};
#endif /* ifndef ARM_CORTEX */

#endif // __ComplexFourierTransform_h__