67 lines
1.9 KiB
C++
67 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include "LuaTools.h"
|
|
#include "MiscUtils.h"
|
|
|
|
#include "df/world.h"
|
|
#include "df/creature_raw.h"
|
|
#include "df/plant_raw.h"
|
|
|
|
// Utility Functions {{{
|
|
// A set of convenience functions for doing common lookups
|
|
|
|
bool call_stockpiles_lua(DFHack::color_ostream* out, const char* fn_name,
|
|
int nargs = 0, int nres = 0,
|
|
DFHack::Lua::LuaLambda&& args_lambda = DFHack::Lua::DEFAULT_LUA_LAMBDA,
|
|
DFHack::Lua::LuaLambda&& res_lambda = DFHack::Lua::DEFAULT_LUA_LAMBDA);
|
|
|
|
/**
|
|
* Retrieve creature raw from index
|
|
*/
|
|
static inline df::creature_raw* find_creature(int32_t idx) {
|
|
return df::global::world->raws.creatures.all[idx];
|
|
}
|
|
|
|
/**
|
|
* Retrieve creature index from id string
|
|
* @return -1 if not found
|
|
*/
|
|
static inline int16_t find_creature(const std::string& creature_id) {
|
|
return linear_index(df::global::world->raws.creatures.all, &df::creature_raw::creature_id, creature_id);
|
|
}
|
|
|
|
/**
|
|
* Retrieve plant raw from index
|
|
*/
|
|
static inline df::plant_raw* find_plant(size_t idx) {
|
|
return df::global::world->raws.plants.all[idx];
|
|
}
|
|
|
|
/**
|
|
* Retrieve plant index from id string
|
|
* @return -1 if not found
|
|
*/
|
|
static inline size_t find_plant(const std::string& plant_id) {
|
|
return linear_index(df::global::world->raws.plants.all, &df::plant_raw::id, plant_id);
|
|
}
|
|
|
|
struct less_than_no_case {
|
|
bool operator () (char x, char y) const {
|
|
return toupper(static_cast<unsigned char>(x)) < toupper(static_cast<unsigned char>(y));
|
|
}
|
|
};
|
|
|
|
static inline bool CompareNoCase(const std::string& a, const std::string& b) {
|
|
return std::lexicographical_compare(a.begin(), a.end(), b.begin(), b.end(), less_than_no_case());
|
|
}
|
|
|
|
/**
|
|
* Checks if the parameter has the dfstock extension.
|
|
* Doesn't check if the file exists or not.
|
|
*/
|
|
static inline bool is_dfstockfile(const std::string& filename) {
|
|
return filename.rfind(".dfstock") != std::string::npos;
|
|
}
|
|
|
|
// }}} utility Functions
|