You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
461 lines
14 KiB
461 lines
14 KiB
/*
|
|
* Copyright (C) 2009-2012 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include "rsContext.h"
|
|
#include "rsScriptC.h"
|
|
|
|
#if !defined(RS_COMPATIBILITY_LIB) && !defined(ANDROID_RS_SERIALIZE)
|
|
#include <bcinfo/BitcodeTranslator.h>
|
|
#include <bcinfo/BitcodeWrapper.h>
|
|
#endif
|
|
|
|
#include <sys/stat.h>
|
|
|
|
#include <sstream>
|
|
#include <string>
|
|
|
|
#ifdef _WIN32
|
|
/* Define the default path separator for the platform. */
|
|
#define OS_PATH_SEPARATOR '\\'
|
|
#define OS_PATH_SEPARATOR_STR "\\"
|
|
|
|
#else /* not _WIN32 */
|
|
|
|
/* Define the default path separator for the platform. */
|
|
#define OS_PATH_SEPARATOR '/'
|
|
#define OS_PATH_SEPARATOR_STR "/"
|
|
|
|
#endif
|
|
|
|
using android::renderscript::ScriptC;
|
|
|
|
#define GET_TLS() Context::ScriptTLSStruct * tls = \
|
|
(Context::ScriptTLSStruct *)pthread_getspecific(Context::gThreadTLSKey); \
|
|
Context * rsc = tls->mContext; \
|
|
ScriptC * sc = (ScriptC *) tls->mScript
|
|
|
|
ScriptC::ScriptC(Context *rsc) : Script(rsc) {
|
|
}
|
|
|
|
ScriptC::~ScriptC() {
|
|
if (mInitialized) {
|
|
mRSC->mHal.funcs.script.invokeFreeChildren(mRSC, this);
|
|
mRSC->mHal.funcs.script.destroy(mRSC, this);
|
|
}
|
|
}
|
|
|
|
#ifndef RS_COMPATIBILITY_LIB
|
|
bool ScriptC::createCacheDir(const char *cacheDir) {
|
|
std::string currentDir;
|
|
const std::string cacheDirString(cacheDir);
|
|
|
|
struct stat statBuf;
|
|
int statReturn = stat(cacheDir, &statBuf);
|
|
if (!statReturn) {
|
|
return true;
|
|
}
|
|
|
|
// Start from the beginning of the cacheDirString.
|
|
int currPos = 0;
|
|
|
|
// Reserve space in currentDir for the entire cacheDir path.
|
|
currentDir.reserve(cacheDirString.length());
|
|
|
|
while (currPos >= 0) {
|
|
/*
|
|
* The character at currPos should be a path separator. We need to look
|
|
* for the next one.
|
|
*/
|
|
int nextPos = cacheDirString.find(OS_PATH_SEPARATOR, currPos + 1);
|
|
|
|
if (nextPos > 0) {
|
|
// A new path separator has been found.
|
|
currentDir += cacheDirString.substr(currPos, nextPos - currPos);
|
|
} else {
|
|
// There are no more path separators.
|
|
currentDir += cacheDirString.substr(currPos);
|
|
}
|
|
|
|
currPos = nextPos;
|
|
|
|
statReturn = stat(currentDir.c_str(), &statBuf);
|
|
|
|
if (statReturn) {
|
|
if (errno == ENOENT) {
|
|
if (mkdir(currentDir.c_str(), S_IRUSR | S_IWUSR | S_IXUSR)) {
|
|
ALOGE("Couldn't create cache directory: %s",
|
|
currentDir.c_str());
|
|
ALOGE("Error: %s", strerror(errno));
|
|
return false;
|
|
}
|
|
} else {
|
|
ALOGE("Stat error: %s", strerror(errno));
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
#endif
|
|
|
|
void ScriptC::setupScript(Context *rsc) {
|
|
mEnviroment.mStartTimeMillis
|
|
= nanoseconds_to_milliseconds(systemTime(SYSTEM_TIME_MONOTONIC));
|
|
|
|
for (uint32_t ct=0; ct < mHal.info.exportedVariableCount; ct++) {
|
|
if (mSlots[ct].get() && !mTypes[ct].get()) {
|
|
mTypes[ct].set(mSlots[ct]->getType());
|
|
}
|
|
|
|
if (!mTypes[ct].get())
|
|
continue;
|
|
rsc->mHal.funcs.script.setGlobalBind(rsc, this, ct, mSlots[ct].get());
|
|
}
|
|
}
|
|
|
|
void ScriptC::setupGLState(Context *rsc) {
|
|
#if !defined(RS_VENDOR_LIB) && !defined(RS_COMPATIBILITY_LIB)
|
|
if (mEnviroment.mFragmentStore.get()) {
|
|
rsc->setProgramStore(mEnviroment.mFragmentStore.get());
|
|
}
|
|
if (mEnviroment.mFragment.get()) {
|
|
rsc->setProgramFragment(mEnviroment.mFragment.get());
|
|
}
|
|
if (mEnviroment.mVertex.get()) {
|
|
rsc->setProgramVertex(mEnviroment.mVertex.get());
|
|
}
|
|
if (mEnviroment.mRaster.get()) {
|
|
rsc->setProgramRaster(mEnviroment.mRaster.get());
|
|
}
|
|
#endif
|
|
}
|
|
|
|
uint32_t ScriptC::run(Context *rsc) {
|
|
if (mHal.info.root == nullptr) {
|
|
rsc->setError(RS_ERROR_BAD_SCRIPT, "Attempted to run bad script");
|
|
return 0;
|
|
}
|
|
|
|
setupGLState(rsc);
|
|
setupScript(rsc);
|
|
|
|
uint32_t ret = 0;
|
|
|
|
if (rsc->props.mLogScripts) {
|
|
ALOGV("%p ScriptC::run invoking root, ptr %p", rsc, mHal.info.root);
|
|
}
|
|
|
|
ret = rsc->mHal.funcs.script.invokeRoot(rsc, this);
|
|
|
|
if (rsc->props.mLogScripts) {
|
|
ALOGV("%p ScriptC::run invoking complete, ret=%i", rsc, ret);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
void ScriptC::runForEach(Context *rsc,
|
|
uint32_t slot,
|
|
const Allocation ** ains,
|
|
size_t inLen,
|
|
Allocation * aout,
|
|
const void * usr,
|
|
size_t usrBytes,
|
|
const RsScriptCall *sc) {
|
|
if (slot >= mHal.info.exportedForEachCount) {
|
|
rsc->setError(RS_ERROR_BAD_SCRIPT,
|
|
"The forEach kernel index is out of bounds");
|
|
return;
|
|
}
|
|
|
|
// Trace this function call.
|
|
// To avoid overhead we only build the string if tracing is actually
|
|
// enabled.
|
|
std::stringstream ss;
|
|
if (ATRACE_ENABLED()) {
|
|
ss << "runForEach slot[" << slot << "]";
|
|
}
|
|
std::string msgStr(ss.str());
|
|
ATRACE_NAME(msgStr.c_str());
|
|
|
|
if (mRSC->hadFatalError()) return;
|
|
|
|
Context::PushState ps(rsc);
|
|
|
|
setupGLState(rsc);
|
|
setupScript(rsc);
|
|
|
|
if (rsc->props.mLogScripts) {
|
|
ALOGV("%p ScriptC::runForEach invoking slot %i, ptr %p", rsc, slot, this);
|
|
}
|
|
|
|
if (rsc->mHal.funcs.script.invokeForEachMulti != nullptr) {
|
|
rsc->mHal.funcs.script.invokeForEachMulti(rsc, this, slot, ains, inLen,
|
|
aout, usr, usrBytes, sc);
|
|
|
|
} else if (inLen == 1) {
|
|
rsc->mHal.funcs.script.invokeForEach(rsc, this, slot, ains[0], aout,
|
|
usr, usrBytes, sc);
|
|
|
|
} else {
|
|
rsc->setError(RS_ERROR_FATAL_DRIVER,
|
|
"Driver support for multi-input not present");
|
|
}
|
|
}
|
|
|
|
void ScriptC::runReduce(Context *rsc, uint32_t slot,
|
|
const Allocation ** ains, size_t inLen,
|
|
Allocation *aout, const RsScriptCall *sc) {
|
|
// TODO: Record the name of the kernel in the tracing information.
|
|
ATRACE_CALL();
|
|
|
|
if (slot >= mHal.info.exportedReduceCount) {
|
|
rsc->setError(RS_ERROR_BAD_SCRIPT, "The general reduce kernel index is out of bounds");
|
|
return;
|
|
}
|
|
if (mRSC->hadFatalError()) return;
|
|
|
|
setupScript(rsc);
|
|
|
|
if (rsc->props.mLogScripts) {
|
|
ALOGV("%p ScriptC::runReduce invoking slot %i, ptr %p", rsc, slot, this);
|
|
}
|
|
|
|
rsc->mHal.funcs.script.invokeReduce(rsc, this, slot, ains, inLen, aout, sc);
|
|
}
|
|
|
|
void ScriptC::Invoke(Context *rsc, uint32_t slot, const void *data, size_t len) {
|
|
ATRACE_CALL();
|
|
|
|
if (slot >= mHal.info.exportedFunctionCount) {
|
|
rsc->setError(RS_ERROR_BAD_SCRIPT, "The invokable index is out of bounds");
|
|
return;
|
|
}
|
|
if (mRSC->hadFatalError()) return;
|
|
|
|
setupScript(rsc);
|
|
|
|
if (rsc->props.mLogScripts) {
|
|
ALOGV("%p ScriptC::Invoke invoking slot %i, ptr %p", rsc, slot, this);
|
|
}
|
|
rsc->mHal.funcs.script.invokeFunction(rsc, this, slot, data, len);
|
|
}
|
|
|
|
static const bool kDebugBitcode = false;
|
|
|
|
#ifndef RS_COMPATIBILITY_LIB
|
|
|
|
static bool dumpBitcodeFile(const char *cacheDir, const char *resName,
|
|
const char *suffix, const uint8_t *bitcode,
|
|
size_t bitcodeLen) {
|
|
std::string f(cacheDir);
|
|
f.append("/");
|
|
f.append(resName);
|
|
f.append("#");
|
|
f.append(suffix);
|
|
f.append(".bc");
|
|
|
|
if (!ScriptC::createCacheDir(cacheDir)) {
|
|
return false;
|
|
}
|
|
|
|
FILE *fp = fopen(f.c_str(), "we");
|
|
if (!fp) {
|
|
ALOGE("Could not open %s", f.c_str());
|
|
return false;
|
|
}
|
|
|
|
size_t nWritten = fwrite(bitcode, 1, bitcodeLen, fp);
|
|
fclose(fp);
|
|
if (nWritten != bitcodeLen) {
|
|
ALOGE("Could not write %s", f.c_str());
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
#endif // !RS_COMPATIBILITY_LIB
|
|
|
|
|
|
bool ScriptC::runCompiler(Context *rsc,
|
|
const char *resName,
|
|
const char *cacheDir,
|
|
const uint8_t *bitcode,
|
|
size_t bitcodeLen) {
|
|
ATRACE_CALL();
|
|
//ALOGE("runCompiler %p %p %p %p %p %i", rsc, this, resName, cacheDir, bitcode, bitcodeLen);
|
|
#ifndef RS_COMPATIBILITY_LIB
|
|
uint32_t sdkVersion = 0;
|
|
bcinfo::BitcodeWrapper bcWrapper((const char *)bitcode, bitcodeLen);
|
|
if (!bcWrapper.unwrap()) {
|
|
ALOGE("Bitcode is not in proper container format (raw or wrapper)");
|
|
return false;
|
|
}
|
|
|
|
if (bcWrapper.getBCFileType() == bcinfo::BC_WRAPPER) {
|
|
sdkVersion = bcWrapper.getTargetAPI();
|
|
}
|
|
|
|
if (sdkVersion == 0) {
|
|
// This signals that we didn't have a wrapper containing information
|
|
// about the bitcode.
|
|
sdkVersion = rsc->getTargetSdkVersion();
|
|
}
|
|
|
|
// Save off the sdkVersion, so that we can handle broken cases later.
|
|
// Bug 19734267
|
|
mApiLevel = sdkVersion;
|
|
|
|
bcinfo::BitcodeTranslator BT((const char *)bitcode, bitcodeLen,
|
|
sdkVersion);
|
|
if (!BT.translate()) {
|
|
ALOGE("Failed to translate bitcode from version: %u", sdkVersion);
|
|
return false;
|
|
}
|
|
bitcode = (const uint8_t *) BT.getTranslatedBitcode();
|
|
bitcodeLen = BT.getTranslatedBitcodeSize();
|
|
|
|
if (kDebugBitcode) {
|
|
if (!dumpBitcodeFile(cacheDir, resName, "after", bitcode, bitcodeLen)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
|
|
// Set the optimization level of bcc to be the same as the
|
|
// optimization level used to compile the bitcode.
|
|
rsc->setOptLevel(bcWrapper.getOptimizationLevel());
|
|
|
|
if (!cacheDir) {
|
|
// MUST BE FIXED BEFORE ANYTHING USING C++ API IS RELEASED
|
|
cacheDir = getenv("EXTERNAL_STORAGE");
|
|
ALOGV("Cache dir changed to %s", cacheDir);
|
|
}
|
|
|
|
// ensure that cache dir exists
|
|
if (cacheDir && !createCacheDir(cacheDir)) {
|
|
return false;
|
|
}
|
|
#endif
|
|
|
|
if (!rsc->mHal.funcs.script.init(rsc, this, resName, cacheDir, bitcode, bitcodeLen, 0)) {
|
|
return false;
|
|
}
|
|
|
|
mInitialized = true;
|
|
#if !defined(RS_VENDOR_LIB) && !defined(RS_COMPATIBILITY_LIB)
|
|
mEnviroment.mFragment.set(rsc->getDefaultProgramFragment());
|
|
mEnviroment.mVertex.set(rsc->getDefaultProgramVertex());
|
|
mEnviroment.mFragmentStore.set(rsc->getDefaultProgramStore());
|
|
mEnviroment.mRaster.set(rsc->getDefaultProgramRaster());
|
|
#endif
|
|
|
|
rsc->mHal.funcs.script.invokeInit(rsc, this);
|
|
|
|
for (size_t i=0; i < mHal.info.exportedPragmaCount; ++i) {
|
|
const char * key = mHal.info.exportedPragmaKeyList[i];
|
|
const char * value = mHal.info.exportedPragmaValueList[i];
|
|
//ALOGE("pragma %s %s", keys[i], values[i]);
|
|
if (!strcmp(key, "version")) {
|
|
if (!strcmp(value, "1")) {
|
|
continue;
|
|
}
|
|
ALOGE("Invalid version pragma value: %s\n", value);
|
|
return false;
|
|
}
|
|
|
|
#if !defined(RS_VENDOR_LIB) && !defined(RS_COMPATIBILITY_LIB)
|
|
if (!strcmp(key, "stateVertex")) {
|
|
if (!strcmp(value, "default")) {
|
|
continue;
|
|
}
|
|
if (!strcmp(value, "parent")) {
|
|
mEnviroment.mVertex.clear();
|
|
continue;
|
|
}
|
|
ALOGE("Unrecognized value %s passed to stateVertex", value);
|
|
return false;
|
|
}
|
|
|
|
if (!strcmp(key, "stateRaster")) {
|
|
if (!strcmp(value, "default")) {
|
|
continue;
|
|
}
|
|
if (!strcmp(value, "parent")) {
|
|
mEnviroment.mRaster.clear();
|
|
continue;
|
|
}
|
|
ALOGE("Unrecognized value %s passed to stateRaster", value);
|
|
return false;
|
|
}
|
|
|
|
if (!strcmp(key, "stateFragment")) {
|
|
if (!strcmp(value, "default")) {
|
|
continue;
|
|
}
|
|
if (!strcmp(value, "parent")) {
|
|
mEnviroment.mFragment.clear();
|
|
continue;
|
|
}
|
|
ALOGE("Unrecognized value %s passed to stateFragment", value);
|
|
return false;
|
|
}
|
|
|
|
if (!strcmp(key, "stateStore")) {
|
|
if (!strcmp(value, "default")) {
|
|
continue;
|
|
}
|
|
if (!strcmp(value, "parent")) {
|
|
mEnviroment.mFragmentStore.clear();
|
|
continue;
|
|
}
|
|
ALOGE("Unrecognized value %s passed to stateStore", value);
|
|
return false;
|
|
}
|
|
#endif
|
|
|
|
}
|
|
|
|
mSlots = new ObjectBaseRef<Allocation>[mHal.info.exportedVariableCount];
|
|
mTypes = new ObjectBaseRef<const Type>[mHal.info.exportedVariableCount];
|
|
|
|
return true;
|
|
}
|
|
|
|
namespace android {
|
|
namespace renderscript {
|
|
|
|
RsScript rsi_ScriptCCreate(Context *rsc,
|
|
const char *resName, size_t resName_length,
|
|
const char *cacheDir, size_t cacheDir_length,
|
|
const char *text, size_t text_length)
|
|
{
|
|
ScriptC *s = new ScriptC(rsc);
|
|
|
|
if (!s->runCompiler(rsc, resName, cacheDir, (uint8_t *)text, text_length)) {
|
|
// Error during compile, destroy s and return null.
|
|
ObjectBase::checkDelete(s);
|
|
return nullptr;
|
|
}
|
|
|
|
s->incUserRef();
|
|
return s;
|
|
}
|
|
|
|
} // namespace renderscript
|
|
} // namespace android
|