Add ctmodel::models() method that allows assigning of a model container for modelling.

Jürgen Knödlseder
1 parent 413461b4
Showing 3 changed files with 20 additions and 1 deletions
pyext/ctmodel.i
src/ctmodel/ctmodel.cpp
src/ctmodel/ctmodel.hpp
 /***************************************************************************
  *                  ctmodel - Model cube generation tool                   *
  * ----------------------------------------------------------------------- *
- *  copyright (C) 2012-2014 by Juergen Knoedlseder                         *
+ *  copyright (C) 2012-2015 by Juergen Knoedlseder                         *
  * ----------------------------------------------------------------------- *
  *                                                                         *
  *  This program is free software: you can redistribute it and/or modify   *
@@ -50,6 +50,7 @@ public:
     const GObservations& obs(void) const;
     const GCTAEventCube& cube(void) const;
     void                 cube(const GCTAEventCube& cube);
+    void                 models(const GModels& models);
 };
@@ -432,6 +432,8 @@ void ctmodel::get_parameters(void)
     // Read model definition file if required
     if (m_obs.models().size() == 0) {
+
+        // Get model filename
         std::string inmodel = (*this)["inmodel"].filename();
         // Load models from file
@@ -71,6 +71,7 @@ public:
     const GObservations& obs(void) const;
     const GCTAEventCube& cube(void) const;
     void                 cube(const GCTAEventCube& cube);
+    void                 models(const GModels& models);
 protected:
     // Protected methods
@@ -130,4 +131,19 @@ bool ctmodel::has_cube(void) const
     return m_has_cube;
 }
+
+/***********************************************************************//**
+ * @brief Set models
+ *
+ * @param[in] models Model container.
+ *
+ * Set model container that should be used for model generation.
+ ***************************************************************************/
+inline
+void ctmodel::models(const GModels& models)
+{
+    m_obs.models(models);
+    return;
+}
+
 #endif /* CTMODEL_HPP */