OilfieldToolsNet/ResInsight
4
0
Fork 0
mirror of https://github.com/OPM/ResInsight.git synced 2025-02-25 18:55:39 -06:00
Code Issues Packages Projects Releases Wiki Activity

Move GrpcInterface into top level

Browse Source
This commit is contained in:
Gaute Lindkvist
2021-01-06 14:57:14 +01:00
parent 81699db187
commit 2de5c9ce0b
113 changed files with 0 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

230
GrpcInterface/RiaGrpcProjectService.cpp Normal file
View File

@@ -0,0 +1,230 @@
#include "RiaGrpcProjectService.h"
#include "RiaApplication.h"
#include "RiaGrpcCallbacks.h"
#include "RiaSocketTools.h"
#include "RimCase.h"
#include "RimCaseCollection.h"
#include "RimEclipseCase.h"
#include "RimEclipseCaseCollection.h"
#include "RimGeoMechCase.h"
#include "RimGridView.h"
#include "RimIdenticalGridCaseGroup.h"
#include "RimOilField.h"
#include "RimProject.h"
#include "cafSelectionManager.h"
#include "Case.grpc.pb.h"
using grpc::ServerCompletionQueue;
using grpc::ServerContext;
using grpc::Status;
using namespace rips;
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
Status RiaGrpcProjectService::GetCurrentCase( ServerContext* context, const rips::Empty* request, rips::CaseRequest* reply )
{
int scriptCaseId = RiaApplication::instance()->currentScriptCaseId();
if ( scriptCaseId != -1 )
{
reply->set_id( scriptCaseId );
return Status::OK;
}
else
{
RimGridView* view = RiaApplication::instance()->activeGridView();
if ( view )
{
RimCase* currentCase = view->ownerCase();
if ( currentCase )
{
reply->set_id( currentCase->caseId() );
return Status::OK;
}
}
}
return Status( grpc::NOT_FOUND, "No current case found" );
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
Status RiaGrpcProjectService::GetSelectedCases( ServerContext* context, const rips::Empty* request, rips::CaseInfoArray* reply )
{
std::vector<RimCase*> cases;
caf::SelectionManager::instance()->objectsByType( &cases );
if ( cases.empty() )
{
return Status( grpc::NOT_FOUND, "No cases selected" );
}
for ( RimCase* rimCase : cases )
{
qint64 caseId = rimCase->caseId();
qint64 caseGroupId = -1;
QString caseName, caseType;
RiaSocketTools::getCaseInfoFromCase( rimCase, caseId, caseName, caseType, caseGroupId );
rips::CaseInfo* caseInfo = reply->add_data();
caseInfo->set_id( caseId );
caseInfo->set_group_id( caseGroupId );
caseInfo->set_name( caseName.toStdString() );
caseInfo->set_type( caseType.toStdString() );
}
return Status::OK;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status RiaGrpcProjectService::GetAllCaseGroups( grpc::ServerContext* context,
const rips::Empty* request,
rips::CaseGroups* reply )
{
RimProject* proj = RimProject::current();
RimEclipseCaseCollection* analysisModels =
( proj && proj->activeOilField() ) ? proj->activeOilField()->analysisModels() : nullptr;
if ( analysisModels )
{
for ( RimIdenticalGridCaseGroup* cg : analysisModels->caseGroups() )
{
rips::CaseGroup* caseGroup = reply->add_case_groups();
caseGroup->set_id( cg->groupId() );
caseGroup->set_name( cg->name().toStdString() );
}
return Status::OK;
}
return Status( grpc::NOT_FOUND, "No case groups found" );
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status RiaGrpcProjectService::GetAllCases( grpc::ServerContext* context,
const rips::Empty* request,
rips::CaseInfoArray* reply )
{
std::vector<RimCase*> cases;
RimProject::current()->allCases( cases );
if ( cases.empty() )
{
return Status( grpc::NOT_FOUND, "No cases found" );
}
for ( RimCase* rimCase : cases )
{
qint64 caseId = rimCase->caseId();
qint64 caseGroupId = -1;
QString caseName, caseType;
RiaSocketTools::getCaseInfoFromCase( rimCase, caseId, caseName, caseType, caseGroupId );
rips::CaseInfo* caseInfo = reply->add_data();
caseInfo->set_id( caseId );
caseInfo->set_group_id( caseGroupId );
caseInfo->set_name( caseName.toStdString() );
caseInfo->set_type( caseType.toStdString() );
}
return Status::OK;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status RiaGrpcProjectService::GetCasesInGroup( grpc::ServerContext* context,
const rips::CaseGroup* request,
rips::CaseInfoArray* reply )
{
RimProject* proj = RimProject::current();
RimEclipseCaseCollection* analysisModels =
( proj && proj->activeOilField() ) ? proj->activeOilField()->analysisModels() : nullptr;
if ( analysisModels )
{
int groupId = request->id();
RimIdenticalGridCaseGroup* caseGroup = nullptr;
for ( size_t i = 0; i < analysisModels->caseGroups().size(); i++ )
{
RimIdenticalGridCaseGroup* cg = analysisModels->caseGroups()[i];
if ( groupId == cg->groupId() )
{
caseGroup = cg;
}
}
std::vector<RimCase*> cases;
if ( caseGroup )
{
for ( size_t i = 0; i < caseGroup->statisticsCaseCollection()->reservoirs.size(); i++ )
{
cases.push_back( caseGroup->statisticsCaseCollection()->reservoirs[i] );
}
for ( size_t i = 0; i < caseGroup->caseCollection()->reservoirs.size(); i++ )
{
cases.push_back( caseGroup->caseCollection()->reservoirs[i] );
}
}
if ( !cases.empty() )
{
for ( RimCase* rimCase : cases )
{
qint64 caseId = rimCase->caseId();
qint64 caseGroupId = -1;
QString caseName, caseType;
RiaSocketTools::getCaseInfoFromCase( rimCase, caseId, caseName, caseType, caseGroupId );
rips::CaseInfo* caseInfo = reply->add_data();
caseInfo->set_id( caseId );
caseInfo->set_group_id( caseGroupId );
caseInfo->set_name( caseName.toStdString() );
caseInfo->set_type( caseType.toStdString() );
}
}
}
return Status( grpc::NOT_FOUND, "No cases found" );
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
grpc::Status
RiaGrpcProjectService::GetPdmObject( grpc::ServerContext* context, const rips::Empty* request, rips::PdmObject* reply )
{
RimProject* project = RimProject::current();
if ( project )
{
copyPdmObjectFromCafToRips( project, reply );
}
return grpc::Status::OK;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::vector<RiaGrpcCallbackInterface*> RiaGrpcProjectService::createCallbacks()
{
typedef RiaGrpcProjectService Self;
return { new RiaGrpcUnaryCallback<Self, Empty, CaseRequest>( this, &Self::GetCurrentCase, &Self::RequestGetCurrentCase ),
new RiaGrpcUnaryCallback<Self, Empty, CaseInfoArray>( this,
&Self::GetSelectedCases,
&Self::RequestGetSelectedCases ),
new RiaGrpcUnaryCallback<Self, Empty, CaseGroups>( this, &Self::GetAllCaseGroups, &Self::RequestGetAllCaseGroups ),
new RiaGrpcUnaryCallback<Self, Empty, CaseInfoArray>( this, &Self::GetAllCases, &Self::RequestGetAllCases ),
new RiaGrpcUnaryCallback<Self, CaseGroup, CaseInfoArray>( this,
&Self::GetCasesInGroup,
&Self::RequestGetCasesInGroup ),
new RiaGrpcUnaryCallback<Self, Empty, PdmObject>( this, &Self::GetPdmObject, &Self::RequestGetPdmObject ) };
}
static bool RiaGrpcProjectService_init = RiaGrpcServiceFactory::instance()->registerCreator<RiaGrpcProjectService>(
typeid( RiaGrpcProjectService ).hash_code() );