From 00b7891e67bf4981b0c85e9f1f6fa43a6f758e35 Mon Sep 17 00:00:00 2001
From: minzhang0929 <78373564+minzhang0929@users.noreply.github.com>
Date: Thu, 10 Nov 2022 13:19:58 +0800
Subject: [PATCH 1/2] Update dfChemistryModel.H

---
 src/dfChemistryModel/dfChemistryModel.H | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/dfChemistryModel/dfChemistryModel.H b/src/dfChemistryModel/dfChemistryModel.H
index 369f107e..75fab9a7 100644
--- a/src/dfChemistryModel/dfChemistryModel.H
+++ b/src/dfChemistryModel/dfChemistryModel.H
@@ -125,7 +125,8 @@ public IOdictionary
     volScalarField& psi_;
     // heat release rate, [J/m^3/s]
     volScalarField Qdot_;
-
+    // DNN selection fields 
+    volScalarField selectDNN_;
 
 #ifdef USE_LIBTORCH
     Switch torchSwitch_;

From 1e99b239f86a5f019fe864e2a59a0c7973b8e000 Mon Sep 17 00:00:00 2001
From: minzhang0929 <78373564+minzhang0929@users.noreply.github.com>
Date: Thu, 10 Nov 2022 13:24:24 +0800
Subject: [PATCH 2/2] Update dfChemistryModel.C

---
 src/dfChemistryModel/dfChemistryModel.C | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/src/dfChemistryModel/dfChemistryModel.C b/src/dfChemistryModel/dfChemistryModel.C
index 334be4d0..8ac219dc 100644
--- a/src/dfChemistryModel/dfChemistryModel.C
+++ b/src/dfChemistryModel/dfChemistryModel.C
@@ -83,6 +83,19 @@ Foam::dfChemistryModel<ThermoType>::dfChemistryModel
         mesh_,
         dimensionedScalar(dimEnergy/dimVolume/dimTime, 0)
     ),
+    selectDNN_
+    (
+        IOobject
+        (
+            "selectDNN",
+            mesh_.time().timeName(),
+            mesh_,
+            IOobject::NO_READ,
+            IOobject::AUTO_WRITE
+        ),
+        mesh_,
+        dimensionedScalar(dimless, -1)
+    ),
     balancer_(createBalancer()),
     cpuTimes_
     (
@@ -522,18 +535,21 @@ Foam::dfChemistryModel<ThermoType>::getGPUProblems
         {
             problem.DNNid = 0;
             problemList.append(problem);
+            selectDNN_[cellI]=0;
             continue;
         }
         if(((Qdot_[cellI] >= Qdotact2_) && (T_[cellI] < Tact2_)&&(T_[cellI] >= Tact1_))||((Qdot_[cellI] > Qdotact3_) && T_[cellI] > Tact2_))  //choose2
         {
             problem.DNNid = 1;
             problemList.append(problem);
+            selectDNN_[cellI]=1;
             continue;
         }
         if  ((Qdot_[cellI] < Qdotact3_) && (T_[cellI] >= Tact2_) && (Qdot_[cellI]!=0)) //choose3
         {
             problem.DNNid = 2;
             problemList.append(problem);
+            selectDNN_[cellI]=2;
             continue;
         }