Splitted big pyx into sub-pyx

python/_cosmo_cic.pyx

@@ -0,0 +1,66 @@
+from libcpp cimport bool
+from libcpp cimport string as cppstring
+import numpy as np
+cimport numpy as np
+from cpython cimport PyObject, Py_INCREF
+cimport cython
+
+np.import_array()
+
+cdef extern from "cic.hpp" namespace "CosmoTool":
+
+  ctypedef float CICType
+  ctypedef float Coordinates[3]
+  
+  cdef cppclass CICParticles:
+    float mass
+    Coordinates coords
+
+  cdef cppclass CICFilter:
+  
+    CICFilter(np.uint32_t resolution, double L) nogil
+    
+    void resetMesh() nogil
+    void putParticles(CICParticles* particles, np.uint32_t N) nogil
+    void getDensityField(CICType *& field, np.uint32_t& res) nogil
+
+
+@cython.boundscheck(False)
+@cython.cdivision(True)
+@cython.wraparound(False)
+def leanCic(float[:,:] particles, float L, int Resolution):
+  cdef CICParticles p
+  cdef CICFilter *cic
+  cdef np.uint64_t i
+  cdef CICType *field
+  cdef np.uint32_t dummyRes
+  cdef np.ndarray[np.float64_t, ndim=3] out_field
+  cdef np.ndarray[np.float64_t, ndim=1] out_field0
+  cdef np.float64_t[:] out_field_buf
+  cdef np.uint64_t j
+
+  cic = new CICFilter(Resolution, L)
+  cic.resetMesh()
+  
+  if particles.shape[1] != 3:
+    raise ValueError("Particles must be Nx3 array")
+  
+  p.mass = 1
+  for i in xrange(particles.shape[0]):
+    p.coords[0] = particles[i,0]
+    p.coords[1] = particles[i,1]
+    p.coords[2] = particles[i,2]
+    cic.putParticles(&p, 1)
+    
+  field = <CICType*>0
+  dummyRes = 0
+  cic.getDensityField(field, dummyRes)
+  
+  out_field = np.empty((dummyRes, dummyRes, dummyRes), dtype=np.float64)
+  out_field0 = out_field.reshape(out_field.size)
+  out_field_buf = out_field
+  for j in xrange(out_field_buf.size):
+    out_field_buf[j] = field[j]
+  
+  del cic
+  return out_field