src/helper/topology-reader-helper.cc
author Hajime Tazaki <tazaki@sfc.wide.ad.jp>
Wed, 23 Jun 2010 18:16:45 +0900
changeset 6374 787aef59db0b
parent 6127 b5bc10de166d
permissions -rw-r--r--
Bug 946 - Rocketfuel topology dataset support for topology reader
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
6127
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     1
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     2
/*
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     3
 * Copyright (c) 2010 Universita' di Firenze, Italy
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     4
 *
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     5
 * This program is free software; you can redistribute it and/or modify
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     6
 * it under the terms of the GNU General Public License version 2 as
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     7
 * published by the Free Software Foundation;
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     8
 *
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
     9
 * This program is distributed in the hope that it will be useful,
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    12
 * GNU General Public License for more details.
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    13
 *
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    14
 * You should have received a copy of the GNU General Public License
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    15
 * along with this program; if not, write to the Free Software
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    16
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    17
 *
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    18
 * Author: Tommaso Pecorella (tommaso.pecorella@unifi.it)
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    19
 * Author: Valerio Sartini (valesar@gmail.com)
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    20
 */
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    21
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    22
#include "ns3/object.h"
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    23
#include "ns3/topology-reader-helper.h"
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    24
#include "ns3/inet-topology-reader.h"
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    25
#include "ns3/orbis-topology-reader.h"
6374
787aef59db0b Bug 946 - Rocketfuel topology dataset support for topology reader
Hajime Tazaki <tazaki@sfc.wide.ad.jp>
parents: 6127
diff changeset
    26
#include "ns3/rocketfuel-topology-reader.h"
6127
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    27
#include "ns3/log.h"
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    28
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    29
namespace ns3 {
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    30
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    31
NS_LOG_COMPONENT_DEFINE ("TopologyReaderHelper");
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    32
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    33
TopologyReaderHelper::TopologyReaderHelper ()
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    34
{
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    35
  m_inFile = 0;
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    36
}
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    37
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    38
void
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    39
TopologyReaderHelper::SetFileName (const std::string fileName)
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    40
{
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    41
  m_fileName = fileName;
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    42
}
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    43
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    44
void
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    45
TopologyReaderHelper::SetFileType (const std::string fileType)
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    46
{
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    47
  m_fileType = fileType;
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    48
}
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    49
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    50
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    51
Ptr<TopologyReader>
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    52
TopologyReaderHelper::GetTopologyReader ()
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    53
{
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    54
  if (!m_inFile)
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    55
    {
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    56
      NS_ASSERT_MSG (!m_fileType.empty (), "Missing File Type");
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    57
      NS_ASSERT_MSG (!m_fileName.empty (), "Missing File Name");
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    58
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    59
      if (m_fileType == "Orbis")
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    60
        {
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    61
          NS_LOG_INFO ("Creating Orbis formatted data input.");
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    62
          m_inFile = CreateObject<OrbisTopologyReader> ();
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    63
        }
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    64
      else if (m_fileType == "Inet")
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    65
        {
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    66
          NS_LOG_INFO ("Creating Inet formatted data input.");
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    67
          m_inFile = CreateObject<InetTopologyReader> ();
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    68
        }
6374
787aef59db0b Bug 946 - Rocketfuel topology dataset support for topology reader
Hajime Tazaki <tazaki@sfc.wide.ad.jp>
parents: 6127
diff changeset
    69
      else if (m_fileType == "Rocketfuel")
787aef59db0b Bug 946 - Rocketfuel topology dataset support for topology reader
Hajime Tazaki <tazaki@sfc.wide.ad.jp>
parents: 6127
diff changeset
    70
        {
787aef59db0b Bug 946 - Rocketfuel topology dataset support for topology reader
Hajime Tazaki <tazaki@sfc.wide.ad.jp>
parents: 6127
diff changeset
    71
          NS_LOG_INFO ("Creating Rocketfuel formatted data input.");
787aef59db0b Bug 946 - Rocketfuel topology dataset support for topology reader
Hajime Tazaki <tazaki@sfc.wide.ad.jp>
parents: 6127
diff changeset
    72
          m_inFile = CreateObject<RocketfuelTopologyReader> ();
787aef59db0b Bug 946 - Rocketfuel topology dataset support for topology reader
Hajime Tazaki <tazaki@sfc.wide.ad.jp>
parents: 6127
diff changeset
    73
        }
6127
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    74
      else
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    75
        {
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    76
          NS_ASSERT_MSG (false, "Wrong (unknown) File Type");
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    77
        }
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    78
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    79
      m_inFile->SetFileName (m_fileName);
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    80
    }
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    81
  return m_inFile;
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    82
}
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    83
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    84
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    85
b5bc10de166d merge topology read system
Tommaso Pecorella <tpecorella@mac.com>
parents:
diff changeset
    86
} // namespace ns3