#pragma once

// Custom Classes
#include "CDTable.h"

/*!
 \file CDRarityTableTable.hpp
 \brief Contains data for the RarityTable table
 */

 //! RarityTable Entry Struct
struct CDRarityTable {
    unsigned int id;
    float randmax;
    unsigned int rarity;
    unsigned int RarityTableIndex;

    friend bool operator> (const CDRarityTable& c1, const CDRarityTable& c2)
    {
        return c1.rarity > c2.rarity;
    }

    friend bool operator>= (const CDRarityTable& c1, const CDRarityTable& c2)
    {
        return c1.rarity >= c2.rarity;
    }

    friend bool operator< (const CDRarityTable& c1, const CDRarityTable& c2)
    {
        return c1.rarity < c2.rarity;
    }

    friend bool operator<= (const CDRarityTable& c1, const CDRarityTable& c2)
    {
        return c1.rarity <= c2.rarity;
    }
};


//! RarityTable table
class CDRarityTableTable : public CDTable {
private:
    std::vector<CDRarityTable> entries;

public:

    //! Constructor
    CDRarityTableTable(void);

    //! Destructor
    ~CDRarityTableTable(void);

    //! Returns the table's name
    /*!
      \return The table name
     */
    std::string GetName(void) const override;

    //! Queries the table with a custom "where" clause
    /*!
      \param predicate The predicate
     */
    std::vector<CDRarityTable> Query(std::function<bool(CDRarityTable)> predicate);

    //! Gets all the entries in the table
    /*!
      \return The entries
     */
    const std::vector<CDRarityTable>& GetEntries(void) const;

};