/*
* Copyright 2017, Haiku. All rights reserved.
* Distributed under the terms of the MIT License.
*
* Authors:
*		Adrien Destugues <pulkomandy@pulkomandy.tk>
*/
#include "Lpstyl.h"


LpstylDriver::LpstylDriver(BMessage* message, PrinterData* printerData,
	const PrinterCap* printerCap)
	: GraphicsDriver(message, printerData, printerCap)
{
}


bool
LpstylDriver::StartDocument()
{
	_EjectAndReset();
	_IdentifyPrinter();
	_ColorCartridge();
	return true;
}


bool
LpstylDriver::StartPage()
{
	if (fPrinterType < kStyleWriter2400)
		WriteSpoolData("nuA", 3);
	else
		WriteSpoolChar('L');
	return true;
}


bool
LpstylDriver::NextBand(BBitmap* bitmap, BPoint* offset)
{
	fprintf(stderr, "Next band at %f %f\n", offset->x, offset->y);

	int page_height = GetPageHeight();

	// Advance the cursor
	offset->y += bitmap->Bounds().Height();

	// Have we reached the end of the page yet?
	if (offset->y >= page_height)
	{
		offset->y = -1;
		offset->x = -1;
	}

	return true;
}


bool
LpstylDriver::EndPage(int page)
{
	fprintf(stderr, "end page %d\n", page);
	return true;
}


bool
LpstylDriver::EndDocument(bool success)
{
	return true;
}


/** Eject the current page (if any) and reset the printer.
 */
void
LpstylDriver::_EjectAndReset(void)
{
	_WriteFFFx('I');

	int s1;

	do {
		for (;;) {
			try {
				s1 = _GetStatus('1');
				break;
			} catch(const TransportException&) {
				continue;
			}
		}

		if (s1 == 1) {
			// Check for stylewriter1, where status 1 doesn't g to 0 on init.
			if (_GetStatus('2') == 0 && _GetStatus('B') == 0xa0)
				break;
		}
	} while (s1 == 1);
}


void
LpstylDriver::_IdentifyPrinter(void)
{
	WriteSpoolChar('?');

	char smallBuf[32];
	int i = 0;
	for (i = 0; i < 31; i++) {
		smallBuf[i] = ReadSpoolChar();
		if (smallBuf[i] == 0x0D)
			break;
	}
	smallBuf[i] = 0;

	if (strcmp(smallBuf, "IJ10\x0D") == 0)
		fPrinterType = kStyleWriter;
	else if (strcmp(smallBuf, "SW\x0D") == 0)
		fPrinterType = kStyleWriter2;
	else if (strcmp(smallBuf, "SW3\x0D") == 0)
		fPrinterType = kStyleWriter3;
	else if (strcmp(smallBuf, "CS\x0D") == 0) {
		switch (_GetStatus('p'))
		{
			default:
			case 1:
				fPrinterType = kStyleWriter2400;
				break;
			case 2:
				fPrinterType = kStyleWriter2200;
				break;
			case 4:
				fPrinterType = kStyleWriter1500;
				break;
			case 5:
				fPrinterType = kStyleWriter2500;
				break;
		}
	}
}


bool
LpstylDriver::_ColorCartridge()
{
	WriteSpoolChar('D');
	unsigned char i = _GetStatus('H');
	return i & 0x80;
}


/** Send a 4-byte command to the printer.
 *
 * These commands can be sent at any time, because their prefix is FFFFFF, a
 * sequence which can't be generated by the data compression algorithm
 */
void
LpstylDriver::_WriteFFFx(char x)
{
	unsigned char str[4];
	str[0] = str[1] = str[2] = 0xFF;
	str[3] = x;
	WriteSpoolData(str, 4);
}


/** Get one of the printer status bytes.
 *
 * There are 3 status registers, 1, 2, and B. Each returns some different
 * information about the printer state.
 */
int
LpstylDriver::_GetStatus(char reg)
{
	_WriteFFFx(reg);
	return ReadSpoolChar();
}