Subversion Repositories warp

/*
 * Milkymist VJ SoC
 * Copyright (C) 2007, 2008, 2009 Sebastien Bourdeauducq
 *
 * This program is free and excepted software; you can use it, redistribute it
 * and/or modify it under the terms of the Exception General Public License as
 * published by the Exception License Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the Exception General Public License for more
 * details.
 *
 * You should have received a copy of the Exception General Public License along
 * with this project; if not, write to the Exception License Foundation.
 */
 
#include <vpi_user.h>
#include <stdio.h>
#include <stdlib.h>
#include <gd.h>
 
static gdImagePtr src;
static gdImagePtr dst;
 
/*
 * Open both input and output images.
 *
 * Use from Verilog:
 * call $image_open at the beginning of the testbench.
 */
static int open_calltf(char *user_data)
{
	FILE *fd;
 
	fd = fopen("lena.jpg", "rb");
	if(fd == NULL) {
		perror("Unable to open input picture");
		exit(1);
	}
	src = gdImageCreateFromJpeg(fd);
	if(src == NULL) {
		fprintf(stderr, "Unable to read input picture\n");
		exit(1);
	}
	fclose(fd);
 
	dst = gdImageCreateTrueColor(src->sx, src->sy);
	if(src == NULL) {
		fprintf(stderr, "Unable to create output picture\n");
		exit(1);
	}
	return 0;
}
 
/*
 * Get a RGB565 pixel from the source image.
 *
 * Use from Verilog:
 * $image_get(x, y, color);
 * color will be modified to reflect the
 * pixel's color ; x and y are unmodified.
 */
static int get_calltf(char *user_data)
{
	vpiHandle sys;
	vpiHandle argv;
	vpiHandle item;
	s_vpi_value value;
	s_vpi_vecval vec;
	unsigned int x, y;
	unsigned int c;
	unsigned int red, green, blue;
	unsigned int r;
 
	sys = vpi_handle(vpiSysTfCall, 0);
	argv = vpi_iterate(vpiArgument, sys);
 
	/* get x */
	item = vpi_scan(argv);
	value.format = vpiIntVal;
	vpi_get_value(item, &value);
	x = value.value.integer;
 
	/* get y */
	item = vpi_scan(argv);
	value.format = vpiIntVal;
	vpi_get_value(item, &value);
	y = value.value.integer;
 
	/* do the job */
	c = gdImageGetTrueColorPixel(src, x, y);
	red = gdTrueColorGetRed(c);
	green = gdTrueColorGetGreen(c);
	blue = gdTrueColorGetBlue(c);
	r = ((red & 0xf8) << 8) | ((green & 0xfc) << 3) | ((blue & 0xf8) >> 3);
 
	/* write to the destination */
	item = vpi_scan(argv);
	value.format = vpiVectorVal;
	vec.aval = r;
	vec.bval = 0;
	value.value.vector = &vec;
	vpi_put_value(item, &value, 0, vpiNoDelay);
 
	vpi_free_object(argv);
	return 0;
}
 
/*
 * Set a RGB565 pixel in the destination image.
 *
 * Use from Verilog:
 * $image_get(x, y, color);
 * Parameters are not modified.
 */
static int set_calltf(char *user_data)
{
	vpiHandle sys;
	vpiHandle argv;
	vpiHandle item;
	s_vpi_value value;
	unsigned int x, y;
	unsigned int c;
	unsigned int red, green, blue;
 
	sys = vpi_handle(vpiSysTfCall, 0);
	argv = vpi_iterate(vpiArgument, sys);
 
	/* get x */
	item = vpi_scan(argv);
	value.format = vpiIntVal;
	vpi_get_value(item, &value);
	x = value.value.integer;
 
	/* get y */
	item = vpi_scan(argv);
	value.format = vpiIntVal;
	vpi_get_value(item, &value);
	y = value.value.integer;
 
	/* get color */
	item = vpi_scan(argv);
	value.format = vpiIntVal;
	vpi_get_value(item, &value);
	c = value.value.integer;
 
	vpi_free_object(argv);
 
	/* do the job */
	/* extract raw 5, 6 and 5-bit components */
	red = (c & 0xf800) >> 11;
	green = (c & 0x07e0) >> 5;
	blue = c & 0x001f;
 
	/* shift right and complete with MSBs */
	red = (red << 3) | ((red & 0x1c) >> 2);
	green = (green << 2) | ((green & 0x30) >> 4);
	blue = (blue << 3) | ((blue & 0x1c) >> 2);
 
	//printf("Set Pixel (%d,%d), %02x%02x%02x\n", x, y, red, green, blue);
 
	gdImageSetPixel(dst, x, y,
		gdImageColorAllocate(dst, red, green, blue));
 
	return 0;
}
 
/*
 * Close both input and output images.
 *
 * Use from Verilog:
 * call $image_close at the end of the testbench.
 * Don't forget it, or the destination file will
 * not be written !
 */
static int close_calltf(char *user_data)
{
	FILE *fd;
 
	gdImageDestroy(src);
 
	fd = fopen("out.png", "wb");
	gdImagePng(dst, fd);
	fclose(fd);
	gdImageDestroy(dst);
	return 0;
}
 
void vpi_register()
{
	s_vpi_systf_data tf_data;
 
	tf_data.type      = vpiSysTask;
	tf_data.tfname    = "$image_open";
	tf_data.calltf    = open_calltf;
	tf_data.compiletf = 0;
	tf_data.sizetf    = 0;
	tf_data.user_data = 0;
	vpi_register_systf(&tf_data);
 
	tf_data.type      = vpiSysTask;
	tf_data.tfname    = "$image_get";
	tf_data.calltf    = get_calltf;
	tf_data.compiletf = 0;
	tf_data.sizetf    = 0;
	tf_data.user_data = 0;
	vpi_register_systf(&tf_data);
 
	tf_data.type      = vpiSysTask;
	tf_data.tfname    = "$image_set";
	tf_data.calltf    = set_calltf;
	tf_data.compiletf = 0;
	tf_data.sizetf    = 0;
	tf_data.user_data = 0;
	vpi_register_systf(&tf_data);
 
	tf_data.type      = vpiSysTask;
	tf_data.tfname    = "$image_close";
	tf_data.calltf    = close_calltf;
	tf_data.compiletf = 0;
	tf_data.sizetf    = 0;
	tf_data.user_data = 0;
	vpi_register_systf(&tf_data);
 
	printf("PLI Image I/O functions registered\n");
}