TransportSettings.cpp

Go to the documentation of this file.

#include "TransportSettings.h"
#include "Fragmentation.h"
#include "madara/knowledge/KnowledgeBase.h"
#include "madara/knowledge/containers/StringVector.h"
#include "madara/knowledge/containers/Map.h"

namespace containers = madara::knowledge::containers;
typedef madara::knowledge::KnowledgeRecord::Integer  Integer;

std::string
madara::transport::types_to_string (int id)
{
  if (NO_TRANSPORT == id)
  {
    return "None";
  }
  if (SPLICE == id)
  {
    return "Splice DDS";
  }
  if (NDDS == id)
  {
    return "RTI DDS";
  }
  if (UDP == id)
  {
    return "UDP";
  }
  if (TCP == id)
  {
    return "TCP (unsupported)";
  }
  if (MULTICAST == id)
  {
    return "UDP Multicast";
  }
  if (BROADCAST == id)
  {
    return "UDP Broadcast";
  }
  if (REGISTRY_SERVER == id)
  {
    return "UDP Registry Server";
  }
  if (REGISTRY_CLIENT == id)
  {
    return "UDP Registry Client";
  }
  if (ZMQ == id)
  {
    return "0MQ";
  }

  // otherwise, it's a custom transport
  return "Custom";
}

madara::transport::TransportSettings::TransportSettings () : 
  write_domain (DEFAULT_DOMAIN), 
  read_threads (1),
  queue_length (DEFAULT_QUEUE_LENGTH), 
  type (DEFAULT_TRANSPORT),
  max_fragment_size (62000),
  resend_attempts (MAXIMUM_RESEND_ATTEMPTS),
  fragment_queue_length (5),
  reliability (DEFAULT_RELIABILITY),
  id (DEFAULT_ID),
  processes (DEFAULT_PROCESSES),
  on_data_received_logic (),
  delay_launch (false),
  never_exit (false),
  send_reduced_message_header (false),
  slack_time (0),
  read_thread_hertz (0.0),
  hosts (),
  no_sending (false),
  no_receiving (false)
{
}

madara::transport::TransportSettings::TransportSettings (
  const TransportSettings & settings) : 
  write_domain (settings.write_domain),
  read_threads (settings.read_threads),
  queue_length (settings.queue_length),
  type (settings.type),
  max_fragment_size (settings.max_fragment_size),
  resend_attempts (settings.resend_attempts),
  fragment_queue_length (settings.fragment_queue_length),
  reliability (settings.reliability),
  id (settings.id),
  processes (settings.processes),
  on_data_received_logic (settings.on_data_received_logic),
  delay_launch (settings.delay_launch),
  never_exit (settings.never_exit),
  send_reduced_message_header (settings.send_reduced_message_header),
  slack_time (settings.slack_time),
  read_thread_hertz (settings.read_thread_hertz),
  hosts (),
  no_sending (settings.no_sending),
  no_receiving (settings.no_receiving),
  read_domains_ (settings.read_domains_)
{
  hosts.resize (settings.hosts.size ());
  for (unsigned int i = 0; i < settings.hosts.size (); ++i)
    hosts[i] = settings.hosts[i];
}

void
madara::transport::TransportSettings::operator= (
  const TransportSettings & settings)
{
  read_threads = settings.read_threads;
  write_domain = settings.write_domain;
  read_domains_ = settings.read_domains_;
  queue_length = settings.queue_length;
  type = settings.type;
  max_fragment_size = settings.max_fragment_size;
  resend_attempts = settings.resend_attempts;
  fragment_queue_length = settings.fragment_queue_length;
  reliability = settings.reliability;
  id = settings.id;
  processes = settings.processes;

  on_data_received_logic = settings.on_data_received_logic;
  delay_launch = settings.delay_launch;
  never_exit = settings.never_exit;

  send_reduced_message_header = settings.send_reduced_message_header;
  slack_time = settings.slack_time;
  read_thread_hertz = settings.read_thread_hertz;

  hosts.resize (settings.hosts.size ());
  for (unsigned int i = 0; i < settings.hosts.size (); ++i)
    hosts[i] = settings.hosts[i];

  no_sending = settings.no_sending;
  no_receiving = settings.no_receiving;
}

madara::transport::TransportSettings::~TransportSettings ()
{
  for (OriginatorFragmentMap::iterator originator = fragment_map.begin ();
       originator != fragment_map.end (); ++originator)
  {
    for (ClockFragmentMap::iterator clock = originator->second.begin ();
        clock != originator->second.end (); ++clock)
    {
      delete_fragments (clock->second);
    }
  }
}

void
madara::transport::TransportSettings::load (const std::string & filename,
  const std::string & prefix)
{
  knowledge::KnowledgeBase knowledge;
  knowledge.load_context (filename);

  read_threads = (uint32_t) knowledge.get (prefix + ".read_threads").to_integer ();
  write_domain = knowledge.get (prefix + ".write_domain").to_string ();
  queue_length = (uint32_t)knowledge.get (prefix + ".queue_length").to_integer ();
  type = (uint32_t)knowledge.get (prefix + ".type").to_integer ();
  max_fragment_size = (uint32_t)knowledge.get (prefix + ".max_fragment_size").to_integer ();
  resend_attempts = (uint32_t)knowledge.get (prefix + ".resend_attempts").to_integer ();
  fragment_queue_length = (uint32_t)knowledge.get (prefix + ".fragment_queue_length").to_integer ();
  reliability = (uint32_t)knowledge.get (prefix + ".reliability").to_integer ();
  id = (uint32_t)knowledge.get (prefix + ".id").to_integer ();
  processes = (uint32_t)knowledge.get (prefix + ".processes").to_integer ();

  on_data_received_logic = knowledge.get (prefix + ".on_data_received_logic").to_string ();
  delay_launch = knowledge.get (prefix + ".delay_launch").is_true ();
  never_exit = knowledge.get (prefix + ".never_exit").is_true ();

  send_reduced_message_header = knowledge.get (prefix + ".send_reduced_message_header").is_true ();
  slack_time = knowledge.get (prefix + ".slack_time").to_double ();
  read_thread_hertz = knowledge.get (prefix + ".read_thread_hertz").to_double ();

  containers::StringVector kb_hosts (prefix + ".hosts", knowledge);

  hosts.resize (kb_hosts.size ());
  for (unsigned int i = 0; i < hosts.size (); ++i)
    hosts[i] = kb_hosts[i];


  containers::Map kb_read_domains (prefix + ".read_domains", knowledge);

  std::vector <std::string> keys;
  kb_read_domains.keys (keys);

  for (unsigned int i = 0; i < keys.size (); ++i)
  {
    read_domains_[keys[i]] = 1;
  }

  no_sending = knowledge.get (prefix + ".no_sending").is_true ();
  no_receiving = knowledge.get (prefix + ".no_receiving").is_true ();
}

void
madara::transport::TransportSettings::load_text (const std::string & filename,
  const std::string & prefix)
{
  knowledge::KnowledgeBase knowledge;
  knowledge.evaluate (madara::utility::file_to_string (filename));

  read_threads = (uint32_t) knowledge.get (prefix + ".read_threads").to_integer ();
  write_domain = knowledge.get (prefix + ".write_domain").to_string ();
  queue_length = (uint32_t)knowledge.get (prefix + ".queue_length").to_integer ();
  type = (uint32_t)knowledge.get (prefix + ".type").to_integer ();
  max_fragment_size = (uint32_t)knowledge.get (prefix + ".max_fragment_size").to_integer ();
  resend_attempts = (uint32_t)knowledge.get (prefix + ".resend_attempts").to_integer ();
  fragment_queue_length = (uint32_t)knowledge.get (prefix + ".fragment_queue_length").to_integer ();
  reliability = (uint32_t)knowledge.get (prefix + ".reliability").to_integer ();
  id = (uint32_t)knowledge.get (prefix + ".id").to_integer ();
  processes = (uint32_t)knowledge.get (prefix + ".processes").to_integer ();

  on_data_received_logic = knowledge.get (prefix + ".on_data_received_logic").to_string ();
  delay_launch = knowledge.get (prefix + ".delay_launch").is_true ();
  never_exit = knowledge.get (prefix + ".never_exit").is_true ();

  send_reduced_message_header = knowledge.get (prefix + ".send_reduced_message_header").is_true ();
  slack_time = knowledge.get (prefix + ".slack_time").to_double ();
  read_thread_hertz = knowledge.get (prefix + ".read_thread_hertz").to_double ();

  containers::StringVector kb_hosts (prefix + ".hosts", knowledge);

  hosts.resize (kb_hosts.size ());
  for (unsigned int i = 0; i < hosts.size (); ++i)
    hosts[i] = kb_hosts[i];


  containers::Map kb_read_domains (prefix + ".read_domains", knowledge);

  std::vector <std::string> keys;
  kb_read_domains.keys (keys);

  for (unsigned int i = 0; i < keys.size (); ++i)
  {
    read_domains_[keys[i]] = 1;
  }

  no_sending = knowledge.get (prefix + ".no_sending").is_true ();
  no_receiving = knowledge.get (prefix + ".no_receiving").is_true ();
}

void
madara::transport::TransportSettings::save (const std::string & filename,
  const std::string & prefix) const
{
  knowledge::KnowledgeBase knowledge;

  // load what exists at the file so we can append/overwrite it
  knowledge.load_context (madara::utility::file_to_string (filename));

  containers::StringVector kb_hosts (prefix + ".hosts", knowledge,
    (int)hosts.size ());

  knowledge.set (prefix + ".read_threads", Integer (read_threads));
  knowledge.set (prefix + ".write_domain", write_domain);
  knowledge.set (prefix + ".queue_length", Integer (queue_length));
  knowledge.set (prefix + ".type", Integer (type));
  knowledge.set (prefix + ".max_fragment_size", Integer (max_fragment_size));
  knowledge.set (prefix + ".resend_attempts", Integer (resend_attempts));
  knowledge.set (prefix + ".fragment_queue_length",
    Integer (fragment_queue_length));
  knowledge.set (prefix + ".reliability", Integer (reliability));
  knowledge.set (prefix + ".id", Integer (id));
  knowledge.set (prefix + ".processes", Integer (processes));

  knowledge.set (prefix + ".on_data_received_logic", on_data_received_logic);
  knowledge.set (prefix + ".delay_launch", Integer (delay_launch));
  knowledge.set (prefix + ".never_exit", Integer (never_exit));

  knowledge.set (prefix + ".send_reduced_message_header",
    Integer (send_reduced_message_header));
  knowledge.set (prefix + ".slack_time", slack_time);
  knowledge.set (prefix + ".read_thread_hertz", read_thread_hertz);

  for (size_t i = 0; i < hosts.size (); ++i)
    kb_hosts.set (i, hosts[i]);

  knowledge.set (prefix + ".no_sending", Integer (no_sending));
  knowledge.set (prefix + ".no_receiving", Integer (no_receiving));

  knowledge::containers::Map kb_read_domains (prefix + ".read_domains", knowledge);
  for (std::map <std::string, int>::const_iterator i = read_domains_.begin ();
    i != read_domains_.end (); ++i)
  {
    kb_read_domains.set (i->first,
      (knowledge::KnowledgeRecord::Integer)i->second);
  }

  knowledge.save_context (filename);
}

void
madara::transport::TransportSettings::save_text (const std::string & filename,
  const std::string & prefix) const
{
  knowledge::KnowledgeBase knowledge;

  // load what exists at the file so we can append/overwrite it
  knowledge.evaluate (madara::utility::file_to_string (filename));

  containers::StringVector kb_hosts (prefix + ".hosts", knowledge,
    (int)hosts.size ());

  knowledge.set (prefix + ".read_threads", Integer (read_threads));
  knowledge.set (prefix + ".write_domain", write_domain);
  knowledge.set (prefix + ".queue_length", Integer (queue_length));
  knowledge.set (prefix + ".type", Integer (type));
  knowledge.set (prefix + ".max_fragment_size", Integer (max_fragment_size));
  knowledge.set (prefix + ".resend_attempts", Integer (resend_attempts));
  knowledge.set (prefix + ".fragment_queue_length",
    Integer (fragment_queue_length));
  knowledge.set (prefix + ".reliability", Integer (reliability));
  knowledge.set (prefix + ".id", Integer (id));
  knowledge.set (prefix + ".processes", Integer (processes));

  knowledge.set (prefix + ".on_data_received_logic", on_data_received_logic);
  knowledge.set (prefix + ".delay_launch", Integer (delay_launch));
  knowledge.set (prefix + ".never_exit", Integer (never_exit));

  knowledge.set (prefix + ".send_reduced_message_header",
    Integer (send_reduced_message_header));
  knowledge.set (prefix + ".slack_time", slack_time);
  knowledge.set (prefix + ".read_thread_hertz", read_thread_hertz);

  for (size_t i = 0; i < hosts.size (); ++i)
    kb_hosts.set (i, hosts[i]);

  knowledge.set (prefix + ".no_sending", Integer (no_sending));
  knowledge.set (prefix + ".no_receiving", Integer (no_receiving));

  knowledge::containers::Map kb_read_domains (prefix + ".read_domains", knowledge);
  for (std::map <std::string, int>::const_iterator i = read_domains_.begin ();
    i != read_domains_.end (); ++i)
  {
    kb_read_domains.set (i->first,
      (knowledge::KnowledgeRecord::Integer)i->second);
  }

  knowledge.save_as_karl (filename);
}