ResInsight/ApplicationCode/GrpcInterface/RiaGrpcPdmObjectService.cpp

/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2019-     Equinor ASA
//
//  ResInsight is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
//  WARRANTY; without even the implied warranty of MERCHANTABILITY or
//  FITNESS FOR A PARTICULAR PURPOSE.
//
//  See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
//  for more details.
//
//////////////////////////////////////////////////////////////////////////////////
#include "RiaGrpcPdmObjectService.h"

#include "RiaApplication.h"
#include "RiaGrpcCallbacks.h"
#include "Rim3dView.h"
#include "RimEclipseResultDefinition.h"
#include "RimProject.h"

#include "cafPdmObject.h"

using namespace rips;

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status RiaGrpcPdmObjectService::GetAncestorPdmObject(grpc::ServerContext*                context,
                                                         const rips::PdmParentObjectRequest* request,
                                                         rips::PdmObject*                    reply)
{
    RimProject* project = RiaApplication::instance()->project();
    std::vector<caf::PdmObject*> objectsOfCurrentClass;
    project->descendantsIncludingThisFromClassKeyword(QString::fromStdString(request->object().class_keyword()),
        objectsOfCurrentClass);

    caf::PdmObject* matchingObject = nullptr;
    for (caf::PdmObject* testObject : objectsOfCurrentClass)
    {
        if (reinterpret_cast<uint64_t>(testObject) == request->object().address())
        {
            matchingObject = testObject;
        }
    }

    if (matchingObject)
    {
        caf::PdmObject* parentObject = nullptr;
        matchingObject->firstAncestorOrThisFromClassKeyword(QString::fromStdString(request->parent_keyword()), parentObject);
        if (parentObject)
        {
            copyPdmObjectFromCafToRips(parentObject, reply);
            return grpc::Status::OK;
        }
    }
    return grpc::Status(grpc::NOT_FOUND, "Parent PdmObject not found");
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status RiaGrpcPdmObjectService::GetDescendantPdmObjects(grpc::ServerContext*               context,
                                                         const rips::PdmDescendantObjectRequest* request,
                                                         rips::PdmObjectArray*                   reply)
{
    RimProject*                  project = RiaApplication::instance()->project();
    std::vector<caf::PdmObject*> objectsOfCurrentClass;
    project->descendantsIncludingThisFromClassKeyword(QString::fromStdString(request->object().class_keyword()),
                                                      objectsOfCurrentClass);

    caf::PdmObject* matchingObject = nullptr;
    for (caf::PdmObject* testObject : objectsOfCurrentClass)
    {
        if (reinterpret_cast<uint64_t>(testObject) == request->object().address())
        {
            matchingObject = testObject;
        }
    }

    if (matchingObject)
    {
        std::vector<caf::PdmObject*> childObjects;
        matchingObject->descendantsIncludingThisFromClassKeyword(QString::fromStdString(request->child_keyword()), childObjects);
        for (auto pdmChild : childObjects)
        {
            rips::PdmObject* ripsChild = reply->add_objects();
            copyPdmObjectFromCafToRips(pdmChild, ripsChild);            
        }
        return grpc::Status::OK;
    }
    return grpc::Status(grpc::NOT_FOUND, "Current PdmObject not found");
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status RiaGrpcPdmObjectService::GetChildPdmObjects(grpc::ServerContext*               context,
                                                              const rips::PdmChildObjectRequest* request,
                                                              rips::PdmObjectArray*              reply)
{
    RimProject*                  project = RiaApplication::instance()->project();
    std::vector<caf::PdmObject*> objectsOfCurrentClass;
    project->descendantsIncludingThisFromClassKeyword(QString::fromStdString(request->object().class_keyword()),
                                                      objectsOfCurrentClass);

    caf::PdmObject* matchingObject = nullptr;
    for (caf::PdmObject* testObject : objectsOfCurrentClass)
    {
        if (reinterpret_cast<uint64_t>(testObject) == request->object().address())
        {
            matchingObject = testObject;
        }
    }

    if (matchingObject)
    {
        QString fieldName = QString::fromStdString(request->child_field());
        std::vector<caf::PdmFieldHandle*> fields;
        matchingObject->fields(fields);
        for (auto field : fields)
        {
            if (field->keyword() == fieldName)
            {
                std::vector<caf::PdmObjectHandle*> childObjects;
                field->childObjects(&childObjects);
                for (auto pdmChild : childObjects)
                {
                    rips::PdmObject* ripsChild = reply->add_objects();
                    copyPdmObjectFromCafToRips(static_cast<caf::PdmObject*>(pdmChild), ripsChild);
                }
                return grpc::Status::OK;
            }
        }
        return grpc::Status(grpc::NOT_FOUND, "Child field not found");
    }
    return grpc::Status(grpc::NOT_FOUND, "Current PdmObject not found");
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status RiaGrpcPdmObjectService::UpdateExistingPdmObject(grpc::ServerContext*   context,
                                                              const rips::PdmObject* request,
                                                              rips::Empty*           response)
{
    RimProject*                  project = RiaApplication::instance()->project();
    std::vector<caf::PdmObject*> objectsOfCurrentClass;
    project->descendantsIncludingThisFromClassKeyword(QString::fromStdString(request->class_keyword()), objectsOfCurrentClass);

    caf::PdmObject* matchingObject = nullptr;
    for (caf::PdmObject* pdmObject : objectsOfCurrentClass)
    {
        if (reinterpret_cast<uint64_t>(pdmObject) == request->address())
        {
            matchingObject = pdmObject;
        }
    }

    if (matchingObject)
    {
        copyPdmObjectFromRipsToCaf(request, matchingObject);
        RimEclipseResultDefinition* resultDefinition = dynamic_cast<RimEclipseResultDefinition*>(matchingObject);
        // TODO: Make this more general. Perhaps we need an interface method for updating UI fields
        if (resultDefinition)
        {
            resultDefinition->updateUiFieldsFromActiveResult();
        }

        matchingObject->updateAllRequiredEditors();
        project->scheduleCreateDisplayModelAndRedrawAllViews();
        Rim3dView* view = dynamic_cast<Rim3dView*>(matchingObject);
        if (view)
        {
            view->applyBackgroundColorAndFontChanges();
        }
        return grpc::Status::OK;
    }
    return grpc::Status(grpc::NOT_FOUND, "PdmObject not found");
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status RiaGrpcPdmObjectService::CreateChildPdmObject(grpc::ServerContext*                  context,
                                                        const rips::CreatePdmChildObjectRequest* request,
                                                        rips::PdmObject*                         reply)
{
    RimProject*                  project = RiaApplication::instance()->project();
    std::vector<caf::PdmObject*> objectsOfCurrentClass;
    project->descendantsIncludingThisFromClassKeyword(QString::fromStdString(request->object().class_keyword()),
                                                      objectsOfCurrentClass);

    caf::PdmObject* matchingObject = nullptr;
    for (caf::PdmObject* testObject : objectsOfCurrentClass)
    {
        if (reinterpret_cast<uint64_t>(testObject) == request->object().address())
        {
            matchingObject = testObject;
        }
    }

    if (matchingObject)
    {
        CAF_ASSERT(request);
       
        caf::PdmObject* pdmObject = emplaceChildArrayField(matchingObject,
            QString::fromStdString(request->child_field()),
            QString::fromStdString(request->child_class()));
        if (pdmObject)
        {
            copyPdmObjectFromCafToRips(pdmObject, reply);
            return grpc::Status::OK;
        }
        return grpc::Status(grpc::NOT_FOUND, "Could not create PdmObject");
    }
    return grpc::Status(grpc::NOT_FOUND, "Could not find PdmObject");
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<RiaGrpcCallbackInterface*> RiaGrpcPdmObjectService::createCallbacks()
{
    typedef RiaGrpcPdmObjectService Self;
    return
    {
        new RiaGrpcUnaryCallback<Self, PdmParentObjectRequest, PdmObject>(this, &Self::GetAncestorPdmObject, &Self::RequestGetAncestorPdmObject),
        new RiaGrpcUnaryCallback<Self, PdmDescendantObjectRequest, PdmObjectArray>(this, &Self::GetDescendantPdmObjects, &Self::RequestGetDescendantPdmObjects),
        new RiaGrpcUnaryCallback<Self, PdmChildObjectRequest, PdmObjectArray>(this, &Self::GetChildPdmObjects, &Self::RequestGetChildPdmObjects),
        new RiaGrpcUnaryCallback<Self, PdmObject, Empty>(this, &Self::UpdateExistingPdmObject, &Self::RequestUpdateExistingPdmObject),
        new RiaGrpcUnaryCallback<Self, CreatePdmChildObjectRequest, PdmObject>(this, &Self::CreateChildPdmObject, &Self::RequestCreateChildPdmObject)
    };
}

static bool RiaGrpcPdmObjectService_init =
    RiaGrpcServiceFactory::instance()->registerCreator<RiaGrpcPdmObjectService>(typeid(RiaGrpcPdmObjectService).hash_code());