ArticyObjectWithAttachments.h

//  
// Copyright (c) 2023 articy Software GmbH & Co. KG. All rights reserved.  
//
 
#pragma once
 
#include "ArticyBaseTypes.h"
#include "ArticyPrimitive.h"
#include "ArticyObjectWith_Base.h"
#include "ArticyObjectWithAttachments.generated.h"
 
UINTERFACE(MinimalAPI, BlueprintType, meta=(CannotImplementInterfaceInBlueprint))
class UArticyObjectWithAttachments : public UArticyObjectWith_Base { GENERATED_BODY() };

class IArticyObjectWithAttachments : public IArticyObjectWith_Base
{
    GENERATED_BODY()
 
public:
    
    UFUNCTION(BlueprintCallable, Category="ArticyObjectWithAttachments")
    virtual TArray<FArticyId>& GetAttachmentIds()
    {
        static const auto& PropName = FName("Attachments");
        return GetProperty<TArray<FArticyId>>(PropName);
    }
 
    virtual const TArray<FArticyId>& GetAttachmentIds() const
    {
        return const_cast<IArticyObjectWithAttachments*>(this)->GetAttachmentIds();
    }

    UFUNCTION(BlueprintCallable, Category="ArticyObjectWithAttachments")
    virtual TArray<UArticyPrimitive*> GetAttachments() const
    {
        TArray<UArticyPrimitive*> attachments;
        
        auto ids = GetAttachmentIds();
        for(const auto& id : ids)
        {
            auto att = id.GetObject(_getUObject());
            if(att)
                attachments.Add(att);
        }
        
        return attachments;
    }
 
    //---------------------------------------------------------------------------//
 
    UFUNCTION(BlueprintCallable, Category="ArticyObjectWithAttachments")
    virtual TArray<FArticyId>& SetAttachmentIds(UPARAM(ref) const TArray<FArticyId>& IDs)
    {
        return GetAttachmentIds() = IDs;
    }

    UFUNCTION(BlueprintCallable, Category="ArticyObjectWithAttachments")
    virtual TArray<UArticyPrimitive*> SetAttachments(UPARAM(ref) const TArray<UArticyPrimitive*>& Objects)
    {
        TArray<FArticyId> IDs;
        for(const auto& obj : Objects)
            IDs.Add(obj->GetId());
 
        SetAttachmentIds(IDs);
 
        return Objects;
    }
};