Grok  7.6.2
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 line_buf
136  {
137  template<typename T>
138  void pre_alloc(mem_fixed_allocator *p, size_t num_ele, ui32 pre_size)
139  {
140  memset(this, 0, sizeof(line_buf));
141  p->pre_alloc_data<T>(num_ele, pre_size);
142  size = num_ele;
143  this->pre_size = pre_size;
144  }
145 
146  template<typename T>
148 
149  template<typename T>
150  void wrap(T *buffer, size_t num_ele, ui32 pre_size);
151 
152  size_t size;
154  union {
156  float* f32;
157  };
158  };
159 
161  struct coded_lists
162  {
164  {
165  next_list = NULL;
167  this->buf = (ui8*)this + sizeof(coded_lists);
168  }
169 
174  };
175 
178  {
179  /*
180  advantage: allocate large chunks of memory
181  */
182 
183  public:
186  { cur_store = store = NULL; total_allocated = 0; }
187 
189  {
190  while (store) {
192  free(store);
193  store = t;
194  }
195  }
196 
197  void get_buffer(ui32 needed_bytes, coded_lists*& p);
198 
199  private:
200  struct stores_list
201  {
203  {
204  this->next_store = NULL;
205  this->available = chunk_size - (ui32)sizeof(stores_list);
206  this->data = (char*)this + sizeof(stores_list);
207  }
210  char* data;
211  };
212 
216  };
217 
218 
219 }
220 
221 
222 #endif // !OJPH_MEM_H
ojph::ui8
uint8_t ui8
Definition: ojph_defs.h:50
ojph::mem_fixed_allocator
Definition: ojph_mem.h:53
ojph::size
Definition: ojph_base.h:48
ojph::mem_fixed_allocator::post_alloc_local
T * post_alloc_local(size_t num_ele, ui32 pre_size, size_t &avail_sz, void *&avail_p)
Definition: ojph_mem.h:115
ojph::mem_fixed_allocator::avail_size_obj
size_t avail_size_obj
Definition: ojph_mem.h:131
ojph::mem_fixed_allocator::size_obj
size_t size_obj
Definition: ojph_mem.h:131
ojph::line_buf::size
size_t size
Definition: ojph_mem.h:152
ojph::mem_elastic_allocator::store
stores_list * store
Definition: ojph_mem.h:213
ojph::mem_fixed_allocator::~mem_fixed_allocator
~mem_fixed_allocator()
Definition: ojph_mem.h:60
ojph::mem_fixed_allocator::store
void * store
Definition: ojph_mem.h:130
ojph::mem_fixed_allocator::avail_size_data
size_t avail_size_data
Definition: ojph_mem.h:131
ojph::line_buf
Definition: ojph_mem.h:136
ojph::mem_fixed_allocator::pre_alloc_obj
void pre_alloc_obj(size_t num_ele)
Definition: ojph_mem.h:72
ojph::mem_fixed_allocator::post_alloc_obj
T * post_alloc_obj(size_t num_ele)
Definition: ojph_mem.h:96
ojph::coded_lists::buf_size
ui32 buf_size
Definition: ojph_mem.h:171
ojph::mem_fixed_allocator::size_data
size_t size_data
Definition: ojph_mem.h:131
ojph::mem_elastic_allocator::mem_elastic_allocator
mem_elastic_allocator(ui32 chunk_size)
Definition: ojph_mem.h:184
ojph::coded_lists::next_list
coded_lists * next_list
Definition: ojph_mem.h:170
ojph_mem.h
ojph::coded_lists::avail_size
ui32 avail_size
Definition: ojph_mem.h:172
ojph::mem_elastic_allocator::stores_list
Definition: ojph_mem.h:201
ojph::mem_elastic_allocator::~mem_elastic_allocator
~mem_elastic_allocator()
Definition: ojph_mem.h:188
ojph::ui32
uint32_t ui32
Definition: ojph_defs.h:54
ojph::mem_fixed_allocator::avail_obj
void * avail_obj
Definition: ojph_mem.h:130
ojph::line_buf::pre_alloc
void pre_alloc(mem_fixed_allocator *p, size_t num_ele, ui32 pre_size)
Definition: ojph_mem.h:138
ojph::line_buf::f32
float * f32
Definition: ojph_mem.h:156
ojph::coded_lists::buf
ui8 * buf
Definition: ojph_mem.h:173
ojph::coded_lists::coded_lists
coded_lists(ui32 size)
Definition: ojph_mem.h:163
ojph::mem_fixed_allocator::pre_alloc_data
void pre_alloc_data(size_t num_ele, ui32 pre_size)
Definition: ojph_mem.h:66
ojph::mem_elastic_allocator::get_buffer
void get_buffer(ui32 needed_bytes, coded_lists *&p)
Definition: ojph_mem.cpp:95
ojph::mem_elastic_allocator::stores_list::data
char * data
Definition: ojph_mem.h:210
ojph::line_buf::i32
si32 * i32
Definition: ojph_mem.h:155
ojph::coded_lists
Definition: ojph_mem.h:162
ojph::mem_elastic_allocator::cur_store
stores_list * cur_store
Definition: ojph_mem.h:213
ojph::si32
int32_t si32
Definition: ojph_defs.h:55
ojph::mem_elastic_allocator::chunk_size
const ui32 chunk_size
Definition: ojph_mem.h:215
ojph::line_buf::wrap
void wrap(T *buffer, size_t num_ele, ui32 pre_size)
ojph::mem_elastic_allocator::total_allocated
size_t total_allocated
Definition: ojph_mem.h:214
ojph_max
#define ojph_max(a, b)
Definition: ojph_defs.h:73
ojph_arch.h
ojph::mem_elastic_allocator::stores_list::next_store
stores_list * next_store
Definition: ojph_mem.h:208
ojph::mem_fixed_allocator::avail_data
void * avail_data
Definition: ojph_mem.h:130
ojph::mem_fixed_allocator::post_alloc_data
T * post_alloc_data(size_t num_ele, ui32 pre_size)
Definition: ojph_mem.h:89
ojph::mem_fixed_allocator::mem_fixed_allocator
mem_fixed_allocator()
Definition: ojph_mem.h:55
ojph::mem_elastic_allocator
Definition: ojph_mem.h:178
ojph::mem_fixed_allocator::pre_alloc_local
void pre_alloc_local(size_t num_ele, ui32 pre_size, size_t &sz)
Definition: ojph_mem.h:104
ojph::mem_elastic_allocator::stores_list::stores_list
stores_list(ui32 chunk_size)
Definition: ojph_mem.h:202
ojph
Definition: ojph_block_decoder.cpp:49
ojph::mem_elastic_allocator::stores_list::available
ui32 available
Definition: ojph_mem.h:209
ojph::line_buf::pre_size
ui32 pre_size
Definition: ojph_mem.h:153
ojph::mem_fixed_allocator::alloc
void alloc()
Definition: ojph_mem.h:77
ojph::line_buf::finalize_alloc
void finalize_alloc(mem_fixed_allocator *p)