/*
* #%L
* gitools-ui-app
* %%
* Copyright (C) 2013 Universitat Pompeu Fabra - Biomedical Genomics group
* %%
* 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 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public
* License along with this program. If not, see
* <http://www.gnu.org/licenses/gpl-3.0.html>.
* #L%
*/
package org.gitools.ui.app.actions.data;
import com.google.common.collect.Sets;
import org.gitools.api.analysis.IProgressMonitor;
import org.gitools.api.components.IEditor;
import org.gitools.api.matrix.MatrixDimensionKey;
import org.gitools.api.matrix.view.IMatrixView;
import org.gitools.heatmap.Heatmap;
import org.gitools.heatmap.HeatmapDimension;
import org.gitools.matrix.filter.FilterByLabelPredicate;
import org.gitools.matrix.filter.PatternFunction;
import org.gitools.ui.app.dialog.filter.StringAnnotationsFilterPage;
import org.gitools.ui.core.Application;
import org.gitools.ui.core.actions.AbstractAction;
import org.gitools.ui.platform.progress.JobRunnable;
import org.gitools.ui.platform.progress.JobThread;
import org.gitools.ui.platform.wizard.PageDialog;
import java.awt.event.ActionEvent;
import java.util.ArrayList;
import static com.google.common.collect.Collections2.filter;
public class SortByPredefinedListAction extends AbstractAction {
private MatrixDimensionKey dimension;
public SortByPredefinedListAction(MatrixDimensionKey dimension) {
super("<html><i>Sort</i> by predefined list</html>");
setDesc("Sort by predefined list");
this.dimension = dimension;
}
@Override
public boolean isEnabledByModel(Object model) {
return model instanceof Heatmap || model instanceof IMatrixView;
}
@Override
public void actionPerformed(ActionEvent e) {
IEditor editor = Application.get().getEditorsPanel().getSelectedEditor();
Object model = editor != null ? editor.getModel() : null;
if (!(model instanceof Heatmap)) {
return;
}
final Heatmap hm = (Heatmap) model;
final StringAnnotationsFilterPage page = new StringAnnotationsFilterPage(hm, dimension);
page.setTitle("Sort by predefined list");
PageDialog dlg = new PageDialog(Application.get(), page);
dlg.open();
if (dlg.isCancelled()) {
return;
}
JobThread.execute(Application.get(), new JobRunnable() {
@Override
public void run(IProgressMonitor monitor) {
monitor.begin("Applying order ...", page.getValues().size());
HeatmapDimension heatmapDimension = hm.getDimension(page.getFilterDimension());
PatternFunction patternFunction = new PatternFunction(page.getPattern(), heatmapDimension.getAnnotations());
ArrayList<String> newOrder = new ArrayList<>();
for (String input : page.getValues()) {
for (String output : filter(heatmapDimension.toList(), new FilterByLabelPredicate(
patternFunction,
Sets.newHashSet(input),
page.isUseRegexChecked()))) {
if (!newOrder.contains(output)) {
newOrder.add(output);
}
}
monitor.worked(1);
}
heatmapDimension.show(newOrder);
}
});
Application.get().showNotification("Sorting by predefined list applied.");
}
}