#include "SuperMenuText.h"

#include "../others/Hanz2Piny.h"

#include "OpenCCConverter.h"

#include "PotConv.h"

#include "strfunc.h"

#include <algorithm>

#include <cmath>

#include <utility>

SuperMenuText::SuperMenuText(const std::string& title, int font_size, const std::vector<std::pair<int, std::string>>& allItems, int itemsPerPage) : InputBox(title, font_size), items_(allItems), itemsPerPage_(itemsPerPage)

{

previous_ = std::make_shared<Button>();

previous_->setText("上一頁PgUp");

next_ = std::make_shared<Button>();

next_->setText("下一頁PgDown");

addChild(previous_);

addChild(next_);

selections_ = std::make_shared<MenuText>();

addChild(selections_);

setAllChildState(NodeNormal);

defaultPage();

std::function<bool(const std::string&, const std::string&)> match = [&](const std::string& text, const std::string& name) -> bool

{

std::string pinyin = Hanz2Piny::hanz2pinyin(name);

int p = 0;

for (int i = 0; i < text.size(); i++)

{

int p1 = pinyin.find_first_of(text[i], p);

if (p1 < 0)

{

return false;

}

p = p1 + 1;

}

return true;

};

setMatchFunction(match);

for (const auto& pairName : items_)

{

//// 拼音

//auto u8Name = PotConv::cp936toutf8(pairName.second);

//std::string pinyin = Hanz2Piny::hanz2pinyin(u8Name);

//auto pys = strfunc::splitString(pinyin, " ");

//std::vector<std::string> acceptables;

//std::function<void(const std::string& curStr, int idx)> comboGenerator = [&](const std::string& curStr, int idx)

//{

// if (idx == pys.size())

// {

// acceptables.push_back(curStr);

// return;

// }

// comboGenerator(curStr + pys[idx][0], idx + 1);

// comboGenerator(curStr + pys[idx], idx + 1);

//};

//comboGenerator("", 0);

//for (auto& acc : acceptables)

//{

// for (int i = 0; i < acc.size(); i++)

// {

// matches_[acc.substr(0, i + 1)].insert(pairName.second);

// }

//}

//// 直接对字，可以两个跳，但是不管了

//for (int i = 0; i < pairName.second.size(); i++)

//{

// matches_[pairName.second.substr(0, i + 1)].insert(pairName.second);

//}

// 对id

std::string strID = std::to_string(pairName.first);

for (int i = 0; i < strID.size(); i++)

{

matches_[strID.substr(0, i + 1)].insert(pairName.second);

}

}

}

void SuperMenuText::setInputPosition(int x, int y)

{

InputBox::setInputPosition(x, y);

selections_->setPosition(x, y + font_size_ * 3);

previous_->setPosition(x, y - font_size_ * 1.5);

next_->setPosition(x + font_size_ * 5, y - font_size_ * 1.5);

}

void SuperMenuText::addDrawableOnCall(std::shared_ptr<DrawableOnCall> doc)

{

docs_.push_back(doc);

addChild(doc);

}

void SuperMenuText::setMatchFunction(std::function<bool(const std::string&, const std::string&)> match)

{

matchFunc_ = match;

}

void SuperMenuText::defaultPage()

{

if (curDefault_)

{

return;

}

std::vector<std::string> displays;

searchResultIndices_.clear();

for (int i = 0; i < items_.size(); i++)

{

if (i < std::min((std::size_t)itemsPerPage_, items_.size()))

{

activeIndices_.push_back(i);

displays.push_back(items_[i].second);

}

searchResultIndices_.push_back(i);

}

curPage_ = 0;

updateMaxPages();

selections_->setStrings(displays);

if (displays.size() != 0)

{

selections_->forceActiveChild(0);

}

curDefault_ = true;

}

void SuperMenuText::flipPage(int pInc)

{

if (curPage_ + pInc >= 0 && curPage_ + pInc < maxPages_)

{

// 允许你翻页！

curPage_ += pInc;

int startIdx = curPage_ * itemsPerPage_;

std::vector<std::string> displays;

activeIndices_.clear();

for (std::size_t i = startIdx; i < std::min(searchResultIndices_.size(), (std::size_t)startIdx + itemsPerPage_); i++)

{

activeIndices_.push_back(searchResultIndices_[i]);

displays.push_back(items_[searchResultIndices_[i]].second);

}

selections_->setStrings(displays);

}

// curDefault_ = false;

}

bool SuperMenuText::defaultMatch(const std::string& input, const std::string& name)

{

auto iterInput = matches_.find(input);

if (iterInput != matches_.end())

{

auto iterName = iterInput->second.find(name);

if (iterName != iterInput->second.end())

{

return true;

}

}

return false;

}

void SuperMenuText::search(const std::string& text)

{

std::vector<std::string> results;

activeIndices_.clear();

searchResultIndices_.clear();

// 只返回itemsPerPage_数量

// 更高级的Trie，LCS编辑距离等等 有缘人来搞

for (int i = 0; i < items_.size(); i++)

{

bool matched = false;

auto& opt = items_[i];

if (matchFunc_ && matchFunc_(text, opt.second))

{

matched = true;

}

else

{

matched = defaultMatch(text, opt.second);

}

if (matched)

{

if (results.size() < itemsPerPage_)

{

results.emplace_back(opt.second);

activeIndices_.push_back(i);

}

searchResultIndices_.push_back(i);

}

}

updateMaxPages();

curPage_ = 0;

selections_->setStrings(results);

selections_->forceActiveChild(0);

curDefault_ = false;

}

void SuperMenuText::updateMaxPages()

{

maxPages_ = std::ceil((searchResultIndices_.size() / (double)itemsPerPage_));

}

void SuperMenuText::dealEvent(BP_Event& e)

{

// get不到result 为何

// 不知道这玩意儿在干嘛，瞎搞即可

if (previous_->getState() == NodePress && e.type == BP_MOUSEBUTTONUP)

{

flipPage(-1);

previous_->setState(NodeNormal);

previous_->setResult(-1);

}

else if (next_->getState() == NodePress && e.type == BP_MOUSEBUTTONUP)

{

flipPage(1);

next_->setState(NodeNormal);

next_->setResult(-1);

}

bool research = false;

// 为什么switch自动缩进是这样

switch (e.type)

{

case BP_TEXTINPUT:

{

auto converted = OpenCCConverter::getInstance()->UTF8s2t(e.text.text);

converted = PotConv::conv(converted, "utf-8", "cp936");

text_ += converted;

research = true;

break;

}

case BP_TEXTEDITING:

{

break;

}

case BP_KEYDOWN:

{

if (e.key.keysym.sym == BPK_BACKSPACE)

{

if (text_.size() >= 1)

{

text_.pop_back();

if (text_.size() >= 1 && uint8_t(text_.back()) >= 128)

{

text_.pop_back();

}

}

research = true;

}

break;

}

case BP_KEYUP:

{

switch (e.key.keysym.sym)

{

case BPK_PAGEUP:

{

flipPage(-1);

break;

}

case BPK_PAGEDOWN:

{

flipPage(1);

break;

}

}

break;

}

case BP_MOUSEWHEEL:

{

if (e.wheel.y > 0)

{

flipPage(-1);

}

else if (e.wheel.y < 0)

{

flipPage(1);

}

break;

}

}

if (text_.empty())

{

defaultPage();

}

if (research)

{

search(text_);

}

int selectionIdx = selections_->getActiveChildIndex();

if (selectionIdx >= 0 && selectionIdx < activeIndices_.size())

{

int idx = activeIndices_[selectionIdx];

for (auto& doc : docs_)

{

doc->updateScreenWithID(items_[idx].first);

}

}

else

{

for (auto& doc : docs_)

{

doc->updateScreenWithID(-1);

}

}

if (selections_->getResult() >= 0)

{

auto selected = selections_->getResultString();

result_ = activeIndices_[selections_->getResult()];

if (result_ >= 0)

{

result_ = items_[result_].first;

}

setExit(true);

}

}