Adding RSSI samples averaging

c4fmdemod
Andy CA6JAU 2016-12-27 00:40:14 -03:00
parent fc200c4cd7
commit 442dbe0b6a
4 changed files with 62 additions and 22 deletions

View File

@ -60,7 +60,7 @@ m_state(DMORXS_NONE),
m_n(0U), m_n(0U),
m_type(0U), m_type(0U),
m_rssiCount(0U), m_rssiCount(0U),
m_rssi(0U) m_rssi()
{ {
} }
@ -88,7 +88,10 @@ void CDMRDMORX::samples(const q15_t* samples, const uint16_t* rssi, uint8_t leng
bool CDMRDMORX::processSample(q15_t sample, uint16_t rssi) bool CDMRDMORX::processSample(q15_t sample, uint16_t rssi)
{ {
uint16_t rssi_avg;
m_buffer[m_dataPtr] = sample; m_buffer[m_dataPtr] = sample;
m_rssi[m_dataPtr] = rssi;
m_bitBuffer[m_bitPtr] <<= 1; m_bitBuffer[m_bitPtr] <<= 1;
if (sample < 0) if (sample < 0)
@ -97,9 +100,6 @@ bool CDMRDMORX::processSample(q15_t sample, uint16_t rssi)
if (m_state == DMORXS_NONE) { if (m_state == DMORXS_NONE) {
correlateSync(true); correlateSync(true);
} else { } else {
// Grab the RSSI data during the frame
if (m_state == DMORXS_VOICE && m_dataPtr == m_syncPtr)
m_rssi = rssi;
uint16_t min = m_syncPtr + DMO_BUFFER_LENGTH_SAMPLES - 1U; uint16_t min = m_syncPtr + DMO_BUFFER_LENGTH_SAMPLES - 1U;
uint16_t max = m_syncPtr + 1U; uint16_t max = m_syncPtr + 1U;
@ -132,6 +132,10 @@ bool CDMRDMORX::processSample(q15_t sample, uint16_t rssi)
samplesToBits(ptr, DMR_FRAME_LENGTH_SYMBOLS, frame, 8U, centre, threshold); samplesToBits(ptr, DMR_FRAME_LENGTH_SYMBOLS, frame, 8U, centre, threshold);
#if defined(SEND_RSSI_DATA)
rssi_avg = avgRSSI(m_startPtr+120, 420);
#endif
if (m_control == CONTROL_DATA) { if (m_control == CONTROL_DATA) {
// Data sync // Data sync
uint8_t colorCode; uint8_t colorCode;
@ -193,13 +197,13 @@ bool CDMRDMORX::processSample(q15_t sample, uint16_t rssi)
#if defined(SEND_RSSI_DATA) #if defined(SEND_RSSI_DATA)
// Send RSSI data approximately every second // Send RSSI data approximately every second
if (m_rssiCount == 2U) { if (m_rssiCount == 2U) {
frame[34U] = (m_rssi >> 8) & 0xFFU; frame[34U] = (rssi_avg >> 8) & 0xFFU;
frame[35U] = (m_rssi >> 0) & 0xFFU; frame[35U] = (rssi_avg >> 0) & 0xFFU;
serial.writeDMRData(true, frame, DMR_FRAME_LENGTH_BYTES + 3U); serial.writeDMRData(true, frame, DMR_FRAME_LENGTH_BYTES + 3U);
} else { } else {
serial.writeDMRData(true, frame, DMR_FRAME_LENGTH_BYTES + 1U); serial.writeDMRData(true, frame, DMR_FRAME_LENGTH_BYTES + 1U);
} }
m_rssiCount++; m_rssiCount++;
if (m_rssiCount >= 16U) if (m_rssiCount >= 16U)
m_rssiCount = 0U; m_rssiCount = 0U;
@ -228,13 +232,13 @@ bool CDMRDMORX::processSample(q15_t sample, uint16_t rssi)
#if defined(SEND_RSSI_DATA) #if defined(SEND_RSSI_DATA)
// Send RSSI data approximately every second // Send RSSI data approximately every second
if (m_rssiCount == 2U) { if (m_rssiCount == 2U) {
frame[34U] = (m_rssi >> 8) & 0xFFU; frame[34U] = (rssi_avg >> 8) & 0xFFU;
frame[35U] = (m_rssi >> 0) & 0xFFU; frame[35U] = (rssi_avg >> 0) & 0xFFU;
serial.writeDMRData(true, frame, DMR_FRAME_LENGTH_BYTES + 3U); serial.writeDMRData(true, frame, DMR_FRAME_LENGTH_BYTES + 3U);
} else { } else {
serial.writeDMRData(true, frame, DMR_FRAME_LENGTH_BYTES + 1U); serial.writeDMRData(true, frame, DMR_FRAME_LENGTH_BYTES + 1U);
} }
m_rssiCount++; m_rssiCount++;
if (m_rssiCount >= 16U) if (m_rssiCount >= 16U)
m_rssiCount = 0U; m_rssiCount = 0U;
@ -398,6 +402,21 @@ void CDMRDMORX::correlateSync(bool first)
} }
} }
uint16_t CDMRDMORX::avgRSSI(uint16_t start, uint16_t count)
{
float rssi_tmp = 0;
for (uint16_t i = 0U; i < count; i++) {
rssi_tmp += (float) m_rssi[start];
start++;
if (start >= DMO_BUFFER_LENGTH_SAMPLES)
start -= DMO_BUFFER_LENGTH_SAMPLES;
}
return (uint16_t) (rssi_tmp / count);
}
void CDMRDMORX::samplesToBits(uint16_t start, uint8_t count, uint8_t* buffer, uint16_t offset, q15_t centre, q15_t threshold) void CDMRDMORX::samplesToBits(uint16_t start, uint8_t count, uint8_t* buffer, uint16_t offset, q15_t centre, q15_t threshold)
{ {
for (uint8_t i = 0U; i < count; i++) { for (uint8_t i = 0U; i < count; i++) {

View File

@ -59,11 +59,12 @@ private:
uint8_t m_n; uint8_t m_n;
uint8_t m_type; uint8_t m_type;
uint16_t m_rssiCount; uint16_t m_rssiCount;
uint16_t m_rssi; uint16_t m_rssi[DMO_BUFFER_LENGTH_SAMPLES];
bool processSample(q15_t sample, uint16_t rssi); bool processSample(q15_t sample, uint16_t rssi);
void correlateSync(bool first); void correlateSync(bool first);
void samplesToBits(uint16_t start, uint8_t count, uint8_t* buffer, uint16_t offset, q15_t centre, q15_t threshold); void samplesToBits(uint16_t start, uint8_t count, uint8_t* buffer, uint16_t offset, q15_t centre, q15_t threshold);
uint16_t avgRSSI(uint16_t start, uint16_t count);
}; };
#endif #endif

View File

@ -66,7 +66,7 @@ m_state(DMRRXS_NONE),
m_n(0U), m_n(0U),
m_type(0U), m_type(0U),
m_rssiCount(0U), m_rssiCount(0U),
m_rssi(0U) m_rssi()
{ {
} }
@ -96,6 +96,8 @@ void CDMRSlotRX::reset()
bool CDMRSlotRX::processSample(q15_t sample, uint16_t rssi) bool CDMRSlotRX::processSample(q15_t sample, uint16_t rssi)
{ {
uint16_t rssi_avg;
m_delayPtr++; m_delayPtr++;
if (m_delayPtr < m_delay) if (m_delayPtr < m_delay)
return m_state != DMRRXS_NONE; return m_state != DMRRXS_NONE;
@ -105,7 +107,8 @@ bool CDMRSlotRX::processSample(q15_t sample, uint16_t rssi)
return m_state != DMRRXS_NONE; return m_state != DMRRXS_NONE;
m_buffer[m_dataPtr] = sample; m_buffer[m_dataPtr] = sample;
m_rssi[m_dataPtr] = rssi;
m_bitBuffer[m_bitPtr] <<= 1; m_bitBuffer[m_bitPtr] <<= 1;
if (sample < 0) if (sample < 0)
m_bitBuffer[m_bitPtr] |= 0x01U; m_bitBuffer[m_bitPtr] |= 0x01U;
@ -114,9 +117,6 @@ bool CDMRSlotRX::processSample(q15_t sample, uint16_t rssi)
if (m_dataPtr >= SCAN_START && m_dataPtr <= SCAN_END) if (m_dataPtr >= SCAN_START && m_dataPtr <= SCAN_END)
correlateSync(true); correlateSync(true);
} else { } else {
// Grab the RSSI data during the frame
if (m_state == DMRRXS_VOICE && m_dataPtr == m_syncPtr)
m_rssi = rssi;
uint16_t min = m_syncPtr - 1U; uint16_t min = m_syncPtr - 1U;
uint16_t max = m_syncPtr + 1U; uint16_t max = m_syncPtr + 1U;
@ -134,6 +134,10 @@ bool CDMRSlotRX::processSample(q15_t sample, uint16_t rssi)
uint16_t ptr = m_endPtr - DMR_FRAME_LENGTH_SAMPLES + DMR_RADIO_SYMBOL_LENGTH + 1U; uint16_t ptr = m_endPtr - DMR_FRAME_LENGTH_SAMPLES + DMR_RADIO_SYMBOL_LENGTH + 1U;
samplesToBits(ptr, DMR_FRAME_LENGTH_SYMBOLS, frame, 8U, centre, threshold); samplesToBits(ptr, DMR_FRAME_LENGTH_SYMBOLS, frame, 8U, centre, threshold);
#if defined(SEND_RSSI_DATA)
rssi_avg = avgRSSI(m_startPtr+120, 420);
#endif
if (m_control == CONTROL_DATA) { if (m_control == CONTROL_DATA) {
// Data sync // Data sync
@ -198,8 +202,8 @@ bool CDMRSlotRX::processSample(q15_t sample, uint16_t rssi)
#if defined(SEND_RSSI_DATA) #if defined(SEND_RSSI_DATA)
// Send RSSI data approximately every second // Send RSSI data approximately every second
if (m_rssiCount == 2U) { if (m_rssiCount == 2U) {
frame[34U] = (m_rssi >> 8) & 0xFFU; frame[34U] = (rssi_avg >> 8) & 0xFFU;
frame[35U] = (m_rssi >> 0) & 0xFFU; frame[35U] = (rssi_avg >> 0) & 0xFFU;
serial.writeDMRData(m_slot, frame, DMR_FRAME_LENGTH_BYTES + 3U); serial.writeDMRData(m_slot, frame, DMR_FRAME_LENGTH_BYTES + 3U);
} else { } else {
serial.writeDMRData(m_slot, frame, DMR_FRAME_LENGTH_BYTES + 1U); serial.writeDMRData(m_slot, frame, DMR_FRAME_LENGTH_BYTES + 1U);
@ -234,8 +238,8 @@ bool CDMRSlotRX::processSample(q15_t sample, uint16_t rssi)
#if defined(SEND_RSSI_DATA) #if defined(SEND_RSSI_DATA)
// Send RSSI data approximately every second // Send RSSI data approximately every second
if (m_rssiCount == 2U) { if (m_rssiCount == 2U) {
frame[34U] = (m_rssi >> 8) & 0xFFU; frame[34U] = (rssi_avg >> 8) & 0xFFU;
frame[35U] = (m_rssi >> 0) & 0xFFU; frame[35U] = (rssi_avg >> 0) & 0xFFU;
serial.writeDMRData(m_slot, frame, DMR_FRAME_LENGTH_BYTES + 3U); serial.writeDMRData(m_slot, frame, DMR_FRAME_LENGTH_BYTES + 3U);
} else { } else {
serial.writeDMRData(m_slot, frame, DMR_FRAME_LENGTH_BYTES + 1U); serial.writeDMRData(m_slot, frame, DMR_FRAME_LENGTH_BYTES + 1U);
@ -364,6 +368,21 @@ void CDMRSlotRX::correlateSync(bool first)
} }
} }
uint16_t CDMRSlotRX::avgRSSI(uint16_t start, uint16_t count)
{
float rssi_tmp = 0;
for (uint16_t i = 0U; i < count; i++) {
rssi_tmp += (float) m_rssi[start];
start++;
if (start >= DMO_BUFFER_LENGTH_SAMPLES)
start -= DMO_BUFFER_LENGTH_SAMPLES;
}
return (uint16_t) (rssi_tmp / count);
}
void CDMRSlotRX::samplesToBits(uint16_t start, uint8_t count, uint8_t* buffer, uint16_t offset, q15_t centre, q15_t threshold) void CDMRSlotRX::samplesToBits(uint16_t start, uint8_t count, uint8_t* buffer, uint16_t offset, q15_t centre, q15_t threshold)
{ {
for (uint8_t i = 0U; i < count; i++, start += DMR_RADIO_SYMBOL_LENGTH) { for (uint8_t i = 0U; i < count; i++, start += DMR_RADIO_SYMBOL_LENGTH) {

View File

@ -63,10 +63,11 @@ private:
uint8_t m_n; uint8_t m_n;
uint8_t m_type; uint8_t m_type;
uint16_t m_rssiCount; uint16_t m_rssiCount;
uint16_t m_rssi; uint16_t m_rssi[DMO_BUFFER_LENGTH_SAMPLES];
void correlateSync(bool first); void correlateSync(bool first);
void samplesToBits(uint16_t start, uint8_t count, uint8_t* buffer, uint16_t offset, q15_t centre, q15_t threshold); void samplesToBits(uint16_t start, uint8_t count, uint8_t* buffer, uint16_t offset, q15_t centre, q15_t threshold);
uint16_t avgRSSI(uint16_t start, uint16_t count);
}; };
#endif #endif