location_set.h

// Description: A set with a locator; a point-set.
// Documentation: pointset.txt

#ifndef PASTELSYS_LOCATION_SET_H
#define PASTELSYS_LOCATION_SET_H

#include "pastel/sys/range/range_concept.h"
#include "pastel/sys/range/transformed_range.h"
#include "pastel/sys/locator/locator_concept.h"
#include "pastel/sys/locator/location.h"

#include <range/v3/all.hpp>

namespace Pastel
{

    //! A set with a locator; a point-set.
    template <
        Range_Concept Set,
        Locator_Concept Locator,
        typename Base>
    class LocationSet
    : public Base
    {
    public:
        using Index = Range_Iterator<std::add_const_t<Set>>;

        LocationSet(
            const Set& set,
            const Locator& locator,
            const Base& base)
            : Base(base)
            , set_(set)
            , locator_(locator)
        {
        }

        const Set& pointSetRange() const
        {
            return set_;
        }

        const Locator& pointSetLocator() const
        {
            return locator_;
        }

    private:
        const Set& set_;
        Locator locator_;
    };

    template <typename Type>
    struct IsLocationSet
        : std::false_type
    {};

    template <
        typename Set, 
        typename Locator,
        typename Base>
    struct IsLocationSet<LocationSet<Set, Locator, Base>>
        : std::true_type
    {};

    //! Constructs a location-set from a point-set and a locator.
    template <
        Range_Concept Set, 
        Locator_Concept Locator
    >
    decltype(auto) locationSet(
        const Set& set,
        const Locator& locator)
    {
        auto transformedSet_ = transformRange(
            set,
            [=](auto&& element){return location(element, locator);});

        using Base = decltype(transformedSet_);

        return LocationSet<Set, Locator, Base>(set, locator, transformedSet_);
    }

}

#endif