CodecMP3Helix.h

#pragma once
 
#include "Stream.h"
#include "AudioCodecs/AudioEncoded.h"
#include "AudioMetaData/MetaDataFilter.h"
#include "MP3DecoderHelix.h"
 
namespace audio_tools {
 
class MP3DecoderHelix : public AudioDecoder  {
    public:
 
        MP3DecoderHelix() {
            TRACED();
            mp3 = new libhelix::MP3DecoderHelix();
            if (mp3!=nullptr){
                mp3->setReference(this);
                filter.setDecoder(mp3);
            } else {
                LOGE("Not enough memory for libhelix");
            }
        }
        MP3DecoderHelix(Print &out_stream){
            TRACED();
            mp3 = new libhelix::MP3DecoderHelix();
            if (mp3!=nullptr){
                mp3->setReference(this);
                filter.setDecoder(mp3);
            } else {
                LOGE("Not enough memory for libhelix");
            }
            setOutput(out_stream);
        }  
 
        MP3DecoderHelix(Print &out_stream, AudioInfoSupport &bi){
            TRACED();
            mp3 = new libhelix::MP3DecoderHelix();
            if (mp3!=nullptr){
                mp3->setReference(this);
                filter.setDecoder(mp3);
            } else {
                LOGE("Not enough memory for libhelix");
            }
            setOutput(out_stream);
            addNotifyAudioChange(bi);
        }  
 
        ~MP3DecoderHelix(){
            if (mp3!=nullptr) delete mp3;
        }
 
        virtual void setOutput(Print &outStream){
            if (mp3!=nullptr) mp3->setOutput(outStream);
        }
 
        bool begin(){
            TRACED();
            if (mp3!=nullptr) {
                mp3->setDelay(CODEC_DELAY_MS);   
                mp3->begin();
                filter.begin();
            } 
            return true;
        }
 
        void end(){
            TRACED();
            if (mp3!=nullptr) mp3->end();
        }
 
        MP3FrameInfo audioInfoEx(){
            return mp3->audioInfo();
        }
 
        AudioInfo audioInfo() override {
            MP3FrameInfo i = audioInfoEx();
            AudioInfo baseInfo;
            baseInfo.channels = i.nChans;
            baseInfo.sample_rate = i.samprate;
            baseInfo.bits_per_sample = i.bitsPerSample;
            return baseInfo;
        }
 
        size_t write(const void* mp3Data, size_t len) {
            LOGD("%s: %zu", LOG_METHOD, len);
            if (mp3==nullptr) return 0;
            return use_filter ? filter.write((uint8_t*)mp3Data, len): mp3->write((uint8_t*)mp3Data, len);
        }
 
        operator bool(){
            return mp3!=nullptr && (bool) *mp3;
        }
 
        libhelix::MP3DecoderHelix *driver() {
            return mp3;
        }
 
        void addNotifyAudioChange(AudioInfoSupport &bi) override {
            TRACED();
            AudioDecoder::addNotifyAudioChange(bi);
            if (mp3!=nullptr) mp3->setInfoCallback(infoCallback, this);
        }
 
        static void infoCallback(MP3FrameInfo &i, void * ref){
            MP3DecoderHelix* p_helix = (MP3DecoderHelix* )ref;
            if (p_helix!=nullptr){
                TRACED();
                AudioInfo baseInfo;
                baseInfo.channels = i.nChans;
                baseInfo.sample_rate = i.samprate;
                baseInfo.bits_per_sample = i.bitsPerSample;
                baseInfo.logInfo("MP3DecoderHelix");
                p_helix->notifyAudioChange(baseInfo);   
            } else {
                LOGE("Wrong Libhelix Version");
            }
        }
 
        void setFilterMetaData(bool filter){
            use_filter = filter;
        }
        
        bool isFilterMetaData() {
            return use_filter;
        }
 
        size_t maxFrameSize() {
            return mp3->maxFrameSize();
        }
 
        void setMaxFrameSize(size_t len){
            mp3->setMaxFrameSize(len);
        }
 
        size_t maxPCMSize() {
            return mp3->maxPCMSize();
        }
 
        void setMaxPCMSize(size_t len) {
            mp3->setMaxPCMSize(len);
        }
    protected:
        libhelix::MP3DecoderHelix *mp3=nullptr;
        MetaDataFilter<libhelix::MP3DecoderHelix> filter;
        bool use_filter = false;
 
};
 
 
} // namespace