summaryrefslogtreecommitdiffstats
path: root/third_party/jpeg-xl/lib/jpegli/memory_manager.h
blob: 238f85a3084c517253e9c26220062b79adcacd2c (plain)
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
// Copyright (c) the JPEG XL Project Authors. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

#ifndef LIB_JPEGLI_MEMORY_MANAGER_H_
#define LIB_JPEGLI_MEMORY_MANAGER_H_

/* clang-format off */
#include <stdio.h>
#include <jpeglib.h>
#include <stdlib.h>
/* clang-format on */

#define JPOOL_PERMANENT_ALIGNED (JPOOL_NUMPOOLS + JPOOL_PERMANENT)
#define JPOOL_IMAGE_ALIGNED (JPOOL_NUMPOOLS + JPOOL_IMAGE)

namespace jpegli {

void InitMemoryManager(j_common_ptr cinfo);

template <typename T>
T* Allocate(j_common_ptr cinfo, size_t len, int pool_id = JPOOL_PERMANENT) {
  void* p = (*cinfo->mem->alloc_small)(cinfo, pool_id, len * sizeof(T));
  return reinterpret_cast<T*>(p);
}

template <typename T>
T* Allocate(j_decompress_ptr cinfo, size_t len, int pool_id = JPOOL_PERMANENT) {
  return Allocate<T>(reinterpret_cast<j_common_ptr>(cinfo), len, pool_id);
}

template <typename T>
T* Allocate(j_compress_ptr cinfo, size_t len, int pool_id = JPOOL_PERMANENT) {
  return Allocate<T>(reinterpret_cast<j_common_ptr>(cinfo), len, pool_id);
}

}  // namespace jpegli

#endif  // LIB_JPEGLI_MEMORY_MANAGER_H_