// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once
#include "ExtensionsSystem/ChaosVDExtension.h"
#include "Templates/SubclassOf.h"
#include "Templates/SharedPointer.h"

class FChaosVDTraceProvider;
class UActorComponent;
class SChaosVDMainTab;

class FChaosVDGenericDebugDrawExtension final : public FChaosVDExtension
{
public:
	
	FChaosVDGenericDebugDrawExtension();
	virtual ~FChaosVDGenericDebugDrawExtension() override;

	virtual void RegisterDataProcessorsInstancesForProvider(const TSharedRef<FChaosVDTraceProvider>& InTraceProvider) override;
	virtual TConstArrayView<TSubclassOf<UActorComponent>> GetSolverDataComponentsClasses() override;
	virtual void RegisterComponentVisualizers(const TSharedRef<SChaosVDMainTab>& InCVDToolKit) override;

private:
	TArray<TSubclassOf<UActorComponent>> DataComponentsClasses;
};