diff --git a/src/base/bittorrent/torrentcreatorthread.cpp b/src/base/bittorrent/torrentcreatorthread.cpp index aaec19324..bb2dfecab 100644 --- a/src/base/bittorrent/torrentcreatorthread.cpp +++ b/src/base/bittorrent/torrentcreatorthread.cpp @@ -38,8 +38,12 @@ #include #include +#include #include +#include +#include +#include "base/global.h" #include "base/utils/fs.h" #include "base/utils/misc.h" #include "base/utils/string.h" @@ -86,13 +90,51 @@ void TorrentCreatorThread::run() emit updateProgress(0); try { - libt::file_storage fs; + const QString parentPath = Utils::Fs::branchPath(m_params.inputPath) + "/"; + // Adding files to the torrent - libt::add_files(fs, Utils::Fs::toNativePath(m_params.inputPath).toStdString(), fileFilter); + libt::file_storage fs; + if (QFileInfo(m_params.inputPath).isFile()) { + libt::add_files(fs, Utils::Fs::toNativePath(m_params.inputPath).toStdString(), fileFilter); + } + else { + // need to sort the file names by natural sort order + QStringList dirs = {m_params.inputPath}; + + QDirIterator dirIter(m_params.inputPath, (QDir::AllDirs | QDir::NoDotAndDotDot), QDirIterator::Subdirectories); + while (dirIter.hasNext()) { + dirIter.next(); + dirs += dirIter.filePath(); + } + std::sort(dirs.begin(), dirs.end(), Utils::String::naturalLessThan); + + QStringList fileNames; + QHash fileSizeMap; + + for (const auto &dir : qAsConst(dirs)) { + QStringList tmpNames; // natural sort files within each dir + + QDirIterator fileIter(dir, QDir::Files); + while (fileIter.hasNext()) { + fileIter.next(); + + const QString relFilePath = fileIter.filePath().mid(parentPath.length()); + tmpNames += relFilePath; + fileSizeMap[relFilePath] = fileIter.fileInfo().size(); + } + + std::sort(tmpNames.begin(), tmpNames.end(), Utils::String::naturalLessThan); + fileNames += tmpNames; + } + + for (const auto &fileName : qAsConst(fileNames)) + fs.add_file(fileName.toStdString(), fileSizeMap[fileName]); + } if (isInterruptionRequested()) return; - libt::create_torrent newTorrent(fs, m_params.pieceSize); + libt::create_torrent newTorrent(fs, m_params.pieceSize, -1 + , (m_params.isAlignmentOptimized ? libt::create_torrent::optimize_alignment : 0)); // Add url seeds foreach (QString seed, m_params.urlSeeds) { @@ -112,7 +154,6 @@ void TorrentCreatorThread::run() if (isInterruptionRequested()) return; // calculate the hash for all pieces - const QString parentPath = Utils::Fs::branchPath(m_params.inputPath) + "/"; libt::set_piece_hashes(newTorrent, Utils::Fs::toNativePath(parentPath).toStdString() , [this, &newTorrent](const int n) { sendProgressSignal(n, newTorrent.num_pieces()); }); // Set qBittorrent as creator and add user comment to @@ -156,12 +197,13 @@ void TorrentCreatorThread::run() } } -int TorrentCreatorThread::calculateTotalPieces(const QString &inputPath, const int pieceSize) +int TorrentCreatorThread::calculateTotalPieces(const QString &inputPath, const int pieceSize, const bool isAlignmentOptimized) { if (inputPath.isEmpty()) return 0; libt::file_storage fs; libt::add_files(fs, Utils::Fs::toNativePath(inputPath).toStdString(), fileFilter); - return libt::create_torrent(fs, pieceSize).num_pieces(); + return libt::create_torrent(fs, pieceSize, -1 + , (isAlignmentOptimized ? libt::create_torrent::optimize_alignment : 0)).num_pieces(); } diff --git a/src/base/bittorrent/torrentcreatorthread.h b/src/base/bittorrent/torrentcreatorthread.h index 58785cfa2..5704315be 100644 --- a/src/base/bittorrent/torrentcreatorthread.h +++ b/src/base/bittorrent/torrentcreatorthread.h @@ -39,6 +39,7 @@ namespace BitTorrent struct TorrentCreatorParams { bool isPrivate; + bool isAlignmentOptimized; int pieceSize; QString inputPath; QString savePath; @@ -58,7 +59,7 @@ namespace BitTorrent void create(const TorrentCreatorParams ¶ms); - static int calculateTotalPieces(const QString &inputPath, const int pieceSize); + static int calculateTotalPieces(const QString &inputPath, const int pieceSize, const bool isAlignmentOptimized); protected: void run(); diff --git a/src/gui/torrentcreatordlg.cpp b/src/gui/torrentcreatordlg.cpp index 793f31f66..fa1e8b2e2 100644 --- a/src/gui/torrentcreatordlg.cpp +++ b/src/gui/torrentcreatordlg.cpp @@ -55,6 +55,7 @@ TorrentCreatorDlg::TorrentCreatorDlg(QWidget *parent, const QString &defaultPath , m_storePrivateTorrent(SETTINGS_KEY("PrivateTorrent")) , m_storeStartSeeding(SETTINGS_KEY("StartSeeding")) , m_storeIgnoreRatio(SETTINGS_KEY("IgnoreRatio")) + , m_storeOptimizeAlignment(SETTINGS_KEY("OptimizeAlignment"), true) , m_storeLastAddPath(SETTINGS_KEY("LastAddPath"), QDir::homePath()) , m_storeTrackerList(SETTINGS_KEY("TrackerList")) , m_storeWebSeedList(SETTINGS_KEY("WebSeedList")) @@ -169,7 +170,8 @@ void TorrentCreatorDlg::onCreateButtonClicked() const QString source = m_ui->lineEditSource->text(); // run the creator thread - m_creatorThread->create({ m_ui->checkPrivate->isChecked(), getPieceSize() + m_creatorThread->create({ m_ui->checkPrivate->isChecked() + , m_ui->checkOptimizeAlignment->isChecked(), getPieceSize() , input, destination, comment, source, trackers, urlSeeds }); } @@ -213,8 +215,9 @@ void TorrentCreatorDlg::updateProgressBar(int progress) void TorrentCreatorDlg::updatePiecesCount() { const QString path = m_ui->textInputPath->text().trimmed(); + const bool isAlignmentOptimized = m_ui->checkOptimizeAlignment->isChecked(); - const int count = BitTorrent::TorrentCreatorThread::calculateTotalPieces(path, getPieceSize()); + const int count = BitTorrent::TorrentCreatorThread::calculateTotalPieces(path, getPieceSize(), isAlignmentOptimized); m_ui->labelTotalPieces->setText(QString::number(count)); } @@ -241,6 +244,7 @@ void TorrentCreatorDlg::saveSettings() m_storePrivateTorrent = m_ui->checkPrivate->isChecked(); m_storeStartSeeding = m_ui->checkStartSeeding->isChecked(); m_storeIgnoreRatio = m_ui->checkIgnoreShareLimits->isChecked(); + m_storeOptimizeAlignment = m_ui->checkOptimizeAlignment->isChecked(); m_storeTrackerList = m_ui->trackersList->toPlainText(); m_storeWebSeedList = m_ui->URLSeedsList->toPlainText(); @@ -258,6 +262,7 @@ void TorrentCreatorDlg::loadSettings() m_ui->checkPrivate->setChecked(m_storePrivateTorrent); m_ui->checkStartSeeding->setChecked(m_storeStartSeeding); m_ui->checkIgnoreShareLimits->setChecked(m_storeIgnoreRatio); + m_ui->checkOptimizeAlignment->setChecked(m_storeOptimizeAlignment); m_ui->checkIgnoreShareLimits->setEnabled(m_ui->checkStartSeeding->isChecked()); m_ui->trackersList->setPlainText(m_storeTrackerList); diff --git a/src/gui/torrentcreatordlg.h b/src/gui/torrentcreatordlg.h index 514dedf0c..48c9eacc3 100644 --- a/src/gui/torrentcreatordlg.h +++ b/src/gui/torrentcreatordlg.h @@ -80,6 +80,7 @@ private: CachedSettingValue m_storePrivateTorrent; CachedSettingValue m_storeStartSeeding; CachedSettingValue m_storeIgnoreRatio; + CachedSettingValue m_storeOptimizeAlignment; CachedSettingValue m_storeLastAddPath; CachedSettingValue m_storeTrackerList; CachedSettingValue m_storeWebSeedList; diff --git a/src/gui/torrentcreatordlg.ui b/src/gui/torrentcreatordlg.ui index 7ac6cbf59..b4238661d 100644 --- a/src/gui/torrentcreatordlg.ui +++ b/src/gui/torrentcreatordlg.ui @@ -224,6 +224,16 @@ + + + + Optimize alignment + + + true + + + @@ -329,6 +339,7 @@ checkPrivate checkStartSeeding checkIgnoreShareLimits + checkOptimizeAlignment trackersList URLSeedsList txtComment