131 lines
4.6 KiB
C++
131 lines
4.6 KiB
C++
|
/**
|
||
|
|
* @file
|
||
|
|
*
|
||
|
|
* @author Mamadou Babaei <mamadou@senseglove.com>
|
||
|
|
*
|
||
|
|
* @section LICENSE
|
||
|
|
*
|
||
|
|
* (The MIT License)
|
||
|
|
*
|
||
|
|
* Copyright (c) 2020 - 2024 SenseGlove
|
||
|
|
*
|
||
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
|
|
* of this software and associated documentation files (the "Software"), to deal
|
||
|
|
* in the Software without restriction, including without limitation the rights
|
||
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
|
|
* copies of the Software, and to permit persons to whom the Software is
|
||
|
|
* furnished to do so, subject to the following conditions:
|
||
|
|
*
|
||
|
|
* The above copyright notice and this permission notice shall be included in all
|
||
|
|
* copies or substantial portions of the Software.
|
||
|
|
*
|
||
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||
|
|
* SOFTWARE.
|
||
|
|
*
|
||
|
|
* @section DESCRIPTION
|
||
|
|
*
|
||
|
|
*
|
||
|
|
*/
|
||
|
|
|
||
|
|
|
||
|
|
#include "SGEditor/SGContentBrowserExtension_Skeleton.h"
|
||
|
|
|
||
|
|
#include "AssetRegistry/AssetData.h"
|
||
|
|
#include "ContentBrowserDelegates.h"
|
||
|
|
#include "ContentBrowserModule.h"
|
||
|
|
#include "Engine/SkeletalMeshSocket.h"
|
||
|
|
#include "Framework/MultiBox/MultiBoxBuilder.h"
|
||
|
|
#include "Framework/MultiBox/MultiBoxExtender.h"
|
||
|
|
#include "Math/Transform.h"
|
||
|
|
#include "ReferenceSkeleton.h"
|
||
|
|
|
||
|
|
#include "SGEditor/SGPluginStyle.h"
|
||
|
|
#include "SGLog/SGLog.h"
|
||
|
|
|
||
|
|
#define LOCTEXT_NAMESPACE "SGContentBrowserExtension_Skeleton"
|
||
|
|
|
||
|
|
void FSGContentBrowserExtension_Skeleton::Register()
|
||
|
|
{
|
||
|
|
FContentBrowserModule& ContentBrowserModule{
|
||
|
|
FModuleManager::LoadModuleChecked<FContentBrowserModule>(TEXT("ContentBrowser"))
|
||
|
|
};
|
||
|
|
|
||
|
|
TArray<FContentBrowserMenuExtender_SelectedAssets>& ContextMenuDelegates{
|
||
|
|
ContentBrowserModule.GetAllAssetViewContextMenuExtenders()
|
||
|
|
};
|
||
|
|
|
||
|
|
ContextMenuDelegates.Add(FContentBrowserMenuExtender_SelectedAssets::CreateStatic(&OnExtendAssetSelectionMenu));
|
||
|
|
}
|
||
|
|
|
||
|
|
TSharedRef<FExtender> FSGContentBrowserExtension_Skeleton::OnExtendAssetSelectionMenu(
|
||
|
|
const TArray<FAssetData>& SelectedAssets)
|
||
|
|
{
|
||
|
|
TSharedRef<FExtender> Extender{MakeShared<FExtender>()};
|
||
|
|
Extender->AddMenuExtension(
|
||
|
|
"CommonAssetActions",
|
||
|
|
EExtensionHook::After,
|
||
|
|
nullptr,
|
||
|
|
FMenuExtensionDelegate::CreateStatic(&AssetExtenderCallback, SelectedAssets)
|
||
|
|
);
|
||
|
|
return MoveTemp(Extender);
|
||
|
|
}
|
||
|
|
|
||
|
|
void FSGContentBrowserExtension_Skeleton::AssetExtenderCallback(
|
||
|
|
FMenuBuilder& MenuBuilder, const TArray<FAssetData> SelectedAssets)
|
||
|
|
{
|
||
|
|
MenuBuilder.BeginSection(FName{TEXT("SenseGlove")}, LOCTEXT("SenseGloveHeading", "SenseGlove"));
|
||
|
|
{
|
||
|
|
for (const FAssetData& Asset: SelectedAssets)
|
||
|
|
{
|
||
|
|
if (!Asset.IsRedirector()
|
||
|
|
&& FName{*Asset.AssetClassPath.ToString()} != NAME_Class
|
||
|
|
&& !(Asset.PackageFlags & PKG_FilterEditorOnly))
|
||
|
|
{
|
||
|
|
if (Asset.GetClass()->IsChildOf(USkeleton::StaticClass()))
|
||
|
|
{
|
||
|
|
MenuBuilder.AddMenuEntry(
|
||
|
|
LOCTEXT("AddSenseGloveSocketsLabel", "Add SenseGlove Sockets"),
|
||
|
|
LOCTEXT(
|
||
|
|
"AddSenseGloveSocketsToolTip", "Adds and sets up the SenseGlove grab and touch sockets"),
|
||
|
|
FSGPluginStyle::GetStyleSetIcon(),
|
||
|
|
FUIAction{
|
||
|
|
FExecuteAction::CreateLambda([SelectedAssets]()-> void
|
||
|
|
{
|
||
|
|
for (const FAssetData& Asset: SelectedAssets)
|
||
|
|
{
|
||
|
|
USkeleton* Skeleton{Cast<USkeleton>(Asset.GetAsset())};
|
||
|
|
|
||
|
|
if (!IsValid(Skeleton))
|
||
|
|
{
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
AddSenseGloveSockets(Skeleton);
|
||
|
|
}
|
||
|
|
})
|
||
|
|
},
|
||
|
|
NAME_None,
|
||
|
|
EUserInterfaceActionType::Button,
|
||
|
|
NAME_None);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
MenuBuilder.EndSection();
|
||
|
|
}
|
||
|
|
|
||
|
|
void FSGContentBrowserExtension_Skeleton::AddSenseGloveSockets(USkeleton* Skeleton)
|
||
|
|
{
|
||
|
|
if (!ensureAlwaysMsgf(IsValid(Skeleton), TEXT("Invalid Skeleton!")))
|
||
|
|
{
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
#undef LOCTEXT_NAMESPACE
|