/*
 * Copyright 2022, Haiku, Inc.
 * Distributed under the terms of the MIT License.
 */


#ifndef _X86PCICONTROLLER_H_
#define _X86PCICONTROLLER_H_

#include <bus/PCI.h>

#include <AutoDeleterOS.h>
#include <lock.h>

#include "../ecam/ECAMPCIController.h"


#define CHECK_RET(err) {status_t _err = (err); if (_err < B_OK) return _err;}

#define PCI_X86_DRIVER_MODULE_NAME "busses/pci/x86/driver_v1"


class X86PCIController {
public:
	virtual ~X86PCIController() = default;

	static float SupportsDevice(device_node* parent);
	static status_t RegisterDevice(device_node* parent);
	static status_t InitDriver(device_node* node, X86PCIController*& outDriver);
	void UninitDriver();

	virtual status_t ReadConfig(
		uint8 bus, uint8 device, uint8 function,
		uint16 offset, uint8 size, uint32 &value) = 0;

	virtual status_t WriteConfig(
		uint8 bus, uint8 device, uint8 function,
		uint16 offset, uint8 size, uint32 value) = 0;

	virtual status_t GetMaxBusDevices(int32& count) = 0;

	virtual status_t GetRange(uint32 index, pci_resource_range* range);

	virtual status_t Finalize() final;

	status_t ReadIrq(
		uint8 bus, uint8 device, uint8 function,
		uint8 pin, uint8& irq);

	status_t WriteIrq(
		uint8 bus, uint8 device, uint8 function,
		uint8 pin, uint8 irq);

protected:
	static status_t CreateDriver(device_node* node, X86PCIController* driver,
		X86PCIController*& driverOut);
	virtual status_t InitDriverInt(device_node* node);

protected:
	spinlock fLock = B_SPINLOCK_INITIALIZER;

	device_node* fNode{};
};


class X86PCIControllerMeth1: public X86PCIController {
public:
	virtual ~X86PCIControllerMeth1() = default;

	status_t InitDriverInt(device_node* node) override;

	status_t ReadConfig(
		uint8 bus, uint8 device, uint8 function,
		uint16 offset, uint8 size, uint32 &value) override;

	status_t WriteConfig(
		uint8 bus, uint8 device, uint8 function,
		uint16 offset, uint8 size, uint32 value) override;

	status_t GetMaxBusDevices(int32& count) override;
};


class X86PCIControllerMeth2: public X86PCIController {
public:
	virtual ~X86PCIControllerMeth2() = default;

	status_t InitDriverInt(device_node* node) final;

	status_t ReadConfig(
		uint8 bus, uint8 device, uint8 function,
		uint16 offset, uint8 size, uint32 &value) final;

	status_t WriteConfig(
		uint8 bus, uint8 device, uint8 function,
		uint16 offset, uint8 size, uint32 value) final;

	status_t GetMaxBusDevices(int32& count) final;
};


class X86PCIControllerMethPcie: public X86PCIControllerMeth1 {
public:
	virtual ~X86PCIControllerMethPcie() = default;

	status_t InitDriverInt(device_node* node) final;

	status_t ReadConfig(
		uint8 bus, uint8 device, uint8 function,
		uint16 offset, uint8 size, uint32 &value) final;

	status_t WriteConfig(
		uint8 bus, uint8 device, uint8 function,
		uint16 offset, uint8 size, uint32 value) final;

	status_t GetMaxBusDevices(int32& count) final;

	status_t GetRange(uint32 index, pci_resource_range* range) final;

private:
	ECAMPCIControllerACPI fECAMPCIController;
};


extern device_manager_info* gDeviceManager;

#endif	// _X86PCICONTROLLER_H_