BigfootDev/flatbuffers
1
0
Fork 1
mirror of https://github.com/google/flatbuffers.git synced 2026-06-01 19:58:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3 from TGIshib/key

Browse Source 
Update idl_gen_general.cpp
This commit is contained in:
Raman
2016-07-23 15:17:28 +03:00
committed by GitHub
parent 247388a20c 38b3893211
commit 06c1ad5a73
1 changed files with 66 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
src/idl_gen_general.cpp
View File

@@ -955,6 +955,7 @@ void GenStruct(StructDef &struct_def, std::string *code_ptr) {
}
}
code += "\n";
flatbuffers::FieldDef *key_field = nullptr;
if (struct_def.fixed) {
// create a struct constructor function
code += " public static " + GenOffsetType(struct_def) + " ";
@@ -1042,6 +1043,7 @@ void GenStruct(StructDef &struct_def, std::string *code_ptr) {
for (auto it = struct_def.fields.vec.begin();
it != struct_def.fields.vec.end(); ++it) {
auto &field = **it;
if (field.key) key_field = &field;
if (field.deprecated) continue;
code += " public static void " + FunctionStart('A') + "dd";
code += MakeCamel(field.name);
@@ -1125,6 +1127,70 @@ void GenStruct(StructDef &struct_def, std::string *code_ptr) {
code += "); }\n";
}
}
if (struct_def.has_key) {
std::string key_name = lang_.language == IDLOptions::kCSharp ? MakeCamel(key_field->name) : key_field->name + "()";
std::string key_type = GenTypeGet(key_field->value.type);
code += "\n public int " + FunctionStart('K') + "eyCompareLessThan(" + struct_def.name + " o) ";
if (key_field->value.type.base_type != BASE_TYPE_STRING) {
code += "{\n if (" + key_name + " < o." + key_name + ") return -1;\n";
code += " else if (" + key_name + " == o." + key_name + ") return 0;\n";
code += " else return 1;\n }\n";
}
else {
code += "{ return " + key_name + "." + FunctionStart('C') + "ompareTo(o." + key_name + "); }\n";
}
code += " public int " + FunctionStart('K') + "eyCompareWithValue(" + key_type + " val) ";
if (key_field->value.type.base_type != BASE_TYPE_STRING) {
code += "{\n if (" + key_name + " < val) return -1;\n";
code += " else if (" + key_name + " == val) return 0;\n";
code += " else return 1;\n }\n";
}
else {
code += "{ return " + key_name + "." + FunctionStart('C') + "ompareTo(val); }\n";
}
code += "\n public static " + struct_def.name + "[] " + FunctionStart('C') + "reateMyTableVector(";
code += (lang_.language == IDLOptions::kCSharp ? "Offset<" + struct_def.name + ">" : "int") + "[] offsets, ByteBuffer bb) {\n";
code += " " + struct_def.name + "[] tables = new " + struct_def.name + "[offsets." + FunctionStart('L') + "ength];\n";
code += " for (int i = 0; i < offsets." + FunctionStart('L') + "ength; i++)\n {\n";
code += " tables[i] = new " + struct_def.name + "().__init(bb." + (lang_.language == IDLOptions::kCSharp ? "" : "array().");
code += FunctionStart('L') + "ength - offsets[i]" + (lang_.language == IDLOptions::kCSharp ? ".Value" : "") + ", bb);\n";
code += " }\n Array";
code += lang_.language == IDLOptions::kCSharp ? "." : "s.";
code += FunctionStart('S') + "ort(tables, (" + struct_def.name + " first, " + struct_def.name + " second) ";
code += lang_.language == IDLOptions::kCSharp ? "=>" : "->";
code += " first." + FunctionStart('K') + "eyCompareLessThan(second));\n";
code += " return tables;\n }\n";
code += "\n public static " + struct_def.name + " " + FunctionStart('L') + "ookupByKey(" + struct_def.name;
code += "[] tables, " + key_type + " key) {\n";
code += " int span = tables." + FunctionStart('L') + "ength;\n";
code += " int start = 0;\n\n";
code += " while (span != 0)\n";
code += " {\n";
code += " int middle = span / 2;\n";
code += " " + struct_def.name + " table = tables[start + middle];\n";
code += " int comp = table." + FunctionStart('K') + "eyCompareWithValue(key);\n\n";
code += " if (comp > 0)\n";
code += " {\n";
code += " span = middle;\n";
code += " }\n";
code += " else if (comp < 0)\n";
code += " {\n";
code += " middle++;\n";
code += " start += middle;\n";
code += " span -= middle;\n";
code += " }\n";
code += " else\n";
code += " {\n";
code += " return table;\n";
code += " }\n";
code += " }\n";
code += " return null;\n";
code += " }\n";
}
code += "};\n\n";
}
const LanguageParameters & lang_ = language_parameters[parser_.opts.lang];