TrajectoryDatabase.h

Go to the documentation of this file.

// This file is part of the Peano project. For conditions of distribution and
// use, please see the copyright notice at www.peano-framework.org
#pragma once
 
#include <iomanip>
#include <list>
#include <map>
#include <string>
 
#include "peano4/utils/Globals.h"
#include "tarch/la/Vector.h"
#include "tarch/logging/Log.h"
#include "tarch/multicore/BooleanSemaphore.h"
 
namespace toolbox {
  namespace particles {
    class TrajectoryDatabase;
  } // namespace particles
} // namespace toolbox
 
class toolbox::particles::TrajectoryDatabase {
private:
  static tarch::logging::Log _log;
 
  std::string _fileName;
  double      _dataDelta;
  double      _positionDelta;
  double      _maxDataDelta;
  double      _maxPositionDelta;
  int         _numberOfDataPointsPerParticle;
 
  double _timeDelta;
 
  const int _deltaBetweenTwoDatabaseFlushes;
  int       _thresholdForNextDatabaseFlush;
  int       _precision;
 
  bool _clearDatabaseAfterFlush;
 
  bool _deltasAreRelative;
 
  static constexpr double _deltaCutOffThreshold = 1e-6;
 
  int _rank;
 
  double _maxTimestamp;
 
  struct Entry {
    tarch::la::Vector<Dimensions, double> x;
    double                                timestamp;
    double*                               data;
    const int numberOfDataEntries;
 
    Entry(const Entry&);
    ~Entry();
    Entry(const TrajectoryDatabase& database, const tarch::la::Vector<Dimensions, double>& x_, double timestamp_);
  };
 
  std::map<std::pair<int, int>, std::list<Entry>> _data;
 
  tarch::multicore::BooleanSemaphore _semaphore;
 
  enum class AddSnapshotAction { Ignore, Append, Replace };
 
  AddSnapshotAction getAction(
    const std::pair<int, int>& number, const tarch::la::Vector<Dimensions, double>& x, double timestamp
  );
 
  AddSnapshotAction getAction(
    const std::pair<int, int>&                   number,
    const tarch::la::Vector<Dimensions, double>& x,
    double                                       timestamp,
    int                                          numberOfDataEntries,
    double*                                      data
  );
 
  bool dumpDatabaseSnapshot();
 
public:
  TrajectoryDatabase(
    int    growthBetweenTwoDatabaseFlushes,
    double positionDelta           = 1e-8,
    double dataDelta               = 1e-8,
    double timeDelta               = 0.0,
    bool   clearDatabaseAfterFlush = true,
    bool   deltasAreRelative       = false
  );
  ~TrajectoryDatabase();
 
  void clear(bool lockSemaphore = true);
 
  void clearHistory(bool lockSemaphore = true);
 
  void dumpCSVFile();
 
  void setOutputFileName(const std::string& filename);
  void setOutputPrecision(int precision);
  void setDataDeltaBetweenTwoSnapshots(double value, bool deltasAreRelative = false);
  void setPositionDeltaBetweenTwoSnapshots(double value, bool deltasAreRelative = false);
  void setTimeDeltaBetweenTwoSnapshots(double value);
 
  void clearDatabaseAfterFlush(bool value);
 
  void addParticleSnapshot(
    const std::pair<int, int>& number, double timestamp, const tarch::la::Vector<Dimensions, double>& x
  );
 
  void addParticleSnapshot(int number0, int number1, double timestamp, const tarch::la::Vector<Dimensions, double>& x);
 
  void addParticleSnapshot(
    const std::pair<int, int>&                   number,
    double                                       timestamp,
    const tarch::la::Vector<Dimensions, double>& x,
    int                                          numberOfDataEntries,
    double*                                      data
  );
 
  void addParticleSnapshot(
    int                                          number0,
    int                                          number1,
    double                                       timestamp,
    const tarch::la::Vector<Dimensions, double>& x,
    int                                          numberOfDataEntries,
    double*                                      data
  );
};