tahoma2d/toonz/sources/toonzfarm/tfarmclient/filebrowser.cpp
2016-06-15 15:43:10 +09:00

582 lines
16 KiB
C++

#include "filebrowser.h"
#include "thumbnail.h"
#include "thumbnailviewer.h"
#include "tw/treeview.h"
#include "tw/colors.h"
#include "tw/button.h"
#include "tw/scrollbar.h"
#include "tw/mainshell.h"
#include "tfilepath.h"
#include "tsystem.h"
#include "tenv.h"
#include "tw/message.h"
using namespace TwConsts;
//===================================================================
namespace {
#include "filebrowser_icons.h"
//===================================================================
class DirectoryBrowser : public ThumbnailViewer {
TFilePath m_filepath;
vector<string> m_fileTypes;
GenericFileBrowserAction *m_fileSelChangeAction;
GenericFileBrowserAction *m_fileDblClickAction;
public:
DirectoryBrowser(TWidget *parent, const vector<string> &fileTypes)
: ThumbnailViewer(parent)
, m_fileSelChangeAction(0)
, m_fileDblClickAction(0)
, m_fileTypes(fileTypes) {}
~DirectoryBrowser() {
if (m_fileSelChangeAction) delete m_fileSelChangeAction;
if (m_fileDblClickAction) delete m_fileDblClickAction;
}
void setPath(const TFilePath &filepath);
void setFilter(const vector<string> &fileTypes);
TFilePath getPath() const { return m_filepath; }
void onDoubleClick(int index);
void onSelect(int index);
void setFileSelChangeAction(GenericFileBrowserAction *action) {
m_fileSelChangeAction = action;
}
void setFileDblClickAction(GenericFileBrowserAction *action) {
m_fileDblClickAction = action;
}
};
//-------------------------------------------------------------------
void DirectoryBrowser::setPath(const TFilePath &filepath) {
m_filepath = filepath;
loadDirectory(filepath, m_fileTypes);
}
//-------------------------------------------------------------------
void DirectoryBrowser::setFilter(const vector<string> &fileTypes) {
m_fileTypes = fileTypes;
if (!m_filepath.isEmpty()) loadDirectory(m_filepath, m_fileTypes);
}
//-------------------------------------------------------------------
void DirectoryBrowser::onDoubleClick(int index) {
Thumbnail *thumbnail = getItem(index);
if (thumbnail->getType() != Thumbnail::SCENE) return;
TFilePath path =
thumbnail->getPath().getParentDir() + (thumbnail->getName() + "_files");
setPath(path);
invalidate();
}
//-------------------------------------------------------------------
void DirectoryBrowser::onSelect(int index) {
if (index != -1) {
Thumbnail *thumbnail = getItem(index);
if (m_fileSelChangeAction)
m_fileSelChangeAction->sendCommand(thumbnail->getPath());
}
}
//===================================================================
class MyTreeViewItem : public TTreeViewItem {
public:
MyTreeViewItem(TTreeViewItemParent *parent) : TTreeViewItem(parent) {}
TDimension getIconSize() const { return TDimension(26, 26); }
void drawIcon(TTreeView *w, const TPoint &origin) {
TRect rect(origin, getIconSize());
w->setColor(TWidget::Black);
w->drawRect(rect);
rect = rect.enlarge(-4);
w->setColor(Red);
w->drawRect(rect);
w->setColor(Blue);
w->fillRect(rect.enlarge(-1));
}
void drawName(TTreeView *w, const TPoint &origin) {
string name = toString(getName());
TDimension textSize = w->getTextSize(name);
TDimension iconSize = getIconSize();
TPoint pos = origin;
pos.x += iconSize.lx;
if (isSelected()) {
w->setColor(ToonzHighlightColor);
w->fillRect(pos.x + 2, pos.y + 2, pos.x + textSize.lx + 8,
pos.y + iconSize.ly - 1 - 2);
}
w->setColor(TWidget::Black);
w->drawText(pos + TPoint(5, iconSize.ly / 2 - textSize.ly / 2), name);
}
/*
void draw(TTreeView *w, const TPoint &origin) {
drawIcon(w,origin);
drawName(w,origin);
}
*/
};
//===================================================================
class MyTreeView : public TTreeView {
public:
MyTreeView(TWidget *parent, string name = "MyTreeView")
: TTreeView(parent, name) {
setBackgroundColor(White);
}
void drawButton(const TPoint &p, bool open) {
int r1 = 3, r2 = 1;
TRect rect(p.x - r1, p.y - r1, p.x + r1, p.y + r1);
setColor(getBackgroundColor());
fillRect(rect.enlarge(-1));
setColor(Gray210);
drawRect(rect);
setColor(Black);
drawLine(p.x - r2, p.y, p.x + r2, p.y);
if (!open) drawLine(p.x, p.y - r2, p.x, p.y + r2);
}
};
//===================================================================
class SimpleFileItem : public MyTreeViewItem {
TFilePath m_path;
public:
SimpleFileItem(TTreeViewItemParent *parent, const TFilePath &path)
: MyTreeViewItem(parent), m_path(path) {
setIsLeaf(true);
}
wstring getName() const { return m_path.withoutParentDir().getWideString(); }
TFilePath getPath() const { return m_path; }
};
//-------------------------------------------------------------------
bool caseInsensitiveLessThan(const TFilePath &fp1, const TFilePath &fp2) {
return (stricmp(toString(fp1.getWideString()).c_str(),
toString(fp2.getWideString()).c_str()) < 0);
}
//-------------------------------------------------------------------
class FileFolderItem : public MyTreeViewItem {
TFilePath m_path;
public:
FileFolderItem(TTreeViewItemParent *parent, const TFilePath &path)
: MyTreeViewItem(parent), m_path(path) {}
wstring getName() const {
return m_path.isRoot() ? m_path.getWideString()
: m_path.withoutParentDir().getWideString();
}
TDimension getIconSize() const { return TDimension(16, 16); }
void onOpen() {
clearItems();
try {
TFilePathSet fps = TSystem::readDirectory(m_path);
fps.sort(caseInsensitiveLessThan);
for (TFilePathSet::iterator it = fps.begin(); it != fps.end(); it++) {
if (TFileStatus(*it).isDirectory()) new FileFolderItem(this, *it);
// else
// new SimpleFileItem(this, *it);
}
} catch (TException &e) {
TMessage::error(toString(e.getMessage()));
}
}
TFilePath getPath() const { return m_path; }
void drawIcon(TTreeView *w, const TPoint &origin) {
w->rectwrite(isOpen() ? openfolder : closedfolder, origin);
}
};
//-------------------------------------------------------------------
class MyComputerFolder : public MyTreeViewItem {
public:
MyComputerFolder(TTreeViewItemParent *parent) : MyTreeViewItem(parent) {}
wstring getName() const { return toWideString("My Computer"); }
void onOpen() {
clearItems();
TFilePathSet fps = TSystem::getDisks();
for (TFilePathSet::iterator it = fps.begin(); it != fps.end(); it++) {
TTreeViewItem *item = new FileFolderItem(this, *it);
}
}
void drawIcon(TTreeView *w, const TPoint &origin) {
w->rectwrite(mycomputer, origin);
}
};
//-------------------------------------------------------------------
class HistoryFolder : public MyTreeViewItem {
public:
HistoryFolder(TTreeViewItemParent *parent) : MyTreeViewItem(parent) {}
wstring getName() const { return toWideString("History"); }
void onOpen() { clearItems(); }
void drawIcon(TTreeView *w, const TPoint &origin) {
w->rectwrite(history, origin);
}
};
//-------------------------------------------------------------------
class LibraryFolder : public MyTreeViewItem {
public:
LibraryFolder(TTreeViewItemParent *parent) : MyTreeViewItem(parent) {}
wstring getName() const { return toWideString("Library"); }
void onOpen() {
clearItems();
TFilePath root = TEnv::getRootDir();
new FileFolderItem(this, root + "patterns");
new FileFolderItem(this, root + "images");
new FileFolderItem(this, root + "scenes");
}
void drawIcon(TTreeView *w, const TPoint &origin) {
w->rectwrite(library, origin);
}
};
//-------------------------------------------------------------------
class WorkFolder : public FileFolderItem {
public:
WorkFolder(TTreeViewItemParent *parent)
: FileFolderItem(parent, TEnv::getRootDir() + "work") {}
TDimension getIconSize() const { return TDimension(24, 24); }
wstring getName() const { return toWideString("Work"); }
void drawIcon(TTreeView *w, const TPoint &origin) {
w->rectwrite(workinprogress, origin);
}
};
//-------------------------------------------------------------------
class FileFolderTreeView : public MyTreeView {
DirectoryBrowser *m_viewer;
TTreeViewItem *m_myComputer;
public:
FileFolderTreeView(TWidget *parent, DirectoryBrowser *viewer)
: MyTreeView(parent), m_viewer(viewer) {
m_myComputer = new MyComputerFolder(this);
m_myComputer->open();
/*
new HistoryFolder(this);
new LibraryFolder(this);
new WorkFolder(this);
*/
}
void leftButtonDoubleClick(const TMouseEvent &e);
void onSelect(TTreeViewItem *item);
void setCurrentDir(const TFilePath &dirPath);
};
//===================================================================
void FileFolderTreeView::leftButtonDoubleClick(const TMouseEvent &e) {
TTreeViewItem *item = getSelectedItem();
if (item) {
if (!item->isOpen())
item->open();
else
item->close();
updateVisibleItems();
invalidate();
}
}
//-------------------------------------------------------------------
void FileFolderTreeView::onSelect(TTreeViewItem *item) {
FileFolderItem *folder = dynamic_cast<FileFolderItem *>(item);
if (folder && folder->getPath() != m_viewer->getPath()) {
TFilePath path = folder->getPath();
m_viewer->setPath(path);
m_viewer->invalidate();
}
}
//-------------------------------------------------------------------
TTreeViewItem *openChild(TTreeViewItem *folder, const TFilePath &childPath) {
int count = folder->getItemCount();
for (int i = 0; i < count; ++i) {
FileFolderItem *item = dynamic_cast<FileFolderItem *>(folder->getItem(i));
if (item && item->getPath() == childPath) {
item->open();
return item;
}
}
return 0;
}
//-------------------------------------------------------------------
void FileFolderTreeView::setCurrentDir(const TFilePath &dirPath) {
assert(TFileStatus(dirPath).isDirectory());
list<TFilePath> ancestors;
ancestors.push_back(dirPath);
if (!dirPath.isRoot()) {
TFilePath fp = dirPath;
while (!(fp = fp.getParentDir()).isRoot()) ancestors.push_front(fp);
ancestors.push_front(fp);
}
TTreeViewItem *folder = m_myComputer;
while (!ancestors.empty() && folder) {
TFilePath fp = ancestors.front();
folder = openChild(folder, fp);
ancestors.pop_front();
}
if (folder) {
updateVisibleItems();
select(folder);
TRect rect;
getPlacement(m_selectedItemIndex, rect);
int y = rect.getP00().y;
if (y <= 0)
y = m_yoffset - y;
else if (y > getLy())
y = m_yoffset - (rect.y1 - getLy());
else
y = m_yoffset;
scrollToY(y);
updateScrollbars();
}
}
}; // anonymous namespace
//==============================================================================
//==============================================================================
class FileBrowser::Data {
public:
Data() {}
~Data() {}
FileFolderTreeView *m_tree;
DirectoryBrowser *m_panel;
TScrollbar *m_panelScb, *m_treeVscb, *m_treeHscb;
TIconButton *m_upButton, *m_createFolderButton, *m_deleteFileButton;
vector<string> m_fileTypes;
};
//-------------------------------------------------------------------
FileBrowser::FileBrowser(TWidget *parent, string name,
const vector<string> &fileTypes)
: TWidget(parent, name), m_data(new Data()) {
m_data->m_fileTypes = fileTypes;
m_data->m_panel = new DirectoryBrowser(this, fileTypes);
m_data->m_tree = new FileFolderTreeView(this, m_data->m_panel);
m_data->m_panelScb = new TScrollbar(this);
m_data->m_treeHscb = new TScrollbar(this);
m_data->m_treeVscb = new TScrollbar(this);
m_data->m_panel->setScrollbars(0, m_data->m_panelScb);
m_data->m_tree->setScrollbars(m_data->m_treeHscb, m_data->m_treeVscb);
setBackgroundColor(Gray210);
m_data->m_upButton = new TIconButton(this, parentfolder);
tconnect(*m_data->m_upButton, this, &FileBrowser::selectParentDirectory);
m_data->m_createFolderButton = new TIconButton(this, newfolder);
m_data->m_deleteFileButton = new TIconButton(this, trash);
}
//-------------------------------------------------------------------
FileBrowser::~FileBrowser() { delete m_data; }
//-------------------------------------------------------------------
void FileBrowser::setFilter(const vector<string> &fileTypes) {
m_data->m_fileTypes = fileTypes;
m_data->m_panel->setFilter(fileTypes);
}
//-------------------------------------------------------------------
void FileBrowser::draw() {
setColor(White);
fillRect(0, 0, 4, getLy() - 1);
TPoint p;
p = m_data->m_treeVscb->getGeometry().getP11();
setColor(Gray150);
drawLine(p.x + 1, 1, p.x + 1, p.y);
p = m_data->m_panel->getGeometry().getP01();
drawLine(p.x - 1, 1, p.x - 1, p.y);
drawLine(p.x - 1, p.y + 1, getLx() - 2, p.y + 1);
}
//-------------------------------------------------------------------
void FileBrowser::configureNotify(const TDimension &d) {
/*
TRect rect = getBounds();
int x = 200;
int w = 15;
int xa = x-w-4;
TDimension scbSize(w, rect.getLy());
m_data->m_tree->setGeometry(rect.x0,rect.y0,xa-1,rect.y1);
m_data->m_treeVscb->setGeometry(TPoint(xa,rect.y0),scbSize);
m_data->m_treeHscb->setGeometry(rect.x0, rect.y0,
x-1-scbSize,rect.y0+scbSize-1);
int buttonBarHeight = 30;
rect.y1 -= buttonBarHeight;
scbSize.ly -= buttonBarHeight;
m_data->m_panel->setGeometry(x,rect.y0,rect.x1-w,rect.y1);
m_data->m_panelScb->setGeometry(TPoint(rect.x1-w+1,rect.y0),scbSize);
int y = rect.y1+4;
m_data->m_upButton->setPosition(x-1, y);
m_data->m_createFolderButton->setPosition(x+25, y);
m_data->m_deleteFileButton->setPosition(rect.x1-26, y);
invalidate();
*/
const int scbSize = 15;
TRect rect = getBounds();
int x = 196;
m_data->m_tree->setGeometry(rect.x0, rect.y0 + scbSize, x - 1 - scbSize,
rect.y1);
m_data->m_treeVscb->setGeometry(x - scbSize, rect.y0 + scbSize, x - 1,
rect.y1);
m_data->m_treeHscb->setGeometry(rect.x0, rect.y0, x - 1 - scbSize,
rect.y0 + scbSize - 1);
int buttonBarHeight = 29;
rect.y1 -= buttonBarHeight;
m_data->m_panel->setGeometry(x + 4, rect.y0, rect.x1 - scbSize, rect.y1);
m_data->m_panelScb->setGeometry(rect.x1 - scbSize + 1, rect.y0, rect.x1,
rect.y1);
int y = rect.y1 + 4;
x += 4;
m_data->m_upButton->setPosition(x - 1, y);
m_data->m_createFolderButton->setPosition(x + 25, y);
m_data->m_deleteFileButton->setPosition(rect.x1 - 26, y);
m_data->m_upButton->invalidate();
m_data->m_createFolderButton->invalidate();
m_data->m_deleteFileButton->invalidate();
m_data->m_treeVscb->invalidate();
m_data->m_treeHscb->invalidate();
m_data->m_tree->invalidate();
invalidate();
}
//-------------------------------------------------------------------
void FileBrowser::setBackgroundColor(const TGuiColor &color) {
TWidget::setBackgroundColor(color);
m_data->m_panel->setBackgroundColor(color);
m_data->m_tree->setBackgroundColor(color);
}
//-------------------------------------------------------------------
void FileBrowser::setFileSelChangeAction(GenericFileBrowserAction *action) {
m_data->m_panel->setFileSelChangeAction(action);
}
//-------------------------------------------------------------------
void FileBrowser::setFileDblClickAction(GenericFileBrowserAction *action) {
m_data->m_panel->setFileDblClickAction(action);
}
//-------------------------------------------------------------------
TFilePath FileBrowser::getCurrentDir() const {
FileFolderItem *item =
dynamic_cast<FileFolderItem *>(m_data->m_tree->getSelectedItem());
if (item)
return item->getPath();
else
return TFilePath();
}
//-------------------------------------------------------------------
void FileBrowser::setCurrentDir(const TFilePath &dirPath) {
m_data->m_tree->setCurrentDir(dirPath);
}
//-------------------------------------------------------------------
void FileBrowser::selectParentDirectory() { m_data->m_tree->selectParent(); }