{-# OPTIONS_GHC -Wall #-}
{-# LANGUAGE ScopedTypeVariables #-}

module PlotHo.ChartRender
       ( toChartRender
       ) where

import Control.Lens ( (.~) )
import Control.Monad ( void )
import qualified Data.Colour as Colour
import Data.Colour.Names as Colour
import Data.Default.Class ( def )
import qualified Graphics.Rendering.Chart as Chart
import Graphics.Rendering.Chart.Backend.Cairo ( runBackend, defaultEnv )
import qualified Graphics.Rendering.Cairo as Cairo

import PlotHo.PlotTypes (Axes(..), AxisType(..), XY(..))

colorSeq :: [Colour.AlphaColour Double]
colorSeq = cycle $ map Colour.opaque [Colour.blue, Colour.red, Colour.green, Colour.cyan, Colour.magenta]

-- take the data and use Chart to make a Renderable ()
toChartRender :: forall a
                 . (Chart.PlotValue a, Show a, RealFloat a)
                 => Axes a
                 -> XY (a, a)
                 -> Maybe String
                 -> [(String, [[(a,a)]])]
                 -> Chart.RectSize
                 -> Cairo.Render ()
toChartRender axes (XY historyXRange historyYRange) mtitle namePcs rectSize =
  void $ runBackend (defaultEnv Chart.bitmapAlignmentFns) (Chart.render renderable rectSize)
  where
    XY xtype ytype = axesType axes
    XY manualXRange manualYRange = axesManualRange axes

    renderable :: Chart.Renderable ()
    renderable = Chart.toRenderable layout

    drawOne (name,pc) col
      = Chart.plot_lines_values .~ pc
        $ Chart.plot_lines_style  . Chart.line_color .~ col
--        $ Chart.plot_points_style ~. Chart.filledCircles 2 red
        $ Chart.plot_lines_title .~ name
        $ def
    allLines :: [Chart.PlotLines a a]
    allLines = zipWith drawOne namePcs colorSeq

    xscaleFun :: Chart.Layout a a -> Chart.Layout a a
    xscaleFun = case xtype of
      LogScaling -> Chart.layout_x_axis . Chart.laxis_generate .~ Chart.autoScaledLogAxis def
      LinearScalingAutoRange -> id
      LinearScalingManualRange ->
        Chart.layout_x_axis . Chart.laxis_generate .~ Chart.scaledAxis def manualXRange
      LinearScalingHistoryRange ->
        Chart.layout_x_axis . Chart.laxis_generate .~ Chart.scaledAxis def historyXRange


    yscaleFun :: Chart.Layout a a -> Chart.Layout a a
    yscaleFun = case ytype of
      LogScaling -> Chart.layout_y_axis . Chart.laxis_generate .~ Chart.autoScaledLogAxis def
      LinearScalingAutoRange -> id
      LinearScalingManualRange ->
        Chart.layout_y_axis . Chart.laxis_generate .~ Chart.scaledAxis def manualYRange
      LinearScalingHistoryRange ->
        Chart.layout_y_axis . Chart.laxis_generate .~ Chart.scaledAxis def historyYRange

    title = case mtitle of
      Nothing -> id
      Just t ->
        (Chart.layout_title .~ t) .
        ((Chart.layout_title_style . Chart.font_size) .~ 10)
    layout = Chart.layout_plots .~ map Chart.toPlot allLines
             $ title
             $ Chart.layout_x_axis . Chart.laxis_title .~ "time [s]"
             $ xscaleFun
             $ yscaleFun
             def