src/common/ascii-writer.cc
author Guillaume Seguin <guillaume@segu.in>
Thu, 13 Aug 2009 12:54:21 +0200
changeset 4711 3c9b7588ea4c
permissions -rw-r--r--
add an ascii writer class
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
4711
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     1
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     2
/*
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     3
 * Copyright (c) 2009 INRIA
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     4
 *
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     5
 * This program is free software; you can redistribute it and/or modify
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     6
 * it under the terms of the GNU General Public License version 2 as
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     7
 * published by the Free Software Foundation;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     8
 *
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
     9
 * This program is distributed in the hope that it will be useful,
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    12
 * GNU General Public License for more details.
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    13
 *
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    14
 * You should have received a copy of the GNU General Public License
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    15
 * along with this program; if not, write to the Free Software
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    16
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    17
 *
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    18
 * Author: Guillaume Seguin <guillaume@segu.in>
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    19
 */
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    20
#include "ascii-writer.h"
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    21
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    22
#include "ns3/log.h"
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    23
#include "ns3/simulator.h"
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    24
#include "ns3/simulation-singleton.h"
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    25
#include "ns3/packet.h"
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    26
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    27
#include <fstream>
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    28
#include <map>
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    29
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    30
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    31
NS_LOG_COMPONENT_DEFINE ("AsciiWriter");
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    32
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    33
namespace ns3 {
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    34
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    35
typedef std::map<std::ostream*, Ptr<AsciiWriter> > AsciiWritersMap;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    36
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    37
Ptr<AsciiWriter>
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    38
AsciiWriter::Get (std::ostream &os)
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    39
{
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    40
  AsciiWritersMap *map = SimulationSingleton<AsciiWritersMap>::Get ();
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    41
  Ptr<AsciiWriter> writer = (*map)[&os];
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    42
  if (writer == 0)
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    43
    {
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    44
      // don't call Create<> because constructor is private
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    45
      writer = Ptr<AsciiWriter> (new AsciiWriter (&os), false);
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    46
      (*map)[&os] = writer;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    47
    }
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    48
  return writer;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    49
}
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    50
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    51
AsciiWriter::AsciiWriter (std::ostream *os)
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    52
  : m_writer (os)
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    53
{
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    54
  NS_LOG_FUNCTION (this);
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    55
}
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    56
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    57
AsciiWriter::~AsciiWriter (void)
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    58
{
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    59
  NS_LOG_FUNCTION (this);
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    60
}
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    61
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    62
void
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    63
AsciiWriter::WritePacket (enum Type type, std::string message, Ptr<const Packet> packet)
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    64
{
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    65
  std::string typeString;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    66
  switch (type)
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    67
    {
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    68
    case ENQUEUE:
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    69
      typeString = "+";
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    70
      break;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    71
    case DEQUEUE:
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    72
      typeString = "-";
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    73
      break;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    74
    case RX:
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    75
      typeString = "r";
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    76
      break;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    77
    case TX:
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    78
      typeString = "t";
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    79
      break;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    80
    case DROP:
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    81
      typeString = "d";
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    82
      break;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    83
    }
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    84
  NS_LOG_FUNCTION (this << typeString << message);
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    85
  *m_writer << typeString << " " << Simulator::Now ().GetSeconds () << " "
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    86
            << message << " " << *packet << std::endl;
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    87
}
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    88
3c9b7588ea4c add an ascii writer class
Guillaume Seguin <guillaume@segu.in>
parents:
diff changeset
    89
} // namespace ns3