#ifndef SAVE_RESULTS_H
#define SAVE_RESULTS_H

#include "nufi/grids.h"
#include <deal.II/lac/vector.h>
#include <string>
#include <vector>

class NuFISolver;

struct DiagnosticsSnapshot {
  unsigned int Nx = 0;
  unsigned int Nv = 0;
  std::vector<double> x_eval;
  std::vector<double> v_eval;
  std::vector<double> f;
  std::vector<double> rho;
  std::vector<double> E;
};

DiagnosticsSnapshot
compute_diagnostics(const NuFISolver &solver, unsigned int n,
                    std::vector<GridStructure<1>> &grid_struct,
                    std::vector<SolutionSnapshot<1>> &phi_history,
                    unsigned int Nx_out, unsigned int Nv_out);

void save_f(const DiagnosticsSnapshot &snap, const std::string &filepath);
void save_rho(const DiagnosticsSnapshot &snap, const std::string &filepath);
void save_Efield(const DiagnosticsSnapshot &snap, const std::string &filepath);
double compute_int_E_squared(const DiagnosticsSnapshot &snap);

void save_time_series(const std::vector<double> &t,
                      const std::vector<double> &values,
                      const std::string &filepath);

#endif