Merge pull request #16975 from IntelOrca/plugin/track-segments

[Plugin] Add APIs for track segments and track iteration

src/openrct2/libopenrct2.vcxproj

@@ -470,6 +470,8 @@
     <ClInclude Include="scripting\bindings\network\ScPlayer.hpp" />
     <ClInclude Include="scripting\bindings\network\ScPlayerGroup.hpp" />
     <ClInclude Include="scripting\bindings\ride\ScRideStation.hpp" />
+    <ClInclude Include="scripting\bindings\ride\ScTrackIterator.h" />
+    <ClInclude Include="scripting\bindings\ride\ScTrackSegment.h" />
     <ClInclude Include="scripting\bindings\world\ScParkMessage.hpp" />
     <ClInclude Include="scripting\bindings\world\ScTileElement.hpp" />
     <ClInclude Include="scripting\Duktape.hpp" />
@@ -928,6 +930,8 @@
     <ClCompile Include="scripting\bindings\network\ScPlayerGroup.cpp" />
     <ClCompile Include="scripting\bindings\ride\ScRide.cpp" />
     <ClCompile Include="scripting\bindings\ride\ScRideStation.cpp" />
+    <ClCompile Include="scripting\bindings\ride\ScTrackIterator.cpp" />
+    <ClCompile Include="scripting\bindings\ride\ScTrackSegment.cpp" />
     <ClCompile Include="scripting\bindings\world\ScMap.cpp" />
     <ClCompile Include="scripting\bindings\world\ScPark.cpp" />
     <ClCompile Include="scripting\bindings\world\ScParkMessage.cpp" />

src/openrct2/ride/Track.cpp

@@ -664,6 +664,25 @@ bool TrackTypeHasSpeedSetting(track_type_t trackType)
     return trackType == TrackElemType::Brakes || trackType == TrackElemType::Booster;
 }
 
+std::optional<CoordsXYZD> GetTrackSegmentOrigin(const CoordsXYE& posEl)
+{
+    auto trackEl = posEl.element->AsTrack();
+    if (trackEl == nullptr)
+        return {};
+
+    const auto& ted = GetTrackElementDescriptor(trackEl->GetTrackType());
+    auto direction = trackEl->GetDirection();
+    auto coords = CoordsXYZ(posEl.x, posEl.y, trackEl->GetBaseZ());
+
+    // Subtract the current sequence's offset
+    const auto* trackBlock = &ted.Block[trackEl->GetSequenceIndex()];
+    CoordsXY trackBlockOffset = { trackBlock->x, trackBlock->y };
+    coords += trackBlockOffset.Rotate(direction_reverse(direction));
+    coords.z -= trackBlock->z;
+
+    return CoordsXYZD(coords, direction);
+}
+
 uint8_t TrackElement::GetSeatRotation() const
 {
     const auto* ride = get_ride(GetRideIndex());

src/openrct2/ride/Track.h

@@ -14,6 +14,8 @@
 #include "../world/Map.h"
 #include "../world/TileElement.h"
 
+#include <optional>
+
 constexpr const uint32_t RideConstructionSpecialPieceSelected = 0x10000;
 
 constexpr const int32_t BLOCK_BRAKE_BASE_SPEED = 0x20364;
@@ -581,3 +583,4 @@ bool track_remove_station_element(const CoordsXYZD& loc, RideId rideIndex, int32
 money32 maze_set_track(const CoordsXYZD& coords, uint8_t flags, bool initialPlacement, RideId rideIndex, uint8_t mode);
 
 bool TrackTypeHasSpeedSetting(track_type_t trackType);
+std::optional<CoordsXYZD> GetTrackSegmentOrigin(const CoordsXYE& posEl);

src/openrct2/scripting/ScriptEngine.cpp

@@ -413,6 +413,8 @@ void ScriptEngine::Initialise()
     ScRideObjectVehicle::Register(ctx);
     ScTile::Register(ctx);
     ScTileElement::Register(ctx);
+    ScTrackIterator::Register(ctx);
+    ScTrackSegment::Register(ctx);
     ScEntity::Register(ctx);
     ScLitter::Register(ctx);
     ScVehicle::Register(ctx);
@@ -438,6 +440,8 @@ void ScriptEngine::Initialise()
     dukglue_register_global(ctx, std::make_shared<ScProfiler>(ctx), "profiler");
     dukglue_register_global(ctx, std::make_shared<ScScenario>(), "scenario");
 
+    RegisterConstants();
+
     _initialised = true;
     _transientPluginsEnabled = false;
     _transientPluginsStarted = false;
@@ -446,6 +450,72 @@ void ScriptEngine::Initialise()
     ClearParkStorage();
 }
 
+class ConstantBuilder
+{
+private:
+    duk_context* _ctx;
+    DukValue _obj;
+
+public:
+    ConstantBuilder(duk_context* ctx)
+        : _ctx(ctx)
+    {
+        duk_push_global_object(_ctx);
+        _obj = DukValue::take_from_stack(_ctx);
+    }
+
+    ConstantBuilder& Namespace(std::string_view ns)
+    {
+        auto flags = DUK_DEFPROP_ENUMERABLE | DUK_DEFPROP_HAVE_WRITABLE | DUK_DEFPROP_HAVE_ENUMERABLE
+            | DUK_DEFPROP_HAVE_CONFIGURABLE | DUK_DEFPROP_HAVE_VALUE;
+
+        // Create a new object for namespace
+        duk_push_global_object(_ctx);
+        duk_push_lstring(_ctx, ns.data(), ns.size());
+        duk_push_object(_ctx);
+
+        // Keep a reference to the namespace object
+        duk_dup_top(_ctx);
+        _obj = DukValue::take_from_stack(_ctx);
+
+        // Place the namespace object into the global context
+        duk_def_prop(_ctx, -3, flags);
+        duk_pop(_ctx);
+
+        return *this;
+    }
+
+    ConstantBuilder& Constant(std::string_view name, int32_t value)
+    {
+        auto flags = DUK_DEFPROP_ENUMERABLE | DUK_DEFPROP_HAVE_WRITABLE | DUK_DEFPROP_HAVE_ENUMERABLE
+            | DUK_DEFPROP_HAVE_CONFIGURABLE | DUK_DEFPROP_HAVE_VALUE;
+        _obj.push();
+        duk_push_lstring(_ctx, name.data(), name.size());
+        duk_push_int(_ctx, value);
+        duk_def_prop(_ctx, -3, flags);
+        duk_pop(_ctx);
+        return *this;
+    }
+};
+
+void ScriptEngine::RegisterConstants()
+{
+    ConstantBuilder builder(_context);
+    builder.Namespace("TrackSlope")
+        .Constant("None", TRACK_SLOPE_NONE)
+        .Constant("Up25", TRACK_SLOPE_UP_25)
+        .Constant("Up60", TRACK_SLOPE_UP_60)
+        .Constant("Down25", TRACK_SLOPE_DOWN_25)
+        .Constant("Down60", TRACK_SLOPE_DOWN_60)
+        .Constant("Up90", TRACK_SLOPE_UP_90)
+        .Constant("Down90", TRACK_SLOPE_DOWN_90);
+    builder.Namespace("TrackBanking")
+        .Constant("None", TRACK_BANK_NONE)
+        .Constant("BankLeft", TRACK_BANK_LEFT)
+        .Constant("BankRight", TRACK_BANK_RIGHT)
+        .Constant("UpsideDown", TRACK_BANK_UPSIDE_DOWN);
+}
+
 void ScriptEngine::RefreshPlugins()
 {
     // Get a list of removed and added plugin files

src/openrct2/scripting/ScriptEngine.h

@@ -252,6 +252,7 @@ namespace OpenRCT2::Scripting
 #    endif
 
     private:
+        void RegisterConstants();
         void RefreshPlugins();
         std::vector<std::string> GetPluginFiles() const;
         void UnregisterPlugin(std::string_view path);

src/openrct2/scripting/bindings/game/ScContext.hpp

@@ -23,6 +23,7 @@
 #    include "../game/ScConfiguration.hpp"
 #    include "../game/ScDisposable.hpp"
 #    include "../object/ScObject.hpp"
+#    include "../ride/ScTrackSegment.h"
 
 #    include <cstdio>
 #    include <memory>
@@ -218,6 +219,19 @@ namespace OpenRCT2::Scripting
             return result;
         }
 
+        DukValue getTrackSegment(track_type_t type)
+        {
+            auto ctx = GetContext()->GetScriptEngine().GetContext();
+            if (type >= TrackElemType::Count)
+            {
+                return ToDuk(ctx, nullptr);
+            }
+            else
+            {
+                return GetObjectAsDukValue(ctx, std::make_shared<ScTrackSegment>(type));
+            }
+        }
+
         int32_t getRandom(int32_t min, int32_t max)
         {
             ThrowIfGameStateNotMutable();
@@ -457,6 +471,7 @@ namespace OpenRCT2::Scripting
             dukglue_register_method(ctx, &ScContext::captureImage, "captureImage");
             dukglue_register_method(ctx, &ScContext::getObject, "getObject");
             dukglue_register_method(ctx, &ScContext::getAllObjects, "getAllObjects");
+            dukglue_register_method(ctx, &ScContext::getTrackSegment, "getTrackSegment");
             dukglue_register_method(ctx, &ScContext::getRandom, "getRandom");
             dukglue_register_method_varargs(ctx, &ScContext::formatString, "formatString");
             dukglue_register_method(ctx, &ScContext::subscribe, "subscribe");

src/openrct2/scripting/bindings/ride/ScTrackIterator.cpp

@@ -0,0 +1,164 @@
+/*****************************************************************************
+ * Copyright (c) 2022 OpenRCT2 developers
+ *
+ * For a complete list of all authors, please refer to contributors.md
+ * Interested in contributing? Visit https://github.com/OpenRCT2/OpenRCT2
+ *
+ * OpenRCT2 is licensed under the GNU General Public License version 3.
+ *****************************************************************************/
+
+#ifdef ENABLE_SCRIPTING
+
+#    include "ScTrackIterator.h"
+
+#    include "../../../Context.h"
+#    include "../../../ride/Ride.h"
+#    include "../../../ride/TrackData.h"
+#    include "../../ScriptEngine.h"
+#    include "ScTrackSegment.h"
+
+using namespace OpenRCT2::Scripting;
+using namespace OpenRCT2::TrackMetaData;
+
+std::shared_ptr<ScTrackIterator> ScTrackIterator::FromElement(const CoordsXY& position, int32_t elementIndex)
+{
+    auto el = map_get_nth_element_at(position, elementIndex);
+    auto origin = GetTrackSegmentOrigin(CoordsXYE(position, el));
+    if (!origin)
+        return nullptr;
+
+    auto trackEl = el->AsTrack();
+    return std::make_shared<ScTrackIterator>(*origin, trackEl->GetTrackType(), trackEl->GetRideIndex());
+}
+
+ScTrackIterator::ScTrackIterator(const CoordsXYZD& position, track_type_t type, RideId ride)
+    : _position(position)
+    , _type(type)
+    , _ride(ride)
+{
+}
+
+void ScTrackIterator::Register(duk_context* ctx)
+{
+    dukglue_register_property(ctx, &ScTrackIterator::position_get, nullptr, "position");
+    dukglue_register_property(ctx, &ScTrackIterator::segment_get, nullptr, "segment");
+    dukglue_register_property(ctx, &ScTrackIterator::previousPosition_get, nullptr, "previousPosition");
+    dukglue_register_property(ctx, &ScTrackIterator::nextPosition_get, nullptr, "nextPosition");
+    dukglue_register_method(ctx, &ScTrackIterator::previous, "previous");
+    dukglue_register_method(ctx, &ScTrackIterator::next, "next");
+}
+
+DukValue ScTrackIterator::position_get() const
+{
+    auto& scriptEngine = GetContext()->GetScriptEngine();
+    auto ctx = scriptEngine.GetContext();
+    return ToDuk(ctx, _position);
+}
+
+DukValue ScTrackIterator::segment_get() const
+{
+    auto& scriptEngine = GetContext()->GetScriptEngine();
+    auto ctx = scriptEngine.GetContext();
+
+    if (_type >= TrackElemType::Count)
+        return ToDuk(ctx, nullptr);
+
+    return GetObjectAsDukValue(ctx, std::make_shared<ScTrackSegment>(_type));
+}
+
+DukValue ScTrackIterator::previousPosition_get() const
+{
+    auto& scriptEngine = GetContext()->GetScriptEngine();
+    auto ctx = scriptEngine.GetContext();
+
+    auto& ted = GetTrackElementDescriptor(_type);
+    auto& seq0 = ted.Block;
+    auto pos = _position + CoordsXYZ(seq0->x, seq0->y, seq0->z);
+
+    auto el = map_get_track_element_at_of_type_seq(pos, _type, 0);
+    if (el == nullptr)
+        return ToDuk(ctx, nullptr);
+
+    auto posEl = CoordsXYE(pos.x, pos.y, reinterpret_cast<TileElement*>(el));
+    track_begin_end tbe{};
+    track_block_get_previous(posEl, &tbe);
+    CoordsXYZD result(tbe.end_x, tbe.end_y, tbe.begin_z, tbe.begin_direction);
+    return ToDuk(ctx, result);
+}
+
+DukValue ScTrackIterator::nextPosition_get() const
+{
+    auto& scriptEngine = GetContext()->GetScriptEngine();
+    auto ctx = scriptEngine.GetContext();
+
+    auto& ted = GetTrackElementDescriptor(_type);
+    auto& seq0 = ted.Block;
+    auto pos = _position + CoordsXYZ(seq0->x, seq0->y, seq0->z);
+
+    auto el = map_get_track_element_at_of_type_seq(pos, _type, 0);
+    if (el == nullptr)
+        return ToDuk(ctx, nullptr);
+
+    auto posEl = CoordsXYE(_position.x, _position.y, reinterpret_cast<TileElement*>(el));
+    CoordsXYE next;
+    int32_t z{};
+    int32_t direction{};
+    track_block_get_next(&posEl, &next, &z, &direction);
+    CoordsXYZD result(next.x, next.y, z, direction);
+    return ToDuk(ctx, result);
+}
+
+bool ScTrackIterator::previous()
+{
+    auto& ted = GetTrackElementDescriptor(_type);
+    auto& seq0 = ted.Block;
+    auto pos = _position + CoordsXYZ(seq0->x, seq0->y, seq0->z);
+
+    auto el = map_get_track_element_at_of_type_seq(pos, _type, 0);
+    if (el == nullptr)
+        return false;
+
+    auto posEl = CoordsXYE(pos.x, pos.y, reinterpret_cast<TileElement*>(el));
+    track_begin_end tbe{};
+    if (track_block_get_previous(posEl, &tbe))
+    {
+        auto prev = CoordsXYE(tbe.end_x, tbe.end_y, tbe.begin_element);
+        auto origin = GetTrackSegmentOrigin(prev);
+        if (origin)
+        {
+            _position = *origin;
+            _type = prev.element->AsTrack()->GetTrackType();
+            return true;
+        }
+    }
+    return false;
+}
+
+bool ScTrackIterator::next()
+{
+    auto& ted = GetTrackElementDescriptor(_type);
+    auto& seq0 = ted.Block;
+    auto pos = _position + CoordsXYZ(seq0->x, seq0->y, seq0->z);
+
+    auto el = map_get_track_element_at_of_type_seq(pos, _type, 0);
+    if (el == nullptr)
+        return false;
+
+    auto posEl = CoordsXYE(_position.x, _position.y, reinterpret_cast<TileElement*>(el));
+    CoordsXYE next;
+    int32_t z{};
+    int32_t direction{};
+    if (track_block_get_next(&posEl, &next, &z, &direction))
+    {
+        auto origin = GetTrackSegmentOrigin(next);
+        if (origin)
+        {
+            _position = *origin;
+            _type = next.element->AsTrack()->GetTrackType();
+            return true;
+        }
+    }
+    return false;
+}
+
+#endif

src/openrct2/scripting/bindings/ride/ScTrackIterator.h

@@ -0,0 +1,47 @@
+/*****************************************************************************
+ * Copyright (c) 2022 OpenRCT2 developers
+ *
+ * For a complete list of all authors, please refer to contributors.md
+ * Interested in contributing? Visit https://github.com/OpenRCT2/OpenRCT2
+ *
+ * OpenRCT2 is licensed under the GNU General Public License version 3.
+ *****************************************************************************/
+
+#pragma once
+
+#ifdef ENABLE_SCRIPTING
+
+#    include "../../../Identifiers.h"
+#    include "../../../world/TileElement.h"
+#    include "../../Duktape.hpp"
+
+#    include <cstdint>
+
+namespace OpenRCT2::Scripting
+{
+    class ScTrackIterator
+    {
+    private:
+        CoordsXYZD _position;
+        track_type_t _type;
+        [[maybe_unused]] RideId _ride;
+
+    public:
+        static std::shared_ptr<ScTrackIterator> FromElement(const CoordsXY& position, int32_t elementIndex);
+        static void Register(duk_context* ctx);
+
+        ScTrackIterator(const CoordsXYZD& position, track_type_t type, RideId ride);
+
+    private:
+        DukValue position_get() const;
+        DukValue segment_get() const;
+        DukValue previousPosition_get() const;
+        DukValue nextPosition_get() const;
+
+        bool previous();
+        bool next();
+    };
+
+} // namespace OpenRCT2::Scripting
+
+#endif

src/openrct2/scripting/bindings/ride/ScTrackSegment.cpp

@@ -0,0 +1,144 @@
+/*****************************************************************************
+ * Copyright (c) 2022 OpenRCT2 developers
+ *
+ * For a complete list of all authors, please refer to contributors.md
+ * Interested in contributing? Visit https://github.com/OpenRCT2/OpenRCT2
+ *
+ * OpenRCT2 is licensed under the GNU General Public License version 3.
+ *****************************************************************************/
+
+#ifdef ENABLE_SCRIPTING
+
+#    include "ScTrackSegment.h"
+
+#    include "../../../Context.h"
+#    include "../../../ride/TrackData.h"
+#    include "../../ScriptEngine.h"
+
+using namespace OpenRCT2::Scripting;
+using namespace OpenRCT2::TrackMetaData;
+
+ScTrackSegment::ScTrackSegment(track_type_t type)
+    : _type(type)
+{
+}
+
+void ScTrackSegment::Register(duk_context* ctx)
+{
+    dukglue_register_property(ctx, &ScTrackSegment::type_get, nullptr, "type");
+    dukglue_register_property(ctx, &ScTrackSegment::description_get, nullptr, "description");
+    dukglue_register_property(ctx, &ScTrackSegment::elements_get, nullptr, "elements");
+    dukglue_register_property(ctx, &ScTrackSegment::beginZ_get, nullptr, "beginZ");
+    dukglue_register_property(ctx, &ScTrackSegment::beginDirection_get, nullptr, "beginDirection");
+    dukglue_register_property(ctx, &ScTrackSegment::endX_get, nullptr, "endX");
+    dukglue_register_property(ctx, &ScTrackSegment::endY_get, nullptr, "endY");
+    dukglue_register_property(ctx, &ScTrackSegment::endZ_get, nullptr, "endZ");
+    dukglue_register_property(ctx, &ScTrackSegment::endDirection_get, nullptr, "endDirection");
+    dukglue_register_property(ctx, &ScTrackSegment::length_get, nullptr, "length");
+}
+
+int32_t ScTrackSegment::type_get() const
+{
+    return _type;
+}
+
+std::string ScTrackSegment::description_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return language_get_string(ted.Description);
+}
+
+int32_t ScTrackSegment::beginZ_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Coordinates.z_begin;
+}
+
+int32_t ScTrackSegment::beginDirection_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Coordinates.rotation_begin;
+}
+
+int32_t ScTrackSegment::beginAngle_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Definition.vangle_start;
+}
+
+int32_t ScTrackSegment::beginBank_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Definition.bank_start;
+}
+
+int32_t ScTrackSegment::endX_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Coordinates.x;
+}
+
+int32_t ScTrackSegment::endY_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Coordinates.y;
+}
+
+int32_t ScTrackSegment::endZ_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Coordinates.z_end;
+}
+
+int32_t ScTrackSegment::endDirection_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Coordinates.rotation_end;
+}
+
+int32_t ScTrackSegment::endAngle_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Definition.vangle_end;
+}
+
+int32_t ScTrackSegment::endBank_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.Definition.bank_end;
+}
+
+int32_t ScTrackSegment::length_get() const
+{
+    const auto& ted = GetTrackElementDescriptor(_type);
+    return ted.PieceLength;
+}
+
+DukValue ScTrackSegment::elements_get() const
+{
+    auto& scriptEngine = GetContext()->GetScriptEngine();
+    auto ctx = scriptEngine.GetContext();
+
+    const auto& ted = GetTrackElementDescriptor(_type);
+
+    duk_push_array(ctx);
+
+    duk_uarridx_t index = 0;
+    for (auto* block = ted.Block; block->index != 0xFF; block++)
+    {
+        duk_push_object(ctx);
+        duk_push_number(ctx, block->x);
+        duk_put_prop_string(ctx, -2, "x");
+        duk_push_number(ctx, block->y);
+        duk_put_prop_string(ctx, -2, "y");
+        duk_push_number(ctx, block->z);
+        duk_put_prop_string(ctx, -2, "z");
+
+        duk_put_prop_index(ctx, -2, index);
+        index++;
+    }
+
+    return DukValue::take_from_stack(ctx);
+}
+
+#endif

src/openrct2/scripting/bindings/ride/ScTrackSegment.h

@@ -0,0 +1,51 @@
+/*****************************************************************************
+ * Copyright (c) 2022 OpenRCT2 developers
+ *
+ * For a complete list of all authors, please refer to contributors.md
+ * Interested in contributing? Visit https://github.com/OpenRCT2/OpenRCT2
+ *
+ * OpenRCT2 is licensed under the GNU General Public License version 3.
+ *****************************************************************************/
+
+#pragma once
+
+#ifdef ENABLE_SCRIPTING
+
+#    include "../../../world/TileElement.h"
+#    include "../../Duktape.hpp"
+
+#    include <cstdint>
+#    include <string>
+
+namespace OpenRCT2::Scripting
+{
+    class ScTrackSegment
+    {
+    private:
+        track_type_t _type;
+
+    public:
+        ScTrackSegment(track_type_t type);
+
+        static void Register(duk_context* ctx);
+
+    private:
+        int32_t type_get() const;
+        std::string description_get() const;
+        int32_t beginZ_get() const;
+        int32_t beginDirection_get() const;
+        int32_t beginAngle_get() const;
+        int32_t beginBank_get() const;
+        int32_t endX_get() const;
+        int32_t endY_get() const;
+        int32_t endZ_get() const;
+        int32_t endDirection_get() const;
+        int32_t endAngle_get() const;
+        int32_t endBank_get() const;
+        int32_t length_get() const;
+        DukValue elements_get() const;
+    };
+
+} // namespace OpenRCT2::Scripting
+
+#endif

src/openrct2/scripting/bindings/world/ScMap.cpp

@@ -31,6 +31,7 @@
 #    include "../entity/ScStaff.hpp"
 #    include "../entity/ScVehicle.hpp"
 #    include "../ride/ScRide.hpp"
+#    include "../ride/ScTrackIterator.h"
 #    include "../world/ScTile.hpp"
 
 namespace OpenRCT2::Scripting
@@ -303,6 +304,16 @@ namespace OpenRCT2::Scripting
         return res;
     }
 
+    DukValue ScMap::getTrackIterator(const DukValue& dukPosition, int32_t elementIndex) const
+    {
+        auto position = FromDuk<CoordsXY>(dukPosition);
+        auto trackIterator = ScTrackIterator::FromElement(position, elementIndex);
+        if (trackIterator == nullptr)
+            return ToDuk(_context, undefined);
+
+        return GetObjectAsDukValue(_context, trackIterator);
+    }
+
     void ScMap::Register(duk_context* ctx)
     {
         dukglue_register_property(ctx, &ScMap::size_get, nullptr, "size");
@@ -315,6 +326,7 @@ namespace OpenRCT2::Scripting
         dukglue_register_method(ctx, &ScMap::getAllEntities, "getAllEntities");
         dukglue_register_method(ctx, &ScMap::getAllEntitiesOnTile, "getAllEntitiesOnTile");
         dukglue_register_method(ctx, &ScMap::createEntity, "createEntity");
+        dukglue_register_method(ctx, &ScMap::getTrackIterator, "getTrackIterator");
     }
 
     DukValue ScMap::GetEntityAsDukValue(const EntityBase* sprite) const

src/openrct2/scripting/bindings/world/ScMap.hpp

@@ -14,6 +14,7 @@
 #    include "../../../common.h"
 #    include "../../Duktape.hpp"
 #    include "../ride/ScRide.hpp"
+#    include "../ride/ScTrackIterator.h"
 #    include "../world/ScTile.hpp"
 
 namespace OpenRCT2::Scripting
@@ -46,6 +47,8 @@ namespace OpenRCT2::Scripting
 
         DukValue createEntity(const std::string& type, const DukValue& initializer);
 
+        DukValue getTrackIterator(const DukValue& position, int32_t elementIndex) const;
+
         static void Register(duk_context* ctx);
 
     private:

src/openrct2/scripting/bindings/world/ScTileElement.cpp

@@ -1075,6 +1075,27 @@ namespace OpenRCT2::Scripting
         Invalidate();
     }
 
+    DukValue ScTileElement::isHighlighted_get() const
+    {
+        auto ctx = GetContext()->GetScriptEngine().GetContext();
+        auto el = _element->AsTrack();
+        if (el != nullptr)
+            duk_push_boolean(ctx, el->IsHighlighted());
+        else
+            duk_push_null(ctx);
+        return DukValue::take_from_stack(ctx);
+    }
+    void ScTileElement::isHighlighted_set(bool value)
+    {
+        ThrowIfGameStateNotMutable();
+        auto el = _element->AsTrack();
+        if (el != nullptr)
+        {
+            el->SetHighlight(value);
+            Invalidate();
+        }
+    }
+
     DukValue ScTileElement::object_get() const
     {
         auto& scriptEngine = GetContext()->GetScriptEngine();
@@ -2045,6 +2066,7 @@ namespace OpenRCT2::Scripting
         dukglue_register_property(ctx, &ScTileElement::hasChainLift_get, &ScTileElement::hasChainLift_set, "hasChainLift");
         dukglue_register_property(ctx, &ScTileElement::isInverted_get, &ScTileElement::isInverted_set, "isInverted");
         dukglue_register_property(ctx, &ScTileElement::hasCableLift_get, &ScTileElement::hasCableLift_set, "hasCableLift");
+        dukglue_register_property(ctx, &ScTileElement::isHighlighted_get, &ScTileElement::isHighlighted_set, "isHighlighted");
 
         // Small Scenery only
         dukglue_register_property(ctx, &ScTileElement::age_get, &ScTileElement::age_set, "age");

src/openrct2/scripting/bindings/world/ScTileElement.hpp

@@ -114,6 +114,9 @@ namespace OpenRCT2::Scripting
         DukValue hasCableLift_get() const;
         void hasCableLift_set(bool value);
 
+        DukValue isHighlighted_get() const;
+        void isHighlighted_set(bool value);
+
         DukValue object_get() const;
         void object_set(const DukValue& value);