Update rips module, proto files, and Python examples

Author: magnesj

docs/proto/SimulatorTables.proto

@@ -7,13 +7,20 @@ package rips;
 service WellPath {
   rpc GetCompletionData(SimulatorTableRequest)
       returns (SimulatorTableData) {}
+  rpc GetCompletionDataUnified(SimulatorTableUnifiedRequest)
+      returns (SimulatorTableData) {}
 }
 
 message SimulatorTableRequest {
   string wellpath_name = 1;
   int32 case_id = 2;
 }
 
+message SimulatorTableUnifiedRequest {
+  repeated string wellpath_names = 1;
+  int32 case_id = 2;
+}
+
 message SimulatorCompdatEntry {
   string well_name = 1;
   int32 grid_i = 2;
@@ -49,8 +56,87 @@ message SimulatorWelspecsEntry {
   optional int32 fip_region = 13;
 }
 
+message SimulatorWelsegsHeaderEntry {
+  string well_name = 1;
+  double top_depth = 2;
+  double top_length = 3;
+  optional double wellbore_volume = 4;
+  string info_type = 5;
+  optional string pressure_omponents = 6;
+  optional string flow_model = 7;
+}
+
+message SimulatorWelsegsRowEntry {
+  int32 segment_1 = 1;
+  int32 segment_2 = 2;
+  int32 branch = 3;
+  int32 join_segment = 4;
+  double length = 5;
+  double depth = 6;
+  optional double diameter = 7;
+  optional double roughness = 8;
+  optional string description = 9;
+}
+
+message SimulatorWelsegsEntry {
+  SimulatorWelsegsHeaderEntry header = 1;
+  repeated SimulatorWelsegsRowEntry row = 2;
+}
+
+message SimulatorCompsegsEntry {
+  int32 i = 1;
+  int32 j = 2;
+  int32 k = 3;
+  int32 branch = 4;
+  double distance_start = 5;
+  double distance_end = 6;
+  optional string grid_name = 7;
+}
+
+message SimulatorWsegvalvEntry {
+  string well_name = 1;
+  int32 segment_number = 2;
+  double cv = 3;
+  double area = 4;
+  optional double extra_length = 5;
+  optional double pipe_d = 6;
+  optional double roughness = 7;
+  optional double pipe_a = 8;
+  optional string status = 9;
+  optional double max_a = 10;
+}
+
+message SimulatorWsegaicdEntry {
+  string well_name = 1;
+  int32 segment_1 = 2;
+  int32 segment_2 = 3;
+  double strength = 4;
+  optional double length = 5;
+  optional double density_cali = 6;
+  optional double viscosity_cali = 7;
+  optional double critical_value = 8;
+  optional double width_trans = 9;
+  optional double max_visc_ratio = 10;
+  optional int32 method_scaling_factor = 11;
+  double max_abs_rate = 12;
+  double flow_rate_exponent = 13;
+  double visc_exponent = 14;
+  optional string status = 15;
+  optional double oil_flow_fraction = 16;
+  optional double water_flow_fraction = 17;
+  optional double gas_flow_fraction = 18;
+  optional double oil_visc_fraction = 19;
+  optional double water_visc_fraction = 20;
+  optional double gas_visc_fraction = 21;
+  optional string description = 22;
+}
+
 message SimulatorTableData {
   repeated SimulatorCompdatEntry compdat = 1;
   repeated SimulatorWelspecsEntry welspecs = 2;
+  repeated SimulatorWelsegsEntry welsegs = 3;
+  repeated SimulatorCompsegsEntry compsegs = 4;
+  repeated SimulatorWsegvalvEntry wsegvalv = 5;
+  repeated SimulatorWsegaicdEntry wsegaicd = 6;
 }
 

docs/rips/generated/SimulatorTables_pb2.py

@@ -39,6 +39,11 @@ def __init__(self, channel):
                 request_serializer=SimulatorTables__pb2.SimulatorTableRequest.SerializeToString,
                 response_deserializer=SimulatorTables__pb2.SimulatorTableData.FromString,
                 _registered_method=True)
+        self.GetCompletionDataUnified = channel.unary_unary(
+                '/rips.WellPath/GetCompletionDataUnified',
+                request_serializer=SimulatorTables__pb2.SimulatorTableUnifiedRequest.SerializeToString,
+                response_deserializer=SimulatorTables__pb2.SimulatorTableData.FromString,
+                _registered_method=True)
 
 
 class WellPathServicer(object):
@@ -50,6 +55,12 @@ def GetCompletionData(self, request, context):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetCompletionDataUnified(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_WellPathServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -58,6 +69,11 @@ def add_WellPathServicer_to_server(servicer, server):
                     request_deserializer=SimulatorTables__pb2.SimulatorTableRequest.FromString,
                     response_serializer=SimulatorTables__pb2.SimulatorTableData.SerializeToString,
             ),
+            'GetCompletionDataUnified': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetCompletionDataUnified,
+                    request_deserializer=SimulatorTables__pb2.SimulatorTableUnifiedRequest.FromString,
+                    response_serializer=SimulatorTables__pb2.SimulatorTableData.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'rips.WellPath', rpc_method_handlers)
@@ -95,3 +111,30 @@ def GetCompletionData(request,
             timeout,
             metadata,
             _registered_method=True)
+
+    @staticmethod
+    def GetCompletionDataUnified(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/rips.WellPath/GetCompletionDataUnified',
+            SimulatorTables__pb2.SimulatorTableUnifiedRequest.SerializeToString,
+            SimulatorTables__pb2.SimulatorTableData.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)