/*
MIT License

Copyright (c) 2022 Reisen

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
*/

#include "avz.h"
#include "avz_more.h"
#include "DanceCheat/DanceCheat.h"
#include "showme/sm.h"

// ----------------------------------------------------------------
#include <algorithm>
#include <cmath>
#include <set>
#include <utility>
#include <vector>
#include "avz.h"
#include "GetZombieAbscissas.h"

using namespace AvZ;

namespace __cob_manager_internal {
struct cob_list_t {
 ? ?private:
 ? ?int type;

 ? ?public:
 ? ?Grid pos;
 ? ?int col;

 ? ?cob_list_t(Grid g) : type(0), pos(g) {}
 ? ?cob_list_t(int r, int c) : type(0), pos({r, c}) {}
 ? ?cob_list_t(int c) : type(1), col(c) {}

 ? ?bool is_pos() const { return type == 0; }
 ? ?bool is_col() const { return type == 1; }
};

using p_row_t = unsigned;
using pd_row_t = double;
using col_t = float;

const int garg_x[387] = {845, 844, 844, 844, 843, 843, 843, 842, 842, 842, 841, 841, 841, 840, 840, 840, 840, 839, 839, 839, 838, 838, 838, 837, 837, 837, 836, 836, 836, 836, 835, 835, 835, 834, 834, 834, 833, 833, 833, 832, 832, 832, 832, 831, 831, 831, 830, 830, 830, 829, 829, 829, 828, 828, 828, 828, 827, 827, 827, 826, 826, 826, 825, 825, 825, 825, 824, 824, 824, 823, 823, 823, 822, 822, 822, 822, 821, 821, 821, 821, 821, 820, 820, 820, 820, 820, 819, 819, 819, 819, 819, 818, 818, 818, 818, 818, 817, 817, 817, 817, 816, 816, 816, 816, 816, 815, 815, 815, 815, 815, 814, 814, 814, 814, 814, 813, 813, 813, 813, 813, 812, 812, 812, 812, 812, 811, 811, 811, 811, 811, 810, 810, 810, 810, 810, 809, 809, 809, 809, 809, 808, 808, 808, 808, 808, 807, 807, 807, 807, 806, 806, 806, 805, 805, 805, 805, 804, 804, 804, 803, 803, 803, 802, 802, 802, 802, 801, 801, 801, 800, 800, 800, 799, 799, 799, 799, 798, 798, 798, 797, 797, 797, 796, 796, 796, 796, 795, 795, 795, 794, 794, 794, 793, 793, 793, 793, 792, 792, 792, 791, 791, 791, 790, 790, 790, 789, 789, 789, 789, 789, 789, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 788, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 787, 786, 786, 786, 786, 786, 786, 786, 786, 786, 786, 786, 786, 786, 785, 785, 785, 785, 785, 785, 785, 785, 784, 784, 784, 784, 784, 784, 784, 784, 784, 783, 783, 783, 783, 783, 783, 783, 783, 782, 782, 782, 782, 782, 782, 782, 782, 782, 781, 781, 781, 781, 781, 781, 781, 781, 781, 780, 780, 780, 780, 780, 780, 780, 780, 779, 779, 779, 779, 779, 779, 779, 779, 779, 778, 778, 778, 778, 778, 778, 778, 778, 777, 777, 777, 777, 777, 777, 776, 776, 776, 776, 776, 776, 775, 775, 775, 775, 775, 775, 774, 774, 774, 774, 774, 774, 773, 773, 773, 773, 773};
}; // namespace __cob_manager_internal

__cob_manager_internal::p_row_t operator "" _r(unsigned long long x) { return x; }
__cob_manager_internal::pd_row_t operator "" _r(long double x) { return x; }
__cob_manager_internal::col_t operator "" _c(long double x) { return x; }
__cob_manager_internal::col_t operator "" _c(unsigned long long x) { return x; }

enum BehaviorWhenFail {
 ? ?RAISE_ERROR,
 ? ?WAIT_UNTIL_RECOVER
};

struct PDArgs {
 ? ?int t1, t2;
 ? ?__cob_manager_internal::pd_row_t rows;
 ? ?__cob_manager_internal::col_t col1 = 9, col2 = -1;
};

class CobManager {
 ? ?private:
 ? ?PaoOperator po;
 ? ?std::vector<__cob_manager_internal::cob_list_t> cob_list;
 ? ?int last_refresh_cob_list = 0;
 ? ?using rc_t = std::pair<int, float>;
 ? ?using p_row_t = __cob_manager_internal::p_row_t;
 ? ?using pd_row_t = __cob_manager_internal::pd_row_t;
 ? ?using col_t = __cob_manager_internal::col_t;

 ? ?void refresh_cob_list() {
 ? ? ? ?InsertOperation([this](){
 ? ? ? ? ? ?if(last_refresh_cob_list == GetMainObject()->gameClock())
 ? ? ? ? ? ? ? ?return;
 ? ? ? ? ? ?InsertGuard _(false);
 ? ? ? ? ? ?std::set<Grid> fixed_cobs;
 ? ? ? ? ? ?for(auto c : cob_list)
 ? ? ? ? ? ? ? ?if(c.is_pos())
 ? ? ? ? ? ? ? ? ? ?fixed_cobs.insert(c.pos);
 ? ? ? ? ? ?std::set<int> cobs[8];
 ? ? ? ? ? ?for(auto& p : alive_plant_filter)
 ? ? ? ? ? ? ? ?if(p.type() == COB_CANNON)
 ? ? ? ? ? ? ? ? ? ?cobs[p.col()].insert(p.row() + 1);
 ? ? ? ? ? ?std::vector<Grid> pao_list;
 ? ? ? ? ? ?for(auto c : cob_list)
 ? ? ? ? ? ? ? ?if(c.is_pos() && cobs[c.pos.col - 1].count(c.pos.row))
 ? ? ? ? ? ? ? ? ? ?pao_list.push_back(c.pos);
 ? ? ? ? ? ? ? ?else if(c.is_col())
 ? ? ? ? ? ? ? ? ? ?for(auto r : cobs[c.col - 1])
 ? ? ? ? ? ? ? ? ? ? ? ?if(!fixed_cobs.count({r, c.col}))
 ? ? ? ? ? ? ? ? ? ? ? ? ? ?pao_list.push_back({r, c.col});
 ? ? ? ? ? ?po.resetPaoList(pao_list);
 ? ? ? ? ? ?last_refresh_cob_list = GetMainObject()->gameClock();
 ? ? ? ?});
 ? ?}

 ? ?int get_launch_time(int row) {
 ? ? ? ?if(RangeIn(GetMainObject()->scene(), {4, 5}))
 ? ? ? ? ? ?return 387;
 ? ? ? ?else if(RangeIn(GetMainObject()->scene(), {2, 3}))
 ? ? ? ? ? ?return RangeIn(row, {3, 4}) ? 378 : 373;
 ? ? ? ?else
 ? ? ? ? ? ?return 373;
 ? ?}

 ? ?void launch(int row, col_t col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?refresh_cob_list();
 ? ? ? ?if(b == WAIT_UNTIL_RECOVER)
 ? ? ? ? ? ?po.recoverPao(row, col);
 ? ? ? ?else if(RangeIn(GetMainObject()->scene(), {4, 5}))
 ? ? ? ? ? ?po.roofPao(row, col);
 ? ? ? ?else
 ? ? ? ? ? ?po.pao(row, col);
 ? ? ? ?po.setNextPao(1);
 ? ?}

 ? ?std::vector<int> parse_rows(p_row_t rows) {
 ? ? ? ?std::vector<int> result;
 ? ? ? ?while(rows) {
 ? ? ? ? ? ?result.push_back(rows % 10);
 ? ? ? ? ? ?rows /= 10;
 ? ? ? ?}
 ? ? ? ?std::reverse(result.begin(), result.end());
 ? ? ? ?return result;
 ? ?}

 ? ?std::pair<p_row_t, p_row_t> parse_rows_pd(pd_row_t rows) {
 ? ? ? ?unsigned p_rows = rows, n = log10(p_rows) + 1;
 ? ? ? ?auto d_rows = llround((rows - p_rows) * pow(10, 12 - n));
 ? ? ? ?while(d_rows && d_rows % 10 == 0)
 ? ? ? ? ? ?d_rows /= 10;
 ? ? ? ?return {p_rows, d_rows};
 ? ?}

 ? ?void launch_smart_d(int row) {
 ? ? ? ?InsertOperation([=](){
 ? ? ? ? ? ?auto mo = GetMainObject();
 ? ? ? ? ? ?int delay_t = get_launch_time(row);
 ? ? ? ? ? ?int col = 773 - 126;
 ? ? ? ? ? ?if(mo->refreshCountdown() <= 200)
 ? ? ? ? ? ? ? ?col = __cob_manager_internal::garg_x[delay_t - mo->refreshCountdown()] - 126;
 ? ? ? ? ? ?else
 ? ? ? ? ? ? ? ?for(auto& z : alive_zombie_filter)
 ? ? ? ? ? ? ? ? ? ?if(z.type() == GIGA_GARGANTUAR &&
 ? ? ? ? ? ? ? ? ? ? ? z.standState() == mo->wave() &&
 ? ? ? ? ? ? ? ? ? ? ? row - 2 <= z.row() && z.row() <= row) {
 ? ? ? ? ? ? ? ? ? ? ? ?auto x = GetZombieAbscissas((Zombie*)&z - mo->zombieArray(), delay_t);
 ? ? ? ? ? ? ? ? ? ? ? ?if(x.empty())
 ? ? ? ? ? ? ? ? ? ? ? ? ? ?continue;
 ? ? ? ? ? ? ? ? ? ? ? ?col = std::min(col, int(x[delay_t]) - 126);
 ? ? ? ? ? ? ? ? ? ?}
 ? ? ? ? ? ?launch(row, col / 80.0);
 ? ? ? ?});
 ? ?}

 ? ?public:
 ? ?CobManager(const std::vector<__cob_manager_internal::cob_list_t>& c = {1, 2, 3, 4, 5, 6, 7, 8}) : cob_list(c) {}

 ? ?// In Queue
 ? ?void set_cob_list(const std::vector<__cob_manager_internal::cob_list_t>& c) {
 ? ? ? ?InsertOperation([=](){ cob_list = c; });
 ? ?}

 ? ?// In Queue
 ? ?void operator()(p_row_t rows, col_t col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?for(auto r : parse_rows(rows)) {
 ? ? ? ? ? ?Delay(-get_launch_time(r));
 ? ? ? ? ? ?if(col >= 0)
 ? ? ? ? ? ? ? ?launch(r, col, b);
 ? ? ? ? ? ?else
 ? ? ? ? ? ? ? ?launch_smart_d(r);
 ? ? ? ? ? ?Delay(get_launch_time(r));
 ? ? ? ?}
 ? ?}

 ? ?// In Queue
 ? ?void operator()(std::vector<rc_t> rcs, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?for(auto rc : rcs)
 ? ? ? ? ? ?operator()(rc.first, rc.second, b);
 ? ?}

 ? ?// In Queue
 ? ?void p(int time, int wave, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?SetTime(time, wave);
 ? ? ? ?operator()(RangeIn(GetMainObject()->scene(), {2, 3}) ? 25 : 24, 9);
 ? ?}

 ? ?// In Queue
 ? ?void p(int time, int wave, p_row_t rows, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?SetTime(time, wave);
 ? ? ? ?operator()(rows, 9, b);
 ? ?}

 ? ?// In Queue
 ? ?void p(int time, int wave, col_t col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?SetTime(time, wave);
 ? ? ? ?operator()(RangeIn(GetMainObject()->scene(), {2, 3}) ? 25 : 24, col, b);
 ? ?}

 ? ?// In Queue
 ? ?void p(int time, int wave, p_row_t rows, col_t col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?SetTime(time, wave);
 ? ? ? ?operator()(rows, col, b);
 ? ?}

 ? ?// In Queue
 ? ?void d(int time, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?Delay(time);
 ? ? ? ?operator()(RangeIn(GetMainObject()->scene(), {2, 3}) ? 14 : 15, -1, b);
 ? ?}

 ? ?// In Queue
 ? ?void d(int time, p_row_t rows, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?Delay(time);
 ? ? ? ?operator()(rows, -1, b);
 ? ?}

 ? ?// In Queue
 ? ?void d(int time, col_t col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?Delay(time);
 ? ? ? ?operator()(RangeIn(GetMainObject()->scene(), {2, 3}) ? 14 : 15, col, b);
 ? ?}

 ? ?// In Queue
 ? ?void d(int time, p_row_t rows, col_t col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?Delay(time);
 ? ? ? ?operator()(rows, col, b);
 ? ?}

 ? ?void pd(int t_, int d_, int wave, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?p(t_, wave, b);
 ? ? ? ?d(d_, b);
 ? ?}

 ? ?void pd(int t_, int d_, int wave, pd_row_t rows, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?auto rows_ = parse_rows_pd(rows);
 ? ? ? ?p(t_, wave, rows_.first, b);
 ? ? ? ?d(d_, rows_.second, b);
 ? ?}

 ? ?void pd(int t_, int d_, int wave, col_t col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?p(t_, wave, b);
 ? ? ? ?d(d_, col, b);
 ? ?}

 ? ?void pd(int t_, int d_, int wave, std::pair<float, float> col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?p(t_, wave, col.first, b);
 ? ? ? ?d(d_, col.second, b);
 ? ?}

 ? ?void pd(int t_, int d_, int wave, pd_row_t rows, col_t col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?auto rows_ = parse_rows_pd(rows);
 ? ? ? ?p(t_, wave, rows_.first, b);
 ? ? ? ?d(d_, rows_.second, col, b);
 ? ?}

 ? ?void pd(int t_, int d_, int wave, pd_row_t rows, std::pair<float, float> col, BehaviorWhenFail b = RAISE_ERROR) {
 ? ? ? ?auto rows_ = parse_rows_pd(rows);
 ? ? ? ?p(t_, wave, rows_.first, col.first, b);
 ? ? ? ?d(d_, rows_.second, col.second, b);
 ? ?}
};
// ----------------------------------------------------------------

using namespace std;
using namespace AvZ;

MainObject* mo;
class : GlobalVar {
 ? ?public:
 ? ?void beforeScript() {
 ? ? ? ?mo = GetMainObject();
 ? ?}
} __mo_manager;

template <class... Ts>
void Card(const vector<PlantType>& plant_types, Ts... args) {
 ? ?for(auto p: plant_types)
 ? ? ? ?Card(p, args...);
}

void Card(vector<PlantType> plant_types, vector<Position> args) {
 ? ?for(auto p: plant_types)
 ? ? ? ?Card(p, args);
}

void I() {
 ? ?Delay(-299); ice_filler.coffee();
 ? ?Delay(299); SetPlantActiveTime(ICE_SHROOM, -1);
}

CobManager wind({1, 7}), norm({7});

/*
對ch6: PP|IPP-PP*|PP|IPP-PA|PP|IPP-N
w1: PP|IPP-N 起手
w10: PPA|IPP-PP|PP|IPP-N 起手
*/

void Script() {
 ? ?OpenMultipleEffective('Q', MAIN_UI);
 ? ?SetZombies({ZOMBIE, POLE_VAULTING_ZOMBIE, ZOMBONI, POGO_ZOMBIE, JACK_IN_THE_BOX_ZOMBIE, FOOTBALL_ZOMBIE, LADDER_ZOMBIE, CATAPULT_ZOMBIE, GARGANTUAR, GIGA_GARGANTUAR});
 ? ?cresc::AverageSpawn({ZOMBIE, POLE_VAULTING_ZOMBIE, ZOMBONI, POGO_ZOMBIE, JACK_IN_THE_BOX_ZOMBIE, LADDER_ZOMBIE, FOOTBALL_ZOMBIE, CATAPULT_ZOMBIE, GARGANTUAR, GIGA_GARGANTUAR});
 ? ?SelectCards({ICE_SHROOM, M_ICE_SHROOM, COFFEE_BEAN, DOOM_SHROOM, CHERRY_BOMB, SQUASH, FLOWER_POT, KERNEL_PULT, COB_CANNON, WALL_NUT});
 ? ?SetTime(-599, 1);
 ? ?ice_filler.start({{1, 6}, {2, 6}, {3, 6}, {4, 6}, {5, 6}});
 ? ?WriteMemory(0.36f, 0x679670);

 ? ?for(int w = 1; w <= 19; w++)
 ? ? ? ?if(w % 2 != w / 10) {
 ? ? ? ? ? ?SetTime(288, w); wind(2, 9); norm(4, 9);
 ? ? ? ? ? ?SetTime(601 + 113, w); I();
 ? ? ? ? ? ?Delay(1); wind(2, 8.6); norm(4, 8.6);
 ? ? ? ?}
 ? ?SetTime(401 - 100, 10); Card({FLOWER_POT, CHERRY_BOMB}, 4, 9);

 ? ?for(int w : {4, 15}) {
 ? ? ? ?SetTime(937, w); wind(2, 9); norm(4, 9);
 ? ? ? ?int r = (w == 4) ? 5 : 2;
 ? ? ? ?Delay(-359 + 205); Shovel(r, 7); Card(KERNEL_PULT, r, 7);
 ? ? ? ?Delay(751); Card({FLOWER_POT, KERNEL_PULT}, r, 8); Card(COB_CANNON, r, 7);
 ? ?}

 ? ?for(int w : {6, 17}) {
 ? ? ? ?SetTime(937, w); norm(4, 9);
 ? ? ? ?Delay(-99); Card({FLOWER_POT, CHERRY_BOMB}, 1, 9);
 ? ?}
 ? ?SetTime(937, 11); wind(2, 9); norm(4, 9);


 ? ?SetTime(937 - 298, 2); Card({FLOWER_POT, DOOM_SHROOM, COFFEE_BEAN}, 3, 9);
 ? ?for(int w : {8, 13, 19}) {
 ? ? ? ?SetTime(937 - 266, w); Card({FLOWER_POT, DOOM_SHROOM, COFFEE_BEAN}, {{2, 9}, {4, 9}, {5, 9}});
 ? ?}
 ? ?SetTime(937, 19); wind(2, 9);

 ? ?SetTime(601 + 937, 9); wind(2, 9); norm(4, 9);
 ? ?SetTime(2026); wind(2, 9); norm(4, 9);

 ? ?SetTime(1137 + 288, 19); wind(2, 9); norm(4, 9);
 ? ?Delay(1); Card({FLOWER_POT, SQUASH}, 1, 9);
 ? ?SetTime(2000); I();
 ? ?SetTime(1137 + 714); norm(4, 9);
 ? ?SetTime(2758); wind(2, 9); norm(4, 9);

 ? ?InsertTimeOperation(-1, 20, [](){ WriteMemory(0.23f, 0x679670); });
 ? ?SetTime(0); DanceCheat(DanceCheatMode::SLOW);
 ? ?SetTime(200); I();
 ? ?SetTime(201); wind(24, 9.5);
 ? ?SetTime(976); wind(2, 9); norm(4, 9);
 ? ?SetTime(1626); norm(14, 8.0625); wind({{2, 8.0625}, {5, 9}});
}