From dc30c033ab14ae48c1adab08edf272c52db97dde Mon Sep 17 00:00:00 2001
From: bdemsky <bdemsky>
Date: Wed, 5 Apr 2006 23:01:54 +0000
Subject: [PATCH] Add memory allocation

---
 Robust/src/IR/Flat/BuildCode.java | 29 +++++++++++++++++++++++++----
 Robust/src/IR/State.java          |  7 +++++++
 Robust/src/Runtime/runtime.c      |  7 +++++++
 Robust/src/f.test                 |  6 +++++-
 4 files changed, 44 insertions(+), 5 deletions(-)

diff --git a/Robust/src/IR/Flat/BuildCode.java b/Robust/src/IR/Flat/BuildCode.java
index 5842dad5..05d1cdc1 100644
--- a/Robust/src/IR/Flat/BuildCode.java
+++ b/Robust/src/IR/Flat/BuildCode.java
@@ -69,6 +69,7 @@ public class BuildCode {
 	    ClassDescriptor cn=(ClassDescriptor)it.next();
 	    generateCallStructs(cn, outclassdefs, outstructs, outmethodheader);
 	}
+
 	outstructs.close();
 	outmethodheader.close();
 
@@ -76,6 +77,10 @@ public class BuildCode {
 	outmethod.println("#include \"methodheaders.h\"");
 	outmethod.println("#include \"virtualtable.h\"");
 	outmethod.println("#include <runtime.h>");
+
+	outclassdefs.println("extern int classsize[];");
+	generateSizeArray(outmethod);
+
 	Iterator classit=state.getClassSymbolTable().getDescriptorsIterator();
 	while(classit.hasNext()) {
 	    ClassDescriptor cn=(ClassDescriptor)classit.next();
@@ -109,15 +114,13 @@ public class BuildCode {
     private int maxcount=0;
 
     private void buildVirtualTables(PrintWriter outvirtual) {
-	int numclasses=0;
     	Iterator classit=state.getClassSymbolTable().getDescriptorsIterator();
 	while(classit.hasNext()) {
 	    ClassDescriptor cd=(ClassDescriptor)classit.next();
 	    if (virtualcalls.getMethodCount(cd)>maxcount)
 		maxcount=virtualcalls.getMethodCount(cd);
-	    numclasses++;
 	}
-	MethodDescriptor[][] virtualtable=new MethodDescriptor[numclasses][maxcount];
+	MethodDescriptor[][] virtualtable=new MethodDescriptor[state.numClasses()][maxcount];
 
 	/* Fill in virtual table */
 	classit=state.getClassSymbolTable().getDescriptorsIterator();
@@ -127,7 +130,7 @@ public class BuildCode {
 	}
 	outvirtual.print("void * virtualtable[]={");
 	boolean needcomma=false;
-	for(int i=0;i<numclasses;i++) {
+	for(int i=0;i<state.numClasses();i++) {
 	    for(int j=0;j<maxcount;j++) {
 		if (needcomma)
 		    outvirtual.print(", ");
@@ -159,6 +162,24 @@ public class BuildCode {
 	}
     }
 
+    private void generateSizeArray(PrintWriter outclassdefs) {
+	outclassdefs.print("int classsize[]={");
+	Iterator it=state.getClassSymbolTable().getDescriptorsIterator();
+	ClassDescriptor[] cdarray=new ClassDescriptor[state.numClasses()];
+	while(it.hasNext()) {
+	    ClassDescriptor cd=(ClassDescriptor)it.next();
+	    cdarray[cd.getId()]=cd;
+	}
+	boolean needcomma=false;
+	for(int i=0;i<state.numClasses();i++) {
+	    if (needcomma)
+		outclassdefs.print(", ");
+	    outclassdefs.print("sizeof(struct "+cdarray[i].getSafeSymbol()+")");	    
+	    needcomma=true;
+	}
+	outclassdefs.println("};");
+    }
+
     private void generateTempStructs(FlatMethod fm) {
 	MethodDescriptor md=fm.getMethod();
 	ParamsObject objectparams=new ParamsObject(md,tag++);
diff --git a/Robust/src/IR/State.java b/Robust/src/IR/State.java
index 83ec6f46..93b5af68 100644
--- a/Robust/src/IR/State.java
+++ b/Robust/src/IR/State.java
@@ -22,6 +22,8 @@ public class State {
     public Set parsetrees;
     public Hashtable treemethodmap;
     public Hashtable flatmethodmap;
+    private int numclasses=0;
+
 
     public static TypeDescriptor getTypeDescriptor(int t) {
 	TypeDescriptor td=new TypeDescriptor(t);
@@ -37,6 +39,11 @@ public class State {
 	if (classes.contains(tdn.getSymbol()))
 	    throw new Error("Class "+tdn.getSymbol()+" defined twice");
 	classes.add(tdn);
+	numclasses++;
+    }
+
+    public int numClasses() {
+	return numclasses;
     }
 
     public BlockNode getMethodBody(MethodDescriptor md) {
diff --git a/Robust/src/Runtime/runtime.c b/Robust/src/Runtime/runtime.c
index 6dca89fd..05d94789 100644
--- a/Robust/src/Runtime/runtime.c
+++ b/Robust/src/Runtime/runtime.c
@@ -1,5 +1,6 @@
 #include "runtime.h"
 #include "structdefs.h"
+extern int classsize[];
 
 int ___Object______hashcode____(struct ___Object___ * ___this___) {
   return (int) ___this___;
@@ -8,3 +9,9 @@ int ___Object______hashcode____(struct ___Object___ * ___this___) {
 void ___System______printInt____I(int x) {
   printf("%d\n",x);
 }
+
+void * allocate_new(int type) {
+  void * v=calloc(1,classsize[type]);
+  *((int *)v)=type;
+  return v;
+}
diff --git a/Robust/src/f.test b/Robust/src/f.test
index 70b58959..7b2b6fc4 100644
--- a/Robust/src/f.test
+++ b/Robust/src/f.test
@@ -1,6 +1,10 @@
 public class Test {
-
+    public Test() {
+      ;
+    }
+    int a;
     public static void main() {
+	Test t=new Test();
 	for(int i=3;i<100000;i++) {
 		boolean flag=true;
 		for(int j=2;flag&&j<i;j++) {
-- 
2.34.1