#ifndef __MainWindowUI__
|
#define __MainWindowUI__
|
|
/**
|
@file
|
Subclass of UI, which is generated by wxFormBuilder.
|
*/
|
|
#include "MainWindow.h"
|
|
//// end generated include
|
#include "ver.hpp"
|
#include "MainWindowSettings.h"
|
#include "MainWindowImageViewer.h"
|
#include "QueueManager.h"
|
|
#include <filesystem>
|
#include <map>
|
#include <thread>
|
#include <vector>
|
#include <fstream>
|
#include <mutex>
|
|
#include <fmt/format.h>
|
#include "utils.hpp"
|
#include <stable-diffusion.h>
|
|
#include <wx/fileconf.h>
|
#include <wx/stdpaths.h>
|
#include <wx/event.h>
|
#include <wx/notifmsg.h>
|
#include <wx/textdlg.h>
|
#include <wx/menu.h>
|
|
/** Implementing UI */
|
class MainWindowUI : public UI
|
{
|
protected:
|
// Handlers for UI events.
|
void onSettings( wxCommandEvent& event );
|
void onModelsRefresh( wxCommandEvent& event );
|
void onModelSelect( wxCommandEvent& event );
|
void onVaeSelect( wxCommandEvent& event );
|
void onResolutionSwap( wxCommandEvent& event );
|
void onJobsStart( wxCommandEvent& event );
|
void onJobsPause( wxCommandEvent& event );
|
void onJobsDelete( wxCommandEvent& event );
|
void onJoblistItemActivated( wxDataViewEvent& event );
|
void onContextMenu( wxDataViewEvent& event );
|
void onJoblistSelectionChanged( wxDataViewEvent& event );
|
void onGenerate( wxCommandEvent& event );
|
void onSamplerSelect( wxCommandEvent& event );
|
void onSavePreset( wxCommandEvent& event );
|
void onLoadPreset( wxCommandEvent& event );
|
void onSelectPreset( wxCommandEvent& event );
|
void onDeletePreset( wxCommandEvent& event );
|
public:
|
/** Constructor */
|
MainWindowUI( wxWindow* parent );
|
//// end generated class members
|
~MainWindowUI();
|
void OnThreadMessage(wxThreadEvent &e);
|
|
private:
|
std::mutex sdMutex;
|
wxFileConfig *fileConfig;
|
sd_gui_utils::config *cfg;
|
wxString ini_path;
|
MainWindowSettings *settingsWindow;
|
sd_gui_utils::SDParams *sd_params;
|
|
std::map<std::string, std::string> ModelFiles;
|
std::map<std::string, std::string> VaeFiles;
|
std::map<std::string, sd_gui_utils::generator_preset> Presets;
|
|
// the queue manager
|
QM::QueueManager *qmanager;
|
bool modelLoaded = false;
|
std::string currentModel;
|
sd_ctx_t *sd_ctx;
|
std::streambuf *buffer;
|
std::vector<std::thread> threads;
|
|
// row,QueueItem
|
std::map<int, QM::QueueItem*> JobTableItems;
|
std::map<int, wxDataViewColumn *> *JobTableColumns;
|
|
void initConfig();
|
void loadModelList();
|
void loadVaeList();
|
void OnCloseSettings(wxCloseEvent &event);
|
void LoadFileList(sd_gui_utils::DirTypes type = sd_gui_utils::DirTypes::CHECKPOINT);
|
void LoadPresets();
|
|
static void HandleSDLog(sd_log_level_t level, const char *text, void *data);
|
static void HandleSDProgress(int step, int steps, float time, void *data);
|
|
// load the model in a new thread
|
sd_ctx_t *LoadModelv2(wxEvtHandler *eventHandler, QM::QueueItem myItem);
|
// generate in another thread
|
void Generate(wxEvtHandler *eventHandler, QM::QueueItem myItem);
|
|
// start a thread to generate image
|
void StartGeneration(QM::QueueItem myJob);
|
|
// handle queue managers events, manipulate data table by events
|
void OnQueueItemManagerItemAdded(QM::QueueItem item);
|
void OnQueueItemManagerItemUpdated(QM::QueueItem item);
|
void OnQueueItemManagerItemStatusChanged(QM::QueueItem item);
|
|
wxNotificationMessage *notification;
|
|
|
};
|
|
#endif // __MainWindowUI__
|
