mirror of https://github.com/lumapu/ahoy.git
lumapu
2 years ago
8 changed files with 246 additions and 12 deletions
@ -0,0 +1,87 @@ |
|||||
|
//-----------------------------------------------------------------------------
|
||||
|
// 2023 Ahoy, https://github.com/lumpapu/ahoy
|
||||
|
// Creative Commons - http://creativecommons.org/licenses/by-nc-sa/4.0/deed
|
||||
|
//-----------------------------------------------------------------------------
|
||||
|
|
||||
|
#ifndef __COMM_QUEUE_H__ |
||||
|
#define __COMM_QUEUE_H__ |
||||
|
|
||||
|
#include <array> |
||||
|
#include <functional> |
||||
|
#include "hmInverter.h" |
||||
|
|
||||
|
template <uint8_t N=100> |
||||
|
class CommQueue { |
||||
|
public: |
||||
|
CommQueue() {} |
||||
|
|
||||
|
void addImportant(Inverter<> *iv, uint8_t cmd, bool delIfFailed = true) { |
||||
|
dec(&mRdPtr); |
||||
|
mQueue[mRdPtr] = queue_s(iv, cmd, delIfFailed); |
||||
|
} |
||||
|
|
||||
|
void add(Inverter<> *iv, uint8_t cmd, bool delIfFailed = true) { |
||||
|
mQueue[mWrPtr] = queue_s(iv, cmd, delIfFailed); |
||||
|
inc(&mWrPtr); |
||||
|
} |
||||
|
|
||||
|
protected: |
||||
|
struct queue_s { |
||||
|
Inverter<> *iv; |
||||
|
uint8_t cmd; |
||||
|
uint8_t attempts; |
||||
|
bool delIfFailed; |
||||
|
bool done; |
||||
|
uint32_t ts; |
||||
|
queue_s() {} |
||||
|
queue_s(Inverter<> *i, uint8_t c, bool d) : |
||||
|
iv(i), cmd(c), attempts(5), done(false), ts(0), delIfFailed(d) {} |
||||
|
}; |
||||
|
|
||||
|
protected: |
||||
|
void add(queue_s q) { |
||||
|
mQueue[mWrPtr] = q; |
||||
|
inc(&mWrPtr); |
||||
|
} |
||||
|
|
||||
|
void get(std::function<void(bool valid, const queue_s *q)> cb) { |
||||
|
if(mRdPtr == mWrPtr) { |
||||
|
cb(false, &mQueue[mRdPtr]); // empty
|
||||
|
return; |
||||
|
} |
||||
|
cb(true, &mQueue[mRdPtr]); |
||||
|
} |
||||
|
|
||||
|
void pop(void) { |
||||
|
if(!mQueue[mRdPtr].delIfFailed) |
||||
|
add(mQueue[mRdPtr]); // add to the end again
|
||||
|
inc(&mRdPtr); |
||||
|
} |
||||
|
|
||||
|
void setTs(uint32_t *ts) { |
||||
|
mQueue[mRdPtr].ts = *ts; |
||||
|
} |
||||
|
|
||||
|
void setDone(void) { |
||||
|
mQueue[mRdPtr].done = true; |
||||
|
} |
||||
|
|
||||
|
void inc(uint8_t *ptr) { |
||||
|
if(++(*ptr) >= N) |
||||
|
*ptr = 0; |
||||
|
} |
||||
|
void dec(uint8_t *ptr) { |
||||
|
if((*ptr) == 0) |
||||
|
*ptr = N-1; |
||||
|
else |
||||
|
--(*ptr); |
||||
|
} |
||||
|
|
||||
|
protected: |
||||
|
std::array<queue_s, N> mQueue; |
||||
|
uint8_t mWrPtr = 0; |
||||
|
uint8_t mRdPtr = 0; |
||||
|
}; |
||||
|
|
||||
|
|
||||
|
#endif /*__COMM_QUEUE_H__*/ |
||||
@ -0,0 +1,129 @@ |
|||||
|
//-----------------------------------------------------------------------------
|
||||
|
// 2023 Ahoy, https://github.com/lumpapu/ahoy
|
||||
|
// Creative Commons - http://creativecommons.org/licenses/by-nc-sa/4.0/deed
|
||||
|
//-----------------------------------------------------------------------------
|
||||
|
|
||||
|
#ifndef __COMMUNICATION_H__ |
||||
|
#define __COMMUNICATION_H__ |
||||
|
|
||||
|
#include "CommQueue.h" |
||||
|
#include <Arduino.h> |
||||
|
#include "../utils/crc.h" |
||||
|
|
||||
|
class Communication : public CommQueue<> { |
||||
|
public: |
||||
|
void setup(uint32_t *timestamp) { |
||||
|
mTimestamp = timestamp; |
||||
|
} |
||||
|
|
||||
|
void loop() { |
||||
|
get([this](bool valid, const queue_s *q) { |
||||
|
if(!valid) |
||||
|
return; // empty
|
||||
|
|
||||
|
switch(mState) { |
||||
|
case States::IDLE: |
||||
|
setTs(mTimestamp); |
||||
|
q->iv->radio->prepareDevInformCmd(q->iv, q->cmd, q->ts, q->iv->alarmLastId, false); |
||||
|
lastMillis = millis(); |
||||
|
lastFound = false; |
||||
|
mState = States::WAIT; |
||||
|
break; |
||||
|
|
||||
|
case States::WAIT: |
||||
|
if((millis()-lastMillis) < 500) |
||||
|
return; |
||||
|
mState = States::CHECK_FRAMES; |
||||
|
break; |
||||
|
|
||||
|
case States::CHECK_FRAMES: |
||||
|
if(!q->iv->radio->loop()) |
||||
|
break; // radio buffer empty
|
||||
|
|
||||
|
while(!q->iv->radio->mBufCtrl.empty()) { |
||||
|
packet_t *p = &q->iv->radio->mBufCtrl.front(); |
||||
|
q->iv->radio->mBufCtrl.pop(); |
||||
|
|
||||
|
if(!checkIvSerial(&p->packet[1], q->iv)) |
||||
|
continue; // inverter ID incorrect
|
||||
|
|
||||
|
q->iv->radioStatistics.frmCnt++; |
||||
|
|
||||
|
if (p->packet[0] == (TX_REQ_INFO + ALL_FRAMES)) // response from get information command
|
||||
|
parseFrame(p); |
||||
|
else if (p->packet[0] == (TX_REQ_DEVCONTROL + ALL_FRAMES)) // response from dev control command
|
||||
|
parseDevCtrl(p); |
||||
|
|
||||
|
yield(); |
||||
|
} |
||||
|
break; |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
void gotData() { |
||||
|
setDone(); |
||||
|
} |
||||
|
|
||||
|
private: |
||||
|
inline bool checkIvSerial(uint8_t buf[], Inverter<> *iv) { |
||||
|
uint8_t tmp[4]; |
||||
|
CP_U32_BigEndian(tmp, iv->radioId.u64 >> 8); |
||||
|
for(uint8_t i = 0; i < 4; i++) { |
||||
|
if(tmp[i] != buf[i]) |
||||
|
return false; |
||||
|
} |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
inline bool checkFrameCrc(uint8_t buf[], uint8_t len) { |
||||
|
return (ah::crc8(buf, len - 1) == buf[len-1]); |
||||
|
} |
||||
|
|
||||
|
inline void parseFrame(packet_t *p) { |
||||
|
uint8_t *frameId = &p->packet[9]; |
||||
|
if(0x00 == *frameId) |
||||
|
return; // skip current packet
|
||||
|
if((*frameId & 0x7f) > MAX_PAYLOAD_ENTRIES) |
||||
|
return; // local storage is to small for id
|
||||
|
|
||||
|
if(!checkFrameCrc(p->packet, p->len)) |
||||
|
return; // CRC8 is wrong, frame invalid
|
||||
|
|
||||
|
if((*frameId & ALL_FRAMES) == ALL_FRAMES) { |
||||
|
maxFrameId = (*frameId & 0x7f); |
||||
|
if(*frameId > 0x81) |
||||
|
lastFound = true; |
||||
|
} |
||||
|
|
||||
|
frame_t *f = &mLocalBuf[(*frameId & 0x7f) - 1]; |
||||
|
memcpy(f->buf, &p->packet[10], p->len-11); |
||||
|
f->len = p->len - 11; |
||||
|
f->rssi = p->rssi; |
||||
|
} |
||||
|
|
||||
|
inline void parseDevCtrl(packet_t *p) { |
||||
|
//if((p->packet[12] == ActivePowerContr) && (p->packet[13] == 0x00))
|
||||
|
} |
||||
|
|
||||
|
private: |
||||
|
enum class States : uint8_t { |
||||
|
IDLE, WAIT, CHECK_FRAMES |
||||
|
}; |
||||
|
|
||||
|
typedef struct { |
||||
|
uint8_t buf[MAX_RF_PAYLOAD_SIZE]; |
||||
|
uint8_t len; |
||||
|
int8_t rssi; |
||||
|
} frame_t; |
||||
|
|
||||
|
private: |
||||
|
States mState = States::IDLE; |
||||
|
uint32_t *mTimestamp; |
||||
|
uint32_t lastMillis; |
||||
|
std::array<frame_t, MAX_PAYLOAD_ENTRIES> mLocalBuf; |
||||
|
bool lastFound; |
||||
|
uint8_t maxFrameId; |
||||
|
}; |
||||
|
|
||||
|
#endif /*__COMMUNICATION_H__*/ |
||||
Loading…
Reference in new issue