// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "modules/mediasource/SourceBufferTrackBaseSupplement.h"

#include "core/html/track/TrackBase.h"
#include "modules/mediasource/SourceBuffer.h"

namespace blink {

static const char* kSupplementName = "SourceBufferTrackBaseSupplement";

// static
SourceBufferTrackBaseSupplement* SourceBufferTrackBaseSupplement::fromIfExists(TrackBase& track)
{
    return static_cast<SourceBufferTrackBaseSupplement*>(Supplement<TrackBase>::from(track, kSupplementName));
}

// static
SourceBufferTrackBaseSupplement& SourceBufferTrackBaseSupplement::from(TrackBase& track)
{
    SourceBufferTrackBaseSupplement* supplement = fromIfExists(track);
    if (!supplement) {
        supplement = new SourceBufferTrackBaseSupplement();
        Supplement<TrackBase>::provideTo(track, kSupplementName, supplement);
    }
    return *supplement;
}

// static
SourceBuffer* SourceBufferTrackBaseSupplement::sourceBuffer(TrackBase& track)
{
    SourceBufferTrackBaseSupplement* supplement = fromIfExists(track);
    if (supplement)
        return supplement->m_sourceBuffer;
    return nullptr;
}

void SourceBufferTrackBaseSupplement::setSourceBuffer(TrackBase& track, SourceBuffer* sourceBuffer)
{
    from(track).m_sourceBuffer = sourceBuffer;
}

DEFINE_TRACE(SourceBufferTrackBaseSupplement)
{
    visitor->trace(m_sourceBuffer);
    Supplement<TrackBase>::trace(visitor);
}

} // namespace blink