diff --git a/src/main/scala/org/viz/lightning/types/Three.scala b/src/main/scala/org/viz/lightning/types/Three.scala index 95815f4..a541496 100644 --- a/src/main/scala/org/viz/lightning/types/Three.scala +++ b/src/main/scala/org/viz/lightning/types/Three.scala @@ -4,9 +4,20 @@ import org.viz.lightning.Visualization trait Three extends Base { - def scatter3(x: Array[Double], y: Array[Double], z: Array[Double]): Visualization = { + def scatter3(x: Array[Double], + y: Array[Double], + z: Array[Double], + label: Array[Int] = Array[Int]() + size: Array[Double] = Array[Double]()): Visualization = { val out = (x, y, z).zipped.map((x, y, z) => List(x, y, z)).toList - val payload = Map("points" -> out) + var payload = Map[String, Any]() + payload += "points" -> out + if (label.length > 0) { + payload += "label" -> label.toList + } + if (size.length > 0) { + payload += "size" -> size.toList + } plot("scatter3", payload) }