/* -*- 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 "gnuplot.h"

#include "ns3/assert.h"

#include <ostream>

namespace ns3 {

GnuplotDataset::GnuplotDataset ()

  : m_title ("untitled"),

    m_style (LINES),

    m_errorBars (NONE)

{}

GnuplotDataset::GnuplotDataset (std::string title)

  : m_title (title),

    m_style (LINES),

    m_errorBars (NONE)

{}

void 

GnuplotDataset::SetStyle (enum Style style)

{

  m_style = style;

}

void 

GnuplotDataset::SetErrorBars (enum ErrorBars errorBars)

{

  m_errorBars = errorBars;

}

void 

GnuplotDataset::Add (double x, double y)

{

  NS_ASSERT (m_errorBars == NONE);

  struct Data data;

  data.x = x;

  data.y = y;

  data.dx = 0.0;

  data.dy = 0.0;

  m_dataset.push_back (data);

}

void 

GnuplotDataset::Add (double x, double y, double errorDelta)

{

  NS_ASSERT (m_errorBars == X || m_errorBars == Y);

  struct Data data;

  data.x = x;

  data.y = y;

  data.dx = errorDelta;

  data.dy = errorDelta;

  m_dataset.push_back (data);

}

Gnuplot::Gnuplot (std::string pngFilename)

  : m_pngFilename (pngFilename)

{}

Gnuplot::~Gnuplot ()

{

  for (Datasets::const_iterator i = m_datasets.begin (); i != m_datasets.end (); i++)

    {

      delete *i;

    }

  m_datasets.clear ();

}

void 

Gnuplot::SetLegend (std::string xLegend, std::string yLegend)

{

  m_xLegend = xLegend;

  m_yLegend = yLegend;

}

void 

Gnuplot::AddDataset (const GnuplotDataset &dataset)

{

  m_datasets.push_back (new GnuplotDataset (dataset));

}

void 

Gnuplot::GenerateOutput (std::ostream &os)

{

  os << "set terminal png" << std::endl;

  os << "set output '" << m_pngFilename << "'" << std::endl;

  os << "set xlabel '" << m_xLegend << "'" << std::endl;

  os << "set ylabel '" << m_yLegend << "'" << std::endl;

  os << "plot ";

  for (Datasets::const_iterator i = m_datasets.begin (); i != m_datasets.end ();)

    {

      os << "'-' title '" << (*i)->m_title << "'";

      switch ((*i)->m_style) {

      case GnuplotDataset::LINES:

        os << " with lines";

	break;

      case GnuplotDataset::POINTS:

        switch ((*i)->m_errorBars)

          {

          case GnuplotDataset::NONE:

            os << " with points";

            break;

          case GnuplotDataset::X:

            os << " with xerrorbars";

            break;

          case GnuplotDataset::Y:

            os << " with yerrorbars";

            break;

          case GnuplotDataset::XY:

            os << " with xyerrorbars";

            break;

          }

	break;

      case GnuplotDataset::LINES_POINTS:

        switch ((*i)->m_errorBars)

          {

          case GnuplotDataset::NONE:

            os << " with linespoints";

            break;

          case GnuplotDataset::X:

            os << " with xerrorlines";

            break;

          case GnuplotDataset::Y:

            os << " with yerrorlines";

            break;

          case GnuplotDataset::XY:

            os << " with xyerrorlines";

            break;

          }

	break;

      case GnuplotDataset::DOTS:

	os << " with dots";

	break;

      case GnuplotDataset::IMPULSES:

	os << " with impulses";

	break;

      case GnuplotDataset::STEPS:

	os << " with steps";

	break;

      case GnuplotDataset::FSTEPS:

	os << " with fsteps";

	break;

      case GnuplotDataset::HISTEPS:

	os << " with histeps";

	break;

      }

      i++;

      if (i != m_datasets.end ())

	{

	  os << ", ";

	}

    }

  os << std::endl;

  for (Datasets::const_iterator i = m_datasets.begin (); i != m_datasets.end (); i++)

    {

      for (GnuplotDataset::Dataset::const_iterator j = (*i)->m_dataset.begin ();

	   j != (*i)->m_dataset.end (); j++)

	{

          switch ((*i)->m_errorBars) {

          case GnuplotDataset::NONE:

            os << j->x << " " << j->y << std::endl;

            break;

          case GnuplotDataset::X:

            os << j->x << " " << j->y << " " << j->dx << std::endl;

            break;

          case GnuplotDataset::Y:

            os << j->x << " " << j->y << " " << j->dy << std::endl;

            break;

          case GnuplotDataset::XY:

            os << j->x << " " << j->y << " " << j->dx << " " << j->dy << std::endl;

            break;

          }

	}

      os << "e" << std::endl;

    }

}

} // namespace ns3