facility_location_solution.hpp

Go to the documentation of this file.

//=======================================================================
// Copyright (c) 2013 Piotr Wygocki
//
// 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_FACILITY_LOCATION_SOLUTION_HPP
#define PAAL_FACILITY_LOCATION_SOLUTION_HPP

#define BOOST_RESULT_OF_USE_DECLTYPE

#include "facility_location_solution_traits.hpp"

#include "paal/data_structures/voronoi/voronoi.hpp"

#include <unordered_set>
#include <cassert>
#include <type_traits>

namespace paal {
namespace data_structures {

template <typename FacilityCost, typename VoronoiType>
class facility_location_solution {
public:
    typedef voronoi_traits<VoronoiType> VT;
    typedef typename VT::VertexType VertexType;
    typedef typename VT::DistanceType Dist;
    typedef typename VT::GeneratorsSet ChosenFacilitiesSet;
    typedef std::unordered_set<VertexType, boost::hash<VertexType>>
        UnchosenFacilitiesSet;

private:

    VoronoiType m_voronoi;
    UnchosenFacilitiesSet m_unchosen_facilities;
    const FacilityCost &m_fac_costs;
public:

    facility_location_solution(VoronoiType voronoi, UnchosenFacilitiesSet uf,
                               const FacilityCost &c)
        : m_voronoi(std::move(voronoi)), m_unchosen_facilities(std::move(uf)),
          m_fac_costs(c) {}

    Dist add_facility(VertexType f) {
        assert(m_unchosen_facilities.find(f) != m_unchosen_facilities.end());
        m_unchosen_facilities.erase(f);

        return m_fac_costs(f) + m_voronoi.add_generator(f);
    }

    Dist rem_facility(VertexType f) {
        assert(m_unchosen_facilities.find(f) == m_unchosen_facilities.end());
        m_unchosen_facilities.insert(f);

        return -m_fac_costs(f) + m_voronoi.rem_generator(f);
    }

    const UnchosenFacilitiesSet &get_unchosen_facilities() const {
        return m_unchosen_facilities;
    }

    const ChosenFacilitiesSet &get_chosen_facilities() const {
        return m_voronoi.get_generators();
    }

    auto get_clients_for_facility(VertexType f) const ->
    decltype(m_voronoi.get_vertices_for_generator(f))
    {
        return m_voronoi.get_vertices_for_generator(f);
    }

    const VoronoiType &get_voronoi() const { return m_voronoi; }

};

template <typename FacilityCost, typename Voronoi>
class facility_location_solution_traits<
    facility_location_solution<FacilityCost, Voronoi>> {
    typedef voronoi_traits<Voronoi> VT;
    typedef facility_location_solution<FacilityCost, Voronoi> FLS;

  public:
    typedef typename VT::VertexType VertexType;
    typedef typename VT::DistanceType Dist;
    typedef typename VT::GeneratorsSet ChosenFacilitiesSet;
    typedef puretype(std::declval<FLS>().get_unchosen_facilities())
        UnchosenFacilitiesSet;
};

} // !data_structures
} // !paal

#endif // PAAL_FACILITY_LOCATION_SOLUTION_HPP