OpenVFDService.c

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <fcntl.h>
#include <errno.h>
#include <time.h>
#include <sys/ioctl.h>
#include <pthread.h>
#include <stdint.h>
#include <signal.h>
#include "driver/openvfd_drv.h"

#define UNUSED(x)	(void*)(x)
#define DRV_NAME	"/dev/" DEV_NAME
#define PIPE_PATH	"/tmp/" DEV_NAME "_service"

void select_display_type(void);
bool set_display_type(int new_display_type);
const char *get_user_string(int argc, char *argv[]);
const char *get_secondary_user_string(int argc, char *argv[]);
bool is_verbose(int argc, char *argv[]);
bool is_demo_mode(int argc, char *argv[]);
bool is_test_mode(int argc, char *argv[]);
bool is_12h_mode(int argc, char *argv[]);
int get_cmd_display_type(int argc, char *argv[]);
int get_cmd_chars_order(int argc, char *argv[], u_int8 chars[], const int sz);
bool print_usage(int argc, char *argv[]);

struct sync_data {
	bool isActive;
	pthread_mutex_t mutex;
	pthread_cond_t cond;
	struct timespec abs_time;
	bool useBuffer;
	union {
		struct vfd_display_data display_data;
		char buffer[sizeof(struct vfd_display_data)];
	};
};

struct display_setup {
	bool is_demo;
	bool is_12h;
	const char *user_string;
	const char *secondary_user_string;
};

typedef struct _DotLedBitMap {
	uint8_t on;
	uint8_t bitmap;
}DotLedBitMap;

#define LED_MASK_VOID	0x00
static DotLedBitMap dotLeds[LED_DOT_MAX] = {
	{0, 0x01},
	{0, 0x02},
	{0, 0x04},
	{0, 0x08},
	{0, 0x10},
	{0, 0x20},
	{0, 0x40},
};

#define LEDCODES_LEN	(sizeof(LED_decode_tab1)/sizeof(LED_decode_tab1[0]))
static const led_bitmap *ledCodes = LED_decode_tab1;
static struct vfd_display display_type;

int openvfd_fd;
bool verbose = false;
#define VERBOSE_PRINTF(s) if (verbose) printf(s)

uint8_t char_to_mask(uint8_t ch)
{
	unsigned int index = 0;
	if (display_type.controller > CONTROLLER_7S_MAX)
		return ch;
	else {
		for (index = 0; index < LEDCODES_LEN; index++) {
			if (ledCodes[index].character == ch) {
				return ledCodes[index].bitmap;
			}
		}
	}

	return LED_MASK_VOID;
}

void mdelay(int n)
{
	unsigned long msec=(n);
	while (msec--)
		usleep(1000);
}

struct sync_data sync_data;

void led_display_loop(const struct display_setup *setup)
{
	static struct vfd_display_data data = { 0 };
	bool use_user_string = false;
	int ret = -1;

	time_t now;
	struct tm *timenow;

	memset(&data, 0, sizeof(data));

	if (setup->user_string) {
		use_user_string = true;
		data.mode = DISPLAY_MODE_TITLE;
		snprintf(data.string_main, sizeof(data.string_secondary), setup->user_string);
		if (setup->secondary_user_string)
			snprintf(data.string_secondary, sizeof(data.string_secondary), setup->secondary_user_string);
	}

	while(sync_data.isActive) {
		if (!pthread_mutex_lock(&sync_data.mutex)) {
			ret = pthread_cond_timedwait(&sync_data.cond, &sync_data.mutex, &sync_data.abs_time);
			if (!ret || ret == ETIMEDOUT) {
				clock_gettime(CLOCK_REALTIME, &sync_data.abs_time);
				sync_data.abs_time.tv_nsec += (long)5E8;
				if (sync_data.abs_time.tv_nsec >= (long)1E9) {
					sync_data.abs_time.tv_nsec -= (long)1E9;
					sync_data.abs_time.tv_sec++;
				}

				select_display_type();
				if (sync_data.useBuffer && (sync_data.display_data.mode == DISPLAY_MODE_CLOCK ||
						sync_data.display_data.mode == DISPLAY_MODE_DATE)) {
					use_user_string = false;
					sync_data.useBuffer = false;
					sync_data.display_data.colon_on = data.colon_on;
					data = sync_data.display_data;
				}

				if (sync_data.useBuffer) {
					use_user_string = false;
					data = sync_data.display_data;
				} else {
					// Get current time
					time(&now);
					timenow = localtime(&now);

					if (setup->is_demo) {
						data.mode = 1 + timenow->tm_sec / 12;
						data.temperature = timenow->tm_hour + timenow->tm_min + timenow->tm_sec;
						data.channel_data.channel = (u_int16)10*(timenow->tm_hour + timenow->tm_min + timenow->tm_sec);
						data.channel_data.channel_count = (u_int16)86400;
						data.time_date.hours = ((timenow->tm_sec >= 24) && (timenow->tm_sec < 30)) ? 0 : (timenow->tm_hour == 0) ? 24 : timenow->tm_hour;
						data.time_date.minutes = timenow->tm_min;
						data.time_date.seconds = timenow->tm_sec;
						data.time_date.day_of_week = timenow->tm_wday;
						data.time_date.day = timenow->tm_mday;
						data.time_date.month = timenow->tm_mon;
						data.time_date.year = timenow->tm_year + 1900;
						data.time_secondary.hours = timenow->tm_hour;
						data.time_secondary.minutes = timenow->tm_min;
						data.time_secondary.seconds = timenow->tm_sec;
						data.colon_on = !data.colon_on;
						// Really long movie title.
						snprintf(data.string_main, sizeof(data.string_secondary), "The Saga of the Viking Women and their Voyage to the Waters of the Great Sea Serpent");
						snprintf(data.string_secondary, sizeof(data.string_secondary), "Now playing:");
					} else if (!use_user_string) {
						if (data.mode != DISPLAY_MODE_DATE)
							data.mode = DISPLAY_MODE_CLOCK;
						if (setup->is_12h) {
							if (timenow->tm_hour == 0)
								data.time_date.hours = 12;
							else if (timenow->tm_hour > 12)
								data.time_date.hours = timenow->tm_hour - 12;
							else
								data.time_date.hours = timenow->tm_hour;
						} else {
							data.time_date.hours = timenow->tm_hour;
						}
						data.time_date.minutes = timenow->tm_min;
						data.time_date.seconds = timenow->tm_sec;
						data.time_date.day_of_week = timenow->tm_wday;
						data.time_date.day = timenow->tm_mday;
						data.time_date.month = timenow->tm_mon;
						data.time_date.year = timenow->tm_year + 1900;
						data.colon_on = !data.colon_on;
					}
				}
				ret = write(openvfd_fd,&data,sizeof(data));
			}
			pthread_mutex_unlock(&sync_data.mutex);
		} else {
			mdelay(500);
		}
	}
}

void led_test_codes()
{
	unsigned short write_buffer[7];
	unsigned char val = ':';
	unsigned int i = 0;

	// Test chart, sequence of numbers.
	for(i = 0; i < LEDCODES_LEN; i++) {
		val = ~val;
		write_buffer[1] = char_to_mask(ledCodes[i].character);
		write_buffer[2] = char_to_mask(ledCodes[i].character);
		write_buffer[3] = char_to_mask(ledCodes[i].character);
		write_buffer[4] = char_to_mask(ledCodes[i].character);
		write_buffer[0] = val;

		write(openvfd_fd,write_buffer,sizeof(write_buffer[0])*5);
		mdelay(500);
	}

	// Test bit sequence.
	for(i = 0; i < 10; i++) {
		val = ~val;
		write_buffer[1] = char_to_mask(1);
		write_buffer[2] = char_to_mask(2);
		write_buffer[3] = char_to_mask(3);
		write_buffer[4] = char_to_mask(4);
		write_buffer[0] = val;

		write(openvfd_fd,write_buffer,sizeof(write_buffer[0])*5);
		mdelay(500);
	}

	// Test sequence 2
	write_buffer[0] = 0;
	for(i = 0; i < LED_DOT_MAX; i++){
		val = ~val;
		write_buffer[1] = char_to_mask(5);
		write_buffer[2] = char_to_mask(6);
		write_buffer[3] = char_to_mask(7);
		write_buffer[4] = char_to_mask(8);
		write_buffer[0] &= dotLeds[i%LED_DOT_MAX].bitmap;

		write(openvfd_fd,write_buffer,sizeof(write_buffer[0])*5);
		mdelay(500);
	}

	// Test sequence 3
	write_buffer[0] = 0;
	for(i = 0; i < LED_DOT_MAX; i++){
		val = ~val;
		write_buffer[1] = char_to_mask(6);
		write_buffer[2] = char_to_mask(7);
		write_buffer[3] = char_to_mask(8);
		write_buffer[4] = char_to_mask(9);
		write_buffer[0] |= dotLeds[i%LED_DOT_MAX].bitmap;

		write(openvfd_fd,write_buffer,sizeof(write_buffer[0])*5);
		mdelay(500);
	}
}

void led_test_loop(bool cycle_display_types)
{
	int current_type = DISPLAY_TYPE_5D_7S_NORMAL;
	int transposed = 0;
	const pid_t pid = getpid();
	printf("Initializing...\n");
	if (!cycle_display_types)
		printf("Process ID = %d\n", pid);
	while (1) {
		int i;
		const int len = 7;
		unsigned short wb[7];
		const size_t sz = sizeof(wb[0])*len;

		if (cycle_display_types) {
			printf("Process ID = %d\n", pid);
			++current_type;
			current_type %= DISPLAY_TYPE_MAX;
			if (!current_type)
				transposed = (~transposed & DISPLAY_FLAG_TRANSPOSED_INT);
			printf("Set display type to 0x%08X\n", current_type | transposed);
			set_display_type(current_type | transposed);
			select_display_type();
		}

		// Light up all sections and cycle
		// through display brightness levels.
		memset(wb, 0xFF, sz);
		write(openvfd_fd,wb,sz);
		for (i = FD628_Brightness_1; i <= FD628_Brightness_8; i++) {
			ioctl(openvfd_fd, VFD_IOC_SBRIGHT, &i);
			mdelay(1000);
		}

		// Clear display for a second.
		memset(wb, 0x00, sz);
		write(openvfd_fd,wb,sz);
		mdelay(1000);

		// Run original test codes.
		led_test_codes();

		// Cycle through fully lit characters.
		for (i = 0; i < len; i++) {
			memset(wb, 0x00, sz);
			wb[i] = 0xFF;
			write(openvfd_fd,wb,sz);
			mdelay(1000);
		}

		// Cycle through bits in each character.
		for (i = 0; i < len; i++) {
			memset(wb, (1 << i), sz);
			write(openvfd_fd,wb,sz);
			mdelay(1000);
		}
	}
}

void *display_thread_handler(void *arg)
{
	struct display_setup *setup = (struct display_setup*)arg;
	led_display_loop(setup);
	pthread_exit(NULL);
}

void *display_test_thread_handler(void *arg)
{
	bool cycle_display_types = *(bool*)arg;
	led_test_loop(cycle_display_types);
	pthread_exit(NULL);
}

void *named_pipe_thread_handler(void *arg)
{
	int file;
	char buf[1024];
	int ret = 0, i;
	unsigned char skipSignal;

	unlink(PIPE_PATH);
	if ((mkfifo(PIPE_PATH, 0666)) != 0) {
		printf("Unable to create a fifo; errno=%d\n",errno);
		pthread_exit(NULL);                    /* Print error message and return */
	}

	while (sync_data.isActive) {
		file = open(PIPE_PATH, O_RDONLY);
		ret = read(file, buf, sizeof(buf));
		close(file);
		buf[ret] = '\0';
		if (verbose) {
			printf("ret = %d, %s\n", ret, buf);
			for (i = 0; i < ret; i++)
				printf("0x%02X, ", buf[i]);
			printf("\n");
		}
		if (ret > 0 && !pthread_mutex_lock(&sync_data.mutex)) {
			skipSignal = 0;
			if (ret == sizeof(sync_data.display_data)) {
				VERBOSE_PRINTF("Write display data\n");
				memcpy(&sync_data.display_data, buf, sizeof(sync_data.display_data));
				sync_data.useBuffer = true;
			} else {
				VERBOSE_PRINTF("Write unknown data\n");
				switch ((unsigned char)buf[0]) {
				case 0:
				default:
					VERBOSE_PRINTF("case 0, default\n");
					sync_data.useBuffer = true;
					sync_data.display_data.mode = DISPLAY_MODE_CLOCK;
					break;
				case 1:
					// Refresh display. Will signal the led_loop to update display.
					break;
				case 2:
					if (ret >= 3 && buf[1] == DISPLAY_MODE_DATE)
					{
						if (sync_data.display_data.mode == DISPLAY_MODE_DATE)
							skipSignal = 1;
						else
							sync_data.display_data.mode = DISPLAY_MODE_DATE;
						sync_data.display_data.time_secondary._reserved = buf[2];
						sync_data.useBuffer = true;
					}
					break;
				}
			}
			if (!skipSignal)
				pthread_cond_signal(&sync_data.cond);
			pthread_mutex_unlock(&sync_data.mutex);
		}
	}

	unlink(PIPE_PATH);
	pthread_exit(NULL);
}

void select_display_type()
{
	if (!ioctl(openvfd_fd, VFD_IOC_GDISPLAY_TYPE, &display_type)) {
		switch(display_type.type) {
			case DISPLAY_TYPE_5D_7S_T95:
				ledCodes = LED_decode_tab1;
				break;
			case DISPLAY_TYPE_5D_7S_G9SX:
				ledCodes = LED_decode_tab3;
				break;
			case DISPLAY_TYPE_4D_7S_FREESATGTC:
				ledCodes = LED_decode_tab4;
				break;
			case DISPLAY_TYPE_5D_7S_TAP1:
				ledCodes = LED_decode_tab5;
				break;
			default:
				ledCodes = LED_decode_tab2;
				break;
		}
	} else {
		memset(&display_type, 0, sizeof(display_type));
		perror("Failed to read display type, using default.");
	}
}

bool set_display_type(int new_display_type)
{
	long ret = ioctl(openvfd_fd, VFD_IOC_SDISPLAY_TYPE, &new_display_type);
	if (ret) {
		printf("Failed setting a new display type.\n");
		if (ret == ERANGE)
			printf("Unsupported display type. (out of range)\n");
	}

	return ret == 0;
}

void handle_signal(int signal)
{
	int file;
	sync_data.isActive = false;
	file = open(PIPE_PATH, O_WRONLY);
	write(file, "\1", 1);
	close(file);
}

int main(int argc, char *argv[])
{
	u_int8 char_indexes[7];
	int ret, type, char_order_count;
	bool test_mode = false;
	bool cycle_display_types = true;
	pthread_t disp_id, npipe_id = 0;

	if (print_usage(argc, argv))
		return 0;
	openvfd_fd = open(DRV_NAME, O_RDWR);
	if (openvfd_fd < 0) {
		perror("Open device failed.\n");
		exit(1);
	}

	verbose = is_verbose(argc, argv);
	char_order_count = get_cmd_chars_order(argc, argv, char_indexes, (int)sizeof(char_indexes));
	if (char_order_count)
		if (ioctl(openvfd_fd, VFD_IOC_SCHARS_ORDER, char_indexes))
			printf("Error setting new character order.\n");

	type = get_cmd_display_type(argc, argv);
	if (type >= 0)
		cycle_display_types = !set_display_type(type);
	select_display_type();

	test_mode = is_test_mode(argc, argv);
	if (test_mode)
		ret = pthread_create(&disp_id, NULL, display_test_thread_handler, &cycle_display_types);
	else {
		struct display_setup setup = { 0 };
		struct sigaction sig_handler = {.sa_handler=handle_signal};
		memset(&sync_data, 0, sizeof(struct sync_data));
		sync_data.isActive = true;
		sigaction(SIGTERM, &sig_handler, 0);
		sigaction(SIGINT, &sig_handler, 0);
		setup.is_demo = is_demo_mode(argc, argv);
		setup.is_12h = is_12h_mode(argc, argv);
		setup.user_string = get_user_string(argc, argv);
		if (setup.user_string)
			setup.secondary_user_string = get_secondary_user_string(argc, argv);
		ret = pthread_create(&disp_id, NULL, display_thread_handler, &setup);
		if (ret == 0)
			ret = pthread_create(&npipe_id, NULL, named_pipe_thread_handler, NULL);
	}
	if(ret != 0) {
		perror("Create disp_id or npipe_id thread error\n");
		return ret;
	}

	if (npipe_id)
		pthread_join(npipe_id, NULL);
	pthread_join(disp_id, NULL);
	close(openvfd_fd);
	return 0;
}

bool is_cmd_option(int argc, char *argv[], const char *str)
{
	bool ret = false;
	int i;
	for (i = 1; i < argc; i++) {
		if (!strcmp(argv[i], str)) {
			ret = true;
			break;
		}
	}

	return ret;
}

const char *get_user_string(int argc, char *argv[])
{
	int i;
	for (i = 1; i < argc; i++) {
		if ((!strcmp(argv[i], "-s") || !strcmp(argv[i], "--string")) && ++i < argc) {
			return argv[i];
		}
	}
	return NULL;
}

const char *get_secondary_user_string(int argc, char *argv[])
{
	int i;
	for (i = 1; i < argc; i++) {
		if ((!strcmp(argv[i], "-ss") || !strcmp(argv[i], "--secondary-string")) && ++i < argc) {
			return argv[i];
		}
	}
	return NULL;
}

bool is_verbose(int argc, char *argv[])
{
	return is_cmd_option(argc, argv, "-v");
}

bool is_demo_mode(int argc, char *argv[])
{
	return is_cmd_option(argc, argv, "-dm");
}

bool is_test_mode(int argc, char *argv[])
{
	return is_cmd_option(argc, argv, "-t");
}

bool is_12h_mode(int argc, char *argv[])
{
	return is_cmd_option(argc, argv, "-12h");
}

int get_cmd_display_type(int argc, char *argv[])
{
	int ret = -1, i;
	for (i = 1; i < argc; i++) {
		if (!strcmp(argv[i], "-dt")) {
			if (argc >= (i + 2)) {
				long temp = -1;
				char *start, *end;
				start = !strncmp(argv[i+1], "0x", 2) ? argv[i+1] + 2 : argv[i+1];
				temp = strtol(start, &end, 0x10);
				if (end == start || *end != '\0' || errno == ERANGE) {
					printf("Error parsing display type index.\n");
				} else {
					ret = (int)temp;
					printf("Display type 0x%08X\n", ret);
				}
			} else {
				printf("Error parsing display type index, missing argument.\n");
			}
			break;
		}
	}

	return ret;
}

int get_cmd_chars_order(int argc, char *argv[], u_int8 chars[], const int sz)
{
	int ret = 0, i, j;
	for (i = 0; i < sz; i++)
		chars[i] = i;
	for (i = 1; i < argc; i++) {
		if (!strcmp(argv[i], "-co")) {
			for (i++, j = 0; i < argc && j < sz; i++) {
				long temp = -1;
				char *end;
				temp = strtol(argv[i], &end, 10);
				if (end == argv[i] || *end != '\0' || errno == ERANGE)
					break;
				else if (temp >= 0 && temp < sz)
					chars[j++] = temp;
			}

			ret = j;
			break;
		}
	}

	if (ret) {
		printf("Got %d char indexes.\n", ret);
		for (i = 0; i < ret; i++) {
			printf("index[%d] = %d\n", i, chars[i]);
		}
	}

	return ret;
}

bool print_usage(int argc, char *argv[])
{
	bool ret = false;
	int i;
	for (i = 1; i < argc; i++) {
		if (!strcmp(argv[i], "-h") || !strcmp(argv[i], "--help")) {
			ret = true;
			printf("\nUsage: OpenVFDService [-t] [-dt TYPE] [-h]\n\n");
			printf("\t-s USER_STRING\tRun OpenVFDService in custom string mode.\n\t\t\tDisplays the USER_STRING on the screen.");
			printf("\t-ss SECONDARY_USER_STRING\tDisplay a smaller secondary string\n\t\t\tin addtion to USER_STRING.");
			printf("\t-t\t\tRun OpenVFDService in display test mode.\n");
			printf("\t-dm\t\tRun OpenVFDService in display demo mode.\n");
			printf("\t-dt N\t\tSpecifies which display type to use.\n");
			printf("\t-co N...\t< D HH:MM > Order of display chars.\n\t\t\tValid values are 0 - 6.\n\t\t\t(D=dots, represented by a single char)\n");
			printf("\t-h\t\tThis text.\n\n");
		}
	}

	return ret;
}