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
41void 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
53void 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
71void 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
80flag 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
114uint32 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
128uint32 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
142uint32 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
177void 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