libicsneo/include/icsneo/communication/message/a2bmessage.h

140 lines
2.5 KiB
C++

#ifndef __A2BMESSAGE_H_
#define __A2BMESSAGE_H_
#ifdef __cplusplus
#include "icsneo/communication/message/message.h"
#define A2BMESSAGE_UPSTREAM 1
#define A2BMESSAGE_DOWNSTREAM 0
#define A2BPCM_L16 16
#define A2BPCM_L24 24
#define A2BPCM_SAMPLERATE_44100 44100
#define A2BPCM_SAMPLERATE_48000 48000
namespace icsneo
{
typedef uint32_t A2BPCMSample;
typedef std::vector<A2BPCMSample> ChannelBuffer;
class A2BMessage : public Frame
{
public:
enum class A2BDirection : uint8_t
{
DownStream = 0,
UpStream = 1
};
A2BMessage() = delete;
A2BMessage(uint8_t bitDepth, uint8_t bytesPerSample, uint8_t numChannels, bool monitor) :
mBitDepth(bitDepth),
mBytesPerSample(bytesPerSample),
mMonitor(monitor)
{
downstream.resize(numChannels);
upstream.resize(numChannels);
}
void addSample(A2BPCMSample &&sample, A2BDirection dir, uint8_t channel)
{
if(dir == A2BDirection::DownStream)
{
downstream[channel].push_back(std::move(sample));
}
else
{
upstream[channel].push_back(std::move(sample));
}
totalSamples++;
}
const A2BPCMSample *getSamples(A2BDirection dir, uint8_t channel) const
{
if(channel >= getNumChannels())
{
return nullptr;
}
if(dir == A2BDirection::DownStream)
{
return downstream[channel].data();
}
return upstream[channel].data();
}
std::optional<A2BPCMSample> getSample(A2BDirection dir, uint8_t channel, uint32_t sampleIndex) const
{
const A2BPCMSample *samples = getSamples(dir, channel);
auto numSamplesInChannel = getNumSamplesInChannel(dir, channel);
if(
samples == nullptr ||
sampleIndex >= numSamplesInChannel.value_or(0)
)
{
return std::nullopt;
}
return samples[sampleIndex];
}
std::optional<std::size_t> getNumSamplesInChannel(A2BDirection dir, uint8_t channel) const
{
if(channel >= getNumChannels())
{
return std::nullopt;
}
if(dir == A2BDirection::DownStream)
{
return downstream[channel].size();
}
return upstream[channel].size();
}
size_t getNumSamples() const
{
return totalSamples;
}
uint8_t getNumChannels() const
{
return static_cast<uint8_t>(downstream.size());
}
uint8_t getBitDepth() const
{
return mBitDepth;
}
uint8_t getBytesPerSample() const
{
return mBytesPerSample;
}
bool isMonitor() const
{
return mMonitor;
}
private:
std::vector<ChannelBuffer> downstream;
std::vector<ChannelBuffer> upstream;
size_t totalSamples = 0;
uint8_t mBitDepth;
uint8_t mBytesPerSample;
bool mMonitor;
};
}
#endif // __cplusplus
#endif