• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• scala-plotly-client-master
  • src
    • main
      • scala
        • co
          • theasi
            • plotly
              • Color.scala
              • Colorscale.scala
              • AxisType.scala
              • Margins.scala
              • LegendOptions.scala
              • Writable.scala
              • package.scala
              • Readable.scala
              • BarOptions.scala
              • ViewPort.scala
              • CartesianPlotOptions.scala
              • Series.scala
              • AxisOptions.scala
              • SeriesOptions.scala
              • FigureOptions.scala
              • Figure.scala
              • SurfaceOptions.scala
              • Grid.scala
              • TextValue.scala
              • Plot.scala
              • ScatterOptions.scala
              • writer
                • SeriesWriter.scala
                • SeriesReader.scala
                • PlotFile.scala
                • AxisOptionsWriter.scala
                • Api.scala
                • package.scala
                • FigureWriter.scala
                • SeriesWriteInfo.scala
                • ServerWriter.scala
                • ColorscaleWriter.scala
                • ColumnWriter.scala
                • FontWriter.scala
                • Credentials.scala
                • FigureOptionsWriter.scala
                • ThreeDPlotLayoutWriter.scala
                • OptionsWriter.scala
                • Server.scala
                • FileOptions.scala
                • ColorWriter.scala
                • CartesianPlotLayoutWriter.scala
                • GridFile.scala
                • GridWriter.scala
              • ThreeDPlotOptions.scala
              • Font.scala
              • Axis.scala
    • site
      • index.html
    • test
      • scala
        • co
          • theasi
            • plotly
              • ThreeDPlotSpec.scala
              • ScatterOptionsSpec.scala
              • AxisOptionsSpec.scala
              • FigureSpec.scala
              • CartesianPlotSpec.scala
              • BarOptionsSpec.scala
              • writer
                • ThreeDPlotLayoutWriterSpec.scala
                • CartesianPlotLayoutWriterSpec.scala
                • CredentialsSpec.scala
                • FigureWriterSpec.scala
                • WriterSpec.scala
                • OptionsWriterSpec.scala
                • AxisOptionsWriterSpec.scala
                • FigureOptionsWriterSpec.scala
                • FontWriterSpec.scala
                • SeriesWriterSpec.scala
              • SurfaceOptionsSpec.scala
              • testing
                • tags
                  • Slow.java
  • scalastyle-config.xml
  • .mailmap
  • sonatype.sbt
  • LICENSE
  • project
    • build.properties
    • plugins.sbt
  • .travis.yml
  • README.md
  • build.sbt
  • .gitignore

package co.theasi.plotly.writer

import org.json4s._
import org.json4s.native.JsonMethods._
import org.json4s.JsonDSL._

import scala.util.{ Try, Success, Failure }

import co.theasi.plotly.Grid

object GridWriter {

  def draw(
      grid: Grid,
      fileName: String,
      fileOptions: FileOptions = FileOptions()
  )(implicit server: Server): GridFile = {
    if(fileOptions.overwrite) { deleteIfExists(fileName) }
    val request = Api.post("grids",
      compact(render(gridAsJson(grid, fileName))))
    val parsedResponse = Api.despatchAndInterpret(request)
    GridFile.fromResponse(parsedResponse \ "file")
  }

  private def gridAsJson(grid: Grid, fileName: String): JObject = {
    val columnsAsJson = grid.columns.toIterator.zipWithIndex.map {
      case ((columnName, columnValues), index) =>
        ColumnWriter.toJson(columnValues, columnName, index)
    }
    val jsonObj = columnsAsJson.reduce {
      (memo, current) => memo ~ current
    }
    (
      ("data" -> ("cols" -> jsonObj)) ~
      ("filename" -> fileName)
    )
  }

  private def deleteIfExists(fileName: String)(implicit server: Server) {
    Try { GridFile.fromFileName(fileName) } match {
      case Success(grid) => // exists already -> delete
        Api.despatchAndInterpret(Api.delete(s"grids/${grid.fileId}"))
      case Failure(PlotlyException("Not found.")) => // good to go
      case Failure(e) => throw e // some other error -> re-throw
    }
  }

}