Source/Editor/Project/ModifyResourceAction.cpp - master - Metallist64/rbfx

Source/Editor/Project/ModifyResourceAction.cpp
126 строк4 KB
Eugene Kozlov
Add AssetPipeline inspector. Minor fixes.
07 сен 2022, 00:20
07 сен 2022, 00:203f82752
//
// Copyright (c) 2017-2020 the rbfx project.
//
// 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.
//

#include "../Project/ModifyResourceAction.h"

#include "../Project/Project.h"

#include <Urho3D/IO/MemoryBuffer.h>
#include <Urho3D/Resource/ResourceCache.h>

namespace Urho3D
{

ModifyResourceAction::ModifyResourceAction(Project* project)
    : project_(project)
{
}

void ModifyResourceAction::AddResource(Resource* resource)
{
    URHO3D_ASSERT(newData_.empty());
    context_ = resource->GetContext();

    VectorBuffer buffer;
    if (resource->Save(buffer))
    {
        ResourceData oldData;
        oldData.resourceType_ = resource->GetType();
        oldData.fileName_ = resource->GetAbsoluteFileName();
        oldData.bytes_ = ea::make_shared<ByteVector>(buffer.GetBuffer());
        oldData_[resource->GetName()] = ea::move(oldData);
    }
}

void ModifyResourceAction::DisableAutoComplete()
{
    autoComplete_ = false;
}

void ModifyResourceAction::SaveOnComplete()
{
    saveOnComplete_ = true;
}

void ModifyResourceAction::Complete(bool force)
{
    if (!autoComplete_ && !force)
        return;

    auto cache = context_ ? context_->GetSubsystem<ResourceCache>() : nullptr;
    for (const auto& [resourceName, oldData] : oldData_)
    {
        if (Resource* resource = cache->GetResource(oldData.resourceType_, resourceName))
        {
            VectorBuffer buffer;
            if (resource->Save(buffer))
            {
                ResourceData newData;
                newData.resourceType_ = oldData.resourceType_;
                newData.fileName_ = oldData.fileName_;
                newData.bytes_ = ea::make_shared<ByteVector>(buffer.GetBuffer());
                newData_[resourceName] = ea::move(newData);
            }
        }
    }

    if (oldData_.size() != newData_.size())
        throw UndoException("ModifyResourceAction failed to complete action creation");

    if (saveOnComplete_)
    {
        for (const auto& [resourceName, data] : newData_)
            project_->SaveFileDelayed(data.fileName_, resourceName, data.bytes_);
    }
}

void ModifyResourceAction::Redo() const
{
    for (const auto& [resourceName, newData] : newData_)
        ApplyResourceData(resourceName, newData);
}

void ModifyResourceAction::Undo() const
{
    for (const auto& [resourceName, oldData] : oldData_)
        ApplyResourceData(resourceName, oldData);
}

bool ModifyResourceAction::MergeWith(const EditorAction& other)
{
    return false;
}

void ModifyResourceAction::ApplyResourceData(const ea::string& resourceName, const ResourceData& data) const
{
    auto cache = context_->GetSubsystem<ResourceCache>();
    if (Resource* resource = cache->GetResource(data.resourceType_, resourceName, false))
    {
        MemoryBuffer buffer{*data.bytes_};
        buffer.SetName(resourceName);
        resource->Load(buffer);
    }

    project_->SaveFileDelayed(data.fileName_, resourceName, data.bytes_);
}

}