1 /* 2 * Copyright (C) 2011 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 /* $Id: db_utilities.cpp,v 1.4 2011/06/17 14:03:31 mbansal Exp $ */ 18 19 #include "db_utilities.h" 20 #include <string.h> 21 #include <stdio.h> 22 23 float** db_SetupImageReferences_f(float *im,int w,int h) 24 { 25 int i; 26 float **img; 27 assert(im); 28 img=new float* [h]; 29 for(i=0;i<h;i++) 30 { 31 img[i]=im+w*i; 32 } 33 return(img); 34 } 35 36 unsigned char** db_SetupImageReferences_u(unsigned char *im,int w,int h) 37 { 38 int i; 39 unsigned char **img; 40 41 assert(im); 42 43 img=new unsigned char* [h]; 44 for(i=0;i<h;i++) 45 { 46 img[i]=im+w*i; 47 } 48 return(img); 49 } 50 float** db_AllocImage_f(int w,int h,int over_allocation) 51 { 52 float **img,*im; 53 54 im=new float [w*h+over_allocation]; 55 img=db_SetupImageReferences_f(im,w,h); 56 57 return(img); 58 } 59 60 unsigned char** db_AllocImage_u(int w,int h,int over_allocation) 61 { 62 unsigned char **img,*im; 63 64 im=new unsigned char [w*h+over_allocation]; 65 img=db_SetupImageReferences_u(im,w,h); 66 67 return(img); 68 } 69 70 void db_FreeImage_f(float **img,int h) 71 { 72 delete [] (img[0]); 73 delete [] img; 74 } 75 76 void db_FreeImage_u(unsigned char **img,int h) 77 { 78 delete [] (img[0]); 79 delete [] img; 80 } 81 82 // ----------------------------------------------------------------------------------------------------------- ; 83 // 84 // copy image (source to destination) 85 // ---> must be a 2D image array with the same image size 86 // ---> the size of the input and output images must be same 87 // 88 // ------------------------------------------------------------------------------------------------------------ ; 89 void db_CopyImage_u(unsigned char **d,const unsigned char * const *s, int w, int h, int over_allocation) 90 { 91 int i; 92 93 for (i=0;i<h;i++) 94 { 95 memcpy(d[i],s[i],w*sizeof(unsigned char)); 96 } 97 98 memcpy(&d[h],&d[h],over_allocation); 99 100 } 101 102 inline void db_WarpImageLutFast_u(const unsigned char * const * src, unsigned char ** dst, int w, int h, 103 const float * const * lut_x, const float * const * lut_y) 104 { 105 assert(src && dst); 106 int xd=0, yd=0; 107 108 for ( int i = 0; i < w; ++i ) 109 for ( int j = 0; j < h; ++j ) 110 { 111 //xd = static_cast<unsigned int>(lut_x[j][i]); 112 //yd = static_cast<unsigned int>(lut_y[j][i]); 113 xd = (unsigned int)(lut_x[j][i]); 114 yd = (unsigned int)(lut_y[j][i]); 115 if ( xd >= w || yd >= h || 116 xd < 0 || yd < 0) 117 dst[j][i] = 0; 118 else 119 dst[j][i] = src[yd][xd]; 120 } 121 } 122 123 inline void db_WarpImageLutBilinear_u(const unsigned char * const * src, unsigned char ** dst, int w, int h, 124 const float * const * lut_x,const float * const* lut_y) 125 { 126 assert(src && dst); 127 double xd=0.0, yd=0.0; 128 129 for ( int i = 0; i < w; ++i ) 130 for ( int j = 0; j < h; ++j ) 131 { 132 xd = static_cast<double>(lut_x[j][i]); 133 yd = static_cast<double>(lut_y[j][i]); 134 if ( xd > w || yd > h || 135 xd < 0.0 || yd < 0.0) 136 dst[j][i] = 0; 137 else 138 dst[j][i] = db_BilinearInterpolation(yd, xd, src); 139 } 140 } 141 142 143 void db_WarpImageLut_u(const unsigned char * const * src, unsigned char ** dst, int w, int h, 144 const float * const * lut_x,const float * const * lut_y, int type) 145 { 146 switch (type) 147 { 148 case DB_WARP_FAST: 149 db_WarpImageLutFast_u(src,dst,w,h,lut_x,lut_y); 150 break; 151 case DB_WARP_BILINEAR: 152 db_WarpImageLutBilinear_u(src,dst,w,h,lut_x,lut_y); 153 break; 154 default: 155 break; 156 } 157 } 158 159 160 void db_PrintDoubleVector(double *a,long size) 161 { 162 printf("[ "); 163 for(long i=0;i<size;i++) printf("%lf ",a[i]); 164 printf("]"); 165 } 166 167 void db_PrintDoubleMatrix(double *a,long rows,long cols) 168 { 169 printf("[\n"); 170 for(long i=0;i<rows;i++) 171 { 172 for(long j=0;j<cols;j++) printf("%lf ",a[i*cols+j]); 173 printf("\n"); 174 } 175 printf("]"); 176 } 177