git.mutantstargoat.com Git - demo/blob - src/image.cc

   1 #include <stdlib.h>
   2 #include <string.h>
   3
   4 #include "imago2.h"
   5 #include "image.h"
   6
   7 Image::Image()
   8 {
   9         w = h = psz = 0;
  10         pixels = 0;
  11
  12         is_float = false;
  13 }
  14
  15 Image::~Image()
  16 {
  17         free(pixels);
  18 }
  19
  20 Image::Image(const Image &image)
  21 {
  22         w = image.w;
  23         h = image.h;
  24         is_float = image.is_float;
  25         psz = image.psz;
  26
  27         if(!(pixels = malloc(w * h * psz))) {
  28                 fprintf(stderr, "Failed to allocate pixels.\n");
  29                 abort();
  30         }
  31         memcpy(pixels, image.pixels, w * h * psz);
  32 }
  33
  34 Image &Image::operator =(const Image &image)
  35 {
  36         if(&image == this)
  37                 return *this;
  38
  39         free(pixels);
  40
  41         w = image.w;
  42         h = image.h;
  43         psz = image.psz;
  44         is_float = image.is_float;
  45
  46         if(!(pixels = malloc(w * h * psz))) {
  47                 fprintf(stderr, "Failed to allocate pixels.\n");
  48                 abort();
  49         }
  50         memcpy(pixels, image.pixels, w * h * psz);
  51
  52         return *this;
  53 }
  54
  55 Image::Image(Image &&image)
  56 {
  57         w = image.w;
  58         h = image.h;
  59         psz = image.psz;
  60         is_float = image.is_float;
  61
  62         pixels = image.pixels;
  63         image.pixels = 0;
  64 }
  65
  66 Image &Image::operator =(Image &&image)
  67 {
  68         if(&image == this)
  69                 return *this;
  70
  71         free(pixels);
  72
  73         w = image.w;
  74         h = image.h;
  75         psz = image.psz;
  76         is_float = image.is_float;
  77
  78         pixels = image.pixels;
  79         image.pixels = 0;
  80
  81         return *this;
  82 }
  83
  84 bool Image::load(const char *fname)
  85 {
  86         free(pixels);
  87
  88         img_pixmap ipm;
  89         img_init(&ipm);
  90
  91         if(img_load(&ipm, fname) == -1) {
  92                 fprintf(stderr, "Failed to load image: %s.\n", fname);
  93
  94                 img_destroy(&ipm);
  95                 return false;
  96         }
  97
  98         printf("Successfully loaded image: %s\n", fname);
  99
 100         img_fmt format = img_is_float(&ipm) ? IMG_FMT_RGBAF : IMG_FMT_RGBA32;
 101         if(img_convert(&ipm, format) == -1) {
 102                 fprintf(stderr, "Failed to convert image %s.\n", fname);
 103
 104                 img_destroy(&ipm);
 105                 return false;
 106         }
 107
 108         w = ipm.width;
 109         h = ipm.height;
 110         psz = ipm.pixelsz;
 111         is_float = img_is_float(&ipm) ? true : false;
 112
 113         if(!(pixels = malloc(w * h * psz))) {
 114                 fprintf(stderr, "Failed to allocate pixels for image: %s.\n", fname);
 115                 return false;
 116         }
 117
 118         memcpy(pixels, ipm.pixels, w * h * psz);
 119
 120         img_destroy(&ipm);
 121         return true;
 122 }
 123
 124 Vec4 Image::lookup_nearest(float u, float v) const
 125 {
 126         int x = (int)(u * w) % w;
 127         int y = (int)(v * h) % h;
 128
 129         if(is_float) {
 130                 return ((Vec4*)pixels)[y * w + x];
 131         }
 132
 133         Vec4 color;
 134         unsigned char *pptr = ((unsigned char*)pixels) + (y * w + x) * 4;
 135         color.x = pptr[0] / 255.0;
 136         color.y = pptr[1] / 255.0;
 137         color.z = pptr[2] / 255.0;
 138         color.w = pptr[3] / 255.0;
 139
 140         return color;
 141 }