diff options
Diffstat (limited to 'tools/aapt2/JavaClassGenerator.cpp')
-rw-r--r-- | tools/aapt2/JavaClassGenerator.cpp | 189 |
1 files changed, 189 insertions, 0 deletions
diff --git a/tools/aapt2/JavaClassGenerator.cpp b/tools/aapt2/JavaClassGenerator.cpp new file mode 100644 index 0000000..7ec2848 --- /dev/null +++ b/tools/aapt2/JavaClassGenerator.cpp @@ -0,0 +1,189 @@ +/* + * Copyright (C) 2015 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 "JavaClassGenerator.h" +#include "Resource.h" +#include "ResourceTable.h" +#include "ResourceValues.h" +#include "StringPiece.h" + +#include <ostream> +#include <set> +#include <sstream> +#include <tuple> + +namespace aapt { + +// The number of attributes to emit per line in a Styleable array. +constexpr size_t kAttribsPerLine = 4; + +JavaClassGenerator::JavaClassGenerator(std::shared_ptr<const ResourceTable> table, + Options options) : + mTable(table), mOptions(options) { +} + +static void generateHeader(std::ostream& out, const StringPiece16& package) { + out << "/* AUTO-GENERATED FILE. DO NOT MODIFY.\n" + " *\n" + " * This class was automatically generated by the\n" + " * aapt tool from the resource data it found. It\n" + " * should not be modified by hand.\n" + " */\n\n"; + out << "package " << package << ";" + << std::endl + << std::endl; +} + +static const std::set<StringPiece16> sJavaIdentifiers = { + u"abstract", u"assert", u"boolean", u"break", u"byte", + u"case", u"catch", u"char", u"class", u"const", u"continue", + u"default", u"do", u"double", u"else", u"enum", u"extends", + u"final", u"finally", u"float", u"for", u"goto", u"if", + u"implements", u"import", u"instanceof", u"int", u"interface", + u"long", u"native", u"new", u"package", u"private", u"protected", + u"public", u"return", u"short", u"static", u"strictfp", u"super", + u"switch", u"synchronized", u"this", u"throw", u"throws", + u"transient", u"try", u"void", u"volatile", u"while", u"true", + u"false", u"null" +}; + +static bool isValidSymbol(const StringPiece16& symbol) { + return sJavaIdentifiers.find(symbol) == sJavaIdentifiers.end(); +} + +/* + * Java symbols can not contain . or -, but those are valid in a resource name. + * Replace those with '_'. + */ +static std::u16string transform(const StringPiece16& symbol) { + std::u16string output = symbol.toString(); + for (char16_t& c : output) { + if (c == u'.' || c == u'-') { + c = u'_'; + } + } + return output; +} + +bool JavaClassGenerator::generateType(std::ostream& out, const ResourceTableType& type, + size_t packageId) { + const StringPiece finalModifier = mOptions.useFinal ? " final" : ""; + + for (const auto& entry : type.entries) { + ResourceId id = { packageId, type.typeId, entry->entryId }; + assert(id.isValid()); + + if (!isValidSymbol(entry->name)) { + mError = (std::stringstream() + << "invalid symbol name '" + << StringPiece16(entry->name) + << "'").str(); + return false; + } + + out << " " + << "public static" << finalModifier + << " int " << transform(entry->name) << " = " << id << ";" << std::endl; + } + return true; +} + +struct GenArgs : ValueVisitorArgs { + GenArgs(std::ostream& o, const ResourceEntry& e) : out(o), entry(e) { + } + + std::ostream& out; + const ResourceEntry& entry; +}; + +void JavaClassGenerator::visit(const Styleable& styleable, ValueVisitorArgs& a) { + const StringPiece finalModifier = mOptions.useFinal ? " final" : ""; + std::ostream& out = static_cast<GenArgs&>(a).out; + const ResourceEntry& entry = static_cast<GenArgs&>(a).entry; + + // This must be sorted by resource ID. + std::vector<std::pair<ResourceId, StringPiece16>> sortedAttributes; + sortedAttributes.reserve(styleable.entries.size()); + for (const auto& attr : styleable.entries) { + assert(attr.id.isValid() && "no ID set for Styleable entry"); + assert(attr.name.isValid() && "no name set for Styleable entry"); + sortedAttributes.emplace_back(attr.id, attr.name.entry); + } + std::sort(sortedAttributes.begin(), sortedAttributes.end()); + + // First we emit the array containing the IDs of each attribute. + out << " " + << "public static final int[] " << transform(entry.name) << " = {"; + + const size_t attrCount = sortedAttributes.size(); + for (size_t i = 0; i < attrCount; i++) { + if (i % kAttribsPerLine == 0) { + out << std::endl << " "; + } + + out << sortedAttributes[i].first; + if (i != attrCount - 1) { + out << ", "; + } + } + out << std::endl << " };" << std::endl; + + // Now we emit the indices into the array. + for (size_t i = 0; i < attrCount; i++) { + out << " " + << "public static" << finalModifier + << " int " << transform(entry.name) << "_" << transform(sortedAttributes[i].second) + << " = " << i << ";" << std::endl; + } +} + +bool JavaClassGenerator::generate(std::ostream& out) { + const size_t packageId = mTable->getPackageId(); + + generateHeader(out, mTable->getPackage()); + + out << "public final class R {" << std::endl; + + for (const auto& type : *mTable) { + out << " public static final class " << type->type << " {" << std::endl; + bool result; + if (type->type == ResourceType::kStyleable) { + for (const auto& entry : type->entries) { + assert(!entry->values.empty()); + if (!isValidSymbol(entry->name)) { + mError = (std::stringstream() + << "invalid symbol name '" + << StringPiece16(entry->name) + << "'").str(); + return false; + } + entry->values.front().value->accept(*this, GenArgs{ out, *entry }); + } + } else { + result = generateType(out, *type, packageId); + } + + if (!result) { + return false; + } + out << " }" << std::endl; + } + + out << "}" << std::endl; + return true; +} + +} // namespace aapt |