|
|
|
/*
|
|
|
|
* (c) 2005 Alexandre Oliveira <aleprj@gmail.com>
|
|
|
|
* (c) 2003 Frerich Raabe <raabe@kde.org>
|
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#define DEBUG_PREFIX "TagGuesser"
|
|
|
|
|
|
|
|
#include "amarok.h"
|
|
|
|
#include "debug.h"
|
|
|
|
#include "tagguesser.h"
|
|
|
|
|
|
|
|
#include <kapplication.h>
|
|
|
|
#include <kconfig.h>
|
|
|
|
#include <kglobal.h>
|
|
|
|
#include <kmacroexpander.h>
|
|
|
|
|
|
|
|
FileNameScheme::FileNameScheme( const TQString &s )
|
|
|
|
: m_cod( s )
|
|
|
|
, m_titleField( -1 )
|
|
|
|
, m_artistField( -1 )
|
|
|
|
, m_albumField( -1 )
|
|
|
|
, m_trackField( -1 )
|
|
|
|
, m_commentField( -1 )
|
|
|
|
, m_yearField( -1 )
|
|
|
|
, m_composerField( -1 )
|
|
|
|
, m_genreField( -1 )
|
|
|
|
{
|
|
|
|
int artist = s.tqfind( "%artist" );
|
|
|
|
int title = s.tqfind( "%title" );
|
|
|
|
int track = s.tqfind( "%track" );
|
|
|
|
int album = s.tqfind( "%album" );
|
|
|
|
int comment = s.tqfind( "%comment" );
|
|
|
|
int year = s.tqfind( "%year" );
|
|
|
|
int composer = s.tqfind( "%composer" );
|
|
|
|
int genre = s.tqfind( "%genre" );
|
|
|
|
|
|
|
|
int fieldNumber = 1;
|
|
|
|
int i = s.tqfind( '%' );
|
|
|
|
while ( i > -1 ) {
|
|
|
|
if ( title == i )
|
|
|
|
m_titleField = fieldNumber++;
|
|
|
|
if ( artist == i )
|
|
|
|
m_artistField = fieldNumber++;
|
|
|
|
if ( album == i )
|
|
|
|
m_albumField = fieldNumber++;
|
|
|
|
if ( track == i )
|
|
|
|
m_trackField = fieldNumber++;
|
|
|
|
if ( comment == i )
|
|
|
|
m_commentField = fieldNumber++;
|
|
|
|
if ( year == i )
|
|
|
|
m_yearField = fieldNumber++;
|
|
|
|
if ( composer == i )
|
|
|
|
m_composerField = fieldNumber++;
|
|
|
|
if ( genre == i )
|
|
|
|
m_genreField = fieldNumber++;
|
|
|
|
|
|
|
|
i = s.tqfind('%', i + 1);
|
|
|
|
}
|
|
|
|
m_regExp.setPattern( composeRegExp( s ) );
|
|
|
|
}
|
|
|
|
|
|
|
|
bool FileNameScheme::matches( const TQString &fileName ) const
|
|
|
|
{
|
|
|
|
/* Strip extension ('.mp3') because '.' may be part of a title, and thus
|
|
|
|
* does not work as a separator.
|
|
|
|
*/
|
|
|
|
TQString stripped = fileName;
|
|
|
|
stripped.truncate( stripped.tqfindRev( '.' ) );
|
|
|
|
return m_regExp.exactMatch( stripped );
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::title() const
|
|
|
|
{
|
|
|
|
if( m_titleField == -1 )
|
|
|
|
return TQString();
|
|
|
|
return m_regExp.capturedTexts()[ m_titleField ];
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::artist() const
|
|
|
|
{
|
|
|
|
if( m_artistField == -1 )
|
|
|
|
return TQString();
|
|
|
|
return m_regExp.capturedTexts()[ m_artistField ];
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::album() const
|
|
|
|
{
|
|
|
|
if( m_albumField == -1 )
|
|
|
|
return TQString();
|
|
|
|
return m_regExp.capturedTexts()[ m_albumField ];
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::track() const
|
|
|
|
{
|
|
|
|
if( m_trackField == -1 )
|
|
|
|
return TQString();
|
|
|
|
return m_regExp.capturedTexts()[ m_trackField ];
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::comment() const
|
|
|
|
{
|
|
|
|
if( m_commentField == -1 )
|
|
|
|
return TQString();
|
|
|
|
return m_regExp.capturedTexts()[ m_commentField ];
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::year() const
|
|
|
|
{
|
|
|
|
if( m_yearField == -1 )
|
|
|
|
return TQString();
|
|
|
|
return m_regExp.capturedTexts()[ m_yearField ];
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::composer() const
|
|
|
|
{
|
|
|
|
if( m_composerField == -1 )
|
|
|
|
return TQString();
|
|
|
|
return m_regExp.capturedTexts()[ m_composerField ];
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::genre() const
|
|
|
|
{
|
|
|
|
if( m_genreField == -1 )
|
|
|
|
return TQString();
|
|
|
|
return m_regExp.capturedTexts()[ m_genreField ];
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString FileNameScheme::composeRegExp( const TQString &s ) const
|
|
|
|
{
|
|
|
|
TQMap<TQString, TQString> substitutions;
|
|
|
|
|
|
|
|
KConfigGroup config(KGlobal::config(), "TagGuesser");
|
|
|
|
|
|
|
|
substitutions[ "title" ] = config.readEntry( "Title regexp", "([\\w\\s'&_,\\.]+)" );
|
|
|
|
substitutions[ "artist" ] = config.readEntry( "Artist regexp", "([\\w\\s'&_,\\.]+)" );
|
|
|
|
substitutions[ "album" ] = config.readEntry( "Album regexp", "([\\w\\s'&_,\\.]+)" );
|
|
|
|
substitutions[ "track" ] = config.readEntry( "Track regexp", "(\\d+)" );
|
|
|
|
substitutions[ "comment" ] = config.readEntry( "Comment regexp", "([\\w\\s_]+)" );
|
|
|
|
substitutions[ "year" ] = config.readEntry( "Year regexp", "(\\d+)" );
|
|
|
|
substitutions[ "composer" ] = config.readEntry( "Composer regexp", "([\\w\\s'&_,\\.]+)" );
|
|
|
|
substitutions[ "genre" ] = config.readEntry( "Genre regexp", "([\\w\\s'&_,\\.]+)" );
|
|
|
|
|
|
|
|
|
|
|
|
TQString regExp = TQRegExp::escape( s.simplifyWhiteSpace() );
|
|
|
|
regExp = ".*" + regExp;
|
|
|
|
regExp.tqreplace( ' ', "\\s+" );
|
|
|
|
regExp = KMacroExpander::expandMacros( regExp, substitutions );
|
|
|
|
regExp += "[^/]*$";
|
|
|
|
return regExp;
|
|
|
|
}
|
|
|
|
|
|
|
|
TQStringList TagGuesser::schemeStrings()
|
|
|
|
{
|
|
|
|
TQStringList schemes;
|
|
|
|
|
|
|
|
schemes = Amarok::config( "TagGuesser" )->readListEntry( "Filename schemes" );
|
|
|
|
|
|
|
|
if ( schemes.isEmpty() ) {
|
|
|
|
schemes += "%track - %title";
|
|
|
|
schemes += "%artist - (%track) - %title [%comment]";
|
|
|
|
schemes += "%artist - (%track) - %title (%comment)";
|
|
|
|
schemes += "%artist - (%track) - %title";
|
|
|
|
schemes += "%artist - [%track] - %title [%comment]";
|
|
|
|
schemes += "%artist - [%track] - %title (%comment)";
|
|
|
|
schemes += "%artist - [%track] - %title";
|
|
|
|
schemes += "%artist - %track - %title [%comment]";
|
|
|
|
schemes += "%artist - %track - %title (%comment)";
|
|
|
|
schemes += "%artist - %track - %title";
|
|
|
|
schemes += "(%track) %artist - %title [%comment]";
|
|
|
|
schemes += "(%track) %artist - %title (%comment)";
|
|
|
|
schemes += "(%track) %artist - %title";
|
|
|
|
schemes += "[%track] %artist - %title [%comment]";
|
|
|
|
schemes += "[%track] %artist - %title (%comment)";
|
|
|
|
schemes += "[%track] %artist - %title";
|
|
|
|
schemes += "%track %artist - %title [%comment]";
|
|
|
|
schemes += "%track %artist - %title (%comment)";
|
|
|
|
schemes += "%track %artist - %title";
|
|
|
|
schemes += "(%artist) %title [%comment]";
|
|
|
|
schemes += "(%artist) %title (%comment)";
|
|
|
|
schemes += "(%artist) %title";
|
|
|
|
schemes += "%artist - %title [%comment]";
|
|
|
|
schemes += "%artist - %title (%comment)";
|
|
|
|
schemes += "%artist - %title";
|
|
|
|
schemes += "%artist/%album/[%track] %title [%comment]";
|
|
|
|
schemes += "%artist/%album/[%track] %title (%comment)";
|
|
|
|
schemes += "%artist/%album/[%track] %title";
|
|
|
|
}
|
|
|
|
return schemes;
|
|
|
|
}
|
|
|
|
|
|
|
|
void TagGuesser::setSchemeStrings( const TQStringList &schemes )
|
|
|
|
{
|
|
|
|
KConfig *cfg = KGlobal::config();
|
|
|
|
{
|
|
|
|
KConfigGroupSaver saver( cfg, "TagGuesser" );
|
|
|
|
cfg->writeEntry( "Filename schemes", schemes );
|
|
|
|
}
|
|
|
|
cfg->sync();
|
|
|
|
}
|
|
|
|
|
|
|
|
TagGuesser::TagGuesser()
|
|
|
|
{
|
|
|
|
loadSchemes();
|
|
|
|
}
|
|
|
|
|
|
|
|
TagGuesser::TagGuesser( const TQString &absFileName )
|
|
|
|
{
|
|
|
|
loadSchemes();
|
|
|
|
guess( absFileName );
|
|
|
|
}
|
|
|
|
|
|
|
|
void TagGuesser::loadSchemes()
|
|
|
|
{
|
|
|
|
const TQStringList schemes = schemeStrings();
|
|
|
|
TQStringList::ConstIterator it = schemes.begin();
|
|
|
|
TQStringList::ConstIterator end = schemes.end();
|
|
|
|
for ( ; it != end; ++it )
|
|
|
|
m_schemes += FileNameScheme( *it );
|
|
|
|
}
|
|
|
|
|
|
|
|
void TagGuesser::guess( const TQString &absFileName )
|
|
|
|
{
|
|
|
|
m_title = m_artist = m_album = m_track = m_comment = m_year = m_composer = m_genre = TQString();
|
|
|
|
|
|
|
|
FileNameScheme::List::ConstIterator it = m_schemes.begin();
|
|
|
|
FileNameScheme::List::ConstIterator end = m_schemes.end();
|
|
|
|
for ( ; it != end; ++it ) {
|
|
|
|
const FileNameScheme schema( *it );
|
|
|
|
if( schema.matches( absFileName ) ) {
|
|
|
|
debug() <<"Schema used: " << " " << schema.pattern() <<endl;
|
|
|
|
m_title = capitalizeWords( schema.title().tqreplace( '_', " " ) ).stripWhiteSpace();
|
|
|
|
m_artist = capitalizeWords( schema.artist().tqreplace( '_', " " ) ).stripWhiteSpace();
|
|
|
|
m_album = capitalizeWords( schema.album().tqreplace( '_', " " ) ).stripWhiteSpace();
|
|
|
|
m_track = schema.track().stripWhiteSpace();
|
|
|
|
m_comment = schema.comment().tqreplace( '_', " " ).stripWhiteSpace();
|
|
|
|
m_year = schema.year().stripWhiteSpace();
|
|
|
|
m_composer = capitalizeWords( schema.composer().tqreplace( '_', " " ) ).stripWhiteSpace();
|
|
|
|
m_genre = capitalizeWords( schema.genre().tqreplace( '_', " " ) ).stripWhiteSpace();
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
TQString TagGuesser::capitalizeWords( const TQString &s )
|
|
|
|
{
|
|
|
|
if( s.isEmpty() )
|
|
|
|
return s;
|
|
|
|
|
|
|
|
TQString result = s;
|
|
|
|
result[ 0 ] = result[ 0 ].upper();
|
|
|
|
|
|
|
|
const TQRegExp wordRegExp( "\\s\\w" );
|
|
|
|
int i = result.tqfind( wordRegExp );
|
|
|
|
while ( i > -1 ) {
|
|
|
|
result[ i + 1 ] = result[ i + 1 ].upper();
|
|
|
|
i = result.tqfind( wordRegExp, ++i );
|
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|