Python get_array_names Examples

Example #1

File: integrator_cython_helper.py Project: pushkargodbole/GPUPySPH

    def get_array_declarations(self, method):
        arrays = set()
        for dest in self.object.steppers:
            s, d = get_array_names(self.get_args(dest, method))
            arrays.update(s | d)

        decl = []
        for arr in sorted(arrays):
            decl.append('cdef double* %s'%arr)
        return '\n'.join(decl)

Example #2

File: integrator_cython_helper.py Project: mahgadalla/pysph-1

    def get_array_declarations(self, method):
        arrays = set()
        for dest in self.object.steppers:
            s, d = get_array_names(self.get_args(dest, method))
            self._check_arrays_for_properties(dest, s | d)
            arrays.update(s | d)

        known_types = self.acceleration_eval_helper.known_types
        decl = []
        for arr in sorted(arrays):
            decl.append('cdef {type} {arr}'.format(type=known_types[arr].type,
                                                   arr=arr))

        return '\n'.join(decl)

Example #3

 def get_array_setup(self, dest, method):
     s, d = get_array_names(self.get_args(dest, method))
     lines = ['%s = dst.%s.data'%(n, n[2:]) for n in sorted(s|d)]
     return '\n'.join(lines)

Example #4

File: integrator_cython_helper.py Project: pushkargodbole/GPUPySPH

 def get_array_setup(self, dest, method):
     s, d = get_array_names(self.get_args(dest, method))
     lines = ['%s = dst.%s.data'%(n, n[2:]) for n in s|d]
     return '\n'.join(lines)