diff --git a/lib_vector_search/include/word_list.h b/lib_vector_search/include/word_list.h index b2365e2..b5a9300 100644 --- a/lib_vector_search/include/word_list.h +++ b/lib_vector_search/include/word_list.h @@ -11,6 +11,7 @@ public: WordList &multiply(size_t factor); WordList &shuffle(); + static WordList oneCap(); static WordList fourCaps(); static WordList fromFile(const std::filesystem::path &path); diff --git a/lib_vector_search/src/word_list.cpp b/lib_vector_search/src/word_list.cpp index e810451..6934beb 100644 --- a/lib_vector_search/src/word_list.cpp +++ b/lib_vector_search/src/word_list.cpp @@ -25,6 +25,19 @@ WordList &WordList::shuffle() { return *this; } +WordList WordList::oneCap() { + const static std::string charset_ = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"; + + WordList word_list; + word_list.reserve(charset_.length()); + + for (auto char_1 : charset_) { + word_list.emplace_back(std::initializer_list({char_1})); + } + + return word_list; +}; + WordList WordList::fourCaps() { const static std::string charset_ = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";