| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982 |
- #pragma once
- #include <algorithm>
- #include <array>
- #include <cctype>
- #include <cstring>
- #include <filesystem>
- #include <memory>
- #include <set>
- #include <string>
- #include <string_view>
- #include <vector>
- #ifndef IMGUI_VERSION
- # error "include imgui.h before this header"
- #endif
- using ImGuiFileBrowserFlags = int;
- enum ImGuiFileBrowserFlags_
- {
- ImGuiFileBrowserFlags_SelectDirectory = 1 << 0, // select directory instead of regular file
- ImGuiFileBrowserFlags_EnterNewFilename = 1 << 1, // allow user to enter new filename when selecting regular file
- ImGuiFileBrowserFlags_NoModal = 1 << 2, // file browsing window is modal by default. specify this to use a popup window
- ImGuiFileBrowserFlags_NoTitleBar = 1 << 3, // hide window title bar
- ImGuiFileBrowserFlags_NoStatusBar = 1 << 4, // hide status bar at the bottom of browsing window
- ImGuiFileBrowserFlags_CloseOnEsc = 1 << 5, // close file browser when pressing 'ESC'
- ImGuiFileBrowserFlags_CreateNewDir = 1 << 6, // allow user to create new directory
- ImGuiFileBrowserFlags_MultipleSelection = 1 << 7, // allow user to select multiple files. this will hide ImGuiFileBrowserFlags_EnterNewFilename
- };
- namespace ImGui
- {
- class FileBrowser
- {
- public:
- // pwd is set to current working directory by default
- explicit FileBrowser(ImGuiFileBrowserFlags flags = 0);
- FileBrowser(const FileBrowser ©From);
- FileBrowser &operator=(const FileBrowser ©From);
- // set the window position (in pixels)
- // default is centered
- void SetWindowPos(int posx, int posy) noexcept;
- // set the window size (in pixels)
- // default is (700, 450)
- void SetWindowSize(int width, int height) noexcept;
- // set the window title text
- void SetTitle(std::string title);
- // open the browsing window
- void Open();
- // close the browsing window
- void Close();
- // the browsing window is opened or not
- bool IsOpened() const noexcept;
- // display the browsing window if opened
- void Display();
- // returns true when there is a selected filename and the "ok" button was clicked
- bool HasSelected() const noexcept;
- // set current browsing directory
- bool SetPwd(const std::filesystem::path &pwd =
- std::filesystem::current_path());
- // get current browsing directory
- const std::filesystem::path &GetPwd() const noexcept;
- // returns selected filename. make sense only when HasSelected returns true
- // when ImGuiFileBrowserFlags_MultipleSelection is enabled, only one of
- // selected filename will be returned
- std::filesystem::path GetSelected() const;
- // returns all selected filenames.
- // when ImGuiFileBrowserFlags_MultipleSelection is enabled, use this
- // instead of GetSelected
- std::vector<std::filesystem::path> GetMultiSelected() const;
- // set selected filename to empty
- void ClearSelected();
- // (optional) set file type filters. eg. { ".h", ".cpp", ".hpp" }
- // ".*" matches any file types
- void SetTypeFilters(const std::vector<std::string> &typeFilters);
- // set currently applied type filter
- // default value is 0 (the first type filter)
- void SetCurrentTypeFilterIndex(int index);
- // when ImGuiFileBrowserFlags_EnterNewFilename is set
- // this function will pre-fill the input dialog with a filename.
- void SetInputName(std::string_view input);
- private:
- template <class Functor>
- struct ScopeGuard
- {
- ScopeGuard(Functor&& t) : func(std::move(t)) { }
- ~ScopeGuard() { func(); }
- private:
- Functor func;
- };
- struct FileRecord
- {
- bool isDir = false;
- std::filesystem::path name;
- std::string showName;
- std::filesystem::path extension;
- };
- static std::string ToLower(const std::string &s);
- void UpdateFileRecords();
- void SetPwdUncatched(const std::filesystem::path &pwd);
- bool IsExtensionMatched(const std::filesystem::path &extension) const;
- #ifdef _WIN32
- static std::uint32_t GetDrivesBitMask();
- #endif
- // for c++17 compatibility
- #if defined(__cpp_lib_char8_t)
- static std::string u8StrToStr(std::u8string s);
- #endif
- static std::string u8StrToStr(std::string s);
- int width_;
- int height_;
- int posX_;
- int posY_;
- ImGuiFileBrowserFlags flags_;
- std::string title_;
- std::string openLabel_;
- bool openFlag_;
- bool closeFlag_;
- bool isOpened_;
- bool ok_;
- bool posIsSet_;
- std::string statusStr_;
- std::vector<std::string> typeFilters_;
- unsigned int typeFilterIndex_;
- bool hasAllFilter_;
- std::filesystem::path pwd_;
- std::set<std::filesystem::path> selectedFilenames_;
- std::vector<FileRecord> fileRecords_;
- // IMPROVE: truncate when selectedFilename_.length() > inputNameBuf_.size() - 1
- static constexpr size_t INPUT_NAME_BUF_SIZE = 512;
- std::unique_ptr<std::array<char, INPUT_NAME_BUF_SIZE>> inputNameBuf_;
- std::string openNewDirLabel_;
- std::unique_ptr<std::array<char, INPUT_NAME_BUF_SIZE>> newDirNameBuf_;
- #ifdef _WIN32
- uint32_t drives_;
- #endif
- };
- } // namespace ImGui
- inline ImGui::FileBrowser::FileBrowser(ImGuiFileBrowserFlags flags)
- : width_(700), height_(450), posX_(0), posY_(0), flags_(flags),
- openFlag_(false), closeFlag_(false), isOpened_(false), ok_(false), posIsSet_(false),
- inputNameBuf_(std::make_unique<std::array<char, INPUT_NAME_BUF_SIZE>>())
- {
- if(flags_ & ImGuiFileBrowserFlags_CreateNewDir)
- {
- newDirNameBuf_ =
- std::make_unique<std::array<char, INPUT_NAME_BUF_SIZE>>();
- }
- inputNameBuf_->front() = '\0';
- inputNameBuf_->back() = '\0';
- SetTitle("file browser");
- SetPwd(std::filesystem::current_path());
- typeFilters_.clear();
- typeFilterIndex_ = 0;
- hasAllFilter_ = false;
- #ifdef _WIN32
- drives_ = GetDrivesBitMask();
- #endif
- }
- inline ImGui::FileBrowser::FileBrowser(const FileBrowser ©From)
- : FileBrowser()
- {
- *this = copyFrom;
- }
- inline ImGui::FileBrowser &ImGui::FileBrowser::operator=(
- const FileBrowser ©From)
- {
- width_ = copyFrom.width_;
- height_ = copyFrom.height_;
- posX_ = copyFrom.posX_;
- posY_ = copyFrom.posY_;
- flags_ = copyFrom.flags_;
- SetTitle(copyFrom.title_);
- openFlag_ = copyFrom.openFlag_;
- closeFlag_ = copyFrom.closeFlag_;
- isOpened_ = copyFrom.isOpened_;
- ok_ = copyFrom.ok_;
- posIsSet_ = copyFrom.posIsSet_;
- statusStr_ = "";
- typeFilters_ = copyFrom.typeFilters_;
- typeFilterIndex_ = copyFrom.typeFilterIndex_;
- hasAllFilter_ = copyFrom.hasAllFilter_;
- pwd_ = copyFrom.pwd_;
- selectedFilenames_ = copyFrom.selectedFilenames_;
- fileRecords_ = copyFrom.fileRecords_;
- *inputNameBuf_ = *copyFrom.inputNameBuf_;
- openNewDirLabel_ = copyFrom.openNewDirLabel_;
- if(flags_ & ImGuiFileBrowserFlags_CreateNewDir)
- {
- newDirNameBuf_ = std::make_unique<
- std::array<char, INPUT_NAME_BUF_SIZE>>();
- *newDirNameBuf_ = *copyFrom.newDirNameBuf_;
- }
- #ifdef _WIN32
- drives_ = copyFrom.drives_;
- #endif
- return *this;
- }
- inline void ImGui::FileBrowser::SetWindowPos(int posx, int posy) noexcept
- {
- posX_ = posx;
- posY_ = posy;
- posIsSet_ = true;
- }
- inline void ImGui::FileBrowser::SetWindowSize(int width, int height) noexcept
- {
- assert(width > 0 && height > 0);
- width_ = width;
- height_ = height;
- }
- inline void ImGui::FileBrowser::SetTitle(std::string title)
- {
- title_ = std::move(title);
- openLabel_ = title_ + "##filebrowser_" +
- std::to_string(reinterpret_cast<size_t>(this));
- openNewDirLabel_ = "new dir##new_dir_" +
- std::to_string(reinterpret_cast<size_t>(this));
- }
- inline void ImGui::FileBrowser::Open()
- {
- ClearSelected();
- UpdateFileRecords();
- statusStr_ = std::string();
- openFlag_ = true;
- closeFlag_ = false;
- }
- inline void ImGui::FileBrowser::Close()
- {
- ClearSelected();
- statusStr_ = std::string();
- closeFlag_ = true;
- openFlag_ = false;
- }
- inline bool ImGui::FileBrowser::IsOpened() const noexcept
- {
- return isOpened_;
- }
- inline void ImGui::FileBrowser::Display()
- {
- PushID(this);
- ScopeGuard exitThis([this]
- {
- openFlag_ = false;
- closeFlag_ = false;
- PopID();
- });
- if(openFlag_)
- {
- OpenPopup(openLabel_.c_str());
- }
- isOpened_ = false;
- // open the popup window
- if(openFlag_ && (flags_ & ImGuiFileBrowserFlags_NoModal))
- {
- if (posIsSet_)
- SetNextWindowPos(
- ImVec2(static_cast<float>(posX_), static_cast<float>(posY_)));
- SetNextWindowSize(
- ImVec2(static_cast<float>(width_), static_cast<float>(height_)));
- }
- else
- {
- if (posIsSet_)
- SetNextWindowPos(
- ImVec2(static_cast<float>(posX_), static_cast<float>(posY_)),
- ImGuiCond_FirstUseEver);
- SetNextWindowSize(
- ImVec2(static_cast<float>(width_), static_cast<float>(height_)),
- ImGuiCond_FirstUseEver);
- }
- if(flags_ & ImGuiFileBrowserFlags_NoModal)
- {
- if(!BeginPopup(openLabel_.c_str()))
- {
- return;
- }
- }
- else if(!BeginPopupModal(openLabel_.c_str(), nullptr,
- flags_ & ImGuiFileBrowserFlags_NoTitleBar ?
- ImGuiWindowFlags_NoTitleBar : 0))
- {
- return;
- }
- isOpened_ = true;
- ScopeGuard endPopup([] { EndPopup(); });
- // display elements in pwd
- #ifdef _WIN32
- char currentDrive = static_cast<char>(pwd_.c_str()[0]);
- char driveStr[] = { currentDrive, ':', '\0' };
- PushItemWidth(4 * GetFontSize());
- if(BeginCombo("##select_drive", driveStr))
- {
- ScopeGuard guard([&] { EndCombo(); });
- for(int i = 0; i < 26; ++i)
- {
- if(!(drives_ & (1 << i)))
- {
- continue;
- }
- char driveCh = static_cast<char>('A' + i);
- char selectableStr[] = { driveCh, ':', '\0' };
- bool selected = currentDrive == driveCh;
- if(Selectable(selectableStr, selected) && !selected)
- {
- char newPwd[] = { driveCh, ':', '\\', '\0' };
- SetPwd(newPwd);
- }
- }
- }
- PopItemWidth();
- SameLine();
- #endif
- int secIdx = 0, newPwdLastSecIdx = -1;
- for(const auto &sec : pwd_)
- {
- #ifdef _WIN32
- if(secIdx == 1)
- {
- ++secIdx;
- continue;
- }
- #endif
- PushID(secIdx);
- if(secIdx > 0)
- {
- SameLine();
- }
- if(SmallButton(u8StrToStr(sec.u8string()).c_str()))
- {
- newPwdLastSecIdx = secIdx;
- }
- PopID();
- ++secIdx;
- }
- if(newPwdLastSecIdx >= 0)
- {
- int i = 0;
- std::filesystem::path newPwd;
- for(const auto &sec : pwd_)
- {
- if(i++ > newPwdLastSecIdx)
- {
- break;
- }
- newPwd /= sec;
- }
- #ifdef _WIN32
- if(newPwdLastSecIdx == 0)
- {
- newPwd /= "\\";
- }
- #endif
- SetPwd(newPwd);
- }
- SameLine();
- if(SmallButton("*"))
- {
- UpdateFileRecords();
- std::set<std::filesystem::path> newSelectedFilenames;
- for(auto &name : selectedFilenames_)
- {
- auto it = std::find_if(
- fileRecords_.begin(), fileRecords_.end(),
- [&](const FileRecord &record)
- {
- return name == record.name;
- });
- if(it != fileRecords_.end())
- {
- newSelectedFilenames.insert(name);
- }
- }
- if(inputNameBuf_ && (*inputNameBuf_)[0])
- {
- newSelectedFilenames.insert(inputNameBuf_->data());
- }
- }
- if(newDirNameBuf_)
- {
- SameLine();
- if(SmallButton("+"))
- {
- OpenPopup(openNewDirLabel_.c_str());
- (*newDirNameBuf_)[0] = '\0';
- }
- if(BeginPopup(openNewDirLabel_.c_str()))
- {
- ScopeGuard endNewDirPopup([] { EndPopup(); });
- InputText("name", newDirNameBuf_->data(), newDirNameBuf_->size());
- SameLine();
- if(Button("ok") && (*newDirNameBuf_)[0] != '\0')
- {
- ScopeGuard closeNewDirPopup([] { CloseCurrentPopup(); });
- if(create_directory(pwd_ / newDirNameBuf_->data()))
- {
- UpdateFileRecords();
- }
- else
- {
- statusStr_ = "failed to create " +
- std::string(newDirNameBuf_->data());
- }
- }
- }
- }
- // browse files in a child window
- float reserveHeight = GetFrameHeightWithSpacing();
- std::filesystem::path newPwd; bool setNewPwd = false;
- if(!(flags_ & ImGuiFileBrowserFlags_SelectDirectory) &&
- (flags_ & ImGuiFileBrowserFlags_EnterNewFilename))
- reserveHeight += GetFrameHeightWithSpacing();
- {
- BeginChild("ch", ImVec2(0, -reserveHeight), true,
- (flags_ & ImGuiFileBrowserFlags_NoModal) ?
- ImGuiWindowFlags_AlwaysHorizontalScrollbar : 0);
- ScopeGuard endChild([] { EndChild(); });
- for(auto &rsc : fileRecords_)
- {
- if(!rsc.isDir && !IsExtensionMatched(rsc.extension))
- {
- continue;
- }
- if(!rsc.name.empty() && rsc.name.c_str()[0] == '$')
- {
- continue;
- }
- bool selected = selectedFilenames_.find(rsc.name)
- != selectedFilenames_.end();
- if(Selectable(rsc.showName.c_str(), selected,
- ImGuiSelectableFlags_DontClosePopups))
- {
- const bool multiSelect =
- (flags_ & ImGuiFileBrowserFlags_MultipleSelection) &&
- IsWindowFocused(ImGuiFocusedFlags_RootAndChildWindows) &&
- (GetIO().KeyCtrl || GetIO().KeyShift);
- if(selected)
- {
- if(!multiSelect)
- {
- selectedFilenames_.clear();
- }
- else
- {
- selectedFilenames_.erase(rsc.name);
- }
- (*inputNameBuf_)[0] = '\0';
- }
- else if(rsc.name != "..")
- {
- if((rsc.isDir && (flags_ & ImGuiFileBrowserFlags_SelectDirectory)) ||
- (!rsc.isDir && !(flags_ & ImGuiFileBrowserFlags_SelectDirectory)))
- {
- if(multiSelect)
- {
- selectedFilenames_.insert(rsc.name);
- }
- else
- {
- selectedFilenames_ = { rsc.name };
- }
- if(!(flags_ & ImGuiFileBrowserFlags_SelectDirectory))
- {
- #ifdef _MSC_VER
- strcpy_s(
- inputNameBuf_->data(), inputNameBuf_->size(),
- u8StrToStr(rsc.name.u8string()).c_str());
- #else
- std::strncpy(inputNameBuf_->data(),
- u8StrToStr(rsc.name.u8string()).c_str(),
- inputNameBuf_->size() - 1);
- #endif
- }
- }
- }
- else
- {
- if(!multiSelect)
- {
- selectedFilenames_.clear();
- }
- }
- }
- if(IsItemClicked(0) && IsMouseDoubleClicked(0))
- {
- if(rsc.isDir)
- {
- setNewPwd = true;
- newPwd = (rsc.name != "..") ? (pwd_ / rsc.name) :
- pwd_.parent_path();
- }
- else if(!(flags_ & ImGuiFileBrowserFlags_SelectDirectory))
- {
- selectedFilenames_ = { rsc.name };
- ok_ = true;
- CloseCurrentPopup();
- }
- }
- }
- }
- if(setNewPwd)
- {
- SetPwd(newPwd);
- }
- if(!(flags_ & ImGuiFileBrowserFlags_SelectDirectory) &&
- (flags_ & ImGuiFileBrowserFlags_EnterNewFilename))
- {
- PushID(this);
- ScopeGuard popTextID([] { PopID(); });
- PushItemWidth(-1);
- if(InputText("", inputNameBuf_->data(), inputNameBuf_->size()) &&
- inputNameBuf_->at(0) != '\0')
- {
- selectedFilenames_ = { inputNameBuf_->data() };
- }
- PopItemWidth();
- }
- if(!(flags_ & ImGuiFileBrowserFlags_SelectDirectory))
- {
- if(Button(" ok ") && !selectedFilenames_.empty())
- {
- ok_ = true;
- CloseCurrentPopup();
- }
- }
- else
- {
- if(Button(" ok "))
- {
- ok_ = true;
- CloseCurrentPopup();
- }
- }
- SameLine();
- bool shouldExit =
- Button("cancel") || closeFlag_ ||
- ((flags_ & ImGuiFileBrowserFlags_CloseOnEsc) &&
- IsWindowFocused(ImGuiFocusedFlags_RootAndChildWindows) &&
- IsKeyPressed(ImGuiKey_Escape));
- if(shouldExit)
- {
- CloseCurrentPopup();
- }
- if(!statusStr_.empty() && !(flags_ & ImGuiFileBrowserFlags_NoStatusBar))
- {
- SameLine();
- Text("%s", statusStr_.c_str());
- }
- if(!typeFilters_.empty())
- {
- SameLine();
- PushItemWidth(8 * GetFontSize());
- if(BeginCombo(
- "##type_filters", typeFilters_[typeFilterIndex_].c_str()))
- {
- ScopeGuard guard([&] { EndCombo(); });
- for(size_t i = 0; i < typeFilters_.size(); ++i)
- {
- bool selected = i == typeFilterIndex_;
- if(Selectable(typeFilters_[i].c_str(), selected) && !selected)
- {
- typeFilterIndex_ = static_cast<unsigned int>(i);
- }
- }
- }
- PopItemWidth();
- }
- }
- inline bool ImGui::FileBrowser::HasSelected() const noexcept
- {
- return ok_;
- }
- inline bool ImGui::FileBrowser::SetPwd(const std::filesystem::path &pwd)
- {
- try
- {
- SetPwdUncatched(pwd);
- return true;
- }
- catch(const std::exception &err)
- {
- statusStr_ = std::string("last error: ") + err.what();
- }
- catch(...)
- {
- statusStr_ = "last error: unknown";
- }
- SetPwdUncatched(std::filesystem::current_path());
- return false;
- }
- inline const class std::filesystem::path &ImGui::FileBrowser::GetPwd() const noexcept
- {
- return pwd_;
- }
- inline std::filesystem::path ImGui::FileBrowser::GetSelected() const
- {
- // when ok_ is true, selectedFilenames_ may be empty if SelectDirectory
- // is enabled. return pwd in that case.
- if(selectedFilenames_.empty())
- {
- return pwd_;
- }
- return pwd_ / *selectedFilenames_.begin();
- }
- inline std::vector<std::filesystem::path>
- ImGui::FileBrowser::GetMultiSelected() const
- {
- if(selectedFilenames_.empty())
- {
- return { pwd_ };
- }
- std::vector<std::filesystem::path> ret;
- ret.reserve(selectedFilenames_.size());
- for(auto &s : selectedFilenames_)
- {
- ret.push_back(pwd_ / s);
- }
- return ret;
- }
- inline void ImGui::FileBrowser::ClearSelected()
- {
- selectedFilenames_.clear();
- (*inputNameBuf_)[0] = '\0';
- ok_ = false;
- }
- inline void ImGui::FileBrowser::SetTypeFilters(
- const std::vector<std::string> &_typeFilters)
- {
- typeFilters_.clear();
- // remove duplicate filter names due to case unsensitivity on windows
- #ifdef _WIN32
- std::vector<std::string> typeFilters;
- for(auto &rawFilter : _typeFilters)
- {
- std::string lowerFilter = ToLower(rawFilter);
- auto it = std::find(typeFilters.begin(), typeFilters.end(), lowerFilter);
- if(it == typeFilters.end())
- {
- typeFilters.push_back(std::move(lowerFilter));
- }
- }
- #else
- auto &typeFilters = _typeFilters;
- #endif
- // insert auto-generated filter
- if(typeFilters.size() > 1)
- {
- hasAllFilter_ = true;
- std::string allFiltersName = std::string();
- for(size_t i = 0; i < typeFilters.size(); ++i)
- {
- if(typeFilters[i] == std::string_view(".*"))
- {
- hasAllFilter_ = false;
- break;
- }
- if(i > 0)
- {
- allFiltersName += ",";
- }
- allFiltersName += typeFilters[i];
- }
- if(hasAllFilter_)
- {
- typeFilters_.push_back(std::move(allFiltersName));
- }
- }
- std::copy(
- typeFilters.begin(), typeFilters.end(),
- std::back_inserter(typeFilters_));
- typeFilterIndex_ = 0;
- }
- inline void ImGui::FileBrowser::SetCurrentTypeFilterIndex(int index)
- {
- typeFilterIndex_ = static_cast<unsigned int>(index);
- }
- inline void ImGui::FileBrowser::SetInputName(std::string_view input)
- {
- if(flags_ & ImGuiFileBrowserFlags_EnterNewFilename)
- {
- if(input.size() >= static_cast<size_t>(INPUT_NAME_BUF_SIZE))
- {
- // If input doesn't fit trim off characters
- input = input.substr(0, INPUT_NAME_BUF_SIZE - 1);
- }
- std::copy(input.begin(), input.end(), inputNameBuf_->begin());
- inputNameBuf_->at(input.size()) = '\0';
- selectedFilenames_ = { inputNameBuf_->data() };
- }
- }
- inline std::string ImGui::FileBrowser::ToLower(const std::string &s)
- {
- std::string ret = s;
- for(char &c : ret)
- {
- c = static_cast<char>(std::tolower(c));
- }
- return ret;
- }
- inline void ImGui::FileBrowser::UpdateFileRecords()
- {
- fileRecords_ = { FileRecord{ true, "..", "[D] ..", "" } };
- for(auto &p : std::filesystem::directory_iterator(pwd_))
- {
- FileRecord rcd;
- if(p.is_regular_file())
- {
- rcd.isDir = false;
- }
- else if(p.is_directory())
- {
- rcd.isDir = true;
- }
- else
- {
- continue;
- }
- rcd.name = p.path().filename();
- if(rcd.name.empty())
- {
- continue;
- }
- rcd.extension = p.path().filename().extension();
- rcd.showName = (rcd.isDir ? "[D] " : "[F] ") +
- u8StrToStr(p.path().filename().u8string());
- fileRecords_.push_back(rcd);
- }
- std::sort(fileRecords_.begin(), fileRecords_.end(),
- [](const FileRecord &L, const FileRecord &R)
- {
- return (L.isDir ^ R.isDir) ? L.isDir : (L.name < R.name);
- });
- }
- inline void ImGui::FileBrowser::SetPwdUncatched(const std::filesystem::path &pwd)
- {
- pwd_ = absolute(pwd);
- UpdateFileRecords();
- selectedFilenames_.clear();
- (*inputNameBuf_)[0] = '\0';
- }
- inline bool ImGui::FileBrowser::IsExtensionMatched(
- const std::filesystem::path &_extension) const
- {
- #ifdef _WIN32
- std::filesystem::path extension = ToLower(_extension.string());
- #else
- auto &extension = _extension;
- #endif
- // no type filters
- if(typeFilters_.empty())
- {
- return true;
- }
- // invalid type filter index
- if(static_cast<size_t>(typeFilterIndex_) >= typeFilters_.size())
- {
- return true;
- }
- // all type filters
- if(hasAllFilter_ && typeFilterIndex_ == 0)
- {
- for(size_t i = 1; i < typeFilters_.size(); ++i)
- {
- if(extension == typeFilters_[i])
- {
- return true;
- }
- }
- return false;
- }
- // universal filter
- if(typeFilters_[typeFilterIndex_] == std::string_view(".*"))
- {
- return true;
- }
- // regular filter
- return extension == typeFilters_[typeFilterIndex_];
- }
- #if defined(__cpp_lib_char8_t)
- inline std::string ImGui::FileBrowser::u8StrToStr(std::u8string s)
- {
- return std::string(s.begin(), s.end());
- }
- #endif
- inline std::string ImGui::FileBrowser::u8StrToStr(std::string s)
- {
- return s;
- }
- #ifdef _WIN32
- #ifndef _INC_WINDOWS
- #ifndef WIN32_LEAN_AND_MEAN
- #define IMGUI_FILEBROWSER_UNDEF_WIN32_LEAN_AND_MEAN
- #define WIN32_LEAN_AND_MEAN
- #endif // #ifndef WIN32_LEAN_AND_MEAN
- #include <windows.h>
- #ifdef IMGUI_FILEBROWSER_UNDEF_WIN32_LEAN_AND_MEAN
- #undef IMGUI_FILEBROWSER_UNDEF_WIN32_LEAN_AND_MEAN
- #undef WIN32_LEAN_AND_MEAN
- #endif // #ifdef IMGUI_FILEBROWSER_UNDEF_WIN32_LEAN_AND_MEAN
- #endif // #ifdef _INC_WINDOWS
- inline std::uint32_t ImGui::FileBrowser::GetDrivesBitMask()
- {
- DWORD mask = GetLogicalDrives();
- uint32_t ret = 0;
- for(int i = 0; i < 26; ++i)
- {
- if(!(mask & (1 << i)))
- {
- continue;
- }
- char rootName[4] = { static_cast<char>('A' + i), ':', '\\', '\0' };
- UINT type = GetDriveTypeA(rootName);
- if(type == DRIVE_REMOVABLE || type == DRIVE_FIXED || type == DRIVE_REMOTE)
- {
- ret |= (1 << i);
- }
- }
- return ret;
- }
- #endif
|