/*
 * Copyright 2022, Gerasim Troeglazov <3dEyes@gmail.com>
 * Distributed under the terms of the MIT License.
 */

#ifndef _USB_WINCHIPHEAD_H_
#define _USB_WINCHIPHEAD_H_

#include "SerialDevice.h"

#define CH34X_DEFAULT_BAUD_RATE	9600

#define CH34X_INPUT_BUF_SIZE	8

#define CH34X_VER_20			0x20
#define CH34X_VER_30			0x30

#define CH34X_BPS_PRE_IMM		0x80

#define CH34X_BIT_CTS			0x01
#define CH34X_BIT_DSR			0x02
#define CH34X_BIT_RI			0x04
#define CH34X_BIT_DCD			0x08
#define CH34X_BIT_DTR			0x20
#define CH34X_BIT_RTS			0x40
#define CH34X_BITS_MODEM_STAT	0x0f

#define CH34X_MULT_STAT 		0x04

#define CH34X_REQ_READ_VERSION	0x5F
#define CH34X_REQ_WRITE_REG		0x9A
#define CH34X_REQ_READ_REG		0x95
#define CH34X_REQ_SERIAL_INIT	0xA1
#define CH34X_REQ_MODEM_CTRL	0xA4

#define CH34X_REG_BREAK			0x05
#define CH34X_REG_STAT1			0x06
#define CH34X_REG_STAT2			0x07
#define CH34X_REG_BPS_PRE		0x12
#define CH34X_REG_BPS_DIV		0x13
#define CH34X_REG_LCR			0x18
#define CH34X_REG_LCR2			0x25

#define CH34X_LCR_CS5			0x00
#define CH34X_LCR_CS6			0x01
#define CH34X_LCR_CS7			0x02
#define CH34X_LCR_CS8			0x03
#define CH34X_LCR_STOP_BITS_2	0x04
#define CH34X_LCR_ENABLE_PAR	0x08
#define CH34X_LCR_PAR_EVEN		0x10
#define CH34X_LCR_MARK_SPACE	0x20
#define CH34X_LCR_ENABLE_TX		0x40
#define CH34X_LCR_ENABLE_RX		0x80

#define CH34X_SIO_600			0x6401
#define CH34X_SIO_1200			0xB201
#define CH34X_SIO_1800			0xCC01
#define CH34X_SIO_2400			0xD901
#define CH34X_SIO_4800			0x6402
#define CH34X_SIO_9600			0xB202
#define CH34X_SIO_19200			0xD902
#define CH34X_SIO_31250			0x4003
#define CH34X_SIO_38400			0x6403
#define CH34X_SIO_57600			0xF302
#define CH34X_SIO_115200		0xCC03
#define CH34X_SIO_230400		0xE603

/* supported vendor and product ids */
#define VENDOR_WCH				0x4348	// WinChipHead
#define VENDOR_QIN_HENG			0x1a86	// QinHeng Electronics
#define VENDOR_GW_INSTEK		0x2184	// GW Instek

const usb_serial_device kWCHDevices[] = {
	{VENDOR_WCH,		0x5523, "CH341 serial converter"},
	{VENDOR_QIN_HENG,	0x5523, "QinHeng CH341A serial converter"},
	{VENDOR_QIN_HENG,	0x7522, "QinHeng CH340 serial converter"},
	{VENDOR_QIN_HENG,	0x7523, "QinHeng CH340 serial converter"},
	{VENDOR_GW_INSTEK,	0x0057, "GW Instek Oscilloscope"}
};

class WCHDevice : public SerialDevice {
public:
						WCHDevice(usb_device device, uint16 vendorID,
							uint16 productID, const char *description);

	virtual	status_t	AddDevice(const usb_configuration_info *config);

	virtual	status_t	ResetDevice();

	virtual	status_t	SetLineCoding(usb_cdc_line_coding *coding);
	virtual	status_t	SetControlLineState(uint16 state);

private:
	status_t			WriteConfig(uint16 dataRate, uint8 lcr, uint8 mcr);

	uint8				fChipVersion;
	uint8				fStatusMCR;
	uint8				fStatusLCR;
	uint16				fDataRate;
};


#endif //_USB_WINCHIPHEAD_H_