static void emit_c_clone_array(lcmgen_t *lcm, FILE *f, lcm_struct_t *lr)
{
char *tn = lr->structname->lctypename;
char *tn_ = dots_to_underscores(tn);
emit(0,"int __%s_clone_array(const %s *p, %s *q, int elements)", tn_, tn_, tn_);
emit(0,"{");
emit(1, "int element;");
emit(1, "for (element = 0; element < elements; element++) {");
emit(0,"");
for (unsigned int m = 0; m < g_ptr_array_size(lr->members); m++) {
lcm_member_t *lm = (lcm_member_t *) g_ptr_array_index(lr->members, m);
emit_c_array_loops_start(lcm, f, lm, "q", lcm_is_constant_size_array(lm) ? FLAG_NONE : FLAG_EMIT_MALLOCS);
int indent = 2+imax(0, g_ptr_array_size(lm->dimensions) - 1);
emit(indent, "__%s_clone_array(%s, %s, %s);",
dots_to_underscores (lm->type->lctypename),
make_accessor(lm, "p", g_ptr_array_size(lm->dimensions) - 1),
make_accessor(lm, "q", g_ptr_array_size(lm->dimensions) - 1),
make_array_size(lm, "p", g_ptr_array_size(lm->dimensions) - 1));
emit_c_array_loops_end(lcm, f, lm, "p", FLAG_NONE);
emit(0,"");
}
emit(1, "}");
emit(1, "return 0;");
emit(0,"}");
emit(0,"");
}
static void emit_c_encoded_array_size(lcmgen_t *lcm, FILE *f, lcm_struct_t *ls)
{
char *tn = ls->structname->lctypename;
char *tn_ = dots_to_underscores(tn);
emit(0,"int __%s_encoded_array_size(const %s *p, int elements)", tn_, tn_);
emit(0,"{");
emit(1,"int size = 0, element;");
emit(1, "for (element = 0; element < elements; element++) {");
emit(0,"");
for (unsigned int m = 0; m < g_ptr_array_size(ls->members); m++) {
lcm_member_t *lm = (lcm_member_t *) g_ptr_array_index(ls->members, m);
emit_c_array_loops_start(lcm, f, lm, "p", FLAG_NONE);
int indent = 2+imax(0, g_ptr_array_size(lm->dimensions) - 1);
emit(indent, "size += __%s_encoded_array_size(%s, %s);",
dots_to_underscores (lm->type->lctypename),
make_accessor(lm, "p", g_ptr_array_size(lm->dimensions) - 1),
make_array_size(lm, "p", g_ptr_array_size(lm->dimensions) - 1));
emit_c_array_loops_end(lcm, f, lm, "p", FLAG_NONE);
emit(0,"");
}
emit(1,"}");
emit(1, "return size;");
emit(0,"}");
emit(0,"");
}
static void emit_c_decode_array(lcmgen_t *lcm, FILE *f, lcm_struct_t *ls)
{
char *tn = ls->structname->lctypename;
char *tn_ = dots_to_underscores(tn);
emit(0,"int __%s_decode_array(const void *buf, int offset, int maxlen, %s *p, int elements)", tn_, tn_);
emit(0,"{");
emit(1, "int pos = 0, thislen, element;");
emit(0,"");
emit(1, "for (element = 0; element < elements; element++) {");
emit(0,"");
for (unsigned int m = 0; m < g_ptr_array_size(ls->members); m++) {
lcm_member_t *lm = (lcm_member_t *) g_ptr_array_index(ls->members, m);
emit_c_array_loops_start(lcm, f, lm, "p", lcm_is_constant_size_array(lm) ? FLAG_NONE : FLAG_EMIT_MALLOCS);
int indent = 2+imax(0, g_ptr_array_size(lm->dimensions) - 1);
emit(indent, "thislen = __%s_decode_array(buf, offset + pos, maxlen - pos, %s, %s);",
dots_to_underscores (lm->type->lctypename),
make_accessor(lm, "p", g_ptr_array_size(lm->dimensions) - 1),
make_array_size(lm, "p", g_ptr_array_size(lm->dimensions) - 1));
emit(indent, "if (thislen < 0) return thislen; else pos += thislen;");
emit_c_array_loops_end(lcm, f, lm, "p", FLAG_NONE);
emit(0,"");
}
emit(1, "}");
emit(1, "return pos;");
emit(0,"}");
emit(0,"");
}
static void emit_c_array_loops_end(lcmgen_t *lcm, FILE *f, lcm_member_t *lm, const char *n, int flags)
{
if (g_ptr_array_size(lm->dimensions) == 0)
return;
for (unsigned int i = 0; i < g_ptr_array_size(lm->dimensions) - 1; i++) {
int indent = g_ptr_array_size(lm->dimensions) - i;
if (flags & FLAG_EMIT_FREES) {
char *accessor = make_accessor(lm, "p", g_ptr_array_size(lm->dimensions) - 1 - i);
emit(indent+1, "if (%s) free(%s);", accessor, accessor);
}
emit(indent, "}");
emit(indent, "}");
}
if (flags & FLAG_EMIT_FREES) {
char *accessor = make_accessor(lm, "p", 0);
emit(2, "if (%s) free(%s);", accessor, accessor);
}
}
static void emit_c_array_loops_start(lcmgen_t *lcm, FILE *f, lcm_member_t *lm, const char *n, int flags)
{
if (g_ptr_array_size(lm->dimensions) == 0)
return;
for (unsigned int i = 0; i < g_ptr_array_size(lm->dimensions) - 1; i++) {
char var = 'a' + i;
if (flags & FLAG_EMIT_MALLOCS) {
char stars[1000];
for (unsigned int s = 0; s < g_ptr_array_size(lm->dimensions) - 1 - i; s++) {
stars[s] = '*';
stars[s+1] = 0;
}
emit(2+i, "%s = (%s%s*) lcm_malloc(sizeof(%s%s) * %s);",
make_accessor(lm, n, i),
map_type_name(lm->type->lctypename),
stars,
map_type_name(lm->type->lctypename),
stars,
make_array_size(lm, n, i));
}
emit(2+i, "{ int %c;", var);
emit(2+i, "for (%c = 0; %c < %s; %c++) {", var, var, make_array_size(lm, "p", i), var);
}
if (flags & FLAG_EMIT_MALLOCS) {
emit(2 + g_ptr_array_size(lm->dimensions) - 1, "%s = (%s*) lcm_malloc(sizeof(%s) * %s);",
make_accessor(lm, n, g_ptr_array_size(lm->dimensions) - 1),
map_type_name(lm->type->lctypename),
map_type_name(lm->type->lctypename),
make_array_size(lm, n, g_ptr_array_size(lm->dimensions) - 1));
}
}
int emit_java(zcmgen_t *zcm)
{
GHashTable *type_table = g_hash_table_new(g_str_hash, g_str_equal);
g_hash_table_insert(type_table, "byte", prim("byte",
"# = ins.readByte();",
"outs.writeByte(#);"));
g_hash_table_insert(type_table, "int8_t", prim("byte",
"# = ins.readByte();",
"outs.writeByte(#);"));
g_hash_table_insert(type_table, "int16_t", prim("short",
"# = ins.readShort();",
"outs.writeShort(#);"));
g_hash_table_insert(type_table, "int32_t", prim("int",
"# = ins.readInt();",
"outs.writeInt(#);"));
g_hash_table_insert(type_table, "int64_t", prim("long",
"# = ins.readLong();",
"outs.writeLong(#);"));
g_hash_table_insert(type_table, "string", prim("String",
"__strbuf = new char[ins.readInt()-1]; for (int _i = 0; _i < __strbuf.length; _i++) __strbuf[_i] = (char) (ins.readByte()&0xff); ins.readByte(); # = new String(__strbuf);",
"__strbuf = new char[#.length()]; #.getChars(0, #.length(), __strbuf, 0); outs.writeInt(__strbuf.length+1); for (int _i = 0; _i < __strbuf.length; _i++) outs.write(__strbuf[_i]); outs.writeByte(0);"));
// g_hash_table_insert(type_table, "string", prim("String",
// "__strbuf = new byte[ins.readInt()-1]; ins.readFully(__strbuf); ins.readByte(); # = new String(__strbuf, \"UTF-8\");",
// "__strbuf = #.getBytes(\"UTF-8\"); outs.writeInt(__strbuf.length+1); outs.write(__strbuf, 0, __strbuf.length); outs.writeByte(0);"));
g_hash_table_insert(type_table, "boolean", prim("boolean",
"# = ins.readByte()!=0;",
"outs.writeByte( # ? 1 : 0);"));
g_hash_table_insert(type_table, "float", prim("float",
"# = ins.readFloat();",
"outs.writeFloat(#);"));
g_hash_table_insert(type_table, "double", prim("double",
"# = ins.readDouble();",
"outs.writeDouble(#);"));
//////////////////////////////////////////////////////////////
// ENUMS
for (unsigned int en = 0; en < g_ptr_array_size(zcm->enums); en++) {
zcm_enum_t *le = (zcm_enum_t *) g_ptr_array_index(zcm->enums, en);
const char *classname = make_fqn(zcm, le->enumname->lctypename);
char *path = g_strdup_printf("%s%s%s.java",
getopt_get_string(zcm->gopt, "jpath"),
strlen(getopt_get_string(zcm->gopt, "jpath")) > 0 ? G_DIR_SEPARATOR_S : "",
dots_to_slashes(classname));
if (!zcm_needs_generation(zcm, le->zcmfile, path))
continue;
if (getopt_get_bool(zcm->gopt, "jmkdir"))
make_dirs_for_file(path);
FILE *f = fopen(path, "w");
if (f==NULL)
return -1;
if (strlen(le->enumname->package) > 0)
emit(0, "package %s;", le->enumname->package);
else
emit(0, "package %s;", getopt_get_string(zcm->gopt, "jdefaultpkg"));
emit(0, " ");
emit(0, "import java.io.*;");
emit(0, "import java.util.*;");
emit(0, " ");
emit(0, "public final class %s %s", le->enumname->shortname, getopt_get_string(zcm->gopt, "jdecl"));
emit(0, "{");
emit(1, "public int value;");
emit(0, " ");
for (unsigned int v = 0; v < g_ptr_array_size(le->values); v++) {
zcm_enum_value_t *lev = (zcm_enum_value_t *) g_ptr_array_index(le->values, v);
emit(1, "public static final int %-16s = %i;",
lev->valuename, lev->value);
}
emit(0," ");
emit(1,"public %s(int value) { this.value = value; }",
le->enumname->shortname);
emit(0," ");
emit(1,"public int getValue() { return value; }");
emit(0," ");
emit(1,"public void _encodeRecursive(DataOutput outs) throws IOException");
emit(1,"{");
emit(2,"outs.writeInt(this.value);");
emit(1,"}");
emit(0," ");
emit(1,"public void encode(DataOutput outs) throws IOException");
emit(1,"{");
emit(2,"outs.writeLong(ZCM_FINGERPRINT);");
emit(2,"_encodeRecursive(outs);");
emit(1,"}");
emit(0," ");
emit(1,"public static %s _decodeRecursiveFactory(DataInput ins) throws IOException", make_fqn(zcm, le->enumname->lctypename));
emit(1,"{");
emit(2,"%s o = new %s(0);", make_fqn(zcm, le->enumname->lctypename), make_fqn(zcm, le->enumname->lctypename));
emit(2,"o._decodeRecursive(ins);");
emit(2,"return o;");
emit(1,"}");
emit(0," ");
emit(1,"public void _decodeRecursive(DataInput ins) throws IOException");
emit(1,"{");
emit(2,"this.value = ins.readInt();");
emit(1,"}");
emit(0," ");
emit(1,"public %s(DataInput ins) throws IOException", le->enumname->shortname);
emit(1,"{");
emit(2,"long hash = ins.readLong();");
emit(2,"if (hash != ZCM_FINGERPRINT)");
emit(3, "throw new IOException(\"ZCM Decode error: bad fingerprint\");");
emit(2,"_decodeRecursive(ins);");
emit(1,"}");
emit(0," ");
emit(1,"public %s copy()", classname);
emit(1,"{");
emit(2,"return new %s(this.value);", classname);
emit(1,"}");
emit(0," ");
emit(1,"public static final long _hashRecursive(ArrayList<Class<?>> clss)");
emit(1,"{");
emit(2,"return ZCM_FINGERPRINT;");
emit(1,"}");
emit(0," ");
emit(1, "public static final long ZCM_FINGERPRINT = 0x%016"PRIx64"L;", le->hash);
emit(0, "}");
fclose(f);
}
for (unsigned int st = 0; st < g_ptr_array_size(zcm->structs); st++) {
zcm_struct_t *lr = (zcm_struct_t *) g_ptr_array_index(zcm->structs, st);
const char *classname = make_fqn(zcm, lr->structname->lctypename);
char *path = g_strdup_printf("%s%s%s.java",
getopt_get_string(zcm->gopt, "jpath"),
strlen(getopt_get_string(zcm->gopt, "jpath")) > 0 ? G_DIR_SEPARATOR_S : "",
dots_to_slashes(classname));
if (!zcm_needs_generation(zcm, lr->zcmfile, path))
continue;
if (getopt_get_bool(zcm->gopt, "jmkdir"))
make_dirs_for_file(path);
FILE *f = fopen(path, "w");
if (f==NULL)
return -1;
emit(0, "/* ZCM type definition class file\n"
" * This file was automatically generated by zcm-gen\n"
" * DO NOT MODIFY BY HAND!!!!\n"
" */\n");
if (strlen(lr->structname->package) > 0)
emit(0, "package %s;", lr->structname->package);
else
emit(0, "package %s;", getopt_get_string(zcm->gopt, "jdefaultpkg"));
emit(0, " ");
emit(0, "import java.io.*;");
if (0) {
// Determine if we even need the java.nio.* package.
int usenio = 0;
for (unsigned int member = 0; member < g_ptr_array_size(lr->members); member++) {
zcm_member_t *lm = (zcm_member_t *) g_ptr_array_index(lr->members, member);
primitive_info_t *pinfo = (primitive_info_t*) g_hash_table_lookup(type_table, lm->type->lctypename);
if (pinfo!=NULL && !strcmp(pinfo->storage, "float")) {
usenio = 1;
break;
}
}
if (usenio)
emit(0, "import java.nio.*;");
}
emit(0, "import java.util.*;");
emit(0, "import zcm.zcm.*;");
emit(0, " ");
emit(0, "public final class %s %s", lr->structname->shortname, getopt_get_string(zcm->gopt, "jdecl"));
emit(0, "{");
for (unsigned int member = 0; member < g_ptr_array_size(lr->members); member++) {
zcm_member_t *lm = (zcm_member_t *) g_ptr_array_index(lr->members, member);
primitive_info_t *pinfo = (primitive_info_t*) g_hash_table_lookup(type_table, lm->type->lctypename);
emit_start(1, "public ");
if (pinfo==NULL) {
emit_continue("%s", make_fqn(zcm, lm->type->lctypename));
} else {
emit_continue("%s", pinfo->storage);
}
emit_continue(" %s", lm->membername);
for (unsigned int i = 0; i < g_ptr_array_size(lm->dimensions); i++)
emit_continue("[]");
emit_end(";");
}
emit(0," ");
// public constructor
emit(1,"public %s()", lr->structname->shortname);
emit(1,"{");
// pre-allocate any fixed-size arrays.
for (unsigned int member = 0; member < g_ptr_array_size(lr->members); member++) {
zcm_member_t *lm = (zcm_member_t *) g_ptr_array_index(lr->members, member);
primitive_info_t *pinfo = (primitive_info_t*) g_hash_table_lookup(type_table, lm->type->lctypename);
if (g_ptr_array_size(lm->dimensions)==0 || !zcm_is_constant_size_array(lm))
continue;
emit_start(2, "%s = new ", lm->membername);
if (pinfo != NULL)
emit_continue("%s", pinfo->storage);
else
emit_continue("%s", make_fqn(zcm, lm->type->lctypename));
for (unsigned int i = 0; i < g_ptr_array_size(lm->dimensions); i++) {
zcm_dimension_t *dim = (zcm_dimension_t*) g_ptr_array_index(lm->dimensions, i);
emit_continue("[%s]", dim->size);
}
emit_end(";");
}
emit(1,"}");
emit(0," ");
emit(1, "public static final long ZCM_FINGERPRINT;");
emit(1, "public static final long ZCM_FINGERPRINT_BASE = 0x%016"PRIx64"L;", lr->hash);
emit(0," ");
//////////////////////////////////////////////////////////////
// CONSTANTS
for (unsigned int cn = 0; cn < g_ptr_array_size(lr->constants); cn++) {
zcm_constant_t *lc = (zcm_constant_t *) g_ptr_array_index(lr->constants, cn);
assert(zcm_is_legal_const_type(lc->lctypename));
if (!strcmp(lc->lctypename, "int8_t")) {
emit(1, "public static final byte %s = (byte) %s;", lc->membername, lc->val_str);
} else if (!strcmp(lc->lctypename, "int16_t")) {
emit(1, "public static final short %s = (short) %s;", lc->membername, lc->val_str);
} else if (!strcmp(lc->lctypename, "int32_t")) {
emit(1, "public static final int %s = %s;", lc->membername, lc->val_str);
} else if (!strcmp(lc->lctypename, "int64_t")) {
emit(1, "public static final long %s = %sL;", lc->membername, lc->val_str);
} else if (!strcmp(lc->lctypename, "float")) {
emit(1, "public static final float %s = %sf;", lc->membername, lc->val_str);
} else if (!strcmp(lc->lctypename, "double")) {
emit(1, "public static final double %s = %s;", lc->membername, lc->val_str);
} else {
assert(0);
}
}
if (g_ptr_array_size(lr->constants) > 0)
emit(0, "");
///////////////// compute fingerprint //////////////////
emit(1, "static {");
emit(2, "ZCM_FINGERPRINT = _hashRecursive(new ArrayList<Class<?>>());");
emit(1, "}");
emit(0, " ");
emit(1, "public static long _hashRecursive(ArrayList<Class<?>> classes)");
emit(1, "{");
emit(2, "if (classes.contains(%s.class))", make_fqn(zcm, lr->structname->lctypename));
emit(3, "return 0L;");
emit(0, " ");
emit(2, "classes.add(%s.class);", make_fqn(zcm, lr->structname->lctypename));
emit(2, "long hash = ZCM_FINGERPRINT_BASE");
for (unsigned int member = 0; member < g_ptr_array_size(lr->members); member++) {
zcm_member_t *lm = (zcm_member_t *) g_ptr_array_index(lr->members, member);
primitive_info_t *pinfo = (primitive_info_t*) g_hash_table_lookup(type_table, lm->type->lctypename);
if (pinfo)
continue;
emit(3, " + %s._hashRecursive(classes)", make_fqn(zcm, lm->type->lctypename));
}
emit(3,";");
emit(2, "classes.remove(classes.size() - 1);");
emit(2, "return (hash<<1) + ((hash>>63)&1);");
emit(1, "}");
emit(0, " ");
///////////////// encode //////////////////
emit(1,"public void encode(DataOutput outs) throws IOException");
emit(1,"{");
emit(2,"outs.writeLong(ZCM_FINGERPRINT);");
emit(2,"_encodeRecursive(outs);");
emit(1,"}");
emit(0," ");
emit(1,"public void _encodeRecursive(DataOutput outs) throws IOException");
emit(1,"{");
if(struct_has_string_member(lr))
emit(2, "char[] __strbuf = null;");
char accessor[1024];
for (unsigned int member = 0; member < g_ptr_array_size(lr->members); member++) {
zcm_member_t *lm = (zcm_member_t *) g_ptr_array_index(lr->members, member);
primitive_info_t *pinfo = (primitive_info_t*) g_hash_table_lookup(type_table, lm->type->lctypename);
make_accessor(lm, "this", accessor);
encode_recursive(zcm, lm, f, pinfo, accessor, 0);
emit(0," ");
}
emit(1,"}");
emit(0," ");
///////////////// decode //////////////////
// decoding constructors
emit(1, "public %s(byte[] data) throws IOException", lr->structname->shortname);
emit(1, "{");
emit(2, "this(new ZCMDataInputStream(data));");
emit(1, "}");
emit(0, " ");
emit(1,"public %s(DataInput ins) throws IOException", lr->structname->shortname);
emit(1,"{");
emit(2,"if (ins.readLong() != ZCM_FINGERPRINT)");
emit(3, "throw new IOException(\"ZCM Decode error: bad fingerprint\");");
emit(0," ");
emit(2,"_decodeRecursive(ins);");
emit(1,"}");
emit(0," ");
emit(1,"public static %s _decodeRecursiveFactory(DataInput ins) throws IOException", make_fqn(zcm, lr->structname->lctypename));
emit(1,"{");
emit(2,"%s o = new %s();", make_fqn(zcm, lr->structname->lctypename), make_fqn(zcm, lr->structname->lctypename));
emit(2,"o._decodeRecursive(ins);");
emit(2,"return o;");
emit(1,"}");
emit(0," ");
emit(1,"public void _decodeRecursive(DataInput ins) throws IOException");
emit(1,"{");
if(struct_has_string_member(lr))
emit(2, "char[] __strbuf = null;");
for (unsigned int member = 0; member < g_ptr_array_size(lr->members); member++) {
zcm_member_t *lm = (zcm_member_t *) g_ptr_array_index(lr->members, member);
primitive_info_t *pinfo = (primitive_info_t*) g_hash_table_lookup(type_table, lm->type->lctypename);
make_accessor(lm, "this", accessor);
// allocate an array if necessary
if (g_ptr_array_size(lm->dimensions) > 0) {
emit_start(2, "this.%s = new ", lm->membername);
if (pinfo != NULL)
emit_continue("%s", pinfo->storage);
else
emit_continue("%s", make_fqn(zcm, lm->type->lctypename));
for (unsigned int i = 0; i < g_ptr_array_size(lm->dimensions); i++) {
zcm_dimension_t *dim = (zcm_dimension_t*) g_ptr_array_index(lm->dimensions, i);
emit_continue("[(int) %s]", dim->size);
}
emit_end(";");
}
decode_recursive(zcm, lm, f, pinfo, accessor, 0);
emit(0," ");
}
emit(1,"}");
emit(0," ");
///////////////// copy //////////////////
emit(1,"public %s copy()", classname);
emit(1,"{");
emit(2,"%s outobj = new %s();", classname, classname);
for (unsigned int member = 0; member < g_ptr_array_size(lr->members); member++) {
zcm_member_t *lm = (zcm_member_t *) g_ptr_array_index(lr->members, member);
primitive_info_t *pinfo = (primitive_info_t*) g_hash_table_lookup(type_table, lm->type->lctypename);
make_accessor(lm, "", accessor);
// allocate an array if necessary
if (g_ptr_array_size(lm->dimensions) > 0) {
emit_start(2, "outobj.%s = new ", lm->membername);
if (pinfo != NULL)
emit_continue("%s", pinfo->storage);
else
emit_continue("%s", make_fqn(zcm, lm->type->lctypename));
for (unsigned int i = 0; i < g_ptr_array_size(lm->dimensions); i++) {
zcm_dimension_t *dim = (zcm_dimension_t*) g_ptr_array_index(lm->dimensions, i);
emit_continue("[(int) %s]", dim->size);
}
emit_end(";");
}
copy_recursive(zcm, lm, f, pinfo, accessor, 0);
emit(0," ");
}
emit(2,"return outobj;");
emit(1,"}");
emit(0," ");
////////
emit(0, "}\n");
fclose(f);
}
/* XXX deallocate our storage. unfinished since memory leaks are non-critical for zcm-gen.
hashtable_iterator_t *hit = hashtable_iterator_create(type_table);
hashtable_entry_t *entry;
while ((entry = hashtable_iterator_next(hit)) != NULL) {
free((char*) entry->value);
}
hashtable_iterator_destroy(hit);
hashtable_destroy(type_table);
*/
return 0;
}
