#pragma once
#include "../utils/common.h"

class Encoder {
public:
    Encoder(unsigned int multires, unsigned int chns) : _multires(multires), _chns(chns) { _genFreqArray(); }

    unsigned int outDim() const { return _chns * (1 + _multires * 2); }
    void encode(sptr<CudaArray<float>> output, sptr<CudaArray<float>> input);

private:
    unsigned int _multires;
    unsigned int _chns;
    sptr<CudaArray<float>> _freqs;

    void _genFreqArray();
};