blob: d401cd127de9008af4b6a285fe456545730cecd1 [file] [log] [blame]
/*
* Copyright (C) 2009 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef ANDROID_RS_STREAM_H
#define ANDROID_RS_STREAM_H
#include <utils/String8.h>
#include <stdio.h>
// ---------------------------------------------------------------------------
namespace android {
namespace renderscript {
class IStream
{
public:
IStream(const uint8_t *, bool use64);
float loadF() {
mPos = (mPos + 3) & (~3);
float tmp = reinterpret_cast<const float *>(&mData[mPos])[0];
mPos += sizeof(float);
return tmp;
}
int32_t loadI32() {
mPos = (mPos + 3) & (~3);
int32_t tmp = reinterpret_cast<const int32_t *>(&mData[mPos])[0];
mPos += sizeof(int32_t);
return tmp;
}
uint32_t loadU32() {
mPos = (mPos + 3) & (~3);
uint32_t tmp = reinterpret_cast<const uint32_t *>(&mData[mPos])[0];
mPos += sizeof(uint32_t);
return tmp;
}
uint16_t loadU16() {
mPos = (mPos + 1) & (~1);
uint16_t tmp = reinterpret_cast<const uint16_t *>(&mData[mPos])[0];
mPos += sizeof(uint16_t);
return tmp;
}
inline uint8_t loadU8() {
uint8_t tmp = reinterpret_cast<const uint8_t *>(&mData[mPos])[0];
mPos += sizeof(uint8_t);
return tmp;
}
void loadByteArray(void *dest, size_t numBytes);
uint64_t loadOffset();
void loadString(String8 *s);
uint64_t getPos() const {
return mPos;
}
void reset(uint64_t pos) {
mPos = pos;
}
void reset() {
mPos = 0;
}
const uint8_t * getPtr() const {
return mData;
}
protected:
const uint8_t * mData;
uint64_t mPos;
bool mUse64;
};
class OStream
{
public:
OStream(uint64_t length, bool use64);
~OStream();
void align(uint32_t bytes) {
mPos = (mPos + (bytes - 1)) & (~(bytes - 1));
if(mPos >= mLength) {
growSize();
}
}
void addF(float v) {
uint32_t uintV = *reinterpret_cast<uint32_t*> (&v);
addU32(uintV);
}
void addI32(int32_t v) {
mPos = (mPos + 3) & (~3);
if(mPos + sizeof(v) >= mLength) {
growSize();
}
mData[mPos++] = (uint8_t)(v & 0xff);
mData[mPos++] = (uint8_t)((v >> 8) & 0xff);
mData[mPos++] = (uint8_t)((v >> 16) & 0xff);
mData[mPos++] = (uint8_t)((v >> 24) & 0xff);
}
void addU32(uint32_t v) {
mPos = (mPos + 3) & (~3);
if(mPos + sizeof(v) >= mLength) {
growSize();
}
mData[mPos++] = (uint8_t)(v & 0xff);
mData[mPos++] = (uint8_t)((v >> 8) & 0xff);
mData[mPos++] = (uint8_t)((v >> 16) & 0xff);
mData[mPos++] = (uint8_t)((v >> 24) & 0xff);
}
void addU16(uint16_t v) {
mPos = (mPos + 1) & (~1);
if(mPos + sizeof(v) >= mLength) {
growSize();
}
mData[mPos++] = (uint8_t)(v & 0xff);
mData[mPos++] = (uint8_t)(v >> 8);
}
inline void addU8(uint8_t v) {
if(mPos + 1 >= mLength) {
growSize();
}
reinterpret_cast<uint8_t *>(&mData[mPos])[0] = v;
mPos ++;
}
void addByteArray(const void *src, size_t numBytes);
void addOffset(uint64_t v);
void addString(String8 *s);
uint64_t getPos() const {
return mPos;
}
void reset(uint64_t pos) {
mPos = pos;
}
void reset() {
mPos = 0;
}
const uint8_t * getPtr() const {
return mData;
}
protected:
void growSize();
uint8_t * mData;
uint64_t mLength;
uint64_t mPos;
bool mUse64;
};
} // renderscript
} // android
#endif //ANDROID_RS_STREAM_H