breeze.plot.Dataset.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of breeze-viz_2.11 Show documentation
breeze-viz
The newest version!
/*
 * Distributed as part of Scalala, a linear algebra library.
 *
 * Copyright (C) 2008- Daniel Ramage
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.

 * This library 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
 * Lesser General Public License for more details.

 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110 USA
 */
package breeze.plot;;;

import scala.collection.mutable.ArrayBuffer;

/**
 * An XY dataset consisting of some number of named series, each consisting
 * of items of type Item, with an associated x value, y value, and optionally
 * a name and tooltip.
 *
 * @author dramage
 */
class XYDataset[Item](x: Item => Number, y: Item => Number, label: Item => String, tip: Item => String)
    extends org.jfree.data.xy.AbstractXYDataset {
  val names = ArrayBuffer[String]();
  val items = ArrayBuffer[IndexedSeq[Item]]();

  override def getSeriesKey(series: Int) =
    names(series)

  override def getSeriesCount =
    names.length

  override def getItemCount(series: Int) =
    items(series).length

  override def getX(series: Int, item: Int): Number =
    x(items(series)(item))

  override def getY(series: Int, item: Int): Number =
    y(items(series)(item))

  def getLabel(series: Int, item: Int): String =
    label(items(series)(item))

  def getTip(series: Int, item: Int): String =
    tip(items(series)(item))

}

object XYDataset {
  def apply[Item](
      name: String,
      items: IndexedSeq[Item],
      x: Item => Number,
      y: Item => Number,
      label: Item => String,
      tip: Item => String): XYDataset[Item] = {
    val rv = new XYDataset(x, y, label, tip);
    rv.names += name;
    rv.items += items;
    rv;
  }
}

/**
 * An XYX dataset consisting of some number of named series, each consisting
 * of items of type Item, with an associated x value, y value, z value,
 * and optionally a name and tooltip.
 *
 * @author dramage
 */
class XYZDataset[Item](
    x: Item => Number,
    y: Item => Number,
    z: Item => Number,
    label: Item => String,
    tip: Item => String)
    extends org.jfree.data.xy.AbstractXYZDataset {
  val names = ArrayBuffer[String]();
  val items = ArrayBuffer[IndexedSeq[Item]]();

  override def getSeriesKey(series: Int) =
    names(series);

  override def getSeriesCount =
    names.length;

  override def getItemCount(series: Int) =
    items(series).length;

  override def getX(series: Int, item: Int): Number =
    x(items(series)(item));

  override def getY(series: Int, item: Int): Number =
    y(items(series)(item));

  override def getZ(series: Int, item: Int): Number =
    z(items(series)(item));

  def getLabel(series: Int, item: Int): String =
    label(items(series)(item));

  def getTip(series: Int, item: Int): String =
    tip(items(series)(item));
}

object XYZDataset {
  def apply[Item](
      name: String,
      items: IndexedSeq[Item],
      x: Item => Number,
      y: Item => Number,
      z: Item => Number,
      label: Item => String,
      tip: Item => String): XYZDataset[Item] = {
    val rv = new XYZDataset(x, y, z, label, tip);
    rv.names += name;
    rv.items += items;
    rv;
  }
}