123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403 |
- #include "rsScriptGroup.h"
- #include "rsContext.h"
- #include "rsScriptGroup2.h"
- #include <algorithm>
- #include <time.h>
- namespace android {
- namespace renderscript {
- ScriptGroup::ScriptGroup(Context *rsc) : ScriptGroupBase(rsc) {
- }
- ScriptGroup::~ScriptGroup() {
- if (mRSC->mHal.funcs.scriptgroup.destroy) {
- mRSC->mHal.funcs.scriptgroup.destroy(mRSC, this);
- }
- for (size_t ct=0; ct < mLinks.size(); ct++) {
- delete mLinks[ct];
- }
- for (auto input : mInputs) {
- input->mAlloc.clear();
- }
- for (auto output : mOutputs) {
- output->mAlloc.clear();
- }
- }
- ScriptGroup::IO::IO(const ScriptKernelID *kid) {
- mKernel = kid;
- }
- ScriptGroup::Node::Node(Script *s) {
- mScript = s;
- mSeen = false;
- mOrder = 0;
- }
- ScriptGroup::Node * ScriptGroup::findNode(Script *s) const {
-
- for (size_t ct=0; ct < mNodes.size(); ct++) {
- Node *n = mNodes[ct];
- for (size_t ct2=0; ct2 < n->mKernels.size(); ct2++) {
- if (n->mKernels[ct2]->mScript == s) {
- return n;
- }
- }
- }
- return nullptr;
- }
- bool ScriptGroup::calcOrderRecurse(Node *n, int depth) {
- n->mSeen = true;
- if (n->mOrder < depth) {
- n->mOrder = depth;
- }
- bool ret = true;
- for (size_t ct=0; ct < n->mOutputs.size(); ct++) {
- const Link *l = n->mOutputs[ct];
- Node *nt = NULL;
- if (l->mDstField.get()) {
- nt = findNode(l->mDstField->mScript);
- } else {
- nt = findNode(l->mDstKernel->mScript);
- }
- if (nt->mSeen) {
- return false;
- }
- ret &= calcOrderRecurse(nt, n->mOrder + 1);
- }
- return ret;
- }
- class NodeCompare {
- public:
- bool operator() (const ScriptGroup::Node* lhs,
- const ScriptGroup::Node* rhs) {
- return (lhs->mOrder < rhs->mOrder);
- }
- };
- bool ScriptGroup::calcOrder() {
-
- for (size_t ct=0; ct < mKernels.size(); ct++) {
- const ScriptKernelID *k = mKernels[ct].get();
-
- Node *n = findNode(k->mScript);
-
- if (n == NULL) {
- n = new Node(k->mScript);
- mNodes.push_back(n);
- }
- n->mKernels.push_back(k);
- }
-
-
- for (size_t ct=0; ct < mLinks.size(); ct++) {
- Link *l = mLinks[ct];
-
- Node *n = findNode(l->mSource->mScript);
-
- n->mOutputs.push_back(l);
- if (l->mDstKernel.get()) {
-
- n = findNode(l->mDstKernel->mScript);
-
- n->mInputs.push_back(l);
- } else {
- n = findNode(l->mDstField->mScript);
-
- n->mInputs.push_back(l);
- }
- }
-
-
- bool ret = true;
- for (size_t ct=0; ct < mNodes.size(); ct++) {
- Node *n = mNodes[ct];
- if (n->mInputs.size() == 0) {
- for (size_t ct2=0; ct2 < mNodes.size(); ct2++) {
- mNodes[ct2]->mSeen = false;
- }
- ret &= calcOrderRecurse(n, 0);
- }
- }
- for (size_t ct=0; ct < mKernels.size(); ct++) {
- const ScriptKernelID *k = mKernels[ct].get();
- const Node *n = findNode(k->mScript);
- if (k->mHasKernelOutput) {
- bool found = false;
- for (size_t ct2=0; ct2 < n->mOutputs.size(); ct2++) {
- if (n->mOutputs[ct2]->mSource.get() == k) {
- found = true;
- break;
- }
- }
- if (!found) {
-
- mOutputs.push_back(new IO(k));
- }
- }
- if (k->mHasKernelInput) {
- bool found = false;
- for (size_t ct2=0; ct2 < n->mInputs.size(); ct2++) {
- if (n->mInputs[ct2]->mDstKernel.get() == k) {
- found = true;
- break;
- }
- }
- if (!found) {
-
- mInputs.push_back(new IO(k));
- }
- }
- }
-
- std::sort(mNodes.begin(), mNodes.end(), NodeCompare());
- return ret;
- }
- ScriptGroup * ScriptGroup::create(Context *rsc,
- ScriptKernelID ** kernels, size_t kernelsSize,
- ScriptKernelID ** src, size_t srcSize,
- ScriptKernelID ** dstK, size_t dstKSize,
- ScriptFieldID ** dstF, size_t dstFSize,
- const Type ** type, size_t typeSize) {
- size_t kernelCount = kernelsSize / sizeof(ScriptKernelID *);
- size_t linkCount = typeSize / sizeof(Type *);
-
-
- ScriptGroup *sg = new ScriptGroup(rsc);
- sg->mKernels.reserve(kernelCount);
- for (size_t ct=0; ct < kernelCount; ct++) {
- sg->mKernels.push_back(kernels[ct]);
- }
- sg->mLinks.reserve(linkCount);
- for (size_t ct=0; ct < linkCount; ct++) {
- Link *l = new Link();
- l->mType = type[ct];
- l->mSource = src[ct];
- l->mDstField = dstF[ct];
- l->mDstKernel = dstK[ct];
- sg->mLinks.push_back(l);
- }
- sg->calcOrder();
-
- for (size_t ct=0; ct < sg->mNodes.size(); ct++) {
- const Node *n = sg->mNodes[ct];
- for (size_t ct2=0; ct2 < n->mOutputs.size(); ct2++) {
- Link *l = n->mOutputs[ct2];
- if (l->mAlloc.get()) {
- continue;
- }
- Allocation * alloc = Allocation::createAllocation(rsc,
- l->mType.get(), RS_ALLOCATION_USAGE_SCRIPT);
- l->mAlloc = alloc;
- for (size_t ct3=ct2+1; ct3 < n->mOutputs.size(); ct3++) {
- if (n->mOutputs[ct3]->mSource.get() == l->mSource.get()) {
- n->mOutputs[ct3]->mAlloc = alloc;
- }
- }
- }
- }
- if (rsc->mHal.funcs.scriptgroup.init) {
- rsc->mHal.funcs.scriptgroup.init(rsc, sg);
- }
- sg->incUserRef();
- return sg;
- }
- void ScriptGroup::setInput(Context *rsc, ScriptKernelID *kid, Allocation *a) {
- for (size_t ct=0; ct < mInputs.size(); ct++) {
- if (mInputs[ct]->mKernel == kid) {
- mInputs[ct]->mAlloc = a;
- if (rsc->mHal.funcs.scriptgroup.setInput) {
- rsc->mHal.funcs.scriptgroup.setInput(rsc, this, kid, a);
- }
- return;
- }
- }
- rsAssert(!"ScriptGroup:setInput kid not found");
- }
- void ScriptGroup::setOutput(Context *rsc, ScriptKernelID *kid, Allocation *a) {
- for (size_t ct=0; ct < mOutputs.size(); ct++) {
- if (mOutputs[ct]->mKernel == kid) {
- mOutputs[ct]->mAlloc = a;
- if (rsc->mHal.funcs.scriptgroup.setOutput) {
- rsc->mHal.funcs.scriptgroup.setOutput(rsc, this, kid, a);
- }
- return;
- }
- }
- rsAssert(!"ScriptGroup:setOutput kid not found");
- }
- bool ScriptGroup::validateInputAndOutput(Context *rsc) {
- for(size_t i = 0; i < mInputs.size(); i++) {
- if (mInputs[i]->mAlloc.get() == nullptr) {
- rsc->setError(RS_ERROR_BAD_VALUE, "ScriptGroup missing input.");
- return false;
- }
- }
- for(size_t i = 0; i < mOutputs.size(); i++) {
- if (mOutputs[i]->mAlloc.get() == nullptr) {
- rsc->setError(RS_ERROR_BAD_VALUE, "ScriptGroup missing output.");
- return false;
- }
- }
- return true;
- }
- void ScriptGroup::execute(Context *rsc) {
- if (!validateInputAndOutput(rsc)) {
- return;
- }
- if (rsc->mHal.funcs.scriptgroup.execute) {
- rsc->mHal.funcs.scriptgroup.execute(rsc, this);
- return;
- }
- for (size_t ct=0; ct < mNodes.size(); ct++) {
- Node *n = mNodes[ct];
-
- for (size_t ct2=0; ct2 < n->mKernels.size(); ct2++) {
- const ScriptKernelID *k = n->mKernels[ct2];
- Allocation *ain = NULL;
- Allocation *aout = NULL;
- for (size_t ct3=0; ct3 < n->mInputs.size(); ct3++) {
- if (n->mInputs[ct3]->mDstKernel.get() == k) {
- ain = n->mInputs[ct3]->mAlloc.get();
-
- }
- }
- for (size_t ct3=0; ct3 < mInputs.size(); ct3++) {
- if (mInputs[ct3]->mKernel == k) {
- ain = mInputs[ct3]->mAlloc.get();
-
- }
- }
- for (size_t ct3=0; ct3 < n->mOutputs.size(); ct3++) {
- if (n->mOutputs[ct3]->mSource.get() == k) {
- aout = n->mOutputs[ct3]->mAlloc.get();
-
- }
- }
- for (size_t ct3=0; ct3 < mOutputs.size(); ct3++) {
- if (mOutputs[ct3]->mKernel == k) {
- aout = mOutputs[ct3]->mAlloc.get();
-
- }
- }
- if (ain == NULL) {
- n->mScript->runForEach(rsc, k->mSlot, NULL, 0, aout, NULL, 0);
- } else {
- const Allocation *ains[1] = {ain};
- n->mScript->runForEach(rsc, k->mSlot, ains,
- sizeof(ains) / sizeof(RsAllocation),
- aout, NULL, 0);
- }
- }
- }
- }
- ScriptGroup::Link::Link() {
- }
- ScriptGroup::Link::~Link() {
- }
- RsScriptGroup rsi_ScriptGroupCreate(Context *rsc,
- RsScriptKernelID * kernels, size_t kernelsSize,
- RsScriptKernelID * src, size_t srcSize,
- RsScriptKernelID * dstK, size_t dstKSize,
- RsScriptFieldID * dstF, size_t dstFSize,
- const RsType * type, size_t typeSize) {
- return ScriptGroup::create(rsc,
- (ScriptKernelID **) kernels, kernelsSize,
- (ScriptKernelID **) src, srcSize,
- (ScriptKernelID **) dstK, dstKSize,
- (ScriptFieldID **) dstF, dstFSize,
- (const Type **) type, typeSize);
- }
- void rsi_ScriptGroupSetInput(Context *rsc, RsScriptGroup sg, RsScriptKernelID kid,
- RsAllocation alloc) {
-
- ScriptGroup *s = (ScriptGroup *)sg;
- s->setInput(rsc, (ScriptKernelID *)kid, (Allocation *)alloc);
- }
- void rsi_ScriptGroupSetOutput(Context *rsc, RsScriptGroup sg, RsScriptKernelID kid,
- RsAllocation alloc) {
-
- ScriptGroup *s = (ScriptGroup *)sg;
- s->setOutput(rsc, (ScriptKernelID *)kid, (Allocation *)alloc);
- }
- void rsi_ScriptGroupExecute(Context *rsc, RsScriptGroup sg) {
- ScriptGroupBase *s = (ScriptGroupBase *)sg;
- s->execute(rsc);
- }
- }
- }
|