jakub
6a667d8deb
Include the current addon, terrain integrations, tools, regression tests and recorded development evidence.
43 lines
2.8 KiB
Python
43 lines
2.8 KiB
Python
"""Run the production recursive cargo-seat collector against aliased native layouts."""
|
|
from policy_harness import ROOT, body, run
|
|
source = ROOT / 'addons/core/Scripts/Game/ERR/10_AI/ERR_RescueFlightComponent.c'
|
|
s = '\n'.join(body(source, name) for name in ('protected void CargoSeats(', 'protected int CargoCapacity('))
|
|
s = s.replace('protected ', '').replace('IEntity entity', 'IEntity* entity').replace('IEntity child', 'IEntity* child')
|
|
s = s.replace('array<BaseCompartmentSlot> seats)', 'std::vector<BaseCompartmentSlot*>& seats)')
|
|
s = s.replace('array<BaseCompartmentSlot>', 'std::vector<BaseCompartmentSlot*>')
|
|
s = s.replace('BaseCompartmentManagerComponent manager = BaseCompartmentManagerComponent.Cast(entity.FindComponent(BaseCompartmentManagerComponent));', 'auto* manager = entity->manager;')
|
|
s = s.replace('foreach (BaseCompartmentSlot slot : slots)', 'for (auto* slot : slots)')
|
|
s = s.replace('!seats.Contains(slot)', 'std::find(seats.begin(), seats.end(), slot) == seats.end()').replace('seats.Insert(slot)', 'seats.push_back(slot)').replace('seats.Count()', 'seats.size()')
|
|
for name in ('entity', 'manager', 'child', 'slot'):
|
|
s = s.replace(name + '.', name + '->')
|
|
s = s.replace('ECompartmentType.', 'ECompartmentType::')
|
|
run(r'''
|
|
#include <cassert>
|
|
#include <algorithm>
|
|
#include <vector>
|
|
namespace ECompartmentType { enum {PILOT,CARGO,TURRET}; }
|
|
struct BaseCompartmentSlot {
|
|
int type=ECompartmentType::CARGO; bool accessible=true,occupied=false,reserved=false,locked=false;
|
|
int GetType(){return type;} bool IsCompartmentAccessible(){return accessible;}
|
|
bool IsOccupied(){return occupied;} bool IsReserved(){return reserved;} bool IsGetInLocked(){return locked;}
|
|
};
|
|
struct Manager {std::vector<BaseCompartmentSlot*> slots;void GetCompartments(std::vector<BaseCompartmentSlot*>& out){out=slots;}};
|
|
struct IEntity {Manager* manager=nullptr;IEntity* child=nullptr;IEntity* sibling=nullptr;IEntity* GetChildren(){return child;}IEntity* GetSibling(){return sibling;}};
|
|
struct Actual {
|
|
''' + s + r'''
|
|
};
|
|
int main(){
|
|
BaseCompartmentSlot first,second,blocked,taken,reserved,locked,pilot;
|
|
blocked.accessible=false;taken.occupied=true;reserved.reserved=true;locked.locked=true;pilot.type=ECompartmentType::PILOT;
|
|
Manager rootSeats{{&first,&second,nullptr,&blocked,&taken,&reserved,&locked,&pilot}};
|
|
Manager benchSeats{{&first,&second}};
|
|
IEntity bench{&benchSeats},root{&rootSeats,&bench};
|
|
Actual actual;
|
|
assert(actual.CargoCapacity(&root)==2); // Parent and child expose the same native seats.
|
|
second.occupied=true;assert(actual.CargoCapacity(&root)==1);
|
|
first.locked=true;assert(actual.CargoCapacity(&root)==0);
|
|
IEntity empty;assert(actual.CargoCapacity(&empty)==0);
|
|
}
|
|
''')
|
|
print('PASS: unique usable cargo seats; duplicate parent/child slots, crew, occupied, reserved and locked seats excluded')
|