bbb_traj_set.h

Go to the documentation of this file.

/* =========================================================================== *
   |
   |  Copyright (c) 1994-2011 by Kobus Barnard (author)
   |
   |  Personal and educational use of this code is granted, provided that this
   |  header is kept intact, and that the authorship is not misrepresented, that
   |  its use is acknowledged in publications, and relevant papers are cited.
   |
   |  For other use contact the author (kobus AT cs DOT arizona DOT edu).
   |
   |  Please note that the code in this file has not necessarily been adequately
   |  tested. Naturally, there is no guarantee of performance, support, or fitness
   |  for any particular task. Nonetheless, I am interested in hearing about
   |  problems that you encounter.
   |
   |  Author:  Ernesto Brau
 * =========================================================================== */

/* $Id$ */

#ifndef B3_TRAJ_SET_H
#define B3_TRAJ_SET_H

#include <bbb_cpp/bbb_data.h>
#include <set>
#include <boost/foreach.hpp>

namespace kjb {
namespace bbb {

class Traj_set
{
private:
    typedef std::set<size_t> Iset;

public:
    typedef Iset::const_iterator const_iterator;
    typedef Iset::const_reverse_iterator const_reverse_iterator;
    typedef Iset::value_type value_type;

public:
    Traj_set() {}

    template<class TrIt>
    Traj_set(TrIt first, TrIt last) : trajectories_(first, last) {}

    template<class SizeType>
    void insert(const SizeType& j) { trajectories_.insert(j); }

    template<class TrIt>
    void insert(TrIt first, TrIt last) { trajectories_.insert(first, last); }

    void clear() { trajectories_.clear(); }

    const_iterator begin() const { return trajectories_.begin(); }

    const_iterator end() const { return trajectories_.end(); }

    const_reverse_iterator rbegin() const { return trajectories_.rbegin(); }

    const_reverse_iterator rend() const { return trajectories_.rend(); }

    size_t size() const { return trajectories_.size(); }

    template<class OutIt>
    void trajectories(const Data& data, OutIt output) const;

private:
    Iset trajectories_;
};

/* \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ \/ */

template<class OutIt>
void Traj_set::trajectories(const Data& data, OutIt output) const
{
    size_t N = data.size();
    BOOST_FOREACH(size_t j, trajectories_)
    {
        IFT(j < N, Runtime_error,
            "Cannot get trajectories; invalid association.");

        *output++ = &data.trajectory(j);
    }
}

}} // namespace kjb::bbb

#endif /*B3_TRAJ_SET_H */