plaibuildlib.cpp

Go to the documentation of this file.

// g++ -O3 recursedirs.cpp `taglib-config --cflags --libs` -lsqlite3 -o recursedirs

// artists with only one song:
// select artist as a from collection where (select count() from collection where artist==a)==1;
// maybe:
// update collection set album=artist where artist in (select artist as a from collection where (select count() from collection where artist==a)==1);
// update collection set artist="(one song)" where artist in (select artist as a from collection where (select count() from collection where artist==a)==1);
// NB this screws up names with "the " and also typos
// plus we lose the album information, which is ugly. better to have a table of
// single song artists?
// thus:
// create table onesongs(track int);
// insert into onesongs select rowid from collection where sortartist in (select sortartist as a from collection where (select count() from collection where sortartist==a)==1);

// find longest name of a certain artist
// select artist from collection where sortartist=="prodigy" order by length(artist) desc limit 1;

// and thus
// select distinct sortartist as s,artist from collection where artist in (select artist from collection where sortartist==s  order by length(artist) desc limit 1) order by sortartist
// gets a list of all sortartists and what they should be known as

// and hence
// create table artists(artist unicode,sortartist unicode)
// insert into artists select distinct artist,sortartist as s from collection where artist in (select artist from collection where sortartist==s  order by length(artist) desc limit 1) order by sortartist
// create index j on artists (artist,sortartist);
// builds the artists table

#include <iostream>
using std::string; using std::cout; using std::endl; 
#include <vector>
using std::vector;
#include <fstream>
using std::ifstream;

#include <fileref.h>
using TagLib::FileRef;
#include <tag.h>
using TagLib::AudioProperties; using TagLib::Tag;

#include "taglib_wmafiletyperesolver.h"
#include "taglib_mp4filetyperesolver.h"

#include <sqlite3.h>

#include <sys/stat.h>


vector<string> GetCommandOutput(string cmd)
{
        const string tempfname="/tmp/recursedirstemp";
        
        system((cmd+" > "+tempfname).c_str());
        
        vector<string> ret;
        
        ifstream f(tempfname.c_str());
        while(!f.eof())
        {
                string line;
                getline(f,line);
                ret.push_back(line);
        }
        f.close();
        
        return ret;
}


int GetFileModificationTime(const char* path)
{
        struct stat buf;
        stat(path,&buf);
        return buf.st_mtime;
}


class CollectionTrack;

class CollectionDB
{
public:
        CollectionDB(string dbfname);
        ~CollectionDB();
        void InsertTrack(CollectionTrack track);
        void PopulateOneSongsTable();
protected:      
        sqlite3* db;
        sqlite3_stmt* pStmt;
};


class CollectionTrack
{
public:
        CollectionTrack(string fname);
        void Print();
        friend void CollectionDB::InsertTrack(CollectionTrack);
protected:
        bool GetInfoTagLib();
        string fname,artist,album,title,sortartist;
        int length,modified;
};


CollectionTrack::CollectionTrack(string f)
{
        fname=f;
        modified=GetFileModificationTime(f.c_str());
        
        title=fname.substr(f.rfind('/')+1);
        artist=album="(unknown)";
        length=-999;
        
        GetInfoTagLib();
}


void CollectionTrack::Print()
{
        cout<<"fname - " <<fname <<endl;
        cout<<"title - " <<title <<endl;
        cout<<"artist - "<<artist<<endl;
        cout<<"album - " <<album <<endl;
        cout<<"length - "<<length<<endl;
}


bool CollectionTrack::GetInfoTagLib()
{
        FileRef f(fname.c_str(),true,AudioProperties::Fast);

        if(f.isNull()) return false;

        Tag* tag;                       
        if(tag=f.tag())
        {
                title =tag->title(). stripWhiteSpace().to8Bit();
                artist=tag->artist().stripWhiteSpace().to8Bit();
                album =tag->album(). stripWhiteSpace().to8Bit();
                sortartist=tag->artist().stripWhiteSpace().upper().to8Bit();
                if(sortartist.substr(0,4)=="THE ") sortartist=sortartist.substr(4);
        }
                
        AudioProperties* properties;
        if(properties=f.audioProperties())
        {
                length=properties->length();
        }
        
        return true;
}


CollectionDB::CollectionDB(string dbfname)
{
        sqlite3_open(dbfname.c_str(),&db);
        sqlite3_exec(db,"create table collection(fname unicode,title unicode,artist unicode,sortartist unicode,album unicode,length int,modified int)",0,0,0);          
        sqlite3_exec(db,"create index i on collection (sortartist,artist,album,title,length,modified)",0,0,0);
        sqlite3_exec(db,"create table onesongs(track int)",0,0,0);
        sqlite3_prepare_v2(db,"insert into collection values(?,?,?,?,?,?,?)",-1,&pStmt,0);
}


CollectionDB::~CollectionDB()
{
        sqlite3_finalize(pStmt);        
        sqlite3_close(db);
}


void CollectionDB::InsertTrack(CollectionTrack track)
{
        sqlite3_bind_text(pStmt,1,track. fname      .c_str(),-1,SQLITE_TRANSIENT);
        sqlite3_bind_text(pStmt,2,track. title      .c_str(),-1,SQLITE_TRANSIENT);
        sqlite3_bind_text(pStmt,3,track. artist     .c_str(),-1,SQLITE_TRANSIENT);
        sqlite3_bind_text(pStmt,4,track. sortartist .c_str(),-1,SQLITE_TRANSIENT);
        sqlite3_bind_text(pStmt,5,track. album      .c_str(),-1,SQLITE_TRANSIENT);
        sqlite3_bind_int(pStmt, 6,track. length);
        sqlite3_bind_int(pStmt, 7,track. modified);
        sqlite3_step(pStmt);
        sqlite3_reset(pStmt);
}


void CollectionDB::PopulateOneSongsTable()
{
        sqlite3_exec(db,"insert into onesongs select rowid from collection where sortartist in (select sortartist as a from collection where (select count() from collection where sortartist==a)==1)",0,0,0);
}


bool ParseCommandLine(int argc,char** argv,string& dbfname,vector<string>& folders)
{
        if(argc<3)
        {
                cout<<"Usage : "<<argv[0]<<" DATABASEFILE FOLDER(s)\n";
                return false;
        }
        dbfname=argv[1];
        for(int n=2;n<argc;++n) folders.push_back(argv[n]);
                
        return true;
}


vector<string> GetSongsInFolders(vector<string> folders)
{
        string cmd="find ";
        for(int n=0;n<folders.size();++n) cmd+="\""+folders[n]+"\" ";
        
        vector<string> fnames=GetCommandOutput(cmd);

        vector<string> ret;
        for(int n=0;n<fnames.size();++n)
        {
                if(fnames[n].size()>=3)
                {
                        const string ext=fnames[n].substr(fnames[n].size()-3);
                        if(ext=="wma" || ext=="mp3" || ext=="ogg" || ext=="mpc" || ext=="m4a")
                                ret.push_back(fnames[n]);
                }
        }
        
        return ret;
}

int main(int argc,char** argv)
{
        string dbfname;
        vector<string> folders;
        
        if(!ParseCommandLine(argc,argv,dbfname,folders)) return -1;
                
        vector<string> files=GetSongsInFolders(folders);
        
        cout<<"numfiles - "<<files.size()<<endl;
        
        FileRef::addFileTypeResolver(new WMAFileTypeResolver);
        FileRef::addFileTypeResolver(new MP4FileTypeResolver);
        
        CollectionDB dbinf(dbfname);
        
        for(int n=0;n<files.size();++n)
        {
                CollectionTrack track(files[n]);
                dbinf.InsertTrack(track);       
                track.Print();
        }
        
        cout<<"building onesongs table"<<endl;
        dbinf.PopulateOneSongsTable();
        
        cout<<"done"<<endl;
        
        return 0;
};

---