FactoryCraft-Server/FactoryCraft/ItemFilter.h

#pragma once

#include <ReferenceCounter.h>
#include <functional>
#include <Array.h>

class Item;
class ItemType;
class ItemSlot;

class ItemFilter : public virtual Framework::ReferenceCounter
{
public:
	ItemFilter();
	virtual bool matchItem(const Item* zItem) const;
	virtual bool matchSourceSlot(ItemSlot* zSlot) const;
	virtual bool matchTargetSlot(ItemSlot* zSlot) const;
};


class CombinedItemFilter : public ItemFilter
{
private:
	ItemFilter* filterA;
	ItemFilter* filterB;
	std::function<bool(bool, bool)> op;

public:
	CombinedItemFilter(ItemFilter* filterA, ItemFilter* filterB, std::function<bool(bool, bool)> op);
	~CombinedItemFilter();
	bool matchItem(const Item* zItem) const override;
	bool matchSourceSlot(ItemSlot* zSlot) const override;
	bool matchTargetSlot(ItemSlot* zSlot) const override;
};


class AnyItemFilter : public ItemFilter
{
public:
	AnyItemFilter();
	bool matchItem(const Item* zItem) const override;
};


class TypeItemFilter : public ItemFilter
{
private:
	const ItemType* zType;

public:
	TypeItemFilter(const ItemType* zType);
	bool matchItem(const Item* zItem) const override;
};


class SpecificSlotFilter : public ItemFilter
{
private:
	int sourceSlotId;
	int targetSlotId;

public:
	SpecificSlotFilter(int sourceSlotId, int targetSlotId);
	bool matchSourceSlot(ItemSlot* zSlot) const override;
	bool matchTargetSlot(ItemSlot* zSlot) const override;
};


class SourceSlotBlacklistFilter : public ItemFilter
{
private:
	Framework::Array<int> blackList;

public:
	SourceSlotBlacklistFilter();
	void addBlackListSlotId(int id);
	bool matchSourceSlot(ItemSlot* zSlot) const override;
	bool matchTargetSlot(ItemSlot* zSlot) const override;
};