add function to support include_transform

Author: zhangzhanqun

src/ansys/dyna/core/pre/Server/kwprocess.proto

@@ -22,6 +22,21 @@ message kwFileNameReply {
 	bool ret = 1;
 }
 
+message IncludeTransformRequest {
+	string filename = 1;
+	int32 idnoff = 2;
+	int32 ideoff = 3;
+	int32 idpoff = 4;
+	int32 idmoff = 5;
+	int32 idsoff = 6;
+	int32 idfoff = 7;
+	int32 tranid = 8;
+}
+
+message IncludeTransformReply {
+	int32 answer = 1;
+}
+
 //---control_* card
 message TimestepRequest {
 	float tssfac = 1;
@@ -715,6 +730,16 @@ message DefineBoxReply {
 	int32 boxid = 1;
 }
 
+message DefineTransformationRequest {
+    int32 tranid = 1;
+	repeated string option = 2;
+	repeated float param = 3;
+}
+
+message DefineTransformationReply {
+	int32 id = 1;
+}
+
 message DefineDEMeshSurfaceRequest {
     int32 sid = 1;
 	int32 type = 2;
@@ -1906,6 +1931,7 @@ message GeneralKWDReply {
 service kwC2S {
 	rpc kwSetFileName(kwFileName) returns(kwFileNameReply) {}
 	rpc Upload(stream Chunk) returns (kwFileReply) {}
+	rpc CreateIncludeTransform(IncludeTransformRequest) returns (IncludeTransformReply) {}
 	//---CONTROL
 	rpc CreateTimestep(TimestepRequest) returns (TimestepReply) {}
 	rpc CreateTermination(TerminationRequest) returns (TerminationReply) {}
@@ -1994,6 +2020,7 @@ service kwC2S {
 	rpc CreateDefineVector(DefineVectorRequest) returns (DefineVectorReply) {}
 	rpc CreateDefineFunction(DefineFunctionRequest) returns (DefineFunctionReply) {}
 	rpc CreateDefineBox(DefineBoxRequest) returns (DefineBoxReply) {}
+	rpc CreateDefineTransformation(DefineTransformationRequest) returns (DefineTransformationReply) {}
 	rpc CreateDefineDEMeshSurface(DefineDEMeshSurfaceRequest) returns (DefineDEMeshSurfaceReply) {}
 	rpc CreateDefineOrientation(DefineOrientationRequest) returns (DefineOrientationReply) {}
 	rpc CreateDefineSPHMassflowPlane(DefineSPHMassflowPlaneRequest) returns (DefineSPHMassflowPlaneReply) {}

src/ansys/dyna/core/pre/Server/kwprocess_pb2.py

@@ -24,6 +24,11 @@ def __init__(self, channel):
                 request_serializer=kwprocess__pb2.Chunk.SerializeToString,
                 response_deserializer=kwprocess__pb2.kwFileReply.FromString,
                 )
+        self.CreateIncludeTransform = channel.unary_unary(
+                '/kwgrpc.kwC2S/CreateIncludeTransform',
+                request_serializer=kwprocess__pb2.IncludeTransformRequest.SerializeToString,
+                response_deserializer=kwprocess__pb2.IncludeTransformReply.FromString,
+                )
         self.CreateTimestep = channel.unary_unary(
                 '/kwgrpc.kwC2S/CreateTimestep',
                 request_serializer=kwprocess__pb2.TimestepRequest.SerializeToString,
@@ -369,6 +374,11 @@ def __init__(self, channel):
                 request_serializer=kwprocess__pb2.DefineBoxRequest.SerializeToString,
                 response_deserializer=kwprocess__pb2.DefineBoxReply.FromString,
                 )
+        self.CreateDefineTransformation = channel.unary_unary(
+                '/kwgrpc.kwC2S/CreateDefineTransformation',
+                request_serializer=kwprocess__pb2.DefineTransformationRequest.SerializeToString,
+                response_deserializer=kwprocess__pb2.DefineTransformationReply.FromString,
+                )
         self.CreateDefineDEMeshSurface = channel.unary_unary(
                 '/kwgrpc.kwC2S/CreateDefineDEMeshSurface',
                 request_serializer=kwprocess__pb2.DefineDEMeshSurfaceRequest.SerializeToString,
@@ -836,6 +846,12 @@ def Upload(self, request_iterator, context):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CreateIncludeTransform(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 CreateTimestep(self, request, context):
         """---CONTROL
         """
@@ -1261,6 +1277,12 @@ def CreateDefineBox(self, request, context):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CreateDefineTransformation(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 CreateDefineDEMeshSurface(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -1823,6 +1845,11 @@ def add_kwC2SServicer_to_server(servicer, server):
                     request_deserializer=kwprocess__pb2.Chunk.FromString,
                     response_serializer=kwprocess__pb2.kwFileReply.SerializeToString,
             ),
+            'CreateIncludeTransform': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateIncludeTransform,
+                    request_deserializer=kwprocess__pb2.IncludeTransformRequest.FromString,
+                    response_serializer=kwprocess__pb2.IncludeTransformReply.SerializeToString,
+            ),
             'CreateTimestep': grpc.unary_unary_rpc_method_handler(
                     servicer.CreateTimestep,
                     request_deserializer=kwprocess__pb2.TimestepRequest.FromString,
@@ -2168,6 +2195,11 @@ def add_kwC2SServicer_to_server(servicer, server):
                     request_deserializer=kwprocess__pb2.DefineBoxRequest.FromString,
                     response_serializer=kwprocess__pb2.DefineBoxReply.SerializeToString,
             ),
+            'CreateDefineTransformation': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateDefineTransformation,
+                    request_deserializer=kwprocess__pb2.DefineTransformationRequest.FromString,
+                    response_serializer=kwprocess__pb2.DefineTransformationReply.SerializeToString,
+            ),
             'CreateDefineDEMeshSurface': grpc.unary_unary_rpc_method_handler(
                     servicer.CreateDefineDEMeshSurface,
                     request_deserializer=kwprocess__pb2.DefineDEMeshSurfaceRequest.FromString,
@@ -2662,6 +2694,23 @@ def Upload(request_iterator,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def CreateIncludeTransform(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, '/kwgrpc.kwC2S/CreateIncludeTransform',
+            kwprocess__pb2.IncludeTransformRequest.SerializeToString,
+            kwprocess__pb2.IncludeTransformReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def CreateTimestep(request,
             target,
@@ -3835,6 +3884,23 @@ def CreateDefineBox(request,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def CreateDefineTransformation(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, '/kwgrpc.kwC2S/CreateDefineTransformation',
+            kwprocess__pb2.DefineTransformationRequest.SerializeToString,
+            kwprocess__pb2.DefineTransformationReply.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def CreateDefineDEMeshSurface(request,
             target,

src/ansys/dyna/core/pre/Server/kwprocess_pb2_grpc.py

@@ -124,6 +124,23 @@ def CreateTermination(self, request, context):
         self.kwdproc.newkeyword(newk)
         print("Termination Created...")
         return kwprocess_pb2.TerminationReply(answer=0)
+    
+    def CreateIncludeTransform(self, request, context):
+        filename = request.filename
+        idnoff = request.idnoff
+        ideoff = request.ideoff
+        idpoff = request.idpoff
+        idmoff = request.idmoff
+        idsoff = request.idsoff
+        idfoff = request.idfoff
+        tranid = request.tranid
+        card1 = filename
+        card2 = str(idnoff) + "," + str(ideoff) + "," + str(idpoff) + "," + str(idmoff)+ "," + str(idsoff) + "," + str(idfoff)
+        card5 = str(tranid)
+        newk = "*INCLUDE_TRANSFORM\n" + card1 + "\n" + card2 + "\n\n\n" + card5
+        self.kwdproc.newkeyword(newk)
+        print("Include Transform Created...")
+        return kwprocess_pb2.IncludeTransformReply(answer=0)
 
     def CreateControlOutput(self, request, context):
         npopt = request.npopt
@@ -691,6 +708,21 @@ def CreateDefineBox(self, request, context):
         self.kwdproc.newkeyword(newk)
         print(f"DefineBox {boxid} Created...")
         return kwprocess_pb2.DefineBoxReply(boxid=boxid)
+    
+    def CreateDefineTransformation(self, request, context):
+        tranid = self.kwdproc.get_data(gdt.KWD_DEFINE_TRANSFORMATION_LASTID) + 1
+        option = request.option
+        param = request.param
+        card1 = str(tranid)
+        newk = "*DEFINE_TRANSFORMATION\n" + card1
+        i=0
+        for op in option:
+            subcard = str(op)+ ","+str(param[i])+ "," + str(param[i+1])+ ","+str(param[i+2])+ "," + str(param[i+3])+ ","+str(param[i+4])+ "," + str(param[i+5])+ ","+str(param[i+6])
+            newk += "\n" + subcard
+            i+=7
+        self.kwdproc.newkeyword(newk)
+        print(f"Define Transformation {tranid} Created...")
+        return kwprocess_pb2.DefineTransformationReply(id=tranid)
 
     def CreateDefineDEMeshSurface(self, request, context):
         sid = request.sid

src/ansys/dyna/core/pre/Server/kwserver.py

@@ -164,6 +164,33 @@ def __init__(self, x=0, y=0, z=0):
         self.z = z
 
 
+class Transform:
+    """Define a transformation."""
+
+    def __init__(self, option=None, param1=0, param2=0, param3=0, param4=0, param5=0, param6=0, param7=0):
+        param = [option, param1, param2, param3, param4, param5, param6, param7]
+        self.paramlist = []
+        self.paramlist.append(param)
+
+    def add_transform(self, option=None, param1=0, param2=0, param3=0, param4=0, param5=0, param6=0, param7=0):
+        """Define a transformation matrix."""
+        param = [option, param1, param2, param3, param4, param5, param6, param7]
+        self.paramlist.append(param)
+
+    def create(self, stub):
+        """Create transformation."""
+        options = []
+        params = []
+        for obj in self.paramlist:
+            options.append(obj[0])
+            for i in range(1, 8):
+                params.append(obj[i])
+        ret = stub.CreateDefineTransformation(DefineTransformationRequest(option=options, param=params))
+        self.id = ret.id
+        logging.info(f"Transformation {self.id} defined...")
+        return self.id
+
+
 class Velocity:
     """Define translational velocity."""
 
@@ -647,6 +674,45 @@ def add(self, obj):
         """Add entities in this object."""
         self.entities.append(obj)
 
+    def set_transform(self, filename=None, idnoff=0, ideoff=0, idpoff=0, idmoff=0, idsoff=0, idfoff=0, transform=None):
+        """Include independent input files containing model data,allow for node, element, and set
+        IDs to be offset and for coordinates and constitutive parameters to be transformed andnscaled.
+
+        Parameters
+        ----------
+        filename : string
+            File name of file to be included in this keyword file.
+        idnoff : int
+            Offset to node ID.
+        ideoff : int
+            Offset to element ID.
+        idpoff : int
+            Offset to part ID.
+        idmoff : int
+            Offset to material ID.
+        idsoff : int
+            Offset to set ID.
+        idfoff : int
+            Offset to function ID, table ID, and curve ID.
+        transform : Transform
+            Define a transformation.
+        """
+        tranid = transform.create(self.stub)
+        ret = self.stub.CreateIncludeTransform(
+            IncludeTransformRequest(
+                filename=filename,
+                idnoff=idnoff,
+                ideoff=ideoff,
+                idpoff=idpoff,
+                idmoff=idmoff,
+                idsoff=idsoff,
+                idfoff=idfoff,
+                tranid=tranid,
+            )
+        )
+        logging.info("Include transform Created...")
+        return ret
+
     def save_file(self):
         """Save keyword files.
 

src/ansys/dyna/core/pre/Server/lib/linux/keywordreader.so

@@ -624,15 +624,17 @@ class Isopotential:
     set : Set
         Segment Set or Node Set.
     """
+
     isopotlist = []
+
     def __init__(self, set=None):
         self.stub = DynaBase.get_stub()
         self.set = set
         self.id = 0
 
     def create(self):
         """Create Isopotential."""
-        isoinfo = [self.set.type,self.set.nodes]
+        isoinfo = [self.set.type, self.set.nodes]
         if isoinfo in Isopotential.isopotlist:
             pass
         id, settype = 0, 1