Expose Local Speed of Sound via Python Wrapper & Implement Custom Thermal Boundaries

Omgiri01 · Omgiri01 · commit eb45b451fbee · 2026-03-25T14:39:31.000+05:30
diff --git a/Common/include/CConfig.hpp b/Common/include/CConfig.hpp
@@ -326,6 +326,7 @@ class CConfig {
   su2double *Engine_Area;                    /*!< \brief Specified engine area for nacelle boundaries. */
   su2double *Outlet_Pressure;                /*!< \brief Specified back pressures (static) for outlet boundaries. */
   su2double *Isothermal_Temperature;         /*!< \brief Specified isothermal wall temperatures (static). */
+  mutable std::map<unsigned short, std::map<unsigned long, su2double>> Marker_Custom_Temperature; /*!< \brief Spatially varying isothermal wall temperatures. */
   su2double *HeatTransfer_Coeff;             /*!< \brief Specified heat transfer coefficients. */
   su2double *HeatTransfer_WallTemp;          /*!< \brief Specified temperatures at infinity alongside heat transfer coefficients. */
   su2double *Heat_Flux;                      /*!< \brief Specified wall heat fluxes. */
@@ -7404,6 +7405,22 @@ class CConfig {
    */
   su2double GetIsothermal_Temperature(const string& val_index) const;
 
+  /*!
+   * \brief Get the custom wall temperature (static) at an isothermal boundary node.
+   * \param[in] iMarker - Marker index.
+   * \param[in] iVertex - Vertex index on the marker.
+   * \return The custom wall temperature (returns -1.0 if not set).
+   */
+  su2double GetCustom_Temperature(unsigned short iMarker, unsigned long iVertex) const;
+
+  /*!
+   * \brief Set the custom wall temperature (static) at an isothermal boundary node.
+   * \param[in] iMarker - Marker index.
+   * \param[in] iVertex - Vertex index on the marker.
+   * \param[in] val - The custom wall temperature.
+   */
+  void SetCustom_Temperature(unsigned short iMarker, unsigned long iVertex, su2double val);
+
   /*!
    * \brief Get the wall heat flux on a constant heat flux boundary.
    * \param[in] val_index - Index corresponding to the constant heat flux boundary.
diff --git a/Common/src/CConfig.cpp b/Common/src/CConfig.cpp
@@ -9608,6 +9608,21 @@ su2double CConfig::GetIsothermal_Temperature(const string& val_marker) const {
   return Isothermal_Temperature[0];
 }
 
+su2double CConfig::GetCustom_Temperature(unsigned short iMarker, unsigned long iVertex) const {
+  auto it_marker = Marker_Custom_Temperature.find(iMarker);
+  if (it_marker != Marker_Custom_Temperature.end()) {
+    auto it_vertex = it_marker->second.find(iVertex);
+    if (it_vertex != it_marker->second.end()) {
+      return it_vertex->second;
+    }
+  }
+  return -1.0;
+}
+
+void CConfig::SetCustom_Temperature(unsigned short iMarker, unsigned long iVertex, su2double val) {
+  Marker_Custom_Temperature[iMarker][iVertex] = val;
+}
+
 su2double CConfig::GetWall_HeatFlux(const string& val_marker) const {
 
   for (unsigned short iMarker_HeatFlux = 0; iMarker_HeatFlux < nMarker_HeatFlux; iMarker_HeatFlux++)
diff --git a/SU2_CFD/include/drivers/CDriver.hpp b/SU2_CFD/include/drivers/CDriver.hpp
@@ -555,6 +555,30 @@ class CDriver : public CDriverBase {
    */
   void SetMarkerTranslationRate(unsigned short iMarker, passivedouble vel_x, passivedouble vel_y, passivedouble vel_z);
 
+  /*!
+   * \brief Set a custom temperature at a boundary vertex.
+   * \param[in] iMarker - Index of the boundary marker.
+   * \param[in] iVertex - Index of the boundary vertex.
+   * \param[in] Temperature - Temperature value.
+   */
+  void SetMarkerCustomTemperature(unsigned short iMarker, unsigned long iVertex, su2double Temperature);
+
+  /*!
+   * \brief Get the local speed of sound at a boundary vertex.
+   * \param[in] iMarker - Index of the boundary marker.
+   * \param[in] iVertex - Index of the boundary vertex.
+   * \return Local speed of sound.
+   */
+  passivedouble GetMarkerLocalSpeedOfSound(unsigned short iMarker, unsigned long iVertex);
+
+  /*!
+   * \brief Get the Mach number at a boundary vertex.
+   * \param[in] iMarker - Index of the boundary marker.
+   * \param[in] iVertex - Index of the boundary vertex.
+   * \return Mach number.
+   */
+  passivedouble GetMarkerMachNumber(unsigned short iMarker, unsigned long iVertex);
+
   /*!
    * \brief Get the Freestream Density for nondimensionalization
    * \return Freestream Density
diff --git a/SU2_CFD/src/output/CFlowCompOutput.cpp b/SU2_CFD/src/output/CFlowCompOutput.cpp
@@ -166,6 +166,8 @@ void CFlowCompOutput::SetHistoryOutputFields(CConfig *config){
   AddHistoryOutput("MIN_DELTA_TIME", "Min DT", ScreenOutputFormat::SCIENTIFIC, "CFL_NUMBER", "Current minimum local time step");
   AddHistoryOutput("MAX_DELTA_TIME", "Max DT", ScreenOutputFormat::SCIENTIFIC, "CFL_NUMBER", "Current maximum local time step");
 
+  AddHistoryOutput("AVG_SOUND_SPEED", "Avg[a]", ScreenOutputFormat::FIXED, "PRIMITIVE", "Average local speed of sound.", HistoryFieldType::DEFAULT);
+
   AddHistoryOutput("MIN_CFL", "Min CFL", ScreenOutputFormat::SCIENTIFIC, "CFL_NUMBER", "Current minimum of the local CFL numbers");
   AddHistoryOutput("MAX_CFL", "Max CFL", ScreenOutputFormat::SCIENTIFIC, "CFL_NUMBER", "Current maximum of the local CFL numbers");
   AddHistoryOutput("AVG_CFL", "Avg CFL", ScreenOutputFormat::SCIENTIFIC, "CFL_NUMBER", "Current average of the local CFL numbers");
@@ -238,6 +240,7 @@ void CFlowCompOutput::SetVolumeOutputFields(CConfig *config){
   AddVolumeOutput("PRESSURE",    "Pressure",                "PRIMITIVE", "Pressure");
   AddVolumeOutput("TEMPERATURE", "Temperature",             "PRIMITIVE", "Temperature");
   AddVolumeOutput("MACH",        "Mach",                    "PRIMITIVE", "Mach number");
+  AddVolumeOutput("LOCAL_SPEED_OF_SOUND", "Local_Speed_of_Sound", "PRIMITIVE", "Local speed of sound");
   AddVolumeOutput("PRESSURE_COEFF", "Pressure_Coefficient", "PRIMITIVE", "Pressure coefficient");
   AddVolumeOutput("VELOCITY-X", "Velocity_x", "PRIMITIVE", "x-component of the velocity vector");
   AddVolumeOutput("VELOCITY-Y", "Velocity_y", "PRIMITIVE", "y-component of the velocity vector");
@@ -325,6 +328,8 @@ void CFlowCompOutput::LoadVolumeData(CConfig *config, CGeometry *geometry, CSolv
     SetVolumeOutputValue("ENERGY",     iPoint, Node_Flow->GetSolution(iPoint, 3));
   }
 
+  SetVolumeOutputValue("LOCAL_SPEED_OF_SOUND", iPoint, Node_Flow->GetSoundSpeed(iPoint));
+
   if (gridMovement){
     SetVolumeOutputValue("GRID_VELOCITY-X", iPoint, Node_Geo->GetGridVel(iPoint)[0]);
     SetVolumeOutputValue("GRID_VELOCITY-Y", iPoint, Node_Geo->GetGridVel(iPoint)[1]);
@@ -429,6 +434,26 @@ void CFlowCompOutput::LoadHistoryData(CConfig *config, CGeometry *geometry, CSol
   SetHistoryOutputValue("MIN_DELTA_TIME", flow_solver->GetMin_Delta_Time());
   SetHistoryOutputValue("MAX_DELTA_TIME", flow_solver->GetMax_Delta_Time());
 
+  /*--- Compute average local speed of sound for screen output ---*/
+  su2double avg_a = 0.0;
+  unsigned long nPoint = geometry->GetnPointDomain();
+  const auto* Node_Flow = flow_solver->GetNodes();
+  for (unsigned long iPoint = 0; iPoint < nPoint; iPoint++) {
+    avg_a += Node_Flow->GetSoundSpeed(iPoint);
+  }
+  su2double total_a = 0.0;
+  unsigned long total_points = 0;
+#ifdef HAVE_MPI
+  unsigned long nPoint_local = nPoint;
+  SU2_MPI::Allreduce(&avg_a, &total_a, 1, MPI_DOUBLE, MPI_SUM, SU2_MPI::GetComm());
+  SU2_MPI::Allreduce(&nPoint_local, &total_points, 1, MPI_UNSIGNED_LONG, MPI_SUM, SU2_MPI::GetComm());
+#else
+  total_a = avg_a;
+  total_points = nPoint;
+#endif
+  if (total_points > 0) total_a /= (su2double)total_points;
+  SetHistoryOutputValue("AVG_SOUND_SPEED", total_a);
+
   SetHistoryOutputValue("MIN_CFL", flow_solver->GetMin_CFL_Local());
   SetHistoryOutputValue("MAX_CFL", flow_solver->GetMax_CFL_Local());
   SetHistoryOutputValue("AVG_CFL", flow_solver->GetAvg_CFL_Local());
diff --git a/SU2_CFD/src/python_wrapper_structure.cpp b/SU2_CFD/src/python_wrapper_structure.cpp
@@ -28,6 +28,7 @@
 #include "../../Common/include/toolboxes/geometry_toolbox.hpp"
 #include "../include/drivers/CDriver.hpp"
 #include "../include/drivers/CSinglezoneDriver.hpp"
+#include <cmath>
 
 void CDriver::PreprocessPythonInterface(CConfig** config, CGeometry**** geometry, CSolver***** solver) {
   int rank = MASTER_NODE;
@@ -179,3 +180,23 @@ void CDriver::SetMarkerTranslationRate(unsigned short iMarker, passivedouble vel
   config_container[selected_zone]->SetMarkerTranslationRate(iMarker, 2, vel_z);
 }
 
+void CDriver::SetMarkerCustomTemperature(unsigned short iMarker, unsigned long iVertex, su2double Temperature) {
+  geometry_container[selected_zone][INST_0][MESH_0]->SetCustomBoundaryTemperature(iMarker, iVertex, Temperature);
+}
+
+passivedouble CDriver::GetMarkerLocalSpeedOfSound(unsigned short iMarker, unsigned long iVertex) {
+  const auto* geometry = geometry_container[selected_zone][INST_0][MESH_0];
+  const auto* flow_solver = solver_container[selected_zone][INST_0][MESH_0][FLOW_SOL];
+  unsigned long iPoint = geometry->vertex[iMarker][iVertex]->GetNode();
+  return SU2_TYPE::GetValue(flow_solver->GetNodes()->GetSoundSpeed(iPoint));
+}
+
+passivedouble CDriver::GetMarkerMachNumber(unsigned short iMarker, unsigned long iVertex) {
+  const auto* geometry = geometry_container[selected_zone][INST_0][MESH_0];
+  const auto* flow_solver = solver_container[selected_zone][INST_0][MESH_0][FLOW_SOL];
+  unsigned long iPoint = geometry->vertex[iMarker][iVertex]->GetNode();
+  su2double V2 = flow_solver->GetNodes()->GetVelocity2(iPoint);
+  su2double a = flow_solver->GetNodes()->GetSoundSpeed(iPoint);
+  return SU2_TYPE::GetValue(sqrt(V2) / a);
+}
+
