combined_vg.h

Go to the documentation of this file.

/* Author (s) :   Francois Mauger <mauger@lpccaen.in2p3.fr>
 * Creation date: 2011-04-22
 * Last modified: 2014-04-09
 *
 * License:
 *
 * Description:
 *
 *   Vertex generator that combines several weighted vertex generators
 *
 * History:
 *
 */

#ifndef GENVTX_COMBINED_VG_H
#define GENVTX_COMBINED_VG_H 1

// Standard library:
#include <vector>

// Third party:
// - Bayeux/datatools
#include <datatools/logger.h>

// This project:
#include <genvtx/vg_macros.h>
#include <genvtx/utils.h>

namespace genvtx {

  class combined_vg : public i_vertex_generator
  {

  public:

    struct entry_type
    {
      std::string name; 
      double weight; 
      double cumulated_weight; 
      genvtx::vg_handle_type vg_handle; 
      entry_type ();
    };

    void add_generator (genvtx::vg_handle_type & a_vg,
                        const std::string & a_name,
                        double a_weight = 1.0);

    virtual void tree_dump (std::ostream & out_ = std::clog,
                            const std::string & title_ = "",
                            const std::string & indent_ = "",
                            bool inherit_ = false) const;

    combined_vg();

    virtual ~combined_vg();

    virtual void initialize(const ::datatools::properties &,
                             ::datatools::service_manager &,
                             ::genvtx::vg_dict_type &);

    virtual void reset();

    virtual bool is_initialized() const;

  protected:

    virtual void _shoot_vertex(::mygsl::rng & random_,
                               ::geomtools::vector_3d & vertex_);

    void _shoot_vertex_combined (mygsl::rng & random_,
                                 geomtools::vector_3d & vertex_);

  private:

    void _init_ ();

    void _reset_ ();

    void _set_defaults_ ();

  private:

    bool _initialized_; 
    std::vector<entry_type> _entries_; 

    GENVTX_VG_REGISTRATION_INTERFACE(combined_vg)

  };

} // end of namespace genvtx

/***************
 * OCD support *
 ***************/
#include <datatools/ocd_macros.h>
DOCD_CLASS_DECLARATION(genvtx::combined_vg)

#endif // GENVTX_COMBINED_VG_H

/*
** Local Variables: --
** mode: c++ --
** c-file-style: "gnu" --
** tab-width: 2 --
** End: --
*/