test-string-value-formatting.cc

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2016 Tom Henderson
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation;
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#include "ns3/core-module.h"

using namespace ns3;

NS_LOG_COMPONENT_DEFINE ("TestStringValueFormatting");

class FormattingTestObject : public Object
{
public:
  static TypeId GetTypeId (void);
  FormattingTestObject ();
  Ptr<RandomVariableStream> GetTestVariable (void) const;
private:
  Ptr<RandomVariableStream> m_testVariable;
};

NS_OBJECT_ENSURE_REGISTERED (FormattingTestObject);

TypeId
FormattingTestObject::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::FormattingTestObject")
    .SetParent<Object> ()
    .AddConstructor<FormattingTestObject> ()
    .AddAttribute ("OnTime", "A RandomVariableStream used to pick the duration of the 'On' state.",
                   StringValue ("ns3::ConstantRandomVariable[Constant=1.0]"),
                   MakePointerAccessor (&FormattingTestObject::m_testVariable),
                   MakePointerChecker <RandomVariableStream>())
  ;
  return tid;
}

FormattingTestObject::FormattingTestObject ()
{
}

Ptr<RandomVariableStream>
FormattingTestObject::GetTestVariable (void) const
{
  return m_testVariable;
}

class FormattingTestObjectHelper
{
public:
  FormattingTestObjectHelper ();
  void SetAttribute (std::string name, const AttributeValue &value);
  Ptr<Object> CreateFromFactory (void);
private:
  ObjectFactory m_factory; 
};

FormattingTestObjectHelper::FormattingTestObjectHelper ()
{
  m_factory.SetTypeId (FormattingTestObject::GetTypeId ());
}

void
FormattingTestObjectHelper::SetAttribute (std::string name, const AttributeValue &value)
{
  m_factory.Set (name, value);
}

Ptr<Object>
FormattingTestObjectHelper::CreateFromFactory (void)
{
  return m_factory.Create ();
}

int 
main (int argc, char *argv[])
{
  // CreateObject parsing
  Ptr<FormattingTestObject> obj = CreateObject<FormattingTestObject> ();
  obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable"));
  obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=0.]"));
  obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=0.|Max=1.]"));
  obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=50.|Max=100.]"));

  Ptr<RandomVariableStream> rvStream = obj->GetTestVariable ();
  // Either GetObject () or DynamicCast may be used to get subclass pointer
  Ptr<UniformRandomVariable> uniformStream = rvStream->GetObject<UniformRandomVariable> ();
  NS_ASSERT (uniformStream);

  // Check that the last setting of Min to 50 and Max to 100 worked
  DoubleValue val;
  uniformStream->GetAttribute ("Min", val);
  NS_ASSERT_MSG (val.Get () == 50, "Minimum not set to 50");
  uniformStream->GetAttribute ("Max", val);
  NS_ASSERT_MSG (val.Get () == 100, "Maximum not set to 100");


  // The following malformed values should result in an error exit
  // if uncommented

  // Attribute doesn't exist
  //obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[A=0.]"));
  // Missing right bracket
  //obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=0."));
  // Comma delimiter fails
  //obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=0.,Max=1.]"));
  // Incomplete specification
  //obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=0.|Max=]"));
  // Incomplete specification
  //obj->SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=0.|Max]"));

  // ObjectFactory parsing
  FormattingTestObjectHelper formattingHelper;
  formattingHelper.SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=30.|Max=60.0]"));
  // Attribute doesn't exist
  //formattingHelper.SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[A=0.]"));
  // Missing right bracket
  //formattingHelper.SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=30."));
  // Comma delimiter fails
  //formattingHelper.SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=30.,Max=60.]"));
  // Incomplete specification
  //formattingHelper.SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=30.|Max=]"));
  // Incomplete specification
  //formattingHelper.SetAttribute ("OnTime", StringValue ("ns3::UniformRandomVariable[Min=30.|Max]"));

  // verify that creation occurs correctly
  Ptr<Object> outputObj = formattingHelper.CreateFromFactory ();
  Ptr<FormattingTestObject> fto = DynamicCast<FormattingTestObject> (outputObj);
  NS_ASSERT_MSG (fto, "object creation failed");
  rvStream = fto->GetTestVariable ();
  uniformStream = rvStream->GetObject<UniformRandomVariable> ();
  NS_ASSERT (uniformStream);
  // Check that the last setting of Min to 30 and Max to 60 worked
  uniformStream->GetAttribute ("Min", val);
  NS_ASSERT_MSG (val.Get () == 30, "Minimum not set to 30");
  uniformStream->GetAttribute ("Max", val);
  NS_ASSERT_MSG (val.Get () == 60, "Maximum not set to 60");
}