#ifndef __JROOT__JROOTCLASSSELECTOR__
#define __JROOT__JROOTCLASSSELECTOR__

#include <string>
#include <set>

#include "TString.h"
#include "TRegexp.h"

#include "JLang/JTypeList.hh"
#include "JLang/JType.hh"
#include "Jeep/JeepToolkit.hh"
#include "JROOT/JRootToolkit.hh"


/**
 * \author mdejong
 */

namespace JROOT {}
namespace JPP { using namespace JROOT; }

namespace JROOT {

  using JLANG::JType;
  using JEEP::getClassname;
      
  
  /**
   * Auxiliary class to select ROOT class based on class name.
   */
  struct JROOTClassSelector :
    public std::string
  {
    /**
     * Default contructor.
     */
    JROOTClassSelector() :
      std::string()
    {}


    /**
     * Contructor.
     *
     * \param  type_name   type_name
     */
    JROOTClassSelector(const char* type_name) :
      std::string(type_name)
    {}


    /**
     * Contructor.
     *
     * \param  type_name   type_name
     */
    JROOTClassSelector(const std::string& type_name) :
      std::string(type_name)
    {}


    /**
     * Contructor.
     *
     * \param  type        data type
     */
    template<class T>
    JROOTClassSelector(const JType<T>& type) : 
      std::string(getClassname(getName(type)))
    {}


    /**
     * Get status of given data type.
     *
     * \param  type        data type
     * \return             true if valid class name; else false
     */
    template<class T>
    bool operator()(const JType<T>& type) const
    {
      return *this == getClassname(T::Class_Name());
    }
  };

  
  /**
   * Auxiliary class for ROOT class selection.
   */
  struct JROOTClassSelection :
    public std::set<JROOTClassSelector>
  {
    /**
     * Default contructor.
     */
    JROOTClassSelection()
    {}


    /**
     * Copy contructor.
     *
     * \param  selection   selection
     */
    JROOTClassSelection(const std::set<JROOTClassSelector>& selection) :
      std::set<JROOTClassSelector>(selection)
    {}


    /**
     * Contructor.
     *
     * \param  typelist    type list
     */
    template<class T>
    JROOTClassSelection(const JType<T>& typelist)
    {
      JLANG::for_each(*this, typelist);
    }


    /**
     * Add given data type.
     */
    template<class T>
    void add()
    {
      this->insert(JROOTClassSelector(JType<T>()));
    }


    /**
     * Remove data type.
     */
    template<class T>
    void remove()
    {
      this->erase(JROOTClassSelector(JType<T>()));
    }


    /**
     * Add data type.
     *
     * \param  type        data type
     */
    template<class T>
    void operator()(const JType<T>& type)
    {
      add<T>();
    }


    /**
     * Get status of given data type.
     *
     * \param  type        data type
     * \return             true if valid class name; else false
     */
    template<class T>
    bool operator()(const JType<T>& type) const
    {
      return this->find(JROOTClassSelector(JType<T>())) != this->end();
    }


    /**
     * Get status of given data type.
     *
     * \return             true if valid class name; else false
     */
    template<class T>
    bool is_valid() const
    {
      return (*this)(JType<T>());
    }
 

    /**
     * Read ROOT class selection from input.
     *
     * Note that if the first character of the read class name is:
     *  - a '+', the remainder of the class name is added;   or
     *  - a '-', the remainder of the class name is removed; otherwise 
     *  - the complete class name is added.
     *
     * \param  in               input stream
     * \param  object           ROOT class selection
     * \return                  input stream
     */
    friend inline std::istream& operator>>(std::istream& in, JROOTClassSelection& object)
    {
      for (std::string buffer; in >> buffer; ) {

	const TRegexp regexp(buffer.substr(1).c_str());

	switch (buffer[0]) {

	case '-':

	  for (JROOTClassSelection::iterator i = object.begin(); i != object.end(); ) {

	    if (TString(i->c_str()).Index(regexp) != -1)
	      object.erase(i++);
	    else
	      ++i;
	  }

	  break;

	case '+':

	  object.insert(buffer.substr(1));
	  break;

	default:

	  object.insert(buffer);
	  break;
	}
      }

      return in;
    }


    /**
     * Write ROOT class selection to output.
     *
     * \param  out              output stream
     * \param  object           ROOT class selection
     * \return                  output stream
     */
    friend inline std::ostream& operator<<(std::ostream& out, const JROOTClassSelection& object)
    {
      for (const_iterator i = object.begin(); i != object.end(); ++i) {
        out << *i << std::endl;
      }

      return out;
    }
  };

  
  /**
   * Get ROOT class selection.
   *
   * \return             ROOT class selection
   */
  template<class T>
  inline std::set<JROOTClassSelector> getROOTClassSelection()
  {
    return JROOTClassSelection(JType<T>());
  }
}

#endif