1 /* 2 * Copyright (C) 2012 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 #include <malloc.h> 18 #include <string.h> 19 20 #include "RenderScript.h" 21 #include "rsCppInternal.h" 22 23 using namespace android; 24 using namespace RSC; 25 26 android::RSC::sp<const Element> Element::getSubElement(uint32_t index) { 27 if (!mVisibleElementMap.size()) { 28 mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Element contains no sub-elements"); 29 return NULL; 30 } 31 if (index >= mVisibleElementMap.size()) { 32 mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Illegal sub-element index"); 33 return NULL; 34 } 35 return mElements[mVisibleElementMap[index]]; 36 } 37 38 const char * Element::getSubElementName(uint32_t index) { 39 if (!mVisibleElementMap.size()) { 40 mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Element contains no sub-elements"); 41 return NULL; 42 } 43 if (index >= mVisibleElementMap.size()) { 44 mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Illegal sub-element index"); 45 return NULL; 46 } 47 return mElementNames[mVisibleElementMap[index]].c_str(); 48 } 49 50 size_t Element::getSubElementArraySize(uint32_t index) { 51 if (!mVisibleElementMap.size()) { 52 mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Element contains no sub-elements"); 53 return 0; 54 } 55 if (index >= mVisibleElementMap.size()) { 56 mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Illegal sub-element index"); 57 return 0; 58 } 59 return mArraySizes[mVisibleElementMap[index]]; 60 } 61 62 uint32_t Element::getSubElementOffsetBytes(uint32_t index) { 63 if (mVisibleElementMap.size()) { 64 mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Element contains no sub-elements"); 65 return 0; 66 } 67 if (index >= mVisibleElementMap.size()) { 68 mRS->throwError(RS_ERROR_INVALID_PARAMETER, "Illegal sub-element index"); 69 return 0; 70 } 71 return mOffsetInBytes[mVisibleElementMap[index]]; 72 } 73 74 75 #define CREATE_USER(N, T) android::RSC::sp<const Element> Element::N(android::RSC::sp<RS> rs) { \ 76 if (rs->mElements.N == NULL) { \ 77 rs->mElements.N = (createUser(rs, RS_TYPE_##T)); \ 78 } \ 79 return rs->mElements.N; \ 80 } 81 82 CREATE_USER(BOOLEAN, BOOLEAN); 83 CREATE_USER(U8, UNSIGNED_8); 84 CREATE_USER(I8, SIGNED_8); 85 CREATE_USER(U16, UNSIGNED_16); 86 CREATE_USER(I16, SIGNED_16); 87 CREATE_USER(U32, UNSIGNED_32); 88 CREATE_USER(I32, SIGNED_32); 89 CREATE_USER(U64, UNSIGNED_64); 90 CREATE_USER(I64, SIGNED_64); 91 CREATE_USER(F32, FLOAT_32); 92 CREATE_USER(F64, FLOAT_64); 93 CREATE_USER(ELEMENT, ELEMENT); 94 CREATE_USER(TYPE, TYPE); 95 CREATE_USER(ALLOCATION, ALLOCATION); 96 CREATE_USER(SAMPLER, SAMPLER); 97 CREATE_USER(SCRIPT, SCRIPT); 98 CREATE_USER(MATRIX_4X4, MATRIX_4X4); 99 CREATE_USER(MATRIX_3X3, MATRIX_3X3); 100 CREATE_USER(MATRIX_2X2, MATRIX_2X2); 101 102 #define CREATE_PIXEL(N, T, K) android::RSC::sp<const Element> Element::N(android::RSC::sp<RS> rs) { \ 103 if (rs->mElements.N == NULL) { \ 104 rs->mElements.N = createPixel(rs, RS_TYPE_##T, RS_KIND_##K); \ 105 } \ 106 return rs->mElements.N; \ 107 } 108 109 CREATE_PIXEL(A_8, UNSIGNED_8, PIXEL_A); 110 CREATE_PIXEL(RGB_565, UNSIGNED_5_6_5, PIXEL_RGB); 111 CREATE_PIXEL(RGB_888, UNSIGNED_8, PIXEL_RGB); 112 CREATE_PIXEL(RGBA_4444, UNSIGNED_4_4_4_4, PIXEL_RGBA); 113 CREATE_PIXEL(RGBA_8888, UNSIGNED_8, PIXEL_RGBA); 114 CREATE_PIXEL(YUV, UNSIGNED_8, PIXEL_YUV); 115 CREATE_PIXEL(RGBA_5551, UNSIGNED_5_5_5_1, PIXEL_RGBA); 116 117 #define CREATE_VECTOR(N, T) android::RSC::sp<const Element> Element::N##_2(android::RSC::sp<RS> rs) { \ 118 if (rs->mElements.N##_2 == NULL) { \ 119 rs->mElements.N##_2 = createVector(rs, RS_TYPE_##T, 2); \ 120 } \ 121 return rs->mElements.N##_2; \ 122 } \ 123 android::RSC::sp<const Element> Element::N##_3(android::RSC::sp<RS> rs) { \ 124 if (rs->mElements.N##_3 == NULL) { \ 125 rs->mElements.N##_3 = createVector(rs, RS_TYPE_##T, 3); \ 126 } \ 127 return rs->mElements.N##_3; \ 128 } \ 129 android::RSC::sp<const Element> Element::N##_4(android::RSC::sp<RS> rs) { \ 130 if (rs->mElements.N##_4 == NULL) { \ 131 rs->mElements.N##_4 = createVector(rs, RS_TYPE_##T, 4); \ 132 } \ 133 return rs->mElements.N##_4; \ 134 } 135 CREATE_VECTOR(U8, UNSIGNED_8); 136 CREATE_VECTOR(I8, SIGNED_8); 137 CREATE_VECTOR(U16, UNSIGNED_16); 138 CREATE_VECTOR(I16, SIGNED_16); 139 CREATE_VECTOR(U32, UNSIGNED_32); 140 CREATE_VECTOR(I32, SIGNED_32); 141 CREATE_VECTOR(U64, UNSIGNED_64); 142 CREATE_VECTOR(I64, SIGNED_64); 143 CREATE_VECTOR(F32, FLOAT_32); 144 CREATE_VECTOR(F64, FLOAT_64); 145 146 147 void Element::updateVisibleSubElements() { 148 if (!mElements.size()) { 149 return; 150 } 151 mVisibleElementMap.clear(); 152 153 int noPaddingFieldCount = 0; 154 size_t fieldCount = mElementNames.size(); 155 // Find out how many elements are not padding 156 for (size_t ct = 0; ct < fieldCount; ct ++) { 157 if (mElementNames[ct].c_str()[0] != '#') { 158 noPaddingFieldCount ++; 159 } 160 } 161 162 // Make a map that points us at non-padding elements 163 for (size_t ct = 0; ct < fieldCount; ct ++) { 164 if (mElementNames[ct].c_str()[0] != '#') { 165 mVisibleElementMap.push_back((uint32_t)ct); 166 } 167 } 168 } 169 170 Element::Element(void *id, android::RSC::sp<RS> rs, 171 std::vector<android::RSC::sp<Element> > &elements, 172 std::vector<std::string> &elementNames, 173 std::vector<uint32_t> &arraySizes) : BaseObj(id, rs) { 174 mSizeBytes = 0; 175 mVectorSize = 1; 176 mElements = elements; 177 mArraySizes = arraySizes; 178 mElementNames = elementNames; 179 180 mType = RS_TYPE_NONE; 181 mKind = RS_KIND_USER; 182 183 for (size_t ct = 0; ct < mElements.size(); ct++ ) { 184 mOffsetInBytes.push_back(mSizeBytes); 185 mSizeBytes += mElements[ct]->mSizeBytes * mArraySizes[ct]; 186 } 187 updateVisibleSubElements(); 188 } 189 190 191 static uint32_t GetSizeInBytesForType(RsDataType dt) { 192 switch(dt) { 193 case RS_TYPE_NONE: 194 return 0; 195 case RS_TYPE_SIGNED_8: 196 case RS_TYPE_UNSIGNED_8: 197 case RS_TYPE_BOOLEAN: 198 return 1; 199 200 case RS_TYPE_FLOAT_16: 201 case RS_TYPE_SIGNED_16: 202 case RS_TYPE_UNSIGNED_16: 203 case RS_TYPE_UNSIGNED_5_6_5: 204 case RS_TYPE_UNSIGNED_5_5_5_1: 205 case RS_TYPE_UNSIGNED_4_4_4_4: 206 return 2; 207 208 case RS_TYPE_FLOAT_32: 209 case RS_TYPE_SIGNED_32: 210 case RS_TYPE_UNSIGNED_32: 211 return 4; 212 213 case RS_TYPE_FLOAT_64: 214 case RS_TYPE_SIGNED_64: 215 case RS_TYPE_UNSIGNED_64: 216 return 8; 217 218 case RS_TYPE_MATRIX_4X4: 219 return 16 * 4; 220 case RS_TYPE_MATRIX_3X3: 221 return 9 * 4; 222 case RS_TYPE_MATRIX_2X2: 223 return 4 * 4; 224 225 case RS_TYPE_TYPE: 226 case RS_TYPE_ALLOCATION: 227 case RS_TYPE_SAMPLER: 228 case RS_TYPE_SCRIPT: 229 case RS_TYPE_MESH: 230 case RS_TYPE_PROGRAM_FRAGMENT: 231 case RS_TYPE_PROGRAM_VERTEX: 232 case RS_TYPE_PROGRAM_RASTER: 233 case RS_TYPE_PROGRAM_STORE: 234 return 4; 235 236 default: 237 break; 238 } 239 240 ALOGE("Missing type %i", dt); 241 return 0; 242 } 243 244 Element::Element(void *id, android::RSC::sp<RS> rs, 245 RsDataType dt, RsDataKind dk, bool norm, uint32_t size) : 246 BaseObj(id, rs) 247 { 248 uint32_t tsize = GetSizeInBytesForType(dt); 249 if ((dt != RS_TYPE_UNSIGNED_5_6_5) && 250 (dt != RS_TYPE_UNSIGNED_4_4_4_4) && 251 (dt != RS_TYPE_UNSIGNED_5_5_5_1)) { 252 if (size == 3) { 253 mSizeBytes = tsize * 4; 254 } else { 255 mSizeBytes = tsize * size; 256 } 257 } else { 258 mSizeBytes = tsize; 259 } 260 mType = dt; 261 mKind = dk; 262 mNormalized = norm; 263 mVectorSize = size; 264 } 265 266 Element::~Element() { 267 } 268 269 void Element::updateFromNative() { 270 BaseObj::updateFromNative(); 271 updateVisibleSubElements(); 272 } 273 274 android::RSC::sp<const Element> Element::createUser(android::RSC::sp<RS> rs, RsDataType dt) { 275 void * id = RS::dispatch->ElementCreate(rs->getContext(), dt, RS_KIND_USER, false, 1); 276 return new Element(id, rs, dt, RS_KIND_USER, false, 1); 277 } 278 279 android::RSC::sp<const Element> Element::createVector(android::RSC::sp<RS> rs, RsDataType dt, uint32_t size) { 280 if (size < 2 || size > 4) { 281 rs->throwError(RS_ERROR_INVALID_PARAMETER, "Vector size out of range 2-4."); 282 return NULL; 283 } 284 void *id = RS::dispatch->ElementCreate(rs->getContext(), dt, RS_KIND_USER, false, size); 285 return new Element(id, rs, dt, RS_KIND_USER, false, size); 286 } 287 288 android::RSC::sp<const Element> Element::createPixel(android::RSC::sp<RS> rs, RsDataType dt, RsDataKind dk) { 289 if (!(dk == RS_KIND_PIXEL_L || 290 dk == RS_KIND_PIXEL_A || 291 dk == RS_KIND_PIXEL_LA || 292 dk == RS_KIND_PIXEL_RGB || 293 dk == RS_KIND_PIXEL_RGBA || 294 dk == RS_KIND_PIXEL_DEPTH)) { 295 rs->throwError(RS_ERROR_INVALID_PARAMETER, "Unsupported DataKind"); 296 return NULL; 297 } 298 if (!(dt == RS_TYPE_UNSIGNED_8 || 299 dt == RS_TYPE_UNSIGNED_16 || 300 dt == RS_TYPE_UNSIGNED_5_6_5 || 301 dt == RS_TYPE_UNSIGNED_4_4_4_4 || 302 dt == RS_TYPE_UNSIGNED_5_5_5_1)) { 303 rs->throwError(RS_ERROR_INVALID_PARAMETER, "Unsupported DataType"); 304 return NULL; 305 } 306 if (dt == RS_TYPE_UNSIGNED_5_6_5 && dk != RS_KIND_PIXEL_RGB) { 307 rs->throwError(RS_ERROR_INVALID_PARAMETER, "Bad kind and type combo"); 308 return NULL; 309 } 310 if (dt == RS_TYPE_UNSIGNED_5_5_5_1 && dk != RS_KIND_PIXEL_RGBA) { 311 rs->throwError(RS_ERROR_INVALID_PARAMETER, "Bad kind and type combo"); 312 return NULL; 313 } 314 if (dt == RS_TYPE_UNSIGNED_4_4_4_4 && dk != RS_KIND_PIXEL_RGBA) { 315 rs->throwError(RS_ERROR_INVALID_PARAMETER, "Bad kind and type combo"); 316 return NULL; 317 } 318 if (dt == RS_TYPE_UNSIGNED_16 && dk != RS_KIND_PIXEL_DEPTH) { 319 rs->throwError(RS_ERROR_INVALID_PARAMETER, "Bad kind and type combo"); 320 return NULL; 321 } 322 323 int size = 1; 324 switch (dk) { 325 case RS_KIND_PIXEL_LA: 326 size = 2; 327 break; 328 case RS_KIND_PIXEL_RGB: 329 size = 3; 330 break; 331 case RS_KIND_PIXEL_RGBA: 332 size = 4; 333 break; 334 case RS_KIND_PIXEL_DEPTH: 335 size = 2; 336 break; 337 default: 338 break; 339 } 340 341 void * id = RS::dispatch->ElementCreate(rs->getContext(), dt, dk, true, size); 342 return new Element(id, rs, dt, dk, true, size); 343 } 344 345 bool Element::isCompatible(android::RSC::sp<const Element>e) const { 346 // Try strict BaseObj equality to start with. 347 if (this == e.get()) { 348 return true; 349 } 350 351 // Ignore mKind because it is allowed to be different (user vs. pixel). 352 // We also ignore mNormalized because it can be different. The mType 353 // field must be non-null since we require name equivalence for 354 // user-created Elements. 355 return ((mSizeBytes == e->mSizeBytes) && 356 (mType != RS_TYPE_NONE) && 357 (mType == e->mType) && 358 (mVectorSize == e->mVectorSize)); 359 } 360 361 Element::Builder::Builder(android::RSC::sp<RS> rs) { 362 mRS = rs.get(); 363 mSkipPadding = false; 364 } 365 366 void Element::Builder::add(android::RSC::sp</*const*/ Element>e, std::string &name, uint32_t arraySize) { 367 // Skip padding fields after a vector 3 type. 368 if (mSkipPadding) { 369 const char *s1 = "#padding_"; 370 const char *s2 = name.c_str(); 371 size_t len = strlen(s1); 372 if (strlen(s2) >= len) { 373 if (!memcmp(s1, s2, len)) { 374 mSkipPadding = false; 375 return; 376 } 377 } 378 } 379 380 if (e->mVectorSize == 3) { 381 mSkipPadding = true; 382 } else { 383 mSkipPadding = false; 384 } 385 386 mElements.push_back(e); 387 mElementNames.push_back(name); 388 mArraySizes.push_back(arraySize); 389 } 390 391 android::RSC::sp<const Element> Element::Builder::create() { 392 size_t fieldCount = mElements.size(); 393 const char ** nameArray = (const char **)calloc(fieldCount, sizeof(char *)); 394 const Element ** elementArray = (const Element **)calloc(fieldCount, sizeof(Element *)); 395 size_t* sizeArray = (size_t*)calloc(fieldCount, sizeof(size_t)); 396 397 for (size_t ct = 0; ct < fieldCount; ct++) { 398 nameArray[ct] = mElementNames[ct].c_str(); 399 elementArray[ct] = mElements[ct].get(); 400 sizeArray[ct] = mElementNames[ct].length(); 401 } 402 403 void *id = RS::dispatch->ElementCreate2(mRS->getContext(), 404 (RsElement *)elementArray, fieldCount, 405 nameArray, fieldCount * sizeof(size_t), sizeArray, 406 (const uint32_t *)&mArraySizes[0], fieldCount); 407 408 409 free(nameArray); 410 free(sizeArray); 411 free(elementArray); 412 return new Element(id, mRS, mElements, mElementNames, mArraySizes); 413 } 414 415