blosc2-sys 0.4.0+2.15.2

Bindings to C Blosc2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79

/*********************************************************************
  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.
**********************************************************************/

# include <b2nd.h>

int main() {

  blosc2_init();

  int8_t ndim = 2;
  int64_t shape[] = {10, 10};
  int32_t chunkshape[] = {4, 4};
  int32_t blockshape[] = {2, 2};
  int32_t typesize = 8;

  int64_t nelem = 1;
  for (int i = 0; i < ndim; ++i) {
    nelem *= shape[i];
  }
  int64_t size = nelem * typesize;
  double *data = malloc(size);

  for (int i = 0; i < nelem; ++i) {
    data[i] = i;
  }

  blosc2_cparams cparams = BLOSC2_CPARAMS_DEFAULTS;
  cparams.typesize = typesize;
  blosc2_dparams dparams = BLOSC2_DPARAMS_DEFAULTS;
  blosc2_storage b2_storage = {.cparams=&cparams, .dparams=&dparams};
  b2_storage.contiguous = false;

  b2nd_context_t *ctx = b2nd_create_ctx(&b2_storage, ndim, shape, chunkshape, blockshape, NULL, 0,
                                        NULL, 0);

  b2nd_array_t *arr;
  BLOSC_ERROR(b2nd_from_cbuffer(ctx, &arr, data, size));

  uint8_t *cframe;
  int64_t cframe_len;
  bool needs_free;
  BLOSC_ERROR(b2nd_to_cframe(arr, &cframe, &cframe_len, &needs_free));

  b2nd_array_t *dest;
  BLOSC_ERROR(b2nd_from_cframe(cframe, cframe_len, true, &dest));

  if (needs_free) {
    free(cframe);
  }

  /* Fill dest array with b2nd_array_t data */
  uint8_t *data_dest = malloc(size);
  BLOSC_ERROR(b2nd_to_cbuffer(dest, data_dest, size));

  for (int i = 0; i < nelem; ++i) {
    if (data[i] != data_dest[i] && data[i] != i) {
      return -1;
    }
  }

  /* Free mallocs */
  free(data);
  free(data_dest);

  BLOSC_ERROR(b2nd_free(arr));
  BLOSC_ERROR(b2nd_free(dest));
  BLOSC_ERROR(b2nd_free_ctx(ctx));

  blosc2_destroy();

  return 0;
}