Passing dimensions from preCICE config to adapter

Author: IshaanDesai

cht/adapter.py

@@ -142,8 +142,10 @@ def configure(self, names):
 
 		self.computeNormals()
 
+		dims = self.precice.get_dimensions()
+
 		self.nodeCoordinates = np.array([p for p in self.SHMESH.sdj.COORDO.VALE.get()])
-		self.nodeCoordinates = np.resize(self.nodeCoordinates, (int(len(self.nodeCoordinates)/3), 3))
+		self.nodeCoordinates = np.resize(self.nodeCoordinates, (int(len(self.nodeCoordinates)/dims), dims))
 		self.shiftMesh()
 
 		self.faces = [self.mesh.correspondance_mailles[idx] for idx in self.mesh.gma[self.groupName]]
@@ -170,7 +172,10 @@ def computeNormals(self):
 			OPERATION='NORMALE'
 		)
 		self.normals = N.EXTR_COMP().valeurs
-		self.normals = np.resize(np.array(self.normals), (int(len(self.normals)/3), 3))
+
+		dims = self.precice.get_dimensions()
+
+		self.normals = np.resize(np.array(self.normals), (int(len(self.normals)/dims), dims))
 		DETRUIRE(CONCEPT=({"NOM": N}, {"NOM": DUMMY}))
 
 	def setVertices(self):
@@ -266,10 +271,11 @@ def setLoad(self, LOAD):
 		self.LOAD = LOAD
 
 	def shiftMesh(self):
+		dims = self.precice.get_dimensions()
 		coords = [p for p in self.SHMESH.sdj.COORDO.VALE.get()]
 		for i in range(len(self.normals)):
-			for c in range(3):
-				coords[i*3 + c] = coords[i*3 + c] - self.normals[i][c] * self.delta
+			for c in range(dims):
+				coords[i*dims + c] = coords[i*dims + c] - self.normals[i][c] * self.delta
 		self.SHMESH.sdj.COORDO.VALE.changeJeveuxValues(len(coords), tuple(range(1, len(coords)+1)), tuple(coords), tuple(coords), 1)
 
 	def updateConductivity(self, T):