contextTracker.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_CONTEXTTRACKER
#define PRESAGE_CONTEXTTRACKER

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

#include <iostream>
#include <sstream>
#include <string>
#include <vector>
#include <assert.h>

#include "contextChangeDetector.h"

#include "../tokenizer/reverseTokenizer.h"
#include "../charsets.h"
#include "../configuration.h"
#include "../logger.h"
#include "../dispatcher.h"

#include "../../presageCallback.h"

class PredictorRegistry;

class ContextTracker : public Observer {
public:
    ContextTracker(Configuration* config,
                   PredictorRegistry* predictorRegistry,
                   PresageCallback* callback,
                   const char[]=DEFAULT_WORD_CHARS,
                   const char[]=DEFAULT_SEPARATOR_CHARS,
                   const char[]=DEFAULT_BLANKSPACE_CHARS,
                   const char[]=DEFAULT_CONTROL_CHARS );
    ~ContextTracker();

    const PresageCallback* callback(const PresageCallback* callback);

    bool contextChange();

    std::string getPrefix() const;
    std::string getToken (const int) const;

    std::string getExtraTokenToLearn(const int index,
                                     const std::vector<std::string>& change) const;

    std::string getFutureStream() const;
    std::string getPastStream  () const;

    bool isCompletionValid(const std::string&) const;

    //eliminate after testing phase
    std::string getWordChars()       const;
    std::string getSeparatorChars()  const;
    std::string getBlankspaceChars() const;
    std::string getControlChars()    const;

    std::string toString() const;

    void update();

    void learn(const std::string& text) const;

    virtual void update (const Observable* variable);

    void set_logger (const std::string& value);
    void set_sliding_window_size (const std::string& value);
    void set_lowercase_mode (const std::string& value);
    void set_online_learning (const std::string& value);

    static const char* LOGGER;
    static const char* SLIDING_WINDOW_SIZE;
    static const char* LOWERCASE_MODE;
    static const char* ONLINE_LEARNING;

private:
    std::string wordChars;
    std::string separatorChars;
    std::string blankspaceChars;
    std::string controlChars;

    bool lowercase_mode;
    bool online_learning;

    // REVISIT: this was removed since a tokenizer is created with
    // each method invocation that needs it
    //ReverseTokenizer tokenizer;

    // utility functions
    bool isWordChar      (const char) const;
    bool isSeparatorChar (const char) const;
    bool isControlChar   (const char) const;
    bool isBlankspaceChar(const char) const;

    const PresageCallback* context_tracker_callback;
    PredictorRegistry* predictorRegistry;
    ContextChangeDetector* contextChangeDetector;
    Logger<char> logger;
    
    Dispatcher<ContextTracker> dispatcher;
};

#endif // PRESAGE_CONTEXTTRACKER