<NME> MediaSourceFactory.java
<BEF> /*
 * Copyright @ 2015 - Present, 8x8 Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.jitsi.videobridge.xmpp;

import org.jitsi.nlj.*;
import org.jitsi.nlj.rtp.*;
import org.jitsi.utils.logging2.*;
import org.jitsi.xmpp.extensions.colibri.*;
import org.jitsi.xmpp.extensions.jingle.*;
import org.jitsi.xmpp.extensions.jitsimeet.*;

import java.util.*;
import java.util.stream.*;
import java.util.Objects;
import java.util.stream.*;

/**
 * A factory of {@link MediaSourceDesc}s from jingle signaling.
 *
 * @author George Politis
 */
public class MediaSourceFactory
{
    /**
     * The {@link Logger} used by the {@link MediaSourceDesc} class and its
     * instances for logging output.
     */
    private static final Logger logger
        = new LoggerImpl(MediaSourceFactory.class.getName());

    /**
     * The default number of temporal layers to use for VP8 simulcast.
     *
     * FIXME: hardcoded ugh.. this should be either signaled or somehow included
     * in the RTP stream.
     */
    private static final int VP8_SIMULCAST_TEMPORAL_LAYERS = 3;

    /**
     * The resolution of the base stream when activating simulcast for VP8.
     *
     * FIXME: hardcoded ugh.. this should be either signaled or somehow included
     * in the RTP stream.
     */
    private static final int VP8_SIMULCAST_BASE_LAYER_HEIGHT = 180;

    /**
     * libjitsi isn't aware of the group semantics names defined in
     * {@link SourceGroupPacketExtension}, which is how we distinguish secondary
     * ssrcs, so we'll translate them into constants defined in libjitsi
     */
    private static Map<String, SsrcAssociationType> secondarySsrcTypeMap = null;

    private static synchronized
        Map<String, SsrcAssociationType> getSecondarySsrcTypeMap()
    {
        if (secondarySsrcTypeMap == null)
        {
            secondarySsrcTypeMap = new HashMap<>();
            secondarySsrcTypeMap.put(
                SourceGroupPacketExtension.SEMANTICS_FID, SsrcAssociationType.RTX);
        }

        return secondarySsrcTypeMap;
    }

    private static final RtpLayerDesc[] noDependencies = new RtpLayerDesc[0];

    /*
     * Creates layers for an encoding.
     *
     * @param spatialLen the number of spatial encodings per simulcast stream.
     * @param temporalLen the number of temporal encodings per simulcast stream.
     * @param height the maximum height of the top spatial layer
     * @return an array that holds the layer descriptions.
     */
    private static RtpLayerDesc[] createRTPLayerDescs(
        int spatialLen, int temporalLen, int encodingIdx, int height)
    {
        RtpLayerDesc[] rtpLayers
            = new RtpLayerDesc[spatialLen * temporalLen];

        for (int spatialIdx = 0; spatialIdx < spatialLen; spatialIdx++)
        {
            double frameRate = (double) 30 / (1 << (temporalLen - 1));
            for (int temporalIdx = 0;
                 temporalIdx < temporalLen; temporalIdx++)
            {
                int idx = idx(spatialIdx, temporalIdx,
                    temporalLen);

                RtpLayerDesc[] dependencies;
                if (spatialIdx > 0 && temporalIdx > 0)
                {
                    // this layer depends on spatialIdx-1 and temporalIdx-1.
                    dependencies = new RtpLayerDesc[]{
                        rtpLayers[
                            idx(spatialIdx, temporalIdx - 1,
                                temporalLen)],
                        rtpLayers[
                            idx(spatialIdx - 1, temporalIdx,
                                temporalLen)]
                    };
                }
                else if (spatialIdx > 0)
                {
                    // this layer depends on spatialIdx-1.
                    dependencies = new RtpLayerDesc[]
                        {rtpLayers[
                            idx(spatialIdx - 1, temporalIdx,
                                temporalLen)]};
                }
                else if (temporalIdx > 0)
                {
                    // this layer depends on temporalIdx-1.
                    dependencies = new RtpLayerDesc[]
                        {rtpLayers[
                            idx(spatialIdx, temporalIdx - 1,
                                temporalLen)]};
                }
                else
                {
                    // this is a base layer without any dependencies.
                    dependencies = noDependencies;
                }

                int temporalId = temporalLen > 1 ? temporalIdx : -1;
                int spatialId = spatialLen > 1 ? spatialIdx : -1;

                rtpLayers[idx]
                    = new RtpLayerDesc(encodingIdx,
                    temporalId, spatialId, height, frameRate, dependencies);

                frameRate *= 2;
            }


        }
        return rtpLayers;
    }


    /**
     * Creates an RTP encoding.
     * @param primarySsrc the primary SSRC for the encoding.
     * @param spatialLen the number of spatial layers of the encoding.
     * @param temporalLen the number of temporal layers of the encodings.
     * @param secondarySsrcs a list of pairs, where each
     * pair has the secondary ssrc as its key, and the type (rtx, etc.) as its
     * value
     * @param encodingIdx the index of the encoding
     * @return a description of the encoding.
     */
    private static RtpEncodingDesc createRtpEncodingDesc(Long primarySsrc,
        int spatialLen, int temporalLen, SecondarySsrcs secondarySsrcs,
        int encodingIdx, int height)
    {
        RtpLayerDesc[] layers = createRTPLayerDescs(spatialLen, temporalLen,
            encodingIdx, height);

        RtpEncodingDesc enc = new RtpEncodingDesc(primarySsrc, layers, encodingIdx);

        if (secondarySsrcs != null)
        {
            secondarySsrcs.forEach(ssrcSecondarySsrc -> {
                SsrcAssociationType type
                    = getSecondarySsrcTypeMap()
                    .get(ssrcSecondarySsrc.type);
                if (type == null)
                {
                    logger.error("Unable to find a mapping for" +
                        " secondary ssrc type " +
                        ssrcSecondarySsrc.type +
                        " will NOT included this secondary ssrc as" +
                        " an encoding");
                }
                else
                {
                    enc.addSecondarySsrc(
                        ssrcSecondarySsrc.ssrc, type);
                }
            });
        }

        return enc;
    }


    /**
     * Get the 'secondary' ssrcs for the given primary ssrc. 'Secondary' here
     * is defined as things like rtx or fec ssrcs.
     * @param ssrc the primary ssrc for which to get the secondary ssrcs
     * @param sourceGroups the source groups
     * @return a map of secondary ssrc -> type (rtx, fec, etc.)
     */
    private static List<SecondarySsrc> getSecondarySsrcs(
        long ssrc, Collection<SourceGroupPacketExtension> sourceGroups)
    {
        List<SecondarySsrc> secondarySsrcs = new ArrayList<>();
        for (SourceGroupPacketExtension sourceGroup : sourceGroups)
        {
            if (sourceGroup.getSemantics().equalsIgnoreCase(
                SourceGroupPacketExtension.SEMANTICS_SIMULCAST))
            {
                // Simulcast does not fall under the definition of 'secondary'
                // we want here.
                continue;
            }
            long groupPrimarySsrc = sourceGroup.getSources().get(0).getSSRC();
            long groupSecondarySsrc = sourceGroup.getSources().get(1).getSSRC();
            if (groupPrimarySsrc == ssrc)
            {
                secondarySsrcs.add(
                    new SecondarySsrc(
                        groupSecondarySsrc, sourceGroup.getSemantics()));
            }
        }
        return secondarySsrcs;
    }

    /**
     * Build a map of source ssrc -> a list of secondary ssrc, secondary ssrc
     * type
     * @param ssrcs the ssrcs to get secondary ssrcs for
     * @param sourceGroups the signaled source groups
     * @return map of source ssrc -> a list of secondary ssrc, secondary ssrc
     * type
     */
    private static Map<Long, SecondarySsrcs> getAllSecondarySsrcs(
            SourceSsrcs ssrcs, Collection<SourceGroupPacketExtension> sourceGroups)
    {
        Map<Long, SecondarySsrcs> allSecondarySsrcs = new HashMap<>();

        for (long ssrc : ssrcs)
        {
            List<SecondarySsrc> secondarySsrcs
                = getSecondarySsrcs(ssrc, sourceGroups);
            allSecondarySsrcs.put(ssrc, new SecondarySsrcs(secondarySsrcs));
        }
        return allSecondarySsrcs;
    }

    /**
     * Get all groups which have the given semantics
     * @param semantics
     * @param groups
     * @return
     */
    private static List<SourceGroupPacketExtension> getGroups(
        final String semantics, final List<SourceGroupPacketExtension> groups)
    {
        return groups.stream()
            .filter(sg -> sg.getSemantics().equalsIgnoreCase(semantics))
            .collect(Collectors.toList());
    }

    /**
     * Removes all sources which correspond to an ssrc in sourceSsrcs, or any
     * groups with a primary ssrc in sourceSsrcs.  NOTE: modifies
     * sources and sourceGroups in place
     * @param sourceSsrcs the primary ssrcs for which all corresponding ssrcs
     * will be removed from the given sources and groups
     * @param sources the set of sources
     * @param sourceGroups the set of groups
     */
    private static void removeReferences(
            SourceSsrcs sourceSsrcs,
            List<SourcePacketExtension> sources,
            List<SourceGroupPacketExtension> sourceGroups)
    {
        // Remove any groups to which any of the ssrcs of this source belong
        List<SourceGroupPacketExtension> groupsToRemove
            = sourceGroups.stream()
                .filter(
                    group -> group.getSources().stream().anyMatch(
                        source -> sourceSsrcs.contains(source.getSSRC())))
                .collect(Collectors.toList());

        sourceGroups.removeAll(groupsToRemove);

        /*
         * Remove not only the ssrcs in the source itself, but any ssrcs that
         * were in groups along with ssrcs from the source. E.g. if we have:`
         * SIM 1 2 3
         * RTX 1 10
         * RTX 2 20
         * RTX 3 30
         * then we need to make sure ssrcs 10, 20 and 30 don't create sources of
         * their own and are removed along with the processing of the source
         * with ssrcs 1, 2 and 3.
         */
        Set<Long> ssrcsToRemove = extractSsrcs(groupsToRemove);
        sources.removeIf(
            source ->
                sourceSsrcs.contains(source.getSSRC())
                    || ssrcsToRemove.contains(source.getSSRC()));
    }

    /**
     * Extracts all SSRCs from all sources of a list of source groups.
     * @param groups the list of groups.
     * @return the set of SSRCs contained in one of the groups.
     */
    private static Set<Long> extractSsrcs(
        List<SourceGroupPacketExtension> groups)
    {
        Set<Long> ssrcs = new HashSet<>();
        groups.forEach(
            group -> group.getSources().forEach(
                source -> ssrcs.add(source.getSSRC())));
        return ssrcs;
    };

    /**
     * Given the sources and groups, return a list of the ssrcs for each
     * unique source
     * @param sources the set of sources
     * @param sourceGroups the set of source groups
     * @return a list of {@link SourceSsrcs}. Each SourceSsrc object represents
     * a set of primary video ssrcs belonging to a single source (video source)
     */
    private static List<SourceSsrcs> getSourceSsrcs(
            Collection<SourcePacketExtension> sources,
            Collection<SourceGroupPacketExtension> sourceGroups
    )
    {
        //FIXME: determining the individual sources should be done via msid,
        // but somewhere along the line we seem to lose the msid information
        // in the packet extensions
        List<SourceSsrcs> sourceSsrcsList = new ArrayList<>();
        // We'll need to keep track of which sources and groups have been
        // processed, so make copies of the lists we've been given so we can
        // modify them.
        List<SourcePacketExtension> sourcesCopy = new ArrayList<>(sources);
        List<SourceGroupPacketExtension> sourceGroupsCopy
            = new ArrayList<>(sourceGroups);

        // Note that the order we process these groups is important
        Arrays.asList(
            SourceGroupPacketExtension.SEMANTICS_SIMULCAST,
            SourceGroupPacketExtension.SEMANTICS_FID,
            SourceGroupPacketExtension.SEMANTICS_FEC
        ).forEach(groupSem -> {
            List<SourceGroupPacketExtension> groups
                = getGroups(groupSem, sourceGroupsCopy);
            groups.forEach(group -> {
                // An empty group is the signal that we want to clear all
                // the groups.
                // https://github.com/jitsi/jitsi/blob/7eabaab0fca37711813965d66a0720d1545f6c48/src/net/java/sip/communicator/impl/protocol/jabber/extensions/colibri/ColibriBuilder.java#L188
                if (group.getSources() == null || group.getSources().isEmpty())
                {
                    if (groups.size() > 1)
                    {
                        logger.warn("Received empty group, which is " +
                            "a signal to clear all groups, but there were " +
                            "other groups present, which shouldn't happen");
                    }
                    return;
                }
                List<Long> ssrcs;
                // For a simulcast group, all the ssrcs are considered primary
                // ssrcs, but for others, only the main ssrc of the group is
                if (groupSem.equalsIgnoreCase(
                    SourceGroupPacketExtension.SEMANTICS_SIMULCAST))
                {
                    ssrcs
                        = group.getSources().stream()
                            .map(SourcePacketExtension::getSSRC)
                            .collect(Collectors.toList());
                }
                else
                {
                    ssrcs = Arrays.asList(group.getSources().get(0).getSSRC());
                }

                SourceSsrcs sourceSsrcs = new SourceSsrcs(ssrcs);
                // Now we need to remove any groups with these ssrcs as their
                // primary, or sources that correspond to one of these ssrcs
                removeReferences(sourceSsrcs, sourcesCopy, sourceGroupsCopy);

                sourceSsrcsList.add(sourceSsrcs);
            });
        });

        if (!sourceGroupsCopy.isEmpty())
        {
            logger.warn(
                "Unprocessed source groups: " +
                    sourceGroupsCopy.stream()
                        .map(e -> e.toXML(XmlEnvironment.EMPTY))
                        .reduce(String::concat));
        }

        // The remaining sources are not part of any group. Add them as sources
        // with their own primary SSRC.
        // NOTE: we need to ignore sources with an ssrc of -1, because the
        // ColibriBuilder will use that as a signal to remove sources
        // https://github.com/jitsi/jitsi/blob/7eabaab0fca37711813965d66a0720d1545f6c48/src/net/java/sip/communicator/impl/protocol/jabber/extensions/colibri/ColibriBuilder.java#L162
        sourcesCopy.forEach(source -> {
            if (source.getSSRC() != -1)
            {
                sourceSsrcsList.add(new SourceSsrcs(source.getSSRC()));
            }
            else
            {
                if (sourcesCopy.size() > 1)
                {
                    logger.warn("Received an empty source, which is " +
                        "a signal to clear all sources, but there were " +
                        "other sources present, which shouldn't happen");
                }
            }
        });

        setOwnersAndNames(sources, sourceSsrcsList);

        return sourceSsrcsList;
    }

    /**
     * Updates the given list of {@link SourceSsrcs}, setting the {@code owner}
     * field according to the {@code owner} attribute in {@code ssrc-info}
     * extensions contained in {@code sources}, if not specified externally
     * @param sources the list of {@link SourcePacketExtension} which may contain
     * the {@code owner} as an attribute of a {@code ssrc-info} child. The
     * list or the objects in the list will not be modified.
     * @param sourceSsrcsList the list of {@link SourceSsrcs} to update.
     */
    private static void setOwnersAndNames(
        Collection<SourcePacketExtension> sources,
        Collection<SourceSsrcs> sourceSsrcsList
    )
    {
        for (SourceSsrcs sourceSsrcs : sourceSsrcsList)
        {
            // Look for the "owner" tag in the sources. We assume that all
            // sources contain the "owner" tag so we just check for the
            // first SSRC of the source's SSRCs.
            long primarySsrc = sourceSsrcs.get(0);
            SourcePacketExtension sourceSource
                = sources.stream()
                    .filter(source -> source.getSSRC() == primarySsrc)
                    .findAny().orElse(null);

            sourceSsrcs.owner = getOwner(sourceSource);
            sourceSsrcs.name = sourceSource != null ? sourceSource.getName() : null;
        }
    }

    /**
     * Extracts the owner/endpoint ID as a {@link String} from a
     * {@link SourcePacketExtension}.
     * Jicofo includes the full occupant JID of the endpoint as the owner of
     * a {@link SSRCInfoPacketExtension}, but in jitsi-videobridge we only
     * care about the resource part, which coincides with the ID of the Colibri
     * endpoint associated with the owner.
            = source == null
                ? null : source.getFirstChildOfType(
                    SSRCInfoPacketExtension.class);
        if (ssrcInfoPacketExtension != null)
        {
            return
                ssrcInfoPacketExtension.getOwner()
                    .getResourceOrEmpty().toString();
        }
        return null;
    }

    /**
     * Creates {@link MediaSourceDesc}s from signaling params
     *
    }

    /**
     * Creates {@link MediaSourceDesc}s from signaling params
     *
     * will receive the created {@link MediaSourceDesc}s.
     * @param sources The {@link List} of {@link SourcePacketExtension} that
     * describes the list of jingle sources.
     * @param sourceGroups The {@link List} of
     * {@link SourceGroupPacketExtension} that describes the list of jingle
     * source groups.
     * @return an array of {@link MediaSourceDesc} that are described in the
     * jingle sources and source groups.
     * @deprecated Use createMediaSource.
     */
    @Deprecated
    public static MediaSourceDesc[] createMediaSources(
        Collection<SourcePacketExtension> sources,
        Collection<SourceGroupPacketExtension> sourceGroups)
    {
        final Collection<SourceGroupPacketExtension> finalSourceGroups
                = sourceGroups == null ? new ArrayList<>() : sourceGroups;
        if (sources == null)
        {
            sources = new ArrayList<>();
        }

        List<SourceSsrcs> sourceSsrcsList = getSourceSsrcs(sources, finalSourceGroups);
        List<MediaSourceDesc> mediaSources = new ArrayList<>();

        sourceSsrcsList.forEach(sourceSsrcs -> {
            // As of now, we only ever have 1 spatial layer per stream
            int numSpatialLayersPerStream = 1;
            // Previously, we assumed that when simulcast was enabled, 3 temporal layers would
            // be present in each stream and if simulcast wasn't enabled, there would be only
            // 1 temporal layer per stream, but we found that screenshare started sending 2
            // temporal layers, and we'd therefore start dropping layer 1 packets since we didn't
            // recognize them.  The truth is we have no way of _knowing_ how many temporal layers
            // there are going to be (as it's not signaled) unless we dynamically read the layers
            // as the packets come in (which we may want to do in the future)--but for now we'll
            // over-estimate how many layers there are per stream to avoid dropping packets for a
            // layer we don't recognize.  Since there are already situations where chrome sends fewer
            // temporal layers than we assume, this should be safe.
            final int numTemporalLayersPerStream = 3;
            Map<Long, SecondarySsrcs> secondarySsrcs
                = getAllSecondarySsrcs(sourceSsrcs, finalSourceGroups);
            MediaSourceDesc mediaSource
                = createSource(
                        sourceSsrcs,
                        numSpatialLayersPerStream,
                        numTemporalLayersPerStream,
                        secondarySsrcs,
                        sourceSsrcs.owner,
                        sourceSsrcs.name
            );
            mediaSources.add(mediaSource);
        });

        return mediaSources.toArray(new MediaSourceDesc[0]);
    }

    // This method is to replace createMediaSources when Colibri V1 is removed
    public static MediaSourceDesc createMediaSource(
            Collection<SourcePacketExtension> sources,
            Collection<SourceGroupPacketExtension> sourceGroups,
            String owner,
            String name)
    {
        Objects.requireNonNull(owner, "owner is required");
        Objects.requireNonNull(name, "name is required");

        final Collection<SourceGroupPacketExtension> finalSourceGroups
            = sourceGroups == null ? new ArrayList<>() : sourceGroups;
        if (sources == null)
        {
            return null;
        }

        List<SourceSsrcs> sourceSsrcsList = getSourceSsrcs(sources, finalSourceGroups);

        if (sourceSsrcsList.size() != 1)
        {
            logger.warn("sourceSsrcsList.size() != 1 for: " + sources + " groups: " + finalSourceGroups);
        }

        if (sourceSsrcsList.size() > 0)
        {
            SourceSsrcs sourceSsrcs = sourceSsrcsList.get(0);

            // As of now, we only ever have 1 spatial layer per stream
            int numSpatialLayersPerStream = 1;
            // Previously, we assumed that when simulcast was enabled, 3 temporal layers would
            // be present in each stream and if simulcast wasn't enabled, there would be only
            // 1 temporal layer per stream, but we found that screenshare started sending 2
            // temporal layers, and we'd therefore start dropping layer 1 packets since we didn't
            // recognize them.  The truth is we have no way of _knowing_ how many temporal layers
            // there are going to be (as it's not signaled) unless we dynamically read the layers
            // as the packets come in (which we may want to do in the future)--but for now we'll
            // over-estimate how many layers there are per stream to avoid dropping packets for a
            // layer we don't recognize.  Since there are already situations where chrome sends fewer
            // temporal layers than we assume, this should be safe.
            final int numTemporalLayersPerStream = 3;
            Map<Long, SecondarySsrcs> secondarySsrcs = getAllSecondarySsrcs(sourceSsrcs, finalSourceGroups);

            return createSource(
                    sourceSsrcs,
                    numSpatialLayersPerStream,
                    numTemporalLayersPerStream,
                    secondarySsrcs,
                    owner,
                    name
            );
        }
        else
        {
            return null;
        }
    }


    /**
     * Calculates the array position of an RTP layer description specified by its
     * spatial index (SVC) and temporal index (SVC).
     *
     * @param spatialIdx the spatial layer index.
     * @param temporalIdx the temporal layer index.
     *
     * @return the subjective quality index of the flow specified in the
     * arguments.
     */
    private static int idx(int spatialIdx, int temporalIdx,
        int temporalLen)
    {
        return spatialIdx * temporalLen + temporalIdx;
    }

    /**
     * Describes a single secondary ssrc and its group semantics to
     * its primary ssrc (e.g. FID or FEC-FR)
     */
    private static class SecondarySsrc
    {
        public long ssrc;
        public String type;

        /**
         * Initializes a new {@link SecondarySsrc} instance.
         */
        public SecondarySsrc(long ssrc, String type)
        {
            this.ssrc = ssrc;
            this.type = type;
        }
    }

    /**
     * Groups a set of secondary ssrcs for a single primary ssrc (e.g. its
     * rtx and fec ssrcs)
     */
    private static class SecondarySsrcs
        implements Iterable<SecondarySsrc>
    {
        public List<SecondarySsrc> secondarySsrcs;

        /**
         * Initializes a new {@link SecondarySsrcs} instance.
         */
        public SecondarySsrcs(List<SecondarySsrc> secondarySsrcs)
        {
            this.secondarySsrcs = secondarySsrcs;
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public Iterator<SecondarySsrc> iterator()
        {
            return secondarySsrcs.iterator();
        }
    }

    /**
     * Describes one or more primary video ssrcs for a single
     * {@link MediaSourceDesc}
     */
    private static class SourceSsrcs
        implements Iterable<Long>
    {
        private String name;

        private List<Long> sourceSsrcs;

        private String owner;

        /**
         * Initializes a new {@link SourceSsrcs} instance for a single SSRC.
         */
        private SourceSsrcs(Long ssrc)
        {
            this(Collections.singletonList(ssrc));
        }

        /**
         * Initializes a new {@link SourceSsrcs} instance for a list of SSRCs.
         */
        public SourceSsrcs(List<Long> sourceSsrcs)
        {
            this.sourceSsrcs = sourceSsrcs;
        }

        /**
         * Checks whether this instance contains a specific SSRC.
         */
        public boolean contains(Long ssrc)
        {
            return sourceSsrcs.contains(ssrc);
        }

        /**
         * @return the number of SSRCs in this instance.
         */
        public int size()
        {
            return sourceSsrcs.size();
        }

        /**
         * Gets the SSRC at a specific index.
         */
        public Long get(int index)
        {
            return sourceSsrcs.get(index);
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public Iterator<Long> iterator()
        {
            return sourceSsrcs.iterator();
        }
    }

    /**
     * Creates a single MediaSourceDesc with the given information
     * @param primarySsrcs the set of primary video ssrcs belonging to this source
     * @param numSpatialLayersPerStream the number of spatial layers per stream
     * for this source
     * @param numTemporalLayersPerStream the number of temporal layers per stream
     * for this source
     * @param allSecondarySsrcs a map of primary ssrc -> SecondarySsrcs, which lists
     * the ssrc and type of all the secondary ssrcs for a given primary (e.g.
     * its corresponding rtx and fec ssrcs)
     * @param owner An externally specified owner, if available.
     * @param name An externally specified name, if available.
     * @return the created MediaSourceDesc
     */
    private static MediaSourceDesc createSource(
            SourceSsrcs primarySsrcs,
            int numSpatialLayersPerStream,
            int numTemporalLayersPerStream,
            Map<Long, SecondarySsrcs> allSecondarySsrcs,
            String owner,
            String name
    )
    {
        RtpEncodingDesc[] encodings =
            new RtpEncodingDesc[primarySsrcs.size()];

        // this loop builds a subjective quality index array that looks like
        // this:
        //
        // [s0t0, s0t1, s0t2, s1t0, s1t1, s1t2, s2t0, s2t1, s2t2]
        //
        // The spatial layer is offered either by simulcast (VP8) or spatial
        // scalability (VP9). Exotic cases might do simulcast + spatial
        // scalability.

        //TODO(brian): this is only correct if the highest res is 720p
        int height = VP8_SIMULCAST_BASE_LAYER_HEIGHT;

        for (int encodingIdx = 0; encodingIdx < primarySsrcs.size(); encodingIdx++) {
            Long primarySsrc = primarySsrcs.get(encodingIdx);
            SecondarySsrcs ssrcSecondarySsrcs = allSecondarySsrcs.get(primarySsrc);

            encodings[encodingIdx] = createRtpEncodingDesc(primarySsrc,
                numSpatialLayersPerStream, numTemporalLayersPerStream,
                ssrcSecondarySsrcs, encodingIdx, height);

            height *= 2;
        }

        if (name == null)
        {
            throw new IllegalArgumentException("The 'name' is missing in the source description");
        }
        if (owner == null)
        {
            throw new IllegalArgumentException("The 'owner' is missing in the source description");
        }

        return new MediaSourceDesc(encodings, owner, name);
    }
}

<MSG> Optimize MediaSourceFactory#getOwner(). (#1579)


<DFF> @@ -21,6 +21,8 @@
 import org.jitsi.xmpp.extensions.colibri.*;
 import org.jitsi.xmpp.extensions.jingle.*;
 import org.jitsi.xmpp.extensions.jitsimeet.*;
+import org.jxmpp.jid.parts.*;
+import org.jxmpp.util.*;
 
 import java.util.*;
 import java.util.stream.*;
@@ -470,15 +472,45 @@ public static String getOwner(SourcePacketExtension source)
             = source == null
                 ? null : source.getFirstChildOfType(
                     SSRCInfoPacketExtension.class);
+
         if (ssrcInfoPacketExtension != null)
         {
-            return
-                ssrcInfoPacketExtension.getOwner()
-                    .getResourceOrEmpty().toString();
+            /*
+             * N.B.: this is semantically the same as
+             *    return
+             *       ssrcInfoPacketExtension.getOwner()
+             *          .getResourceOrEmpty().toString();
+             *
+             * but avoids expensive construction of a lot of JID parts we just throw away.
+             *
+             * This function is called repeatedly by {@link ConfOctoTransport#setSources}
+             * so needs to be fast.
+             */
+            String ownerAttr = ssrcInfoPacketExtension.getAttributeAsString(SSRCInfoPacketExtension.OWNER_ATTR_NAME);
+            if (ownerAttr != null)
+            {
+                return getResourceFromJid(ownerAttr);
+            }
         }
         return null;
     }
 
+    /**
+     * Given a string containing a JID, return its StringPrep'd Resource part,
+     * or {@link Resourcepart#EMPTY}, as a string.
+     *
+     * Helper for {@link #getOwner}
+     */
+    private static String getResourceFromJid(String jid)
+    {
+        String unpreppedResource = XmppStringUtils.parseResource(jid);
+        if (unpreppedResource.isEmpty())
+        {
+            return Resourcepart.EMPTY.toString();
+        }
+        return Resourcepart.fromOrThrowUnchecked(unpreppedResource).toString();
+    }
+
     /**
      * Creates {@link MediaSourceDesc}s from signaling params
      *

{'addition_count': 35, 'commit_subject': 'Optimize MediaSourceFactory#getOwner(). (#1579)', 'deletion_count': 3, 'file_extension': '.java', 'lang': 'java', 'license': 'apache-2.0', 'repo_name': 'jitsi/jitsi-videobridge'}