/*
* Copyright (c) 2008, 2009, 2010 Denis Tulskiy
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* version 3 along with this work. If not, see <http://www.gnu.org/licenses/>.
*/
package com.tulskiy.musique.gui;
import java.util.ArrayList;
import java.util.Iterator;
import javax.swing.SwingWorker;
import org.jaudiotagger.tag.FieldKey;
import com.tulskiy.musique.gui.model.FieldValues;
import com.tulskiy.musique.playlist.Playlist;
import com.tulskiy.musique.playlist.Track;
import com.tulskiy.musique.util.Util;
/**
* Author: Denis Tulskiy
* Date: 11/12/10
*/
public abstract class SearchWorker extends SwingWorker<Playlist, Integer> {
private final FieldKey[] fields = FieldKey.values();
// TODO apply correct search metas, looking through all fields at the moment
// {
// "artist", "title", "album", "albumArtist", "fileName"
// };
private Playlist playlist;
private String search;
private boolean fillEmpty;
protected ArrayList<Integer> viewToModelList = new ArrayList<Integer>();
public SearchWorker(Playlist playlist, String search, boolean fillEmpty) {
this.playlist = playlist;
this.search = search;
this.fillEmpty = fillEmpty;
}
@Override
protected Playlist doInBackground() throws Exception {
Playlist newPlaylist = new Playlist();
String str = search.toLowerCase().trim();
String[] text = str.split("\\s+");
if (!str.isEmpty() && text.length > 0) {
for (int i = 0, playlistSize = playlist.size(); i < playlistSize; i++) {
Track track = playlist.get(i);
boolean hasText[] = new boolean[text.length];
for (FieldKey field : fields) {
FieldValues values = track.getTrackData().getTagFieldValues(field);
if (values != null) {
for (int k = 0; k < values.size(); k++) {
String value = values.get(k);
if (!Util.isEmpty(value)) {
value = value.toLowerCase();
String[] vals = value.split("\\s+");
for (String val : vals) {
for (int j = 0, textLength = text.length; j < textLength; j++) {
String s = text[j];
if (val.startsWith(s)) {
hasText[j] = true;
}
}
}
}
}
}
}
boolean toAdd = true;
for (boolean b : hasText) {
toAdd &= b;
}
if (toAdd) {
newPlaylist.add(track);
viewToModelList.add(i);
}
}
} else if (fillEmpty) {
newPlaylist.addAll(playlist);
}
return newPlaylist;
}
@Override
protected abstract void done();
}