NetCDFArchive.h

Go to the documentation of this file.

//===-- serialbox/core/archive/NetCDFArchive.h --------------------------------------*- C++ -*-===//
//
//                                    S E R I A L B O X
//
// This file is distributed under terms of BSD license.
// See LICENSE.txt for more information
//
//===------------------------------------------------------------------------------------------===//
//
//===------------------------------------------------------------------------------------------===//

#ifndef SERIALBOX_CORE_ARCHIVE_NETCDFARCHIVE_H
#define SERIALBOX_CORE_ARCHIVE_NETCDFARCHIVE_H

#include "serialbox/core/Compiler.h"
#ifdef SERIALBOX_HAS_NETCDF

#include "serialbox/core/Filesystem.h"
#include "serialbox/core/Json.h"
#include "serialbox/core/archive/Archive.h"
#include <string>
#include <unordered_map>
#include <vector>

namespace serialbox {

class NetCDFArchive : public Archive {
public:
  static const std::string Name;

  static const int Version;

  NetCDFArchive(OpenModeKind mode, const std::string& directory, const std::string& prefix);

  void readMetaDataFromJson();

  void writeMetaDataToJson();

  virtual FieldID write(const StorageView& storageView, const std::string& fieldID,
                        const std::shared_ptr<FieldMetainfoImpl> info) override;

  virtual void read(StorageView& storageView, const FieldID& fieldID,
                    std::shared_ptr<FieldMetainfoImpl> info) const override;

  virtual void updateMetaData() override;

  virtual OpenModeKind mode() const override { return mode_; }

  virtual std::string directory() const override { return directory_.string(); }

  virtual std::string prefix() const override { return prefix_; }

  virtual std::string name() const override { return NetCDFArchive::Name; }

  virtual std::string metaDataFile() const override { return metaDatafile_.string(); }

  virtual std::ostream& toStream(std::ostream& stream) const override;

  virtual void clear() override;

  virtual bool isReadingThreadSafe() const override { return false; }

  virtual bool isWritingThreadSafe() const override { return false; }

  virtual bool isSlicedReadingSupported() const override { return false; }


  static std::unique_ptr<Archive> create(OpenModeKind mode, const std::string& directory,
                                         const std::string& prefix);

  static void writeToFile(std::string filename, const StorageView& storageView,
                          const std::string& field);

  static void readFromFile(std::string filename, StorageView& storageView,
                           const std::string& field);

private:
  OpenModeKind mode_;
  filesystem::path directory_;
  std::string prefix_;
  filesystem::path metaDatafile_;

  std::unordered_map<std::string, int> fieldMap_;
  json::json json_;
};

} // namespace serialbox

#endif // SERIALBOX_HAS_NETCDF

#endif