MessageProfiling.h

Go to the documentation of this file.

 
#ifndef _GAMS_ALGORITHMS_MESSAGE_PROFILING_H_
#define _GAMS_ALGORITHMS_MESSAGE_PROFILING_H_
 
#include "gams/algorithms/BaseAlgorithm.h"
 
#include <map>
#include <vector>
#include <string>
 
#include "gams/variables/Sensor.h"
#include "gams/platforms/BasePlatform.h"
#include "gams/variables/AlgorithmStatus.h"
#include "gams/variables/Self.h"
 
#include "gams/algorithms/AlgorithmFactory.h"
 
#include "madara/filters/AggregateFilter.h"
#include "madara/transport/QoSTransportSettings.h"
 
namespace gams
{
  namespace algorithms
  {
    class GAMS_EXPORT MessageProfiling : public BaseAlgorithm
    {
    public:
      MessageProfiling(
        const madara::knowledge::KnowledgeRecord& send,
        madara::knowledge::KnowledgeBase * knowledge = 0,
        platforms::Base * platform = 0,
        variables::Sensors * sensors = 0,
        variables::Self * self = 0);
 
      ~MessageProfiling();
 
      void init_filtered_transport(madara::transport::QoSTransportSettings 
        settings);
 
      void operator=(const MessageProfiling & rhs);
      
      virtual int analyze(void);
      
      virtual int execute(void);
 
      virtual int plan(void);
 
    private:
      const static std::string key_prefix_;
 
      madara::knowledge::KnowledgeBase * local_knowledge_;
 
      madara::knowledge::containers::String data_;
 
      //madara::knowledge::containers::Integer count_;
 
      class MessageFilter : public madara::filters::AggregateFilter
      {
      public:
        virtual ~MessageFilter();
 
        void filter(madara::knowledge::KnowledgeMap& records, 
          const madara::transport::TransportContext& transport_context,
          madara::knowledge::Variables& var);
 
        std::string missing_messages_string() const;
 
        struct MessageData
        {
          size_t first;
          size_t last;
          std::vector<bool> present;
        };
 
        //std::map<std::string, MessageData> msg_map;
        std::map<std::string, size_t> msg_map;
      };
 
      MessageFilter filter_;
 
      size_t send_size_;
    };
 
    class GAMS_EXPORT MessageProfilingFactory : public AlgorithmFactory
    {
    public:
 
      virtual BaseAlgorithm * create(
        const madara::knowledge::KnowledgeMap & args,
        madara::knowledge::KnowledgeBase * knowledge,
        platforms::BasePlatform * platform,
        variables::Sensors * sensors,
        variables::Self * self,
        variables::Agents * agents);
    };
  }
}
 
#endif // _GAMS_ALGORITHMS_MESSAGE_PROFILING_H_