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