Archive.h

Go to the documentation of this file.

// Copyright Contributors to the OpenVDB Project
// SPDX-License-Identifier: Apache-2.0
 
#ifndef OPENVDB_IO_ARCHIVE_HAS_BEEN_INCLUDED
#define OPENVDB_IO_ARCHIVE_HAS_BEEN_INCLUDED
 
#include <openvdb/version.h>
#include <openvdb/Grid.h>
#include <openvdb/MetaMap.h>
#include <openvdb/Platform.h>
#include <openvdb/version.h> // for VersionId
 
#include "Codec.h"
#include "Compression.h" // for COMPRESS_ZIP, etc.
#include "io.h" // for StreamMetadata
 
#include <cstdint>
#include <iosfwd>
#include <map>
#include <memory>
#include <string>
 
 
class TestFile;
 
namespace openvdb {
OPENVDB_USE_VERSION_NAMESPACE
namespace OPENVDB_VERSION_NAME {
namespace io {
 
class GridDescriptor;
 
 
/// Grid serializer/unserializer
class OPENVDB_API Archive
{
public:
    using Ptr = SharedPtr<Archive>;
    using ConstPtr = SharedPtr<const Archive>;
 
    static const uint32_t DEFAULT_COMPRESSION_FLAGS;
 
    Archive();
    Archive(const Archive&) = default;
    Archive& operator=(const Archive&) = default;
    virtual ~Archive();
 
    /// @brief Return a copy of this archive.
    virtual Ptr copy() const;
 
    /// @brief Return the UUID that was most recently written (or read,
    /// if no UUID has been written yet).
    std::string getUniqueTag() const;
    /// @brief Return @c true if the given UUID matches this archive's UUID.
    bool isIdentical(const std::string& uuidStr) const;
 
    /// @brief Return the file format version number of the input stream.
    uint32_t fileVersion() const { return mFileVersion; }
    /// @brief Return the (major, minor) version number of the library that was
    /// used to write the input stream.
    VersionId libraryVersion() const { return mLibraryVersion; }
    /// @brief Return a string of the form "<major>.<minor>/<format>", giving the
    /// library and file format version numbers associated with the input stream.
    std::string version() const;
 
    /// @brief Return @c true if trees shared by multiple grids are written out
    /// only once, @c false if they are written out once per grid.
    bool isInstancingEnabled() const { return mEnableInstancing; }
    /// @brief Specify whether trees shared by multiple grids should be
    /// written out only once (@c true) or once per grid (@c false).
    /// @note Instancing is enabled by default.
    void setInstancingEnabled(bool b) { mEnableInstancing = b; }
 
    /// Return @c true if the OpenVDB library includes support for the Blosc compressor.
    /// @note Blosc is always built in, so this always returns @c true.
    static bool hasBloscCompression();
 
    /// Return @c true if the OpenVDB library includes support for the ZLib compressor.
    static bool hasZLibCompression();
 
    /// Return a bit mask specifying compression options for the data stream.
    uint32_t compression() const { return mCompression; }
    /// @brief Specify whether and how the data stream should be compressed.
    /// @param c bitwise OR (e.g., COMPRESS_ZIP | COMPRESS_ACTIVE_MASK) of
    ///     compression option flags (see Compression.h for the available flags)
    /// @note Not all combinations of compression options are supported.
    void setCompression(uint32_t c) { mCompression = c; }
 
    /// @brief Return @c true if grid statistics (active voxel count and
    /// bounding box, etc.) are computed and written as grid metadata.
    bool isGridStatsMetadataEnabled() const { return mEnableGridStats; }
    /// @brief Specify whether grid statistics (active voxel count and
    /// bounding box, etc.) should be computed and written as grid metadata.
    void setGridStatsMetadataEnabled(bool b) { mEnableGridStats = b; }
 
    /// @brief Write the grids in the given container to this archive's output stream.
    virtual void write(const GridCPtrVec&, const MetaMap& = MetaMap(),
        const io::WriteOptions& = io::WriteOptions{}) const {}
 
    /// @brief Return @c false (delayed loading has been removed).
    static bool isDelayedLoadingEnabled() { return false; }
 
    /// @brief Enable collection of read diagnostics (warnings, etc.) during I/O operations.
    void enableReadDiagnostics();
 
    /// @brief Disable collection of read diagnostics.
    void disableReadDiagnostics();
 
    /// @brief Return a const reference to the diagnostics collector.
    const ReadDiagnostics& readDiagnostics() const;
 
    /// @brief Clear the diagnostics list while keeping collection active.
    void clearReadDiagnostics();
 
protected:
    /// @brief Return @c true if the input stream contains grid offsets
    /// that allow for random access or partial reading.
    bool inputHasGridOffsets() const { return mInputHasGridOffsets; }
    void setInputHasGridOffsets(bool b) { mInputHasGridOffsets = b; }
 
    /// @brief Tag the given input stream with the input file format version number.
    ///
    /// The tag can be retrieved with getFormatVersion().
    /// @sa getFormatVersion()
    void setFormatVersion(std::istream&);
 
    /// @brief Tag the given input stream with the version number of
    /// the library with which the input stream was created.
    ///
    /// The tag can be retrieved with getLibraryVersion().
    /// @sa getLibraryVersion()
    void setLibraryVersion(std::istream&);
 
    /// @brief Tag the given input stream with flags indicating whether
    /// the input stream contains compressed data and how it is compressed.
    void setDataCompression(std::istream&);
 
    /// @brief Tag an output stream with flags specifying only those
    /// compression options that are applicable to the given grid.
    void setGridCompression(std::ostream&, const GridBase&) const;
    /// @brief Read in the compression flags for a grid and
    /// tag the given input stream with those flags.
    static void readGridCompression(std::istream&);
 
    /// Read in and return the number of grids on the input stream.
    static int32_t readGridCount(std::istream&);
 
    /// @brief Find the codec for the given grid type and options.
    static io::Codec* findCodec(const std::string& gridType, const io::ReadOptions& options = io::ReadOptions{});
 
    /// @brief Read in and create the grid represented by the given grid descriptor using the
    /// given input stream, using the provided options if given.
    static GridBase::Ptr readGrid(const GridDescriptor&, std::istream&,
        const io::ReadOptions& readOptions, ReadDiagnostics& diagnostics);
    static GridBase::Ptr readGrid(const GridDescriptor&, std::istream&,
        const io::ReadOptions& readOptions = io::ReadOptions{});
 
    using NamedGridMap = std::map<Name /*uniqueName*/, GridBase::Ptr>;
 
    /// @brief If the grid represented by the given grid descriptor
    /// is an instance, connect it with its instance parent.
    void connectInstance(const GridDescriptor&, const NamedGridMap&) const;
 
    /// Write the given grid descriptor and grid to an output stream
    /// and update the GridDescriptor offsets.
    /// @param seekable  if true, the output stream supports seek operations
    /// @param writeOptions options controlling how grid data is written
    void writeGrid(GridDescriptor&, GridBase::ConstPtr, std::ostream&, bool seekable,
        const io::WriteOptions& writeOptions = io::WriteOptions{}) const;
    /// Write the given grid descriptor and grid metadata to an output stream
    /// and update the GridDescriptor offsets, but don't write the grid's tree,
    /// since it is shared with another grid.
    /// @param seekable  if true, the output stream supports seek operations
    /// @param writeOptions options controlling how grid data is written
    void writeGridInstance(GridDescriptor&, GridBase::ConstPtr,
        std::ostream&, bool seekable, const io::WriteOptions& writeOptions = io::WriteOptions{}) const;
 
    /// @brief Read the magic number, version numbers, UUID, etc. from the given input stream.
    /// @return @c true if the input UUID differs from the previously-read UUID.
    bool readHeader(std::istream&);
    /// @brief Write the magic number, version numbers, UUID, etc. to the given output stream.
    /// @param seekable  if true, the output stream supports seek operations
    /// @todo This method should not be const since it actually redefines the UUID!
    void writeHeader(std::ostream&, bool seekable) const;
 
    //@{
    /// Write the given grids to an output stream.
    void write(std::ostream&, const GridPtrVec&, bool seekable, const MetaMap&,
        const io::WriteOptions& writeOptions = io::WriteOptions{}) const;
    void write(std::ostream&, const GridCPtrVec&, bool seekable, const MetaMap&,
        const io::WriteOptions& writeOptions = io::WriteOptions{}) const;
    //@}
 
    /// Diagnostics collector for read operations (always valid; enabled/disabled via flag)
    mutable ReadDiagnostics mReadDiagnostics;
 
private:
    friend class ::TestFile;
 
    /// The version of the file that was read
    uint32_t mFileVersion;
    /// The version of the library that was used to create the file that was read
    VersionId mLibraryVersion;
    /// Unique tag, a random 16-byte (128-bit) value, stored as a string format.
    mutable std::string mUuid;// needs to be mutable since writeHeader is const!
    /// Flag indicating whether the input stream contains grid offsets
    /// and therefore supports partial reading
    bool mInputHasGridOffsets;
    /// Flag indicating whether a tree shared by multiple grids should be
    /// written out only once (true) or once per grid (false)
    bool mEnableInstancing;
    /// Flags indicating whether and how the data stream is compressed
    uint32_t mCompression;
    /// Flag indicating whether grid statistics metadata should be written
    bool mEnableGridStats;
}; // class Archive
 
} // namespace io
} // namespace OPENVDB_VERSION_NAME
} // namespace openvdb
 
#endif // OPENVDB_IO_ARCHIVE_HAS_BEEN_INCLUDED