91 lines
2.9 KiB
C++
91 lines
2.9 KiB
C++
#include "plannedbuilding.h"
|
|
#include "buildingplan.h"
|
|
|
|
#include "Debug.h"
|
|
#include "MiscUtils.h"
|
|
|
|
#include "modules/World.h"
|
|
|
|
#include "df/job.h"
|
|
|
|
namespace DFHack {
|
|
DBG_EXTERN(buildingplan, status);
|
|
}
|
|
|
|
using std::string;
|
|
using std::vector;
|
|
using namespace DFHack;
|
|
|
|
static vector<vector<df::job_item_vector_id>> get_vector_ids(color_ostream &out, int bld_id) {
|
|
vector<vector<df::job_item_vector_id>> ret;
|
|
|
|
df::building *bld = df::building::find(bld_id);
|
|
|
|
if (!bld || bld->jobs.size() != 1)
|
|
return ret;
|
|
|
|
auto &job = bld->jobs[0];
|
|
for (auto &jitem : job->job_items) {
|
|
ret.emplace_back(getVectorIds(out, jitem));
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static vector<vector<df::job_item_vector_id>> deserialize(color_ostream &out, PersistentDataItem &bld_config) {
|
|
vector<vector<df::job_item_vector_id>> ret;
|
|
|
|
DEBUG(status,out).print("deserializing state for building %d: %s\n",
|
|
get_config_val(bld_config, BLD_CONFIG_ID), bld_config.val().c_str());
|
|
|
|
vector<string> joined;
|
|
split_string(&joined, bld_config.val(), "|");
|
|
for (auto &str : joined) {
|
|
vector<string> lst;
|
|
split_string(&lst, str, ",");
|
|
vector<df::job_item_vector_id> ids;
|
|
for (auto &s : lst)
|
|
ids.emplace_back(df::job_item_vector_id(string_to_int(s)));
|
|
ret.emplace_back(ids);
|
|
}
|
|
|
|
if (!ret.size())
|
|
ret = get_vector_ids(out, get_config_val(bld_config, BLD_CONFIG_ID));
|
|
|
|
return ret;
|
|
}
|
|
|
|
static string serialize(const vector<vector<df::job_item_vector_id>> &vector_ids) {
|
|
vector<string> joined;
|
|
for (auto &vec_list : vector_ids) {
|
|
joined.emplace_back(join_strings(",", vec_list));
|
|
}
|
|
return join_strings("|", joined);
|
|
}
|
|
|
|
PlannedBuilding::PlannedBuilding(color_ostream &out, df::building *building)
|
|
: id(building->id), vector_ids(get_vector_ids(out, id)) {
|
|
DEBUG(status,out).print("creating persistent data for building %d\n", id);
|
|
bld_config = World::AddPersistentData(BLD_CONFIG_KEY);
|
|
set_config_val(bld_config, BLD_CONFIG_ID, id);
|
|
bld_config.val() = serialize(vector_ids);
|
|
DEBUG(status,out).print("serialized state for building %d: %s\n", id, bld_config.val().c_str());
|
|
}
|
|
|
|
PlannedBuilding::PlannedBuilding(color_ostream &out, PersistentDataItem &bld_config)
|
|
: id(get_config_val(bld_config, BLD_CONFIG_ID)),
|
|
vector_ids(deserialize(out, bld_config)),
|
|
bld_config(bld_config) { }
|
|
|
|
// Ensure the building still exists and is in a valid state. It can disappear
|
|
// for lots of reasons, such as running the game with the buildingplan plugin
|
|
// disabled, manually removing the building, modifying it via the API, etc.
|
|
df::building * PlannedBuilding::getBuildingIfValidOrRemoveIfNot(color_ostream &out) {
|
|
auto bld = df::building::find(id);
|
|
bool valid = bld && bld->getBuildStage() == 0;
|
|
if (!valid) {
|
|
remove(out);
|
|
return NULL;
|
|
}
|
|
return bld;
|
|
}
|