Squashed 'rtengine/libraw/' content from commit cccb97647

git-subtree-dir: rtengine/libraw
git-subtree-split: cccb97647fcee56801fa68231fa8a38aa8b52ef7

libraw/libraw_alloc.h

@@ -0,0 +1,148 @@
+/* -*- C++ -*-
+ * File: libraw_alloc.h
+ * Copyright 2008-2021 LibRaw LLC (info@libraw.org)
+ * Created: Sat Mar  22, 2008
+ *
+ * LibRaw C++ interface
+ *
+LibRaw is free software; you can redistribute it and/or modify
+it under the terms of the one of two licenses as you choose:
+
+1. GNU LESSER GENERAL PUBLIC LICENSE version 2.1
+   (See file LICENSE.LGPL provided in LibRaw distribution archive for details).
+
+2. COMMON DEVELOPMENT AND DISTRIBUTION LICENSE (CDDL) Version 1.0
+   (See file LICENSE.CDDL provided in LibRaw distribution archive for details).
+
+ */
+
+#ifndef __LIBRAW_ALLOC_H
+#define __LIBRAW_ALLOC_H
+
+#include <stdlib.h>
+#include <string.h>
+#include "libraw_const.h"
+
+#ifdef __cplusplus
+
+#define LIBRAW_MSIZE 512
+
+class DllDef libraw_memmgr
+{
+public:
+  libraw_memmgr(unsigned ee) : extra_bytes(ee)
+  {
+    size_t alloc_sz = LIBRAW_MSIZE * sizeof(void *);
+    mems = (void **)::malloc(alloc_sz);
+    memset(mems, 0, alloc_sz);
+  }
+  ~libraw_memmgr()
+  {
+    cleanup();
+    ::free(mems);
+  }
+  void *malloc(size_t sz)
+  {
+#ifdef LIBRAW_USE_CALLOC_INSTEAD_OF_MALLOC
+    void *ptr = ::calloc(sz + extra_bytes, 1);
+#else
+    void *ptr = ::malloc(sz + extra_bytes);
+#endif
+    mem_ptr(ptr);
+    return ptr;
+  }
+  void *calloc(size_t n, size_t sz)
+  {
+    void *ptr = ::calloc(n + (extra_bytes + sz - 1) / (sz ? sz : 1), sz);
+    mem_ptr(ptr);
+    return ptr;
+  }
+  void *realloc(void *ptr, size_t newsz)
+  {
+    void *ret = ::realloc(ptr, newsz + extra_bytes);
+    forget_ptr(ptr);
+    mem_ptr(ret);
+    return ret;
+  }
+  void free(void *ptr)
+  {
+    forget_ptr(ptr);
+    ::free(ptr);
+  }
+  void cleanup(void)
+  {
+    for (int i = 0; i < LIBRAW_MSIZE; i++)
+      if (mems[i])
+      {
+        ::free(mems[i]);
+        mems[i] = NULL;
+      }
+  }
+
+private:
+  void **mems;
+  unsigned extra_bytes;
+  void mem_ptr(void *ptr)
+  {
+#if defined(LIBRAW_USE_OPENMP)
+      bool ok = false; /* do not return from critical section */
+#endif
+
+#if defined(LIBRAW_USE_OPENMP)
+#pragma omp critical
+      {
+#endif
+          if (ptr)
+          {
+              for (int i = 0; i < LIBRAW_MSIZE - 1; i++)
+                  if (!mems[i])
+                  {
+                      mems[i] = ptr;
+#if defined(LIBRAW_USE_OPENMP)
+		      ok = true;
+		      break;
+#else
+                      return;
+#endif
+                  }
+#ifdef LIBRAW_MEMPOOL_CHECK
+#if !defined(LIBRAW_USE_OPENMP)
+              /* remember ptr in last mems item to be free'ed at cleanup */
+              if (!mems[LIBRAW_MSIZE - 1])
+                  mems[LIBRAW_MSIZE - 1] = ptr;
+              throw LIBRAW_EXCEPTION_MEMPOOL;
+#endif
+#endif
+          }
+#if defined(LIBRAW_USE_OPENMP)
+      }
+      if(!ok)
+      {
+          if (!mems[LIBRAW_MSIZE - 1])
+              mems[LIBRAW_MSIZE - 1] = ptr;
+          throw LIBRAW_EXCEPTION_MEMPOOL;
+      }
+#endif
+  }
+  void forget_ptr(void *ptr)
+  {
+#if defined(LIBRAW_USE_OPENMP)
+#pragma omp critical
+    {
+#endif
+     if (ptr)
+      for (int i = 0; i < LIBRAW_MSIZE; i++)
+        if (mems[i] == ptr)
+        {
+          mems[i] = NULL;
+          break;
+        }
+#if defined(LIBRAW_USE_OPENMP)
+    }
+#endif
+  }
+};
+
+#endif /* C++ */
+
+#endif