output.hpp

Go to the documentation of this file.

#include <sys/stat.h> 
#include <petscmat.h>
#include <iostream> 
#include "config.hpp"
#include "mastereq.hpp"
#pragma once
 
class Output{
  protected:
 
  int mpirank_world; 
  int mpirank_petsc; 
  int mpisize_petsc; 
  int mpirank_init; 
 
  bool quietmode; 
  
  FILE* optimfile; 
  int output_timestep_stride; 
  std::vector<OutputType> output_observables; 
 
  size_t noscillators; 
  bool writeFullState; 
  bool writeExpectedEnergy; 
  bool writeExpectedEnergy_comp; 
  bool writePopulation; 
  bool writePopulation_comp; 
  FILE *ufile; 
  FILE *vfile; 
  std::vector<FILE *>expectedfile; 
  std::vector<FILE *>populationfile; 
  FILE *expectedfile_comp; 
  FILE *populationfile_comp; 
 
  // VecScatter scat; ///< PETSc's scatter context for state communication across cores
  // Vec xseq; ///< Sequential vector for I/O operations
 
  public:
    std::string output_dir; 
    int output_optimization_stride; 
 
  public:
    Output();
 
    Output(const Config& config, MPI_Comm comm_petsc, MPI_Comm comm_init, bool quietmode=false);
 
    ~Output();
 
    void writeOptimFile(int optim_iter, double objective, double gnorm, double stepsize, double Favg, double cost, double tikh_regul,  double penalty_leakage, double penalty_dpdm, double penalty_energy, double penalty_variation, double penalty_weightedcost);
 
    void writeControls(Vec params, MasterEq* mastereq, int ntime, double dt);
 
    void writeGradient(Vec grad);
 
    void openTrajectoryDataFiles(std::string prefix, int initid);
 
    void writeTrajectoryDataFiles(int timestep, double time, const Vec state, MasterEq* mastereq);
 
    void closeTrajectoryDataFiles();
 
};