psopeeps-newserv/src/BattleParamsIndex.hh

#pragma once

#include <inttypes.h>

#include <array>
#include <memory>
#include <phosg/Encoding.hh>
#include <random>
#include <string>
#include <vector>

#include "EnemyType.hh"
#include "LevelTable.hh"
#include "StaticGameData.hh"
#include "Text.hh"

class BattleParamsIndex {
public:
  // These files are little-endian, even on PSO GC.

  struct AttackData {
    /* 00 */ le_int16_t min_atp;
    /* 02 */ le_int16_t max_atp;
    /* 04 */ le_int16_t min_ata;
    /* 06 */ le_int16_t max_ata;
    /* 08 */ le_float distance_x;
    /* 0C */ le_uint32_t angle; // Out of 0x10000 (high 16 bits are unused)
    /* 10 */ le_float distance_y;
    /* 14 */ le_uint16_t unknown_a8;
    /* 16 */ le_uint16_t unknown_a9;
    /* 18 */ le_uint16_t unknown_a10;
    /* 1A */ le_uint16_t unknown_a11;
    /* 1C */ le_uint32_t unknown_a12;
    /* 20 */ le_uint32_t unknown_a13;
    /* 24 */ le_uint32_t unknown_a14;
    /* 28 */ le_uint32_t unknown_a15;
    /* 2C */ le_uint32_t unknown_a16;
    static AttackData from_json(const phosg::JSON& json);
    phosg::JSON json() const;
  } __packed_ws__(AttackData, 0x30);

  struct ResistData {
    /* 00 */ le_int16_t evp_bonus;
    /* 02 */ le_uint16_t efr;
    /* 04 */ le_uint16_t eic;
    /* 06 */ le_uint16_t eth;
    /* 08 */ le_uint16_t elt;
    /* 0A */ le_uint16_t edk;
    /* 0C */ le_uint32_t unknown_a6;
    /* 10 */ le_uint32_t unknown_a7;
    /* 14 */ le_uint32_t unknown_a8;
    /* 18 */ le_uint32_t unknown_a9;
    /* 1C */ le_int32_t dfp_bonus;
    static ResistData from_json(const phosg::JSON& json);
    phosg::JSON json() const;
  } __packed_ws__(ResistData, 0x20);

  struct MovementData {
    /* 00 */ le_float fparam1;
    /* 04 */ le_float fparam2;
    /* 03 */ le_float fparam3;
    /* 0C */ le_float fparam4;
    /* 10 */ le_float fparam5;
    /* 14 */ le_float fparam6;
    /* 18 */ le_uint32_t iparam1;
    /* 1C */ le_uint32_t iparam2;
    /* 20 */ le_uint32_t iparam3;
    /* 24 */ le_uint32_t iparam4;
    /* 28 */ le_uint32_t iparam5;
    /* 2C */ le_uint32_t iparam6;
    static MovementData from_json(const phosg::JSON& json);
    phosg::JSON json() const;
  } __packed_ws__(MovementData, 0x30);

  struct Table {
    /* 0000 */ parray<parray<PlayerStats, 0x60>, 4> stats; // [difficulty][bp_index]
    /* 3600 */ parray<parray<AttackData, 0x60>, 4> attack_data; // [difficulty][bp_index]
    /* 7E00 */ parray<parray<ResistData, 0x60>, 4> resist_data; // [difficulty][bp_index]
    /* AE00 */ parray<parray<MovementData, 0x60>, 4> movement_data; // [difficulty][bp_index]
    /* F600 */

    inline const PlayerStats& stats_for_index(Difficulty difficulty, uint8_t index) const {
      return this->stats.at(static_cast<size_t>(difficulty)).at(index);
    }
    inline const AttackData& attack_data_for_index(Difficulty difficulty, uint8_t index) const {
      return this->attack_data.at(static_cast<size_t>(difficulty)).at(index);
    }
    inline const ResistData& resist_data_for_index(Difficulty difficulty, uint8_t index) const {
      return this->resist_data.at(static_cast<size_t>(difficulty)).at(index);
    }
    inline const MovementData& movement_data_for_index(Difficulty difficulty, uint8_t index) const {
      return this->movement_data.at(static_cast<size_t>(difficulty)).at(index);
    }

    static Table from_json(const phosg::JSON& json);
    phosg::JSON json() const;

    void print(FILE* stream, Episode episode) const;
  } __packed_ws__(Table, 0xF600);

  virtual ~BattleParamsIndex() = default;

  virtual const Table& get_table(bool solo, Episode episode) const = 0;
  phosg::JSON json() const;

protected:
  BattleParamsIndex() = default;
};

class JSONBattleParamsIndex : public BattleParamsIndex {
public:
  explicit JSONBattleParamsIndex(const phosg::JSON& json);

  virtual const Table& get_table(bool solo, Episode episode) const;

protected:
  // Indexed as [online/offline][episode]
  std::array<std::array<Table, 3>, 2> tables;
};

class BinaryBattleParamsIndex : public BattleParamsIndex {
public:
  BinaryBattleParamsIndex(
      std::shared_ptr<const std::string> data_on_ep1, // BattleParamEntry_on.dat
      std::shared_ptr<const std::string> data_on_ep2, // BattleParamEntry_lab_on.dat
      std::shared_ptr<const std::string> data_on_ep4, // BattleParamEntry_ep4_on.dat
      std::shared_ptr<const std::string> data_off_ep1, // BattleParamEntry.dat
      std::shared_ptr<const std::string> data_off_ep2, // BattleParamEntry_lab.dat
      std::shared_ptr<const std::string> data_off_ep4); // BattleParamEntry_ep4.dat

  virtual const Table& get_table(bool solo, Episode episode) const;

protected:
  struct File {
    std::shared_ptr<const std::string> data;
    const Table* table = nullptr;
  };

  // Indexed as [online/offline][episode]
  std::array<std::array<File, 3>, 2> files;
};