mapElitesSelector.h

 
#ifndef MAP_ELITES_SELECTOR_H
#define MAP_ELITES_SELECTOR_H
 
#include "selector/mapElites/mapElitesArchive.h"
#include "selector/mapElites/mapElitesDescriptor.h"
#include "selector/mapElites/mapElitesSelectionMetrics.h"
#include "selector/selector.h"
 
#include "learn/learningEnvironment.h"
 
namespace Selector {
    namespace MapElites {
 
        class MapElitesSelector : public Selector
        {
          protected:
            std::map<std::shared_ptr<const MapElitesDescriptor>,
                     std::shared_ptr<MapElitesArchive>>
                mapEliteArchives;
 
          public:
            MapElitesSelector(std::shared_ptr<TPG::TPGGraph> graph,
                              const Learn::LearningParameters& params)
                : Selector{graph, params}
            {
            }
 
            virtual std::shared_ptr<const MapElitesArchive>
            getMapElitesArchiveAt(
                std::shared_ptr<const MapElitesDescriptor> descriptor);
 
            virtual const std::map<std::shared_ptr<const MapElitesDescriptor>,
                                   std::shared_ptr<MapElitesArchive>>&
            getMapElitesArchives();
 
            virtual std::shared_ptr<SelectionMetrics> createSelectionMetrics()
                override;
 
            virtual std::shared_ptr<const MapElitesArchive>
            addArchiveFromDescriptor(
                size_t nbBins,
                std::shared_ptr<const MapElitesDescriptor> descriptor,
                Learn::LearningEnvironment& le);
 
            virtual std::shared_ptr<const MapElitesArchive>
            addCvtArchiveFromDescriptor(
                size_t nbCentroids,
                std::shared_ptr<const MapElitesDescriptor> descriptor,
                Learn::LearningEnvironment& le, Mutator::RNG& rng,
                size_t nbIterationInit = 300, size_t nbDotsInit = 1000,
                double a1 = 0.5, double b1 = 0.5, double a2 = 0.5,
                double b2 = 0.5);
 
            virtual void doSelection(
                std::multimap<std::shared_ptr<Learn::EvaluationResult>,
                              const TPG::TPGVertex*>& results,
                Mutator::RNG& rng) override;
 
            virtual const SelectionContext& updateContext() override;
        };
    }; // namespace MapElites
 
    using MapElitesSelector = MapElites::MapElitesSelector;
}; // namespace Selector
 
#endif // MAP_ELITES_SELECTOR_H