Add publish methods to server models.

This commit is contained in:
h2zero 2020-09-01 08:43:58 -06:00
parent bc2e11637e
commit 4d78c3013b
3 changed files with 110 additions and 28 deletions

View file

@ -56,7 +56,7 @@ void NimBLEMeshElement::createModel(uint16_t type, NimBLEMeshModelCallbacks *pCa
return; return;
} }
m_modelsVec.push_back(bt_mesh_model{{type},0,0,0, pModel->opPub,{0},{0},pModel->opList, pModel}); m_modelsVec.push_back(bt_mesh_model{{type},0,0,0, &pModel->m_opPub,{0},{0},pModel->m_opList, pModel});
} }
/** /**

View file

@ -27,8 +27,7 @@ NimBLEMeshModel::NimBLEMeshModel(NimBLEMeshModelCallbacks* pCallbacks) {
m_callbacks = pCallbacks; m_callbacks = pCallbacks;
} }
opList = nullptr; m_opList = nullptr;
opPub = nullptr;
m_lastTid = 0; m_lastTid = 0;
m_lastSrcAddr = 0; m_lastSrcAddr = 0;
m_lastDstAddr = 0; m_lastDstAddr = 0;
@ -40,6 +39,8 @@ NimBLEMeshModel::NimBLEMeshModel(NimBLEMeshModelCallbacks* pCallbacks) {
m_levelValue = 0; m_levelValue = 0;
m_levelTarget = 0; m_levelTarget = 0;
m_transStep = 0; m_transStep = 0;
memset(&m_opPub, 0, sizeof(m_opPub));
} }
@ -47,12 +48,8 @@ NimBLEMeshModel::NimBLEMeshModel(NimBLEMeshModelCallbacks* pCallbacks) {
* @brief destructor * @brief destructor
*/ */
NimBLEMeshModel::~NimBLEMeshModel() { NimBLEMeshModel::~NimBLEMeshModel() {
if(opList != nullptr) { if(m_opList != nullptr) {
delete[] opList; delete[] m_opList;
}
if(opPub != nullptr) {
delete[] opPub;
} }
} }
@ -116,6 +113,16 @@ void NimBLEMeshModel::startTdTimer(ble_npl_time_t timerMs) {
} }
void NimBLEMeshModel::publish() {
ble_npl_callout_reset(&m_pubTimer, 1);
}
void NimBLEMeshModel::setPubMsg() {
NIMBLE_LOGD(LOG_TAG,"Base setPubMsg");
m_opPub.msg = NULL;
}
/** /**
* @brief Generic on/off server model constructor * @brief Generic on/off server model constructor
* @param [in] pCallbacks, a pointer to a callback instance for model operations * @param [in] pCallbacks, a pointer to a callback instance for model operations
@ -124,7 +131,7 @@ NimBLEGenOnOffSrvModel::NimBLEGenOnOffSrvModel(NimBLEMeshModelCallbacks* pCallba
:NimBLEMeshModel(pCallbacks) :NimBLEMeshModel(pCallbacks)
{ {
// Register the opcodes for this model with the required callbacks // Register the opcodes for this model with the required callbacks
opList = new bt_mesh_model_op[4]{ m_opList = new bt_mesh_model_op[4]{
{ BT_MESH_MODEL_OP_2(0x82, 0x01), 0, NimBLEGenOnOffSrvModel::getOnOff }, { BT_MESH_MODEL_OP_2(0x82, 0x01), 0, NimBLEGenOnOffSrvModel::getOnOff },
{ BT_MESH_MODEL_OP_2(0x82, 0x02), 2, NimBLEGenOnOffSrvModel::setOnOff }, { BT_MESH_MODEL_OP_2(0x82, 0x02), 2, NimBLEGenOnOffSrvModel::setOnOff },
{ BT_MESH_MODEL_OP_2(0x82, 0x03), 2, NimBLEGenOnOffSrvModel::setOnOffUnack }, { BT_MESH_MODEL_OP_2(0x82, 0x03), 2, NimBLEGenOnOffSrvModel::setOnOffUnack },
@ -132,6 +139,10 @@ NimBLEGenOnOffSrvModel::NimBLEGenOnOffSrvModel(NimBLEMeshModelCallbacks* pCallba
ble_npl_callout_init(&m_tdTimer, nimble_port_get_dflt_eventq(), ble_npl_callout_init(&m_tdTimer, nimble_port_get_dflt_eventq(),
NimBLEGenOnOffSrvModel::tdTimerCb, this); NimBLEGenOnOffSrvModel::tdTimerCb, this);
ble_npl_callout_init(&m_pubTimer, nimble_port_get_dflt_eventq(),
NimBLEGenOnOffSrvModel::pubTimerCb, this);
m_opPub.msg = NET_BUF_SIMPLE(2 + 3);
} }
@ -143,6 +154,8 @@ void NimBLEGenOnOffSrvModel::getOnOff(bt_mesh_model *model,
os_mbuf *buf) os_mbuf *buf)
{ {
NimBLEMeshModel *pModel = (NimBLEMeshModel*)model->user_data; NimBLEMeshModel *pModel = (NimBLEMeshModel*)model->user_data;
pModel->setPubMsg();
/*
struct os_mbuf *msg = NET_BUF_SIMPLE(2 + 3); struct os_mbuf *msg = NET_BUF_SIMPLE(2 + 3);
bt_mesh_model_msg_init(msg, BT_MESH_MODEL_OP_2(0x82, 0x04)); bt_mesh_model_msg_init(msg, BT_MESH_MODEL_OP_2(0x82, 0x04));
@ -159,7 +172,11 @@ void NimBLEGenOnOffSrvModel::getOnOff(bt_mesh_model *model,
pModel->m_transTime : pModel->m_transTime + 1); pModel->m_transTime : pModel->m_transTime + 1);
} }
pModel->sendMessage(model, ctx, msg); pModel->sendMessage(model, ctx, msg);*/
if (bt_mesh_model_send(model, ctx, pModel->m_opPub.msg, NULL, NULL)) {
NIMBLE_LOGE(LOG_TAG, "Send status failed");
}
} }
@ -233,16 +250,39 @@ void NimBLEGenOnOffSrvModel::tdTimerCb(ble_npl_event *event) {
} }
if((pModel->m_transTime & 0x3F) && pModel->m_onOffTarget == 0) { if((pModel->m_transTime & 0x3F) && pModel->m_onOffTarget == 0) {
ble_npl_time_t ticks = 0; pModel->startTdTimer(NimBLEUtils::meshTransTimeMs(pModel->m_transTime));
ble_npl_time_ms_to_ticks(NimBLEUtils::meshTransTimeMs(pModel->m_transTime), &ticks);
ble_npl_callout_reset(&pModel->m_tdTimer, ticks);
pModel->m_transTime -= 1; pModel->m_transTime -= 1;
pModel->publish();
return; return;
} }
pModel->m_transTime = 0; pModel->m_transTime = 0;
pModel->m_onOffValue = pModel->m_onOffTarget; pModel->m_onOffValue = pModel->m_onOffTarget;
pModel->m_callbacks->setOnOff(pModel->m_onOffValue); pModel->m_callbacks->setOnOff(pModel->m_onOffValue);
pModel->publish();
}
void NimBLEGenOnOffSrvModel::pubTimerCb(ble_npl_event *event) {
NimBLEMeshModel *pModel = (NimBLEMeshModel*)event->arg;
pModel->setPubMsg();
int err = bt_mesh_model_publish(pModel->m_opPub.mod);
if(err != 0) {
NIMBLE_LOGD(LOG_TAG, "Publish rc: %d",err);
}
}
void NimBLEGenOnOffSrvModel::setPubMsg() {
bt_mesh_model_msg_init(m_opPub.msg, BT_MESH_MODEL_OP_2(0x82, 0x04));
net_buf_simple_add_u8(m_opPub.msg, m_onOffValue);
if(m_transTime > 0) {
net_buf_simple_add_u8(m_opPub.msg, m_onOffTarget);
// If we started the transition timer in setOnOff we need to correct the reported remaining time.
net_buf_simple_add_u8(m_opPub.msg, (m_delayTime > 0) ?
m_transTime : m_transTime + 1);
}
} }
/** /**
@ -253,7 +293,7 @@ NimBLEGenLevelSrvModel::NimBLEGenLevelSrvModel(NimBLEMeshModelCallbacks* pCallba
:NimBLEMeshModel(pCallbacks) :NimBLEMeshModel(pCallbacks)
{ {
// Register the opcodes for this model with the required callbacks // Register the opcodes for this model with the required callbacks
opList = new bt_mesh_model_op[8]{ m_opList = new bt_mesh_model_op[8]{
{ BT_MESH_MODEL_OP_2(0x82, 0x05), 0, NimBLEGenLevelSrvModel::getLevel }, { BT_MESH_MODEL_OP_2(0x82, 0x05), 0, NimBLEGenLevelSrvModel::getLevel },
{ BT_MESH_MODEL_OP_2(0x82, 0x06), 3, NimBLEGenLevelSrvModel::setLevel }, { BT_MESH_MODEL_OP_2(0x82, 0x06), 3, NimBLEGenLevelSrvModel::setLevel },
{ BT_MESH_MODEL_OP_2(0x82, 0x07), 3, NimBLEGenLevelSrvModel::setLevelUnack }, { BT_MESH_MODEL_OP_2(0x82, 0x07), 3, NimBLEGenLevelSrvModel::setLevelUnack },
@ -265,6 +305,9 @@ NimBLEGenLevelSrvModel::NimBLEGenLevelSrvModel(NimBLEMeshModelCallbacks* pCallba
ble_npl_callout_init(&m_tdTimer, nimble_port_get_dflt_eventq(), ble_npl_callout_init(&m_tdTimer, nimble_port_get_dflt_eventq(),
NimBLEGenLevelSrvModel::tdTimerCb, this); NimBLEGenLevelSrvModel::tdTimerCb, this);
ble_npl_callout_init(&m_pubTimer, nimble_port_get_dflt_eventq(),
NimBLEGenLevelSrvModel::pubTimerCb, this);
m_opPub.msg = NET_BUF_SIMPLE(2 + 5);
} }
@ -276,7 +319,8 @@ void NimBLEGenLevelSrvModel::getLevel(bt_mesh_model *model,
os_mbuf *buf) os_mbuf *buf)
{ {
NimBLEMeshModel *pModel = (NimBLEMeshModel*)model->user_data; NimBLEMeshModel *pModel = (NimBLEMeshModel*)model->user_data;
pModel->setPubMsg();
/*
struct os_mbuf *msg = NET_BUF_SIMPLE(4 + 3); struct os_mbuf *msg = NET_BUF_SIMPLE(4 + 3);
bt_mesh_model_msg_init(msg, BT_MESH_MODEL_OP_2(0x82, 0x08)); bt_mesh_model_msg_init(msg, BT_MESH_MODEL_OP_2(0x82, 0x08));
@ -290,6 +334,10 @@ void NimBLEGenLevelSrvModel::getLevel(bt_mesh_model *model,
} }
pModel->sendMessage(model, ctx, msg); pModel->sendMessage(model, ctx, msg);
*/
if (bt_mesh_model_send(model, ctx, pModel->m_opPub.msg, NULL, NULL)) {
NIMBLE_LOGE(LOG_TAG, "Send status failed");
}
} }
@ -423,16 +471,38 @@ void NimBLEGenLevelSrvModel::tdTimerCb(ble_npl_event *event) {
if((pModel->m_transTime & 0x3F) && pModel->m_onOffTarget == 0) { if((pModel->m_transTime & 0x3F) && pModel->m_onOffTarget == 0) {
pModel->m_levelValue += pModel->m_transStep; pModel->m_levelValue += pModel->m_transStep;
pModel->m_callbacks->setLevel(pModel->m_levelValue); pModel->m_callbacks->setLevel(pModel->m_levelValue);
ble_npl_time_t ticks = 0; pModel->startTdTimer(NimBLEUtils::meshTransTimeMs(pModel->m_transTime));
ble_npl_time_ms_to_ticks(NimBLEUtils::meshTransTimeMs(pModel->m_transTime), &ticks);
ble_npl_callout_reset(&pModel->m_tdTimer, ticks);
pModel->m_transTime -= 1; pModel->m_transTime -= 1;
pModel->publish();
return; return;
} }
pModel->m_transTime = 0; pModel->m_transTime = 0;
pModel->m_levelValue = pModel->m_levelTarget; pModel->m_levelValue = pModel->m_levelTarget;
pModel->m_callbacks->setLevel(pModel->m_levelValue); pModel->m_callbacks->setLevel(pModel->m_levelValue);
pModel->publish();
}
void NimBLEGenLevelSrvModel::pubTimerCb(ble_npl_event *event) {
NimBLEMeshModel *pModel = (NimBLEMeshModel*)event->arg;
pModel->setPubMsg();
int err = bt_mesh_model_publish(pModel->m_opPub.mod);
if(err != 0) {
NIMBLE_LOGD(LOG_TAG, "Publish rc: %d",err);
}
}
void NimBLEGenLevelSrvModel::setPubMsg() {
bt_mesh_model_msg_init(m_opPub.msg, BT_MESH_MODEL_OP_2(0x82, 0x08));
net_buf_simple_add_le16(m_opPub.msg, m_levelValue);
if(m_transTime > 0) {
net_buf_simple_add_le16(m_opPub.msg, m_levelTarget);
// If we started the transition timer in setOnOff we need to correct the reported remaining time.
net_buf_simple_add_u8(m_opPub.msg, (m_delayTime > 0) ?
m_transTime : m_transTime + 1);
}
} }
/** /**

View file

@ -20,16 +20,18 @@ class NimBLEMeshModelCallbacks;
class NimBLEMeshModel { class NimBLEMeshModel {
public: public:
NimBLEMeshModel(NimBLEMeshModelCallbacks* pCallbacks); NimBLEMeshModel(NimBLEMeshModelCallbacks* pCallbacks);
~NimBLEMeshModel(); virtual ~NimBLEMeshModel();
bt_mesh_model_op* opList; bt_mesh_model_op* m_opList;
bt_mesh_model_pub* opPub; bt_mesh_model_pub m_opPub;
NimBLEMeshModelCallbacks* m_callbacks; NimBLEMeshModelCallbacks* m_callbacks;
int extractTransTimeDelay(os_mbuf *buf); int extractTransTimeDelay(os_mbuf *buf);
bool checkRetransmit(uint8_t tid, bt_mesh_msg_ctx *ctx); bool checkRetransmit(uint8_t tid, bt_mesh_msg_ctx *ctx);
void sendMessage(bt_mesh_model *model, bt_mesh_msg_ctx *ctx, os_mbuf *msg); void sendMessage(bt_mesh_model *model, bt_mesh_msg_ctx *ctx, os_mbuf *msg);
void startTdTimer(ble_npl_time_t timerMs); void startTdTimer(ble_npl_time_t timerMs);
void publish();
virtual void setPubMsg();
uint8_t m_lastTid; uint8_t m_lastTid;
uint16_t m_lastSrcAddr; uint16_t m_lastSrcAddr;
@ -42,7 +44,9 @@ public:
int16_t m_levelValue; int16_t m_levelValue;
int16_t m_levelTarget; int16_t m_levelTarget;
int16_t m_transStep; int16_t m_transStep;
ble_npl_callout m_tdTimer; ble_npl_callout m_tdTimer;
ble_npl_callout m_pubTimer;
}; };
class NimBLEGenOnOffSrvModel : NimBLEMeshModel { class NimBLEGenOnOffSrvModel : NimBLEMeshModel {
@ -50,7 +54,7 @@ class NimBLEGenOnOffSrvModel : NimBLEMeshModel {
friend class NimBLEMeshNode; friend class NimBLEMeshNode;
NimBLEGenOnOffSrvModel(NimBLEMeshModelCallbacks* pCallbacks); NimBLEGenOnOffSrvModel(NimBLEMeshModelCallbacks* pCallbacks);
~NimBLEGenOnOffSrvModel(); ~NimBLEGenOnOffSrvModel(){};
static void getOnOff(bt_mesh_model *model, static void getOnOff(bt_mesh_model *model,
bt_mesh_msg_ctx *ctx, bt_mesh_msg_ctx *ctx,
@ -62,6 +66,10 @@ class NimBLEGenOnOffSrvModel : NimBLEMeshModel {
bt_mesh_msg_ctx *ctx, bt_mesh_msg_ctx *ctx,
os_mbuf *buf); os_mbuf *buf);
static void tdTimerCb(ble_npl_event *event); static void tdTimerCb(ble_npl_event *event);
static void pubTimerCb(ble_npl_event *event);
void setPubMsg() override;
}; };
class NimBLEGenLevelSrvModel : NimBLEMeshModel { class NimBLEGenLevelSrvModel : NimBLEMeshModel {
@ -69,7 +77,7 @@ class NimBLEGenLevelSrvModel : NimBLEMeshModel {
friend class NimBLEMeshNode; friend class NimBLEMeshNode;
NimBLEGenLevelSrvModel(NimBLEMeshModelCallbacks* pCallbacks); NimBLEGenLevelSrvModel(NimBLEMeshModelCallbacks* pCallbacks);
~NimBLEGenLevelSrvModel(); ~NimBLEGenLevelSrvModel(){};
static void getLevel(bt_mesh_model *model, static void getLevel(bt_mesh_model *model,
bt_mesh_msg_ctx *ctx, bt_mesh_msg_ctx *ctx,
@ -93,6 +101,10 @@ class NimBLEGenLevelSrvModel : NimBLEMeshModel {
bt_mesh_msg_ctx *ctx, bt_mesh_msg_ctx *ctx,
os_mbuf *buf); os_mbuf *buf);
static void tdTimerCb(ble_npl_event *event); static void tdTimerCb(ble_npl_event *event);
static void pubTimerCb(ble_npl_event *event);
void setPubMsg() override;
}; };
class NimBLEMeshModelCallbacks { class NimBLEMeshModelCallbacks {