|
|
|
@ -18,12 +18,78 @@
|
|
|
|
|
|
|
|
|
|
#include "../../safeguards.h"
|
|
|
|
|
|
|
|
|
|
ScriptVehicleList::ScriptVehicleList()
|
|
|
|
|
ScriptVehicleList::ScriptVehicleList(HSQUIRRELVM vm)
|
|
|
|
|
{
|
|
|
|
|
EnforceDeityOrCompanyModeValid_Void();
|
|
|
|
|
|
|
|
|
|
int nparam = sq_gettop(vm) - 1;
|
|
|
|
|
if (nparam >= 1) {
|
|
|
|
|
/* Make sure the filter function is really a function, and not any
|
|
|
|
|
* other type. It's parameter 2 for us, but for the user it's the
|
|
|
|
|
* first parameter they give. */
|
|
|
|
|
SQObjectType valuator_type = sq_gettype(vm, 2);
|
|
|
|
|
if (valuator_type != OT_CLOSURE && valuator_type != OT_NATIVECLOSURE) {
|
|
|
|
|
throw sq_throwerror(vm, "parameter 1 has an invalid type (expected function)");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Push the function to call */
|
|
|
|
|
sq_push(vm, 2);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Don't allow docommand from a Valuator, as we can't resume in
|
|
|
|
|
* mid C++-code. */
|
|
|
|
|
bool backup_allow = ScriptObject::GetAllowDoCommand();
|
|
|
|
|
ScriptObject::SetAllowDoCommand(false);
|
|
|
|
|
|
|
|
|
|
for (const Vehicle *v : Vehicle::Iterate()) {
|
|
|
|
|
if ((v->owner == ScriptObject::GetCompany() || ScriptCompanyMode::IsDeity()) && (v->IsPrimaryVehicle() || (v->type == VEH_TRAIN && ::Train::From(v)->IsFreeWagon()))) this->AddItem(v->index);
|
|
|
|
|
if (v->owner != ScriptObject::GetCompany() && !ScriptCompanyMode::IsDeity()) continue;
|
|
|
|
|
if (!v->IsPrimaryVehicle() && !(v->type == VEH_TRAIN && ::Train::From(v)->IsFreeWagon())) continue;
|
|
|
|
|
|
|
|
|
|
if (nparam < 1) {
|
|
|
|
|
/* No filter, just add the item. */
|
|
|
|
|
this->AddItem(v->index);
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Push the root table as instance object, this is what squirrel does for meta-functions. */
|
|
|
|
|
sq_pushroottable(vm);
|
|
|
|
|
/* Push all arguments for the valuator function. */
|
|
|
|
|
sq_pushinteger(vm, v->index);
|
|
|
|
|
for (int i = 0; i < nparam - 1; i++) {
|
|
|
|
|
sq_push(vm, i + 3);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Call the function. Squirrel pops all parameters and pushes the return value. */
|
|
|
|
|
if (SQ_FAILED(sq_call(vm, nparam + 1, SQTrue, SQTrue))) {
|
|
|
|
|
ScriptObject::SetAllowDoCommand(backup_allow);
|
|
|
|
|
throw sq_throwerror(vm, "failed to run filter");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Retrieve the return value */
|
|
|
|
|
switch (sq_gettype(vm, -1)) {
|
|
|
|
|
case OT_BOOL: {
|
|
|
|
|
SQBool add;
|
|
|
|
|
sq_getbool(vm, -1, &add);
|
|
|
|
|
if (add) this->AddItem(v->index);
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
default: {
|
|
|
|
|
ScriptObject::SetAllowDoCommand(backup_allow);
|
|
|
|
|
throw sq_throwerror(vm, "return value of filter is not valid (not bool)");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* Pop the return value. */
|
|
|
|
|
sq_poptop(vm);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (nparam >= 1) {
|
|
|
|
|
/* Pop the filter function */
|
|
|
|
|
sq_poptop(vm);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ScriptObject::SetAllowDoCommand(backup_allow);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ScriptVehicleList_Station::ScriptVehicleList_Station(StationID station_id)
|
|
|
|
|