/*
* Copyright 1999-2000 Y.Takagi. All Rights Reserved.
*
* Copyright 2010, Haiku. All rights reserved.
* Distributed under the terms of the MIT License.
*
* Authors:
*		Michael Pfeiffer
*/


#include "GPCapabilities.h"

#include "GPBinding.h"
#include "GPData.h"
#include "PrinterData.h"


GPCapabilities::GPCapabilities(const PrinterData* printer_data)
	:
	PrinterCap(printer_data)
{
	InitCapabilitiesFromGutenprint();
}


GPCapabilities::~GPCapabilities()
{
}

int
GPCapabilities::CountCap(CapID category) const
{
	const GPArray<struct BaseCap>* capabilities;

	switch (category) {
		case kPaper:
			return fPageSizes.Size();
		case kPaperSource:
			return fInputSlots.Size();
		case kResolution:
			return fResolutions.Size();
		case kColor:
			return fPrintingModes.Size();
		case kDriverSpecificCapabilities:
			return fDriverSpecificCategories.Size();

		default:
			capabilities = DriverSpecificCapabilities(category);
			if (capabilities == NULL)
				return 0;
			return capabilities->Size();
	}
}


const BaseCap**
GPCapabilities::GetCaps(CapID category) const
{
	typedef const BaseCap** ArrayType;
	const GPArray<struct BaseCap>* capabilities;

	switch (category) {
		case kPaper:
			return (ArrayType)fPageSizes.Array();
		case kPaperSource:
			return (ArrayType)fInputSlots.Array();
		case kResolution:
			return (ArrayType)fResolutions.Array();
		case kColor:
			return (ArrayType)fPrintingModes.Array();
		case kDriverSpecificCapabilities:
			return (ArrayType)fDriverSpecificCategories.Array();

		default:
			capabilities = DriverSpecificCapabilities(category);
			if (capabilities == NULL)
				return NULL;
			return (ArrayType)capabilities->Array();
	}
}


bool
GPCapabilities::Supports(CapID category) const
{
	switch (category) {
		case kPaper:
		case kPaperSource:
		case kResolution:
		case kColor:
			return true;

		default:
			return CountCap(category) > 0;
	}
}

void
GPCapabilities::InitCapabilitiesFromGutenprint()
{
	const GPData* data = dynamic_cast<const GPData*>(GetPrinterData());
	ASSERT(data != NULL);
	// capabilities are available only after printer model
	// has been selected
	if (data->fGutenprintDriverName == "")
		return;


	GPBinding binding;
	binding.GetCapabilities(data->fGutenprintDriverName.String(), this);
}


const GPArray<struct BaseCap>*
GPCapabilities::DriverSpecificCapabilities(int32 category) const
{
	DriverSpecificCapabilitiesType::const_iterator it =
		fDriverSpecificCapabilities.find(category);
	if (it == fDriverSpecificCapabilities.end())
		return NULL;
	return &it->second;
}