include/N3AnaBase/cuts/source_cut.h

#ifndef __N3AnaBase__source_cut_h
#define __N3AnaBase__source_cut_h 1

#include <cstdlib>
#include <iostream>
#include <stdexcept>
#include <string>
#include <list>

#include <N3AnaBase/cuts/i_ana_event_cut.h>
#include <N3AnaBase/utils/source_utils.h>

namespace nemo3 {
  

  class source_cut : public nemo3::i_ana_event_cut
    {
    public:
      enum mode_t
        {
          MODE_INVALID            = -1,
          MODE_ALL_TRACKS         = 0,
          MODE_AT_LEAST_ONE_TRACK = 1
        };

      static int get_mode_from_label (const std::string & label_);

      static std::string get_label_from_mode (int mode_);

      static bool is_mode_valid (int mode_);

    private:
      bool __debug;
      int  __mode;
      bool __delayed;
      std::list<int>  __source_mask;
      std::string __tracking_name;

    public:

      source_cut (bool debug_ = false); 

      virtual ~source_cut (); 

    public:

      bool is_source_molybdenum_metallic () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_MO_METAL) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_molybdenum_composite () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_MO_COMPOSITE) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_selenium () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_SE) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_tellurium_natural () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_TE_NATURAL) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_tellurium_enriched () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_TE_ENRICHED) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_zirconium () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_ZR) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_neodymum () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_ND) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_calcium () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_CA) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_copper () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_CU) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_cadmium () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_CD) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_calibration_tube () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_CALIB_TUBE) == __source_mask.end())
          return false;
        return true;
      }

      bool is_source_endcap () const
      {
        if(find (__source_mask.begin(), __source_mask.end(), source_utils::SOURCE_ENDCAP) == __source_mask.end())
          return false;
        return true;
      }

      bool is_prompt () const
      {
        return ! __delayed;
      }

      bool is_delayed () const
      {
        return __delayed;
      }

      bool is_mode_all_tracks () const
      {
        return __mode == MODE_ALL_TRACKS;
      }

      bool is_mode_at_least_one_track () const
      {
        return __mode == MODE_AT_LEAST_ONE_TRACK;
      }

      bool is_debug () const;

      void set_debug (bool);

      void set (int mode_, 
                std::list<int> source_mask_,
                bool delayed_,
                const std::string & tracking_name_ = "");

      void dump (std::ostream & out_ = std::clog) const;

    public:

      bool accept (); 

    protected:

      virtual void __prepare_cut ();
      
    private:
      
      static creator_registration<source_cut> __CR;

    public:

      static const std::string CUT_ID;

      virtual std::string cut_id () const;

      virtual cut_creator_t cut_creator () const;

      static nemo3::i_ana_event_cut * create (const nemocuts::ICutFactory::cut_properties_t & cut_properties_);

    };

} // end of namespace nemo3 

#endif /* __N3AnaBase__source_cut_h */

// end of source_cut.h