python loop & more doc

1 files changed, 85 insertions, 0 deletions

diff --git a/src/chill.cc b/src/chill.cc
new file mode 100644
index 0000000..6ca0c4c
--- /dev/null
+++ b/src/chill.cc
@@ -0,0 +1,85 @@
+#include "chilldebug.h"
+
+#include <csignal>
+#include <cstdio>
+#include <cstdlib>
+#include <cstring>
+
+
+#include "loop.hh"
+#include <omega.h>
+#include "ir_code.hh"
+#include "ir_rose.hh"
+
+#include "chillmodule.hh" // Python wrapper functions for CHiLL
+
+//---
+// CHiLL globals
+//---
+Loop *myloop = NULL;
+IR_Code *ir_code = NULL;
+bool is_interactive = false;
+
+std::vector<IR_Control *> ir_controls;
+std::vector<int> loops;
+
+//---
+// CHiLL program main
+// Initialize state and run script or interactive mode
+//---
+int main( int argc, char* argv[] )
+{
+  DEBUG_PRINT("%s  main()\n", argv[0]);
+  if (argc > 2) {
+    fprintf(stderr, "Usage: %s [script_file]\n", argv[0]);
+    exit(-1);
+  }
+  
+  // Create PYTHON interpreter
+  /* Pass argv[0] to the Python interpreter */
+  Py_SetProgramName(argv[0]);
+  
+  /* Initialize the Python interpreter.  Required. */
+  Py_Initialize();
+  
+  /* Add a static module */
+  initchill();
+  
+  if (argc == 2) {
+    FILE* f = fopen(argv[1], "r");
+    if(!f){
+      printf("can't open script file \"%s\"\n", argv[1]);
+      exit(-1);
+    }
+    PyRun_SimpleFile(f, argv[1]);
+    fclose(f);
+  }
+  if (argc == 1) {
+    //---
+    // Run a CHiLL interpreter
+    //---
+    printf("CHiLL v" CHILL_BUILD_VERSION " (built on " CHILL_BUILD_DATE ")\n");
+    printf("Copyright (C) 2008 University of Southern California\n");
+    printf("Copyright (C) 2009-2012 University of Utah\n");
+    fflush(stdout);
+    is_interactive=true;
+    PyRun_InteractiveLoop(stdin,"-");
+    printf("CHiLL ending...\n");
+    fflush(stdout);
+  }
+
+  // TODO Codegen is the major part that prevent CHiLL to be a pure Python library
+  //    Other than exporting a class of course
+  if (ir_code != NULL && myloop != NULL && myloop->stmt.size() != 0 && !myloop->stmt[0].xform.is_null()) {
+    int lnum_start;
+    int lnum_end;
+    lnum_start = get_loop_num_start();
+    lnum_end = get_loop_num_end();
+    DEBUG_PRINT("calling ROSE code gen?    loop num %d\n", lnum);
+    finalize_loop(lnum_start, lnum_end);
+    ((IR_roseCode*)(ir_code))->finalizeRose();
+    delete ir_code;
+  }
+  Py_Finalize();
+  return 0;
+}