1 ///////////////////////////////////////////////////////////////////////////////
2 // LameXP - Audio Encoder Front-End
3 // Copyright (C) 2004-2014 LoRd_MuldeR <MuldeR2@GMX.de>
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; either version 2 of the License, or
8 // (at your option) any later version, but always including the *additional*
9 // restrictions defined in the "License.txt" file.
11 // This program is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 // GNU General Public License for more details.
16 // You should have received a copy of the GNU General Public License along
17 // with this program; if not, write to the Free Software Foundation, Inc.,
18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20 // http://www.gnu.org/licenses/gpl-2.0.txt
21 ///////////////////////////////////////////////////////////////////////////////
23 #include "Thread_CueSplitter.h"
27 #include "LockedFile.h"
28 #include "Model_AudioFile.h"
29 #include "Model_CueSheet.h"
30 #include "Registry_Decoder.h"
31 #include "Decoder_Abstract.h"
34 #include <MUtils/Global.h>
35 #include <MUtils/OSSupport.h>
50 ////////////////////////////////////////////////////////////
52 ////////////////////////////////////////////////////////////
54 CueSplitter::CueSplitter(const QString
&outputDir
, const QString
&baseName
, CueSheetModel
*model
, const QList
<AudioFileModel
> &inputFilesInfo
)
57 m_outputDir(outputDir
),
59 m_soxBin(lamexp_lookup_tool("sox.exe"))
61 if(m_soxBin
.isEmpty())
63 qFatal("Invalid path to SoX binary. Tool not initialized properly.");
66 m_decompressedFiles
.clear();
69 qDebug("\n[CueSplitter]");
71 int nInputFiles
= inputFilesInfo
.count();
72 for(int i
= 0; i
< nInputFiles
; i
++)
74 m_inputFilesInfo
.insert(inputFilesInfo
[i
].filePath(), inputFilesInfo
[i
]);
75 qDebug("File %02d: <%s>", i
, MUTILS_UTF8(inputFilesInfo
[i
].filePath()));
78 qDebug("All input files added.");
82 CueSplitter::~CueSplitter(void)
84 while(!m_tempFiles
.isEmpty())
86 MUtils::remove_file(m_tempFiles
.takeFirst());
90 ////////////////////////////////////////////////////////////
92 ////////////////////////////////////////////////////////////
94 void CueSplitter::run()
100 m_nTracksSkipped
= 0;
101 m_decompressedFiles
.clear();
102 m_activeFile
.clear();
104 if(!QDir(m_outputDir
).exists())
106 qWarning("Output directory \"%s\" does not exist!", MUTILS_UTF8(m_outputDir
));
110 QStringList inputFileList
= m_inputFilesInfo
.keys();
111 int nInputFiles
= inputFileList
.count();
113 emit
progressMaxChanged(nInputFiles
);
114 emit
progressValChanged(0);
116 //Decompress all input files
117 for(int i
= 0; i
< nInputFiles
; i
++)
119 const AudioFileModel_TechInfo
&inputFileInfo
= m_inputFilesInfo
[inputFileList
.at(i
)].techInfo();
120 if(inputFileInfo
.containerType().compare("Wave", Qt::CaseInsensitive
) || inputFileInfo
.audioType().compare("PCM", Qt::CaseInsensitive
))
122 AbstractDecoder
*decoder
= DecoderRegistry::lookup(inputFileInfo
.containerType(), inputFileInfo
.containerProfile(), inputFileInfo
.audioType(), inputFileInfo
.audioProfile(), inputFileInfo
.audioVersion());
125 m_activeFile
= shortName(QFileInfo(inputFileList
.at(i
)).fileName());
127 emit
fileSelected(m_activeFile
);
128 emit
progressValChanged(i
+1);
130 QString tempFile
= QString("%1/~%2.wav").arg(m_outputDir
, MUtils::rand_str());
131 connect(decoder
, SIGNAL(statusUpdated(int)), this, SLOT(handleUpdate(int)), Qt::DirectConnection
);
133 if(decoder
->decode(inputFileList
.at(i
), tempFile
, &m_abortFlag
))
135 m_decompressedFiles
.insert(inputFileList
.at(i
), tempFile
);
136 m_tempFiles
.append(tempFile
);
140 qWarning("Failed to decompress file: <%s>", inputFileList
.at(i
).toLatin1().constData());
141 MUtils::remove_file(tempFile
);
144 m_activeFile
.clear();
145 MUTILS_DELETE(decoder
);
149 qWarning("Unsupported input file: <%s>", inputFileList
.at(i
).toLatin1().constData());
154 m_decompressedFiles
.insert(inputFileList
.at(i
), inputFileList
.at(i
));
160 qWarning("The user has requested to abort the process!");
165 int nFiles
= m_model
->getFileCount();
166 int nTracksTotal
= 0, nTracksComplete
= 0;
168 for(int i
= 0; i
< nFiles
; i
++)
170 nTracksTotal
+= m_model
->getTrackCount(i
);
173 emit
progressMaxChanged(10 * nTracksTotal
);
174 emit
progressValChanged(0);
176 const AudioFileModel_MetaInfo
*albumInfo
= m_model
->getAlbumInfo();
178 //Now split all files
179 for(int i
= 0; i
< nFiles
; i
++)
181 int nTracks
= m_model
->getTrackCount(i
);
182 QString trackFile
= m_model
->getFileName(i
);
185 for(int j
= 0; j
< nTracks
; j
++)
187 const AudioFileModel_MetaInfo
*trackInfo
= m_model
->getTrackInfo(i
, j
);
188 const int trackNo
= trackInfo
->position();
189 double trackOffset
= std::numeric_limits
<double>::quiet_NaN();
190 double trackLength
= std::numeric_limits
<double>::quiet_NaN();
191 m_model
->getTrackIndex(i
, j
, &trackOffset
, &trackLength
);
193 if((trackNo
< 0) || _isnan(trackOffset
) || _isnan(trackLength
))
195 qWarning("Failed to fetch information for track #%d of file #%d!", j
, i
);
200 AudioFileModel_MetaInfo
trackMetaInfo(*trackInfo
);
202 //Apply album meta data on files
203 if(trackMetaInfo
.title().trimmed().isEmpty())
205 trackMetaInfo
.setTitle(QString().sprintf("Track %02d", trackNo
));
207 trackMetaInfo
.update(*albumInfo
, false);
209 //Generate output file name
210 QString trackTitle
= trackMetaInfo
.title().isEmpty() ? QString().sprintf("Track %02d", trackNo
) : trackMetaInfo
.title();
211 QString outputFile
= QString("%1/[%2] %3 - %4.wav").arg(m_outputDir
, QString().sprintf("%02d", trackNo
), MUtils::clean_file_name(m_baseName
), MUtils::clean_file_name(trackTitle
));
212 for(int n
= 2; QFileInfo(outputFile
).exists(); n
++)
214 outputFile
= QString("%1/[%2] %3 - %4 (%5).wav").arg(m_outputDir
, QString().sprintf("%02d", trackNo
), MUtils::clean_file_name(m_baseName
), MUtils::clean_file_name(trackTitle
), QString::number(n
));
217 //Call split function
218 emit
fileSelected(shortName(QFileInfo(outputFile
).fileName()));
219 splitFile(outputFile
, trackNo
, trackFile
, trackOffset
, trackLength
, trackMetaInfo
, nTracksComplete
);
220 emit
progressValChanged(nTracksComplete
+= 10);
225 qWarning("The user has requested to abort the process!");
231 emit
progressValChanged(10 * nTracksTotal
);
232 MUtils::OS::sleep_ms(333);
234 qDebug("All files were split.\n");
238 ////////////////////////////////////////////////////////////
240 ////////////////////////////////////////////////////////////
242 void CueSplitter::handleUpdate(int progress
)
244 //QString("%1 [%2]").arg(m_activeFile, QString::number(progress)))
247 ////////////////////////////////////////////////////////////
249 ////////////////////////////////////////////////////////////
251 void CueSplitter::splitFile(const QString
&output
, const int trackNo
, const QString
&file
, const double offset
, const double length
, const AudioFileModel_MetaInfo
&metaInfo
, const int baseProgress
)
253 qDebug("[Track %02d]", trackNo
);
254 qDebug("File: <%s>", MUTILS_UTF8(file
));
255 qDebug("Offset: <%f> <%s>", offset
, indexToString(offset
).toLatin1().constData());
256 qDebug("Length: <%f> <%s>", length
, indexToString(length
).toLatin1().constData());
257 qDebug("Artist: <%s>", MUTILS_UTF8(metaInfo
.artist()));
258 qDebug("Title: <%s>", MUTILS_UTF8(metaInfo
.title()));
259 qDebug("Album: <%s>", MUTILS_UTF8(metaInfo
.album()));
261 int prevProgress
= baseProgress
;
263 if(!m_decompressedFiles
.contains(file
))
265 qWarning("Unknown or unsupported input file, skipping!");
270 QString baseName
= shortName(QFileInfo(output
).fileName());
271 QString decompressedInput
= m_decompressedFiles
[file
];
272 qDebug("Input: <%s>", MUTILS_UTF8(decompressedInput
));
274 AudioFileModel
outFileInfo(output
);
275 outFileInfo
.setMetaInfo(metaInfo
);
277 AudioFileModel_TechInfo
&outFileTechInfo
= outFileInfo
.techInfo();
278 outFileTechInfo
.setContainerType("Wave");
279 outFileTechInfo
.setAudioType("PCM");
280 outFileTechInfo
.setDuration(static_cast<unsigned int>(abs(length
)));
283 args
<< "-S" << "-V3";
284 args
<< "--guard" << "--temp" << ".";
285 args
<< QDir::toNativeSeparators(decompressedInput
);
286 args
<< QDir::toNativeSeparators(output
);
288 //Add trim parameters, if needed
292 args
<< indexToString(offset
);
296 args
<< indexToString(length
);
300 QRegExp
rxProgress("In:(\\d+)(\\.\\d+)*%", Qt::CaseInsensitive
);
301 QRegExp
rxChannels("Channels\\s*:\\s*(\\d+)", Qt::CaseInsensitive
);
302 QRegExp
rxSamplerate("Sample Rate\\s*:\\s*(\\d+)", Qt::CaseInsensitive
);
303 QRegExp
rxPrecision("Precision\\s*:\\s*(\\d+)-bit", Qt::CaseInsensitive
);
304 QRegExp
rxDuration("Duration\\s*:\\s*(\\d\\d):(\\d\\d):(\\d\\d).(\\d\\d)", Qt::CaseInsensitive
);
307 MUtils::init_process(process
, m_outputDir
);
309 process
.start(m_soxBin
, args
);
311 if(!process
.waitForStarted())
313 qWarning("SoX process failed to create!");
314 qWarning("Error message: \"%s\"\n", process
.errorString().toLatin1().constData());
316 process
.waitForFinished(-1);
321 while(process
.state() != QProcess::NotRunning
)
326 qWarning("Process was aborted on user request!");
329 process
.waitForReadyRead(m_processTimeoutInterval
);
330 if(!process
.bytesAvailable() && process
.state() == QProcess::Running
)
333 qWarning("SoX process timed out <-- killing!");
336 while(process
.bytesAvailable() > 0)
338 QByteArray line
= process
.readLine();
339 QString text
= QString::fromUtf8(line
.constData()).simplified();
340 if(rxProgress
.lastIndexIn(text
) >= 0)
343 int progress
= rxProgress
.cap(1).toInt(&ok
);
346 const int newProgress
= baseProgress
+ qRound(static_cast<double>(qBound(0, progress
, 100)) / 10.0);
347 if(newProgress
> prevProgress
)
349 emit
progressValChanged(newProgress
);
350 prevProgress
= newProgress
;
354 else if(rxChannels
.lastIndexIn(text
) >= 0)
357 unsigned int channels
= rxChannels
.cap(1).toUInt(&ok
);
358 if(ok
) outFileInfo
.techInfo().setAudioChannels(channels
);
360 else if(rxSamplerate
.lastIndexIn(text
) >= 0)
363 unsigned int samplerate
= rxSamplerate
.cap(1).toUInt(&ok
);
364 if(ok
) outFileInfo
.techInfo().setAudioSamplerate(samplerate
);
366 else if(rxPrecision
.lastIndexIn(text
) >= 0)
369 unsigned int precision
= rxPrecision
.cap(1).toUInt(&ok
);
370 if(ok
) outFileInfo
.techInfo().setAudioBitdepth(precision
);
372 else if(rxDuration
.lastIndexIn(text
) >= 0)
374 bool ok1
= false, ok2
= false, ok3
= false;
375 unsigned int hh
= rxDuration
.cap(1).toUInt(&ok1
);
376 unsigned int mm
= rxDuration
.cap(2).toUInt(&ok2
);
377 unsigned int ss
= rxDuration
.cap(3).toUInt(&ok3
);
378 if(ok1
&& ok2
&& ok3
)
380 unsigned intputLen
= (hh
* 3600) + (mm
* 60) + ss
;
381 if(length
== std::numeric_limits
<double>::infinity())
383 qDebug("Duration updated from SoX info!");
384 int duration
= intputLen
- static_cast<int>(floor(offset
+ 0.5));
385 if(duration
< 0) qWarning("Track is out of bounds: Track offset exceeds input file duration!");
386 outFileInfo
.techInfo().setDuration(qMax(0, duration
));
390 unsigned int trackEnd
= static_cast<unsigned int>(floor(offset
+ 0.5)) + static_cast<unsigned int>(floor(length
+ 0.5));
391 if(trackEnd
> intputLen
) qWarning("Track is out of bounds: End of track exceeds input file duration!");
398 process
.waitForFinished();
399 if(process
.state() != QProcess::NotRunning
)
402 process
.waitForFinished(-1);
405 if(process
.exitCode() != EXIT_SUCCESS
|| QFileInfo(output
).size() == 0)
407 qWarning("Splitting has failed !!!");
412 emit
fileSplit(outFileInfo
);
416 QString
CueSplitter::indexToString(const double index
) const
418 if(!_finite(index
) || (index
< 0.0) || (index
> 86400.0))
423 QTime time
= QTime().addMSecs(static_cast<int>(floor(0.5 + (index
* 1000.0))));
424 return time
.toString(time
.hour() ? "H:mm:ss.zzz" : "m:ss.zzz");
427 QString
CueSplitter::shortName(const QString
&longName
) const
429 static const int maxLen
= 54;
431 if(longName
.length() > maxLen
)
433 return QString("%1...%2").arg(longName
.left(maxLen
/2).trimmed(), longName
.right(maxLen
/2).trimmed());
439 ////////////////////////////////////////////////////////////
441 ////////////////////////////////////////////////////////////