org.caleydo.view.domino.internal.tourguide.DataTourGuideDataMode.java Source code

Introduction

Here is the source code for org.caleydo.view.domino.internal.tourguide.DataTourGuideDataMode.java

Source

/*******************************************************************************
 * Caleydo - Visualization for Molecular Biology - http://caleydo.org
 * Copyright (c) The Caleydo Team. All rights reserved.
 * Licensed under the new BSD license, available at http://caleydo.org/license
 *******************************************************************************/
package org.caleydo.view.domino.internal.tourguide;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Set;

import org.caleydo.core.data.collection.EDataClass;
import org.caleydo.core.data.datadomain.ATableBasedDataDomain;
import org.caleydo.core.data.datadomain.DataSupportDefinitions;
import org.caleydo.core.data.datadomain.IDataDomain;
import org.caleydo.core.id.IDCategory;
import org.caleydo.core.view.opengl.layout2.renderer.GLRenderers;
import org.caleydo.view.domino.internal.tourguide.ui.EntityTypeSelector;
import org.caleydo.view.tourguide.api.adapter.ATourGuideDataMode;
import org.caleydo.view.tourguide.api.model.ADataDomainQuery;
import org.caleydo.view.tourguide.api.model.AScoreRow;
import org.caleydo.view.tourguide.api.model.InhomogenousDataDomainQuery;
import org.caleydo.vis.lineup.model.CategoricalRankColumnModel;
import org.caleydo.vis.lineup.model.IRow;
import org.caleydo.vis.lineup.model.MultiCategoricalRankColumnModel;
import org.caleydo.vis.lineup.model.RankTableModel;
import org.caleydo.vis.lineup.model.StringRankColumnModel;

import com.google.common.base.Function;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.ImmutableSortedSet;
import com.google.common.collect.Iterables;
import com.google.common.collect.Sets;

/**
 * @author Samuel Gratzl
 *
 */
public class DataTourGuideDataMode extends ATourGuideDataMode {

    @Override
    public Iterable<? extends ADataDomainQuery> createDataDomainQuery(IDataDomain dd) {
        return Collections.singleton(createFor(dd));
    }

    @Override
    public Iterable<? extends ADataDomainQuery> createDataDomainQueries() {
        return super.createDataDomainQueries(); // , createLabelQueries());
    }

    private Iterable<ADataDomainQuery> createLabelQueries() {
        List<ADataDomainQuery> r = new ArrayList<>();
        for (IDCategory cat : EntityTypeSelector.findAllUsedIDCategories()) {
            r.add(new LabelDataDomainQuery(cat));
        }
        return r;
    }

    protected ADataDomainQuery createFor(IDataDomain dd) {
        if (!DataSupportDefinitions.homogenousTables.apply(dd))
            return new InhomogenousDataDomainQuery((ATableBasedDataDomain) dd, Sets
                    .immutableEnumSet(EDataClass.NATURAL_NUMBER, EDataClass.REAL_NUMBER, EDataClass.UNIQUE_OBJECT));
        return new DataDomainQuery((ATableBasedDataDomain) dd);
    }

    @Override
    public void addDefaultColumns(RankTableModel table) {
        addDataDomainRankColumn(table);
        final StringRankColumnModel base = new StringRankColumnModel(GLRenderers.drawText("Data Set"),
                StringRankColumnModel.DEFAULT);
        table.add(base);
        base.setWidth(150);
        base.orderByMe();
        table.add(
                CategoricalRankColumnModel.createSimple(GLRenderers.drawText("Type"), new Function<IRow, String>() {
                    @Override
                    public String apply(IRow input) {
                        return input instanceof DataDomainScoreRow ? "Matrix" : "Numerical";
                    }

                }, Arrays.asList("Numerical", "Matrix")));

        ImmutableSortedSet.Builder<String> s = ImmutableSortedSet.naturalOrder();
        for (IDCategory cat : EntityTypeSelector.findAllUsedIDCategories())
            s.add(cat.getCategoryName());
        table.add(MultiCategoricalRankColumnModel
                .createSimple(GLRenderers.drawText("Item Type"), new Function<IRow, Set<String>>() {
                    @Override
                    public Set<String> apply(IRow input) {
                        if (input instanceof DataDomainScoreRow) {
                            IDCategory a = ((DataDomainScoreRow) input).getIdType().getIDCategory();
                            IDCategory b = ((DataDomainScoreRow) input).getDimensionIdType().getIDCategory();
                            return ImmutableSet.of(a.getCategoryName(), b.getCategoryName());
                        }
                        return ImmutableSet.of(((AScoreRow) input).getIdType().getIDCategory().getCategoryName());
                    }

                }, s.build(), "").setWidth(150));

        table.add(new SizeRankColumnModel("#Records", new Function<IRow, Integer>() {
            @Override
            public Integer apply(IRow in) {
                return ((AScoreRow) in).size();
            }
        }).setWidth(75));
        table.add(new SizeRankColumnModel("#Dimensions", new Function<IRow, Integer>() {
            @Override
            public Integer apply(IRow in) {
                return Iterables.size(((AScoreRow) in).getDimensionIDs());
            }
        }).setWidth(75));
    }

    @Override
    public boolean apply(IDataDomain input) {
        return input instanceof ATableBasedDataDomain;
    }

}