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_