/*
 * CDCCommandStream.h
 *
 *  Created on: 17 Jan 2020
 *      Author: andrewcapon
 */

#ifndef SRC_CDCCOMMANDSTREAM_H_
#define SRC_CDCCOMMANDSTREAM_H_

#include <map>

#include "CDCCommandHandler.h"

// for receiving at fullspeed CDC we need as many fifo elements as the time taken in ms in the main loop
// Also for code that does not require fast streaming this could be set to 1
// So this will need tuning and define should come from cmake.

struct CDCFifoElement
{
	uint8_t  m_data[64];
	uint8_t  m_uLen = 0;
};


class CDCCommandStream
{
public:
	CDCCommandStream() : m_uFifoReadPos(0), m_uFifoWritePos(0)
	{
		Init();
	}

	void Init(void);
	void AddCommandHandler(CDCCommandHandler::CDCFourCC uCommand, CDCCommandHandler *pCommandHandler);


	uint32_t GetTimeTaken(void);
	void     Stream(void);
	uint8_t  Stream(uint8_t *data, uint32_t len);

  void SetParsingEnabled(bool enabled);

  uint32_t Read(uint8_t *data, uint32_t len);

	// USB Packet fifo
	uint8_t					*GetFifoWriteBuffer(void);
	void						ReleaseFifoWriteBuffer(uint8_t uLen);
	CDCFifoElement  *GetFifoReadElement(void);
	void 						ReleaseFifoReadElement(void);

private:
	typedef enum { stDetect, stDetectCommandWord, stDispatch, stProcessing, stError, stDisabled } StreamState;

	StreamState m_state;

	uint8_t			m_uHeaderScanPos = 0;
	uint8_t			m_uCommandScanPos = 0;
	uint8_t			m_uRetryCount = 0;
	CDCCommandHandler::CDCFourCC uCommand = 0;


	CDCCommandHandler														*m_pCurrentCommandHandler;
	std::map<uint32_t, CDCCommandHandler*> 	m_commandHandlers;

	uint32_t m_uDispatchTime;

	CDCDataStream	m_dataStream;

  uint32_t uLastResumeTime;

	CDCFifoElement m_fifoElements[CDC_FIFO_BUFFERS];
	uint8_t				 m_uFifoReadPos;
	uint8_t				 m_uFifoWritePos;

	bool					m_bNeedsUSBResume;

  uint8_t m_uCurElementOff = 0;


	void 			LogTimeTaken(CDCCommandHandler::StreamResult result, uint32_t uBytesHandled);

  void Resume();
};

#endif /* SRC_CDCCOMMANDSTREAM_H_ */