adding 'ENERGY' in CPrimitiveIndices to get index of energy equation

Author: Cristopher-Morales

SU2_CFD/include/variables/CPrimitiveIndices.hpp

@@ -124,6 +124,9 @@ std::map<std::string, IndexType> PrimitiveNameToIndexMap(const CPrimitiveIndices
   nameToIndex["VELOCITY_Y"] = idx.Velocity() + 1;
   if (idx.NDim() == 3) {
     nameToIndex["VELOCITY_Z"] = idx.Velocity() + 2;
+    nameToIndex["ENERGY"] = idx.Velocity() + 3;
+  } else {
+    nameToIndex["ENERGY"] = idx.Velocity() + 2;
   }
   for (IndexType iSpecies = 0; iSpecies < idx.NSpecies(); ++iSpecies) {
     nameToIndex["DENSITY_" + std::to_string(iSpecies)] = idx.SpeciesDensities() + iSpecies;

TestCases/py_wrapper/turbulent_premixed_psi/run.py

@@ -95,8 +95,10 @@ def update_temperature(SU2Driver, iPoint):
     iFLOWSOLVER = SU2Driver.GetSolverIndices()['INC.FLOW']
     # the list with names
     solindex = getsolvar(SU2Driver)
-    iTEMP = solindex.get("TEMPERATURE")
-    SU2Driver.Solution(iFLOWSOLVER).Set(iPoint,iTEMP,T)
+    # get solution index for energy equation
+    iEnergy = solindex.get("ENERGY")
+    # set solution in energy equation
+    SU2Driver.Solution(iFLOWSOLVER).Set(iPoint,iEnergy,T)
 
 
 # ################################################################## #