CarlaRecorderAnimVehicleWheels.cpp

Go to the documentation of this file.

// Copyright (c) 2026 Computer Vision Center (CVC) at the Universitat Autonoma
// de Barcelona (UAB).
//
// This work is licensed under the terms of the MIT license.
// For a copy, see <https://opensource.org/licenses/MIT>.
 
#include "CarlaRecorderAnimVehicleWheels.h"
#include "CarlaRecorder.h"
#include "CarlaRecorderHelpers.h"
 
#include <algorithm>
#include <limits>
 
void WheelInfo::Write(std::ostream &OutFile) const
{
  WriteValue<EVehicleWheelLocation>(OutFile, Location);
  WriteValue<float>(OutFile, SteeringAngle);
  WriteValue<float>(OutFile, TireRotation);
}
 
void WheelInfo::Read(std::istream &InFile)
{
  ReadValue<EVehicleWheelLocation>(InFile, Location);
  ReadValue<float>(InFile, SteeringAngle);
  ReadValue<float>(InFile, TireRotation);
}
 
void CarlaRecorderAnimWheels::Write(std::ostream &OutFile)
{
  WriteValue<uint32_t>(OutFile, DatabaseId);
  WriteValue<uint32_t>(OutFile, WheelValues.size());
  for (const WheelInfo& Wheel : WheelValues)
  {
    Wheel.Write(OutFile);
  }
}
 
void CarlaRecorderAnimWheels::Read(std::istream &InFile)
{
  ReadValue<uint32_t>(InFile, DatabaseId);
  uint32_t NumWheels = 0;
  ReadValue<uint32_t>(InFile, NumWheels);
  // The same object is read again for every vehicle of a packet.
  WheelValues.clear();
  for (size_t i = 0; (i < NumWheels) && InFile; ++i)
  {
    WheelInfo Wheel;
    Wheel.Read(InFile);
    WheelValues.push_back(Wheel);
  }
}
 
// ---------------------------------------------
 
void CarlaRecorderAnimVehicleWheels::Clear(void)
{
  VehicleWheels.clear();
}
 
void CarlaRecorderAnimVehicleWheels::Add(const CarlaRecorderAnimWheels &Vehicle)
{
  VehicleWheels.push_back(Vehicle);
}
 
void CarlaRecorderAnimVehicleWheels::Write(std::ostream &OutFile)
{
  // write the packet id
  WriteValue<char>(OutFile, static_cast<char>(CarlaRecorderPacketId::AnimVehicleWheels));
 
  std::streampos PosStart = OutFile.tellp();
 
  // write a dummy packet size
  uint32_t Total = 0;
  WriteValue<uint32_t>(OutFile, Total);
 
  // write total records
  Total = std::min<size_t>(VehicleWheels.size(), std::numeric_limits<uint16_t>::max());
  WriteValue<uint16_t>(OutFile, Total);
 
  for (uint16_t i=0; i<Total; ++i)
    VehicleWheels[i].Write(OutFile);
 
  // write the real packet size
  std::streampos PosEnd = OutFile.tellp();
  Total = PosEnd - PosStart - sizeof(uint32_t);
  OutFile.seekp(PosStart, std::ios::beg);
  WriteValue<uint32_t>(OutFile, Total);
  OutFile.seekp(PosEnd, std::ios::beg);
}
 
void CarlaRecorderAnimVehicleWheels::Read(std::istream &InFile)
{
  uint16_t i, Total = 0;
  CarlaRecorderAnimWheels Wheels;
 
  // read Total Vehicles
  ReadValue<uint16_t>(InFile, Total);
  for (i = 0; (i < Total) && InFile; ++i)
  {
    Wheels.Read(InFile);
    Add(Wheels);
  }
}
 
const std::vector<CarlaRecorderAnimWheels>& CarlaRecorderAnimVehicleWheels::GetVehicleWheels()
{
  return VehicleWheels;
}