remoteinputbuffer.h

Go to the documentation of this file.

// Copyright (C) 2016 Edouard Griffiths, F4EXB                                   //
//                                                                               //
// This program is free software; you can redistribute it and/or modify          //
// it under the terms of the GNU General Public License as published by          //
// the Free Software Foundation as version 3 of the License, or                  //
// (at your option) any later version.                                           //
//                                                                               //
// This program is distributed in the hope that it will be useful,               //
// but WITHOUT ANY WARRANTY; without even the implied warranty of                //
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the                  //
// GNU General Public License V3 for more details.                               //
//                                                                               //
// You should have received a copy of the GNU General Public License             //
// along with this program. If not, see <http://www.gnu.org/licenses/>.          //

#ifndef PLUGINS_SAMPLESOURCE_REMOTEINPUT_REMOTEINPUTBUFFER_H_
#define PLUGINS_SAMPLESOURCE_REMOTEINPUT_REMOTEINPUTBUFFER_H_

#include <channel/remotedatablock.h>
#include <QString>
#include <QDebug>
#include <cstdlib>
#include "cm256cc/cm256.h"
#include "util/movingaverage.h"


#define REMOTEINPUT_UDPSIZE 512               // UDP payload size
#define REMOTEINPUT_NBORIGINALBLOCKS 128      // number of sample blocks per frame excluding FEC blocks
#define REMOTEINPUT_NBDECODERSLOTS 16         // power of two sub multiple of uint16_t size. A too large one is superfluous.

class RemoteInputBuffer
{
public:
    RemoteInputBuffer();
    ~RemoteInputBuffer();

    // R/W operations
    void writeData(char *array); 
    uint8_t *readData(int32_t length);            

    // meta data
    const RemoteMetaDataFEC& getCurrentMeta() const { return m_currentMeta; }

    // samples timestamp
    uint32_t getTVOutSec() const { return m_tvOut_sec; }
    uint32_t getTVOutUsec() const { return m_tvOut_usec; }
    uint64_t getTVOutMSec() const { return (m_tvOut_sec * 1000LL) + (m_tvOut_usec/ 1000LL); }

    // stats

    int getCurNbBlocks() const { return m_curNbBlocks; }
    int getCurOriginalBlocks() const { return m_curOriginalBlocks; }
    int getCurNbRecovery() const { return m_curNbRecovery; }
    float getAvgNbBlocks() const { return m_avgNbBlocks; }
    float getAvgOriginalBlocks() const { return m_avgOrigBlocks; }
    float getAvgNbRecovery() const { return m_avgNbRecovery; }

    int getMinNbBlocks()
    {
        int minNbBlocks = m_minNbBlocks;
        m_minNbBlocks = 256;
        return minNbBlocks;
    }

    int getMinOriginalBlocks()
    {
        int minOriginalBlocks = m_minOriginalBlocks;
        m_minOriginalBlocks = 128;
        return minOriginalBlocks;
    }

    int getMaxNbRecovery()
    {
        int maxNbRecovery = m_maxNbRecovery;
        m_maxNbRecovery = 0;
        return maxNbRecovery;
    }

    bool allFramesDecoded()
    {
        bool framesDecoded = m_framesDecoded;
        m_framesDecoded = true;
        return framesDecoded;
    }

    float getBufferLengthInSecs() const { return m_bufferLenSec; }
    int32_t getRWBalanceCorrection() const { return m_balCorrection; }

    inline int32_t getBufferGauge() const
    {
        if (m_framesNbBytes)
        {
            int32_t val = (m_wrDeltaEstimate * 100) / (int32_t) m_framesNbBytes;
            int32_t ret = val < 0 ? -val - 50 : 50 -val;
            return ret;
        }
        else
        {
            return 0; // default position
        }
    }

    static const int framesSize = REMOTEINPUT_NBDECODERSLOTS * (RemoteNbOrginalBlocks - 1) * RemoteNbBytesPerBlock;

private:
    static const int nbDecoderSlots = REMOTEINPUT_NBDECODERSLOTS;

#pragma pack(push, 1)
    struct BufferFrame
    {
        RemoteProtectedBlock  m_blocks[RemoteNbOrginalBlocks - 1];
    };
#pragma pack(pop)

    struct DecoderSlot
    {
        RemoteProtectedBlock m_blockZero;                                       
        RemoteProtectedBlock m_originalBlocks[RemoteNbOrginalBlocks];        
        RemoteProtectedBlock m_recoveryBlocks[RemoteNbOrginalBlocks];        
        CM256::cm256_block      m_cm256DescriptorBlocks[RemoteNbOrginalBlocks]; 
        int                     m_blockCount;         
        int                     m_originalCount;      
        int                     m_recoveryCount;      
        bool                    m_decoded;            
        bool                    m_metaRetrieved;      
    };

    RemoteMetaDataFEC m_currentMeta;          
    CM256::cm256_encoder_params m_paramsCM256;          
    DecoderSlot          m_decoderSlots[nbDecoderSlots]; 
    BufferFrame          m_frames[nbDecoderSlots];       
    int                  m_framesNbBytes;                
    int                  m_decoderIndexHead;     
    int                  m_frameHead;            
    int                  m_curNbBlocks;          
    int                  m_minNbBlocks;          
    int                  m_curOriginalBlocks;    
    int                  m_minOriginalBlocks;    
    int                  m_curNbRecovery;        
    int                  m_maxNbRecovery;        
    MovingAverageUtil<int, int, 10> m_avgNbBlocks;   
    MovingAverageUtil<int, int, 10> m_avgOrigBlocks; 
    MovingAverageUtil<int, int, 10> m_avgNbRecovery; 
    bool                 m_framesDecoded;        
    int                  m_readIndex;            
    int                  m_wrDeltaEstimate;      
    uint32_t             m_tvOut_sec;            
    uint32_t             m_tvOut_usec;           
    int                  m_readNbBytes;          

    uint8_t* m_readBuffer;         
    int      m_readSize;           

    float    m_bufferLenSec;

    int      m_nbReads;       
    int      m_nbWrites;      
    int      m_balCorrection; 
    int      m_balCorrLimit;  
    CM256    m_cm256;         
    bool     m_cm256_OK;      

    inline RemoteProtectedBlock* storeOriginalBlock(int slotIndex, int blockIndex, const RemoteProtectedBlock& protectedBlock)
    {
        if (blockIndex == 0) {
            // m_decoderSlots[slotIndex].m_originalBlocks[0] = protectedBlock;
            // return &m_decoderSlots[slotIndex].m_originalBlocks[0];
            m_decoderSlots[slotIndex].m_blockZero = protectedBlock;
            return &m_decoderSlots[slotIndex].m_blockZero;
        } else {
            // m_decoderSlots[slotIndex].m_originalBlocks[blockIndex] = protectedBlock;
            // return &m_decoderSlots[slotIndex].m_originalBlocks[blockIndex];
            m_frames[slotIndex].m_blocks[blockIndex - 1] = protectedBlock;
            return &m_frames[slotIndex].m_blocks[blockIndex - 1];
        }
    }

    inline RemoteProtectedBlock& getOriginalBlock(int slotIndex, int blockIndex)
    {
        if (blockIndex == 0) {
            // return m_decoderSlots[slotIndex].m_originalBlocks[0];
            return m_decoderSlots[slotIndex].m_blockZero;
        } else {
            // return m_decoderSlots[slotIndex].m_originalBlocks[blockIndex];
            return m_frames[slotIndex].m_blocks[blockIndex - 1];
        }
    }

    inline RemoteMetaDataFEC *getMetaData(int slotIndex)
    {
        // return (MetaDataFEC *) &m_decoderSlots[slotIndex].m_originalBlocks[0];
        return (RemoteMetaDataFEC *) &m_decoderSlots[slotIndex].m_blockZero;
    }

    inline void resetOriginalBlocks(int slotIndex)
    {
        // memset((void *) m_decoderSlots[slotIndex].m_originalBlocks, 0, m_nbOriginalBlocks * sizeof(ProtectedBlock));
        memset((void *) &m_decoderSlots[slotIndex].m_blockZero, 0, sizeof(RemoteProtectedBlock));
        memset((void *) m_frames[slotIndex].m_blocks, 0, (RemoteNbOrginalBlocks - 1) * sizeof(RemoteProtectedBlock));
    }

    void initDecodeAllSlots();
    void initReadIndex();
    void rwCorrectionEstimate(int slotIndex);
    void checkSlotData(int slotIndex);
    void initDecodeSlot(int slotIndex);

    static void printMeta(const QString& header, RemoteMetaDataFEC *metaData);
};



#endif /* PLUGINS_SAMPLESOURCE_REMOTEINPUT_REMOTEINPUTBUFFER_H_ */