spieleprogrammierer.de - Forum und Wiki zur Spieleprogrammierung und Spieleentwicklung

C-/C++-Quelltext

#include "../common/cpu_bitmap.h"
#include "../common/cpu_anim.h"
#include "../common/book.h"
#include "math.h"
#include "stdio.h"
const int DIM = 500;


struct DataBlock {
    unsigned char *dev_bitmap;
    CPUAnimBitmap *bitmap;
};
__device__ float cosinus( float y1) {
    return cos(y1 / 2000);
}

__device__ float sinus( float x1) {
    return sin(x1 / 500);
}

__device__ float rechnung1( float z1) {
    z1 = 1400 - z1;
    return (150 - z1 / 10) / 150;
}

__device__ float rechnung2() {
    return 255 / 2;
}

__device__ float formel1(float zahl1, float sin1, float cos1, float zahl2 ) {
    return  (zahl1 * (sin1 * cos1) * zahl2);
}



__global__ void kernel( unsigned char *ptr, int ticks, float d_wave[DIM][DIM]) {
    
    int x = threadIdx.x + blockIdx.x * blockDim.x;
    int y = threadIdx.y + blockIdx.y * blockDim.y;
    float x1 = threadIdx.x + blockIdx.x * blockDim.x + 530;
    float y1 = threadIdx.y + blockIdx.y * blockDim.y;
    float time = ticks;
    int offset = x + y * blockDim.x * gridDim.x; 
    __syncthreads(); 

    float sin1 = sinus(x1);
    float cos1 = cosinus(y1);
    float zahl1 = rechnung1(time);
    float zahl2 = rechnung2();
    __syncthreads();

    d_wave[x][y] = formel1(zahl1, sin1, cos1, zahl2);
    ptr[offset*4 + 0] = 0;
    ptr[offset*4 + 1] = 0;
    ptr[offset*4 + 2] = 255 * d_wave[x][y];
    ptr[offset*4 + 3] = 255;

}

void generate_frame( DataBlock *d, int ticks) {
    dim3 blocks(DIM,DIM);
    float wave[DIM][DIM];
    kernel<<<blocks,1>>>( d->dev_bitmap, ticks, wave);

    cudaMemcpy( d->bitmap->get_ptr(),
                d->dev_bitmap,
                d->bitmap->image_size(),
                cudaMemcpyDeviceToHost);
}


void cleanup( DataBlock *d) {
    cudaFree( d->dev_bitmap );
}

int main( void ) {
    DataBlock data;
    CPUAnimBitmap bitmap( DIM, DIM, &data );
    data.bitmap = &bitmap;
    cudaMalloc( (void**) &data.dev_bitmap, bitmap.image_size() );
 
    bitmap.anim_and_exit( (void (*) (void*,int))generate_frame, (void (*)(void*))cleanup );
    cudaFree( &data.dev_bitmap);


}