LoadAssetMaterialsCommandlet.h

Go to the documentation of this file.

// Copyright (c) 2019 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>.

#pragma once

#include "PrepareAssetsForCookingCommandlet.h"
#include "Util/RoadPainterWrapper.h"
#include "Runtime/Engine/Classes/Engine/ObjectLibrary.h"
#include "Carla/OpenDrive/OpenDrive.h"
#include <carla/opendrive/OpenDriveParser.h>

#include <compiler/disable-ue4-macros.h>
#include <carla/opendrive/OpenDriveParser.h>
#include <carla/geom/Math.h>
#include <carla/rpc/String.h>
#include <compiler/enable-ue4-macros.h>


#include "LoadAssetMaterialsCommandlet.generated.h"

USTRUCT()
struct CARLA_API FLargeMapTileData
{
  GENERATED_USTRUCT_BODY()

  UPROPERTY()
  float FirstTileCenterX;

  UPROPERTY()
  float FirstTileCenterY;

  UPROPERTY()
  float Size;
};

struct FDecalsProperties;

UCLASS()
class CARLA_API ULoadAssetMaterialsCommandlet
  : public UCommandlet
{
  GENERATED_BODY()

public:

  /// Default constructor.
  ULoadAssetMaterialsCommandlet();

#if WITH_EDITORONLY_DATA

  /// Gets the Path of all the Assets contained in the package to cook with name
  /// @a PackageName
  FAssetsPaths GetAssetsPathFromPackage(const FString &PackageName) const;

  /// Parses the command line parameters provided through @a InParams
  FPackageParams ParseParams(const FString& InParams) const;

  void LoadAssetsMaterials(const FString &PackageName, const TArray<FMapData> &MapsPaths);

  void ApplyRoadPainterMaterials(const FString &LoadedMapName, const FString &PackageName, bool IsInTiles = false);

  FDecalsProperties ReadDecalsConfigurationFile(const FString &PackageName);

  /// Main method and entry of the commandlet, taking as input parameters @a
  /// Params.
  virtual int32 Main(const FString &Params) override;

#endif // WITH_EDITORONLY_DATA

private:

  /// Loaded assets from any object library
  UPROPERTY()
  TArray<FAssetData> AssetDatas;

  UPROPERTY()
  UWorld *World;

  UPROPERTY()
  UWorld *NewWorldToLoad;

  /// Used for loading maps in object library. Loaded Data is stored in
  /// AssetDatas.
  UPROPERTY()
  UObjectLibrary *MapObjectLibrary;

  /// Subclass for acquiring the RoadPainter blueprint
  UPROPERTY()
  TSubclassOf<ARoadPainterWrapper> RoadPainterSubclass;

  /// Dictionary for translating the JSON file "decal_names" array
  /// to already loaded Material Instances, which are used to apply on the roads
  UPROPERTY()
  TMap<FString, FString> DecalNamesMap;

  //UPROPERTY()
  //TSet<FVector> WaypointsUsedForDecals;

  boost::optional<carla::road::Map> XODRMap;

  /// Gets the first .Package.json file found in Unreal Content Directory with
  /// @a PackageName
  FString GetFirstPackagePath(const FString &PackageName) const;

  FLargeMapTileData TileData;

  FDecalsProperties DecalsProperties;

  bool FilledData;

  bool ReadConfigFile;
};