Code so far

2009-07-24 08:35:21 +00:00
commit da046a9a90
21 changed files with 1908 additions and 0 deletions
--- a/pycdas.cpp
+++ b/pycdas.cpp
@@ -0,0 +1,116 @@
+#include <cstdio>
+#include <cstdarg>
+#include "module.h"
+#include "bytecode.h"
+#include "numeric.h"
+
+static void ivprintf(int indent, const char* fmt, va_list varargs)
+{
+    for (int i=0; i<indent; i++)
+        printf("    ");
+    vprintf(fmt, varargs);
+}
+
+static void iprintf(int indent, const char* fmt, ...)
+{
+    va_list varargs;
+    va_start(varargs, fmt);
+    ivprintf(indent, fmt, varargs);
+    va_end(varargs);
+}
+
+void output_object(PycRef<PycObject> obj, PycModule* mod, int indent)
+{
+    switch (obj->type()) {
+    case PycObject::TYPE_CODE:
+        {
+            PycRef<PycCode> codeObj = obj.cast<PycCode>();
+            iprintf(indent, "[Code]\n");
+            iprintf(indent + 1, "File Name: %s\n", codeObj->fileName()->value());
+            iprintf(indent + 1, "Object Name: %s\n", codeObj->name()->value());
+            iprintf(indent + 1, "Arg Count: %d\n", codeObj->argCount());
+            iprintf(indent + 1, "Locals: %d\n", codeObj->numLocals());
+            iprintf(indent + 1, "Stack Size: %d\n", codeObj->stackSize());
+            iprintf(indent + 1, "Flags: 0x%08X\n", codeObj->flags());
+
+            iprintf(indent + 1, "[Names]\n");
+            PycTuple::value_t names = codeObj->names()->values();
+            for (PycTuple::value_t::iterator i = names.begin(); i != names.end(); i++)
+                output_object(*i, mod, indent + 2);
+
+            iprintf(indent + 1, "[Var Names]\n");
+            names = codeObj->varNames()->values();
+            for (PycTuple::value_t::iterator i = names.begin(); i != names.end(); i++)
+                output_object(*i, mod, indent + 2);
+
+            iprintf(indent + 1, "[Free Vars]\n");
+            names = codeObj->freeVars()->values();
+            for (PycTuple::value_t::iterator i = names.begin(); i != names.end(); i++)
+                output_object(*i, mod, indent + 2);
+
+            iprintf(indent + 1, "[Cell Vars]\n");
+            names = codeObj->cellVars()->values();
+            for (PycTuple::value_t::iterator i = names.begin(); i != names.end(); i++)
+                output_object(*i, mod, indent + 2);
+
+            iprintf(indent + 1, "[Constants]\n");
+            PycTuple::value_t consts = codeObj->consts()->values();
+            for (PycTuple::value_t::iterator i = consts.begin(); i != consts.end(); i++)
+                output_object(*i, mod, indent + 2);
+
+            iprintf(indent + 1, "[Disassembly]\n");
+            bc_disasm(codeObj, mod, indent + 2);
+        }
+        break;
+    case PycObject::TYPE_STRING:
+    case PycObject::TYPE_STRINGREF:
+    case PycObject::TYPE_INTERNED:
+        iprintf(indent, "\"%s\"\n", obj.cast<PycString>()->value());
+        break;
+    case PycObject::TYPE_TUPLE:
+        {
+            iprintf(indent, "(\n");
+            PycTuple::value_t values = obj.cast<PycTuple>()->values();
+            for (PycTuple::value_t::iterator i = values.begin(); i != values.end(); i++)
+                output_object(*i, mod, indent + 1);
+            iprintf(indent, ")\n");
+        }
+        break;
+    case PycObject::TYPE_LIST:
+        {
+            iprintf(indent, "[\n");
+            PycList::value_t values = obj.cast<PycList>()->values();
+            for (PycList::value_t::iterator i = values.begin(); i != values.end(); i++)
+                output_object(*i, mod, indent + 1);
+            iprintf(indent, "]\n");
+        }
+        break;
+    case PycObject::TYPE_NONE:
+        iprintf(indent, "None\n");
+        break;
+    case PycObject::TYPE_INT:
+        iprintf(indent, "%d\n", obj.cast<PycInt>()->value());
+        break;
+    case PycObject::TYPE_FLOAT:
+        iprintf(indent, "%s\n", obj.cast<PycFloat>()->value());
+        break;
+    default:
+        iprintf(indent, "<TYPE: %d>\n", obj->type());
+    }
+}
+
+int main(int argc, char* argv[])
+{
+    if (argc < 2) {
+        fprintf(stderr, "No input file specified\n");
+        return 1;
+    }
+
+    PycModule mod;
+    mod.loadFromFile(argv[1]);
+    printf("%s (Python %d.%d%s)\n", argv[1], mod.majorVer(), mod.minorVer(),
+           mod.isUnicode() ? " -U" : "");
+    output_object(mod.code().cast<PycObject>(), &mod, 0);
+
+    return 0;
+}