ImCreate_cube.c

/*
 * Example code to write image in shared memory
 *
 * compile with:
 * gcc ImCreate_cube.c ImageStreamIO.c -o ImCreate_cube -lm -lpthread
 * gcc ImCreate_cube.c ImageStreamIO.c -DHAVE_CUDA -o ImCreate_test -lm -lpthread -I/opt/cuda/include -L/opt/cuda/lib64 -lcudart
 *
 * Required files in compilation directory :
 * ImCreate_cube.c   : source code (this file)
 * ImageStreamIO.c   : ImageStreamIO source code
 * ImageStreamIO.h   : ImageCreate function prototypes
 * ImageStruct.h     : Image structure definition
 *
 * EXECUTION:
 * ./ImCreate_cube
 * (no argument)
 *
 * Creates a circular buffer imtest00 in shared memory
 * Updates the image every ~ 10ms, forever...
 * A square is rotating around the center of the image
 *
 */




#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <time.h>
#include <string.h>
#include "ImageStruct.h"
#include "ImageStreamIO.h"




int main()
{
	IMAGE imarray;    // pointer to array of images
	long naxis;        // number of axis
	uint8_t atype;     // data type
	uint32_t *imsize;  // image size
	int shared;        // 1 if image in shared memory
	int NBkw;          // number of keywords supported


	// image will be a cube
	naxis = 3;

	// image size will be 512 x 512
	imsize = (uint32_t *) malloc(sizeof(uint32_t)*naxis);
	imsize[0] = 512;
	imsize[1] = 512;
	imsize[2] = 10;

	// image will be float type
	// see file ImageStruct.h for list of supported types
	atype = _DATATYPE_FLOAT;

	// image will be in shared memory
	shared = 1;

	// allocate space for 10 keywords
	NBkw = 3;

	// create an image in shared memory
	ImageStreamIO_createIm_gpu(&imarray, "imtest00", naxis, imsize, atype, -1, shared, 10, NBkw, 2);

    strncpy(imarray.kw[0].name, "symcode", KEYWORD_MAX_STRING-1);
    imarray.kw[0].type = 'L';
    imarray.kw[0].value.numl = 5;
    strncpy(imarray.kw[0].comment, "symcode value", KEYWORD_MAX_COMMENT-1);

    strncpy(imarray.kw[1].name, "exposure", KEYWORD_MAX_STRING-1);
    imarray.kw[1].type = 'D';
    imarray.kw[1].value.numf = 8000.;
    strncpy(imarray.kw[1].comment, "in us, exposure value", KEYWORD_MAX_COMMENT-1);

    strncpy(imarray.kw[2].name, "source", KEYWORD_MAX_STRING-1);
    imarray.kw[2].type = 'S';
    strncpy(imarray.kw[2].value.valstr, "ImCreate_cube", KEYWORD_MAX_STRING-1);
    strncpy(imarray.kw[2].comment, "source value", KEYWORD_MAX_COMMENT-1);


	free(imsize);


	float angle;
	float r;
	float r1;
	long ii, jj;
	float x, y, x0, y0, xc, yc;
	// float squarerad=20;
	long dtus = 100000; // update every 1ms
	float dangle = 0.02;

	int s;
	int semval;
	unsigned int index;
	float *current_image;

	// writes a square in image
	// square location rotates around center
	angle = 0.0;
	r = 100.0;
	x0 = 0.5*imarray.md->size[0];
	y0 = 0.5*imarray.md->size[1];
	while (1)
	{
		// disk location
		xc = x0 + r*cos(angle);
		yc = y0 + r*sin(angle);


		imarray.md->write = 1; // set this flag to 1 when writing data

		index = imarray.md->cnt1 +1;
		if(index == imarray.md->size[2])
			index=0;

		current_image = imarray.array.F + index * imarray.md->size[0] * imarray.md->size[1];
       // printf("%lu %lu %lu %x\r", imarray.md->cnt0, imarray.md->cnt1, index,
		// 				current_image);
       // printf("%d, %x \n", index * imarray.md->size[0] * imarray.md->size[1], current_image);
		for(ii=0; ii<imarray.md->size[0]; ii++)
			for(jj=0; jj<imarray.md->size[1]; jj++)
			{
				x = 1.0*ii;
				y = 1.0*jj;
				float dx = x-xc;
				float dy = y-yc;
				current_image[ii*imarray.md->size[1]+jj] = cos(0.03*dx)*cos(0.03*dy)*exp(-1.0e-4*(dx*dx+dy*dy));

				//if( (x-xc<squarerad) && (x-xc>-squarerad) && (y-yc<squarerad) && (y-yc>-squarerad))
				//	imarray.array.F[jj*imarray.md->size[0]+ii] = 1.0;
				//else
				//	imarray.array.F[jj*imarray.md->size[0]+ii] = 0.0;
			}
		imarray.md->cnt1 = index;
		imarray.md->cnt0++;
		clock_gettime(CLOCK_ISIO, &imarray.md[0].lastaccesstime);

		// POST ALL SEMAPHORES
		ImageStreamIO_sempost(&imarray, -1);

		imarray.md->write = 0; // Done writing data

		usleep(dtus);
		angle += dangle;
		if(angle > 2.0*3.141592)
			angle -= 2.0*3.141592;
		//printf("Wrote square at position xc = %16f  yc = %16f\n", xc, yc);
		//fflush(stdout);
	}

	return 0;
}