/* -*-  Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2007 INRIA

 *

 * 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

 *

 * Author: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>

 */

#include "position-allocator.h"

#include "ns3/random-variable.h"

#include "ns3/double.h"

#include "ns3/uinteger.h"

#include "ns3/enum.h"

#include "ns3/log.h"

#include <cmath>

NS_LOG_COMPONENT_DEFINE ("PositionAllocator");

namespace ns3 {

NS_OBJECT_ENSURE_REGISTERED (PositionAllocator);

TypeId 

PositionAllocator::GetTypeId (void)

{

    .SetParent<Object> ();

  return tid;

}

PositionAllocator::PositionAllocator ()

{

}

PositionAllocator::~PositionAllocator ()

{}

NS_OBJECT_ENSURE_REGISTERED (ListPositionAllocator);

TypeId 

ListPositionAllocator::GetTypeId (void)

{

    .SetParent<PositionAllocator> ()

    .AddConstructor<ListPositionAllocator> ()

    ;

  return tid;

}

ListPositionAllocator::ListPositionAllocator ()

{}

void 

ListPositionAllocator::Add (Vector v)

{

  m_positions.push_back (v);

  m_current = m_positions.begin ();

}

Vector 

ListPositionAllocator::GetNext (void) const

{

  Vector v = *m_current;

  m_current++;

  return v;

}

NS_OBJECT_ENSURE_REGISTERED (GridPositionAllocator);

TypeId 

GridPositionAllocator::GetTypeId (void)

{

    .SetParent<PositionAllocator> ()

    .SetGroupName ("Mobility")

    .AddConstructor<GridPositionAllocator> ()

    .AddAttribute ("GridWidth", "The number of objects layed out on a line.",

                   Uinteger (10),

                   MakeUintegerAccessor (&GridPositionAllocator::m_n),

                   MakeUintegerChecker<uint32_t> ())

    .AddAttribute ("MinX", "The x coordinate where the grid starts.",

                   Double (1.0),

                   MakeDoubleAccessor (&GridPositionAllocator::m_xMin),

                   MakeDoubleChecker<double> ())

    .AddAttribute ("MinY", "The y coordinate where the grid starts.",

                   Double (0.0),

                   MakeDoubleAccessor (&GridPositionAllocator::m_yMin),

                   MakeDoubleChecker<double> ())

    .AddAttribute ("DeltaX", "The x space between objects.",

                   Double (1.0),

                   MakeDoubleAccessor (&GridPositionAllocator::m_deltaX),

                   MakeDoubleChecker<double> ())

    .AddAttribute ("DeltaY", "The y space between objects.",

                   Double (1.0),

                   MakeDoubleAccessor (&GridPositionAllocator::m_deltaY),

                   MakeDoubleChecker<double> ())

    .AddAttribute ("LayoutType", "The type of layout.",

                   Enum (ROW_FIRST),

                   MakeEnumAccessor (&GridPositionAllocator::m_layoutType),

                   MakeEnumChecker (ROW_FIRST, "RowFirst",

                                    COLUMN_FIRST, "ColumnFirst"))

    ;

  return tid;

}

GridPositionAllocator::GridPositionAllocator ()

  : m_current (0)

{}

Vector 

GridPositionAllocator::GetNext (void) const

{

  double x = 0.0, y = 0.0;

  switch (m_layoutType) {

  case ROW_FIRST:

    x = m_xMin + m_deltaX * (m_current % m_n);

    y = m_yMin + m_deltaY * (m_current / m_n);

    break;

  case COLUMN_FIRST:

    x = m_xMin + m_deltaX * (m_current / m_n);

    y = m_yMin + m_deltaY * (m_current % m_n);

    break;

  }

  m_current++;

  return Vector (x, y, 0.0);

}

NS_OBJECT_ENSURE_REGISTERED (RandomRectanglePositionAllocator);

TypeId

RandomRectanglePositionAllocator::GetTypeId (void)

{

    .SetParent<PositionAllocator> ()

    .SetGroupName ("Mobility")

    .AddConstructor<RandomRectanglePositionAllocator> ()

    .AddAttribute ("X",

                   "A random variable which represents the x coordinate of a position in a random rectangle.",

                   UniformVariable (0.0, 1.0),

                   MakeRandomVariableAccessor (&RandomRectanglePositionAllocator::m_x),

                   MakeRandomVariableChecker ())

    .AddAttribute ("Y",

                   "A random variable which represents the y coordinate of a position in a random rectangle.",

                   UniformVariable (0.0, 1.0),

                   MakeRandomVariableAccessor (&RandomRectanglePositionAllocator::m_y),

                   MakeRandomVariableChecker ());

  return tid;

}

RandomRectanglePositionAllocator::RandomRectanglePositionAllocator ()

{}

RandomRectanglePositionAllocator::~RandomRectanglePositionAllocator ()

{}

Vector

RandomRectanglePositionAllocator::GetNext (void) const

{

  double x = m_x.GetValue ();

  double y = m_y.GetValue ();

  return Vector (x, y, 0.0);

}

NS_OBJECT_ENSURE_REGISTERED (RandomDiscPositionAllocator);

TypeId

RandomDiscPositionAllocator::GetTypeId (void)

{

    .SetParent<PositionAllocator> ()

    .SetGroupName ("Mobility")

    .AddConstructor<RandomDiscPositionAllocator> ()

    .AddAttribute ("Theta",

                   "A random variable which represents the angle (gradients) of a position in a random disc.",

                   UniformVariable (0.0, 6.2830),

                   MakeRandomVariableAccessor (&RandomDiscPositionAllocator::m_theta),

                   MakeRandomVariableChecker ())

    .AddAttribute ("Rho",

                   "A random variable which represents the radius of a position in a random disc.",

                   UniformVariable (0.0, 200.0),

                   MakeRandomVariableAccessor (&RandomDiscPositionAllocator::m_rho),

                   MakeRandomVariableChecker ())

    .AddAttribute ("X",

                   "The x coordinate of the center of the random position disc.",

                   Double (0.0),

                   MakeDoubleAccessor (&RandomDiscPositionAllocator::m_x),

                   MakeDoubleChecker<double> ())

    .AddAttribute ("Y",

                   "The y coordinate of the center of the random position disc.",

                   Double (0.0),

                   MakeDoubleAccessor (&RandomDiscPositionAllocator::m_y),

                   MakeDoubleChecker<double> ())

    ;

  return tid;

}   

RandomDiscPositionAllocator::RandomDiscPositionAllocator ()

{}

RandomDiscPositionAllocator::~RandomDiscPositionAllocator ()

{}

Vector

RandomDiscPositionAllocator::GetNext (void) const

{

  double theta = m_theta.GetValue ();

  double rho = m_rho.GetValue ();

  double x = m_x + std::cos (theta) * rho;

  double y = m_y + std::sin (theta) * rho;

  NS_LOG_DEBUG ("Disc position x=" << x << ", y=" << y);

  return Vector (x, y, 0.0);

}

} // namespace ns3