/*

* Ecowatt

*

* get ecowatt device's data

* Copyright (C) 2012 Alexandre Coffignal <alexandre.github@gmail.com>

*

* 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 3 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 General Public License for more details.

* You should have received a copy of the GNU General Public License

* along with this program. If not, see <http://www.gnu.org/licenses/>.

*

*/

#include <stdio.h>

#include <stdlib.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <termios.h>

#include <time.h>

#include <unistd.h>

#include <signal.h>

#include <string.h>

#include <errno.h>

#include <sys/time.h>

#include <time.h>

#include <mysql.h>

#define MAX_BUF 1024

#define MAX_DEV 30

#define MAX_CHAINE 255

#define MIN_DETECT 7

#define MAX_VERBOSE_LEVEL 2

#define CVS_FILE_NAME "data.csv"

#define NB_TRY 10

#define NB_JOUR 256

#define NB_HEURE 24

#define HEADER_DAY 4

float afKWatt[NB_JOUR][NB_HEURE];

static int _iVerbose=0;

static char _acDevice[MAX_DEV];

static int _iFunction=-1;

static char _bEmon =0;

static char _bSql=0;

static time_t _tTime;

struct termios term, old;

int (*pvPrint1) (char * acFormat, ...);

int (*pvPrint2) (char * acFormat, ...);

char _acServer[MAX_CHAINE] = {0};

char _acUser[MAX_CHAINE] = {0};

char _acPassword[MAX_CHAINE]= {0};

char _acDatabase[MAX_CHAINE] = {0};

char _acFeed[MAX_CHAINE]= {0};

char _acApiKey[MAX_CHAINE]= {0};

int iQuiet (char * acFormat, ...)

{

return 0;

}

static void vSetVerbose(int iLevel)

{

_iVerbose=1;

if(iLevel>0)

pvPrint1=(void*)printf;

if(iLevel>1)

pvPrint2=(void*)printf;

}

static void _vUsage(char * acName)

{

printf("usage : %s [-v 1] [-d device] [-h]\n",acName);

printf("-h, --help : aide\n");

printf("-v n, --verbose : verbose n est le niveau de trace 2, 1 ou 0\n");

printf("-d \"/dev/ttyUSB0\", --device : device\n");

printf("-e <api_key>,--emoncms3 : envoi donnees au format emoncms3\n");

printf("-s <server> <user> <pass> <database> <feed> ,--sql : envoi donnees vers sql\n");

printf("-s 192.168.0.14 <user> <pass> emoncms feed_37 ,--sql : envoi donnees vers sql\n");

}

int iGetData(int fd,const unsigned char * aucTx,int iTxLen, unsigned char pucRx[MAX_BUF])

{

struct sigaction action;

int i=0,ret;

int iOut=0;

unsigned int uiRead=0;

unsigned char buf[MAX_BUF];

pvPrint2("=> ");

ret=write(fd,aucTx,iTxLen);

if(ret<0)

{

perror("write");

uiRead=ret;

}

else

{

for(i=0;i<ret;i++)

{

pvPrint2("0x%02X ",aucTx[i]);

}

pvPrint2("\n");

do

{

ret=read(fd,(void*)buf,MAX_BUF);

if(ret>0)

{

pvPrint2("<= ");

for(i=0;i<ret;i++)

{

pvPrint2("0x%02X ",(unsigned char)buf[i]);

pucRx[i+uiRead]=buf[i];

}

uiRead+=ret;

pvPrint2(" (%d)\n",uiRead);

}

else

{

if(ret==0)

{

//timeout

iOut=1;

}

else

{

//erreur

perror("read");

uiRead=ret;

printf("ret =%d\n",ret);

}

}

}while(iOut==0);

}

return uiRead;

}

//function 1 : trame ping

int iTestCx(int fd)

{

const unsigned char aucTestCx[]={0x45,0x4C,0x01,0x00};

unsigned char aucRecu[MAX_BUF];

int i=0,ret=-1;

bzero(aucRecu, sizeof(unsigned char)*MAX_BUF);

if(iGetData(fd,aucTestCx,sizeof(aucTestCx),aucRecu)==4)

{

if(aucRecu[3]==0xA5)

{

ret=0;

}

}

return ret;

}

//function 2 : not used here

int iFonction2(int fd,int iNb)

{

unsigned char aucFunction[]={0x45,0x4C,0x02,0x00};

unsigned char aucRecu[MAX_BUF];

int i=0,ret=-1,irecu;

pvPrint1("Fonction 2 arg %d\n",iNb);

aucFunction[3]=iNb;

bzero(aucRecu, sizeof(unsigned char)*MAX_BUF);

if(irecu = iGetData(fd,aucFunction,sizeof(aucFunction),aucRecu)==10)

{

printf("arg = %d ",iNb);

for(i=0;i<6;i++)

{

printf("0x%02x ",aucRecu[i + HEADER_DAY]);

}

printf("\n");

ret=0;

}

pvPrint1("%d\n",irecu);

return ret;

}

float fGetKWatt(unsigned char ucMSB,unsigned char ucLsb,unsigned char ucExp)

{

unsigned long ulExp;

float fKWatt;

switch(ucExp)

{

case 248: ulExp = 8388608;break;

case 249: ulExp = 4194304;break;

case 250: ulExp = 2097152;break;

case 251: ulExp = 1048576;break;

case 252: ulExp = 524288;break;

case 253: ulExp = 262144;break;

case 254: ulExp = 131072;break;

case 255: ulExp = 65536;break;

case 0: ulExp = 32768;break;

case 1: ulExp = 16384;break;

case 2: ulExp = 8192;break;

case 3: ulExp = 4096;break;

case 4: ulExp = 2048;break;

case 5: ulExp = 1024;break;

case 6: ulExp = 512;break;

case 7: ulExp = 256;break;

case 8: ulExp = 128;break;

case 9: ulExp = 64;break;

case 10: ulExp = 32;break;

case 11: ulExp = 16;break;

case 12: ulExp = 8;break;

case 13: ulExp = 4;break;

case 14: ulExp = 2;break;

case 15: ulExp = 1;break;

default :

printf("Exposant non correct (0x%02x)\n",ucExp);

//return -1;

break;

}

fKWatt = (float)((int)ucMSB * 256 + ucLsb) / ulExp;

return fKWatt;

}

//function 3 : get ecowatt data hour per hour

int iGetDay(int fd,int iDay)

{

unsigned char aucGetDay[]={0x45,0x4C,0x03,0x00};

unsigned char aucRecu[MAX_BUF];

int i=0,ret=-1,iNb=0;

aucGetDay[3]=iDay;

bzero(aucRecu, sizeof(unsigned char)*MAX_BUF);

if((iNb=iGetData(fd,aucGetDay,sizeof(aucGetDay),aucRecu))==76)

{

for(i=0;i<NB_HEURE;i++)

{

unsigned char ucMSB = aucRecu[i*3 + HEADER_DAY];

unsigned char ucLsb = aucRecu[i*3 + HEADER_DAY +1 ];

unsigned char ucExp = aucRecu[i*3 + HEADER_DAY +2 ];

float fkWatt = fGetKWatt(ucMSB,ucLsb,ucExp);

pvPrint1("%x %x %x %f\n",ucMSB,ucLsb,ucExp,fkWatt);

afKWatt[iDay][i]=fkWatt;

}

ret=0;

}

else

{

pvPrint1("Probleme : Recu %d\n",iNb);

}

return ret;

}

//function 4 : stop ecowatt com

int iFonction4(int fd,int iNb)

{

unsigned char aucFunction[]={0x45,0x4C,0x04,0x00};

unsigned char aucRecu[MAX_BUF];

int i=0,ret=-1,irecu;

pvPrint1("Fonction 4 arg %d\n",iNb);

aucFunction[3]=iNb;

bzero(aucRecu, sizeof(unsigned char)*MAX_BUF);

irecu = iGetData(fd,aucFunction,sizeof(aucFunction),aucRecu);

printf("recu %d ",irecu);

{

printf("arg = %d ",iNb);

for(i=0;i<irecu;i++)

{

printf("0x%02x ",aucRecu[i + HEADER_DAY]);

}

printf("\n");

ret=0;

}

pvPrint1("%d\n",irecu);

return ret;

}

//fonction 5 : not used here

int iFonction5(int fd,int iNb)

{

unsigned char aucFunction[]={0x45,0x4C,0x05,0x00};

unsigned char aucRecu[MAX_BUF];

int i=0,ret=-1,irecu;

pvPrint1("Fonction 4 arg %d\n",iNb);

aucFunction[3]=iNb;

bzero(aucRecu, sizeof(unsigned char)*MAX_BUF);

irecu = iGetData(fd,aucFunction,sizeof(aucFunction),aucRecu);

if(irecu==7)

{

unsigned char ucMSB = aucRecu[HEADER_DAY];

unsigned char ucLsb = aucRecu[HEADER_DAY +1 ];

unsigned char ucExp = aucRecu[HEADER_DAY +2 ];

float fKWatt = fGetKWatt(ucMSB,ucLsb,ucExp);

pvPrint1("%x %x %x %f\n",ucMSB,ucLsb,ucExp,fKWatt);

for(i=0;i<irecu;i++)

{

printf("0x%02x ",aucRecu[i + HEADER_DAY]);

}

printf("\n");

ret=0;

}

pvPrint1("%d\n",irecu);

return ret;

}

//function 6 : week average

int iFonction6(int fd,int iNb)

{

unsigned char aucFunction[]={0x45,0x4C,0x06,0x00};

unsigned char aucRecu[MAX_BUF];

int i=0,ret=-1,irecu;

pvPrint1("Fonction 6 arg %d\n",iNb);

aucFunction[3]=iNb;

bzero(aucRecu, sizeof(unsigned char)*MAX_BUF);

irecu = iGetData(fd,aucFunction,sizeof(aucFunction),aucRecu);

if(irecu==7)

{

unsigned char ucMSB = aucRecu[HEADER_DAY];

unsigned char ucLsb = aucRecu[HEADER_DAY +1 ];

unsigned char ucExp = aucRecu[HEADER_DAY +2 ];

float fKWatt = fGetKWatt(ucMSB,ucLsb,ucExp);

pvPrint1("%x %x %x %f\n",ucMSB,ucLsb,ucExp,fKWatt);

for(i=0;i<irecu;i++)

{

printf("0x%02x ",aucRecu[i + HEADER_DAY]);

}

printf("\n");

ret=0;

}

pvPrint1("%d\n",irecu);

return ret;

}

//function 8 : get ecowatt time

int iGetDate(int fd)

{

unsigned char aucFunction[]={0x45,0x4C,0x08,0x00};

unsigned char aucRecu[MAX_BUF];

int i=0,ret=-1,irecu;

pvPrint1("Get EcoWatt Time\n");

bzero(aucRecu, sizeof(unsigned char)*MAX_BUF);

irecu = iGetData(fd,aucFunction,sizeof(aucFunction),aucRecu);

if(irecu==9)

{

struct tm t;

t.tm_year = aucRecu[HEADER_DAY] + 100;

t.tm_mon = aucRecu[HEADER_DAY+1] - 1;

t.tm_mday = aucRecu[HEADER_DAY+2];

t.tm_hour = 25;

t.tm_min = 0;

t.tm_sec = 0;

_tTime = mktime(&t);

printf("Date et heure: %s", ctime(&_tTime));

ret=0;

}

return ret;

}

//generic function used for debug

int iFonction(int fd,int iFonction,int iarg)

{

unsigned char aucFunction[]={0x45,0x4C,0x00,0x00};

unsigned char aucRecu[MAX_BUF];

int i=0,ret=-1,irecu;

pvPrint1("Fonction %d arg %d\n",iFonction,iarg);

aucFunction[2]=iFonction;

aucFunction[3]=iarg;

bzero(aucRecu, sizeof(unsigned char)*MAX_BUF);

irecu = iGetData(fd,aucFunction,sizeof(aucFunction),aucRecu);

printf("recu %d ",irecu);

{

printf("arg = %d ",iarg);

for(i=0;i<irecu;i++)

{

printf("0x%02x ",aucRecu[i]);

}

printf("\n");

ret=0;

}

pvPrint1("%d\n",irecu);

return ret;

}

void vStop(int fd)

{

if(fd)

{

pvPrint1("close comm\n");

close(fd);

}

}

/*

* init serial com over USB

* speed 38400 baud; line = 0;

* eof = ^A; min = 1; time = 5;

* ignbrk -brkint -icrnl -imaxbel

* -opost -onlcr

* -isig -icanon -iexten -echo -echoe -echok -echoctl -echoke

*/

void vInitCom(int fd)

{

/* Lecture des paramètres */

tcgetattr(fd, &term);

tcgetattr(fd, &old);

cfsetospeed(&term, B38400);

/* mode RAW, pas de mode canonique, pas d'echo */

term.c_iflag = IGNBRK;

term.c_lflag = 0;

term.c_oflag = 0;

term.c_cc[VMIN] = 0;

term.c_cc[VTIME] = 1;

/* Inhibe le controle de flux XON/XOFF */

term.c_iflag &= ~(IXON|IXOFF|IXANY);

/* 8 bits de données, pas de parité */

term.c_cflag &= ~(PARENB | CSIZE);

term.c_cflag |= CS8;

tcsetattr(fd, TCSANOW, &term);

}

//write data in cvs file

static void vSaveData(void)

{

FILE * fp;

int iDay,iHour;

struct termios term;

struct tm *tb;

time_t tTime=_tTime;

time_t intps;

char buf[255];

tTime=time(NULL);

tb=localtime(&tTime);

sprintf(buf,"%02d%02d%02d_%02d%d_%s",tb->tm_year-100,tb->tm_mon+1,tb->tm_mday,tb->tm_hour,tb->tm_min,CVS_FILE_NAME);

if ((fp = fopen(buf,"w")) == NULL)

{

fprintf(stderr,"Impossible d'ouvrir le fichier données en lecture\n");

}

else

{

fprintf(fp,"KWatt / jour\n");

fprintf(fp,";");

for(iDay=0;iDay<NB_JOUR;iDay++)

{

struct tm *tb;

tb = localtime(&tTime);

fprintf(fp,"%02d/%02d/%02d;", tb->tm_mday,tb->tm_mon+1,tb->tm_year-100);

tTime-=24*60*60;

}

fprintf(fp,"\n");

//save KWatt per hour

for(iHour=0;iHour<NB_HEURE;iHour++)

{

fprintf(fp,"%02d:00;",iHour);

for(iDay=0;iDay<NB_JOUR;iDay++)

{

fprintf(fp,"%f;", afKWatt[iDay][iHour]);

}

fprintf(fp,"\n");

}

//save KWatt per day

fprintf(fp,"\nKWatt / jour\n");

fprintf(fp,"Jour;Mesure\n");

tTime=_tTime;

float fToTalJour;

for(iDay=0;iDay<NB_JOUR;iDay++)

{

tb = localtime(&tTime);

fToTalJour=0;

for(iHour=0;iHour<NB_HEURE;iHour++)

{

fToTalJour+=afKWatt[iDay][iHour];

}

fprintf(fp,"%02d/%02d/%02d;%f\n", tb->tm_mday,tb->tm_mon+1,tb->tm_year-100,fToTalJour);

tTime-=24*60*60;

}

//save KWatt per month

tTime=_tTime;

float fToTalMois=0;

float fToTalAn;

int iMoisEnCours=0,iAnneeEnCours;

tb = localtime(&tTime);

iMoisEnCours=tb->tm_mon;

iAnneeEnCours=tb->tm_year;

fprintf(fp,"\nKWatt / mois\n");

fprintf(fp,"Mois;Mesure\n");

for(iDay=0;iDay<NB_JOUR;iDay++)

{

tb = localtime(&tTime);

if(iMoisEnCours!=tb->tm_mon)

{

fprintf(fp,"%02d/%02d;%f\n",iMoisEnCours+1,iAnneeEnCours-100,fToTalMois);

fToTalMois=0;

}

iMoisEnCours=tb->tm_mon;

iAnneeEnCours=tb->tm_year;

for(iHour=0;iHour<NB_HEURE;iHour++)

{

fToTalMois+=afKWatt[iDay][iHour];

fToTalAn+=afKWatt[iDay][iHour];

}

tTime-=24*60*60;

}

fprintf(fp,"\nKWatt total ;%f;\n",fToTalAn);

fprintf(fp,"KWatt / an (estimation);%f;\n",fToTalAn*365/255);

fclose(fp);

}

}

//write data on emoncms3 server

static void vSendDataEmoncs3(void)

{

int iDay,iHour;

struct termios term;

struct tm *tb;

time_t tTime=_tTime;

char buf[255];

//save KWatt per hour

tTime=_tTime-(NB_JOUR*24)*3600;

for(iDay=NB_JOUR-1;iDay>=0;iDay--)

{

for(iHour=0;iHour<NB_HEURE;iHour++)

{

char acCmd[255];

sprintf(acCmd,"wget \"http://192.168.0.14/emoncms3/emoncms3//api/post?apikey=%s&time=%ld&node=1000&json={P:%f}\" -qO - >/dev/null",_acApiKey,tTime, afKWatt[iDay][iHour]*1000);

system(acCmd);

printf("%ld;%f;%s",tTime,afKWatt[iDay][iHour]*1000,ctime(&tTime));

tTime=_tTime-(iDay*24+(NB_HEURE-iHour-1))*60*60;

}

}

}

#define MAX_REQ_SQL 255

//write data on emoncms3 server via mysql

static void vSendDataMysql(void)

{

MYSQL *conn;

MYSQL_RES *res;

MYSQL_ROW row;

int iTime=0;

time_t tTime=_tTime;

int iDay,iHour;

char acBuf[MAX_REQ_SQL];

conn = mysql_init(NULL);

/* Connect to database */

if (!mysql_real_connect(conn, _acServer,

_acUser, _acPassword, _acDatabase, 0, NULL, 0)) {

fprintf(stderr, "%s\n", mysql_error(conn));

return;

}

/* GET last entry */

sprintf(acBuf,"SELECT MAX(time) FROM `%s`",_acFeed);

if (mysql_query(conn, acBuf)) {

fprintf(stderr, "%s\n", mysql_error(conn));

return;

}

res = mysql_use_result(conn);

while ((row = mysql_fetch_row(res)) != NULL)

iTime=atoi(row[0]);

//save KWatt per hour

tTime=_tTime-(NB_JOUR*24)*3600;

for(iDay=NB_JOUR-1;iDay>=0;iDay--)

{

for(iHour=0;iHour<NB_HEURE;iHour++)

{

char acCmd[255];

float fWatt=afKWatt[iDay][iHour]*1000;

//update last entry

if((tTime==iTime)&&(afKWatt[iDay][iHour]!=0)&&(iDay*NB_HEURE+NB_HEURE-iHour!=6144))

{

printf("update %ld;%f;%s",tTime,fWatt,ctime(&tTime));

sprintf(acBuf,"UPDATE %s SET data = %f WHERE time=%ld",_acFeed ,fWatt,tTime);

if (mysql_query(conn, acBuf)) {

fprintf(stderr, "%s\n", mysql_error(conn));

return;

}

}

//insert new one

if((tTime>iTime)&&(afKWatt[iDay][iHour]!=0)&&(iDay*NB_HEURE+NB_HEURE-iHour!=6144))

{

printf("insert %ld;%f;%s",tTime,fWatt,ctime(&tTime));

sprintf(acBuf,"INSERT INTO %s (time,data) VALUES(%ld,%f)",_acFeed ,tTime,fWatt);

if (mysql_query(conn, acBuf)) {

fprintf(stderr, "%s\n", mysql_error(conn));

return;

}

}

tTime=_tTime-(iDay*24+(NB_HEURE-iHour-1))*60*60;

}

}

/* close connection */

mysql_free_result(res);

mysql_close(conn);

}

static int iRun()

{

int fd=0,i=0,j,ret=0,try;

fd=open(_acDevice,O_RDWR);

if(fd<=0)

{

perror("open");

return 0;

}

vInitCom(fd);

pvPrint1("Test du port comm\n");

if(iTestCx(fd))

{

printf("Ecowatt not found\n");

vStop(fd);

ret=-1;

}

else

{

if(_iFunction==-1)

{

pvPrint1("Ecowatt ready\n");

if(iGetDate(fd))

{

printf("Ecowatt Date not found\n");

ret=-1;

}

for(i=0;i<NB_JOUR;i++)

{

printf("%d pct\n\r",(i*100)/255);

for(try=0;try<NB_TRY;try++)

{

if(iGetDay(fd,i))

{

printf("Recuperation jour %d => ko\n",i);

}

else

{

pvPrint1("Recuperation jour %d => ok\n",i);

break;

}

}

if(try==NB_TRY)

{

return -1;

}

}

if(_bEmon)

vSendDataEmoncs3();

if(_bSql)

vSendDataMysql();

vSaveData();

}

else

{

for(i=0;i<255;i++)

{

iFonction(fd,_iFunction,i);

}

}

vStop(fd);

}

return ret;

}

int iGetArg(int argc, char *argv[])

{

int i;

int level;

_bEmon=0;

for(i=0;i<argc;i++)

{

//mode verbeux ou pas

if((!strcmp(argv[i],"-v"))||(!strcmp(argv[i],"--verbose")))

{

if(argc>=i+1)

{

level = atoi(argv[i+1]);

if((level<=MAX_VERBOSE_LEVEL) && (level>0))

{

i++;

}

else

{

level=1;

}

}

else

{

level=1;

}

printf("Verbose mode level %d\n",level);

vSetVerbose(level);

}

if((!strcmp(argv[i],"-h"))||(!strcmp(argv[i],"--help")))

{

_vUsage(argv[0]);

return -1;

}

if((!strcmp(argv[i],"-e"))||(!strcmp(argv[i],"--emoncms3")))

{

_bEmon=1;

if(argc>=i+1)

{

i++;

pvPrint1("ApiKey %s\n",argv[i]);

strncpy(_acApiKey,argv[i],MAX_CHAINE);

}

}

if((!strcmp(argv[i],"-d"))||(!strcmp(argv[i],"--device")))

{

if(argc>=i+1)

{

i++;

pvPrint1("Device %s\n",argv[i]);

strncpy(_acDevice,argv[i],MAX_DEV);

}

}

if((!strcmp(argv[i],"-f"))||(!strcmp(argv[i],"--function")))

{

if(argc>=i+1)

{

_iFunction = atoi(argv[i+1]);

i++;

}

else

{

_iFunction=-1;

}

printf("mode function %d\n",_iFunction);

}

if((!strcmp(argv[i],"-s"))||(!strcmp(argv[i],"--sql")))

{

_bSql=1;

if(argc>=i+5)

{

i++;

pvPrint1("Server %s\n",argv[i]);

strncpy(_acServer,argv[i],MAX_CHAINE);

i++;

pvPrint1("User %s\n",argv[i]);

strncpy(_acUser,argv[i],MAX_CHAINE);

i++;

pvPrint1("Password %s\n",argv[i]);

strncpy(_acPassword,argv[i],MAX_CHAINE);

i++;

pvPrint1("Database %s\n",argv[i]);

strncpy(_acDatabase,argv[i],MAX_CHAINE);

i++;

pvPrint1("Feed %s\n",argv[i]);

strncpy(_acFeed,argv[i],MAX_CHAINE);

}

}

}

if(_bSql==1)

{

if ((_acServer[0]==0)||(_acUser[0]==0)||(_acPassword[0]==0)||(_acDatabase[0]==0)||(_acFeed[0]==0))

{

_vUsage(argv[0]);

return -1;

}

}

if(_bEmon==1)

{

if (_acApiKey[0]==0)

{

_vUsage(argv[0]);

return -1;

}

}

return 0;

}

int main (int argc, char *argv[])

{

//init

pvPrint1=iQuiet;

pvPrint2=iQuiet;

strncpy(_acDevice,"/dev/ttyUSB0",MAX_DEV);

//get arguments

iGetArg(argc,argv );

//process

return iRun();

}