1 files changed, 85 insertions, 0 deletions
diff --git a/weave/examples/functional.py b/weave/examples/functional.py
new file mode 100644
index 000000000..cd907aec4
--- /dev/null
+++ b/weave/examples/functional.py
@@ -0,0 +1,85 @@
+import sys
+sys.path.insert(0,'..')
+import inline_tools
+from types import *
+def c_list_map(func,seq):
+    """ Uses CXX C code to implement a simple map-like function.
+        It does not provide any error checking.
+    """
+    assert(type(func) in [FunctionType,MethodType,type(len)])
+    code = """
+           #line 12 "functional.py"
+           Py::Tuple args(1);    
+           Py::List result(seq.length());
+           PyObject* this_result = NULL;
+           for(int i = 0; i < seq.length();i++)
+           {
+              args[0] = seq[i];
+              this_result = PyEval_CallObject(func,args.ptr());
+              result[i] = Py::Object(this_result);
+           }           
+           return_val = Py::new_reference_to(result);
+           """   
+    return inline_tools.inline(code,['func','seq'])
+
+def c_list_map2(func,seq):
+    """ Uses Python API more than CXX to implement a simple map-like function.
+        It does not provide any error checking.
+    """
+    assert(type(func) in [FunctionType,MethodType,type(len)])
+    code = """
+           #line 32 "functional.py"
+           Py::Tuple args(1);    
+           Py::List result(seq.length());
+           PyObject* item = NULL;
+           PyObject* this_result = NULL;
+           for(int i = 0; i < seq.length();i++)
+           {
+              item = PyList_GET_ITEM(seq.ptr(),i);
+              PyTuple_SetItem(args.ptr(),0,item);
+              this_result = PyEval_CallObject(func,args.ptr());
+              PyList_SetItem(result.ptr(),i,this_result);              
+           }           
+           return_val = Py::new_reference_to(result);
+           """   
+    return inline_tools.inline(code,['func','seq'])
+    
+def main():
+    seq = ['aa','bbb','cccc']
+    print 'desired:', map(len,seq)
+    print 'actual:', c_list_map(len,seq)
+    print 'actual2:', c_list_map2(len,seq)
+
+def time_it(m,n):
+    import time
+    seq = ['aadasdf'] * n
+    t1 = time.time()
+    for i in range(m):
+        result = map(len,seq)
+    t2 = time.time()
+    py = t2 - t1
+    print 'python speed:', py
+    
+    #load cache
+    result = c_list_map(len,seq)
+    t1 = time.time()
+    for i in range(m):
+        result = c_list_map(len,seq)
+    t2 = time.time()
+    c = t2-t1
+    print 'c speed:', c
+    print 'speed up:', py / c
+
+    #load cache
+    result = c_list_map2(len,seq)
+    t1 = time.time()
+    for i in range(m):
+        result = c_list_map2(len,seq)
+    t2 = time.time()
+    c = t2-t1
+    print 'c speed:', c
+    print 'speed up:', py / c
+
+if __name__ == "__main__":
+    main()
+    time_it(100,1000)
+\ No newline at end of file