1 /*
2  * Copyright (C) 2008 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 #ifndef bts_RBFMAP2D_EM_H
18 #define bts_RBFMAP2D_EM_H
19 
20 /* ---- includes ----------------------------------------------------------- */
21 
22 #include "b_BasicEm/Context.h"
23 #include "b_BasicEm/Basic.h"
24 #include "b_BasicEm/Int16Arr.h"
25 #include "b_TensorEm/Int16Vec2D.h"
26 #include "b_TensorEm/Flt16Vec2D.h"
27 #include "b_TensorEm/Flt16Alt2D.h"
28 #include "b_TensorEm/Functions.h"
29 #include "b_TensorEm/Cluster2D.h"
30 #include "b_TensorEm/Int32Mat.h"
31 
32 /* ---- related objects  --------------------------------------------------- */
33 
34 /* ---- typedefs ----------------------------------------------------------- */
35 
36 /* ---- constants ---------------------------------------------------------- */
37 
38 /* data format version number */
39 #define bts_IRBFMAP2D_VERSION 100
40 
41 /* ---- object definition -------------------------------------------------- */
42 
43 /**
44  * radial basis function transformation (RBF).
45  * T( x ) = A( x ) + R( x ) where,
46  * T is the resulting overall tranformation
47  * A is a possibly linear tranformation of type altTypeE
48  * R is the rbf ( non-linear ) transformation of type typeE
49  * See member declaration for more information on typeE and altTypeE.
50  * See also
51  * 'Image Warping Using few Anchor Points and Radial Functions',
52  * Nur Arad and Daniel Reisfeld, 1994
53  */
54 struct bts_RBFMap2D
55 {
56 	/* ---- private data --------------------------------------------------- */
57 
58 	/* ---- public data ---------------------------------------------------- */
59 
60     /** type of radial basis function (enum bts_RBFType).
61 	 * one of:
62 	 * bts_RBF_IDENTITY   : no rbf deformation
63 	 * bts_RBF_LINEAr     : linear, i.e. ||r||
64 	 */
65 	int32 RBFTypeE;
66 
67 	/** src cluster, part of the RBF trafo */
68 	struct bts_Cluster2D srcClusterE;
69 
70 	/** cluster of rbf coefficients, x and y */
71 	struct bts_Cluster2D rbfCoeffClusterE;
72 
73     /** type of linear transformation (enum bts_AltType) */
74 	int32 altTypeE;
75 
76     /** affine linear transformation */
77 	struct bts_Flt16Alt2D altE;
78 
79 	/** apply only affine lnear transformation */
80 	flag altOnlyE;
81 
82 	/* ---- temporary data ------------------------------------------------- */
83 
84 	/** matrix needed for computation of rbf coefficients */
85 	struct bts_Int32Mat matE;
86 	struct bts_Int32Mat tempMatE;
87 
88 	/** arrays needed for computation of rbf coefficients */
89 	struct bbs_Int32Arr inVecE;
90 	struct bbs_Int32Arr outVecE;
91 	struct bbs_Int32Arr tempVecE;
92 };
93 
94 /* ---- associated objects ------------------------------------------------- */
95 
96 /* ---- external functions ------------------------------------------------- */
97 
98 /* ---- \ghd{ constructor/destructor } ------------------------------------- */
99 
100 /** initializes RBFMap */
101 void bts_RBFMap2D_init( struct bbs_Context* cpA,
102 					    struct bts_RBFMap2D* ptrA );
103 
104 /** destroys RBFMap */
105 void bts_RBFMap2D_exit( struct bbs_Context* cpA,
106 					    struct bts_RBFMap2D* ptrA );
107 
108 /* ---- \ghd{ operators } -------------------------------------------------- */
109 
110 /** copies RBFMap */
111 void bts_RBFMap2D_copy( struct bbs_Context* cpA,
112 					    struct bts_RBFMap2D* ptrA,
113 						const struct bts_RBFMap2D* srcPtrA );
114 
115 /** compares RBFMap */
116 flag bts_RBFMap2D_equal( struct bbs_Context* cpA,
117 						 const struct bts_RBFMap2D* ptrA,
118 						 const struct bts_RBFMap2D* srcPtrA );
119 
120 /* ---- \ghd{ query functions } -------------------------------------------- */
121 
122 /* ---- \ghd{ modify functions } ------------------------------------------- */
123 
124 /** allocates RBFMap */
125 void bts_RBFMap2D_create( struct bbs_Context* cpA,
126 						  struct bts_RBFMap2D* ptrA,
127 						  uint32 sizeA,
128 				          struct bbs_MemSeg* mspA );
129 
130 /** computes rbf transform from 2 given clusters of same size and bbp */
131 void bts_RBFMap2D_compute( struct bbs_Context* cpA,
132 						   struct bts_RBFMap2D* ptrA,
133 						   const struct bts_Cluster2D* srcPtrA,
134 						   const struct bts_Cluster2D* dstPtrA );
135 
136 /* ---- \ghd{ memory I/O } ------------------------------------------------- */
137 
138 /** size object needs when written to memory */
139 uint32 bts_RBFMap2D_memSize( struct bbs_Context* cpA,
140 							 const struct bts_RBFMap2D* ptrA );
141 
142 /** writes object to memory; returns number of bytes written */
143 uint32 bts_RBFMap2D_memWrite( struct bbs_Context* cpA,
144 							  const struct bts_RBFMap2D* ptrA,
145 							  uint16* memPtrA );
146 
147 /** reads object from memory; returns number of bytes read */
148 uint32 bts_RBFMap2D_memRead( struct bbs_Context* cpA,
149 							 struct bts_RBFMap2D* ptrA,
150 							 const uint16* memPtrA,
151 				             struct bbs_MemSeg* mspA );
152 
153 /* ---- \ghd{ exec functions } --------------------------------------------- */
154 
155 /** vector map operation: apply rbf to a vector */
156 struct bts_Flt16Vec2D bts_RBFMap2D_mapVector( struct bbs_Context* cpA,
157 											  const struct bts_RBFMap2D* ptrA,
158 											  struct bts_Flt16Vec2D vecA );
159 
160 /** cluster map operation: apply rbf to all vectors in cluster */
161 void bts_RBFMap2D_mapCluster( struct bbs_Context* cpA,
162 							  const struct bts_RBFMap2D* ptrA,
163 							  const struct bts_Cluster2D* srcPtrA,
164 							  struct bts_Cluster2D* dstPtrA,
165 							  int32 dstBbpA );
166 
167 #endif /* bts_RBFMAP2D_EM_H */
168 
169