Home | History | Annotate | Download | only in rs 
      1 
      2 /*
      3  * Copyright (C) 2009 The Android Open Source Project
      4  *
      5  * Licensed under the Apache License, Version 2.0 (the "License");
      6  * you may not use this file except in compliance with the License.
      7  * You may obtain a copy of the License at
      8  *
      9  *      http://www.apache.org/licenses/LICENSE-2.0
     10  *
     11  * Unless required by applicable law or agreed to in writing, software
     12  * distributed under the License is distributed on an "AS IS" BASIS,
     13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     14  * See the License for the specific language governing permissions and
     15  * limitations under the License.
     16  */
     17 
     18 #include "rsContext.h"
     19 #include "rsStream.h"
     20 
     21 namespace android {
     22 namespace renderscript {
     23 
     24 IStream::IStream(const uint8_t *buf, bool use64) {
     25     mData = buf;
     26     mPos = 0;
     27     mUse64 = use64;
     28 }
     29 
     30 void IStream::loadByteArray(void *dest, size_t numBytes) {
     31     memcpy(dest, mData + mPos, numBytes);
     32     mPos += numBytes;
     33 }
     34 
     35 uint64_t IStream::loadOffset() {
     36     uint64_t tmp;
     37     if (mUse64) {
     38         mPos = (mPos + 7) & (~7);
     39         tmp = reinterpret_cast<const uint64_t *>(&mData[mPos])[0];
     40         mPos += sizeof(uint64_t);
     41         return tmp;
     42     }
     43     return loadU32();
     44 }
     45 
     46 const char * IStream::loadString() {
     47     uint32_t len = loadU32();
     48     const char *s = rsuCopyString((const char *)&mData[mPos], len);
     49     mPos += len;
     50     return s;
     51 }
     52 
     53 // Output stream implementation
     54 OStream::OStream(uint64_t len, bool use64) {
     55     mData = (uint8_t*)malloc(len);
     56     mLength = len;
     57     mPos = 0;
     58     mUse64 = use64;
     59 }
     60 
     61 OStream::~OStream() {
     62     free(mData);
     63 }
     64 
     65 void OStream::addByteArray(const void *src, size_t numBytes) {
     66     // We need to potentially grow more than once if the number of byes we write is substantial
     67     while (mPos + numBytes >= mLength) {
     68         growSize();
     69     }
     70     memcpy(mData + mPos, src, numBytes);
     71     mPos += numBytes;
     72 }
     73 
     74 void OStream::addOffset(uint64_t v) {
     75     if (mUse64) {
     76         mPos = (mPos + 7) & (~7);
     77         if (mPos + sizeof(v) >= mLength) {
     78             growSize();
     79         }
     80         mData[mPos++] = (uint8_t)(v & 0xff);
     81         mData[mPos++] = (uint8_t)((v >> 8) & 0xff);
     82         mData[mPos++] = (uint8_t)((v >> 16) & 0xff);
     83         mData[mPos++] = (uint8_t)((v >> 24) & 0xff);
     84         mData[mPos++] = (uint8_t)((v >> 32) & 0xff);
     85         mData[mPos++] = (uint8_t)((v >> 40) & 0xff);
     86         mData[mPos++] = (uint8_t)((v >> 48) & 0xff);
     87         mData[mPos++] = (uint8_t)((v >> 56) & 0xff);
     88     } else {
     89         addU32(v);
     90     }
     91 }
     92 
     93 void OStream::addString(const char *s, size_t len) {
     94     addU32(len);
     95     if (mPos + len*sizeof(char) >= mLength) {
     96         growSize();
     97     }
     98     char *stringData = reinterpret_cast<char *>(&mData[mPos]);
     99     memcpy(stringData, s, len);
    100     mPos += len*sizeof(char);
    101 }
    102 
    103 void OStream::addString(const char *s) {
    104     addString(s, strlen(s));
    105 }
    106 
    107 void OStream::growSize() {
    108     uint8_t *newData = (uint8_t*)malloc(mLength*2);
    109     memcpy(newData, mData, mLength*sizeof(uint8_t));
    110     mLength = mLength * 2;
    111     free(mData);
    112     mData = newData;
    113 }
    114 
    115 } // namespace renderscript
    116 } // namespace android
    117