From d2c6144d82f93651e03e626388317916fdd5c792 Mon Sep 17 00:00:00 2001 From: Andrew Myers Date: Fri, 5 Apr 2024 17:08:42 -0700 Subject: [PATCH 1/2] Add agent variable names to output --- src/IO.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/IO.cpp b/src/IO.cpp index 8e2fc05..a4ffb48 100644 --- a/src/IO.cpp +++ b/src/IO.cpp @@ -61,8 +61,9 @@ void writePlotFile (const AgentContainer& pc, /*!< Agent (particle) container {"total", "never_infected", "infected", "immune", "susceptible", "unit", "FIPS", "Tract", "comm"}, pc.ParticleGeom(0), cur_time, step); - // uncomment this to write all the particles - pc.WritePlotFile(amrex::Concatenate("plt", step, 5), "agents"); + pc.WritePlotFile(amrex::Concatenate("plt", step, 5), "agents", + {"disease_counter", "treantment_timer", "infection_prob", "incubation_period", "infectious_period", "symptomdev_period"}, + {"status", "strain", "age_group", "family", "home_i", "home_j", "work_i", "work_j", "nborhood", "school", "workgroup", "work_nborhood", "withdrawn", "symptomatic"}); } /*! \brief Writes diagnostic data by FIPS code From c9653ed163f776a863048c359edc2ca7a6b83f6b Mon Sep 17 00:00:00 2001 From: Andrew Myers Date: Fri, 5 Apr 2024 17:10:04 -0700 Subject: [PATCH 2/2] Update src/IO.cpp --- src/IO.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/IO.cpp b/src/IO.cpp index a4ffb48..8f36d48 100644 --- a/src/IO.cpp +++ b/src/IO.cpp @@ -62,7 +62,7 @@ void writePlotFile (const AgentContainer& pc, /*!< Agent (particle) container pc.ParticleGeom(0), cur_time, step); pc.WritePlotFile(amrex::Concatenate("plt", step, 5), "agents", - {"disease_counter", "treantment_timer", "infection_prob", "incubation_period", "infectious_period", "symptomdev_period"}, + {"disease_counter", "treatment_timer", "infection_prob", "incubation_period", "infectious_period", "symptomdev_period"}, {"status", "strain", "age_group", "family", "home_i", "home_j", "work_i", "work_j", "nborhood", "school", "workgroup", "work_nborhood", "withdrawn", "symptomatic"}); }