internal override void GenReadMethod(CodeBuffer cb, string fname, string tag, bool decl) { if (decl) { cb.Append(this.GetType() + " " + fname + ";\n"); } cb.Append("{\n"); JMap.IncrLevel(); cb.Append("org.apache.hadoop.record.Index " + JMap.GetId(Consts.RioPrefix + "midx" ) + " = " + Consts.RecordInput + ".startMap(\"" + tag + "\");\n"); cb.Append(fname + "=new " + this.GetType() + "();\n"); cb.Append("for (; !" + JMap.GetId(Consts.RioPrefix + "midx") + ".done(); " + JMap .GetId(Consts.RioPrefix + "midx") + ".incr()) {\n"); this.key.GenReadMethod(cb, JMap.GetId(Consts.RioPrefix + "k"), JMap.GetId(Consts. RioPrefix + "k"), true); this.value.GenReadMethod(cb, JMap.GetId(Consts.RioPrefix + "v"), JMap.GetId(Consts .RioPrefix + "v"), true); cb.Append(fname + ".put(" + JMap.GetId(Consts.RioPrefix + "k") + "," + JMap.GetId (Consts.RioPrefix + "v") + ");\n"); cb.Append("}\n"); cb.Append(Consts.RecordInput + ".endMap(\"" + tag + "\");\n"); JMap.DecrLevel(); cb.Append("}\n"); }
internal override void GenCompareTo(CodeBuffer cb, string fname, string other) { string setType = "java.util.Set<" + this.key.GetWrapperType() + "> "; string iterType = "java.util.Iterator<" + this.key.GetWrapperType() + "> "; cb.Append("{\n"); cb.Append(setType + JMap.GetId(Consts.RioPrefix + "set1") + " = " + fname + ".keySet();\n" ); cb.Append(setType + JMap.GetId(Consts.RioPrefix + "set2") + " = " + other + ".keySet();\n" ); cb.Append(iterType + JMap.GetId(Consts.RioPrefix + "miter1") + " = " + JMap.GetId (Consts.RioPrefix + "set1") + ".iterator();\n"); cb.Append(iterType + JMap.GetId(Consts.RioPrefix + "miter2") + " = " + JMap.GetId (Consts.RioPrefix + "set2") + ".iterator();\n"); cb.Append("for(; " + JMap.GetId(Consts.RioPrefix + "miter1") + ".hasNext() && " + JMap.GetId(Consts.RioPrefix + "miter2") + ".hasNext();) {\n"); cb.Append(this.key.GetType() + " " + JMap.GetId(Consts.RioPrefix + "k1") + " = " + JMap.GetId(Consts.RioPrefix + "miter1") + ".next();\n"); cb.Append(this.key.GetType() + " " + JMap.GetId(Consts.RioPrefix + "k2") + " = " + JMap.GetId(Consts.RioPrefix + "miter2") + ".next();\n"); this.key.GenCompareTo(cb, JMap.GetId(Consts.RioPrefix + "k1"), JMap.GetId(Consts. RioPrefix + "k2")); cb.Append("if (" + Consts.RioPrefix + "ret != 0) { return " + Consts.RioPrefix + "ret; }\n"); cb.Append("}\n"); cb.Append(Consts.RioPrefix + "ret = (" + JMap.GetId(Consts.RioPrefix + "set1") + ".size() - " + JMap.GetId(Consts.RioPrefix + "set2") + ".size());\n"); cb.Append("}\n"); }
internal override void GenCompareBytes(CodeBuffer cb) { cb.Append("{\n"); JMap.IncrLevel(); cb.Append("int " + JMap.GetId("mi1") + " = org.apache.hadoop.record.Utils.readVInt(b1, s1);\n" ); cb.Append("int " + JMap.GetId("mi2") + " = org.apache.hadoop.record.Utils.readVInt(b2, s2);\n" ); cb.Append("int " + JMap.GetId("mz1") + " = org.apache.hadoop.record.Utils.getVIntSize(" + JMap.GetId("mi1") + ");\n"); cb.Append("int " + JMap.GetId("mz2") + " = org.apache.hadoop.record.Utils.getVIntSize(" + JMap.GetId("mi2") + ");\n"); cb.Append("s1+=" + JMap.GetId("mz1") + "; s2+=" + JMap.GetId("mz2") + "; l1-=" + JMap.GetId("mz1") + "; l2-=" + JMap.GetId("mz2") + ";\n"); cb.Append("for (int " + JMap.GetId("midx") + " = 0; " + JMap.GetId("midx") + " < " + JMap.GetId("mi1") + " && " + JMap.GetId("midx") + " < " + JMap.GetId("mi2") + "; " + JMap.GetId("midx") + "++) {"); this.key.GenCompareBytes(cb); this.value.GenSlurpBytes(cb, "b1", "s1", "l1"); this.value.GenSlurpBytes(cb, "b2", "s2", "l2"); cb.Append("}\n"); cb.Append("if (" + JMap.GetId("mi1") + " != " + JMap.GetId("mi2") + ") { return (" + JMap.GetId("mi1") + "<" + JMap.GetId("mi2") + ")?-1:0; }\n"); JMap.DecrLevel(); cb.Append("}\n"); }
internal override void GenWriteMethod(CodeBuffer cb, string fname, string tag) { string setType = "java.util.Set<java.util.Map.Entry<" + this.key.GetWrapperType() + "," + this.value.GetWrapperType() + ">> "; string entryType = "java.util.Map.Entry<" + this.key.GetWrapperType() + "," + this .value.GetWrapperType() + "> "; string iterType = "java.util.Iterator<java.util.Map.Entry<" + this.key.GetWrapperType () + "," + this.value.GetWrapperType() + ">> "; cb.Append("{\n"); JMap.IncrLevel(); cb.Append(Consts.RecordOutput + ".startMap(" + fname + ",\"" + tag + "\");\n"); cb.Append(setType + JMap.GetId(Consts.RioPrefix + "es") + " = " + fname + ".entrySet();\n" ); cb.Append("for(" + iterType + JMap.GetId(Consts.RioPrefix + "midx") + " = " + JMap .GetId(Consts.RioPrefix + "es") + ".iterator(); " + JMap.GetId(Consts.RioPrefix + "midx") + ".hasNext();) {\n"); cb.Append(entryType + JMap.GetId(Consts.RioPrefix + "me") + " = " + JMap.GetId(Consts .RioPrefix + "midx") + ".next();\n"); cb.Append(this.key.GetType() + " " + JMap.GetId(Consts.RioPrefix + "k") + " = " + JMap.GetId(Consts.RioPrefix + "me") + ".getKey();\n"); cb.Append(this.value.GetType() + " " + JMap.GetId(Consts.RioPrefix + "v") + " = " + JMap.GetId(Consts.RioPrefix + "me") + ".getValue();\n"); this.key.GenWriteMethod(cb, JMap.GetId(Consts.RioPrefix + "k"), JMap.GetId(Consts .RioPrefix + "k")); this.value.GenWriteMethod(cb, JMap.GetId(Consts.RioPrefix + "v"), JMap.GetId(Consts .RioPrefix + "v")); cb.Append("}\n"); cb.Append(Consts.RecordOutput + ".endMap(" + fname + ",\"" + tag + "\");\n"); cb.Append("}\n"); JMap.DecrLevel(); }
internal virtual void GenReadMethod(CodeBuffer cb, string fname, string tag, bool decl) { if (decl) { cb.Append(this.name + " " + fname + ";\n"); } cb.Append(fname + "=" + Consts.RecordInput + ".read" + this.methodSuffix + "(\"" + tag + "\");\n"); }
internal virtual void GenGetSet(CodeBuffer cb, string fname) { cb.Append("virtual " + this.name + " get" + JType.ToCamelCase(fname) + "() const {\n" ); cb.Append("return " + fname + ";\n"); cb.Append("}\n"); cb.Append("virtual void set" + JType.ToCamelCase(fname) + "(" + this.name + " m_) {\n" ); cb.Append(fname + "=m_;\n"); cb.Append("}\n"); }
internal override void GenSlurpBytes(CodeBuffer cb, string b, string s, string l) { cb.Append("{\n"); JMap.IncrLevel(); cb.Append("int " + JMap.GetId("mi") + " = org.apache.hadoop.record.Utils.readVInt(" + b + ", " + s + ");\n"); cb.Append("int " + JMap.GetId("mz") + " = org.apache.hadoop.record.Utils.getVIntSize(" + JMap.GetId("mi") + ");\n"); cb.Append(s + "+=" + JMap.GetId("mz") + "; " + l + "-=" + JMap.GetId("mz") + ";\n" ); cb.Append("for (int " + JMap.GetId("midx") + " = 0; " + JMap.GetId("midx") + " < " + JMap.GetId("mi") + "; " + JMap.GetId("midx") + "++) {"); this.key.GenSlurpBytes(cb, b, s, l); this.value.GenSlurpBytes(cb, b, s, l); cb.Append("}\n"); JMap.DecrLevel(); cb.Append("}\n"); }
internal virtual void GenSetRTIFilter(CodeBuffer cb) { // do nothing by default return; }