Adapt new kernel.core plugins to TMF

[deliverable/tracecompass.git] / lttng / org.lttng.scope.lttng.kernel.core / src / org / lttng / scope / lttng / kernel / core / views / timegraph / resources / ResourcesCpuIrqModelProvider.java

/*
 * Copyright (C) 2017 EfficiOS Inc., Alexandre Montplaisir <alexmonthy@efficios.com>
 *
 * All rights reserved. This program and the accompanying materials are
 * made available under the terms of the Eclipse Public License v1.0 which
 * accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 */

package org.lttng.scope.lttng.kernel.core.views.timegraph.resources;

import static java.util.Objects.requireNonNull;

import java.util.Collections;
import java.util.Comparator;
import java.util.LinkedList;
import java.util.List;
import java.util.Objects;
import java.util.function.Function;
import java.util.stream.Collectors;

import org.eclipse.tracecompass.internal.analysis.os.linux.core.kernel.Attributes;
import org.eclipse.tracecompass.statesystem.core.ITmfStateSystem;
import org.lttng.scope.lttng.kernel.core.views.timegraph.resources.elements.ResourcesCpuTreeElement;
import org.lttng.scope.lttng.kernel.core.views.timegraph.resources.elements.ResourcesIrqTreeElement;
import org.lttng.scope.lttng.kernel.core.views.timegraph.resources.elements.ResourcesIrqTreeElement.IrqType;
import org.lttng.scope.tmf2.views.core.timegraph.model.render.tree.TimeGraphTreeElement;
import org.lttng.scope.tmf2.views.core.timegraph.model.render.tree.TimeGraphTreeRender;

import com.google.common.annotations.VisibleForTesting;
import com.google.common.primitives.Ints;

/**
 * View model for a Resources view, showing CPUs as the first level, then
 * per-cpu IRQs as the second level.
 *
 * @author Alexandre Montplaisir
 */
public class ResourcesCpuIrqModelProvider extends ResourcesBaseModelProvider {

    /**
     * Each "CPU" attribute has the following children:
     *
     * <ul>
     * <li>Current_thread</li>
     * <li>Soft_IRQs</li>
     * <li>IRQs</li>
     * </ul>
     */
    private static final String[] CPUS_QUARK_PATTERN = { Attributes.CPUS, "*" }; //$NON-NLS-1$

    /**
     * Get the tree element name for every cpu.
     */
    @VisibleForTesting
    public static final Function<TreeRenderContext, TimeGraphTreeRender> SS_TO_TREE_RENDER_FUNCTION = (treeContext) -> {
        ITmfStateSystem ss = treeContext.ss;

        List<TimeGraphTreeElement> treeElems = ss.getQuarks(CPUS_QUARK_PATTERN).stream()
                .map(cpuQuark -> {
                    String cpuStr = ss.getAttributeName(cpuQuark);
                    Integer cpu = Ints.tryParse(cpuStr);
                    if (cpu == null) {
                        return null;
                    }

                    List<ResourcesIrqTreeElement> children = new LinkedList<>();

                    /* Add the "IRQ" children. */
                    int irqsQuark = ss.optQuarkRelative(cpuQuark, Attributes.IRQS);
                    for (int irqQuark : ss.getSubAttributes(irqsQuark, false)) {
                        int irqNumber = Ints.tryParse(ss.getAttributeName(irqQuark));
                        children.add(new ResourcesIrqTreeElement(IrqType.IRQ, irqNumber, irqQuark));
                    }

                    /* Add the "SoftIRQ" children. */
                    int softIrqsQuark = ss.optQuarkRelative(cpuQuark, Attributes.SOFT_IRQS);
                    for (int softIrqQuark : ss.getSubAttributes(softIrqsQuark, false)) {
                        int irqNumber = Ints.tryParse(ss.getAttributeName(softIrqQuark));
                        children.add(new ResourcesIrqTreeElement(IrqType.SOFTIRQ, irqNumber, softIrqQuark));
                    }

                    Collections.sort(children, IRQ_SORTER);
                    /* Generic types are not covariant :/ Use a raw type instead... */
                    @SuppressWarnings("rawtypes")
                    List children2 = children;
                    return new ResourcesCpuTreeElement(cpu, children2, cpuQuark);
                })
                .filter(Objects::nonNull)
                /*
                 * Sort entries according to their CPU number (not just an
                 * alphabetical sort!)
                 */
                .sorted(Comparator.comparingInt(ResourcesCpuTreeElement::getCpu))
                .collect(Collectors.toList());

        TimeGraphTreeElement rootElement = new TimeGraphTreeElement(treeContext.traceName, treeElems);
        return new TimeGraphTreeRender(rootElement);
    };

    /**
     * Constructor
     */
    public ResourcesCpuIrqModelProvider() {
        super(requireNonNull(Messages.resourcesCpuIrqProviderName), SS_TO_TREE_RENDER_FUNCTION);
    }

}