diff -r 99ee62eec34c -r 85d201f1c67f src/energy/helper/basic-energy-source-helper.cc --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/energy/helper/basic-energy-source-helper.cc Tue Mar 22 16:40:32 2011 -0700 @@ -0,0 +1,57 @@ +/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */ +/* + * Copyright (c) 2010 Network Security Lab, University of Washington, Seattle. + * + * 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 + * + * Authors: Sidharth Nabar , He Wu + */ + +#include "basic-energy-source-helper.h" +#include "ns3/energy-source.h" + +namespace ns3 { + +BasicEnergySourceHelper::BasicEnergySourceHelper () +{ + m_basicEnergySource.SetTypeId ("ns3::BasicEnergySource"); +} + +BasicEnergySourceHelper::~BasicEnergySourceHelper () +{ +} + +void +BasicEnergySourceHelper::Set (std::string name, const AttributeValue &v) +{ + m_basicEnergySource.Set (name, v); +} + +Ptr +BasicEnergySourceHelper::DoInstall (Ptr node) const +{ + NS_ASSERT (node != NULL); + // check if energy source already exists + Ptr source = node->GetObject (); + if (source != NULL) + { + NS_FATAL_ERROR ("Energy source already installed!"); + } + source = m_basicEnergySource.Create (); + NS_ASSERT (source != NULL); + source->SetNode (node); + return source; +} + +} // namespace ns3