1ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams/* 2ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * Copyright (C) 2012 The Android Open Source Project 3ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * 4ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * Licensed under the Apache License, Version 2.0 (the "License"); 5ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * you may not use this file except in compliance with the License. 6ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * You may obtain a copy of the License at 7ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * 8ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * http://www.apache.org/licenses/LICENSE-2.0 9ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * 10ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * Unless required by applicable law or agreed to in writing, software 11ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * distributed under the License is distributed on an "AS IS" BASIS, 12ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * See the License for the specific language governing permissions and 14ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams * limitations under the License. 15ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams */ 16ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 17ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams#include "rsContext.h" 18ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams#include <time.h> 19ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 20ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsusing namespace android; 21ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsusing namespace android::renderscript; 22ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 23ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsScriptGroup::ScriptGroup(Context *rsc) : ObjectBase(rsc) { 24ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 25ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 26ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsScriptGroup::~ScriptGroup() { 27ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (mRSC->mHal.funcs.scriptgroup.destroy) { 28ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mRSC->mHal.funcs.scriptgroup.destroy(mRSC, this); 29ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 30ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 31ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mLinks.size(); ct++) { 32ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams delete mLinks[ct]; 33ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 34ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 35ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 36ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsScriptGroup::IO::IO(const ScriptKernelID *kid) { 37ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mKernel = kid; 38ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 39ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 40ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsScriptGroup::Node::Node(Script *s) { 41ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mScript = s; 42ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mSeen = false; 43ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mOrder = 0; 44ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 45ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 46ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsScriptGroup::Node * ScriptGroup::findNode(Script *s) const { 47ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("find %p %i", s, (int)mNodes.size()); 48ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mNodes.size(); ct++) { 49ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Node *n = mNodes[ct]; 50ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct2=0; ct2 < n->mKernels.size(); ct2++) { 51ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n->mKernels[ct2]->mScript == s) { 52ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return n; 53ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 54ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 55ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 56ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return NULL; 57ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 58ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 59ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsbool ScriptGroup::calcOrderRecurse(Node *n, int depth) { 60ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n->mSeen = true; 61ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n->mOrder < depth) { 62ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n->mOrder = depth; 63ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 64ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams bool ret = true; 65ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < n->mOutputs.size(); ct++) { 66ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const Link *l = n->mOutputs[ct]; 67ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Node *nt = NULL; 68ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (l->mDstField.get()) { 69ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams nt = findNode(l->mDstField->mScript); 70ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } else { 71ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams nt = findNode(l->mDstKernel->mScript); 72ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 73ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (nt->mSeen) { 74ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return false; 75ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 76ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ret &= calcOrderRecurse(nt, n->mOrder + 1); 77ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 78ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return ret; 79ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 80ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 81ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsstatic int CompareNodeForSort(ScriptGroup::Node *const* lhs, 82ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptGroup::Node *const* rhs) { 83ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (lhs[0]->mOrder > rhs[0]->mOrder) { 84ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return 1; 85ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 86ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return 0; 87ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 88ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 89ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 90ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsbool ScriptGroup::calcOrder() { 91ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams // Make nodes 92ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mKernels.size(); ct++) { 93ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const ScriptKernelID *k = mKernels[ct].get(); 94ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE(" kernel %i, %p s=%p", (int)ct, k, mKernels[ct]->mScript); 95ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Node *n = findNode(k->mScript); 96ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE(" n = %p", n); 97ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n == NULL) { 98ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n = new Node(k->mScript); 99ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mNodes.add(n); 100ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 101ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n->mKernels.add(k); 102ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 103ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 104ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams // add links 105ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("link count %i", (int)mLinks.size()); 106ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mLinks.size(); ct++) { 107ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Link *l = mLinks[ct]; 108ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("link %i %p", (int)ct, l); 109ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Node *n = findNode(l->mSource->mScript); 110ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("link n %p", n); 111ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n->mOutputs.add(l); 112ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 113ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (l->mDstKernel.get()) { 114ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("l->mDstKernel.get() %p", l->mDstKernel.get()); 115ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n = findNode(l->mDstKernel->mScript); 116ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE(" n1 %p", n); 117ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n->mInputs.add(l); 118ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } else { 119ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n = findNode(l->mDstField->mScript); 120ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE(" n2 %p", n); 121ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n->mInputs.add(l); 122ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 123ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 124ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 125ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("node count %i", (int)mNodes.size()); 126ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams // Order nodes 127ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams bool ret = true; 128ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mNodes.size(); ct++) { 129ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Node *n = mNodes[ct]; 130ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n->mInputs.size() == 0) { 131ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct2=0; ct2 < mNodes.size(); ct2++) { 132ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mNodes[ct2]->mSeen = false; 133ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 134ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ret &= calcOrderRecurse(n, 0); 135ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 136ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 137ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 138ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mKernels.size(); ct++) { 139ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const ScriptKernelID *k = mKernels[ct].get(); 140ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const Node *n = findNode(k->mScript); 141ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 142ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (k->mHasKernelOutput) { 143ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams bool found = false; 144ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct2=0; ct2 < n->mOutputs.size(); ct2++) { 145ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n->mOutputs[ct2]->mSource.get() == k) { 146ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams found = true; 147ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams break; 148ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 149ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 150ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (!found) { 151ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("add io out %p", k); 152ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mOutputs.add(new IO(k)); 153ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 154ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 155ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 156ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (k->mHasKernelInput) { 157ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams bool found = false; 158ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct2=0; ct2 < n->mInputs.size(); ct2++) { 159ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n->mInputs[ct2]->mDstKernel.get() == k) { 160ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams found = true; 161ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams break; 162ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 163ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 164ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (!found) { 165ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("add io in %p", k); 166ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mInputs.add(new IO(k)); 167ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 168ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 169ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 170ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 171ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams // sort 172ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mNodes.sort(&CompareNodeForSort); 173ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 174ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return ret; 175ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 176ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 177ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsScriptGroup * ScriptGroup::create(Context *rsc, 178ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptKernelID ** kernels, size_t kernelsSize, 179ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptKernelID ** src, size_t srcSize, 180ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptKernelID ** dstK, size_t dstKSize, 181ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptFieldID ** dstF, size_t dstFSize, 182ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const Type ** type, size_t typeSize) { 183ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 184ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams size_t kernelCount = kernelsSize / sizeof(ScriptKernelID *); 185ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams size_t linkCount = typeSize / sizeof(Type *); 186ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 187ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("ScriptGroup::create kernels=%i links=%i", (int)kernelCount, (int)linkCount); 188ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 189ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 190ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams // Start by counting unique kernel sources 191ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 192ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptGroup *sg = new ScriptGroup(rsc); 193ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 194ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams sg->mKernels.reserve(kernelCount); 195ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < kernelCount; ct++) { 196ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams sg->mKernels.add(kernels[ct]); 197ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 198ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 199ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams sg->mLinks.reserve(linkCount); 200ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < linkCount; ct++) { 201ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Link *l = new Link(); 202ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams l->mType = type[ct]; 203ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams l->mSource = src[ct]; 204ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams l->mDstField = dstF[ct]; 205ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams l->mDstKernel = dstK[ct]; 206ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams sg->mLinks.add(l); 207ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 208ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 209ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams sg->calcOrder(); 210ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 211ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams // allocate links 212ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < sg->mNodes.size(); ct++) { 213ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const Node *n = sg->mNodes[ct]; 214ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct2=0; ct2 < n->mOutputs.size(); ct2++) { 215ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Link *l = n->mOutputs[ct2]; 216ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (l->mAlloc.get()) { 217ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams continue; 218ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 219ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const ScriptKernelID *k = l->mSource.get(); 220ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 221ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Allocation * alloc = Allocation::createAllocation(rsc, 222ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams l->mType.get(), RS_ALLOCATION_USAGE_SCRIPT); 223ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams l->mAlloc = alloc; 224ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 225ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct3=ct2+1; ct3 < n->mOutputs.size(); ct3++) { 226ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n->mOutputs[ct3]->mSource.get() == l->mSource.get()) { 227ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n->mOutputs[ct3]->mAlloc = alloc; 228ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 229ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 230ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 231ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 232ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 233ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (rsc->mHal.funcs.scriptgroup.init) { 234ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams rsc->mHal.funcs.scriptgroup.init(rsc, sg); 235ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 236ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return sg; 237ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 238ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 239ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsvoid ScriptGroup::setInput(Context *rsc, ScriptKernelID *kid, Allocation *a) { 240ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mInputs.size(); ct++) { 241ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (mInputs[ct]->mKernel == kid) { 242ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mInputs[ct]->mAlloc = a; 243ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 244ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (rsc->mHal.funcs.scriptgroup.setInput) { 245ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams rsc->mHal.funcs.scriptgroup.setInput(rsc, this, kid, a); 246ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 247ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return; 248ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 249ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 250ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams rsAssert(!"ScriptGroup:setInput kid not found"); 251ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 252ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 253ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsvoid ScriptGroup::setOutput(Context *rsc, ScriptKernelID *kid, Allocation *a) { 254ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mOutputs.size(); ct++) { 255ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (mOutputs[ct]->mKernel == kid) { 256ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams mOutputs[ct]->mAlloc = a; 257ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 258ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (rsc->mHal.funcs.scriptgroup.setOutput) { 259ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams rsc->mHal.funcs.scriptgroup.setOutput(rsc, this, kid, a); 260ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 261ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return; 262ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 263ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 264ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams rsAssert(!"ScriptGroup:setOutput kid not found"); 265ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 266ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 267ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsvoid ScriptGroup::execute(Context *rsc) { 268ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("ScriptGroup::execute"); 269ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (rsc->mHal.funcs.scriptgroup.execute) { 270ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams rsc->mHal.funcs.scriptgroup.execute(rsc, this); 271ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return; 272ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 273ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 274ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct=0; ct < mNodes.size(); ct++) { 275ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Node *n = mNodes[ct]; 276ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("node %i, order %i, in %i out %i", (int)ct, n->mOrder, (int)n->mInputs.size(), (int)n->mOutputs.size()); 277ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 278ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct2=0; ct2 < n->mKernels.size(); ct2++) { 279ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const ScriptKernelID *k = n->mKernels[ct2]; 280ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Allocation *ain = NULL; 281ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams Allocation *aout = NULL; 282ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 283ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct3=0; ct3 < n->mInputs.size(); ct3++) { 284ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n->mInputs[ct3]->mDstKernel.get() == k) { 285ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ain = n->mInputs[ct3]->mAlloc.get(); 286ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE(" link in %p", ain); 287ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 288ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 289ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct3=0; ct3 < mInputs.size(); ct3++) { 290ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (mInputs[ct3]->mKernel == k) { 291ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ain = mInputs[ct3]->mAlloc.get(); 292ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE(" io in %p", ain); 293ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 294ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 295ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 296ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct3=0; ct3 < n->mOutputs.size(); ct3++) { 297ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (n->mOutputs[ct3]->mSource.get() == k) { 298ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams aout = n->mOutputs[ct3]->mAlloc.get(); 299ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE(" link out %p", aout); 300ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 301ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 302ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams for (size_t ct3=0; ct3 < mOutputs.size(); ct3++) { 303ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams if (mOutputs[ct3]->mKernel == k) { 304ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams aout = mOutputs[ct3]->mAlloc.get(); 305ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE(" io out %p", aout); 306ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 307ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 308ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 309ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams n->mScript->runForEach(rsc, k->mSlot, ain, aout, NULL, 0); 310ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 311ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 312ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams } 313ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 314ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 315ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 316ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsvoid ScriptGroup::serialize(Context *rsc, OStream *stream) const { 317ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 318ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 319ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsRsA3DClassID ScriptGroup::getClassId() const { 320ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return RS_A3D_CLASS_ID_SCRIPT_GROUP; 321ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 322ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 323ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsScriptGroup::Link::Link() { 324ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 325ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 326ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsScriptGroup::Link::~Link() { 327ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 328ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 329ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsnamespace android { 330ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsnamespace renderscript { 331ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 332ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 333ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason SamsRsScriptGroup rsi_ScriptGroupCreate(Context *rsc, 334ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams RsScriptKernelID * kernels, size_t kernelsSize, 335ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams RsScriptKernelID * src, size_t srcSize, 336ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams RsScriptKernelID * dstK, size_t dstKSize, 337ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams RsScriptFieldID * dstF, size_t dstFSize, 338ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams const RsType * type, size_t typeSize) { 339ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 340ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 341ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams return ScriptGroup::create(rsc, 342ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams (ScriptKernelID **) kernels, kernelsSize, 343ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams (ScriptKernelID **) src, srcSize, 344ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams (ScriptKernelID **) dstK, dstKSize, 345ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams (ScriptFieldID **) dstF, dstFSize, 346ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams (const Type **) type, typeSize); 347ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 348ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 349ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 350ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsvoid rsi_ScriptGroupSetInput(Context *rsc, RsScriptGroup sg, RsScriptKernelID kid, 351ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams RsAllocation alloc) { 352ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("rsi_ScriptGroupSetInput"); 353ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptGroup *s = (ScriptGroup *)sg; 354ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams s->setInput(rsc, (ScriptKernelID *)kid, (Allocation *)alloc); 355ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 356ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 357ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsvoid rsi_ScriptGroupSetOutput(Context *rsc, RsScriptGroup sg, RsScriptKernelID kid, 358ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams RsAllocation alloc) { 359ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("rsi_ScriptGroupSetOutput"); 360ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptGroup *s = (ScriptGroup *)sg; 361ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams s->setOutput(rsc, (ScriptKernelID *)kid, (Allocation *)alloc); 362ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 363ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 364ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Samsvoid rsi_ScriptGroupExecute(Context *rsc, RsScriptGroup sg) { 365ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams //ALOGE("rsi_ScriptGroupExecute"); 366ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams ScriptGroup *s = (ScriptGroup *)sg; 367ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams s->execute(rsc); 368ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 369ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 370ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 371ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams} 372ca29b8caf56fa4866752f9cea4ec02b2a271dceeJason Sams 373