StiHitFilters.h

Go to the documentation of this file.

#ifndef StiHitFilters_h
#define StiHitFilters_h

#include <boost/regex.hpp>

#include "StEvent/StEnumerations.h"
#include "Sti/Base/Filter.h"
#include "Sti/StiHit.h"



class StiHitFilterByDetectorGroup : public Filter<StiHit>
{
public:

   StiHitFilterByDetectorGroup(StDetectorId detGroupId) : Filter(), fDetGroupId(detGroupId) {}

   virtual bool accept(const StiHit *hit) const {
      return (hit && hit->detector() && hit->detector()->getGroupId() == fDetGroupId);
   }

private:

   StDetectorId  fDetGroupId;   
};



class StiHitFilterByVolumeName : public Filter<StiHit>
{
public:

   StiHitFilterByVolumeName(const std::set<boost::regex>& patterns) : Filter(),
      fVolNamePattern(patterns) {}

   virtual bool accept(const StiHit *hit) const
   {
      for (const auto& volNamePattern : fVolNamePattern)
      {
         if ( hit && hit->detector() && boost::regex_match(std::string(hit->detector()->getName()), volNamePattern) )
            return true;
      }
   }

private:

   const std::set<boost::regex>&  fVolNamePattern;   
};

#endif