module Chart.Glyph
( GlyphOptions(..)
, hline_
, vline_
, glyph_
, glyphs
, lglyphs
, glyphChart
, glyphChart_
, lglyphChart
, lglyphChart_
, circle
, square
, triangle
) where
import Chart.Core
import Chart.Text
import Diagrams.Prelude hiding (Color, D, scaleX, scaleY)
import NumHask.Pair
import NumHask.Prelude
import NumHask.Rect
data GlyphOptions b = GlyphOptions
{ glyphSize :: Double
, glyphColor :: AlphaColour Double
, glyphBorderColor :: AlphaColour Double
, glyphBorderSize :: Double
, glyphShape :: Double -> Chart b
}
instance Default (GlyphOptions b) where
def = GlyphOptions 0.03 ublue ugrey 0.015 circle
vline_ :: Double -> Double -> Chart b
vline_ fatness x = vrule x # scaleX (1.6 / 0.5 * fatness)
hline_ :: Double -> Double -> Chart b
hline_ fatness x = hrule x # scaleY (1.6 / 0.5 * fatness)
glyph_ :: GlyphOptions b -> Chart b
glyph_ (GlyphOptions s c bc bs shape) = shape s # fcA c # lcA bc # lwN bs
glyphs :: (R2 r, Traversable f) => GlyphOptions b -> f (r Double) -> Chart b
glyphs opts xs = mconcat $ toList $ (\x -> positioned x (glyph_ opts)) <$> xs
glyphChart ::
(Traversable f)
=> [GlyphOptions b]
-> Aspect
-> Rect Double
-> [f (Pair Double)]
-> Chart b
glyphChart optss (Aspect asp) r xyss =
mconcat $ zipWith glyphs optss (projectss r asp xyss)
glyphChart_ ::
(Traversable f)
=> [GlyphOptions b]
-> Aspect
-> [f (Pair Double)]
-> Chart b
glyphChart_ optss asp xyss = glyphChart optss asp (range xyss) xyss
lglyphs ::
(R2 r, Traversable f)
=> LabelOptions
-> GlyphOptions b
-> f (Text, r Double)
-> Chart b
lglyphs lopts gopts xs =
mconcat $
toList $ (\(t, x) -> moveTo (p_ x) $ labelled lopts t (glyph_ gopts)) <$> xs
lglyphChart ::
(Traversable f)
=> [LabelOptions]
-> [GlyphOptions b]
-> Aspect
-> Rect Double
-> [f (Text, Pair Double)]
-> Chart b
lglyphChart ls gs (Aspect asp) r xyss =
mconcat $
getZipList $
lglyphs <$> ZipList ls <*> ZipList gs <*>
ZipList
(zipWith
zip
(map fst . toList <$> xyss)
(projectss r asp (map snd . toList <$> xyss)))
lglyphChart_ ::
(Traversable f)
=> [LabelOptions]
-> [GlyphOptions b]
-> Aspect
-> [f (Text, Pair Double)]
-> Chart b
lglyphChart_ ls gs asp xyss =
lglyphChart ls gs asp (range (map snd . toList <$> xyss)) xyss