SusyNtuple/vec_utils.h

// Dear emacs, this is -*- c++ -*-
#ifndef SUSY_VEC_UTILS_H
#define SUSY_VEC_UTILS_H

/*
  Vector utility functions
  
  Most of these are useful to manipulate the mc truth table.

  davide.gerbaudo@gmail.com
  April 2014
*/

#include <algorithm>
#include <string>
#include <vector>
#include <algorithm> // find

namespace Susy{
namespace utils{

typedef std::vector< int > vint_t;
typedef std::vector< vint_t > vvint_t;


template<typename T, size_t N>
T * end(T (&ra)[N]) {
    return ra + N;
}
template < class T >
bool contains(const std::vector<T> &v, const T& val) {
    return std::find(v.begin(), v.end(), val)!=v.end();
}
bool containsAnyOf(const vint_t &firstVec, const vint_t &subVec);
std::string vecToString(const vint_t &vec);
std::string vecVecToString(const vvint_t &vvec);
void removeDuplicates(vint_t &vec);

template < template <typename, typename> class Container,
           typename Predicate,
           typename Allocator,
           typename A
           >
Container<A, Allocator> filter(Container<A, Allocator> const & container, Predicate const & pred) {
  Container<A, Allocator> filtered(container);
  filtered.erase(remove_if(filtered.begin(), filtered.end(), pred), filtered.end());
  return filtered;
}


} // utils
} // susy
#endif