services-ros1-xbee_mav/include/PacketsHandler.h

201 lines
7.5 KiB
C++

/* PacketsHandler.h-- Packets Handler class for XBee:
Serialize, deserialize, fragment and reassemly mavlink
messages -- */
/* ------------------------------------------------------------------------- */
/* February 06, 2017 -- @Copyright Aymen Soussia. All rights reserved. */
/* (aymen.soussia@gmail.com) */
#pragma once
#include <atomic>
#include <inttypes.h>
#include <mutex>
#include <map>
#include <set>
#include <string>
#include <unistd.h>
#include <vector>
#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/xml_parser.hpp>
#include <boost/foreach.hpp>
#include <boost/filesystem.hpp>
#include <mavros_msgs/Mavlink.h>
#include "MultithreadingDeque.hpp"
#include "SerialDevice.h"
#include "frame_generators.h"
//*****************************************************************************
using boost::property_tree::ptree;
namespace Mist
{
namespace Xbee
{
struct Reassembly_Packet_S
{
uint8_t packet_ID_;
std::string packet_buffer_; // TO DO make it shared ptr
std::set<uint8_t> received_fragments_IDs_;
std::clock_t time_since_creation_; // TO DO use it to delete packets with time out
};
struct On_Line_Node_S
{
uint64_t device_64_bits_address_;
float packet_loss_raw_;
float packet_loss_filtered_;
uint16_t sent_packets_;
uint16_t received_packets_;
uint8_t packet_loss_sent_id_;
uint8_t packet_loss_received_id_;
};
enum RSSI_API_STATUS
{
OLD_VALUE,
TRIGGERED,
NEW_VALUE
};
struct RSSI_Result
{
uint16_t payload_size;
uint16_t iterations;
uint16_t success;
uint16_t retries;
uint8_t result;
uint8_t rr;
uint8_t max_rssi;
uint8_t min_rssi;
uint8_t avg_rssi;
RSSI_API_STATUS status;
};
//*****************************************************************************
class PacketsHandler
{
public:
PacketsHandler();
~PacketsHandler();
bool Init(SerialDevice* serial_device, Thread_Safe_Deque* in_packets);
bool Init_Device_ID();
void Run();
void Handle_Mavlink_Message(const mavros_msgs::Mavlink::ConstPtr& mavlink_msg);
void Process_Fragment(std::shared_ptr<std::string> fragment);
void Process_Ping_Or_Acknowledgement(std::shared_ptr<std::string> frame);
void Process_Command_Response(const char* command_response);
void Quit();
void Delete_Packets_With_Time_Out();
void Deserialize_Mavlink_Message(const char * bytes,
mavros_msgs::Mavlink* mavlink_msg, const std::size_t msg_size);
uint8_t getDeviceId();
uint8_t getDequeFull();
float getSignalStrength();
float getAPISignalStrength(uint8_t short_node_id);
float getRawPacketLoss(uint8_t short_node_id);
float getPacketLoss(uint8_t short_node_id);
void triggerRssiUpdate();
uint8_t triggerAPIRssiUpdate(uint16_t rssi_payload_size,
uint16_t rssi_iterations,
uint8_t target_id);
void processPacketLoss(const char* packet_loss);
void sendPacketLoss();
static uint16_t ucharToUint16(unsigned char msb, unsigned char lsb);
static const uint16_t PACKET_LOSS_UNAVAILABLE;
static const uint8_t ALL_IDS;
static const uint8_t MAX_WAITING_OUT_MSG;
private:
const std::size_t MAX_PACEKT_SIZE; /* MAX packet size in bytes = 63750 bytes */
const std::size_t XBEE_NETWORK_MTU; /* Maximum Transmission Unit of Xbee netwrok = 256 bytes (max payload) - 6 bytes (Header size of each fragment) = 250 bytes */
const std::size_t FRAGMENT_HEADER_SIZE; /* Header size of each fragment = 6 bytes */
const std::clock_t MAX_TIME_TO_SEND_PACKET; /* Maximum time before dropping a packet = 30 seconds*/
;
// RSSI constants
const std::string RSSI_COMMAND;
const float RSSI_FILTER_GAIN; //filter: old * (1.0-GAIN) + new * GAIN
// Packet Loss
const std::string PACKET_LOSS_IDENTIFIER;
const uint8_t MAX_PACKET_LOSS_MSG_ID; // used for synchronisation
const float PACKET_LOSS_FILTER_GAIN; //filter: new * (1.0-GAIN) + old * GAIN
void Insert_Fragment_In_Packet_Buffer(std::string* buffer,
const char* fragment, const uint16_t offset, const std::size_t length);
void Add_New_Node_To_Network(const uint8_t new_node_address);
void Serialize_Mavlink_Message(const mavros_msgs::Mavlink::ConstPtr&
mavlink_msg, std::shared_ptr<std::string> serialized_packet);
void Insert_Fragment_Header(bool single_fragment,
std::shared_ptr<std::string> fragment, const uint8_t packet_ID,
const uint8_t fragment_ID, const uint16_t offset);
void Process_Out_Standard_Messages();
void Process_Out_Packets();
void Send_Packet(const Out_Packet_S& packet);
void Send_End_Of_Packet_Ping(const uint8_t packet_ID, const uint8_t total_NBR_of_fragments);
bool Load_Database_Addresses();
bool Get_Local_Address();
bool Check_Packet_Transmitted_To_All_Nodes();
void Init_Network_Nodes_For_New_Transmission(const uint8_t packet_ID,
std::vector<std::string>* frames,
std::shared_ptr<std::vector<std::shared_ptr<std::string> > > packet);
void Transmit_Fragments(const std::vector<std::string>& frames);
void Adjust_Optimum_MT_Number(const std::clock_t elapsed_time,
const std::size_t NBR_of_transmission);
void Send_SL_and_SH_Commands();
float computePercentage(const int16_t numerator, const int16_t denumerator) const;
float filterIIR(const float new_val, const float old_val, const float gain) const;
void updatePacketLoss(On_Line_Node_S& node, const uint16_t recieved_packet);
void processAPIRssi(const char* command_response);
uint64_t get64BitsAddress(const char* bytes, const int offset);
std::set<uint8_t> fragments_indexes_to_transmit_;
SerialDevice* serial_device_;
std::atomic<bool> quit_;
Thread_Safe_Deque out_std_messages_;
Thread_Safe_Deque_Of_Vectors out_packets_;
Thread_Safe_Deque* in_packets_;
std::map<uint8_t, bool> connected_network_nodes_;
std::map<uint8_t, bool>::iterator connected_network_nodes_it_;
std::map<uint8_t, Reassembly_Packet_S> packets_assembly_map_;
std::map<uint8_t, Reassembly_Packet_S>::iterator assembly_map_it_;
std::map<uint64_t, uint8_t> database_addresses_;
std::map<uint64_t, uint8_t>::iterator database_addresses_it_;
std::map<uint8_t, uint64_t> database_addresses_inv_;
std::map<uint8_t, uint64_t>::iterator database_addresses_inv_it_;
std::map<uint8_t, On_Line_Node_S> packet_loss_map_;
std::map<uint8_t, On_Line_Node_S>::iterator packet_loss_it_;
std::map<uint8_t, RSSI_Result> rssi_result_map_;
std::map<uint8_t, RSSI_Result>::iterator rssi_result_map_it_;
std::mutex mutex_;
uint8_t device_address_;
uint64_t local_64_bits_address_;
std::string device_64_bits_address_;
bool loaded_SL_;
bool loaded_SH_;
bool deque_full_;
float rssi_float_;
uint8_t current_processed_packet_ID_;
std::size_t optimum_MT_NBR_;
// TO DO & after auto !?
useconds_t delay_interframes_;
};
}
}