gered/BWMirror-Generator
Archived
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases 3 Wiki Activity

add hardcoded equals/hashCode impls to certain generated classes

Browse source
This commit is contained in:
Gered 2017-04-14 12:01:15 -04:00
parent 6b810bcd2e
commit e5721392c6
1 changed files with 68 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

68
generator/src/main/java/bwmirror/generator/java/ClassMirror.java
View file

@ -41,6 +41,7 @@ public class ClassMirror extends Mirror {
out.println();
writeFields(cClass.getFields());
if (!cClass.isStatic()) {
writeEqualsAndHashCodeMethods();
writeInstanceMap();
writeConstructor();
writeInstanceGetter();
@ -53,6 +54,73 @@ public class ClassMirror extends Mirror {
nativeBinds = null;
}
private void writeEqualsMethodBeginning() {
out.println(INTEND + "@Override");
out.println(INTEND + "public boolean equals(Object o) {");
out.println(INTEND + INTEND + "if (this == o) return true;");
out.println(INTEND + INTEND + "if (o == null || getClass() != o.getClass()) return false;");
out.println();
out.println(INTEND + INTEND + cClass.getName() + " other = (" + cClass.getName() + ")o;");
out.println();
}
private void writeEqualsMethodEnding() {
out.println();
out.println(INTEND + INTEND + "return true;");
out.println(INTEND + "}");
}
private void writeHashCodeMethodBeginning() {
out.println();
out.println(INTEND + "@Override");
out.println(INTEND + "public int hashCode() {");
out.println(INTEND + INTEND + "int result;");
}
private void writeHashCodeMethodEnding() {
out.println(INTEND + INTEND + "return result;");
out.println(INTEND + "}");
out.println();
}
// for the simpler cases where the BWAPI object has an ID, and our equality test
// only needs to be a comparison of this ID. ID is assumed to be an int
private void writeEqualsAndHashCodeMethodsForObjWithID() {
writeEqualsMethodBeginning();
out.println(INTEND + INTEND + "if (getID() != other.getID()) return false;");
writeEqualsMethodEnding();
writeHashCodeMethodBeginning();
out.println(INTEND + INTEND + "result = getID();");
writeHashCodeMethodEnding();
}
// HACK: manually stuffing in some implementations for specific classes for now.
// it would probably take a fair bit of work to rig up some general-purpose
// solution for all classes ....
private void writeEqualsAndHashCodeMethods() {
String className = cClass.getName();
String packageName = context.getPackage();
if (className.equals("BaseLocation")) {
writeEqualsMethodBeginning();
out.println(INTEND + INTEND + "if (!getPosition().equals(other.getPosition())) return false;");
writeEqualsMethodEnding();
writeHashCodeMethodBeginning();
out.println(INTEND + INTEND + "result = getPosition().hashCode();");
writeHashCodeMethodEnding();
} else if (className.equals("Bullet")) {
writeEqualsAndHashCodeMethodsForObjWithID();
} else if (className.equals("Force")) {
writeEqualsAndHashCodeMethodsForObjWithID();
} else if (className.equals("Player")) {
writeEqualsAndHashCodeMethodsForObjWithID();
} else if (className.equals("Region") && packageName.equals("bwapi")) {
writeEqualsAndHashCodeMethodsForObjWithID();
} else if (className.equals("Unit")) {
writeEqualsAndHashCodeMethodsForObjWithID();
}
}
private void writeInstanceMap() {
out.println(INTEND + "private" + SPACE + "static" + SPACE + "Map<Long," + SPACE + cClass.getName() + "> instances = " +
"new HashMap<Long," + SPACE + cClass.getName() + ">()" + SEMICOLON);