/*
* Copyright (C) 2020 The Android Open Source Project
*
* 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 androidx.media3.exoplayer.source;
import android.net.Uri;
import androidx.annotation.Nullable;
import androidx.media3.common.C;
import androidx.media3.common.DataReader;
import androidx.media3.common.util.Assertions;
import androidx.media3.common.util.UnstableApi;
import androidx.media3.common.util.Util;
import androidx.media3.extractor.DefaultExtractorInput;
import androidx.media3.extractor.Extractor;
import androidx.media3.extractor.ExtractorInput;
import androidx.media3.extractor.ExtractorOutput;
import androidx.media3.extractor.ExtractorsFactory;
import androidx.media3.extractor.PositionHolder;
import androidx.media3.extractor.mp3.Mp3Extractor;
import java.io.EOFException;
import java.io.IOException;
import java.util.List;
import java.util.Map;
/**
* {@link ProgressiveMediaExtractor} built on top of {@link Extractor} instances, whose
* implementation classes are bundled in the app.
*/
@UnstableApi
public final class BundledExtractorsAdapter implements ProgressiveMediaExtractor {
private final ExtractorsFactory extractorsFactory;
@Nullable private Extractor extractor;
@Nullable private ExtractorInput extractorInput;
/**
* Creates a holder that will select an extractor and initialize it using the specified output.
*
* @param extractorsFactory The {@link ExtractorsFactory} providing the extractors to choose from.
*/
public BundledExtractorsAdapter(ExtractorsFactory extractorsFactory) {
this.extractorsFactory = extractorsFactory;
}
@Override
public void init(
DataReader dataReader,
Uri uri,
Map<String, List<String>> responseHeaders,
long position,
long length,
ExtractorOutput output)
throws IOException {
ExtractorInput extractorInput = new DefaultExtractorInput(dataReader, position, length);
this.extractorInput = extractorInput;
if (extractor != null) {
return;
}
Extractor[] extractors = extractorsFactory.createExtractors(uri, responseHeaders);
if (extractors.length == 1) {
this.extractor = extractors[0];
} else {
for (Extractor extractor : extractors) {
try {
if (extractor.sniff(extractorInput)) {
this.extractor = extractor;
break;
}
} catch (EOFException e) {
// Do nothing.
} finally {
Assertions.checkState(this.extractor != null || extractorInput.getPosition() == position);
extractorInput.resetPeekPosition();
}
}
if (extractor == null) {
throw new UnrecognizedInputFormatException(
"None of the available extractors ("
+ Util.getCommaDelimitedSimpleClassNames(extractors)
+ ") could read the stream.",
Assertions.checkNotNull(uri));
}
}
extractor.init(output);
}
@Override
public void release() {
if (extractor != null) {
extractor.release();
extractor = null;
}
extractorInput = null;
}
@Override
public void disableSeekingOnMp3Streams() {
if (extractor == null) {
return;
}
Extractor underlyingExtractor = extractor.getUnderlyingImplementation();
if (underlyingExtractor instanceof Mp3Extractor) {
((Mp3Extractor) underlyingExtractor).disableSeeking();
}
}
@Override
public long getCurrentInputPosition() {
return extractorInput != null ? extractorInput.getPosition() : C.INDEX_UNSET;
}
@Override
public void seek(long position, long seekTimeUs) {
Assertions.checkNotNull(extractor).seek(position, seekTimeUs);
}
@Override
public int read(PositionHolder positionHolder) throws IOException {
return Assertions.checkNotNull(extractor)
.read(Assertions.checkNotNull(extractorInput), positionHolder);
}
}