Line data    Source code

       1             : #ifndef CRPROPA_MODULETOOLS_H
       2             : #define CRPROPA_MODULETOOLS_H
       3             : 
       4             : #include "crpropa/Module.h"
       5             : #include "crpropa/EmissionMap.h"
       6             : 
       7             : #include <vector>
       8             : #include <set>
       9             : 
      10             : namespace crpropa {
      11             : /**
      12             :  * \addtogroup Tools
      13             :  * @{
      14             :  */
      15             : 
      16             : /**
      17             :  @class PerformanceModule
      18             :  @brief Module to monitor the simulation performance
      19             : 
      20             :  Add modules under investigation to this module instead of the ModuleList.
      21             :  */
      22           0 : class PerformanceModule: public Module {
      23             : private:
      24           0 :         struct _module_info {
      25             :                 double time;
      26             :                 ref_ptr<Module> module;
      27             :         };
      28             : 
      29             :         mutable std::vector<_module_info> modules;
      30             :         mutable size_t calls;
      31             : 
      32             : public:
      33             :         ~PerformanceModule();
      34             :         void add(Module* module);
      35             :         void process(Candidate* candidate) const;
      36             :         std::string getDescription() const;
      37             : };
      38             : 
      39             : /**
      40             :   @class ParticleFilter
      41             :   @brief Reject Particles not listed in filter.
      42             : */
      43             : class ParticleFilter: public AbstractCondition {
      44             :         std::set<int> ids;
      45             : 
      46             : public:
      47             :         ParticleFilter();
      48             :         ParticleFilter(const std::set<int> &ids);
      49             :         void addId(int id);
      50             :         void removeId(int remove);
      51             :         std::set<int> &getIds();
      52             : 
      53             :         void process(Candidate* candidate) const;
      54             :         std::string getDescription() const;
      55             : };
      56             : 
      57             : 
      58             : /**
      59             :   @class EmissionMapFiller
      60             :   @brief Fill EmissionMap with source particle state
      61             : */
      62             : class EmissionMapFiller: public Module {
      63             :         ref_ptr<EmissionMap> emissionMap;
      64             : public:
      65             :         EmissionMapFiller(EmissionMap *emissionMap);
      66             :         void setEmissionMap(EmissionMap *emissionMap);
      67             :         void process(Candidate* candidate) const;
      68             :         std::string getDescription() const;
      69             : };
      70             : 
      71             : /** @}*/
      72             : } // namespace crpropa
      73             : 
      74             : #endif // CRPROPA_MODULETOOLS_H