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
db_SetupImageReferences_f(float * im,int w,int h)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
db_SetupImageReferences_u(unsigned char * im,int w,int h)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 }
db_AllocImage_f(int w,int h,int over_allocation)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
db_AllocImage_u(int w,int h,int over_allocation)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
db_FreeImage_f(float ** img,int h)70 void db_FreeImage_f(float **img,int h)
71 {
72 delete [] (img[0]);
73 delete [] img;
74 }
75
db_FreeImage_u(unsigned char ** img,int h)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 // ------------------------------------------------------------------------------------------------------------ ;
db_CopyImage_u(unsigned char ** d,const unsigned char * const * s,int w,int h,int over_allocation)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
db_WarpImageLutFast_u(const unsigned char * const * src,unsigned char ** dst,int w,int h,const float * const * lut_x,const float * const * lut_y)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
db_WarpImageLutBilinear_u(const unsigned char * const * src,unsigned char ** dst,int w,int h,const float * const * lut_x,const float * const * lut_y)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
db_WarpImageLut_u(const unsigned char * const * src,unsigned char ** dst,int w,int h,const float * const * lut_x,const float * const * lut_y,int type)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
db_PrintDoubleVector(double * a,long size)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
db_PrintDoubleMatrix(double * a,long rows,long cols)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