Anton Tarasenko
2 years ago
3 changed files with 271 additions and 0 deletions
@ -0,0 +1,80 @@
|
||||
/** |
||||
* Iterator for going through all the entities inside the game world. |
||||
* Copyright 2022 Anton Tarasenko |
||||
*------------------------------------------------------------------------------ |
||||
* This file is part of Acedia. |
||||
* |
||||
* Acedia is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* Acedia is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with Acedia. If not, see <https://www.gnu.org/licenses/>. |
||||
*/ |
||||
class EntityIterator extends Iter |
||||
abstract; |
||||
|
||||
/** |
||||
* Returns only `EPlaceable` interfaces for world entities. |
||||
* |
||||
* Resulting `EPlaceable` can refer to now non-existing entities if they were |
||||
* destroyed after the start of iteration. |
||||
*/ |
||||
public function AcediaObject Get() { return none; } |
||||
|
||||
/** |
||||
* Returns `EPlaceable` caller `EntityIterator` is currently at. |
||||
* Guaranteed to be not `none` as long as iteration hasn't finished. |
||||
* |
||||
* Resulting `EPlaceable` can refer to now non-existing entities if they were |
||||
* destroyed after the start of iteration. |
||||
* |
||||
* @return `EPlaceable` caller `EntityIterator` is currently at. |
||||
*/ |
||||
public function EPlaceable GetPlaceable(); |
||||
|
||||
/** |
||||
* Returns `EPlaceable` caller `EntityIterator` is currently at as `EPawn`, |
||||
* assuming that its entity support that interface. |
||||
* |
||||
* Resulting `EPawn` can refer to now non-existing entities if they were |
||||
* destroyed after the start of iteration. |
||||
* |
||||
* @return `EPawn` interface for `EPlaceable` that `Get()` would have returned. |
||||
* If `EPawn` is not supported by that `EPlaceable` - returns `none`. |
||||
*/ |
||||
public function EPawn GetPawn(); |
||||
|
||||
/** |
||||
* Makes caller iterator skip any entities that do not support `EPawn` |
||||
* interface during iteration. |
||||
* |
||||
* @return Reference to caller `EntityIterator` to allow for method chaining. |
||||
*/ |
||||
public function EntityIterator LeaveOnlyPawns(); |
||||
|
||||
/** |
||||
* Makes caller iterator skip any entities that are not visible in the game |
||||
* world. |
||||
* |
||||
* @return Reference to caller `EntityIterator` to allow for method chaining. |
||||
*/ |
||||
public function EntityIterator LeaveOnlyPlaceables(); |
||||
|
||||
/** |
||||
* Makes caller iterator skip any entities that are not visible in the game |
||||
* world. |
||||
* |
||||
* @return Reference to caller `EntityIterator` to allow for method chaining. |
||||
*/ |
||||
public function EntityIterator LeaveOnlyVisible(); |
||||
|
||||
defaultproperties |
||||
{ |
||||
} |
@ -0,0 +1,184 @@
|
||||
/** |
||||
* `EntityIterator` implementation for `KF1_Frontend`. |
||||
* Copyright 2022 Anton Tarasenko |
||||
*------------------------------------------------------------------------------ |
||||
* This file is part of Acedia. |
||||
* |
||||
* Acedia is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU General Public License as published by |
||||
* the Free Software Foundation, version 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* Acedia is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU General Public License |
||||
* along with Acedia. If not, see <https://www.gnu.org/licenses/>. |
||||
*/ |
||||
class KF1_EntityIterator extends EntityIterator; |
||||
|
||||
var private bool initialized; |
||||
|
||||
var private int currentIndex; |
||||
// Simply store all traced `Actor`s here at the moment of user first |
||||
// interacting with iterator's items: when either `Next()` or one of |
||||
// the `Get...()` methods were called. |
||||
var private array<EPlaceable> foundActors; |
||||
// Did we already perform iteration through actors? |
||||
var private bool iterated; |
||||
|
||||
// Iterator filters |
||||
var private bool onlyPawns; |
||||
var private bool onlyPlaceables; |
||||
var private bool onlyVisible; |
||||
|
||||
protected function Finalizer() |
||||
{ |
||||
_.memory.FreeMany(foundActors); |
||||
foundActors.length = 0; |
||||
initialized = false; |
||||
} |
||||
|
||||
/** |
||||
* Initializes `TracingIterator` that traces entities between `start` and |
||||
* `end` positions, in order starting from the `start` |
||||
*/ |
||||
public final function Initialize() |
||||
{ |
||||
initialized = true; |
||||
} |
||||
|
||||
private final function bool IsActorVisible(Actor actorToCheck) |
||||
{ |
||||
if (actorToCheck == none) return false; |
||||
if (actorToCheck.bHidden && !actorToCheck.bWorldGeometry) return false; |
||||
if (actorToCheck.drawType == DT_None) return false; |
||||
|
||||
return true; |
||||
} |
||||
|
||||
// Does actual tracing, but only once per iterator's lifecycle. |
||||
// Assumes `initialized` is `true`. |
||||
private final function TryIterating() |
||||
{ |
||||
local Pawn nextPawn; |
||||
local Actor nextActor; |
||||
local class<Actor> targetClass; |
||||
local ServerLevelCore core; |
||||
|
||||
// Checking `initialized` flag is already done by every method that |
||||
// calls `TryTracing()` |
||||
if (iterated) { |
||||
return; |
||||
} |
||||
currentIndex = 0; |
||||
if (onlyPawns) { |
||||
targetClass = class'Pawn'; |
||||
} |
||||
else { |
||||
targetClass = class'Actor'; |
||||
} |
||||
core = ServerLevelCore(class'ServerLevelCore'.static.GetInstance()); |
||||
// TODO: We should not always use slow `AllActors()` method |
||||
foreach core.AllActors(targetClass, nextActor) |
||||
{ |
||||
if (onlyVisible && !IsActorVisible(nextActor)) { |
||||
continue; |
||||
} |
||||
nextPawn = Pawn(nextActor); |
||||
if (nextPawn != none) |
||||
{ |
||||
foundActors[foundActors.length] = |
||||
class'EKFPawn'.static.Wrap(nextPawn); |
||||
} |
||||
else { |
||||
foundActors[foundActors.length] = |
||||
class'EKFUnknownPlaceable'.static.Wrap(nextActor); |
||||
} |
||||
} |
||||
iterated = true; |
||||
} |
||||
|
||||
public function Iter Next(optional bool skipNone) |
||||
{ |
||||
if (!initialized) { |
||||
return self; |
||||
} |
||||
TryIterating(); |
||||
currentIndex += 1; |
||||
return self; |
||||
} |
||||
|
||||
public function AcediaObject Get() |
||||
{ |
||||
if (!initialized) { |
||||
return none; |
||||
} |
||||
TryIterating(); |
||||
if (HasFinished()) { |
||||
return none; |
||||
} |
||||
return foundActors[currentIndex].NewRef(); |
||||
} |
||||
|
||||
public function EPlaceable GetPlaceable() |
||||
{ |
||||
// We only create `EPlaceable` child classes in this class |
||||
return EPlaceable(Get()); |
||||
} |
||||
|
||||
public function EPawn GetPawn() |
||||
{ |
||||
local AcediaObject result; |
||||
local EPawn pawnResult; |
||||
|
||||
if (!initialized) { |
||||
return none; |
||||
} |
||||
result = Get(); |
||||
pawnResult = EPawn(result); |
||||
if (pawnResult == none) { |
||||
_.memory.Free(result); |
||||
} |
||||
return pawnResult; |
||||
} |
||||
|
||||
public function bool HasFinished() |
||||
{ |
||||
TryIterating(); |
||||
return (currentIndex >= foundActors.length); |
||||
} |
||||
|
||||
public function Iter LeaveOnlyNotNone() |
||||
{ |
||||
// We cannot tracer `none` actors, so no need to do anything |
||||
return self; |
||||
} |
||||
|
||||
public function EntityIterator LeaveOnlyPawns() |
||||
{ |
||||
if (initialized && !iterated) { |
||||
onlyPawns = true; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
public function EntityIterator LeaveOnlyPlaceables() { |
||||
// Doesn't do anything for now |
||||
// TODO: make it actually do something |
||||
return self; |
||||
} |
||||
|
||||
public function EntityIterator LeaveOnlyVisible() |
||||
{ |
||||
if (initialized && !iterated) { |
||||
onlyVisible = true; |
||||
} |
||||
return self; |
||||
} |
||||
|
||||
defaultproperties |
||||
{ |
||||
} |
Loading…
Reference in new issue