steiner_components.hpp

Go to the documentation of this file.

//=======================================================================
// Copyright (c)
//
// Distributed under the Boost Software License, Version 1.0. (See
// accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
//=======================================================================
#ifndef PAAL_STEINER_COMPONENTS_HPP
#define PAAL_STEINER_COMPONENTS_HPP

#include "paal/iterative_rounding/steiner_tree/steiner_component.hpp"

namespace paal {
namespace ir {

template <typename Vertex, typename Dist> class steiner_components {
  public:
    steiner_components() : m_size(0) {}

    void add(const steiner_component<Vertex, Dist> &component) {
        for (int i = 0; i < (int)component.count_terminals(); ++i) {
            ++m_size;
            m_original_id.push_back(m_components.size());
            m_version.push_back(i);
        }
        m_components.push_back(component);
    }

    int size() const { return m_size; }

    const steiner_component<Vertex, Dist> &find(int id) const {
        assert(id >= 0 && id < m_size);
        return m_components[m_original_id[id]];
    }

    int find_version(int id) const { return m_version[id]; }

    void clear() {
        m_components.clear();
        m_version.clear();
        m_original_id.clear();
        m_size = 0;
    }

  private:
    std::vector<steiner_component<Vertex, Dist>> m_components; // generated
                                                               // components
    std::vector<int> m_version;     // id -> version map
    std::vector<int> m_original_id; // id -> place in m_components vector
    int m_size; // m_version.size(), but not m_components.size()
};

} // ir
} // paal

#endif // PAAL_STEINER_COMPONENTS_HPP