module ggplotd.ggplotd;

import cconfig = cairo.c.config;
import cpdf = cairo.pdf;
import csvg = cairo.svg;
import cairo = cairo;

import ggplotd.colour;
import ggplotd.geom : Geom;
import ggplotd.bounds : Bounds;
import ggplotd.colourspace : RGBA;

version (unittest)
{
    import dunit.toolkit;
}

/// delegate that takes a Title struct and returns a changed Title struct
alias TitleFunction = Title delegate(Title);

/// Currently only holds the title. In the future could also be used to store details on location etc.
struct Title
{
    /// The actual title
    string[] title;
}

/** 
Draw the title

Examples:
--------------------
GGPlotD().put( title( "My title" ) );
--------------------
*/
TitleFunction title( string title )
{
    return delegate(Title t) { t.title = [title]; return t; };
}

/** 
Draw the multiline title

Examples:
--------------------
GGPlotD().put( title( ["My title line1", "line2", "line3"] ) );
--------------------
*/
TitleFunction title( string[] title )
{
    return delegate(Title t) { t.title = title; return t; };
}

private auto createEmptySurface( string fname, int width, int height,
    RGBA colour )
{
    cairo.Surface surface;

    static if (cconfig.CAIRO_HAS_PDF_SURFACE)
        {
        if (fname[$ - 3 .. $] == "pdf")
            {
            surface = new cpdf.PDFSurface(fname, width, height);
        }
    }
    else
        {
        if (fname[$ - 3 .. $] == "pdf")
            assert(0, "PDF support not enabled by cairoD");
    }
    static if (cconfig.CAIRO_HAS_SVG_SURFACE)
        {
        if (fname[$ - 3 .. $] == "svg")
            {
            surface = new csvg.SVGSurface(fname, width, height);
        }
    }
    else
    {
        if (fname[$ - 3 .. $] == "svg")
            assert(0, "SVG support not enabled by cairoD");
    }
    if (fname[$ - 3 .. $] == "png")
    {
        surface = new cairo.ImageSurface(cairo.Format.CAIRO_FORMAT_ARGB32, width, height);
    }

    import ggplotd.colourspace : toCairoRGBA;
    auto backcontext = cairo.Context(surface);
    backcontext.setSourceRGBA(colour.toCairoRGBA);
    backcontext.paint;

    return surface;
}

///
private auto drawTitle( in Title title, ref cairo.Surface surface,
    in Margins margins, int width )
{
    auto context = cairo.Context(surface);
    context.setFontSize(16.0);
    context.moveTo( width/2, margins.top/2 );
    
    auto f = context.fontExtents();
    foreach(t; title.title)
    {
        auto e = context.textExtents(t);
        context.relMoveTo( -e.width/2, 0 );
        context.showText(t);
        context.relMoveTo( -e.width/2, f.height );
    }

    return surface;
}

import ggplotd.scale : ScaleType;
import ggplotd.guide : GuideToDoubleFunction, GuideToColourFunction;
private auto drawGeom( in Geom geom, ref cairo.Surface surface,
     in GuideToDoubleFunction xFunc, in GuideToDoubleFunction yFunc,
     in GuideToColourFunction cFunc, in GuideToDoubleFunction sFunc, 
     in ScaleType scaleFunction, 
     in Bounds bounds, 
     in Margins margins, int width, int height )
{
    if (geom.draw.isNull)
        return surface;
    cairo.Context context;
    if (geom.mask) {
        auto plotSurface = cairo.Surface.createForRectangle(surface,
            cairo.Rectangle!double(margins.left, margins.top,
            width - (margins.left+margins.right), 
            height - (margins.top+margins.bottom)));
        context = cairo.Context(plotSurface);
    } else {
        context = cairo.Context(surface);
        context.translate(margins.left, margins.top);
    }
    import std.conv : to;
    context = scaleFunction(context, bounds,
        width.to!double - (margins.left+margins.right),
        height.to!double - (margins.top+margins.bottom));
    context = geom.draw(context, xFunc, yFunc, cFunc, sFunc);
    return surface;
}

/// Specify margins in number of pixels
struct Margins
{
    /// Create new Margins object based on old one
    this(in Margins copy) {
        this(copy.left, copy.right, copy.bottom, copy.top);
    }

    /// Create new Margins object based on specified sizes
    this(in size_t l, in size_t r, in size_t b, in size_t t) {
        left = l;
        right = r;
        bottom = b;
        top = t;
    }

    /// left margin
    size_t left = 50;
    /// right margin
    size_t right = 20; 
    /// bottom margin
    size_t bottom = 50; 
    /// top margin
    size_t top = 40; 
}

Margins defaultMargins(int size1, int size2)
{
    import std.conv : to;
    Margins margins;
    auto scale = defaultScaling(size1, size2);
    margins.left = (margins.left*scale).to!size_t;
    margins.right = (margins.right*scale).to!size_t;
    margins.top = (margins.top*scale).to!size_t;
    margins.bottom = (margins.bottom*scale).to!size_t;
    return margins;
}

private auto defaultScaling( int size ) 
{
    if (size > 500)
        return 1;
    if (size < 100)
        return 0.6;
    return 0.6+(1.0-0.6)*(size-100)/(500-100);
}

private auto defaultScaling( int size1, int size2 ) 
{
    return (defaultScaling(size1) + defaultScaling(size2))/2.0;
}

unittest 
{
    assertEqual(defaultScaling(50), 0.6);
    assertEqual(defaultScaling(600), 1.0);
    assertEqual(defaultScaling(100), 0.6);
    assertEqual(defaultScaling(500), 1.0);
    assertEqual(defaultScaling(300), 0.8);
}

/// GGPlotD contains the needed information to create a plot
struct GGPlotD
{
    import ggplotd.bounds : height, width;
    import ggplotd.colour : ColourGradientFunction;
    import ggplotd.scale : ScaleType;

    /**
    Draw the plot to a cairoD cairo surface.

    Params:
        surface = Surface object of type cairo.Surface from cairoD library, on top of which this plot is drawn.
        width = Width of the given surface.
        height = Height of the given surface.

    Returns:
        Resulting surface of the same type as input surface, with this plot drawn on top of it.
    */
    ref cairo.Surface drawToSurface( ref cairo.Surface surface, int width, int height ) const
    {
        import std.range : empty, front;
        import std.typecons : Tuple;

        import ggplotd.bounds : AdaptiveBounds;
        import ggplotd.guide : GuideStore;

        Tuple!(double, string)[] xAxisTicks;
        Tuple!(double, string)[] yAxisTicks;

        GuideStore!"x" xStore;
        GuideStore!"y" yStore;
        GuideStore!"colour" colourStore;
        GuideStore!"size" sizeStore;

        foreach (geom; geomRange.data)
        {
            xStore.put(geom.xStore);
            yStore.put(geom.yStore);
            colourStore.put(geom.colourStore);
            sizeStore.put(geom.sizeStore);
        }

        AdaptiveBounds bounds;
        bounds.adapt(xStore.min(), yStore.min());
        bounds.adapt(xStore.max(), yStore.max());

        import std.algorithm : map;
        import std.array : array;
        import std.typecons : tuple;
        if (xStore.hasDiscrete)
            xAxisTicks = xStore
                .storeHash
                .byKeyValue()
                .map!((kv) => tuple(kv.value, kv.key))
                .array;
        if (yStore.hasDiscrete)
            yAxisTicks = yStore
                .storeHash
                .byKeyValue()
                .map!((kv) => tuple(kv.value, kv.key))
                .array;

        // Axis
        import std.algorithm : sort, uniq, min, max;
        import std.range : chain;
        import std.array : array;

        import ggplotd.axes : initialized, axisAes;

        // TODO move this out of here and add some tests
        // If ticks are provided then we make sure the bounds include them
        auto xSortedTicks = xAxisTicks.sort().uniq.array;
        if (!xSortedTicks.empty)
        {
            bounds.min_x = min( bounds.min_x, xSortedTicks[0][0] );
            bounds.max_x = max( bounds.max_x, xSortedTicks[$-1][0] );
        }
        if (initialized(xaxis))
        {
            bounds.min_x = xaxis.min;
            bounds.max_x = xaxis.max;
        }

        // This needs to happen before the offset of x axis is set
        auto ySortedTicks = yAxisTicks.sort().uniq.array;
        if (!ySortedTicks.empty)
        {
            bounds.min_y = min( bounds.min_y, ySortedTicks[0][0] );
            bounds.max_y = max( bounds.max_y, ySortedTicks[$-1][0] );
        }
        if (initialized(yaxis))
        {
            bounds.min_y = yaxis.min;
            bounds.max_y = yaxis.max;
        }

        import std.math : isNaN;
        auto offset = bounds.min_y;
        if (!isNaN(xaxis.offset))
            offset = xaxis.offset;
        if (!xaxis.show) // Trixk to draw the axis off screen if it is hidden
            offset = yaxis.min - bounds.height;

        // TODO: Should really take separate scaling for number of ticks (defaultScaling(width)) 
        // and for font: defaultScaling(widht, height)
        auto aesX = axisAes("x", bounds.min_x, bounds.max_x, offset, defaultScaling(width, height),
            xSortedTicks );

        offset = bounds.min_x;
        if (!isNaN(yaxis.offset))
            offset = yaxis.offset;
        if (!yaxis.show) // Trixk to draw the axis off screen if it is hidden
            offset = xaxis.min - bounds.width;
        auto aesY = axisAes("y", bounds.min_y, bounds.max_y, offset, defaultScaling(height, width),
            ySortedTicks );

        import ggplotd.geom : geomAxis;
        import ggplotd.axes : tickLength;

        auto currentMargins = margins(width, height);

        auto gR = chain(
                geomAxis(aesX, 
                    bounds.height.tickLength(height - currentMargins.bottom - currentMargins.top, 
                        defaultScaling(width), defaultScaling(height)), xaxis.label), 
                geomAxis(aesY, 
                    bounds.width.tickLength(width - currentMargins.left - currentMargins.right, 
                        defaultScaling(width), defaultScaling(height)), yaxis.label), 
            );
        auto plotMargins = Margins(currentMargins);
        if (!legends.empty)
            plotMargins.right += legends[0].width;

        // Plot axis and geomRange
        import ggplotd.guide : guideFunction;
        auto xFunc = guideFunction(xStore);
        auto yFunc = guideFunction(yStore);
        auto cFunc = guideFunction(colourStore, this.colourGradient());
        auto sFunc = guideFunction(sizeStore);

        foreach (geom; chain(geomRange.data, gR) )
        {
            surface = geom.drawGeom( surface,
                xFunc, yFunc, cFunc, sFunc,
                scale(), bounds, 
                plotMargins, width, height );
        }

        // Plot title
        surface = title.drawTitle( surface, currentMargins, width );

        import std.range : iota, zip, dropOne;
        foreach(ly; zip(legends, iota(0.0, height, height/(legends.length+1.0)).dropOne)) 
        {
            auto legend = ly[0];
            auto y = ly[1] - legend.height*.5;
            if (legend.type == "continuous") {
                import ggplotd.legend : drawContinuousLegend; 
                auto legendSurface = cairo.Surface.createForRectangle(surface,
                    cairo.Rectangle!double(width - currentMargins.right - legend.width, 
                    y, legend.width, legend.height ));//margins.right, margins.right));
                legendSurface = drawContinuousLegend( legendSurface, 
                legend.width, legend.height, 
                    colourStore, this.colourGradient );
            } else if (legend.type == "discrete") {
                import ggplotd.legend : drawDiscreteLegend; 
                auto legendSurface = cairo.Surface.createForRectangle(surface,
                    cairo.Rectangle!double(width - currentMargins.right - legend.width, 
                    y, legend.width, legend.height ));//margins.right, margins.right));
                legendSurface = drawDiscreteLegend( legendSurface, 
                legend.width, legend.height, 
                    colourStore, this.colourGradient );
            }
        }

        return surface;
    }
 
    version(ggplotdGTK) 
    {
        import gtkdSurface = cairo.Surface; // cairo surface module in GtkD package.

        /**
        Draw the plot to a GtkD cairo surface.

        Params:
            surface = Surface object of type cairo.Surface from GtkD library, on top of which this plot is drawn.
            width = Width of the given surface.
            height = Height of the given surface.

        Returns:
            Resulting surface of the same type as input surface, with this plot drawn on top of it.
        */
        auto drawToSurface( ref gtkdSurface.Surface surface, int width, int height ) const
        {
            import gtkc = gtkc.cairotypes;
            import cairod = cairo.c.cairo;

            alias gtkd_surface_t = gtkc.cairo_surface_t;
            alias cairod_surface_t = cairod.cairo_surface_t;

            cairo.Surface cairodSurface = new cairo.Surface(cast(cairod_surface_t*)surface.getSurfaceStruct());
            drawToSurface(cairodSurface, width, height);

            return surface;
        }
    }

    /// save the plot to a file
    void save( string fname, int width = 470, int height = 470 ) const
    {
        bool pngWrite = false;
        auto surface = createEmptySurface( fname, width, height,
            theme.backgroundColour );

        surface = drawToSurface( surface, width, height );

        if (fname[$ - 3 .. $] == "png")
        {
            pngWrite = true;
        }

        if (pngWrite)
            (cast(cairo.ImageSurface)(surface)).writeToPNG(fname);
    }

    /// Using + to extend the plot for compatibility to ggplot2 in R
    ref GGPlotD opBinary(string op, T)(T rhs) if (op == "+")
    {
        import ggplotd.axes : XAxisFunction, YAxisFunction;
        import ggplotd.colour : ColourGradientFunction;
        static if (is(ElementType!T==Geom))
        {
            geomRange.put( rhs );
        }
        static if (is(T==ScaleType))
        {
            scaleFunction = rhs;
        }
        static if (is(T==XAxisFunction))
        {
            xaxis = rhs( xaxis );
        }
        static if (is(T==YAxisFunction))
        {
            yaxis = rhs( yaxis );
        }
        static if (is(T==TitleFunction))
        {
            title = rhs( title );
        }
        static if (is(T==ThemeFunction))
        {
            theme = rhs( theme );
        }
        static if (is(T==Margins))
        {
            _margins = rhs;
        }
        static if (is(T==Legend))
        {
            legends ~= rhs;
        }
        static if (is(T==ColourGradientFunction)) {
            colourGradientFunction = rhs;
        }
        return this;
    }
/// put/add to the plot
    ref GGPlotD put(T)(T rhs)
    {
        return this.opBinary!("+", T)(rhs);
    }

    /// Active scale
    ScaleType scale() const
    {
        import ggplotd.scale : defaultScale = scale;
        // Return active function or the default
        if (!scaleFunction.isNull)
            return scaleFunction;
        else 
            return defaultScale();
    }

    /// Active colourGradient
    ColourGradientFunction colourGradient() const
    {
        import ggplotd.colour : defaultColourGradient = colourGradient;
        import ggplotd.colourspace : HCY;
        if (!colourGradientFunction.isNull)
            return colourGradientFunction;
        else
            return defaultColourGradient!HCY("");
    }

    /// Active margins
    Margins margins(int width, int height) const
    {
        if (!_margins.isNull)
            return _margins;
        else
            return defaultMargins(width, height);
    }

private:
    import std.range : Appender;
    import ggplotd.theme : Theme, ThemeFunction;
    import ggplotd.legend : Legend;
    Appender!(Geom[]) geomRange;

    import ggplotd.axes : XAxis, YAxis;
    XAxis xaxis;
    YAxis yaxis;


    Title title;
    Theme theme;

    import std.typecons : Nullable;
    Nullable!(Margins) _margins;
    Nullable!(ScaleType) scaleFunction;
    Nullable!(ColourGradientFunction) colourGradientFunction;

    Legend[] legends;
}

unittest
{
    import std.range : zip;
    import std.algorithm : map;
    import ggplotd.geom;
    import ggplotd.aes;

    const win_width = 1024;
    const win_height = 1024;

    const radius = 400.;


    auto gg = GGPlotD();
    gg = zip([ 0, radius*0.45 ], [ 0, radius*0.45])
        .map!((a) => aes!("x","y")(a[0], a[1]))
        .geomLine.putIn(gg);
    gg = zip([ 300, radius*0.45 ], [ 210, radius*0.45])
        .map!((a) => aes!("x","y")(a[0], a[1]))
        .geomLine.putIn(gg);

    import ggplotd.theme : Theme, ThemeFunction;
    Theme theme;

    auto surface = createEmptySurface( "test.png", win_width, win_height,
        theme.backgroundColour );

    auto dim = gg.geomRange.data.length;
    surface = gg.drawToSurface( surface, win_width, win_height );
    assertEqual( dim, gg.geomRange.data.length );
    surface = gg.drawToSurface( surface, win_width, win_height );
    assertEqual( dim, gg.geomRange.data.length );
    surface = gg.drawToSurface( surface, win_width, win_height );
    assertEqual( dim, gg.geomRange.data.length );
}

version(ggplotdGTK) 
{
    unittest 
    {
        import std.range : zip;
        import std.algorithm : map;
        // Draw same plot on cairod.ImageSurface, and on gtkd.cairo.ImageSurface,
        // and prove resulting images are the same.

        import ggplotd.geom;
        import ggplotd.aes;

        import gtkSurface = cairo.Surface;
        import gtkImageSurface = cairo.ImageSurface;
        import gtkCairoTypes = gtkc.cairotypes;

        const win_width = 1024;
        const win_height = 1024;

        const radius = 400.;

        auto gg = GGPlotD();
        gg = zip([ 0, radius*0.45 ], [ 0, radius*0.45])
            .map!((a) => aes!("x","y")(a[0], a[1]))
            .geomLine.putIn(gg);
        gg = zip([ 300, radius*0.45 ], [ 210, radius*0.45])
            .map!((a) => aes!("x","y")(a[0], a[1]))
            .geomLine.putIn(gg);

        cairo.Surface cairodSurface = 
            new cairo.ImageSurface(cairo.Format.CAIRO_FORMAT_RGB24, win_width, win_height);
        gtkSurface.Surface gtkdSurface = 
            gtkImageSurface.ImageSurface.create(gtkCairoTypes.cairo_format_t.RGB24, 
                win_width, win_height);

        auto cairodImageSurface = cast(cairo.ImageSurface)cairodSurface;
        auto gtkdImageSurface = cast(gtkImageSurface.ImageSurface)gtkdSurface;

        gg.drawToSurface(cairodSurface, win_width, win_height);
        gg.drawToSurface(gtkdSurface, win_width, win_height);

        auto byteSize = win_width*win_height*4;

        assertEqual(cairodImageSurface.getData()[0..byteSize], 
            gtkdImageSurface.getData()[0..byteSize]);
    }
}

unittest
{
    import ggplotd.axes : yaxisLabel, yaxisRange;
    auto gg = GGPlotD()
        .put( yaxisLabel( "My ylabel" ) )
        .put( yaxisRange( 0, 2.0 ) );
    assertEqual( gg.yaxis.max, 2.0 );
    assertEqual( gg.yaxis.label, "My ylabel" );

    gg = GGPlotD(); 
    gg.put( yaxisLabel( "My ylabel" ) )
        .put( yaxisRange( 0, 2.0 ) );
    assertEqual( gg.yaxis.max, 2.0 );
    assertEqual( gg.yaxis.label, "My ylabel" );
}


///
unittest
{
    import std.range : zip;
    import std.algorithm : map;

    import ggplotd.aes : aes;
    import ggplotd.geom : geomLine;
    import ggplotd.scale : scale;
    auto gg = zip(["a", "b", "c", "b"], ["x", "y", "y", "x"], ["b", "b", "b", "b"])
        .map!((a) => aes!("x", "y", "colour")(a[0], a[1], a[2]))
        .geomLine
        .putIn(GGPlotD());
    gg + scale();
    gg.save( "test6.png");
}

///
unittest
{
    // http://blackedder.github.io/ggplotd/images/noise.png
    import std.array : array;
    import std.math : sqrt;
    import std.algorithm : map;
    import std.range : zip, iota;
    import std.random : uniform;

    import ggplotd.aes : aes;
    import ggplotd.geom : geomLine, geomPoint;
    // Generate some noisy data with reducing width
    auto f = (double x) { return x/(1+x); };
    auto width = (double x) { return sqrt(0.1/(1+x)); };
    auto xs = iota( 0, 10, 0.1 ).array;

    auto ysfit = xs.map!((x) => f(x));
    auto ysnoise = xs.map!((x) => f(x) + uniform(-width(x),width(x))).array;

    auto gg = xs.zip(ysnoise)
        .map!((a) => aes!("x", "y", "colour")(a[0], a[1], "a"))
        .geomPoint
        .putIn(GGPlotD());

    gg = xs.zip(ysfit).map!((a) => aes!("x", "y")(a[0], a[1])).geomLine.putIn(gg);

    //  
    auto ys2fit = xs.map!((x) => 1-f(x));
    auto ys2noise = xs.map!((x) => 1-f(x) + uniform(-width(x),width(x))).array;

    gg = xs.zip(ys2fit).map!((a) => aes!("x", "y")(a[0], a[1]))
        .geomLine
        .putIn(gg);
    gg = xs.zip(ys2noise)
        .map!((a) => aes!("x", "y", "colour")(a[0], a[1], "b"))
        .geomPoint
        .putIn(gg);

    gg.save( "noise.png" );
}

///
unittest
{
    // http://blackedder.github.io/ggplotd/images/hist.png
    import std.array : array;
    import std.algorithm : map;
    import std.range : iota, zip;
    import std.random : uniform;

    import ggplotd.aes : aes;
    import ggplotd.geom : geomHist, geomPoint;
    import ggplotd.range : mergeRange;

    auto xs = iota(0,25,1).map!((x) => uniform(0.0,5)+uniform(0.0,5)).array;
    auto gg = xs 
        .map!((a) => aes!("x")(a))
        .geomHist
        .putIn(GGPlotD());

    gg = xs.map!((a) => aes!("x", "y")(a, 0.0))
        .geomPoint
        .putIn(gg);

    gg.save( "hist.png" );
}

/// Setting background colour
unittest
{
    /// http://blackedder.github.io/ggplotd/images/background.svg
    import std.range : zip;
    import std.algorithm : map;
    import ggplotd.aes : aes;
    import ggplotd.theme : background;
    import ggplotd.geom : geomPoint;

    // http://blackedder.github.io/ggplotd/images/polygon.png
    auto gg = zip([1, 0, 0.0], [1, 1, 0.0], [1, 0.1, 0])
        .map!((a) => aes!("x", "y", "colour")(a[0], a[1], a[2]))
        .geomPoint
        .putIn(GGPlotD());
    gg.put(background(RGBA(0.7, 0.7, 0.7, 1)));
    gg.save( "background.svg" );
}

/// Other data type
unittest
{
    /// http://blackedder.github.io/ggplotd/images/data.png
    import std.array : array;
    import std.math : sqrt;
    import std.algorithm : map;
    import std.range : iota;
    import std.random : uniform;

    import ggplotd.geom : geomPoint;

    struct Point { double x; double y; }
    // Generate some noisy data with reducing width
    auto f = (double x) { return x/(1+x); };
    auto width = (double x) { return sqrt(0.1/(1+x)); };
    immutable xs = iota( 0, 10, 0.1 ).array;

    auto points = xs.map!((x) => Point(x,
        f(x) + uniform(-width(x),width(x))));

    auto gg = GGPlotD().put( geomPoint( points ) );

    gg.save( "data.png" );
}

import std.range : ElementType;

/**
Put an element into a plot/facets struct

This basically reverses a call to put and allows one to write more idiomatic D code where code flows from left to right instead of right to left.

Examples:
--------------------
auto gg = data.aes.geomPoint.putIn(GGPlotD());
// instead of
auto gg = GGPlotD().put(geomPoint(aes(data)));
--------------------
*/
ref auto putIn(T, U)(T t, U u) 
{
    return u.put(t);
}

/**
Plot multiple (sub) plots
*/
struct Facets
{
    ///
    ref Facets put(GGPlotD facet)
    {
        ggs.put( facet );
        return this;
    }

    ///
    auto drawToSurface( ref cairo.Surface surface, int dimX, int dimY, 
            int width, int height ) const
    {
        import std.conv : to;
        import std.math : floor;
        import std.range : save, empty, front, popFront;
        import cairo.cairo : Rectangle;
        int w = floor( width.to!double/dimX ).to!int;
        int h = floor( height.to!double/dimY ).to!int;

        auto gs = ggs.data.save;
        foreach( i; 0..dimX )
        {
            foreach( j; 0..dimY )
            {
                if (!gs.empty) 
                {
                    auto rect = Rectangle!double( w*i, h*j, w, h );
                    auto subS = cairo.Surface.createForRectangle( surface, rect );
                    gs.front.drawToSurface( subS, w, h ),
                    gs.popFront;
                }
            }
        }

        return surface;
    }

    ///
    auto drawToSurface( ref cairo.Surface surface,
            int width, int height ) const
    {
        import std.conv : to;
        // Calculate dimX/dimY from width/height
        auto grid = gridLayout( ggs.data.length, width.to!double/height );
        return drawToSurface( surface, grid[0], grid[1], width, height );
    }
 
 
    ///
    void save( string fname, int dimX, int dimY, int width = 470, int height = 470 ) const
    {
        bool pngWrite = false;
        auto surface = createEmptySurface( fname, width, height,
            RGBA(1,1,1,1) );

        surface = drawToSurface( surface, dimX, dimY, width, height );

        if (fname[$ - 3 .. $] == "png")
        {
            pngWrite = true;
        }

        if (pngWrite)
            (cast(cairo.ImageSurface)(surface)).writeToPNG(fname);
    }

    ///
    void save( string fname, int width = 470, int height = 470 ) const
    {
        import std.conv : to;
        // Calculate dimX/dimY from width/height
        auto grid = gridLayout( ggs.data.length, width.to!double/height );
        save( fname, grid[0], grid[1], width, height );
    }

    import std.range : Appender;

    Appender!(GGPlotD[]) ggs;
 }

auto gridLayout( size_t length, double ratio )
{
    import std.conv : to;
    import std.math : ceil, sqrt;
    import std.typecons : Tuple;
    auto h = ceil( sqrt(length/ratio) );
    auto w = ceil(length/h);
    return Tuple!(int, int)( w.to!int, h.to!int );
}

unittest
{
    import std.typecons : Tuple;
    assertEqual(gridLayout(4, 1), Tuple!(int, int)(2, 2));
    assertEqual(gridLayout(2, 1), Tuple!(int, int)(1, 2));
    assertEqual(gridLayout(3, 1), Tuple!(int, int)(2, 2));
    assertEqual(gridLayout(2, 2), Tuple!(int, int)(2, 1));
}

///
unittest
{
    // Drawing different shapes
    import ggplotd.aes : aes, Pixel;
    import ggplotd.axes : xaxisRange, yaxisRange;
    import ggplotd.geom : geomDiamond, geomRectangle;

    auto gg = GGPlotD();

    auto aes1 = [aes!("x", "y", "width", "height")(1.0, -1.0, 3.0, 5.0)];
    gg.put( geomDiamond( aes1 ) );
    gg.put( geomRectangle( aes1 ) );
    gg.put( xaxisRange( -5, 11.0 ) );
    gg.put( yaxisRange( -9, 9.0 ) );


    auto aes2 = [aes!("x", "y", "width", "height")(8.0, 5.0, Pixel(10), Pixel(20))];
    gg.put( geomDiamond( aes2 ) );
    gg.put( geomRectangle( aes2 ) );

    auto aes3 = [aes!("x", "y", "width", "height")(6.0, -5.0, Pixel(25), Pixel(25))];
    gg.put( geomDiamond( aes3 ) );
    gg.put( geomRectangle( aes3 ) );
 
    gg.save( "shapes1.png", 300, 300 );
}

///
unittest
{
    // Drawing different shapes
    import ggplotd.aes : aes, Pixel;
    import ggplotd.axes : xaxisRange, yaxisRange;

    import ggplotd.geom : geomEllipse, geomTriangle;

    auto gg = GGPlotD();

    auto aes1 = [aes!("x", "y", "width", "height")( 1.0, -1.0, 3.0, 5.0 )];
    gg.put( geomEllipse( aes1 ) );
    gg.put( geomTriangle( aes1 ) );
    gg.put( xaxisRange( -5, 11.0 ) );
    gg.put( yaxisRange( -9, 9.0 ) );


    auto aes2 = [aes!("x", "y", "width", "height")(8.0, 5.0, Pixel(10), Pixel(20))];
    gg.put( geomEllipse( aes2 ) );
    gg.put( geomTriangle( aes2 ) );

    auto aes3 = [aes!("x", "y", "width", "height")( 6.0, -5.0, Pixel(25), Pixel(25))];
    gg.put( geomEllipse( aes3 ) );
    gg.put( geomTriangle( aes3 ) );
 
    gg.save( "shapes2.png", 300, 300 );
}