Grok  9.7.5
ojph_mem.h
Go to the documentation of this file.
1 //***************************************************************************/
2 // This software is released under the 2-Clause BSD license, included
3 // below.
4 //
5 // Copyright (c) 2019, Aous Naman
6 // Copyright (c) 2019, Kakadu Software Pty Ltd, Australia
7 // Copyright (c) 2019, The University of New South Wales, Australia
8 //
9 // Redistribution and use in source and binary forms, with or without
10 // modification, are permitted provided that the following conditions are
11 // met:
12 //
13 // 1. Redistributions of source code must retain the above copyright
14 // notice, this list of conditions and the following disclaimer.
15 //
16 // 2. Redistributions in binary form must reproduce the above copyright
17 // notice, this list of conditions and the following disclaimer in the
18 // documentation and/or other materials provided with the distribution.
19 //
20 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
21 // IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
22 // TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
23 // PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24 // HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
26 // TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
27 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
28 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
29 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
30 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 //***************************************************************************/
32 // This file is part of the OpenJPH software implementation.
33 // File: ojph_mem.h
34 // Author: Aous Naman
35 // Date: 28 August 2019
36 //***************************************************************************/
37 
38 
39 #ifndef OJPH_MEM_H
40 #define OJPH_MEM_H
41 
42 #include <cstdlib>
43 #include <cassert>
44 #include <cstring>
45 #include <type_traits>
46 
47 #include "ojph_arch.h"
48 
49 namespace ojph {
50 
53  {
54  public:
56  {
57  avail_obj = avail_data = store = NULL;
59  }
61  {
62  if (store) free(store);
63  }
64 
65  template<typename T>
66  void pre_alloc_data(size_t num_ele, ui32 pre_size)
67  {
68  pre_alloc_local<T, byte_alignment>(num_ele, pre_size, size_data);
69  }
70 
71  template<typename T>
72  void pre_alloc_obj(size_t num_ele)
73  {
74  pre_alloc_local<T, object_alignment>(num_ele, 0, size_obj);
75  }
76 
77  void alloc()
78  {
79  assert(store == NULL);
80  avail_obj = store = malloc(size_data + size_obj);
82  if (store == NULL)
83  throw "malloc failed";
86  }
87 
88  template<typename T>
89  T* post_alloc_data(size_t num_ele, ui32 pre_size)
90  {
91  return post_alloc_local<T, byte_alignment>
92  (num_ele, pre_size, avail_size_data, avail_data);
93  }
94 
95  template<typename T>
96  T* post_alloc_obj(size_t num_ele)
97  {
98  return post_alloc_local<T, object_alignment>
99  (num_ele, 0, avail_size_obj, avail_obj);
100  }
101 
102  private:
103  template<typename T, int N>
104  void pre_alloc_local(size_t num_ele, ui32 pre_size, size_t& sz)
105  {
106  assert(store == NULL);
107  num_ele = calc_aligned_size<T, N>(num_ele);
108  size_t total = (num_ele + pre_size) * sizeof(T);
109  total += 2*N - 1;
110 
111  sz += total;
112  }
113 
114  template<typename T, int N>
115  T* post_alloc_local(size_t num_ele, ui32 pre_size,
116  size_t& avail_sz, void*& avail_p)
117  {
118  assert(store != NULL);
119  num_ele = calc_aligned_size<T, N>(num_ele);
120  size_t total = (num_ele + pre_size) * sizeof(T);
121  total += 2*N - 1;
122 
123  T* p = align_ptr<T, N>((T*)avail_p + pre_size);
124  avail_p = (ui8*)avail_p + total;
125  avail_sz -= total;
126  assert((avail_sz & 0x8000000000000000llu) == 0);
127  return p;
128  }
129 
132  };
133 
135  struct coded_lists
136  {
138  {
139  next_list = NULL;
140  avail_size = buf_size = size;
141  this->buf = (ui8*)this + sizeof(coded_lists);
142  }
143 
148  };
149 
152  {
153  /*
154  advantage: allocate large chunks of memory
155  */
156 
157  public:
160  { cur_store = store = NULL; total_allocated = 0; }
161 
163  {
164  while (store) {
166  free(store);
167  store = t;
168  }
169  }
170 
171  void get_buffer(ui32 needed_bytes, coded_lists*& p);
172 
173  private:
174  struct stores_list
175  {
177  {
178  this->next_store = NULL;
179  this->available = chunk_size - (ui32)sizeof(stores_list);
180  this->data = (char*)this + sizeof(stores_list);
181  }
184  char* data;
185  };
186 
190  };
191 
192 
193 }
194 
195 
196 #endif // !OJPH_MEM_H
Definition: ojph_mem.h:152
const ui32 chunk_size
Definition: ojph_mem.h:189
void get_buffer(ui32 needed_bytes, coded_lists *&p)
Definition: ojph_mem.cpp:54
stores_list * store
Definition: ojph_mem.h:187
~mem_elastic_allocator()
Definition: ojph_mem.h:162
mem_elastic_allocator(ui32 chunk_size)
Definition: ojph_mem.h:158
stores_list * cur_store
Definition: ojph_mem.h:187
size_t total_allocated
Definition: ojph_mem.h:188
Definition: ojph_mem.h:53
void pre_alloc_data(size_t num_ele, ui32 pre_size)
Definition: ojph_mem.h:66
void alloc()
Definition: ojph_mem.h:77
void pre_alloc_obj(size_t num_ele)
Definition: ojph_mem.h:72
void * avail_obj
Definition: ojph_mem.h:130
void * store
Definition: ojph_mem.h:130
void pre_alloc_local(size_t num_ele, ui32 pre_size, size_t &sz)
Definition: ojph_mem.h:104
size_t size_data
Definition: ojph_mem.h:131
T * post_alloc_obj(size_t num_ele)
Definition: ojph_mem.h:96
T * post_alloc_local(size_t num_ele, ui32 pre_size, size_t &avail_sz, void *&avail_p)
Definition: ojph_mem.h:115
T * post_alloc_data(size_t num_ele, ui32 pre_size)
Definition: ojph_mem.h:89
void * avail_data
Definition: ojph_mem.h:130
size_t avail_size_data
Definition: ojph_mem.h:131
size_t avail_size_obj
Definition: ojph_mem.h:131
~mem_fixed_allocator()
Definition: ojph_mem.h:60
mem_fixed_allocator()
Definition: ojph_mem.h:55
size_t size_obj
Definition: ojph_mem.h:131
N
Definition: rvv-inl.h:1656
Definition: ojph_block_decoder.cpp:50
uint32_t ui32
Definition: ojph_defs.h:50
uint8_t ui8
Definition: ojph_defs.h:46
Definition: ojph_mem.h:136
ui32 buf_size
Definition: ojph_mem.h:145
coded_lists(ui32 size)
Definition: ojph_mem.h:137
ui8 * buf
Definition: ojph_mem.h:147
coded_lists * next_list
Definition: ojph_mem.h:144
ui32 avail_size
Definition: ojph_mem.h:146
ui32 available
Definition: ojph_mem.h:183
char * data
Definition: ojph_mem.h:184
stores_list * next_store
Definition: ojph_mem.h:182
stores_list(ui32 chunk_size)
Definition: ojph_mem.h:176