/*
 * Copyright (c) 2014 Eric Sandeen <sandeen@sandeen.net>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software 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
 *  GNU Library General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

/*
 * tt-status: Show status of a Triangle Tube Solo Prestige boiler via ModBus
 *
 * Usually pointed at a RS-485 serial port device, but may also query through
 * a ModBus/TCP gateway such as mbusd (http://http://mbus.sourceforge.net/)
 */

#include <errno.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <time.h>
#include <modbus/modbus.h>

#define CHECK_BIT(var,pos) ((var) & (1<<(pos)))

void usage(void)
{
	printf("Usage: tt-status [-h] [-d] [-S slave] [-c|-t] -s serial port][-i ip addr [-p port]]\n\n");
	printf("-h\tShow this help\n");
	printf("-d\tEnable debug, default 0\n");
	printf("-S\tModbus slave id, default 1\n");
	printf("-c\tExport in CSV\n");
	printf("-t\tExport in thingspeak\n");
	printf("-s\tSerial Port Device for ModBus/RTU\n");
	printf("-i\tIP Address for ModBus/TCP\n");
	printf("-p\tTCP Port for ModBus/TCP (optional, default 502)\n");
	exit(1);
}

float c_to_f(float c)
{
	return ((c * 9)/5 + 32);
}

struct status_bits {
	int	bit;
	char	*desc;
};

struct status_bits status[] = {
	{ 0, "PC Manual Mode" },
	{ 1, "DHW Mode" },
	{ 2, "CH Mode" },
	{ 3, "Freeze Protection Mode" },
	{ 4, "Flame Present" },
	{ 5, "CH(1) Pump" },
	{ 6, "DHW Pump" },
	{ 7, "System / CH2 Pump" }
};

uint16_t status_regs[1];/* Holds results of register reads */
uint16_t temp_regs[9];	/* Holds results of register reads */
uint16_t setpt_regs[2];	/* Holds results of register reads */

void csv_print(void)
{
	int i;

	printf("%d", time(NULL));
	printf(",%d", status_regs[0]);

	for (i = 0; i < 9; i++)
		printf(",%d", temp_regs[i]);
	
	for (i = 0; i < 2; i++)
		printf(",%d", setpt_regs[i]);

	printf("\n");
}

void thingspeak_print(void)
{
	/* Setpoint */
	if (temp_regs[8] != 0x8000)
		printf("field1=%.1f&", c_to_f(temp_regs[8]));
	else
		printf("field1=%d&", 0);
	
	/* firing rate */
	printf("field2=%d&", temp_regs[7]);
	/* supply temp */
	printf("field3=%.1f&", c_to_f(temp_regs[0]/10));
	/* return temp */
	printf("field4=%.1f&", c_to_f(temp_regs[1]));
	/* outdoor temp */
	printf("field5=%.1f&", c_to_f((int16_t)temp_regs[4]));
	/* DHW temp */
	printf("field6=%.1f&", c_to_f((int16_t)temp_regs[2]));
	/* Status */
	printf("field7=%d\n", status_regs[0]);
}

void pretty_print(void)
{
	int i;

	printf("Status:\n");
	if (status_regs[0] == 0)
		printf(" Standby\n");

	for (i = 0; i < 7; i++) {
		if (CHECK_BIT(status_regs[0], i))
			printf(" %s\n", status[i].desc);
	}

	/* Supply temp: 0.1 degree C, 16 bits */
	printf("Supply temp:\t\t%3.0f °F\n", c_to_f(temp_regs[0]/10));

	/* Return temp: degrees C, 8 bits */
	printf("Return temp:\t\t%3.0f °F\n", c_to_f(temp_regs[1]));

	/* DHW storage temp */
	printf("DHW Storage temp:\t%3.0f °F\n", c_to_f(temp_regs[2]));

	/* Flue temp: degrees C, 8 bits */
	printf("Flue temp:\t\t%3.0f °F\n", c_to_f(temp_regs[3]));

	/* Outdoor temp: degrees C, 8 bits */
	printf("Outdoor temp:\t\t%3.0f °F\n", c_to_f((int16_t)temp_regs[4]));

	/* Future use, 5 */

	/* Flame Ionization: μA, 8 bits */
	printf("Flame Ionization:\t%3d μA\n", temp_regs[6]);

	/* Firing rate: % 8 bits */
	printf("Firing rate:\t\t%3d %\n", temp_regs[7]);

	/* Boiler setpoint: degrees C, 8 bits, only if firing */
	if (temp_regs[8] != 0x8000)
		printf("Boiler Setpoint:\t\t%3.0f °F\n", c_to_f(temp_regs[8]));

	/* CH1 Maximum Setpoint C, 8 bits */
	printf("CH1 Maximum Setpoint:\t%3.0f °F\n", c_to_f(setpt_regs[0]));

	/* DHW setpoint: degrees C, 8 bits, only if set */
	if (setpt_regs[1] != 0x8000)
		printf("DHW Setpoint:\t\t%3.0f °F\n", c_to_f(setpt_regs[1]));
}

typedef enum { PRETTY, CSV, THINGSPEAK } printtype;

int modbus_read_boiler(modbus_t *mb)
{
	/* Read 1 register from the address 0 for status bitfield */
	if (modbus_read_input_registers(mb, 0, 1, status_regs) != 1) {
		printf("Error: Modbus read of 1 byte at addr 0 failed\n");
		return 1;
	}

	/* Read 9 registers from the address 0x300 */
	if (modbus_read_input_registers(mb, 0x300, 9, temp_regs) != 9) {
		printf("Error: Modbus read of 9 bytes at addr 0x300 failed\n");
		return 1;
	}

	/* Read 2 registers from the address 0x500 */
	if (modbus_read_registers(mb, 0x500, 2, setpt_regs) != 2) {
		printf("Error: Modbus read of 2 bytes at addr 0x500 failed\n");
		return 1;
	}

	return 0;
}

int main(int argc, char **argv)
{
	int c;
	int err = 1;
	int slave = 1;		/* default modbus slave ID */
	int debug = 0;		/* enable debug */
	int port = 502;		/* default ModBus/TCP port */
	char ipaddr[16] = "";	/* ModBus/TCP ip address */
	char serport[32] = "";	/* ModBus/RTU serial port */
	modbus_t *mb;		/* ModBus context */
	uint16_t regs[8];	/* Holds results of register reads */

	printtype output = PRETTY;

	while ((c = getopt(argc, argv, "cdths:i:p:S")) != -1) {
		switch (c) {
		case 'c':
			output = CSV;
			break;
		case 't':
			output = THINGSPEAK;
			break;
		case 'h':
			usage();
			break;
		case 'd':
			debug++;
			break;
		case 's':
			strncpy(serport, optarg, sizeof(serport));
			serport[31] = '\0';
			break;
		case 'i':
			strncpy(ipaddr, optarg, sizeof(ipaddr));
			ipaddr[15] = '\0';
			break;
		case 'p':
			port = atoi(optarg);
			break;
		case 'S':
			slave = atoi(optarg);
			break;
		default:
			usage();
		}
	}

	if (!ipaddr[0] && !serport[0]) {
		printf("Error: Must specify either ip addresss or serial port\n\n");
		usage();
	}
	if (ipaddr[0] && serport[0]) {
		printf("Error: Must specify only one of ip addresss or serial port\n\n");
		usage();
	}

	if (ipaddr[0])
		mb = modbus_new_tcp(ipaddr, port);
	else
		mb = modbus_new_rtu(serport, 38400, 'N', 8, 1);

	if (!mb) {
		perror("Error: modbus_new failed");
		goto out;
	}

	if (debug)
		modbus_set_debug(mb,TRUE);

	if (modbus_set_slave(mb, slave)) {
		perror("Error: modbus_set_slave failed");
		goto out;
	}

	if (modbus_connect(mb)) {
		perror("Error: modbus_connect failed");
		goto out;
	}

	if ((err = modbus_read_boiler(mb)))
		goto out;

	if (output == PRETTY)
		pretty_print();
	else if (output == CSV)
		csv_print();
	else if (output == THINGSPEAK)
		thingspeak_print();

	err = 0;
out:
	modbus_close(mb);
	modbus_free(mb);
	return err;
}