/*
 * I2C通信関連
 *
 * Copyright (c) 2016 Wakayama.rb Ruby Board developers
 *
 * This software is released under the MIT License.
 * https://github.com/wakayamarb/wrbb-v2lib-firm/blob/master/MITL
 *
 */
#include <arduino.h>
/*#include <wire.h>*/

#include <mruby.h>
#include <mruby/data.h>
#include <mruby/class.h>

#include "../llbruby.h"
#include "sKernel.h"

#define WIRE_MAX	6

i2c_t *RbWire[WIRE_MAX];		/*0:Wire1, 1:Wire3, 2:Wire2, 3:Wire6 4:Wire7*/

/******************************************************/
/* メモリの開放時に走る*/
/******************************************************/
static void i2c_free(mrb_state *mrb, void *ptr)
{
	i2c_t* i2c = (i2c_t*)(ptr);
	mrb_free(mrb, i2c);
}

/******************************************************/
/* この構造体の意味はよくわかっていない*/
/******************************************************/
static struct mrb_data_type i2c_type = {"I2c", i2c_free};

/******************************************************/
/* I2Cを初期化します: I2c.new*/
/*  I2c.new(num)　*/
/*  num: 通信番号(0:SDA-0/SCL-1, 1:SDA-5/SCL-6, 2:SDA-7/SCL-8, 3:SDA-12/SCL-11, 4:SDA-9(26)/SCL-3)*/
/**/
/* 戻り値*/
/*  I2cのインスタンス*/
/******************************************************/
static mrb_value mrb_i2c_initialize(mrb_state *mrb, mrb_value self)
{
/* Initialize data type first, otherwise segmentation fault occurs.*/
	DATA_TYPE(self) = &i2c_type;
	DATA_PTR(self) = NULL;

	mrb_int num;

	mrb_get_args(mrb, "i", &num);

	if (num <= 0 && num >= WIRE_MAX) {
		return mrb_nil_value();			/*戻り値は無しですよ。*/
	}

	i2c_t* i2c = mrb_malloc(mrb, sizeof(i2c_t));
	i2c_init(i2c, num, num);

	DATA_PTR(self) = i2c;
	return self;
}

/******************************************************/
/* アドレスにデータを書き込みます: I2c.write*/
/*  I2c.write( deviceID, address, data )*/
/*	deviceID: デバイスID*/
/*  address: 書き込みアドレス*/
/*  data: データ*/
/**/
/*  戻り値は以下のとおり*/
/*		0: 成功*/
/*		4: その他のエラー*/
/******************************************************/
mrb_value mrb_i2c_write(mrb_state *mrb, mrb_value self)
{
	int deviceID, addr, dat;

	mrb_get_args(mrb, "iii", &deviceID, &addr, &dat);

	i2c_t* i2c = (i2c_t *)(mrb_get_datatype(mrb, self, &i2c_type));

	return mrb_fixnum_value(i2c_write(i2c, addr, (char *)&dat, sizeof(dat), 1));
}

/******************************************************/
/* アドレスからデータを読み込み: I2c.read*/
/*  I2c.read( deviceID, addressL[, addressH] )*/
/*	deviceID: デバイスID*/
/*  addressL: 読み込み下位アドレス*/
/*  addressH: 読み込み上位アドレス*/
/**/
/*  戻り値は読み込んだ値*/
/******************************************************/
mrb_value mrb_i2c_read(mrb_state *mrb, mrb_value self)
{
	int deviceID, addrL;
	int addrH = 0;
	int dat = 0;
	/*byte datH;*/

	int n = mrb_get_args(mrb, "ii|i", &deviceID, &addrL, &addrH);

	i2c_t* i2c = (i2c_t *)(mrb_get_datatype(mrb, self, &i2c_type));

	i2c_read(i2c, addrL | (addrH << 8), (char *)&dat, sizeof(dat), 1);

	return mrb_fixnum_value(dat);
}

/******************************************************/
/* I2Cデバイスに対して送信を開始するための準備をする: I2c.begin*/
/*	I2c.begin( deviceID )*/
/*	この関数は送信バッファを初期化するだけで、実際の動作は行わない。繰り返し呼ぶと、送信バッファが先頭に戻る。*/
/*	deviceID: デバイスID 0～0x7Fまでの純粋なアドレス*/
/******************************************************/
mrb_value mrb_i2c_beginTransmission(mrb_state *mrb, mrb_value self)
{
	int deviceID;

	mrb_get_args(mrb, "i", &deviceID);

	i2c_t* i2c = (i2c_t *)(mrb_get_datatype(mrb, self, &i2c_type));

	i2c_start(i2c);
	i2c_byte_write(i2c, deviceID);

	return mrb_nil_value();			/*戻り値は無しですよ。*/
}

/******************************************************/
/* 送信バッファの末尾に数値を追加する: I2c.lwrite*/
/*	I2c.lwrite( data )*/
/*	data: セットする値*/
/**/
/* 戻り値は、送信したバイト数(バッファに溜めたバイト数)を返す。*/
/*	送信バッファ(260バイト)に空き容量が無ければ失敗して0を返す*/
/******************************************************/
mrb_value mrb_i2c_lwrite(mrb_state *mrb, mrb_value self)
{
	int dat;

	mrb_get_args(mrb, "i", &dat);

	i2c_t* i2c = (i2c_t *)(mrb_get_datatype(mrb, self, &i2c_type));

	return mrb_fixnum_value(i2c_byte_write(i2c, dat));
}

/******************************************************/
/* デバイスに対してI2Cの送信シーケンスを発行する: I2c.end*/
/*	I2c.end()*/
/*	I2Cの送信はこの関数を実行して初めて実際に行われる。*/
/**/
/* 戻り値は以下のとおり*/
/*	0: 成功*/
/*	4: その他のエラー*/
/******************************************************/
mrb_value mrb_i2c_endTransmission(mrb_state *mrb, mrb_value self)
{
	i2c_t* i2c = (i2c_t *)(mrb_get_datatype(mrb, self, &i2c_type));

	return mrb_fixnum_value(i2c_stop(i2c));
}

/******************************************************/
/* デバイスに対して受信シーケンスを発行しデータを読み出す: I2c.request*/
/*	I2c.request( address, count )*/
/*  num: 通信番号(0:SDA-0/SCL-1, 1:SDA-5/SCL-6, 2:SDA-7/SCL-8, 3:SDA-12/SCL-11, 4:SDA-9(26)/SCL-3)*/
/*	address: 読み込み開始アドレス*/
/*	count: 読み出す数*/
/**/
/*  戻り値は、実際に受信したバイト数。*/
/******************************************************/
mrb_value mrb_i2c_requestFrom(mrb_state *mrb, mrb_value self)
{
	int addr, cnt, ret;
	char rxdata[16];

	mrb_get_args(mrb, "ii", &addr, &cnt);

	i2c_t* i2c = (i2c_t *)(mrb_get_datatype(mrb, self, &i2c_type));

	ret = i2c_read(i2c, addr, rxdata, cnt, 1);

	return mrb_fixnum_value(ret);
}

/******************************************************/
/* デバイスに対して受信シーケンスを発行しデータを読み出す: I2c.lread*/
/*	I2c.lread()*/
/**/
/*  戻り値は読み込んだ値*/
/******************************************************/
mrb_value mrb_i2c_lread(mrb_state *mrb, mrb_value self)
{
	i2c_t* i2c = (i2c_t *)(mrb_get_datatype(mrb, self, &i2c_type));
	return mrb_fixnum_value(i2c_byte_read(i2c, 1));
}

/******************************************************/
/* デバイスに対して受信バッファ内にあるデータ数を調べる: I2c.available*/
/*	I2c.available()*/
/**/
/*  戻り値はデータ数*/
/******************************************************/
mrb_value mrb_i2c_available(mrb_state *mrb, mrb_value self)
{
	i2c_t* i2c = (i2c_t *)(mrb_get_datatype(mrb, self, &i2c_type));
	return mrb_fixnum_value(1/*i2c_available(i2c)*/);
}
#if 0
/******************************************************/
/* 周波数を変更する: I2c.freq*/
/*  I2c.freq( Hz )*/
/*  Hz: クロックの周波数をHz単位で指定する。*/
/*      有効な値は1～200000程度。基本的にソフトでやっているので400kHzは出ない。*/
/******************************************************/
mrb_value mrb_i2c_freq(mrb_state *mrb, mrb_value self)
{
	int fq;

	mrb_get_args(mrb, "i", &fq);

	i2c_frequency(i2c, fq);

	return mrb_nil_value();			/* 戻り値は無しですよ。 */
}
#endif
/******************************************************/
/* ライブラリを定義します*/
/******************************************************/
void i2c_Init(mrb_state *mrb)
{
	/*0:Wire1, 1:Wire3, 2:Wire2, 3:Wire6 4:Wire7*/
	RbWire[0] = NULL;
	RbWire[1] = &Wire1;
	RbWire[2] = &Wire2;
	RbWire[3] = &Wire3;
	RbWire[4] = &Wire4;
	RbWire[5] = &Wire5;

	struct RClass *i2cModule = mrb_define_class(mrb, "I2c", mrb->object_class);
	MRB_SET_INSTANCE_TT(i2cModule, MRB_TT_DATA);

	mrb_define_method(mrb, i2cModule, "initialize", mrb_i2c_initialize, MRB_ARGS_REQ(1));
	mrb_define_method(mrb, i2cModule, "write", mrb_i2c_write, MRB_ARGS_REQ(3));
	mrb_define_method(mrb, i2cModule, "read", mrb_i2c_read, MRB_ARGS_REQ(2) | MRB_ARGS_OPT(1));
	mrb_define_method(mrb, i2cModule, "begin", mrb_i2c_beginTransmission, MRB_ARGS_REQ(1));
	mrb_define_method(mrb, i2cModule, "lwrite", mrb_i2c_lwrite, MRB_ARGS_REQ(1));
	mrb_define_method(mrb, i2cModule, "end", mrb_i2c_endTransmission, MRB_ARGS_NONE());
	mrb_define_method(mrb, i2cModule, "request", mrb_i2c_requestFrom, MRB_ARGS_REQ(2));
	mrb_define_method(mrb, i2cModule, "lread", mrb_i2c_lread, MRB_ARGS_NONE());
	mrb_define_method(mrb, i2cModule, "available", mrb_i2c_available, MRB_ARGS_NONE());
}