spell_filter.hpp

Go to the documentation of this file.

#pragma once
 
#include "spell_base.hpp"
#include "unicode.hpp"
#include <vector>
#include "rune_sequence.hpp"
 
namespace rune_caster {
 
namespace spell {
namespace filter {

class CategoryFilter : public sequence_spell {
private:
    std::vector<unicode::Category> filtered_categories_;
    bool remove_mode_; // true = remove, false = keep only
 
public:
    using input_type = RuneSequence;
    using output_type = RuneSequence;

    explicit CategoryFilter(std::vector<unicode::Category> categories, bool remove = true);
 
    RuneSequence operator()(const RuneSequence& input) const override;
    std::string description() const override;
    std::string name() const override { return "CategoryFilter"; }

    const std::vector<unicode::Category>& filtered_categories() const noexcept {
        return filtered_categories_;
    }

    bool is_remove_mode() const noexcept { return remove_mode_; }
};

class ScriptFilter : public sequence_spell {
private:
    std::vector<unicode::Script> filtered_scripts_;
    bool remove_mode_; // true = remove, false = keep only
 
public:
    using input_type = RuneSequence;
    using output_type = RuneSequence;

    explicit ScriptFilter(std::vector<unicode::Script> scripts, bool remove = true);
 
    RuneSequence operator()(const RuneSequence& input) const override;
    std::string description() const override;
    std::string name() const override { return "ScriptFilter"; }

    const std::vector<unicode::Script>& filtered_scripts() const noexcept {
        return filtered_scripts_;
    }

    bool is_remove_mode() const noexcept { return remove_mode_; }
};

class PunctuationFilter : public sequence_spell {
private:
    bool remove_mode_; // true = remove punctuation, false = keep only punctuation
 
public:
    using input_type = RuneSequence;
    using output_type = RuneSequence;
 
    explicit PunctuationFilter(bool remove = true) : remove_mode_(remove) {}
 
    RuneSequence operator()(const RuneSequence& input) const override;
 
    std::string description() const override {
        return remove_mode_ ? "Remove punctuation" : "Keep only punctuation";
    }
 
    std::string name() const override { return "PunctuationFilter"; }
 
    bool is_remove_mode() const noexcept { return remove_mode_; }
};
 
} // namespace filter
 
// Import commonly used filters
using filter::CategoryFilter;
using filter::ScriptFilter;
using filter::PunctuationFilter;
 
} // namespace spell
} // namespace rune_caster