mass_properties.py 文件源码

def __init__(self, queue, grid_size, program_buffer, block_sizes):
        self.queue = queue
        self.grid_size = grid_size
        self.program_buffer = program_buffer
        self.block_sizes = block_sizes

        self.integral_one = 0
        self.integral_x = 0
        self.integral_y = 0
        self.integral_z = 0
        self.integral_xx = 0
        self.integral_yy = 0
        self.integral_zz = 0
        self.integral_xy = 0
        self.integral_xz = 0
        self.integral_yz = 0

        self.index_sums = cl_util.Buffer(queue, numpy.uint32, 10, pyopencl.mem_flags.READ_WRITE)

        self.counter = cl_util.Buffer(queue, numpy.uint32, 1, pyopencl.mem_flags.READ_WRITE)
        self.list = cl_util.Buffer(queue, cl_util.Buffer.quad_dtype(numpy.uint8), grid_size * grid_size * grid_size, pyopencl.mem_flags.WRITE_ONLY)

        self.box_corner = None
        self.level = None