#define _USE_MATH_DEFINES

#include <iostream>

#include <cmath>

#include "../matplotlibcpp.h"

namespace plt = matplotlibcpp;

int main()

{

// Prepare data.

std::vector<std::vector<double>> events;

events.push_back({2,5,10,25});

events.push_back({20,21,22,24});

events.push_back({4,8,12,14,16,17,18});

events.push_back({10,11,12,13.5});

// Set the size of output image = 1200x780 pixels

plt::figure_size(1200, 780);

std::map<std::string, std::string> keywords;

keywords["linewidths"] = "0.5";

keywords["colors"] = "black";

plt::eventplot(events, keywords);

// Add graph title

plt::title("Event plot");

plt::xlabel("Time in ms");

plt::ylabel("Neuron ID");

plt::show();

// save figure

const char* filename = "./events.png";

std::cout << "Saving result to " << filename << std::endl;;

plt::save(filename);

}