Библиотека Haskell Chart: создайте линейный график журнала с одинаковыми осями X и Y

Хорошо, я попадаю в тупик библиотекой Haskell Chart. Я разобрался таким образом, чтобы сделать линейный график log-log Vector значений в Kronos Haskell:

import Data.Vector (Vector, (!))
import qualified Data.Vector as V
import Graphics.Rendering.Chart.Easy hiding (Vector)

logLogChart name points = toRenderable $ execEC $ plot chart
    where chart = line name [V.toList $ V.imap makePoint points]
          makePoint x y = (LogValue (fromIntegral (x+1)), LogValue y)

Это, безусловно, дает разумную линейную диаграмму log-log с автоматически выбранными диапазонами для осей x и y на основе данных. Один пример (представленный в Kronos Haskell):

Пример лог-лога

Проблема в том, что у меня есть специализированное приложение, где мне нужны эти две вещи:

  1. Диапазон двух осей должен быть одинаковым. (Так как моя ось X представляет собой индексы на основе 1 в Vectorэто можно упростить, чтобы диапазон оси y определялся диапазоном оси x.)
  2. Размеры визуализированной диаграммы должны быть квадратными, а не прямоугольными, как в примере выше.

Я попытался просмотреть документацию для библиотеки, но это просто поставило меня в тупик. Есть указатели?

1 ответ

Решение

Это будет сделано, но может быть значительно улучшено (не в последнюю очередь путем удаления unsafePerformIO).

{-# OPTIONS_GHC -Wall                      #-}
{-# OPTIONS_GHC -fno-warn-name-shadowing   #-}
{-# OPTIONS_GHC -fno-warn-type-defaults    #-}
{-# OPTIONS_GHC -fno-warn-unused-do-bind   #-}
{-# OPTIONS_GHC -fno-warn-missing-methods  #-}
{-# OPTIONS_GHC -fno-warn-orphans          #-}

import Graphics.Rendering.Chart hiding ( translate )
import Graphics.Rendering.Chart.Backend.Diagrams
import Diagrams.Backend.Cairo.CmdLine
import Diagrams.Prelude hiding ( render, Renderable )
import Data.Default.Class
import Diagrams.Backend.CmdLine

import System.IO.Unsafe


pointVals :: [(Double, Double)]
pointVals = map (\(x,y) -> (log x, log y)) [(1,10), (10, 100), (100, 1000)]

dataPts :: PlotPoints Double Double
dataPts = plot_points_style .~ filledCircles 2 (opaque red)
          $ plot_points_values .~ pointVals
          $ plot_points_title .~ "Data points"
          $ def

layout :: Layout Double Double
layout = layout_title .~ "Log vs Log"
         $ layout_y_axis . laxis_generate .~ scaledAxis def (0,10)
         $ layout_x_axis . laxis_generate .~ scaledAxis def (0,10)
         $ layout_plots .~ [toPlot dataPts]
         $ def

myChart :: Renderable ()
myChart = toRenderable layout

denv :: DEnv
denv = unsafePerformIO $ defaultEnv vectorAlignmentFns 500 500

displayHeader :: FilePath -> Diagram B R2 -> IO ()
displayHeader fn =
  mainRender ( DiagramOpts (Just 900) (Just 700) fn
             , DiagramLoopOpts False Nothing 0
             )

myDiagram :: Diagram Cairo R2
myDiagram = fst $ runBackend denv (render myChart (500, 500))

main :: IO ()
main = displayHeader "LogChart.png" myDiagram
Другие вопросы по тегам