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
23float** 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
36unsigned 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}
50float** 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
60unsigned 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
70void db_FreeImage_f(float **img,int h)
71{
72    delete [] (img[0]);
73    delete [] img;
74}
75
76void 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// ------------------------------------------------------------------------------------------------------------ ;
89void 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
102inline 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
123inline 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
143void 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
160void 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
167void 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