#include "xplot/xfigure.hpp"
#include "xplot/xmarks.hpp"
#include "xplot/xaxes.hpp"

std::size_t n = 200;

std::vector<double> x(n);

double h = 1./(n-1);
for (std::size_t i=0; i<n; ++i)
{
    x[i] = -5 + 10. * i * h;
}

std::vector<std::vector<double>> color(n, std::vector<double>(n));

for (std::size_t i=0; i<n; ++i)
{
    for (std::size_t j=0; j<n; ++j)
    {
        color[i][j] = std::cos(x[j]*x[j] + x[i]*x[i]);
    }
}

xpl::linear_scale xs, ys;
xpl::color_scale cs;
cs.scheme = "RdYlBu";
xpl::heat_map heatmap(x, x, color, xs, ys, cs);

xpl::figure fig;
fig.add_mark(heatmap);
fig

fig.max_aspect_ratio = 1.;

fig.title = "Cosine";

xpl::axis hx(xs), hy(ys);
xpl::color_axis hc(cs);
hy.orientation = "vertical";
fig.add_axis(hx);
fig.add_axis(hy);
fig.add_axis(hc);