/*
 * Copyright 1993-2010 NVIDIA Corporation.  All rights reserved.
 *
 * NVIDIA Corporation and its licensors retain all intellectual property and 
 * proprietary rights in and to this software and related documentation. 
 * Any use, reproduction, disclosure, or distribution of this software 
 * and related documentation without an express license agreement from
 * NVIDIA Corporation is strictly prohibited.
 *
 * Please refer to the applicable NVIDIA end user license agreement (EULA) 
 * associated with this source code for terms and conditions that govern 
 * your use of this NVIDIA software.
 * 
 */


#include "cuda.h"
#include "../BitmapAnim/src/BitmapAnim.h"

#define DIM 512
#define PI 3.1415926535897932f

__global__ void kernel( unsigned char *ptr, int ticks ) {
    int x = threadIdx.x + blockIdx.x * blockDim.x;
    int y = threadIdx.y + blockIdx.y * blockDim.y;
    int offset = x + y * blockDim.x * gridDim.x;

    float fx = x - DIM/2;
    float fy = y - DIM/2;
    float d = sqrtf( fx * fx + fy * fy );
    unsigned char grey = (unsigned char)(128.0f + 127.0f *
                                         cos(d/10.0f - ticks/100.0f) /
                                         (d/10.0f + 1.0f));    
    ptr[offset*4 + 0] = grey;
    ptr[offset*4 + 1] = grey;
    ptr[offset*4 + 2] = grey;
    ptr[offset*4 + 3] = 255;
}

void generateFrame( BitmapAnim* animObj, int time ) {
    dim3    blocks(DIM/16,DIM/16);
    dim3    threads(16,16);
    kernel<<<blocks,threads>>>( (unsigned char*)animObj->extraData, time );

    cudaMemcpy( animObj->pixels,
                animObj->extraData,
                animObj->image_size(),
                cudaMemcpyDeviceToHost );
}

int main( void ) {

    BitmapAnim bitmap( DIM, DIM);
	unsigned char* dev_bitmap;

    cudaMalloc( (void**)&dev_bitmap, bitmap.image_size() );
	bitmap.extraData = dev_bitmap;

	bitmap.genFrame = generateFrame;

    bitmap.startAnim();
}