/* * * $Id: k3bmusicbrainzjob.cpp 619556 2007-01-03 17:38:12Z trueg $ * Copyright (C) 2005 Sebastian Trueg * * This file is part of the K3b project. * Copyright (C) 1998-2007 Sebastian Trueg * * 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. * See the file "COPYING" for the exact licensing terms. */ #include #ifdef HAVE_MUSICBRAINZ #include "k3bmusicbrainzjob.h" #include "k3btrm.h" #include "k3bmusicbrainz.h" #include #include #include #include #include #include #include #include // determine one trm class K3bMusicBrainzJob::TRMThread : public K3bThread { public: TRMThread(); void init() { m_canceled = false; } void run(); void cancel(); const TQCString& signature() const { return m_trm.signature(); } K3bAudioTrack* track; private: bool m_canceled; K3bTRM m_trm; }; class K3bMusicBrainzJob::MusicBrainzThread : public K3bThread { public: MusicBrainzThread() { } void run(); void setSignature( const TQCString& sig ) { m_sig = sig; } unsigned int results() { return m_results; } const TQString& title( unsigned int i = 0 ) const { return m_mb.title( i ); } const TQString& artist( unsigned int i = 0 ) const { return m_mb.artist( i ); } private: K3bMusicBrainz m_mb; int m_results; TQCString m_sig; }; K3bMusicBrainzJob::TRMThread::TRMThread() : m_canceled(false) { } void K3bMusicBrainzJob::TRMThread::run() { emitStarted(); track->seek(0); m_trm.start( track->length() ); char buffer[2352*10]; int len = 0; long long dataRead = 0; while( !m_canceled && (len = track->read( buffer, 2352*10 )) > 0 ) { dataRead += len; // swap data char buf; for( int i = 0; i < len-1; i+=2 ) { buf = buffer[i]; buffer[i] = buffer[i+1]; buffer[i+1] = buf; } if( m_trm.generate( buffer, len ) ) { len = 0; break; } // FIXME: useless since libmusicbrainz does never need all the data emitPercent( 100*dataRead/track->length().audioBytes() ); } if( m_canceled ) { emitCanceled(); emitFinished( false ); } else if( len == 0 ) { emitFinished( m_trm.finalize() ); } else emitFinished( false ); } void K3bMusicBrainzJob::TRMThread::cancel() { m_canceled = true; } void K3bMusicBrainzJob::MusicBrainzThread::run() { emitStarted(); m_results = m_mb.query( m_sig ); emitFinished( m_results > 0 ); } // cannot use this as tqparent for the K3bSimpleJobHandler since this has not been constructed yet K3bMusicBrainzJob::K3bMusicBrainzJob( TQWidget* tqparent, const char* name ) : K3bJob( new K3bSimpleJobHandler( 0 ), TQT_TQOBJECT(tqparent), name ), m_canceled( false ) { m_trmThread = new TRMThread(); m_mbThread = new MusicBrainzThread(); m_trmJob = new K3bThreadJob( m_trmThread, this, this ); m_mbJob = new K3bThreadJob( m_mbThread, this, this ); connect( m_trmJob, TQT_SIGNAL(percent(int)), this, TQT_SIGNAL(subPercent(int)) ); connect( m_trmJob, TQT_SIGNAL(percent(int)), this, TQT_SLOT(slotTrmPercent(int)) ); connect( m_trmJob, TQT_SIGNAL(finished(bool)), this, TQT_SLOT(slotTrmJobFinished(bool)) ); connect( m_mbJob, TQT_SIGNAL(finished(bool)), this, TQT_SLOT(slotMbJobFinished(bool)) ); } K3bMusicBrainzJob::~K3bMusicBrainzJob() { delete m_trmThread; delete m_trmJob; delete m_mbThread; delete m_mbJob; delete jobHandler(); } void K3bMusicBrainzJob::start() { jobStarted(); m_canceled = false; m_trmThread->track = m_tracks.first(); emit infoMessage( i18n("Generating fingerprint for track %1.") .tqarg(m_tracks.current()->trackNumber()), INFO ); m_trmJob->start(); } void K3bMusicBrainzJob::cancel() { m_canceled = true; m_trmJob->cancel(); m_mbJob->cancel(); } void K3bMusicBrainzJob::slotTrmPercent( int p ) { // the easy way (inaccurate) emit percent( (100*m_tracks.at() + p) / m_tracks.count() ); } void K3bMusicBrainzJob::slotTrmJobFinished( bool success ) { if( success ) { // now query musicbrainz m_mbThread->setSignature( m_trmThread->signature() ); emit infoMessage( i18n("Querying MusicBrainz for track %1.") .tqarg(m_tracks.current()->trackNumber()), INFO ); m_mbJob->start(); } else { if( hasBeenCanceled() ) emit canceled(); jobFinished(false); } } void K3bMusicBrainzJob::slotMbJobFinished( bool success ) { if( hasBeenCanceled() ) { emit canceled(); jobFinished(false); } else { emit trackFinished( m_tracks.current(), success ); if( success ) { // found entries TQStringList resultStrings, resultStringsUnique; for( unsigned int i = 0; i < m_mbThread->results(); ++i ) resultStrings.append( m_mbThread->artist(i) + " - " + m_mbThread->title(i) ); // since we are only using the title and the artist a lot of entries are alike to us // so to not let the user have to choose between two equal entries we trim the list down for( TQStringList::const_iterator it = resultStrings.begin(); it != resultStrings.end(); ++it ) if( resultStringsUnique.tqfind( *it ) == resultStringsUnique.end() ) resultStringsUnique.append( *it ); TQString s; bool ok = true; if( resultStringsUnique.count() > 1 ) s = KInputDialog::getItem( i18n("MusicBrainz Query"), i18n("Found multiple matches for track %1 (%2). Please select one.") .tqarg(m_tracks.current()->trackNumber()) .tqarg(m_tracks.current()->firstSource()->sourceComment()), resultStringsUnique, 0, false, &ok, dynamic_cast(tqparent()) ); else s = resultStringsUnique.first(); if( ok ) { int i = resultStrings.tqfindIndex( s ); m_tracks.current()->setTitle( m_mbThread->title(i) ); m_tracks.current()->setArtist( m_mbThread->artist(i) ); } } // query next track if( m_tracks.next() ) { emit infoMessage( i18n("Generating fingerprint for track %1.") .tqarg(m_tracks.current()->trackNumber()), INFO ); m_trmThread->track = m_tracks.current(); m_trmJob->start(); } else jobFinished( true ); } } #include "k3bmusicbrainzjob.moc" #endif