Files
oak-editor/app/project/projectimportmanager.cpp
T
itsmattkc 75a9291cd2 refactored import function for greater robustness
The import function was written early on in the rewrite as a multithreaded
background task that was considered somewhat flawed. While it worked for the
most part, there were possibilities of race conditions that could have
potentially been fatal, particularly since media could theoretically be
deleted while the import/probe tasks were running in the background.

With the save/load functions coming in, it became even more complicated as
projects may include metadata about the footage that can't be implemented
easily when the footage is imported/probed in the background. Making importing
a modal task fixes all of these issues, it's still done in a background thread
to not hang the GUI thread, but the GUI thread can be briefly "paused" in a
user friendly way so that all these functions can be safer.
2020-01-15 15:19:14 +11:00

113 lines
3.1 KiB
C++

#include "projectimportmanager.h"
#include <QDir>
#include <QFileInfo>
#include "core.h"
#include "codec/decoder.h"
#include "project/item/footage/footage.h"
ProjectImportManager::ProjectImportManager(ProjectViewModel *model, Folder *folder, const QStringList &filenames) :
model_(model),
folder_(folder)
{
foreach (const QString& f, filenames) {
filenames_.append(f);
}
file_count_ = Core::CountFilesInFileList(filenames_);
}
const int &ProjectImportManager::GetFileCount()
{
return file_count_;
}
void ProjectImportManager::Action()
{
QUndoCommand* command = new QUndoCommand();
int imported = 0;
Import(folder_, filenames_, imported, command);
if (IsCancelled()) {
delete command;
} else {
emit ImportComplete(command);
}
}
void ProjectImportManager::Import(Folder *folder, const QFileInfoList &import, int &counter, QUndoCommand* parent_command)
{
foreach (const QFileInfo& file_info, import) {
if (IsCancelled()) {
break;
}
// Check if this file is a diretory
if (file_info.isDir()) {
// QDir::entryList only returns filenames, we can use entryInfoList() to get full paths
QFileInfoList entry_list = QDir(file_info.absoluteFilePath()).entryInfoList();
// Strip out "." and ".." (for some reason QDir::NoDotAndDotDot doesn't work with entryInfoList, so we have to
// check manually)
for (int i=0;i<entry_list.size();i++) {
if (entry_list.at(i).fileName() == "." || entry_list.at(i).fileName() == "..") {
entry_list.removeAt(i);
i--;
}
}
// Only proceed if the empty actually has files in it
if (!entry_list.isEmpty()) {
// Create a folder corresponding to the directory
ItemPtr f = std::make_shared<Folder>();
f->set_name(file_info.fileName());
// Create undoable command that adds the items to the model
new ProjectViewModel::AddItemCommand(model_,
folder,
f,
parent_command);
// Recursively follow this path
Import(static_cast<Folder*>(f.get()), entry_list, counter, parent_command);
}
} else {
FootagePtr f = std::make_shared<Footage>();
f->set_filename(file_info.absoluteFilePath());
f->set_name(file_info.fileName());
f->set_timestamp(file_info.lastModified());
// Probe will fail if a project isn't set because ImageStream and its derivatives try to connect to the project's
// ColorManager instance
// FIXME: Perhaps re-think this approach at some point
f->set_project(model_->project());
Decoder::ProbeMedia(f.get());
f->set_project(nullptr);
if (f->status() != Footage::kInvalid) {
// Create undoable command that adds the items to the model
new ProjectViewModel::AddItemCommand(model_,
folder,
f,
parent_command);
}
counter++;
emit ProgressChanged((counter * 100) / file_count_);
}
}
}