/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <https://www.gnu.org/licenses/old-licenses/gpl-2.0>.
 */

/** @file newgrf_airport.h NewGRF handling of airports. */

#ifndef NEWGRF_AIRPORT_H
#define NEWGRF_AIRPORT_H

#include "airport.h"
#include "timer/timer_game_calendar.h"
#include "newgrf_badge_type.h"
#include "newgrf_class.h"
#include "newgrf_commons.h"
#include "newgrf_spritegroup.h"
#include "newgrf_town.h"
#include "tilearea_type.h"

/** Copy from station_map.h */
typedef uint8_t StationGfx;

/** Tile-offset / AirportTileID pair. */
struct AirportTileTable {
	TileIndexDiffC ti; ///< Tile offset from  the top-most airport tile.
	StationGfx gfx; ///< AirportTile to use for this tile.
};

/** Iterator to iterate over all tiles belonging to an airport spec. */
class AirportTileTableIterator : public TileIterator {
private:
	std::span<const AirportTileTable> att; ///< The offsets.
	TileIndex base_tile; ///< The tile we base the offsets off.
	std::span<const AirportTileTable>::iterator iter;

public:
	/**
	 * Construct the iterator.
	 * @param att The TileTable we want to iterate over.
	 * @param base_tile The basetile for all offsets.
	 */
	AirportTileTableIterator(std::span<const AirportTileTable> att, TileIndex base_tile)
		: TileIterator(base_tile + ToTileIndexDiff(att.front().ti))
		, att(att), base_tile(base_tile), iter(att.begin())
	{
	}

	inline TileIterator& operator ++() override
	{
		++this->iter;
		if (this->iter == std::end(att)) {
			this->tile = INVALID_TILE;
		} else {
			this->tile = this->base_tile + ToTileIndexDiff(this->iter->ti);
		}
		return *this;
	}

	/**
	 * Get the StationGfx for the current tile.
	 * @return The identifier of the graphics for this tile.
	 */
	StationGfx GetStationGfx() const
	{
		return this->iter->gfx;
	}

	std::unique_ptr<TileIterator> Clone() const override
	{
		return std::make_unique<AirportTileTableIterator>(*this);
	}
};

/** Class IDs for airports. */
using AirportClassID = PoolID<uint8_t, struct AirportClassIDTag, 16, UINT8_MAX>;

static constexpr AirportClassID APC_SMALL{0}; ///< id for small airports class.
static constexpr AirportClassID APC_LARGE{1}; ///< id for large airports class.
static constexpr AirportClassID APC_HUB{2}; ///< id for hub airports class.
static constexpr AirportClassID APC_HELIPORT{3}; ///< id for heliports.

/** TTDP airport types. Used to map our types to TTDPatch's */
enum TTDPAirportType : uint8_t {
	ATP_TTDP_SMALL,    ///< Same as AT_SMALL
	ATP_TTDP_LARGE,    ///< Same as AT_LARGE
	ATP_TTDP_HELIPORT, ///< Same as AT_HELIPORT
	ATP_TTDP_OILRIG,   ///< Same as AT_OILRIG
};

/** A list of all hangar tiles in an airport */
struct HangarTileTable {
	TileIndexDiffC ti; ///< Tile offset from the top-most airport tile.
	Direction dir;     ///< Direction of the exit.
	uint8_t hangar_num;   ///< The hangar to which this tile belongs.
};

struct AirportTileLayout {
	std::vector<AirportTileTable> tiles; ///< List of all tiles in this layout.
	Direction rotation; ///< The rotation of this layout.
};

/**
 * Defines the data structure for an airport.
 */
struct AirportSpec : NewGRFSpecBase<AirportClassID> {
	const struct AirportFTAClass *fsm;     ///< the finite statemachine for the default airports
	std::vector<AirportTileLayout> layouts; ///< List of layouts composing the airport.
	std::span<const HangarTileTable> depots; ///< Position of the depots on the airports.
	uint8_t size_x;                           ///< size of airport in x direction
	uint8_t size_y;                           ///< size of airport in y direction
	uint8_t noise_level;                      ///< noise that this airport generates
	uint8_t catchment;                        ///< catchment area of this airport
	TimerGameCalendar::Year min_year;      ///< first year the airport is available
	TimerGameCalendar::Year max_year;      ///< last year the airport is available
	StringID name;                         ///< name of this airport
	TTDPAirportType ttd_airport_type;      ///< ttdpatch airport type (Small/Large/Helipad/Oilrig)
	SpriteID preview_sprite;               ///< preview sprite for this airport
	uint16_t maintenance_cost;               ///< maintenance cost multiplier
	/* Newgrf data */
	bool enabled;                          ///< Entity still available (by default true). Newgrf can disable it, though.
	SubstituteGRFFileProps grf_prop; ///< Properties related to the grf file.
	std::vector<BadgeID> badges;

	static const AirportSpec *Get(uint8_t type);
	static AirportSpec *GetWithoutOverride(uint8_t type);

	bool IsAvailable() const;
	bool IsWithinMapBounds(uint8_t table, TileIndex index) const;

	static void ResetAirports();

	/**
	 * Get the index of this spec.
	 * @return The offset in the specs table.
	 */
	uint8_t GetIndex() const
	{
		assert(this >= std::begin(specs) && this < std::end(specs));
		return static_cast<uint8_t>(std::distance(std::cbegin(specs), this));
	}

	static const AirportSpec dummy; ///< The dummy airport.

private:
	static AirportSpec specs[NUM_AIRPORTS]; ///< Specs of the airports.

	friend void AirportOverrideManager::SetEntitySpec(AirportSpec &&as);
};

/** Information related to airport classes. */
using AirportClass = NewGRFClass<AirportSpec, AirportClassID>;

void BindAirportSpecs();

/** Resolver for the airport scope. */
struct AirportScopeResolver : public ScopeResolver {
	struct Station *st; ///< Station of the airport for which the callback is run, or \c nullptr for build gui.
	const AirportSpec *spec; ///< AirportSpec for which the callback is run.
	uint8_t layout;        ///< Layout of the airport to build.
	TileIndex tile;     ///< Tile for the callback, only valid for airporttile callbacks.

	/**
	 * Constructor of the scope resolver for an airport.
	 * @param ro Surrounding resolver.
	 * @param tile %Tile for the callback, only valid for airporttile callbacks.
	 * @param st %Station of the airport for which the callback is run, or \c nullptr for build gui.
	 * @param spec AirportSpec for which the callback is run.
	 * @param layout Layout of the airport to build.
	 */
	AirportScopeResolver(ResolverObject &ro, TileIndex tile, Station *st, const AirportSpec *spec, uint8_t layout)
		: ScopeResolver(ro), st(st), spec(spec), layout(layout), tile(tile)
	{
	}

	uint32_t GetRandomBits() const override;
	uint32_t GetVariable(uint8_t variable, [[maybe_unused]] uint32_t parameter, bool &available) const override;
	void StorePSA(uint pos, int32_t value) override;
};


/** Resolver object for airports. */
struct AirportResolverObject : public ResolverObject {
	AirportScopeResolver airport_scope;
	std::optional<TownScopeResolver> town_scope = std::nullopt; ///< The town scope resolver (created on the first call).

	AirportResolverObject(TileIndex tile, Station *st, const AirportSpec *spec, uint8_t layout,
			CallbackID callback = CBID_NO_CALLBACK, uint32_t callback_param1 = 0, uint32_t callback_param2 = 0);

	TownScopeResolver *GetTown();

	ScopeResolver *GetScope(VarSpriteGroupScope scope = VSG_SCOPE_SELF, uint8_t relative = 0) override
	{
		switch (scope) {
			case VSG_SCOPE_SELF: return &this->airport_scope;
			case VSG_SCOPE_PARENT:
			{
				TownScopeResolver *tsr = this->GetTown();
				if (tsr != nullptr) return tsr;
				[[fallthrough]];
			}
			default: return ResolverObject::GetScope(scope, relative);
		}
	}

	GrfSpecFeature GetFeature() const override;
	uint32_t GetDebugID() const override;
};

StringID GetAirportTextCallback(const AirportSpec *as, uint8_t layout, uint16_t callback);

#endif /* NEWGRF_AIRPORT_H */
