tools.aqua.stars.data.av.dataclasses.TickData.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2023-2024 The STARS Project Authors
 * SPDX-License-Identifier: Apache-2.0
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package tools.aqua.stars.data.av.dataclasses

import tools.aqua.stars.core.types.TickDataType

/**
 * Json format containing data for current tick.
 *
 * @property currentTick Current tick value.
 * @property entities List of all [Actor]s.
 * @property trafficLights List of all [TrafficLight]s.
 * @property blocks ist of all [Block]s.
 * @property weather The current [WeatherParameters].
 * @property daytime The current [Daytime].
 */
data class TickData(
    override val currentTick: TickDataUnitSeconds,
    override var entities: List,
    val trafficLights: List,
    val blocks: List,
    val weather: WeatherParameters,
    val daytime: Daytime
) : TickDataType {

  override lateinit var segment: Segment

  /** All pedestrians. */
  val pedestrians: List
    get() = entities.filterIsInstance()

  /** All vehicles. */
  val vehicles: List
    get() = entities.filterIsInstance()

  /** The ego vehicle. */
  val egoVehicle: Vehicle?
    get() = vehicles.firstOrNull { it.isEgo }

  /** Returns all [Vehicle]s in given [Block]. */
  fun vehiclesInBlock(block: Block): List = vehicles.filter { it.lane.road.block == block }

  /** Clones current [TickData]. */
  fun clone(): TickData {
    val newTickData = TickData(currentTick, emptyList(), trafficLights, blocks, weather, daytime)
    newTickData.entities = entities.map { it.clone(newTickData) }
    return newTickData
  }

  override fun toString(): String = "$currentTick"
}