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