MetzNet
/
dfhack
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rework autofarm persistence

develop
Kelly Kinkade 2023-01-06 23:45:16 -06:00
parent f0aa7a76de
commit e649255c8e
1 changed files with 43 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
plugins/autofarm.cpp
Unescape Escape View File

@ -348,32 +348,42 @@ public:
out << std::flush; out << std::flush;
} }
private:
PersistentDataItem cfg_default_threshold;
PersistentDataItem cfg_enabled;
public:
void load_state(color_ostream& out) void load_state(color_ostream& out)
{ {
initialize(); initialize();
PersistentDataItem cfg_default_threshold = World::GetPersistentData("autofarm/default_threshold"); cfg_enabled = World::GetPersistentData("autofarm/enabled");
if (cfg_enabled.isValid())
enabled = cfg_enabled.ival(0) != 0;
else
cfg_enabled.ival(0) = enabled;
cfg_default_threshold = World::GetPersistentData("autofarm/default_threshold");
if (cfg_default_threshold.isValid()) if (cfg_default_threshold.isValid())
defaultThreshold = cfg_default_threshold.ival(0); defaultThreshold = cfg_default_threshold.ival(0);
else
cfg_default_threshold.ival(0) = defaultThreshold;
PersistentDataItem cfg_num_thresholds = World::GetPersistentData("autofarm/num_threshold"); std::vector<PersistentDataItem> items;
if (cfg_num_thresholds.isValid()) World::GetPersistentData(&items, "autofarm/threshold/", true);
{ for (auto& i: items) {
int n = cfg_num_thresholds.ival(0); if (i.isValid())
for (int i = 0; i < n; i++)
{
const std::string keyName = "autofarm/threshold/" + std::to_string(i);
PersistentDataItem cfg_threshold = World::GetPersistentData(keyName);
if (cfg_threshold.isValid())
{ {
const auto allPlants = world->raws.plants.all; const auto allPlants = world->raws.plants.all;
const std::string id = cfg_threshold.val(); const std::string id = i.val();
const int val = cfg_threshold.ival(0); const int val = i.ival(0);
const auto plant = std::find_if(std::begin(allPlants), std::end(allPlants), [id](df::plant_raw* p) { return p->id == id; }); const auto plant = std::find_if(std::begin(allPlants), std::end(allPlants), [id](df::plant_raw* p) { return p->id == id; });
if (plant != std::end(allPlants)) if (plant != std::end(allPlants))
{ {
setThreshold((*plant)->index, val); setThreshold((*plant)->index, val);
INFO(config, out).print("threshold of %d for unknown plant %s in saved configuration loaded\n", val, id); INFO(config, out).print("threshold of %d for plant %s in saved configuration loaded\n", val, id);
} }
else else
{ {
@ -381,25 +391,27 @@ public:
} }
} }
} }
}
PersistentDataItem cfg_enabled = World::GetPersistentData("autofarm/enabled");
enabled = (cfg_enabled.isValid() && cfg_enabled.ival(0) != 0);
} }
void save_state(color_ostream& out) void save_state(color_ostream& out)
{ {
World::AddPersistentData("autofarm/default_threshold").ival(0) = defaultThreshold; cfg_default_threshold.ival(0) = defaultThreshold;
World::AddPersistentData("autofarm/num_threshold").ival(0) = thresholds.size(); cfg_enabled.ival(0) = enabled;
std::vector<PersistentDataItem> items;
World::GetPersistentData(&items, "autofarm/threshold/", true);
for (auto& i : items)
World::DeletePersistentData(i);
for (const auto& t : thresholds) for (const auto& t : thresholds)
{ {
const std::string& plantName = world->raws.plants.all[t.first]->id; const std::string& plantID = world->raws.plants.all[t.first]->id;
const std::string keyName = "autofarm/threshold/" + plantName; const std::string keyName = "autofarm/threshold/" + plantID;
PersistentDataItem cfgThreshold = World::AddPersistentData(keyName); PersistentDataItem cfgThreshold = World::AddPersistentData(keyName);
cfgThreshold.val() = plantName; cfgThreshold.val() = plantID;
cfgThreshold.ival(0) = t.second; cfgThreshold.ival(0) = t.second;
} }
World::AddPersistentData("autofarm/enabled").ival(0) = enabled;
} }
}; };
@ -416,6 +428,7 @@ DFhackCExport command_result plugin_init(color_ostream& out, std::vector <Plugin
autofarm)); autofarm));
} }
autofarmInstance = std::move(dts::make_unique<AutoFarm>()); autofarmInstance = std::move(dts::make_unique<AutoFarm>());
autofarmInstance->load_state(out);
return CR_OK; return CR_OK;
} }