w3d/wavelet.c

   1 #include "mem.h"
   2 #include "wavelet.h"
   3
   4 /**
   5  *   (The transform code is in wavelet_xform.c)
   6  */
   7
   8
   9 #define MAX(a,b) ((a) > (b) ? (a) : (b))
  10 #define MAX3(a,b,c) (MAX(a,MAX(b,c)))
  11
  12
  13
  14 Wavelet3DBuf* wavelet_3d_buf_new (uint32_t width, uint32_t height,
  15                                   uint32_t frames)
  16 {
  17    Wavelet3DBuf* buf = (Wavelet3DBuf*) MALLOC (sizeof (Wavelet3DBuf));
  18    uint32_t _w = width;
  19    uint32_t _h = height;
  20    uint32_t _f = frames;
  21    int level;
  22
  23    if (!buf)
  24       return NULL;
  25
  26    buf->data = (TYPE*) MALLOC (width * height * frames * sizeof (TYPE));
  27
  28    if (!buf->data) {
  29       wavelet_3d_buf_destroy (buf);
  30       return NULL;
  31    }
  32
  33    buf->width = width;
  34    buf->height = height;
  35    buf->frames = frames;
  36    buf->scales = 1;
  37
  38    while (_w > 1 || _h > 1 || _f > 1) {
  39       buf->scales++;
  40       _w = (_w+1)/2;
  41       _h = (_h+1)/2;
  42       _f = (_f+1)/2;
  43    }
  44
  45    buf->w = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
  46    buf->h = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
  47    buf->f = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
  48    buf->offset = (uint32_t (*) [8]) MALLOC (8 * buf->scales * sizeof (uint32_t));
  49
  50    buf->scratchbuf = (TYPE*) MALLOC (MAX3(width, height, frames) * sizeof (TYPE));
  51
  52    if (!buf->w || !buf->h || !buf->f || !buf->offset || !buf->scratchbuf) {
  53       wavelet_3d_buf_destroy (buf);
  54       return NULL;
  55    }
  56
  57    buf->w [buf->scales-1] = width;
  58    buf->h [buf->scales-1] = height;
  59    buf->f [buf->scales-1] = frames;
  60
  61    for (level=buf->scales-2; level>=0; level--) {
  62       buf->w [level] = (buf->w [level+1] + 1) / 2;
  63       buf->h [level] = (buf->h [level+1] + 1) / 2;
  64       buf->f [level] = (buf->f [level+1] + 1) / 2;
  65       buf->offset[level][0] = 0;
  66       buf->offset[level][1] = buf->w [level];
  67       buf->offset[level][2] = buf->h [level] * width;
  68       buf->offset[level][3] = buf->f [level] * width * height;
  69       buf->offset[level][4] = buf->offset [level][2] + buf->w [level];
  70       buf->offset[level][5] = buf->offset [level][3] + buf->w [level];
  71       buf->offset[level][6] = buf->offset [level][3] + buf->offset [level][2];
  72       buf->offset[level][7] = buf->offset [level][6] + buf->w [level];
  73    }
  74
  75    return buf;
  76 }
  77
  78
  79 void wavelet_3d_buf_destroy (Wavelet3DBuf* buf)
  80 {
  81    if (buf) {
  82       if (buf->data)
  83          FREE (buf->data);
  84       if (buf->w)
  85          FREE (buf->w);
  86       if (buf->h)
  87          FREE (buf->h);
  88       if (buf->f)
  89          FREE (buf->f);
  90       if (buf->offset)
  91          FREE (buf->offset);
  92       if (buf->scratchbuf)
  93          FREE (buf->scratchbuf);
  94       FREE (buf);
  95    }
  96 }
  97
  98
  99 #if defined(DBG_XFORM)
 100
 101 #include "pnm.h"
 102
 103 void wavelet_3d_buf_dump (char *fmt,
 104                           uint32_t first_frame_in_buf,
 105                           uint32_t id,
 106                           Wavelet3DBuf* buf,
 107                           int16_t offset)
 108 {
 109    char fname [256];
 110    uint32_t f;
 111
 112    for (f=0; f<buf->frames; f++) {
 113       snprintf (fname, 256, fmt, id, first_frame_in_buf + f);
 114
 115       write_pgm16 (fname, buf->data + f * buf->width * buf->height,
 116                    buf->width, buf->height, offset);
 117    }
 118 }
 119 #endif
 120
 121