Printing variants count

UofUEpiBio · Dec 6, 2023 · e9a460e · e9a460e
1 parent 68de576
commit e9a460e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/epiworld.hpp b/epiworld.hpp
@@ -9004,6 +9004,14 @@ inline const Model<TSeq> & Model<TSeq>::print(bool lite) const
 
     }
 
+    auto nvariants = db.get_n_viruses() - n_viruses_model;
+    if (nvariants > 0)
+    {
+
+        printf_epiworld(" ...and %li more variants...\n", nvariants);
+
+    }
+
     if (viruses.size() == 0u)
     {
         printf_epiworld(" (none)\n");
@@ -18352,7 +18360,7 @@ inline ModelSEIRDCONN<TSeq> & ModelSEIRDCONN<TSeq>::initial_states(
 ) {
 
     Model<TSeq>::initial_states_fun =
-        create_init_function_seir<TSeq>(proportions_)
+        create_init_function_seird<TSeq>(proportions_)
         ;
 
     return *this;

diff --git a/include/epiworld/model-meat-print.hpp b/include/epiworld/model-meat-print.hpp
@@ -184,6 +184,14 @@ inline const Model<TSeq> & Model<TSeq>::print(bool lite) const
 
     }
 
+    auto nvariants = db.get_n_viruses() - n_viruses_model;
+    if (nvariants > 0)
+    {
+
+        printf_epiworld(" ...and %li more variants...\n", nvariants);
+
+    }
+
     if (viruses.size() == 0u)
     {
         printf_epiworld(" (none)\n");