mirror of
				https://github.com/f4exb/sdrangel.git
				synced 2025-11-03 21:20:31 -05:00 
			
		
		
		
	
		
			
				
	
	
		
			103 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			103 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
///////////////////////////////////////////////////////////////////////////////////
 | 
						|
// Copyright (C) 2019, 2021 Edouard Griffiths, F4EXB <f4exb06@gmail.com>         //
 | 
						|
// Copyright (C) 2022 Jiří Pinkava <jiri.pinkava@rossum.ai>                      //
 | 
						|
//                                                                               //
 | 
						|
// 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 INCLUDE_SAMPLEMIFIFO_H
 | 
						|
#define INCLUDE_SAMPLEMIFIFO_H
 | 
						|
 | 
						|
#include <QObject>
 | 
						|
#include <QRecursiveMutex>
 | 
						|
#include <vector>
 | 
						|
#include "dsp/dsptypes.h"
 | 
						|
#include "export.h"
 | 
						|
 | 
						|
class SDRBASE_API SampleMIFifo : public QObject {
 | 
						|
	Q_OBJECT
 | 
						|
 | 
						|
public:
 | 
						|
    SampleMIFifo(QObject *parent = nullptr);
 | 
						|
    SampleMIFifo(unsigned int nbStreams, unsigned int size, QObject *parent = nullptr);
 | 
						|
    ~SampleMIFifo();
 | 
						|
    void init(unsigned int nbStreams, unsigned int size);
 | 
						|
    void reset();
 | 
						|
 | 
						|
    void writeSync(const quint8* data, unsigned int count); //!< de-interleaved data in input with count bytes for each stream
 | 
						|
    void writeSync(const std::vector<SampleVector::const_iterator>& vbegin, unsigned int size);
 | 
						|
    void readSync(
 | 
						|
		std::vector<SampleVector::const_iterator*> vpart1Begin, std::vector<SampleVector::const_iterator*> vpart1End,
 | 
						|
		std::vector<SampleVector::const_iterator*> vpart2Begin, std::vector<SampleVector::const_iterator*> vpart2End
 | 
						|
    );
 | 
						|
    void readSync(
 | 
						|
        std::vector<unsigned int>& vPart1Begin, std::vector<unsigned int>& vPart1End,
 | 
						|
        std::vector<unsigned int>& vPart2Begin, std::vector<unsigned int>& vPart2End
 | 
						|
    );
 | 
						|
    void readSync(
 | 
						|
		unsigned int& ipart1Begin, unsigned int& ipart1End,
 | 
						|
		unsigned int& ipart2Begin, unsigned int& ipart2End
 | 
						|
    );
 | 
						|
 | 
						|
    void writeAsync(const quint8* data, unsigned int count, unsigned int stream);
 | 
						|
    void writeAsync(const SampleVector::const_iterator& begin, unsigned int inputSize, unsigned int stream);
 | 
						|
    void readAsync(
 | 
						|
		SampleVector::const_iterator* part1Begin, SampleVector::const_iterator* part1End,
 | 
						|
		SampleVector::const_iterator* part2Begin, SampleVector::const_iterator* part2End,
 | 
						|
        unsigned int stream);
 | 
						|
    void readAsync(
 | 
						|
		unsigned int& ipart1Begin, unsigned int& ipart1End,
 | 
						|
		unsigned int& ipart2Begin, unsigned int& ipart2End,
 | 
						|
        unsigned int stream);
 | 
						|
 | 
						|
 | 
						|
    const std::vector<SampleVector>& getData() { return m_data; }
 | 
						|
    const SampleVector& getData(unsigned int stream) { return m_data[stream]; }
 | 
						|
    unsigned int getNbStreams() const { return m_data.size(); }
 | 
						|
 | 
						|
    inline unsigned int fillSync()
 | 
						|
    {
 | 
						|
        QMutexLocker mutexLocker(&m_mutex);
 | 
						|
        unsigned int fill = m_head <= m_fill ? m_fill - m_head : m_size - (m_head - m_fill);
 | 
						|
        return fill;
 | 
						|
    }
 | 
						|
 | 
						|
    inline unsigned int fillAsync(unsigned int stream)
 | 
						|
    {
 | 
						|
        if (stream >= m_nbStreams) {
 | 
						|
            return 0;
 | 
						|
        }
 | 
						|
 | 
						|
        QMutexLocker mutexLocker(&m_mutex);
 | 
						|
        unsigned int fill = m_vHead[stream] <= m_vFill[stream] ? m_vFill[stream] - m_vHead[stream] : m_size - (m_vHead[stream] - m_vFill[stream]);
 | 
						|
        return fill;
 | 
						|
    }
 | 
						|
 | 
						|
signals:
 | 
						|
	void dataSyncReady();
 | 
						|
    void dataAsyncReady(int streamIndex);
 | 
						|
 | 
						|
private:
 | 
						|
    std::vector<SampleVector> m_data;
 | 
						|
    unsigned int m_nbStreams;
 | 
						|
    unsigned int m_size;
 | 
						|
    unsigned int m_fill;               //!< Number of samples written from beginning of samples vector (sync)
 | 
						|
    unsigned int m_head;               //!< Number of samples read from beginning of samples vector (sync)
 | 
						|
    std::vector<unsigned int> m_vFill; //!< Number of samples written from beginning of samples vector (async)
 | 
						|
    std::vector<unsigned int> m_vHead; //!< Number of samples read from beginning of samples vector (async)
 | 
						|
	QRecursiveMutex m_mutex;
 | 
						|
};
 | 
						|
 | 
						|
#endif // INCLUDE_SAMPLEMIFIFO_H
 |