From 6a8ef4952936e5bd018b2b595a3fd1bb40fbb103 Mon Sep 17 00:00:00 2001
From: scuri <scuri>
Date: Wed, 12 Aug 2009 04:08:57 +0000
Subject: *** empty log message ***

---
 src/im.def               |   2 +
 src/im_image.cpp         | 210 +++++++++++++++++++++++++++++++++++++++--------
 src/lua5/imlua_capture.c |  58 +++++++++----
 src/lua5/imlua_image.c   |  35 +++++++-
 src/lua5/imlua_process.c |   8 +-
 5 files changed, 258 insertions(+), 55 deletions(-)

(limited to 'src')

diff --git a/src/im.def b/src/im.def
index abd5ec4..3f49105 100644
--- a/src/im.def
+++ b/src/im.def
@@ -99,6 +99,7 @@ EXPORTS
   imImageLineCount
   imImageLineSize
   imImageIsBitmap
+  imImageGetOpenGLData
   imImageMatch
   imImageMatchColor
   imImageMatchColorSpace
@@ -112,6 +113,7 @@ EXPORTS
   imImageSetAttribute
   imImageSetBinary
   imImageMakeBinary
+  imImageMakeGray
   imImageSetPalette
   imImageCopy
   imImageCopyData
diff --git a/src/im_image.cpp b/src/im_image.cpp
index 7acae43..80553a7 100644
--- a/src/im_image.cpp
+++ b/src/im_image.cpp
@@ -2,7 +2,7 @@
  * \brief Image Manipulation
  *
  * See Copyright Notice in im_lib.h
- * $Id: im_image.cpp,v 1.1 2008/10/17 06:10:16 scuri Exp $
+ * $Id: im_image.cpp,v 1.2 2009/08/12 04:09:16 scuri Exp $
  */
 
 #include <stdlib.h>
@@ -15,6 +15,7 @@
 #include "im_util.h"
 #include "im_attrib.h"
 #include "im_file.h"
+#include "im_convert.h"
 
 
 int imImageCheckFormat(int color_mode, int data_type)
@@ -52,7 +53,7 @@ int imImageLineSize(int width, int color_mode, int data_type)
   return imImageLineCount(width, color_mode) * imDataTypeSize(data_type);
 }
 
-static void iImageInit(imImage* image, int width, int height, int color_space, int data_type)
+static void iImageInit(imImage* image, int width, int height, int color_space, int data_type, int has_alpha)
 {
   assert(width>0);
   assert(height>0);
@@ -63,7 +64,7 @@ static void iImageInit(imImage* image, int width, int height, int color_space, i
   image->height = height;
   image->color_space = color_space;
   image->data_type = data_type;
-  image->has_alpha = 0;
+  image->has_alpha = has_alpha;
 
   image->depth = imColorModeDepth(color_space);
   image->line_size = image->width * imDataTypeSize(data_type); 
@@ -71,15 +72,20 @@ static void iImageInit(imImage* image, int width, int height, int color_space, i
   image->size = image->plane_size * image->depth;
   image->count = image->width * image->height; 
 
+  int depth = image->depth+1;  // add room for an alpha plane pointer, even if does not have alpha now.
+
   if (image->data)
   {
+    /* if reallocating, preserve the data buffer */
     void* data0 = image->data[0];
+
     free(image->data);
-    image->data = (void**)malloc((image->depth+1) * sizeof(void*));  // add room for alpha
+    image->data = (void**)malloc(depth * sizeof(void*));  
+
     image->data[0] = data0;
   }
   else
-    image->data = (void**)malloc((image->depth+1) * sizeof(void*));
+    image->data = (void**)malloc(depth * sizeof(void*));
 }
 
 imImage* imImageInit(int width, int height, int color_space, int data_type, void* data_buffer, long* palette, int palette_count)
@@ -90,7 +96,7 @@ imImage* imImageInit(int width, int height, int color_space, int data_type, void
   imImage* image = (imImage*)malloc(sizeof(imImage));
   image->data = 0;
     
-  iImageInit(image, width, height, color_space, data_type);
+  iImageInit(image, width, height, color_space, data_type, 0);
 
   if (data_buffer)
   {
@@ -119,6 +125,7 @@ imImage* imImageCreate(int width, int height, int color_space, int data_type)
   imImage* image = imImageInit(width, height, color_space, data_type, NULL, NULL, 0);
   if (!image) return NULL;
 
+  /* palette is available to BINARY, MAP and GRAY */
   if (imColorModeDepth(color_space) == 1)
   {
     image->palette = (long*)malloc(256*sizeof(long));
@@ -137,6 +144,7 @@ imImage* imImageCreate(int width, int height, int color_space, int data_type)
     }
   }
   
+  /* allocate data buffer */
   image->data[0] = malloc(image->size);
   if (!image->data[0])
   {
@@ -144,29 +152,11 @@ imImage* imImageCreate(int width, int height, int color_space, int data_type)
     return NULL;
   }
 
+  /* initialize data plane pointers */
   for (int d = 1; d < image->depth; d++)
     image->data[d] = (imbyte*)(image->data[0]) + d*image->plane_size;
 
-  if ((image->color_space == IM_YCBCR || image->color_space == IM_LAB || image->color_space == IM_LUV) && 
-      (image->data_type == IM_BYTE || image->data_type == IM_USHORT))
-  {
-    memset(image->data[0], 0, image->plane_size);
-
-    if (image->data_type == IM_BYTE)
-    {
-      imbyte* usdata = (imbyte*)image->data[1];
-      for (int i = 0; i < 2*image->count; i++)
-        *usdata++ = 128;
-    }
-    else
-    {
-      imushort* usdata = (imushort*)image->data[1];
-      for (int i = 0; i < 2*image->count; i++)
-        *usdata++ = 32768;
-    }
-  }
-  else
-    memset(image->data[0], 0, image->size);
+  imImageClear(image);
 
   return image;
 }
@@ -182,6 +172,10 @@ imImage* imImageCreateBased(const imImage* image, int width, int height, int col
 
   imImage* new_image = imImageCreate(width, height, color_space, data_type);
   imImageCopyAttributes(image, new_image);
+
+  if (image->has_alpha)
+    imImageAddAlpha(new_image);
+
   return new_image;
 }
 
@@ -213,21 +207,19 @@ void imImageReshape(imImage* image, int width, int height)
       old_width = width, 
       old_height = height;
 
-  iImageInit(image, width, height, image->color_space, image->data_type);
+  iImageInit(image, width, height, image->color_space, image->data_type, image->has_alpha);
 
   if (old_size < image->size)
   {
     void* data0 = realloc(image->data[0], image->has_alpha? image->size+image->plane_size: image->size);
     if (!data0) // if failed restore the previous size
-      iImageInit(image, old_width, old_height, image->color_space, image->data_type);
+      iImageInit(image, old_width, old_height, image->color_space, image->data_type, image->has_alpha);
     else
       image->data[0] = data0;
   }
 
-  memset(image->data[0], 0, image->has_alpha? image->size+image->plane_size: image->size);
-
+  /* initialize data plane pointers */
   int depth = image->has_alpha? image->depth+1: image->depth;
-
   for (int d = 1; d < depth; d++)
     image->data[d] = (imbyte*)image->data[0] + d*image->plane_size;
 }
@@ -256,7 +248,30 @@ void imImageDestroy(imImage* image)
 void imImageClear(imImage* image)
 {
   assert(image);
-  memset(image->data[0], 0, image->has_alpha? image->size+image->plane_size: image->size);
+
+  if ((image->color_space == IM_YCBCR || image->color_space == IM_LAB || image->color_space == IM_LUV) && 
+      (image->data_type == IM_BYTE || image->data_type == IM_USHORT))
+  {
+    memset(image->data[0], 0, image->plane_size);
+
+    if (image->data_type == IM_BYTE)
+    {
+      imbyte* usdata = (imbyte*)image->data[1];
+      for (int i = 0; i < 2*image->count; i++)
+        *usdata++ = 128;
+    }
+    else
+    {
+      imushort* usdata = (imushort*)image->data[1];
+      for (int i = 0; i < 2*image->count; i++)
+        *usdata++ = 32768;
+    }
+  }
+  else
+    memset(image->data[0], 0, image->size);
+
+  if (image->has_alpha)
+    memset(image->data[image->depth], 0, image->plane_size);
 }
 
 int imImageIsBitmap(const imImage* image)
@@ -320,6 +335,124 @@ imImage* imImageClone(const imImage* image)
   return new_image;
 }
 
+static void iImageMakeGray(imbyte *map, int count, int step)
+{
+  for(int i = 0; i < count; i++)
+  {
+    if (*map)
+      *map = 255;
+    map += step;
+  }
+}
+
+static void iImageCopyMapAlpha(imbyte *map, imbyte *gldata, int depth, int count)
+{
+  gldata += depth-1; /* position at first alpha */
+  for(int i = 0; i < count; i++)
+  {
+    *gldata = *map;
+    map++;
+    gldata += depth;  /* skip to next alpha */
+  }
+}
+
+static void iImageExpandTranspIndex(imbyte *map, imbyte *gldata, int depth, int count, imbyte index)
+{
+  gldata += depth-1; /* position at first alpha */
+  for(int i = 0; i < count; i++)
+  {
+    if (*map == index)
+      *gldata = 255;
+
+    map++;
+    gldata += depth;  /* skip to next alpha */
+  }
+}
+
+/* To avoid including gl.h */
+#define GL_RGB                            0x1907
+#define GL_RGBA                           0x1908
+#define GL_LUMINANCE                      0x1909
+#define GL_LUMINANCE_ALPHA                0x190A
+
+void* imImageGetOpenGLData(imImage* image, int *format)
+{
+  if (!imImageIsBitmap(image))
+    return NULL;
+
+  int transp_count;
+  imbyte* transp_index = (imbyte*)imImageGetAttribute(image, "TransparencyIndex", NULL, &transp_count);
+
+  int glformat;
+  switch(image->color_space)
+  {
+  case IM_MAP:
+    if (image->has_alpha || transp_index)
+      glformat = GL_RGBA;
+    else
+      glformat = GL_RGB;
+    break;
+  case IM_RGB:
+    if (image->has_alpha)
+      glformat = GL_RGBA;
+    else
+      glformat = GL_RGB;
+    break;
+  case IM_BINARY:
+  default: /* IM_GRAY */
+    if (image->has_alpha || transp_index)
+      glformat = GL_LUMINANCE_ALPHA;
+    else
+      glformat = GL_LUMINANCE;
+    break;
+  }
+
+  int depth;
+  switch (glformat)
+  {
+  case GL_RGB:
+    depth = 3;
+    break;
+  case GL_RGBA:
+    depth = 4;
+    break;
+  case GL_LUMINANCE_ALPHA:
+    depth = 2;
+    break;
+  default: /* GL_LUMINANCE */
+    depth = 1;
+    break;
+  }
+
+  int size = image->count*depth;
+  imImageSetAttribute(image, "GLDATA", IM_BYTE, size, NULL);
+  imbyte* gldata = (imbyte*)imImageGetAttribute(image, "GLDATA", NULL, NULL);
+
+  if (image->color_space == IM_RGB)
+    imConvertPacking(image->data[0], gldata, image->width, image->height, depth, IM_BYTE, 0);
+  else
+  {
+    memcpy(gldata, image->data[0], image->size);
+
+    if (image->color_space == IM_MAP)
+      imConvertMapToRGB(gldata, image->count, depth, 1, image->palette, image->palette_count);
+    else if (image->color_space == IM_BINARY)
+      iImageMakeGray(gldata, image->count, (glformat==GL_LUMINANCE_ALPHA)? 2: 1);
+
+    if (image->has_alpha)
+      iImageCopyMapAlpha((imbyte*)image->data[1], gldata, depth, image->count);
+    else if (transp_index)
+    {
+      int i;
+      for (i=0; i<transp_count; i++)
+        iImageExpandTranspIndex((imbyte*)image->data[0], gldata, depth, image->count, transp_index[i]);
+    }
+  }
+
+  if (format) *format = glformat;
+  return gldata;
+}
+
 void imImageSetAttribute(imImage* image, const char* attrib, int data_type, int count, const void* data)
 {
   assert(image);
@@ -474,6 +607,19 @@ void imImageMakeBinary(imImage *image)
   }
 }
 
+void imImageMakeGray(imImage *image)
+{
+  assert(image);
+
+  imbyte *map = (imbyte*)image->data[0];
+  for(int i = 0; i < image->count; i++)
+  {
+    if (*map)
+      *map = 255;
+    map++;
+  }
+}
+
 static void iImageGrayCheckChange(imImage *image)
 {
   int i, do_remap = 0;
diff --git a/src/lua5/imlua_capture.c b/src/lua5/imlua_capture.c
index 15d52ce..59ec0fb 100644
--- a/src/lua5/imlua_capture.c
+++ b/src/lua5/imlua_capture.c
@@ -2,7 +2,7 @@
  * \brief IM Lua 5 Binding
  *
  * See Copyright Notice in im_lib.h
- * $Id: imlua_capture.c,v 1.1 2008/10/17 06:16:32 scuri Exp $
+ * $Id: imlua_capture.c,v 1.2 2009/08/12 04:09:17 scuri Exp $
  */
 
 #include <string.h>
@@ -92,7 +92,10 @@ static int imluaVideoCaptureConnect (lua_State *L)
 {
   imVideoCapture *vc = imlua_checkvideocapture(L, 1);
   int device = luaL_optint(L, 2, -1);
-  lua_pushnumber(L, imVideoCaptureConnect(vc, device));
+  if (device == -1)
+    lua_pushnumber(L, imVideoCaptureConnect(vc, device));
+  else
+    lua_pushboolean(L, imVideoCaptureConnect(vc, device));
   return 1;
 }
 
@@ -123,7 +126,7 @@ static int imluaVideoCaptureShowDialog (lua_State *L)
   int dialog = luaL_checkint(L, 2);
   void *parent = lua_touserdata(L, 3); 
 
-  lua_pushnumber(L, imVideoCaptureShowDialog(vc, dialog, parent));
+  lua_pushboolean(L, imVideoCaptureShowDialog(vc, dialog, parent));
   return 1;
 }
 
@@ -158,7 +161,7 @@ static int imluaVideoCaptureGetFormat (lua_State *L)
   int width, height;
   char desc[10];
 
-  lua_pushnumber(L, imVideoCaptureGetFormat(vc, format, &width, &height, desc));
+  lua_pushboolean(L, imVideoCaptureGetFormat(vc, format, &width, &height, desc));
   lua_pushnumber(L, width);
   lua_pushnumber(L, height);
   lua_pushstring(L, desc);
@@ -190,24 +193,40 @@ static int imluaVideoCaptureSetImageSize (lua_State *L)
   int width = luaL_checkint(L, 2);
   int height = luaL_checkint(L, 3);
 
-  lua_pushnumber(L, imVideoCaptureSetImageSize(vc, width, height));
+  lua_pushboolean(L, imVideoCaptureSetImageSize(vc, width, height));
 
   return 1;
 }
 
 /*****************************************************************************\
- vc:SetFormat()
+ vc:SetInOut()
 \*****************************************************************************/
-static int imluaVideoCaptureSetFormat (lua_State *L)
+static int imluaVideoCaptureSetInOut(lua_State *L)
 {
   imVideoCapture *vc = imlua_checkvideocapture(L, 1);
-  int format = luaL_checkint(L, 2);
+  int input = luaL_checkint(L, 2);
+  int output = luaL_checkint(L, 3);
+  int cross = luaL_checkint(L, 4);
 
-  lua_pushnumber(L, imVideoCaptureSetFormat(vc, format));
+  lua_pushboolean(L, imVideoCaptureSetInOut(vc, input, output, cross));
 
   return 1;
 }
 
+/*****************************************************************************\
+ vc:SetFormat()
+\*****************************************************************************/
+static int imluaVideoCaptureSetFormat (lua_State *L)
+{
+  imVideoCapture *vc = imlua_checkvideocapture(L, 1);
+  int format = luaL_optint(L, 2, -1);
+  if (format == -1)
+    lua_pushnumber(L, imVideoCaptureSetFormat(vc, format));
+  else
+    lua_pushboolean(L, imVideoCaptureSetFormat(vc, format));
+  return 1;
+}
+
 /*****************************************************************************\
  vc:ResetAttribute(attrib, fauto)
 \*****************************************************************************/
@@ -215,9 +234,9 @@ static int imluaVideoCaptureResetAttribute (lua_State *L)
 {
   imVideoCapture *vc = imlua_checkvideocapture(L, 1);
   const char *attrib = luaL_checkstring(L, 2);
-  int fauto = luaL_checkint(L, 3);
+  int fauto = lua_toboolean(L, 3);
 
-  lua_pushnumber(L, imVideoCaptureResetAttribute(vc, attrib, fauto));
+  lua_pushboolean(L, imVideoCaptureResetAttribute(vc, attrib, fauto));
   return 1;
 }
 
@@ -230,7 +249,7 @@ static int imluaVideoCaptureSetAttribute (lua_State *L)
   const char *attrib = luaL_checkstring(L, 2);
   float percent = (float) luaL_checknumber(L, 3);
 
-  lua_pushnumber(L, imVideoCaptureSetAttribute(vc, attrib, percent));
+  lua_pushboolean(L, imVideoCaptureSetAttribute(vc, attrib, percent));
   return 1;
 }
 
@@ -243,7 +262,7 @@ static int imluaVideoCaptureGetAttribute (lua_State *L)
   const char *attrib = luaL_checkstring(L, 2);
   float percent;
 
-  lua_pushnumber(L, imVideoCaptureGetAttribute(vc, attrib, &percent));
+  lua_pushboolean(L, imVideoCaptureGetAttribute(vc, attrib, &percent));
   lua_pushnumber(L, percent);
   return 2;
 }
@@ -282,7 +301,7 @@ static int imluaVideoCaptureFrame (lua_State *L)
     luaL_argerror(L, 2, "image must be of RGB or Gray color spaces");
   imlua_checkdatatype(L, 2, image, IM_BYTE);
 
-  lua_pushnumber(L, imVideoCaptureFrame(vc, image->data[0], image->color_space, timeout));
+  lua_pushboolean(L, imVideoCaptureFrame(vc, image->data[0], image->color_space, timeout));
 
   return 1;
 }
@@ -299,7 +318,7 @@ static int imluaVideoCaptureOneFrame (lua_State *L)
     luaL_argerror(L, 2, "image must be of RGB or Gray color spaces");
   imlua_checkdatatype(L, 2, image, IM_BYTE);
 
-  lua_pushnumber(L, imVideoCaptureOneFrame(vc, image->data[0], image->color_space));
+  lua_pushboolean(L, imVideoCaptureOneFrame(vc, image->data[0], image->color_space));
 
   return 1;
 }
@@ -310,9 +329,11 @@ static int imluaVideoCaptureOneFrame (lua_State *L)
 static int imluaVideoCaptureLive (lua_State *L)
 {
   imVideoCapture *vc = imlua_checkvideocapture(L, 1);
-  int live = luaL_checkint(L, 2);
-
-  lua_pushnumber(L, imVideoCaptureLive(vc, live));
+  int live = luaL_optint(L, 2, -1);
+  if (live == -1)
+    lua_pushnumber(L, imVideoCaptureLive(vc, live));
+  else
+    lua_pushboolean(L, imVideoCaptureLive(vc, live));
 
   return 1;
 }
@@ -377,6 +398,7 @@ static const luaL_reg imcapture_metalib[] = {
   {"SetFormat", imluaVideoCaptureSetFormat},
   {"GetImageSize", imluaVideoCaptureGetImageSize},
   {"SetImageSize", imluaVideoCaptureSetImageSize},
+  {"SetInOut", imluaVideoCaptureSetInOut},
   {"ResetAttribute", imluaVideoCaptureResetAttribute},
   {"GetAttribute", imluaVideoCaptureGetAttribute},
   {"SetAttribute", imluaVideoCaptureSetAttribute},
diff --git a/src/lua5/imlua_image.c b/src/lua5/imlua_image.c
index 22ed38c..dc3322a 100644
--- a/src/lua5/imlua_image.c
+++ b/src/lua5/imlua_image.c
@@ -2,7 +2,7 @@
  * \brief IM Lua 5 Binding
  *
  * See Copyright Notice in im_lib.h
- * $Id: imlua_image.c,v 1.1 2008/10/17 06:16:32 scuri Exp $
+ * $Id: imlua_image.c,v 1.2 2009/08/12 04:09:17 scuri Exp $
  */
 
 #include <string.h>
@@ -437,6 +437,28 @@ static int imluaImageIsBitmap (lua_State *L)
   return 1;
 }
 
+
+/*****************************************************************************\
+ image:GetOpenGLData()
+\*****************************************************************************/
+static int imluaImageGetOpenGLData (lua_State *L)
+{
+  int format;
+  imbyte* gldata;
+  imImage *image = imlua_checkimage(L, 1);
+
+  gldata = imImageGetOpenGLData(image, &format);
+  if (!gldata)
+  {
+    lua_pushnil(L);
+    return 1;
+  }
+
+  lua_pushlightuserdata(L, gldata);
+  lua_pushinteger(L, format);
+  return 2;
+}
+
 /*****************************************************************************\
  image:GetPalette()
 \*****************************************************************************/
@@ -550,6 +572,15 @@ static int imluaImageMakeBinary (lua_State *L)
   return 0;
 }
 
+/*****************************************************************************\
+ image:MakeGray()
+\*****************************************************************************/
+static int imluaImageMakeGray (lua_State *L)
+{
+  imImageMakeGray(imlua_checkimage(L, 1));
+  return 0;
+}
+
 /*****************************************************************************\
  image:Width()
 \*****************************************************************************/
@@ -965,6 +996,7 @@ static const luaL_reg imimage_metalib[] = {
   {"GetAttributeList", imluaImageGetAttributeList},
   {"Clear", imluaImageClear},
   {"IsBitmap", imluaImageIsBitmap},
+  {"GetOpenGLData", imluaImageGetOpenGLData},
   {"SetPalette", imluaImageSetPalette},
   {"GetPalette", imluaImageGetPalette},
   {"CopyAttributes", imluaImageCopyAttributes},
@@ -975,6 +1007,7 @@ static const luaL_reg imimage_metalib[] = {
   {"Match", imluaImageMatch},
   {"SetBinary", imluaImageSetBinary},
   {"MakeBinary", imluaImageMakeBinary},
+  {"MakeGray", imluaImageMakeGray},
   {"Width", imluaImageWidth},
   {"Height", imluaImageHeight},
   {"Depth", imluaImageDepth},
diff --git a/src/lua5/imlua_process.c b/src/lua5/imlua_process.c
index 492a1ba..1c0dfc8 100644
--- a/src/lua5/imlua_process.c
+++ b/src/lua5/imlua_process.c
@@ -2,7 +2,7 @@
  * \brief IM Lua 5 Binding
  *
  * See Copyright Notice in im_lib.h
- * $Id: imlua_process.c,v 1.5 2009/08/04 21:35:26 scuri Exp $
+ * $Id: imlua_process.c,v 1.6 2009/08/12 04:09:17 scuri Exp $
  */
 
 #include <memory.h>
@@ -77,7 +77,7 @@ static int imluaCalcHistogram (lua_State *L)
 {
   imImage* src_image = imlua_checkimage(L, 1);
   int plane = luaL_checkint(L, 2);
-  int cumulative = luaL_checkint(L, 3);
+  int cumulative = lua_toboolean(L, 3);
 
   switch (src_image->data_type)
   {
@@ -112,7 +112,7 @@ static int imluaCalcGrayHistogram (lua_State *L)
 {
   unsigned long hist[256];
   imImage* src_image = imlua_checkimage(L, 1);
-  int cumulative = luaL_checkint(L, 2);
+  int cumulative = lua_toboolean(L, 2);
 
   imlua_checkdatatype(L, 1, src_image, IM_BYTE);
   if (src_image->color_space >= IM_CMYK)
@@ -1679,7 +1679,7 @@ static int imluaProcessMergeComplex (lua_State *L)
   imImage *src_image1 = imlua_checkimage(L, 1);
   imImage *src_image2 = imlua_checkimage(L, 2);
   imImage *dst_image = imlua_checkimage(L, 3);
-  int polar = luaL_checkint(L, 5);
+  int polar = lua_toboolean(L, 4);
 
   imlua_checkdatatype(L, 1, src_image1, IM_FLOAT);
   imlua_checkdatatype(L, 2, src_image2, IM_FLOAT);
-- 
cgit v1.2.3