add descriptor box for IC fields

E3SM-Project · Nov 21, 2024 · cf476bb · cf476bb
1 parent 2e6e29e
commit cf476bb
Showing 1 changed file with 23 additions and 3 deletions.
diff --git a/components/eamxx/src/share/atm_process/atmosphere_process_dag.cpp b/components/eamxx/src/share/atm_process/atmosphere_process_dag.cpp
@@ -183,6 +183,7 @@ void AtmProcDAG::write_dag (const std::string& fname, const int verbosity) const
   ofile << "strict digraph G {\n"
         << "rankdir=\"LR\"\n";
 
+  bool has_IC_field = false;
   for (const auto& n : m_nodes) {
     const auto& unmet = m_unmet_deps.at(n.id);
 
@@ -268,10 +269,11 @@ void AtmProcDAG::write_dag (const std::string& fname, const int verbosity) const
           }
           fc += "\">  ";
           ofile << "      <tr><td align=\"left\">" << fc << html_fix(print_fid(m_fids[fid],fid_verb));
-          if (ekat::contains(m_unmet_deps.at(n.id), fid)) {
+          if (ekat::contains(unmet, fid)) {
             ofile << "<b>  *** MISSING ***</b>";
-          } else if (ekat::contains(m_unmet_deps.at(n.id), -fid)) {
+          } else if (ekat::contains(unmet, -fid)) {
             ofile << "<b>  (Init. Cond.)</b>";
+            has_IC_field = true;
           }
           ofile << "</font></td></tr>\n";
         }
@@ -339,7 +341,7 @@ void AtmProcDAG::write_dag (const std::string& fname, const int verbosity) const
           fc += (ekat::contains(unmet,gr_fid) ? "red" : "black");
           fc += "\">  ";
           ofile << "      <tr><td align=\"left\">" << fc << html_fix(print_fid(m_fids[gr_fid],fid_verb));
-          if (ekat::contains(m_unmet_deps.at(n.id),gr_fid)) {
+          if (ekat::contains(unmet, gr_fid)) {
             ofile << "<b>  *** MISSING ***</b>";
           }
           ofile << "</font></td></tr>\n";
@@ -407,6 +409,24 @@ void AtmProcDAG::write_dag (const std::string& fname, const int verbosity) const
           << "];\n";
   }
 
+  if (m_IC_processed && has_IC_field) {
+    int this_node_id = m_nodes.size() + 1;
+    ofile << this_node_id << " [\n"
+          << "  shape=box\n"
+          << "  color=\"#605d57\"\n"
+          << "  fontcolor=\"#031576\"\n"
+          << "  penwidth=8\n"
+          << "  fontsize=40\n"
+          << "  style=filled\n"
+          << "  fillcolor=\"#cccccc\"\n"
+          << "  align=\"center\"\n"
+          << "  label=<<b><font color=\"#3d2906\">NOTE:</font> Fields denoted "
+             "with <font color=\"#006219\"><b>green text</b></font> "
+             "<br align=\"center\"/>indicate the field was provided by the "
+             "<br align=\"center\"/>initial conditions and never updated</b>>\n"
+          << "];\n";
+  }
+
   // Close the file
   ofile << "}";
   ofile.close();