RangeComp.h

#pragma once
 
#include <complex>
#include <vector>
 
#include <isce3/fft/FFT.h>
 
namespace isce3 { namespace focus {

class RangeComp {
public:
    enum class Mode {
        Full,

        Valid,

        Same
    };

    RangeComp(const std::vector<std::complex<float>> & chirp,
              int inputsize,
              int maxbatch = 1,
              Mode mode = Mode::Full);

    int chirpSize() const { return _chirpsize; }

    int inputSize() const { return _inputsize; }

    int fftSize() const { return _fftsize; }

    int maxBatch() const { return _maxbatch; }

    Mode mode() const { return _mode; }

    int outputSize() const;

    int firstValidSample() const;

    void applyNotch(double frequency, double bandwidth = 0.0);

    const std::complex<float>* chirp_spectrum() const {
        return _reffn.data();
    }

    void rangecompress(std::complex<float> * out, const std::complex<float> * in, int batch = 1);
 
private:
    int _chirpsize;
    int _inputsize;
    int _fftsize;
    int _maxbatch;
    Mode _mode;
    std::vector<std::complex<float>> _reffn;
    std::vector<std::complex<float>> _wkspc;
    isce3::fft::FwdFFTPlan<float> _fftplan;
    isce3::fft::InvFFTPlan<float> _ifftplan;
};
 
}}