/*********************************************************************
  Blosc - Blocked Shuffling and Compression Library

  Copyright (c) 2021  Blosc Development Team <blosc@blosc.org>
  https://blosc.org
  License: BSD 3-Clause (see LICENSE.txt)

  See LICENSE.txt for details about copyright and rights to use.
**********************************************************************/

#ifndef BLOSC_BLOSC2_BLOSC2_STDIO_H
#define BLOSC_BLOSC2_BLOSC2_STDIO_H

#include "blosc2-export.h"

#if defined(_MSC_VER)
#include <io.h>
#else
#include <unistd.h>
#endif

#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <stdbool.h>

#if defined(_WIN32)
#include <windows.h>
#endif

#ifdef __cplusplus
extern "C" {
#endif

typedef struct {
  FILE *file;
} blosc2_stdio_file;

BLOSC_EXPORT void *blosc2_stdio_open(const char *urlpath, const char *mode, void* params);
BLOSC_EXPORT int blosc2_stdio_close(void *stream);
BLOSC_EXPORT int64_t blosc2_stdio_size(void *stream);
BLOSC_EXPORT int64_t blosc2_stdio_write(const void *ptr, int64_t size, int64_t nitems, int64_t position, void *stream);
BLOSC_EXPORT int64_t blosc2_stdio_read(void **ptr, int64_t size, int64_t nitems, int64_t position, void *stream);
BLOSC_EXPORT int blosc2_stdio_truncate(void *stream, int64_t size);
BLOSC_EXPORT int blosc2_stdio_destroy(void* params);


/**
 * @brief Parameters for memory-mapped I/O. You can use the blosc2_schunk_open*_udio functions to memory-map existing
 * schunk files from disk. To create a new schunk which is backed up by a memory-mapped file on disk, set the io member
 * of the #blosc2_storage struct (see test_mmap for examples). Please note that memory-mapped I/O is only available for
 * cframes and not sframes.
 */
typedef struct {
  /* Arguments of the mapping */
  const char* mode;
  //!< The opening mode of the memory-mapped file (r, r+, w+ or c) similar to Numpy's np.memmap
  //!< (https://numpy.org/doc/stable/reference/generated/numpy.memmap.html). Set to r if the file should only be read,
  //!< r+ if you want to extend data to an existing file, w+ to create a new file and c to use an existing file as basis
  //!<  but keep all modifications in-memory. On Windows, the size of the mapping cannot change in the c mode.
  size_t initial_mapping_size;
  //!< The initial size of the memory mapping used as a large enough write buffer for the r+, w+ and c modes (for
  //!< Windows, only the r+ and w+ modes). On Windows, this will also be the size of the file while the file is opened.
  //!< It will be truncated to the target size when the file is closed (e.g., when the schunk is destroyed).
  bool needs_free;
  //!< Indicates whether this object should be freed in the blosc2_destroy_cb callback (set to true if the
  //!< blosc2_stdio_mmap struct was created on the heap).

  /* Internal attributes of the mapping */
  char* addr;
  //!< The starting address of the mapping.
  char* urlpath;
  //!< The path to the file which is associated with this object.
  size_t file_size;
  //!< The size of the file.
  size_t mapping_size;
  //!< The size of the mapping (mapping_size >= file_size).
  bool is_memory_only;
  //!< Whether the mapping is only in-memory and changes are not reflected to the file on disk (c mode).
  FILE* file;
  //!< The underlying file handle.
  int fd;
  //!< The underlying file descriptor.
  int64_t access_flags;
  //!< The access attributes for the memory pages.
  int64_t map_flags;
  //!< The attributes of the mapping.
#if defined(_WIN32)
  HANDLE mmap_handle;
  //!< The Windows handle to the memory mapping.
#endif
} blosc2_stdio_mmap;

/**
 * @brief Default struct for memory-mapped I/O for user initialization.
 */
static const blosc2_stdio_mmap BLOSC2_STDIO_MMAP_DEFAULTS = {
  "r", ((size_t)1 << 30), false, NULL, NULL, 0, 0, false, NULL, -1, -1, -1
#if defined(_WIN32)
  , INVALID_HANDLE_VALUE
#endif
};

BLOSC_EXPORT void *blosc2_stdio_mmap_open(const char *urlpath, const char *mode, void* params);
BLOSC_EXPORT int blosc2_stdio_mmap_close(void *stream);
BLOSC_EXPORT int64_t blosc2_stdio_mmap_size(void *stream);
BLOSC_EXPORT int64_t blosc2_stdio_mmap_write(
  const void *ptr, int64_t size, int64_t nitems, int64_t position, void *stream);
BLOSC_EXPORT int64_t blosc2_stdio_mmap_read(void **ptr, int64_t size, int64_t nitems, int64_t position, void *stream);
BLOSC_EXPORT int blosc2_stdio_mmap_truncate(void *stream, int64_t size);
BLOSC_EXPORT int blosc2_stdio_mmap_destroy(void* params);

#ifdef __cplusplus
}
#endif

#endif /* BLOSC_BLOSC2_BLOSC2_STDIO_H */
