summaryrefslogtreecommitdiffstats
path: root/ext/tarkin/wavelet.c
diff options
context:
space:
mode:
Diffstat (limited to 'ext/tarkin/wavelet.c')
-rw-r--r--ext/tarkin/wavelet.c181
1 files changed, 90 insertions, 91 deletions
diff --git a/ext/tarkin/wavelet.c b/ext/tarkin/wavelet.c
index 9288e15b..cecce80a 100644
--- a/ext/tarkin/wavelet.c
+++ b/ext/tarkin/wavelet.c
@@ -15,88 +15,90 @@
-Wavelet3DBuf* wavelet_3d_buf_new (uint32_t width, uint32_t height,
- uint32_t frames)
+Wavelet3DBuf *
+wavelet_3d_buf_new (uint32_t width, uint32_t height, uint32_t frames)
{
- Wavelet3DBuf* buf = (Wavelet3DBuf*) MALLOC (sizeof (Wavelet3DBuf));
- uint32_t _w = width;
- uint32_t _h = height;
- uint32_t _f = frames;
- int level;
-
- if (!buf)
- return NULL;
-
- buf->data = (TYPE*) MALLOC (width * height * frames * sizeof (TYPE));
-
- if (!buf->data) {
- wavelet_3d_buf_destroy (buf);
- return NULL;
- }
-
- buf->width = width;
- buf->height = height;
- buf->frames = frames;
- buf->scales = 1;
-
- while (_w > 1 || _h > 1 || _f > 1) {
- buf->scales++;
- _w = (_w+1)/2;
- _h = (_h+1)/2;
- _f = (_f+1)/2;
- }
-
- buf->w = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
- buf->h = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
- buf->f = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
- buf->offset = (uint32_t (*) [8]) MALLOC (8 * buf->scales * sizeof (uint32_t));
-
- buf->scratchbuf = (TYPE*) MALLOC (MAX3(width, height, frames) * sizeof (TYPE));
-
- if (!buf->w || !buf->h || !buf->f || !buf->offset || !buf->scratchbuf) {
- wavelet_3d_buf_destroy (buf);
- return NULL;
- }
-
- buf->w [buf->scales-1] = width;
- buf->h [buf->scales-1] = height;
- buf->f [buf->scales-1] = frames;
-
- for (level=buf->scales-2; level>=0; level--) {
- buf->w [level] = (buf->w [level+1] + 1) / 2;
- buf->h [level] = (buf->h [level+1] + 1) / 2;
- buf->f [level] = (buf->f [level+1] + 1) / 2;
- buf->offset[level][0] = 0;
- buf->offset[level][1] = buf->w [level];
- buf->offset[level][2] = buf->h [level] * width;
- buf->offset[level][3] = buf->f [level] * width * height;
- buf->offset[level][4] = buf->offset [level][2] + buf->w [level];
- buf->offset[level][5] = buf->offset [level][3] + buf->w [level];
- buf->offset[level][6] = buf->offset [level][3] + buf->offset [level][2];
- buf->offset[level][7] = buf->offset [level][6] + buf->w [level];
- }
-
- return buf;
+ Wavelet3DBuf *buf = (Wavelet3DBuf *) MALLOC (sizeof (Wavelet3DBuf));
+ uint32_t _w = width;
+ uint32_t _h = height;
+ uint32_t _f = frames;
+ int level;
+
+ if (!buf)
+ return NULL;
+
+ buf->data = (TYPE *) MALLOC (width * height * frames * sizeof (TYPE));
+
+ if (!buf->data) {
+ wavelet_3d_buf_destroy (buf);
+ return NULL;
+ }
+
+ buf->width = width;
+ buf->height = height;
+ buf->frames = frames;
+ buf->scales = 1;
+
+ while (_w > 1 || _h > 1 || _f > 1) {
+ buf->scales++;
+ _w = (_w + 1) / 2;
+ _h = (_h + 1) / 2;
+ _f = (_f + 1) / 2;
+ }
+
+ buf->w = (uint32_t *) MALLOC (buf->scales * sizeof (uint32_t));
+ buf->h = (uint32_t *) MALLOC (buf->scales * sizeof (uint32_t));
+ buf->f = (uint32_t *) MALLOC (buf->scales * sizeof (uint32_t));
+ buf->offset = (uint32_t (*)[8]) MALLOC (8 * buf->scales * sizeof (uint32_t));
+
+ buf->scratchbuf =
+ (TYPE *) MALLOC (MAX3 (width, height, frames) * sizeof (TYPE));
+
+ if (!buf->w || !buf->h || !buf->f || !buf->offset || !buf->scratchbuf) {
+ wavelet_3d_buf_destroy (buf);
+ return NULL;
+ }
+
+ buf->w[buf->scales - 1] = width;
+ buf->h[buf->scales - 1] = height;
+ buf->f[buf->scales - 1] = frames;
+
+ for (level = buf->scales - 2; level >= 0; level--) {
+ buf->w[level] = (buf->w[level + 1] + 1) / 2;
+ buf->h[level] = (buf->h[level + 1] + 1) / 2;
+ buf->f[level] = (buf->f[level + 1] + 1) / 2;
+ buf->offset[level][0] = 0;
+ buf->offset[level][1] = buf->w[level];
+ buf->offset[level][2] = buf->h[level] * width;
+ buf->offset[level][3] = buf->f[level] * width * height;
+ buf->offset[level][4] = buf->offset[level][2] + buf->w[level];
+ buf->offset[level][5] = buf->offset[level][3] + buf->w[level];
+ buf->offset[level][6] = buf->offset[level][3] + buf->offset[level][2];
+ buf->offset[level][7] = buf->offset[level][6] + buf->w[level];
+ }
+
+ return buf;
}
-void wavelet_3d_buf_destroy (Wavelet3DBuf* buf)
+void
+wavelet_3d_buf_destroy (Wavelet3DBuf * buf)
{
- if (buf) {
- if (buf->data)
- FREE (buf->data);
- if (buf->w)
- FREE (buf->w);
- if (buf->h)
- FREE (buf->h);
- if (buf->f)
- FREE (buf->f);
- if (buf->offset)
- FREE (buf->offset);
- if (buf->scratchbuf)
- FREE (buf->scratchbuf);
- FREE (buf);
- }
+ if (buf) {
+ if (buf->data)
+ FREE (buf->data);
+ if (buf->w)
+ FREE (buf->w);
+ if (buf->h)
+ FREE (buf->h);
+ if (buf->f)
+ FREE (buf->f);
+ if (buf->offset)
+ FREE (buf->offset);
+ if (buf->scratchbuf)
+ FREE (buf->scratchbuf);
+ FREE (buf);
+ }
}
@@ -104,22 +106,19 @@ void wavelet_3d_buf_destroy (Wavelet3DBuf* buf)
#include "pnm.h"
-void wavelet_3d_buf_dump (char *fmt,
- uint32_t first_frame_in_buf,
- uint32_t id,
- Wavelet3DBuf* buf,
- int16_t offset)
+void
+wavelet_3d_buf_dump (char *fmt,
+ uint32_t first_frame_in_buf,
+ uint32_t id, Wavelet3DBuf * buf, int16_t offset)
{
- char fname [256];
- uint32_t f;
+ char fname[256];
+ uint32_t f;
- for (f=0; f<buf->frames; f++) {
- snprintf (fname, 256, fmt, id, first_frame_in_buf + f);
+ for (f = 0; f < buf->frames; f++) {
+ snprintf (fname, 256, fmt, id, first_frame_in_buf + f);
- write_pgm16 (fname, buf->data + f * buf->width * buf->height,
- buf->width, buf->height, offset);
- }
+ write_pgm16 (fname, buf->data + f * buf->width * buf->height,
+ buf->width, buf->height, offset);
+ }
}
#endif
-
-
generated by cgit v1.2.1 (git 2.18.0) at 2024-11-01 03:04:37 +0000