Add HA filter

Admiral-Fish · Mar 9, 2021 · 04bb0d9 · 04bb0d9
1 parent 30de903
commit 04bb0d9
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 0 deletions.
diff --git a/Source/Forms/Controls/Filter.cpp b/Source/Forms/Controls/Filter.cpp
@@ -187,3 +187,8 @@ void Filter::setGenderRatio(u8 genderRatio)
 {
     ui->comboBoxGenderRatio->setCurrentIndex(ui->comboBoxGenderRatio->findData(genderRatio));
 }
+
+void Filter::enableHiddenAbility()
+{
+    ui->comboBoxAbility->addItem("H", 2);
+}
diff --git a/Source/Forms/Controls/Filter.hpp b/Source/Forms/Controls/Filter.hpp
@@ -66,6 +66,7 @@ class Filter  : public QWidget
     bool getDisableFilters();
     void disableControls(u16 control);
     void setGenderRatio(u8 genderRatio);
+    void enableHiddenAbility();
 
 private:
     Ui::Filter *ui;

diff --git a/Source/Forms/Gen5/DreamRadar.cpp b/Source/Forms/Gen5/DreamRadar.cpp
@@ -106,6 +106,9 @@ void DreamRadar::setupModels()
     ui->comboBoxSearcherSpecies5->addItem(tr("None"), 0);
     ui->comboBoxSearcherSpecies6->addItem(tr("None"), 0);
 
+    ui->filterGenerator->enableHiddenAbility();
+    ui->filterSearcher->enableHiddenAbility();
+
     std::vector<u16> species
         = { 641, 642, 645, 483, 484, 487, 249, 250, 79, 120, 137, 163, 174, 175, 213, 238, 280, 333, 425, 436, 442, 447, 479, 517, 561 };
     std::vector<std::string> names = Translator::getSpecies(species);

diff --git a/Source/Forms/Gen5/Eggs5.cpp b/Source/Forms/Gen5/Eggs5.cpp
@@ -102,6 +102,9 @@ void Eggs5::setupModels()
     ui->eggSettingsGenerator->setup(static_cast<Game>(Game::BW | Game::BW2));
     ui->eggSettingsSearcher->setup(static_cast<Game>(Game::BW | Game::BW2));
 
+    ui->filterGenerator->enableHiddenAbility();
+    ui->filterSearcher->enableHiddenAbility();
+
     QAction *outputTXTGenerator = generatorMenu->addAction(tr("Output Results to TXT"));
     QAction *outputCSVGenerator = generatorMenu->addAction(tr("Output Results to CSV"));
     connect(outputTXTGenerator, &QAction::triggered, [=]() { ui->tableViewGenerator->outputModel(false); });

diff --git a/Source/Forms/Gen5/Event5.cpp b/Source/Forms/Gen5/Event5.cpp
@@ -111,6 +111,9 @@ void Event5::setupModels()
     ui->filterGenerator->disableControls(Controls::EncounterSlots);
     ui->filterSearcher->disableControls(Controls::EncounterSlots | Controls::DisableFilter | Controls::UseDelay);
 
+    ui->filterGenerator->enableHiddenAbility();
+    ui->filterSearcher->enableHiddenAbility();
+
     std::vector<u16> species(649);
     std::iota(species.begin(), species.end(), 1);
     for (const std::string &specie : Translator::getSpecies(species))