rpcs3/rpcs3/Gui/VHDDManager.cpp
DH 321d323beb Improved VFS
- Implemended vfsDir.
- Improved vfsDevice.
- Improved vfsFile.
2014-02-16 17:19:06 +02:00

549 lines
14 KiB
C++

#include "stdafx.h"
#include "VHDDManager.h"
#include "TextInputDialog.h"
#include <wx/busyinfo.h>
VHDDListDropTarget::VHDDListDropTarget(wxListView* parent) : m_parent(parent)
{
SetDataObject(new wxDataObjectSimple(wxDF_PRIVATE));
}
wxDragResult VHDDListDropTarget::OnDragOver(wxCoord x, wxCoord y, wxDragResult def)
{
int flags = 0;
int indx = m_parent->HitTest(wxPoint(x, y), flags);
for(int i=0; i<m_parent->GetItemCount(); ++i)
{
m_parent->SetItemState(i, i == indx ? wxLIST_STATE_SELECTED : ~wxLIST_STATE_SELECTED, wxLIST_STATE_SELECTED);
}
return def;
}
void VHDDListDropTarget::OnLeave()
{
for(int i=0; i<m_parent->GetItemCount(); ++i)
{
m_parent->SetItemState(i, ~wxLIST_STATE_SELECTED, wxLIST_STATE_SELECTED);
}
}
wxDragResult VHDDListDropTarget::OnData(wxCoord x, wxCoord y, wxDragResult def)
{
int flags = 0;
int dst_indx = m_parent->HitTest(wxPoint(x, y), flags);
ConLog.Write("OnData(%d -> %d)", m_src_indx, dst_indx);
return def;
}
enum
{
id_open = 0x777,
id_rename,
id_remove,
id_create_dir,
id_create_file,
id_import,
id_export,
id_create_hdd,
id_add_hdd
};
VHDDExplorer::VHDDExplorer(wxWindow* parent, const wxString& hdd_path) : wxDialog(parent, wxID_ANY, "Virtual HDD Explorer", wxDefaultPosition)
{
m_list = new wxListView(this);
m_drop_target = new VHDDListDropTarget(m_list);
m_list->SetDropTarget(m_drop_target);
m_list->DragAcceptFiles(true);
wxBoxSizer& s_main(*new wxBoxSizer(wxVERTICAL));
s_main.Add(m_list, 1, wxEXPAND | wxALL, 5);
SetSizerAndFit(&s_main);
SetSize(800, 600);
m_list->InsertColumn(0, "Name");
m_list->InsertColumn(1, "Type");
m_list->InsertColumn(2, "Size");
m_list->InsertColumn(3, "Creation time");
m_hdd = new vfsHDD(nullptr, hdd_path);
UpdateList();
Connect(m_list->GetId(), wxEVT_COMMAND_LIST_BEGIN_DRAG, wxListEventHandler(VHDDExplorer::OnListDrag));
Connect(m_list->GetId(), wxEVT_COMMAND_LIST_ITEM_ACTIVATED, wxListEventHandler(VHDDExplorer::DClick));
Connect(m_list->GetId(), wxEVT_COMMAND_RIGHT_CLICK, wxCommandEventHandler(VHDDExplorer::OnContextMenu));
m_list->Connect(wxEVT_DROP_FILES, wxDropFilesEventHandler(VHDDExplorer::OnDropFiles), (wxObject*)0, this);
Connect(id_open, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDExplorer::OnOpen));
Connect(id_rename, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDExplorer::OnRename));
Connect(id_remove, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDExplorer::OnRemove));
Connect(id_create_dir, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDExplorer::OnCreateDir));
Connect(id_create_file, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDExplorer::OnCreateFile));
Connect(id_import, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDExplorer::OnImport));
Connect(id_export, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDExplorer::OnExport));
}
void VHDDExplorer::UpdateList()
{
m_list->Freeze();
m_list->DeleteAllItems();
m_entries.Clear();
m_names.Clear();
u64 block;
vfsHDD_Entry entry;
wxString name;
for(bool is_ok = m_hdd->GetFirstEntry(block, entry, name); is_ok; is_ok = m_hdd->GetNextEntry(block, entry, name))
{
int item = m_list->GetItemCount();
m_list->InsertItem(item, name);
m_list->SetItem(item, 1, entry.type == vfsHDD_Entry_Dir ? "Dir" : "File");
m_list->SetItem(item, 2, wxString::Format("%lld", entry.size));
m_list->SetItem(item, 3, wxDateTime().Set(time_t(entry.ctime)).Format());
m_entries.AddCpy(entry);
m_names.Add(name);
}
m_list->SetColumnWidth(0, wxLIST_AUTOSIZE_USEHEADER);
m_list->SetColumnWidth(1, wxLIST_AUTOSIZE_USEHEADER);
m_list->SetColumnWidth(2, wxLIST_AUTOSIZE_USEHEADER);
m_list->SetColumnWidth(3, wxLIST_AUTOSIZE_USEHEADER);
m_list->Thaw();
}
void VHDDExplorer::Import(const wxString& path, const wxString& to)
{
if(!m_hdd->Create(vfsHDD_Entry_File, to))
{
wxMessageBox("IMPORT ERROR: file creation error.");
return;
}
if(!m_hdd->Open(to, vfsWrite))
{
wxMessageBox("IMPORT ERROR: file open error.");
return;
}
wxFile f(path);
char buf[256];
while(!f.Eof())
{
m_hdd->Write(buf, f.Read(buf, 256));
}
m_hdd->Close();
}
void VHDDExplorer::Export(const wxString& path, const wxString& to)
{
if(!m_hdd->Open(path))
{
wxMessageBox(wxString::Format("EXPORT ERROR: file open error. (%s)", path.wx_str()));
return;
}
wxFile f(to, wxFile::write);
char buf[256];
while(u64 size = m_hdd->Read(buf, 256))
{
f.Write(buf, size);
}
}
void VHDDExplorer::OpenDir(int sel)
{
if(sel == wxNOT_FOUND)
{
return;
}
if(m_entries[sel].type == vfsHDD_Entry_Dir)
{
m_hdd->OpenDir(m_names[sel]);
UpdateList();
}
}
void VHDDExplorer::OnListDrag(wxListEvent& event)
{
m_drop_target->SetSrcIndx(event.GetIndex());
wxDataObjectSimple obj(wxDF_PRIVATE);
wxDropSource drag(obj, m_list);
drag.DoDragDrop(wxDrag_AllowMove);
}
void VHDDExplorer::OnDropFiles(wxDropFilesEvent& event)
{
int count = event.GetNumberOfFiles();
wxString* dropped = event.GetFiles();
wxBusyCursor busyCursor;
wxWindowDisabler disabler;
wxBusyInfo busyInfo("Adding files, wait please...");
for(int i=0; i<count; ++i)
{
ConLog.Write("Importing '%s'", dropped[i].wx_str());
Import(dropped[i], wxFileName(dropped[i]).GetFullName());
}
UpdateList();
}
void VHDDExplorer::DClick(wxListEvent& event)
{
OpenDir(event.GetIndex());
}
void VHDDExplorer::OnContextMenu(wxCommandEvent& event)
{
wxMenu* menu = new wxMenu();
int idx = m_list->GetFirstSelected();
menu->Append(id_open, "Open")->Enable(idx != wxNOT_FOUND && m_entries[idx].type == vfsHDD_Entry_Dir);
menu->Append(id_rename, "Rename")->Enable(idx != wxNOT_FOUND && m_names[idx] != "." && m_names[idx] != "..");
menu->Append(id_remove, "Remove")->Enable(idx != wxNOT_FOUND && m_names[idx] != "." && m_names[idx] != "..");
menu->AppendSeparator();
menu->Append(id_create_dir, "Create dir");
menu->Append(id_create_file, "Create file");
menu->AppendSeparator();
menu->Append(id_import, "Import");
menu->Append(id_export, "Export")->Enable(idx != wxNOT_FOUND);
PopupMenu( menu );
}
void VHDDExplorer::OnOpen(wxCommandEvent& event)
{
m_hdd->OpenDir(m_names[m_list->GetFirstSelected()]);
UpdateList();
}
void VHDDExplorer::OnRename(wxCommandEvent& event)
{
TextInputDialog dial(this, m_names[m_list->GetFirstSelected()]);
if(dial.ShowModal() == wxID_OK)
{
m_hdd->Rename(m_names[m_list->GetFirstSelected()], dial.GetResult());
UpdateList();
}
}
void VHDDExplorer::OnRemove(wxCommandEvent& event)
{
for(int sel = m_list->GetNextSelected(-1); sel != wxNOT_FOUND; sel = m_list->GetNextSelected(sel))
{
m_hdd->RemoveEntry(m_names[sel]);
}
UpdateList();
}
void VHDDExplorer::OnCreateDir(wxCommandEvent& event)
{
int i = 1;
static const wxString& fmt = "New Dir (%d)";
while(m_hdd->HasEntry(wxString::Format(fmt, i))) i++;
m_hdd->Create(vfsHDD_Entry_Dir, wxString::Format(fmt, i));
UpdateList();
}
void VHDDExplorer::OnCreateFile(wxCommandEvent& event)
{
int i = 1;
static const wxString& fmt = "New File (%d)";
while(m_hdd->HasEntry(wxString::Format(fmt, i))) i++;
m_hdd->Create(vfsHDD_Entry_File, wxString::Format(fmt, i));
UpdateList();
}
void VHDDExplorer::OnImport(wxCommandEvent& event)
{
wxFileDialog ctrl(this, "Select import files", wxEmptyString, wxEmptyString, wxFileSelectorDefaultWildcardStr,
wxFD_OPEN | wxFD_FILE_MUST_EXIST | wxFD_MULTIPLE);
if(ctrl.ShowModal() == wxID_CANCEL)
{
return;
}
wxArrayString paths;
ctrl.GetPaths(paths);
for(size_t i=0; i<paths.GetCount(); ++i)
{
if(wxFileExists(paths[i]))
{
Import(paths[i], wxFileName(paths[i]).GetFullName());
}
}
UpdateList();
}
void VHDDExplorer::OnExport(wxCommandEvent& event)
{
if(m_list->GetSelectedItemCount() > 1)
{
wxDirDialog ctrl(this, "Select export folder", wxGetCwd());
if(ctrl.ShowModal() == wxID_CANCEL)
{
return;
}
for(int sel = m_list->GetNextSelected(-1); sel != wxNOT_FOUND; sel = m_list->GetNextSelected(sel))
{
Export(m_names[sel], ctrl.GetPath() + '\\' + m_names[sel]);
}
}
else
{
int sel = m_list->GetFirstSelected();
wxFileDialog ctrl(this, "Select export file", wxEmptyString, m_names[sel], wxFileSelectorDefaultWildcardStr, wxFD_SAVE | wxFD_OVERWRITE_PROMPT);
if(ctrl.ShowModal() == wxID_CANCEL)
{
return;
}
Export(m_names[sel], ctrl.GetPath());
}
UpdateList();
}
VHDDSetInfoDialog::VHDDSetInfoDialog(wxWindow* parent) : wxDialog(parent, wxID_ANY, "HDD Settings", wxDefaultPosition)
{
m_spin_size = new wxSpinCtrl(this);
m_ch_type = new wxChoice(this, wxID_ANY);
m_spin_block_size = new wxSpinCtrl(this);
wxBoxSizer& s_sinf(*new wxBoxSizer(wxHORIZONTAL));
s_sinf.Add(m_spin_size, wxSizerFlags().Border(wxALL, 5).Expand());
s_sinf.Add(m_ch_type, wxSizerFlags().Border(wxALL, 5).Expand());
wxBoxSizer& s_binf(*new wxBoxSizer(wxHORIZONTAL));
s_binf.Add(m_spin_block_size, wxSizerFlags().Border(wxALL, 5).Expand());
wxBoxSizer& s_btns(*new wxBoxSizer(wxHORIZONTAL));
s_btns.Add(new wxButton(this, wxID_OK), wxSizerFlags().Align(wxALIGN_LEFT).Border(wxALL, 5));
s_btns.Add(new wxButton(this, wxID_CANCEL), wxSizerFlags().Align(wxALIGN_RIGHT).Border(wxALL, 5));
wxBoxSizer& s_main(*new wxBoxSizer(wxVERTICAL));
s_main.Add(&s_sinf, wxSizerFlags().Align(wxALIGN_TOP).Expand());
s_main.Add(&s_binf, wxSizerFlags().Align(wxALIGN_TOP).Expand());
s_main.Add(&s_btns, wxSizerFlags().Align(wxALIGN_BOTTOM).Expand());
SetSizerAndFit(&s_main);
m_ch_type->Append("B");
m_ch_type->Append("KB");
m_ch_type->Append("MB");
m_ch_type->Append("GB");
m_spin_size->SetRange(1, 0x7fffffff);
m_spin_size->SetValue(64);
m_ch_type->SetSelection(3);
m_spin_block_size->SetRange(64, 0x7fffffff);
m_spin_block_size->SetValue(2048);
Connect(wxID_OK, wxEVT_COMMAND_BUTTON_CLICKED, wxCommandEventHandler(VHDDSetInfoDialog::OnOk));
}
void VHDDSetInfoDialog::OnOk(wxCommandEvent& event)
{
m_res_size = m_spin_size->GetValue();
for(int i=0; i<m_ch_type->GetSelection(); ++i)
{
m_res_size *= 1024;
}
m_res_block_size = m_spin_block_size->GetValue();
EndModal(wxID_OK);
}
void VHDDSetInfoDialog::GetResult(u64& size, u64& block_size)
{
size = m_res_size;
block_size = m_res_block_size;
}
VHDDManagerDialog::VHDDManagerDialog(wxWindow* parent)
: wxDialog(parent, wxID_ANY, "Virtual HDD Manager", wxDefaultPosition)
{
m_list = new wxListView(this);
wxBoxSizer& s_main(*new wxBoxSizer(wxVERTICAL));
s_main.Add(m_list, 1, wxEXPAND | wxALL, 5);
SetSizerAndFit(&s_main);
SetSize(800, 600);
m_list->InsertColumn(0, "Path");
//m_list->InsertColumn(1, "Size");
//m_list->InsertColumn(2, "Block size");
Connect(m_list->GetId(), wxEVT_COMMAND_LIST_ITEM_ACTIVATED, wxListEventHandler(VHDDManagerDialog::DClick));
Connect(m_list->GetId(), wxEVT_COMMAND_RIGHT_CLICK, wxCommandEventHandler(VHDDManagerDialog::OnContextMenu));
Connect(id_add_hdd, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDManagerDialog::AddHDD));
Connect(id_open, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDManagerDialog::OnOpen));
Connect(id_remove, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDManagerDialog::OnRemove));
Connect(id_create_hdd, wxEVT_COMMAND_MENU_SELECTED, wxCommandEventHandler(VHDDManagerDialog::OnCreateHDD));
Connect(wxEVT_CLOSE_WINDOW, wxCloseEventHandler(VHDDManagerDialog::OnClose));
LoadPathes();
UpdateList();
}
void VHDDManagerDialog::UpdateList()
{
m_list->Freeze();
m_list->DeleteAllItems();
for(size_t i=0; i<m_pathes.GetCount(); ++i)
{
m_list->InsertItem(i, m_pathes[i]);
}
m_list->SetColumnWidth(0, wxLIST_AUTOSIZE_USEHEADER);
//m_list->SetColumnWidth(1, wxLIST_AUTOSIZE_USEHEADER);
//m_list->SetColumnWidth(2, wxLIST_AUTOSIZE_USEHEADER);
m_list->Thaw();
}
void VHDDManagerDialog::Open(int sel)
{
VHDDExplorer dial(this, m_pathes[sel]);
dial.ShowModal();
}
void VHDDManagerDialog::DClick(wxListEvent& event)
{
Open(event.GetIndex());
}
void VHDDManagerDialog::AddHDD(wxCommandEvent& event)
{
wxFileDialog ctrl(this, "Select HDDs", wxEmptyString, wxEmptyString, "Virtual HDD (*.hdd) | *.hdd",
wxFD_OPEN | wxFD_FILE_MUST_EXIST | wxFD_MULTIPLE);
if(ctrl.ShowModal() == wxID_CANCEL)
{
return;
}
wxArrayString pathes;
ctrl.GetPaths(pathes);
for(size_t i=0; i<pathes.GetCount(); ++i)
{
bool skip = false;
for(size_t j=0; j<m_pathes.GetCount(); ++j)
{
if(m_pathes[j].CmpNoCase(pathes[i]) == 0)
{
skip = true;
break;
}
}
if(!skip)
{
m_pathes.Move(new wxString(pathes[i].c_str()));
}
}
UpdateList();
}
void VHDDManagerDialog::OnContextMenu(wxCommandEvent& event)
{
wxMenu* menu = new wxMenu();
int idx = m_list->GetFirstSelected();
menu->Append(id_open, "Open")->Enable(idx != wxNOT_FOUND);
menu->Append(id_remove, "Remove")->Enable(idx != wxNOT_FOUND);
menu->AppendSeparator();
menu->Append(id_add_hdd, "Add");
menu->Append(id_create_hdd, "Create");
PopupMenu(menu);
}
void VHDDManagerDialog::OnOpen(wxCommandEvent& event)
{
int idx = m_list->GetFirstSelected();
if(idx >= 0) Open(idx);
}
void VHDDManagerDialog::OnRemove(wxCommandEvent& event)
{
for(int sel = m_list->GetNextSelected(-1), offs = 0; sel != wxNOT_FOUND; sel = m_list->GetNextSelected(sel), --offs)
{
m_pathes.RemoveAt(sel + offs);
}
UpdateList();
}
void VHDDManagerDialog::OnCreateHDD(wxCommandEvent& event)
{
wxFileDialog ctrl(this, "Select HDD path", wxEmptyString, "new_hdd.hdd", "Virtual HDD (*.hdd) | *.hdd",
wxFD_SAVE | wxFD_OVERWRITE_PROMPT);
if(ctrl.ShowModal() == wxID_CANCEL)
{
return;
}
VHDDSetInfoDialog dial(this);
if(dial.ShowModal() == wxID_OK)
{
u64 size, bsize;
dial.GetResult(size, bsize);
vfsHDDManager::CreateHDD(ctrl.GetPath(), size, bsize);
m_pathes.AddCpy(ctrl.GetPath());
UpdateList();
}
}
void VHDDManagerDialog::OnClose(wxCloseEvent& event)
{
SavePathes();
event.Skip();
}
void VHDDManagerDialog::LoadPathes()
{
IniEntry<int> path_count;
path_count.Init("path_count", "HDDManager");
int count = 0;
count = path_count.LoadValue(count);
m_pathes.SetCount(count);
for(size_t i=0; i<m_pathes.GetCount(); ++i)
{
IniEntry<wxString> path_entry;
path_entry.Init(wxString::Format("path[%d]", i), "HDDManager");
new (m_pathes + i) wxString(path_entry.LoadValue(wxEmptyString));
}
}
void VHDDManagerDialog::SavePathes()
{
IniEntry<int> path_count;
path_count.Init("path_count", "HDDManager");
path_count.SaveValue(m_pathes.GetCount());
for(size_t i=0; i<m_pathes.GetCount(); ++i)
{
IniEntry<wxString> path_entry;
path_entry.Init(wxString::Format("path[%d]", i), "HDDManager");
path_entry.SaveValue(m_pathes[i]);
}
}