Martin Michelsen
131 lines
4.4 KiB
C++
131 lines
4.4 KiB
C++
#pragma once
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <array>
|
|
#include <memory>
|
|
#include <phosg/JSON.hh>
|
|
#include <random>
|
|
#include <string>
|
|
|
|
#include "AFSArchive.hh"
|
|
#include "CommonItemSet.hh"
|
|
#include "GSLArchive.hh"
|
|
#include "ItemNameIndex.hh"
|
|
#include "StaticGameData.hh"
|
|
#include "Text.hh"
|
|
#include "Version.hh"
|
|
|
|
class RareItemSet {
|
|
public:
|
|
struct ExpandedDrop {
|
|
uint32_t probability = 0;
|
|
ItemData data;
|
|
|
|
bool operator==(const ExpandedDrop& other) const = default;
|
|
bool operator!=(const ExpandedDrop& other) const = default;
|
|
|
|
std::string str() const;
|
|
std::string str(std::shared_ptr<const ItemNameIndex> name_index) const;
|
|
};
|
|
|
|
RareItemSet();
|
|
RareItemSet(const AFSArchive& afs, bool is_v1);
|
|
RareItemSet(const GSLArchive& gsl, bool is_big_endian);
|
|
RareItemSet(const std::string& rel, bool is_big_endian);
|
|
RareItemSet(const phosg::JSON& json, std::shared_ptr<const ItemNameIndex> name_index = nullptr);
|
|
~RareItemSet() = default;
|
|
|
|
std::vector<ExpandedDrop> get_enemy_specs(
|
|
GameMode mode, Episode episode, Difficulty difficulty, uint8_t secid, EnemyType enemy_type) const;
|
|
std::vector<ExpandedDrop> get_box_specs(
|
|
GameMode mode, Episode episode, Difficulty difficulty, uint8_t secid, uint8_t area_norm) const;
|
|
|
|
bool has_entries_for_game_config(GameMode mode, Episode episode, Difficulty difficulty) const;
|
|
|
|
std::string serialize_afs(bool is_v1) const;
|
|
std::string serialize_gsl(bool big_endian) const;
|
|
std::string serialize_html(
|
|
GameMode mode,
|
|
Episode episode,
|
|
Difficulty difficulty,
|
|
std::shared_ptr<const ItemNameIndex> name_index = nullptr,
|
|
std::shared_ptr<const CommonItemSet> common_item_set = nullptr) const;
|
|
phosg::JSON json(std::shared_ptr<const ItemNameIndex> name_index = nullptr) const;
|
|
|
|
void multiply_all_rates(double factor);
|
|
|
|
void print_collection(
|
|
FILE* stream,
|
|
GameMode mode,
|
|
Episode episode,
|
|
Difficulty difficulty,
|
|
uint8_t section_id,
|
|
std::shared_ptr<const ItemNameIndex> name_index = nullptr) const;
|
|
void print_all_collections(FILE* stream, std::shared_ptr<const ItemNameIndex> name_index = nullptr) const;
|
|
void print_diff(FILE* stream, const RareItemSet& other) const;
|
|
|
|
protected:
|
|
struct SpecCollection {
|
|
std::unordered_map<EnemyType, std::vector<ExpandedDrop>> enemy_specs;
|
|
std::vector<std::vector<ExpandedDrop>> box_specs; // Indexed by area_norm
|
|
|
|
bool operator==(const SpecCollection& other) const = default;
|
|
bool operator!=(const SpecCollection& other) const = default;
|
|
|
|
void print_diff(FILE* stream, const SpecCollection& other) const;
|
|
};
|
|
|
|
struct ParsedRELData {
|
|
struct PackedDrop {
|
|
uint8_t probability = 0;
|
|
parray<uint8_t, 3> item_code;
|
|
|
|
PackedDrop() = default;
|
|
explicit PackedDrop(const ExpandedDrop&);
|
|
ExpandedDrop expand() const;
|
|
} __packed_ws__(PackedDrop, 4);
|
|
|
|
template <bool BE>
|
|
struct RootT {
|
|
/* 00 */ U32T<BE> monster_rares_offset; // -> parray<PackedDrop, 0x65> (or 0x33 on v1)
|
|
/* 04 */ U32T<BE> box_count; // Usually 30 (0x1E)
|
|
/* 08 */ U32T<BE> box_areas_offset; // -> parray<uint8_t, 0x1E>
|
|
/* 0C */ U32T<BE> box_rares_offset; // -> parray<PackedDrop, 0x1E>
|
|
/* 10 */
|
|
} __packed_ws_be__(RootT, 0x10);
|
|
using Root = RootT<false>;
|
|
using RootBE = RootT<true>;
|
|
|
|
struct BoxRare {
|
|
uint8_t area_norm_plus_1;
|
|
ExpandedDrop drop;
|
|
};
|
|
|
|
std::vector<ExpandedDrop> monster_rares; // Indexed by rt_index
|
|
std::vector<BoxRare> box_rares; // Not indexed (area_norm + 1 is in the struct)
|
|
|
|
ParsedRELData() = default;
|
|
ParsedRELData(phosg::StringReader r, bool big_endian, bool is_v1);
|
|
explicit ParsedRELData(const SpecCollection& collection);
|
|
std::string serialize(bool big_endian, bool is_v1) const;
|
|
|
|
template <bool BE>
|
|
void parse_t(phosg::StringReader r, bool is_v1);
|
|
template <bool BE>
|
|
std::string serialize_t(bool is_v1) const;
|
|
|
|
SpecCollection as_collection(Episode episode) const;
|
|
};
|
|
|
|
std::unordered_map<uint16_t, SpecCollection> collections;
|
|
|
|
const SpecCollection& get_collection(GameMode mode, Episode episode, Difficulty difficulty, uint8_t secid) const;
|
|
|
|
static std::string gsl_entry_name_for_table(GameMode mode, Episode episode, Difficulty difficulty, uint8_t section_id);
|
|
static uint16_t key_for_params(GameMode mode, Episode episode, Difficulty difficulty, uint8_t secid);
|
|
|
|
static uint32_t expand_rate(uint8_t pc);
|
|
static uint8_t compress_rate(uint32_t probability);
|
|
};
|