rsAllocation.cpp revision 2d1220c27ae91f0b307f283fe66cb767b63dfe38
1326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams/*
2326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Copyright (C) 2009 The Android Open Source Project
3326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams *
4326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Licensed under the Apache License, Version 2.0 (the "License");
5326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * you may not use this file except in compliance with the License.
6326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * You may obtain a copy of the License at
7326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams *
8326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams *      http://www.apache.org/licenses/LICENSE-2.0
9326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams *
10326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * Unless required by applicable law or agreed to in writing, software
11326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * distributed under the License is distributed on an "AS IS" BASIS,
12326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * See the License for the specific language governing permissions and
14326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams * limitations under the License.
15326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams */
16326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
1777d9f4bd05b2d2a161f30c12a2248f9c97eaac42Alex Sakhartchouk#include "rsContext.h"
18eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams#include "rs_hal.h"
19eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams
2039f2ef6fed00a99c5c389e12c4597884027d4858Alex Sakhartchouk
21326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android;
22326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsusing namespace android::renderscript;
23326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
24a2aab8be7a980791fc9d4e6c4d050d703b20bcf6Alex SakhartchoukAllocation::Allocation(Context *rsc, const Type *type, uint32_t usages,
25a2aab8be7a980791fc9d4e6c4d050d703b20bcf6Alex Sakhartchouk                       RsAllocationMipmapControl mc)
26a2aab8be7a980791fc9d4e6c4d050d703b20bcf6Alex Sakhartchouk    : ObjectBase(rsc) {
27fa84da2cbc271f855b3b1ec75bb688abdf1d1d01Jason Sams
28eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    memset(&mHal, 0, sizeof(mHal));
29eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    mHal.state.mipmapControl = RS_ALLOCATION_MIPMAP_NONE;
30bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.usageFlags = usages;
31bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.mipmapControl = mc;
32366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams
33064aa7ed76db9564b041afcd4b75da5b3d12fabaAlex Sakhartchouk    setType(type);
34eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    updateCache();
35fa84da2cbc271f855b3b1ec75bb688abdf1d1d01Jason Sams}
36fa84da2cbc271f855b3b1ec75bb688abdf1d1d01Jason Sams
37eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason SamsAllocation * Allocation::createAllocation(Context *rsc, const Type *type, uint32_t usages,
38eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams                              RsAllocationMipmapControl mc) {
39eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    Allocation *a = new Allocation(rsc, type, usages, mc);
40fa84da2cbc271f855b3b1ec75bb688abdf1d1d01Jason Sams
41eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    if (!rsc->mHal.funcs.allocation.init(rsc, a, type->getElement()->getHasReferences())) {
42eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams        rsc->setError(RS_ERROR_FATAL_DRIVER, "Allocation::Allocation, alloc failure");
43eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams        delete a;
44eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams        return NULL;
45eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    }
46eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    return a;
47bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams}
48fa84da2cbc271f855b3b1ec75bb688abdf1d1d01Jason Sams
49bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Samsvoid Allocation::updateCache() {
50064aa7ed76db9564b041afcd4b75da5b3d12fabaAlex Sakhartchouk    const Type *type = mHal.state.type;
51bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.dimensionX = type->getDimX();
52bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.dimensionY = type->getDimY();
53bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.dimensionZ = type->getDimZ();
54bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.hasFaces = type->getDimFaces();
55bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.hasMipmaps = type->getDimLOD();
56bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.elementSizeBytes = type->getElementSizeBytes();
57bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.hasReferences = mHal.state.type->getElement()->getHasReferences();
58326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams}
59326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
60afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex SakhartchoukAllocation::~Allocation() {
61c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams    freeChildrenUnlocked();
62eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    mRSC->mHal.funcs.allocation.destroy(mRSC, this);
63326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams}
64326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
65366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Samsvoid Allocation::syncAll(Context *rsc, RsAllocationUsageType src) {
66eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    rsc->mHal.funcs.allocation.syncAll(rsc, this, src);
67cf4c7c9b2f513be77a5b9853319ca82ac2b128edJason Sams}
68cf4c7c9b2f513be77a5b9853319ca82ac2b128edJason Sams
69afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::read(void *data) {
70eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    memcpy(data, getPtr(), mHal.state.type->getSizeBytes());
71e579df42e85d9e00f53c42ef1b78dbd209dba989Jason Sams}
72e579df42e85d9e00f53c42ef1b78dbd209dba989Jason Sams
734b45b8998e0d7038efaea80c70d23c086640b4e3Jason Samsvoid Allocation::data(Context *rsc, uint32_t xoff, uint32_t lod,
744b45b8998e0d7038efaea80c70d23c086640b4e3Jason Sams                         uint32_t count, const void *data, uint32_t sizeBytes) {
75eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    const uint32_t eSize = mHal.state.type->getElementSizeBytes();
769397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams
77eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    if ((count * eSize) != sizeBytes) {
78eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams        LOGE("Allocation::subData called with mismatched size expected %i, got %i",
79eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams             (count * eSize), sizeBytes);
80bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams        mHal.state.type->dumpLOGV("type info");
819397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams        return;
829397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams    }
83e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams
84eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    rsc->mHal.funcs.allocation.data1D(rsc, this, xoff, lod, count, data, sizeBytes);
85eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    sendDirty(rsc);
86326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams}
87326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
884b45b8998e0d7038efaea80c70d23c086640b4e3Jason Samsvoid Allocation::data(Context *rsc, uint32_t xoff, uint32_t yoff, uint32_t lod, RsAllocationCubemapFace face,
89afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk             uint32_t w, uint32_t h, const void *data, uint32_t sizeBytes) {
90eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    const uint32_t eSize = mHal.state.elementSizeBytes;
91eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    const uint32_t lineSize = eSize * w;
92326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
93a23715148f7bda74e904fc553b70c9a49fd52a6eJason Sams    //LOGE("data2d %p,  %i %i %i %i %i %i %p %i", this, xoff, yoff, lod, face, w, h, data, sizeBytes);
949397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams
95a23715148f7bda74e904fc553b70c9a49fd52a6eJason Sams    if ((lineSize * h) != sizeBytes) {
96a23715148f7bda74e904fc553b70c9a49fd52a6eJason Sams        LOGE("Allocation size mismatch, expected %i, got %i", (lineSize * h), sizeBytes);
979397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams        rsAssert(!"Allocation::subData called with mismatched size");
989397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams        return;
999397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams    }
1009397e30ce5fe3f6af9212a93b490836b04fdfffaJason Sams
101eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    rsc->mHal.funcs.allocation.data2D(rsc, this, xoff, yoff, lod, face, w, h, data, sizeBytes);
102eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    sendDirty(rsc);
103326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams}
104326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
105236385b73f21ae65e756b9cb5738f1514d95ea3eJason Samsvoid Allocation::data(Context *rsc, uint32_t xoff, uint32_t yoff, uint32_t zoff,
106236385b73f21ae65e756b9cb5738f1514d95ea3eJason Sams                      uint32_t lod, RsAllocationCubemapFace face,
107236385b73f21ae65e756b9cb5738f1514d95ea3eJason Sams                      uint32_t w, uint32_t h, uint32_t d, const void *data, uint32_t sizeBytes) {
108326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams}
109326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
1104b45b8998e0d7038efaea80c70d23c086640b4e3Jason Samsvoid Allocation::elementData(Context *rsc, uint32_t x, const void *data,
111afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk                                uint32_t cIdx, uint32_t sizeBytes) {
112bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    uint32_t eSize = mHal.state.elementSizeBytes;
1135f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
114bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    if (cIdx >= mHal.state.type->getElement()->getFieldCount()) {
1155f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        LOGE("Error Allocation::subElementData component %i out of range.", cIdx);
1165f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        rsc->setError(RS_ERROR_BAD_VALUE, "subElementData component out of range.");
1175f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        return;
1185f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    }
1195f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
120bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    if (x >= mHal.state.dimensionX) {
1215f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        LOGE("Error Allocation::subElementData X offset %i out of range.", x);
1225f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        rsc->setError(RS_ERROR_BAD_VALUE, "subElementData X offset out of range.");
1235f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        return;
1245f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    }
1255f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
126bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    const Element * e = mHal.state.type->getElement()->getField(cIdx);
1275f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    if (sizeBytes != e->getSizeBytes()) {
128099d7d33e55afeb3399f6e8cf8d665223ca94939Alex Sakhartchouk        LOGE("Error Allocation::subElementData data size %i does not match field size %zu.", sizeBytes, e->getSizeBytes());
1295f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        rsc->setError(RS_ERROR_BAD_VALUE, "subElementData bad size.");
1305f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        return;
1315f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    }
1325f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
133eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    rsc->mHal.funcs.allocation.elementData1D(rsc, this, x, data, cIdx, sizeBytes);
134eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    sendDirty(rsc);
1355f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams}
1365f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
1374b45b8998e0d7038efaea80c70d23c086640b4e3Jason Samsvoid Allocation::elementData(Context *rsc, uint32_t x, uint32_t y,
138afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk                                const void *data, uint32_t cIdx, uint32_t sizeBytes) {
139bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    uint32_t eSize = mHal.state.elementSizeBytes;
1405f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
141bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    if (x >= mHal.state.dimensionX) {
1425f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        LOGE("Error Allocation::subElementData X offset %i out of range.", x);
1435f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        rsc->setError(RS_ERROR_BAD_VALUE, "subElementData X offset out of range.");
1445f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        return;
1455f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    }
1465f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
147bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    if (y >= mHal.state.dimensionY) {
1485f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        LOGE("Error Allocation::subElementData X offset %i out of range.", x);
1495f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        rsc->setError(RS_ERROR_BAD_VALUE, "subElementData X offset out of range.");
1505f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        return;
1515f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    }
1525f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
153bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    if (cIdx >= mHal.state.type->getElement()->getFieldCount()) {
1545f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        LOGE("Error Allocation::subElementData component %i out of range.", cIdx);
1555f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        rsc->setError(RS_ERROR_BAD_VALUE, "subElementData component out of range.");
1565f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        return;
1575f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    }
1585f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
159bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    const Element * e = mHal.state.type->getElement()->getField(cIdx);
1605f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
1615f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    if (sizeBytes != e->getSizeBytes()) {
162099d7d33e55afeb3399f6e8cf8d665223ca94939Alex Sakhartchouk        LOGE("Error Allocation::subElementData data size %i does not match field size %zu.", sizeBytes, e->getSizeBytes());
1635f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        rsc->setError(RS_ERROR_BAD_VALUE, "subElementData bad size.");
1645f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams        return;
1655f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    }
1665f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
167eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    rsc->mHal.funcs.allocation.elementData2D(rsc, this, x, y, data, cIdx, sizeBytes);
168eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    sendDirty(rsc);
1695f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams}
1705f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
171afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::addProgramToDirty(const Program *p) {
172fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    mToDirtyList.push(p);
1735c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams}
1745c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams
175afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::removeProgramToDirty(const Program *p) {
1765c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams    for (size_t ct=0; ct < mToDirtyList.size(); ct++) {
1775c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams        if (mToDirtyList[ct] == p) {
1785c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams            mToDirtyList.removeAt(ct);
1795c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams            return;
1805c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams        }
1815c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams    }
1825c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams    rsAssert(0);
1835c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams}
184326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
185afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::dumpLOGV(const char *prefix) const {
186c21cf40f6ae69091bf24f87b5eeabc95e73dd271Jason Sams    ObjectBase::dumpLOGV(prefix);
187c21cf40f6ae69091bf24f87b5eeabc95e73dd271Jason Sams
188c21cf40f6ae69091bf24f87b5eeabc95e73dd271Jason Sams    String8 s(prefix);
189c21cf40f6ae69091bf24f87b5eeabc95e73dd271Jason Sams    s.append(" type ");
190064aa7ed76db9564b041afcd4b75da5b3d12fabaAlex Sakhartchouk    if (mHal.state.type) {
191bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams        mHal.state.type->dumpLOGV(s.string());
192c21cf40f6ae69091bf24f87b5eeabc95e73dd271Jason Sams    }
193c21cf40f6ae69091bf24f87b5eeabc95e73dd271Jason Sams
1946598201f1c4f409defac9a5af789fb53a7cc00f8Steve Block    ALOGV("%s allocation ptr=%p  mUsageFlags=0x04%x, mMipmapControl=0x%04x",
195eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams         prefix, getPtr(), mHal.state.usageFlags, mHal.state.mipmapControl);
196c21cf40f6ae69091bf24f87b5eeabc95e73dd271Jason Sams}
197326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
1982d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchoukuint32_t Allocation::getPackedSize() const {
1992d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t numItems = mHal.state.type->getSizeBytes() / mHal.state.type->getElementSizeBytes();
2002d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    return numItems * mHal.state.type->getElement()->getSizeBytesUnpadded();
2012d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk}
2022d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2032d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchoukvoid Allocation::writePackedData(const Type *type,
2042d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk                                 uint8_t *dst, const uint8_t *src, bool dstPadded) {
2052d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    const Element *elem = type->getElement();
2062d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t unpaddedBytes = elem->getSizeBytesUnpadded();
2072d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t paddedBytes = elem->getSizeBytes();
2082d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t numItems = type->getSizeBytes() / paddedBytes;
2092d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2102d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t srcInc = !dstPadded ? paddedBytes : unpaddedBytes;
2112d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t dstInc =  dstPadded ? paddedBytes : unpaddedBytes;
2122d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2132d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    // no sub-elements
2142d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t fieldCount = elem->getFieldCount();
2152d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    if (fieldCount == 0) {
2162d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        for (uint32_t i = 0; i < numItems; i ++) {
2172d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk            memcpy(dst, src, unpaddedBytes);
2182d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk            src += srcInc;
2192d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk            dst += dstInc;
2202d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        }
2212d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        return;
2222d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    }
2232d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2242d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    // Cache offsets
2252d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t *offsetsPadded = new uint32_t[fieldCount];
2262d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t *offsetsUnpadded = new uint32_t[fieldCount];
2272d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t *sizeUnpadded = new uint32_t[fieldCount];
2282d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2292d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    for (uint32_t i = 0; i < fieldCount; i++) {
2302d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        offsetsPadded[i] = elem->getFieldOffsetBytes(i);
2312d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        offsetsUnpadded[i] = elem->getFieldOffsetBytesUnpadded(i);
2322d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        sizeUnpadded[i] = elem->getField(i)->getSizeBytesUnpadded();
2332d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    }
2342d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2352d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t *srcOffsets = !dstPadded ? offsetsPadded : offsetsUnpadded;
2362d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t *dstOffsets =  dstPadded ? offsetsPadded : offsetsUnpadded;
2372d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2382d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    // complex elements, need to copy subelem after subelem
2392d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    for (uint32_t i = 0; i < numItems; i ++) {
2402d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        for (uint32_t fI = 0; fI < fieldCount; fI++) {
2412d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk            memcpy(dst + dstOffsets[fI], src + srcOffsets[fI], sizeUnpadded[fI]);
2422d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        }
2432d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        src += srcInc;
2442d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        dst += dstInc;
2452d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    }
2462d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2472d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    delete[] offsetsPadded;
2482d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    delete[] offsetsUnpadded;
2492d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    delete[] sizeUnpadded;
2502d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk}
2512d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2522d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchoukvoid Allocation::unpackVec3Allocation(const void *data, uint32_t dataSize) {
2532d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    const uint8_t *src = (const uint8_t*)data;
2542d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint8_t *dst = (uint8_t*)getPtr();
2552d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2562d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    writePackedData(getType(), dst, src, true);
2572d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk}
2582d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2592d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchoukvoid Allocation::packVec3Allocation(OStream *stream) const {
2602d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t paddedBytes = getType()->getElement()->getSizeBytes();
2612d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t unpaddedBytes = getType()->getElement()->getSizeBytesUnpadded();
2622d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t numItems = mHal.state.type->getSizeBytes() / paddedBytes;
2632d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2642d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    const uint8_t *src = (const uint8_t*)getPtr();
2652d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint8_t *dst = new uint8_t[numItems * unpaddedBytes];
2662d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2672d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    writePackedData(getType(), dst, src, false);
2682d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    stream->addByteArray(dst, getPackedSize());
2692d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
2702d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    delete[] dst;
2712d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk}
2722d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
273afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::serialize(OStream *stream) const {
274fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    // Need to identify ourselves
275fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    stream->addU32((uint32_t)getClassId());
276fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
277fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    String8 name(getName());
278fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    stream->addString(&name);
279fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
280fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    // First thing we need to serialize is the type object since it will be needed
281fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    // to initialize the class
282bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    mHal.state.type->serialize(stream);
283fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
284bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    uint32_t dataSize = mHal.state.type->getSizeBytes();
2852d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    // 3 element vectors are padded to 4 in memory, but padding isn't serialized
2862d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t packedSize = getPackedSize();
287fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    // Write how much data we are storing
2882d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    stream->addU32(packedSize);
2892d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    if (dataSize == packedSize) {
2902d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        // Now write the data
2912d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        stream->addByteArray(getPtr(), dataSize);
2922d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    } else {
2932d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        // Now write the data
2942d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        packVec3Allocation(stream);
2952d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    }
296fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk}
297fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
298afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex SakhartchoukAllocation *Allocation::createFromStream(Context *rsc, IStream *stream) {
299fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    // First make sure we are reading the correct object
300b825f67adb5d1e1751fe108e6dbf9c6f2555c283Alex Sakhartchouk    RsA3DClassID classID = (RsA3DClassID)stream->loadU32();
301afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk    if (classID != RS_A3D_CLASS_ID_ALLOCATION) {
302fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk        LOGE("allocation loading skipped due to invalid class id\n");
303fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk        return NULL;
304fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    }
305fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
306fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    String8 name;
307fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    stream->loadString(&name);
308fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
309fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    Type *type = Type::createFromStream(rsc, stream);
310afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk    if (!type) {
311fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk        return NULL;
312fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    }
313fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    type->compute();
314fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
3152d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    Allocation *alloc = Allocation::createAllocation(rsc, type, RS_ALLOCATION_USAGE_SCRIPT);
3162d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    type->decUserRef();
3172d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk
318fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    // Number of bytes we wrote out for this allocation
319fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    uint32_t dataSize = stream->loadU32();
3202d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    // 3 element vectors are padded to 4 in memory, but padding isn't serialized
3212d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    uint32_t packedSize = alloc->getPackedSize();
3222d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    if (dataSize != type->getSizeBytes() &&
3232d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        dataSize != packedSize) {
324fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk        LOGE("failed to read allocation because numbytes written is not the same loaded type wants\n");
3252d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        ObjectBase::checkDelete(alloc);
326225afd317e101a7be5fe02c0a86361146ea89f05Jason Sams        ObjectBase::checkDelete(type);
327fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk        return NULL;
328fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    }
329fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
330fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    alloc->setName(name.string(), name.size());
3314b45b8998e0d7038efaea80c70d23c086640b4e3Jason Sams
3322d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    if (dataSize == type->getSizeBytes()) {
3332d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        uint32_t count = dataSize / type->getElementSizeBytes();
3342d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        // Read in all of our allocation data
3352d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        alloc->data(rsc, 0, 0, count, stream->getPtr() + stream->getPos(), dataSize);
3362d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    } else {
3372d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk        alloc->unpackVec3Allocation(stream->getPtr() + stream->getPos(), dataSize);
3382d1220c27ae91f0b307f283fe66cb767b63dfe38Alex Sakhartchouk    }
339e6d9fbc31bef01219cc812e819c505ff01673c6fAlex Sakhartchouk    stream->reset(stream->getPos() + dataSize);
340fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
341fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk    return alloc;
342fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk}
343fb6b614bcea88a587a7ea4530be45ff0ffa0210eAlex Sakhartchouk
344eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Samsvoid Allocation::sendDirty(const Context *rsc) const {
3455c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams    for (size_t ct=0; ct < mToDirtyList.size(); ct++) {
3465c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams        mToDirtyList[ct]->forceDirty();
3475c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams    }
348eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    mRSC->mHal.funcs.allocation.markDirty(rsc, this);
3495c3e3bc8af6de6be5e6bd68e1d5168496f99e6cfJason Sams}
350326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
351afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::incRefs(const void *ptr, size_t ct, size_t startOff) const {
352e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams    const uint8_t *p = static_cast<const uint8_t *>(ptr);
353bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    const Element *e = mHal.state.type->getElement();
354e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams    uint32_t stride = e->getSizeBytes();
355e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams
35696abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    p += stride * startOff;
357e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams    while (ct > 0) {
358e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams        e->incRefs(p);
359e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams        ct --;
360e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams        p += stride;
361e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams    }
362e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams}
363e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams
364afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::decRefs(const void *ptr, size_t ct, size_t startOff) const {
3655c4369a3a1b19eaeabb044af2cdeef05474f9069Alex Sakhartchouk    if (!mHal.state.hasReferences || !getIsScript()) {
3665c4369a3a1b19eaeabb044af2cdeef05474f9069Alex Sakhartchouk        return;
3675c4369a3a1b19eaeabb044af2cdeef05474f9069Alex Sakhartchouk    }
368e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams    const uint8_t *p = static_cast<const uint8_t *>(ptr);
369bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    const Element *e = mHal.state.type->getElement();
370e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams    uint32_t stride = e->getSizeBytes();
371e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams
37296abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    p += stride * startOff;
373e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams    while (ct > 0) {
374e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams        e->decRefs(p);
375e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams        ct --;
376e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams        p += stride;
377e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams    }
378e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams}
379e3929c9bc6f3897e132304faf1b40c3cf1f47474Jason Sams
380c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Samsvoid Allocation::freeChildrenUnlocked () {
381c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams    decRefs(getPtr(), mHal.state.type->getSizeBytes() / mHal.state.type->getElementSizeBytes(), 0);
382c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams}
383c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams
384c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Samsbool Allocation::freeChildren() {
385c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams    if (mHal.state.hasReferences) {
386c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams        incSysRef();
387c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams        freeChildrenUnlocked();
388c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams        return decSysRef();
389c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams    }
390c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams    return false;
391c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams}
392c7cec1e3577cc77a5a73d5bd5a82733b1b9936a1Jason Sams
393afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::copyRange1D(Context *rsc, const Allocation *src, int32_t srcOff, int32_t destOff, int32_t len) {
39496abf819e50b59ba8cf886c13f894633eb0a24baJason Sams}
39596abf819e50b59ba8cf886c13f894633eb0a24baJason Sams
396afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::resize1D(Context *rsc, uint32_t dimX) {
397bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    uint32_t oldDimX = mHal.state.dimensionX;
39896abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    if (dimX == oldDimX) {
39996abf819e50b59ba8cf886c13f894633eb0a24baJason Sams        return;
40096abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    }
40196abf819e50b59ba8cf886c13f894633eb0a24baJason Sams
402c700e649ca44d0dcff8b271e42d949ea72fe3c63Alex Sakhartchouk    ObjectBaseRef<Type> t = mHal.state.type->cloneAndResize1D(rsc, dimX);
40396abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    if (dimX < oldDimX) {
404eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams        decRefs(getPtr(), oldDimX - dimX, dimX);
40596abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    }
406c700e649ca44d0dcff8b271e42d949ea72fe3c63Alex Sakhartchouk    rsc->mHal.funcs.allocation.resize(rsc, this, t.get(), mHal.state.hasReferences);
407064aa7ed76db9564b041afcd4b75da5b3d12fabaAlex Sakhartchouk    setType(t.get());
408bad807405b2b9764372af1ad24bcfd4fb1f33d8eJason Sams    updateCache();
40996abf819e50b59ba8cf886c13f894633eb0a24baJason Sams}
41096abf819e50b59ba8cf886c13f894633eb0a24baJason Sams
411afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid Allocation::resize2D(Context *rsc, uint32_t dimX, uint32_t dimY) {
41296abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    LOGE("not implemented");
41396abf819e50b59ba8cf886c13f894633eb0a24baJason Sams}
41496abf819e50b59ba8cf886c13f894633eb0a24baJason Sams
415326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams/////////////////
416565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams//
4176a121811e5d2e56e94747b36d15c7613ab2aedd4Stephen Hines
418326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace android {
419326e0ddf89e8df2837752fbfd7a014814b32082cJason Samsnamespace renderscript {
420326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
421c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Samsstatic void AllocationGenerateScriptMips(RsContext con, RsAllocation va);
422c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams
423afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukstatic void mip565(const Adapter2D &out, const Adapter2D &in) {
424326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams    uint32_t w = out.getDimX();
425326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams    uint32_t h = out.getDimY();
426326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
427e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams    for (uint32_t y=0; y < h; y++) {
428326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams        uint16_t *oPtr = static_cast<uint16_t *>(out.getElement(0, y));
429326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams        const uint16_t *i1 = static_cast<uint16_t *>(in.getElement(0, y*2));
430326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams        const uint16_t *i2 = static_cast<uint16_t *>(in.getElement(0, y*2+1));
431326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
432e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams        for (uint32_t x=0; x < w; x++) {
433565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams            *oPtr = rsBoxFilter565(i1[0], i1[1], i2[0], i2[1]);
434565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams            oPtr ++;
435565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams            i1 += 2;
436565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams            i2 += 2;
437565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams        }
438565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams    }
439565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams}
440565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams
441afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukstatic void mip8888(const Adapter2D &out, const Adapter2D &in) {
442565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams    uint32_t w = out.getDimX();
443565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams    uint32_t h = out.getDimY();
444565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams
445e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams    for (uint32_t y=0; y < h; y++) {
446565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams        uint32_t *oPtr = static_cast<uint32_t *>(out.getElement(0, y));
447565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams        const uint32_t *i1 = static_cast<uint32_t *>(in.getElement(0, y*2));
448565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams        const uint32_t *i2 = static_cast<uint32_t *>(in.getElement(0, y*2+1));
449565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams
450e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams        for (uint32_t x=0; x < w; x++) {
451565ac36ee479f9d7b83e2030ac9646a09cb886a1Jason Sams            *oPtr = rsBoxFilter8888(i1[0], i1[1], i2[0], i2[1]);
452326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams            oPtr ++;
453326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams            i1 += 2;
454326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams            i2 += 2;
455326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams        }
456326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams    }
457326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams}
458326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
459afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukstatic void mip8(const Adapter2D &out, const Adapter2D &in) {
4602f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams    uint32_t w = out.getDimX();
4612f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams    uint32_t h = out.getDimY();
4622f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams
4632f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams    for (uint32_t y=0; y < h; y++) {
4642f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams        uint8_t *oPtr = static_cast<uint8_t *>(out.getElement(0, y));
4652f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams        const uint8_t *i1 = static_cast<uint8_t *>(in.getElement(0, y*2));
4662f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams        const uint8_t *i2 = static_cast<uint8_t *>(in.getElement(0, y*2+1));
4672f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams
4682f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams        for (uint32_t x=0; x < w; x++) {
4692f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams            *oPtr = (uint8_t)(((uint32_t)i1[0] + i1[1] + i2[0] + i2[1]) * 0.25f);
4702f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams            oPtr ++;
4712f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams            i1 += 2;
4722f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams            i2 += 2;
4732f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams        }
4742f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams    }
4752f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams}
4762f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams
477afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukstatic void mip(const Adapter2D &out, const Adapter2D &in) {
478afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchouk    switch (out.getBaseType()->getElement()->getSizeBits()) {
479e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams    case 32:
480e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams        mip8888(out, in);
481e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams        break;
482e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams    case 16:
483e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams        mip565(out, in);
484e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams        break;
4852f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams    case 8:
4862f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams        mip8(out, in);
4872f6d8617769b5fd6287404a31d4e10140ebdcf7eJason Sams        break;
488e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams    }
489e9f5c53929c6c46872c4e9ba7cc3d0e528f5ad01Jason Sams}
490326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
491366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Samsvoid rsi_AllocationSyncAll(Context *rsc, RsAllocation va, RsAllocationUsageType src) {
492366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams    Allocation *a = static_cast<Allocation *>(va);
493eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    a->sendDirty(rsc);
494366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams    a->syncAll(rsc, src);
495366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams}
496366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams
497a23715148f7bda74e904fc553b70c9a49fd52a6eJason Samsvoid rsi_AllocationGenerateMipmaps(Context *rsc, RsAllocation va) {
49839f2ef6fed00a99c5c389e12c4597884027d4858Alex Sakhartchouk    Allocation *texAlloc = static_cast<Allocation *>(va);
499c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams    AllocationGenerateScriptMips(rsc, texAlloc);
50039f2ef6fed00a99c5c389e12c4597884027d4858Alex Sakhartchouk}
50139f2ef6fed00a99c5c389e12c4597884027d4858Alex Sakhartchouk
502837e388700a48084489ba59d1d8cc5ece68b1535Jason Samsvoid rsi_AllocationCopyToBitmap(Context *rsc, RsAllocation va, void *data, size_t dataLen) {
503837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams    Allocation *texAlloc = static_cast<Allocation *>(va);
504837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams    const Type * t = texAlloc->getType();
505837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams
506837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams    size_t s = t->getDimX() * t->getDimY() * t->getElementSizeBytes();
507837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams    if (s != dataLen) {
508837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams        rsc->setError(RS_ERROR_BAD_VALUE, "Bitmap size didn't match allocation size");
509837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams        return;
510837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams    }
511837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams
512837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams    memcpy(data, texAlloc->getPtr(), s);
513837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams}
514837e388700a48084489ba59d1d8cc5ece68b1535Jason Sams
5154b45b8998e0d7038efaea80c70d23c086640b4e3Jason Samsvoid rsi_Allocation1DData(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t lod,
516b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchouk                          uint32_t count, const void *data, size_t sizeBytes) {
517326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams    Allocation *a = static_cast<Allocation *>(va);
5184b45b8998e0d7038efaea80c70d23c086640b4e3Jason Sams    a->data(rsc, xoff, lod, count, data, sizeBytes);
5195f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams}
5205f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
5214b45b8998e0d7038efaea80c70d23c086640b4e3Jason Samsvoid rsi_Allocation2DElementData(Context *rsc, RsAllocation va, uint32_t x, uint32_t y, uint32_t lod, RsAllocationCubemapFace face,
522b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchouk                                 const void *data, size_t eoff, uint32_t sizeBytes) { // TODO: this seems wrong, eoff and sizeBytes may be swapped
5235f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    Allocation *a = static_cast<Allocation *>(va);
5244b45b8998e0d7038efaea80c70d23c086640b4e3Jason Sams    a->elementData(rsc, x, y, data, eoff, sizeBytes);
5255f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams}
5265f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams
5274b45b8998e0d7038efaea80c70d23c086640b4e3Jason Samsvoid rsi_Allocation1DElementData(Context *rsc, RsAllocation va, uint32_t x, uint32_t lod,
528b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchouk                                 const void *data, size_t eoff, uint32_t sizeBytes) { // TODO: this seems wrong, eoff and sizeBytes may be swapped
5295f0c84cf464dda719cef65fdc9b4d0980e86b98fJason Sams    Allocation *a = static_cast<Allocation *>(va);
5304b45b8998e0d7038efaea80c70d23c086640b4e3Jason Sams    a->elementData(rsc, x, data, eoff, sizeBytes);
531326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams}
532326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
5334b45b8998e0d7038efaea80c70d23c086640b4e3Jason Samsvoid rsi_Allocation2DData(Context *rsc, RsAllocation va, uint32_t xoff, uint32_t yoff, uint32_t lod, RsAllocationCubemapFace face,
534b81a0eb8180791e4eaab1253b59fa8bd562b046bAlex Sakhartchouk                          uint32_t w, uint32_t h, const void *data, size_t sizeBytes) {
535326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams    Allocation *a = static_cast<Allocation *>(va);
5364b45b8998e0d7038efaea80c70d23c086640b4e3Jason Sams    a->data(rsc, xoff, yoff, lod, face, w, h, data, sizeBytes);
537326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams}
538326e0ddf89e8df2837752fbfd7a014814b32082cJason Sams
53970b83c111beceaf8fbb700580833e7fec99272cfAlex Sakhartchoukvoid rsi_AllocationRead(Context *rsc, RsAllocation va, void *data, size_t data_length) {
540e579df42e85d9e00f53c42ef1b78dbd209dba989Jason Sams    Allocation *a = static_cast<Allocation *>(va);
541e579df42e85d9e00f53c42ef1b78dbd209dba989Jason Sams    a->read(data);
542e579df42e85d9e00f53c42ef1b78dbd209dba989Jason Sams}
543e579df42e85d9e00f53c42ef1b78dbd209dba989Jason Sams
544afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_AllocationResize1D(Context *rsc, RsAllocation va, uint32_t dimX) {
54596abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    Allocation *a = static_cast<Allocation *>(va);
54696abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    a->resize1D(rsc, dimX);
54796abf819e50b59ba8cf886c13f894633eb0a24baJason Sams}
54896abf819e50b59ba8cf886c13f894633eb0a24baJason Sams
549afb743aca56c18beb7ab924e75cb6e070ef3e55aAlex Sakhartchoukvoid rsi_AllocationResize2D(Context *rsc, RsAllocation va, uint32_t dimX, uint32_t dimY) {
55096abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    Allocation *a = static_cast<Allocation *>(va);
55196abf819e50b59ba8cf886c13f894633eb0a24baJason Sams    a->resize2D(rsc, dimX, dimY);
55296abf819e50b59ba8cf886c13f894633eb0a24baJason Sams}
55396abf819e50b59ba8cf886c13f894633eb0a24baJason Sams
554c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Samsstatic void AllocationGenerateScriptMips(RsContext con, RsAllocation va) {
555f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk    Context *rsc = static_cast<Context *>(con);
556f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk    Allocation *texAlloc = static_cast<Allocation *>(va);
557f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk    uint32_t numFaces = texAlloc->getType()->getDimFaces() ? 6 : 1;
558f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk    for (uint32_t face = 0; face < numFaces; face ++) {
559f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk        Adapter2D adapt(rsc, texAlloc);
560f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk        Adapter2D adapt2(rsc, texAlloc);
561f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk        adapt.setFace(face);
562f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk        adapt2.setFace(face);
563f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk        for (uint32_t lod=0; lod < (texAlloc->getType()->getLODCount() -1); lod++) {
564f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk            adapt.setLOD(lod);
565f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk            adapt2.setLOD(lod + 1);
566f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk            mip(adapt2, adapt);
567f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk        }
568f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk    }
569f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk}
570f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk
571c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason SamsRsAllocation rsi_AllocationCreateTyped(Context *rsc, RsType vtype,
572c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams                                       RsAllocationMipmapControl mips,
573c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams                                       uint32_t usages) {
574eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    Allocation * alloc = Allocation::createAllocation(rsc, static_cast<Type *>(vtype), usages, mips);
575eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    if (!alloc) {
576eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams        return NULL;
577eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    }
578f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams    alloc->incUserRef();
579f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams    return alloc;
580f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams}
581f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams
582c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason SamsRsAllocation rsi_AllocationCreateFromBitmap(Context *rsc, RsType vtype,
583c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams                                            RsAllocationMipmapControl mips,
584c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams                                            const void *data, size_t data_length, uint32_t usages) {
585366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams    Type *t = static_cast<Type *>(vtype);
586f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams
587c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams    RsAllocation vTexAlloc = rsi_AllocationCreateTyped(rsc, vtype, mips, usages);
588f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams    Allocation *texAlloc = static_cast<Allocation *>(vTexAlloc);
589f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams    if (texAlloc == NULL) {
590f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams        LOGE("Memory allocation failure");
591f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams        return NULL;
592f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams    }
593f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams
594366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams    memcpy(texAlloc->getPtr(), data, t->getDimX() * t->getDimY() * t->getElementSizeBytes());
595ebc5019400a129b1f1e57bd1fe8200a21f8da00bJason Sams    if (mips == RS_ALLOCATION_MIPMAP_FULL) {
596c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams        AllocationGenerateScriptMips(rsc, texAlloc);
597f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams    }
598f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams
599eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    texAlloc->sendDirty(rsc);
600f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams    return texAlloc;
601f0c1df480304a72ce41e7d4b088319cbd7f0938aJason Sams}
60284e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk
603c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason SamsRsAllocation rsi_AllocationCubeCreateFromBitmap(Context *rsc, RsType vtype,
604c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams                                                RsAllocationMipmapControl mips,
605c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams                                                const void *data, size_t data_length, uint32_t usages) {
606366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams    Type *t = static_cast<Type *>(vtype);
60784e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk
60884e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    // Cubemap allocation's faces should be Width by Width each.
60984e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    // Source data should have 6 * Width by Width pixels
61084e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    // Error checking is done in the java layer
611c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams    RsAllocation vTexAlloc = rsi_AllocationCreateTyped(rsc, vtype, mips, usages);
61284e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    Allocation *texAlloc = static_cast<Allocation *>(vTexAlloc);
61384e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    if (texAlloc == NULL) {
61484e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk        LOGE("Memory allocation failure");
61584e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk        return NULL;
61684e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    }
61784e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk
6189f8bc4fb7e9e48088dc6b0496afb35b34fc4c5afAlex Sakhartchouk    uint32_t faceSize = t->getDimX();
6199f8bc4fb7e9e48088dc6b0496afb35b34fc4c5afAlex Sakhartchouk    uint32_t strideBytes = faceSize * 6 * t->getElementSizeBytes();
6209f8bc4fb7e9e48088dc6b0496afb35b34fc4c5afAlex Sakhartchouk    uint32_t copySize = faceSize * t->getElementSizeBytes();
6219f8bc4fb7e9e48088dc6b0496afb35b34fc4c5afAlex Sakhartchouk
62284e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    uint8_t *sourcePtr = (uint8_t*)data;
623366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams    for (uint32_t face = 0; face < 6; face ++) {
624366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams        Adapter2D faceAdapter(rsc, texAlloc);
625366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams        faceAdapter.setFace(face);
626366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams
6279f8bc4fb7e9e48088dc6b0496afb35b34fc4c5afAlex Sakhartchouk        for (uint32_t dI = 0; dI < faceSize; dI ++) {
6289f8bc4fb7e9e48088dc6b0496afb35b34fc4c5afAlex Sakhartchouk            memcpy(faceAdapter.getElement(0, dI), sourcePtr + strideBytes * dI, copySize);
6299f8bc4fb7e9e48088dc6b0496afb35b34fc4c5afAlex Sakhartchouk        }
630366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams
631366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams        // Move the data pointer to the next cube face
6329f8bc4fb7e9e48088dc6b0496afb35b34fc4c5afAlex Sakhartchouk        sourcePtr += copySize;
633f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk    }
634366c9c85196675437a8dd74c1cf6b63ddbde3d6aJason Sams
635f8aafcfad92fcf37d4b55c749601de22441ac9bfAlex Sakhartchouk    if (mips == RS_ALLOCATION_MIPMAP_FULL) {
636c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams        AllocationGenerateScriptMips(rsc, texAlloc);
63784e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    }
63884e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk
639eb4fe18dd88634330f9566cbb9e785d8c7ec5813Jason Sams    texAlloc->sendDirty(rsc);
64084e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk    return texAlloc;
64184e4027f83b20af59f5b1fc52be6e45f159d3970Alex Sakhartchouk}
642099d7d33e55afeb3399f6e8cf8d665223ca94939Alex Sakhartchouk
64374a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchoukvoid rsi_AllocationCopy2DRange(Context *rsc,
64474a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                               RsAllocation dstAlloc,
64574a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                               uint32_t dstXoff, uint32_t dstYoff,
64674a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                               uint32_t dstMip, uint32_t dstFace,
64774a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                               uint32_t width, uint32_t height,
64874a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                               RsAllocation srcAlloc,
64974a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                               uint32_t srcXoff, uint32_t srcYoff,
65074a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                               uint32_t srcMip, uint32_t srcFace) {
65174a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk    Allocation *dst = static_cast<Allocation *>(dstAlloc);
65274a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk    Allocation *src= static_cast<Allocation *>(srcAlloc);
65374a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk    rsc->mHal.funcs.allocation.allocData2D(rsc, dst, dstXoff, dstYoff, dstMip,
65474a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                                           (RsAllocationCubemapFace)dstFace,
65574a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                                           width, height,
65674a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                                           src, srcXoff, srcYoff,srcMip,
65774a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk                                           (RsAllocationCubemapFace)srcFace);
65874a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk}
65974a827988567a9d65954bb0d825a3ba4a97e2947Alex Sakhartchouk
660c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams}
661c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams}
662c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams
663c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Samsconst void * rsaAllocationGetType(RsContext con, RsAllocation va) {
664c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams    Allocation *a = static_cast<Allocation *>(va);
665c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams    a->getType()->incUserRef();
666c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams
667c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams    return a->getType();
668c975cf4a71b63ccbd20f1f3b3341c5f2e6025b45Jason Sams}
669