#define _USE_MATH_DEFINES
#include "../matplotlib.h"
#include <cmath>
#include <iostream>

using namespace std;
namespace plt = matplotlibcpp;

int main() {
  // Prepare data
  int ncols = 500, nrows = 300;
  std::vector<float> z(ncols * nrows);
  for (int j = 0; j < nrows; ++j) {
    for (int i = 0; i < ncols; ++i) {
      z.at(ncols * j + i) = std::sin(std::hypot(i - ncols / 2, j - nrows / 2));
    }
  }

  const float *zptr = &(z[0]);
  const int colors = 1;

  plt::title("My matrix");
  plt::imshow(zptr, nrows, ncols, colors);

  // Show plots
  plt::save("imshow.png");
  std::cout << "Result saved to 'imshow.png'.\n";
}