gles_util.cpp

python s60 1.4.5版本的源代码

/**
 * ====================================================================
 * gles_util.h
 *
 * Copyright (c) 2006 Nokia Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include "glesmodule.h"

/**
 * Converts a CFbsBitmap object into an array for OpenGL texturing
 *
 * @param bitmap Bitmap object to convert
 * @param format Format of bitmap. Can be one of: GL_RGB, GL_RGBA, GL_ALPHA, GL_LUMINANCE, GL_LUMINANCE_ALPHA
 * @param type Type of data. Can be one of: GL_UNSIGNED_BYTE, GL_UNSIGNED_SHORT_5_6_5
 * @param size Stores the size of the data. Can be NULL.
 * @return Converted bitmap
 */
void *gles_convert_fbsbitmap(CFbsBitmap *bitmap, GLenum format, GLenum type, unsigned int *datalen) {
  CFbsBitmap *target = NULL;
  CFbsBitmapDevice *dev = NULL;
  CBitmapContext *gc = NULL;
  
  TSize texture_dimensions;
  int height;
  int width;
  int texture_size;
  int real_size = 0; // Actual size of the (final) bitmap in bytes
  TUint8 *pixels = NULL;
  
  int y;
  int x;
  // Bits per pixel in the final bitmap
  int dst_bits_per_pixel = 24;
  // Length of the scanline in source bitmap
  int stride = 0;
  // Desired number of color elements per pixel
  TUint8 elements;
  
  assert(bitmap != NULL);
  
  TDisplayMode displaymode;
  
  switch(format) {
    case GL_LUMINANCE:
    case GL_ALPHA:
      elements = 1;
      break;
    // There is no corresponding display mode for GL_LUMINANCE_ALPHA mode
    //case GL_LUMINANCE_ALPHA:
    //  elements = 2;
    //  break;
    case GL_RGB:
      elements = 3;
      break;
    // Alpha channel is disabled since Image objects do not support it yet.
    // If the support gets added, the corresponding code can be added in this function.
    //case GL_RGBA:
      //break;
    default:
      PyErr_SetString(PyExc_ValueError, "Unsupported image format");
      return NULL;
  }
  
  // Check that we can convert the bitmap
  // The types that are allowed to pass here must
  // be also handled in the conversion loop, or bad things may happen...
  switch(type) {
    case GL_UNSIGNED_BYTE:
      if(elements == 3) {
        displaymode = EColor16M;
        dst_bits_per_pixel = 24;
      } else if(elements == 1) {
        displaymode = EGray256;
        dst_bits_per_pixel = 8;
      } else {
        PyErr_SetString(PyExc_ValueError, "Unsupported type/format combination");
        return NULL;
      }
      break;
    case GL_UNSIGNED_SHORT_5_6_5:
      if(elements == 3) {
        dst_bits_per_pixel = 16;
        displaymode = EColor64K;
      } else {
        PyErr_SetString(PyExc_ValueError, "Unsupported type/format combination");
        return NULL;
      }
      break;
    // There is no corresponding display mode for these formats
    case GL_UNSIGNED_SHORT_5_5_5_1:
    case GL_UNSIGNED_SHORT_4_4_4_4:
      PyErr_SetString(PyExc_ValueError, "Unsupported image type");
      return NULL;
  }
  
  // Create converted bitmap only if needed
  if(bitmap->DisplayMode() != displaymode) {
    // Try to create another bitmap with the desired display mode
    target = new CFbsBitmap();
    target->Create(bitmap->SizeInPixels(), displaymode);
  
    // Blit the image, converting at the same time
    dev = CFbsBitmapDevice::NewL(target);
    dev->CreateBitmapContext(gc);
    gc->BitBlt(TPoint(0,0), bitmap);
    
    // Not used anymore
    delete gc;
    delete dev;
  } else {
    target = bitmap;
  }
  
  // Figure out some sizing information
  texture_dimensions = bitmap->SizeInPixels();
  height = texture_dimensions.iHeight;
  width = texture_dimensions.iWidth;
  texture_size = height * width;
  real_size = texture_size * (dst_bits_per_pixel/8) * sizeof(TUint8);
  stride = target->ScanLineLength(width, target->DisplayMode());
  
#ifdef DEBUG_GLES
  DEBUGMSG1("height: %d", height);
  DEBUGMSG1("width: %d", width);
  DEBUGMSG1("size: %d", real_size);
  DEBUGMSG1("stride: %d", stride);
#endif
  
  pixels = (TUint8*)gles_alloc( real_size );
  if(pixels == NULL) {
    return PyErr_NoMemory();
  }
  
  // Make a copy of the bitmap pixel data
  target->LockHeap();
  
  TUint8 *pixel_data = (TUint8*)target->DataAddress();
  TUint8 *current_row = pixel_data;
  for(y=0;y<height;y++) {
    for(x=0;x<width;x++) {
      // Assign the pixel to the final bitmap.
      //TUint8 *dstpixel = ((TUint8*)pixels + (y*width*(dst_bits_per_pixel/8)) + (x*(dst_bits_per_pixel/8)));
      
      if(type == GL_UNSIGNED_SHORT_5_6_5) {
        // bits_per_pixel not needed here (always 16)
        TUint16 *dstpixel = (TUint16*)((TUint16*)pixels + (y*width) + x);
        TUint16 *pixel =  (TUint16*)current_row+x;
        
        if(format == GL_RGB) {
          // Read the pixels directly
          // Copy 16 bits
          *dstpixel = *pixel;
        }
      } else if(type == GL_UNSIGNED_BYTE) {
        TUint8 *dstpixel = ((TUint8*)pixels + (y*width*(dst_bits_per_pixel/8)) + (x*(dst_bits_per_pixel/8)));
        
        if(elements == 3) {
          // Copy 24 bits
          // Converting from BGR to RGB at the same time
          TUint8 *pixel = (TUint8*)current_row+(x*3);
          *(dstpixel+2) = *pixel;
          *(dstpixel+1) = *(pixel+1);
          *(dstpixel) = *(pixel+2);
        } else if(elements == 1) {
          TUint8 *pixel = (TUint8*)current_row+(x);
          *(dstpixel) = *pixel;
        }
      }
    } // for(x)
    // Move to next row
    current_row = ((TUint8*)current_row + stride);
  } // for(y)
  // Unlock the heap
  bitmap->UnlockHeap();
  
  if(datalen) {
    *datalen = real_size;
  }
  
  // Don't delete if the bitmap isn't ours
  if(target != bitmap) {
    delete target;
    target = NULL;
  }
  return pixels;
}

/** 
 * Return a graphics.Image object as an FbsBitmap object. Copied from pys60/src/ext/graphicsmodule.cpp
 *
 * @param obj Object to check
 * @return FbsBitmap object or NULL
 */
CFbsBitmap *
Bitmap_AsFbsBitmap(PyObject *obj)
{
  PyObject *capi_object=PyCAPI_GetCAPI(obj,"_bitmapapi");
  if (!capi_object)
    return NULL;
  CFbsBitmap *bitmap=(CFbsBitmap *)PyCObject_AsVoidPtr(capi_object);
  Py_DECREF(capi_object);
  return bitmap;
}

/**
 * Helper function for extracting C API's from objects.
 * Copied from pys60/src/ext/graphicsmodule.cpp
 *
 * @param object Object to extract C API from
 * @param name Name of C API to extract
 * @return PyCObject instance
 */
PyObject *
PyCAPI_GetCAPI(PyObject *object, 
	       const char *apiname)
{
  PyObject *capi_func=NULL;
  PyObject *capi_cobject=NULL;
  if (!(capi_func=PyObject_GetAttrString(object,(char *)apiname)) ||
      !PyCallable_Check(capi_func) ||
      !(capi_cobject=PyObject_CallObject(capi_func,NULL)) ||
      !(PyCObject_Check(capi_cobject))) {
    Py_XDECREF(capi_func);
    Py_XDECREF(capi_cobject);
    return NULL;
  }
  Py_DECREF(capi_func);
  return capi_cobject;
}

int gles_PySequence_Dimension(PyObject *seq) {
  assert(PySequence_Check(seq));
  PyObject *item = PySequence_GetItem(seq, 0);
  if( !PySequence_Check(item) ) {
    return -1;
  }
  // The dimension is determined from the lenght of the first element
  int dimension = PySequence_Length(item);
  return dimension;
}

/**
 * Convert a freely structured sequence into a flat sequence
 * Also converts data to the given type.
 *
 * @param type Can be one of: GL_FLOAT, GL_BYTE, GL_UNSIGNED_BYTE, GL_SHORT, GL_UNSIGNED_SHORT, GL_FIXED.
 * @param source A Python sequence where the data is read from.
 * @param target A pointer to an existing target sequence. If NULL, a new sequence is created.
 * @return A collapsed sequence.
 */
PyObject *gles_PySequence_Collapse(GLenum type, PyObject *source, PyObject *target)
{
  int count;
  int i;
  PyObject *k;
  
  // We need a sequence to operate.
  if(!PySequence_Check(source)) {
    PyErr_SetString(PyExc_TypeError, "Expecting a sequence");
    return NULL;
  }
  
  // Make a new list if not supplied by caller.
  if(target == NULL) {
    target = PyList_New(0);
    if(!target) {
      assert(PyErr_Occurred() != NULL);
      return NULL;
    }
  }
  
  count = PySequence_Length(source);
  for(i=0; i < count; i++) {
    k = PySequence_GetItem(source, i);
    // See if its a sequence. Strings are sequences too, but we should ignore them here.
    if(PySequence_Check(k) && !PyString_Check(k)) {
      // Call a new function go get the contents of the sequence.
      if(gles_PySequence_Collapse(type, k, target) == NULL) {
        // There should be an exception
        assert(PyErr_Occurred() != NULL);
        return NULL;
      }
    } else {
      assert(k != NULL);
      if(!PyNumber_Check(k)) {
        PyErr_SetString(PyExc_TypeError, "Expecting a number");
        return NULL;
      }
      
      // Try to convert the object into desired type.
      // Using the number interface we can handle all
      // types that can be converted into a number.
      PyObject *t=NULL;
      switch(type) {
        case GL_FLOAT:
          k = PyNumber_Float(k);
          if(k==NULL) {
            // Something went wrong with the conversion.
            assert(PyErr_Occurred() != NULL);
            return NULL;
          }
          t = Py_BuildValue("f", PyFloat_AsDouble(k));
          break;
        case GL_SHORT:
          k = PyNumber_Int(k);
          if(k==NULL) {
            assert(PyErr_Occurred() != NULL);
            return NULL;
          }
          t = Py_BuildValue("h", PyInt_AsLong(k));
          break;
        case GL_UNSIGNED_SHORT:
          k = PyNumber_Int(k);
          if(k==NULL) {
            assert(PyErr_Occurred() != NULL);
            return NULL;
          }
          t = Py_BuildValue("H", PyInt_AsLong(k));
          break;
        case GL_BYTE:
          k = PyNumber_Int(k);
          if(k==NULL) {
            assert(PyErr_Occurred() != NULL);
            return NULL;
          }
          t = Py_BuildValue("b", PyInt_AsLong(k));
          break;
        case GL_UNSIGNED_BYTE:
          k = PyNumber_Int(k);
          if(k==NULL) {
            assert(PyErr_Occurred() != NULL);
            return NULL;
          }
          t = Py_BuildValue("B", PyInt_AsLong(k));
          break;
        case GL_FIXED:
          k = PyNumber_Int(k);
          if(k==NULL) {
            assert(PyErr_Occurred() != NULL);
            return NULL;
          }
          t = Py_BuildValue("i", PyInt_AsLong(k));
          break;
        default:
          PyErr_SetString(PyExc_TypeError, "Invalid type for conversion");
          return NULL;
      }
      if(t == NULL) {
        assert(PyErr_Occurred() != NULL);
        return NULL;
      }
      // Add item to target.
      PyList_Append(target, t);
    }
  }
  
  return target;
}

/**
 * Place given array into internal memory structure
 *
 * If there already is something in memory, it will be freed,
 * if the memory does not belong to an array object
 *
 * @param targetarr Specifies where the given array should be placed in internal memory. Can be one of: GL_VERTEX_ARRAY, GL_NORMAL_ARRAY, GL_COLOR_ARRAY, GL_TEXTURE_COORD_ARRAY.
 * @param *ptr Pointer to an array.
 * @param object Python object of type gles.array. Can be NULL, in which case the memory will be freed when the next array is placed in the same location.
 * @return The given pointer
*/
void *gles_assign_array(GLenum targetarr, void *ptr, array_object *object) {
  GLES_Arrays *arrays = (GLES_Arrays*)Dll::Tls();
  gles_array_t *arrptr;
  
  assert(arrays!=NULL);
  
  // Get the array from internal struct.
  switch(targetarr) {
    case GL_VERTEX_ARRAY:
      arrptr = &(arrays->vertex);
#ifdef DEBUG_GLES
      DEBUGMSG("Assigning GL_VERTEX_ARRAY\n");
#endif
      break;
    case GL_NORMAL_ARRAY:
      arrptr = &(arrays->normal);
#ifdef DEBUG_GLES
      DEBUGMSG("Assigning GL_NORMAL_ARRAY\n");
#endif
      break;
    case GL_COLOR_ARRAY:
      arrptr = &(arrays->color);
#ifdef DEBUG_GLES
      DEBUGMSG("Assigning GL_COLOR_ARRAY\n");
#endif
      break;
    case GL_TEXTURE_COORD_ARRAY:
      arrptr = &(arrays->texcoord);
#ifdef DEBUG_GLES
      DEBUGMSG("Assigning GL_TEXTURE_COORD_ARRAY\n");
#endif
      break;
#ifdef GL_OES_VERSION_1_1
    case GL_MATRIX_INDEX_ARRAY_OES:
      arrptr = &(arrays->matrix);
#ifdef DEBUG_GLES
      DEBUGMSG("Assigning GL_MATRIX_INDEX_ARRAY_OES\n");
#endif
      break;
    case GL_POINT_SIZE_ARRAY_OES:
      arrptr = &(arrays->pointsize);
#ifdef DEBUG_GLES
      DEBUGMSG("Assigning GL_POINT_SIZE_ARRAY_OES\n");
#endif
      break;
    case GL_WEIGHT_ARRAY_OES:
      arrptr = &(arrays->weight);
#ifdef DEBUG_GLES
      DEBUGMSG("Assigning GL_WEIGHT_ARRAY_OES\n");
#endif
      break;
#endif
    default:
      return NULL;
  }
  // Set the previous array free
  gles_free_array(targetarr);
  
  // See if we got a new object
  if(object != NULL) {
    arrptr->obj = (PyObject*)object;
    Py_XINCREF(object);
    arrptr->ptr = object->arrdata;
  } // Otherwise just take the pointer into custody
  else {
    arrptr->obj = NULL;
    arrptr->ptr = ptr;
  }
  
  return arrptr->ptr;
}

/**
 * Convert a Python sequence to a float array
 * 
 * @param seq A Python sequence object
 * @return A pointer to a C array of GLfloat values
 */
GLfloat *gles_PySequence_AsGLfloatArray(PyObject *seq) {
  int count;
  GLfloat *dataptr;
  int i;
  PyObject *item;
  
  if(! PySequence_Check(seq)) {
    PyErr_SetString(PyExc_TypeError, "Invalid type");
    return NULL;
  }
  
  count = PySequence_Length(seq);
  dataptr = (GLfloat*)gles_alloc( sizeof(GLfloat)*count);
  if(dataptr == NULL) {
    return (GLfloat*)PyErr_NoMemory();
  }