selector.h

Go to the documentation of this file.

/******************************************************
 *  Presage, an extensible predictive text entry system
 *  ---------------------------------------------------
 *
 *  Copyright (C) 2008  Matteo Vescovi <matteo.vescovi@yahoo.co.uk>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License along
    with this program; if not, write to the Free Software Foundation, Inc.,
    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
                                                                             *
                                                                **********(*)*/


#ifndef PRESAGE_SELECTOR
#define PRESAGE_SELECTOR

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "suggestion.h"
#include "prediction.h"
#include "context_tracker/contextTracker.h"
#include "logger.h"
#include "dispatcher.h"

#include <string>
#include <vector>
#include <set>
#include <stdlib.h> // needed for abort()


typedef std::set< std::string, std::less<std::string> > StringSet;


class Selector : public Observer {
public:
    Selector(Configuration*, ContextTracker*);
    ~Selector();

    std::vector<std::string> select(Prediction);

    void update();

    void set_logger(const std::string& value);
    void set_suggestions(const std::string& value);
    void set_repeat_suggestions(const std::string& value);
    void set_greedy_suggestion_threshold(const std::string& value);

    size_t get_suggestions () const;
    bool get_repeat_suggestions () const;
    size_t get_greedy_suggestion_threshold () const;

    static const char* LOGGER;
    static const char* SUGGESTIONS;
    static const char* REPEAT_SUGGESTIONS;
    static const char* GREEDY_SUGGESTION_THRESHOLD;

    virtual void update (const Observable* variable);

private:
    // handle observable notifications
    typedef void (Selector::* mbr_func_ptr_t)(const std::string&);
    std::map<std::string, mbr_func_ptr_t> dispatch_map;

    size_t suggestions;
    bool repeat_suggestions;
    size_t greedy_suggestion_threshold;

    void updateSuggestedWords( const std::vector<std::string>& );
    void clearSuggestedWords();

    void repetitionFilter( std::vector<std::string>& );
    void thresholdFilter( std::vector<std::string>& );

    StringSet suggestedWords;

    std::string previous_prefix;

    ContextTracker* contextTracker;
    Configuration*  config;
    Logger<char> logger;

    Dispatcher<Selector> dispatcher;
};


#endif // PRESAGE_SELECTOR