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 /* ---- includes ----------------------------------------------------------- */
18 
19 #include "b_BasicEm/Functions.h"
20 #include "b_BasicEm/Math.h"
21 #include "b_TensorEm/Mat.h"
22 
23 /* ------------------------------------------------------------------------- */
24 
25 /* ========================================================================= */
26 /*                                                                           */
27 /* ---- \ghd{ auxiliary functions } ---------------------------------------- */
28 /*                                                                           */
29 /* ========================================================================= */
30 
31 /* ------------------------------------------------------------------------- */
32 
33 /* ========================================================================= */
34 /*                                                                           */
35 /* ---- \ghd{ constructor / destructor } ----------------------------------- */
36 /*                                                                           */
37 /* ========================================================================= */
38 
39 /* ------------------------------------------------------------------------- */
40 
bts_Mat_init(struct bbs_Context * cpA,struct bts_Mat * ptrA)41 void bts_Mat_init( struct bbs_Context* cpA,
42 				   struct bts_Mat* ptrA )
43 {
44 	bts_VectorMap_init( cpA, &ptrA->baseE );
45 	ptrA->baseE.typeE = ( uint32 )bts_VM_MAT;
46 	ptrA->baseE.vpMapE = bts_Mat_map;
47 
48 	bts_CompactMat_init( cpA, &ptrA->matE );
49 }
50 
51 /* ------------------------------------------------------------------------- */
52 
bts_Mat_exit(struct bbs_Context * cpA,struct bts_Mat * ptrA)53 void bts_Mat_exit( struct bbs_Context* cpA,
54 				   struct bts_Mat* ptrA )
55 {
56 	bts_CompactMat_exit( cpA, &ptrA->matE );
57 
58 	bts_VectorMap_exit( cpA, &ptrA->baseE );
59 }
60 
61 /* ------------------------------------------------------------------------- */
62 
63 /* ========================================================================= */
64 /*                                                                           */
65 /* ---- \ghd{ operators } -------------------------------------------------- */
66 /*                                                                           */
67 /* ========================================================================= */
68 
69 /* ------------------------------------------------------------------------- */
70 
bts_Mat_copy(struct bbs_Context * cpA,struct bts_Mat * ptrA,const struct bts_Mat * srcPtrA)71 void bts_Mat_copy( struct bbs_Context* cpA,
72 				   struct bts_Mat* ptrA,
73 				   const struct bts_Mat* srcPtrA )
74 {
75 	bts_CompactMat_copy( cpA, &ptrA->matE, &srcPtrA->matE );
76 }
77 
78 /* ------------------------------------------------------------------------- */
79 
bts_Mat_equal(struct bbs_Context * cpA,const struct bts_Mat * ptrA,const struct bts_Mat * srcPtrA)80 flag bts_Mat_equal( struct bbs_Context* cpA,
81 					const struct bts_Mat* ptrA,
82 					const struct bts_Mat* srcPtrA )
83 {
84 	bbs_ERROR0( "bts_Mat_equal:\n Function is not available" );
85 	return FALSE;
86 }
87 
88 /* ------------------------------------------------------------------------- */
89 
90 /* ========================================================================= */
91 /*                                                                           */
92 /* ---- \ghd{ query functions } -------------------------------------------- */
93 /*                                                                           */
94 /* ========================================================================= */
95 
96 /* ------------------------------------------------------------------------- */
97 
98 /* ========================================================================= */
99 /*                                                                           */
100 /* ---- \ghd{ modify functions } ------------------------------------------- */
101 /*                                                                           */
102 /* ========================================================================= */
103 
104 /* ------------------------------------------------------------------------- */
105 
106 /* ========================================================================= */
107 /*                                                                           */
108 /* ---- \ghd{ I/O } -------------------------------------------------------- */
109 /*                                                                           */
110 /* ========================================================================= */
111 
112 /* ------------------------------------------------------------------------- */
113 
bts_Mat_memSize(struct bbs_Context * cpA,const struct bts_Mat * ptrA)114 uint32 bts_Mat_memSize( struct bbs_Context* cpA,
115 					    const struct bts_Mat* ptrA )
116 {
117 	uint32 memSizeL = bbs_SIZEOF16( uint32 ) +
118 					  bbs_SIZEOF16( uint32 ); /* version */
119 
120 	memSizeL += bts_VectorMap_memSize( cpA, &ptrA->baseE );
121 	memSizeL += bts_CompactMat_memSize( cpA, &ptrA->matE );
122 
123 	return memSizeL;
124 }
125 
126 /* ------------------------------------------------------------------------- */
127 
bts_Mat_memWrite(struct bbs_Context * cpA,const struct bts_Mat * ptrA,uint16 * memPtrA)128 uint32 bts_Mat_memWrite( struct bbs_Context* cpA,
129 						 const struct bts_Mat* ptrA,
130 						 uint16* memPtrA )
131 {
132 	uint32 memSizeL = bts_Mat_memSize( cpA, ptrA );
133 	memPtrA += bbs_memWrite32( &memSizeL, memPtrA );
134 	memPtrA += bbs_memWriteUInt32( bts_MAT_VERSION, memPtrA );
135 	memPtrA += bts_VectorMap_memWrite( cpA, &ptrA->baseE, memPtrA );
136 	memPtrA += bts_CompactMat_memWrite( cpA, &ptrA->matE, memPtrA );
137 	return memSizeL;
138 }
139 
140 /* ------------------------------------------------------------------------- */
141 
bts_Mat_memRead(struct bbs_Context * cpA,struct bts_Mat * ptrA,const uint16 * memPtrA,struct bbs_MemTbl * mtpA)142 uint32 bts_Mat_memRead( struct bbs_Context* cpA,
143 						struct bts_Mat* ptrA,
144 						const uint16* memPtrA,
145 						struct bbs_MemTbl* mtpA )
146 {
147 	uint32 memSizeL, versionL;
148 	struct bbs_MemTbl memTblL = *mtpA;
149 	struct bbs_MemSeg* espL = bbs_MemTbl_segPtr( cpA, &memTblL, 0 );
150 
151 	if( bbs_Context_error( cpA ) ) return 0;
152 	memPtrA += bbs_memRead32( &memSizeL, memPtrA );
153 	memPtrA += bbs_memReadVersion32( cpA, &versionL, bts_MAT_VERSION, memPtrA );
154 	memPtrA += bts_VectorMap_memRead( cpA, &ptrA->baseE, memPtrA );
155 	memPtrA += bts_CompactMat_memRead( cpA, &ptrA->matE, memPtrA, espL );
156 
157 	if( memSizeL != bts_Mat_memSize( cpA, ptrA ) )
158 	{
159 		bbs_ERR0( bbs_ERR_CORRUPT_DATA, "uint32 bts_Mat_memRead( struct bem_ScanGradientMove* ptrA, const uint16* memPtrA ):\n"
160 			        "size mismatch" );
161 		return 0;
162 	}
163 
164 	return memSizeL;
165 }
166 
167 /* ------------------------------------------------------------------------- */
168 
169 /* ========================================================================= */
170 /*                                                                           */
171 /* ---- \ghd{ exec functions } --------------------------------------------- */
172 /*                                                                           */
173 /* ========================================================================= */
174 
175 /* ------------------------------------------------------------------------- */
176 
bts_Mat_map(struct bbs_Context * cpA,const struct bts_VectorMap * ptrA,const struct bts_Flt16Vec * inVecPtrA,struct bts_Flt16Vec * outVecPtrA)177 void bts_Mat_map( struct bbs_Context* cpA,
178 				  const struct bts_VectorMap* ptrA,
179 				  const struct bts_Flt16Vec* inVecPtrA,
180 				  struct bts_Flt16Vec* outVecPtrA )
181 {
182 	bbs_DEF_fNameL( "bts_Mat_map" )
183 	const struct bts_Mat* ptrL = ( const struct bts_Mat* )ptrA;
184 
185 	if( inVecPtrA->arrE.sizeE != ptrL->matE.widthE )
186 	{
187 		bbs_ERROR1( "%s:\ninput vector has incorrect size", fNameL );
188 		return;
189 	}
190 
191 	if( outVecPtrA->arrE.allocatedSizeE < ptrL->matE.heightE )
192 	{
193 		bbs_ERROR1( "%s:\noutput vector is insufficiently allocated", fNameL );
194 		return;
195 	}
196 
197 	bts_Flt16Vec_size( cpA, outVecPtrA, ptrL->matE.heightE );
198 
199 	{
200 		int16 expL = 0;
201 		int32 outExpL = inVecPtrA->expE;
202 		bts_CompactMat_map( cpA, &ptrL->matE, inVecPtrA->arrE.arrPtrE, outVecPtrA->arrE.arrPtrE, &expL );
203 		outExpL += expL;
204 
205 		/* precision underflow */
206 		if( outExpL < -32767 ) bts_Flt16Vec_setZero( cpA, outVecPtrA );
207 	}
208 
209 	bts_Flt16Vec_maximizeMantisse( cpA, outVecPtrA );
210 }
211 
212 /* ------------------------------------------------------------------------- */
213 
214 /* ========================================================================= */
215 
216