Apply suggestions from code review

Co-authored-by: Kürşat Yurt <57598663+kursatyurt@users.noreply.github.com>

Author: pcarruscag

SU2_CFD/include/drivers/CDriver.hpp

@@ -535,6 +535,7 @@ class CDriver : public CDriverBase {
    * \param[in] rot_z - Value of Angular velocity about z-axes.
    */
   void SetRotationRate(passivedouble rot_x, passivedouble rot_y, passivedouble rot_z);
+
   /*!
    * \brief Set the moving wall marker rotation rates.
    * \param[in] iMaker - Index of moving wall marker.

SU2_CFD/src/python_wrapper_structure.cpp

@@ -168,18 +168,14 @@ void CDriver::SetRotationRate(passivedouble rot_x, passivedouble rot_y, passived
 }
 
 void CDriver::SetMarkerRotationRate(unsigned short iMarker, passivedouble rot_x, passivedouble rot_y, passivedouble rot_z) {
-  for (iZone = 0; iZone < nZone; iZone++) {
-    config_container[iZone]->SetMarkerRotationRate(iMarker, 0, rot_x);
-    config_container[iZone]->SetMarkerRotationRate(iMarker, 1, rot_y);
-    config_container[iZone]->SetMarkerRotationRate(iMarker, 2, rot_z);
-  }
+  config_container[selected_zone]->SetMarkerRotationRate(iMarker, 0, rot_x);
+  config_container[selected_zone]->SetMarkerRotationRate(iMarker, 1, rot_y);
+  config_container[selected_zone]->SetMarkerRotationRate(iMarker, 2, rot_z);
 }
 
 void CDriver::SetMarkerTranslationRate(unsigned short iMarker, passivedouble vel_x, passivedouble vel_y, passivedouble vel_z) {
-  for (iZone = 0; iZone < nZone; iZone++) {
-    config_container[iZone]->SetMarkerTranslationRate(iMarker, 0, vel_x);
-    config_container[iZone]->SetMarkerTranslationRate(iMarker, 1, vel_y);
-    config_container[iZone]->SetMarkerTranslationRate(iMarker, 2, vel_z);
-  }
+  config_container[selected_zone]->SetMarkerTranslationRate(iMarker, 0, vel_x);
+  config_container[selected_zone]->SetMarkerTranslationRate(iMarker, 1, vel_y);
+  config_container[selected_zone]->SetMarkerTranslationRate(iMarker, 2, vel_z);
 }
 

TestCases/py_wrapper/rotating_cylinder/run.py

@@ -13,9 +13,8 @@
 
 SU2Driver = pysu2.CSinglezoneDriver("spinning_cylinder.cfg",1, comm)
 
-for i in range(n_of_steps):
-    
-    SU2Driver.SetMarkerRotationRate(0,0,0,rotation_vector[n_of_steps-i-1])
+for i, rate in enumerate(rotation_vector):
+    SU2Driver.SetMarkerRotationRate(0,0,0,rate)
     SU2Driver.Preprocess(i)
     SU2Driver.Run()
     SU2Driver.Postprocess()