Added tscalc parameter to GModel and use it to write GModelSky to file

Michael Mayer
1 parent eb131e63
Showing 3 changed files with 10 additions and 0 deletions
include/GModel.hpp
src/model/GModel.cpp
src/model/GModelSky.cpp
@@ -163,6 +163,7 @@ protected:
     std::vector<std::string> m_ids;          //!< Identifiers to which model applies
     std::vector<GModelPar*>  m_pars;         //!< Pointers to all model parameters
     bool                     m_has_ts;       //!< Signals if TS is available
+    bool                     m_has_tscalc; //!< Signals if tscalc attribute is available
     bool                     m_tscalc;       //!< Signals if TS should be computed
     double                   m_ts;           //!< Test Statistic of the model
 };
@@ -304,6 +305,7 @@ inline
 void GModel::tscalc(const bool& tscalc)
 {
     m_tscalc = tscalc;
+    m_has_tscalc = true;
     return;
 }
  
@@ -556,6 +556,7 @@ void GModel::init_members(void)
     m_pars.clear();
     m_ts     = 0.0;
     m_has_ts = false;
+    m_has_tscalc = false;
     m_tscalc = false;
  
     // Return
@@ -578,6 +579,7 @@ void GModel::copy_members(const GModel&amp; model)
     m_pars        = model.m_pars;
     m_ts          = model.m_ts;
     m_has_ts      = model.m_has_ts;
+    m_has_tscalc = model.m_has_tscalc;
     m_tscalc      = model.m_tscalc;
  
     // Return
@@ -761,6 +761,12 @@ void GModelSky::write(GXmlElement&amp; xml) const
         src->attribute("ts", gammalib::str(ts(), 3));
     }
  
+    // If tscalc parameter was available, then write it to XML
+    if (m_has_tscalc) {
+        std::string ts_calc = tscalc() ? "1" : "0";
+        src->attribute("tscalc", ts_calc);
+    }
+
     // Write spectral model
     if (spectral() != NULL) {
         GXmlElement* spec = src->element("spectrum", 0);