-
-
Notifications
You must be signed in to change notification settings - Fork 773
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1619 from pedroSG94/feature/decoder-refactor
Feature/decoder refactor
- Loading branch information
Showing
15 changed files
with
417 additions
and
224 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
151 changes: 151 additions & 0 deletions
151
encoder/src/main/java/com/pedro/encoder/input/decoder/AndroidExtractor.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,151 @@ | ||
| /* | ||
| * | ||
| * * Copyright (C) 2024 pedroSG94. | ||
| * * | ||
| * * 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 com.pedro.encoder.input.decoder | ||
|
|
||
| import android.content.Context | ||
| import android.media.MediaExtractor | ||
| import android.media.MediaFormat | ||
| import android.net.Uri | ||
| import com.pedro.common.frame.MediaFrame | ||
| import com.pedro.common.getIntegerSafe | ||
| import com.pedro.common.getLongSafe | ||
| import com.pedro.common.validMessage | ||
| import java.io.FileDescriptor | ||
| import java.io.IOException | ||
| import java.nio.ByteBuffer | ||
|
|
||
| /** | ||
| * Created by pedro on 18/10/24. | ||
| */ | ||
| class AndroidExtractor: Extractor { | ||
|
|
||
| private var mediaExtractor = MediaExtractor() | ||
| private var sleepTime: Long = 0 | ||
| private var accumulativeTs: Long = 0 | ||
| @Volatile | ||
| private var lastExtractorTs: Long = 0 | ||
| private var format: MediaFormat? = null | ||
|
|
||
| override fun selectTrack(type: MediaFrame.Type): String { | ||
| return when (type) { | ||
| MediaFrame.Type.VIDEO -> selectTrack("video/") | ||
| MediaFrame.Type.AUDIO -> selectTrack("audio/") | ||
| } | ||
| } | ||
|
|
||
| override fun initialize(path: String) { | ||
| try { | ||
| reset() | ||
| mediaExtractor = MediaExtractor() | ||
| mediaExtractor.setDataSource(path) | ||
| } catch (e: Exception) { | ||
| throw IOException(e.validMessage()) | ||
| } | ||
| } | ||
|
|
||
| override fun initialize(context: Context, uri: Uri) { | ||
| try { | ||
| reset() | ||
| mediaExtractor = MediaExtractor() | ||
| mediaExtractor.setDataSource(context, uri, null) | ||
| } catch (e: Exception) { | ||
| throw IOException(e.validMessage()) | ||
| } | ||
| } | ||
|
|
||
| override fun initialize(fileDescriptor: FileDescriptor) { | ||
| try { | ||
| reset() | ||
| mediaExtractor = MediaExtractor() | ||
| mediaExtractor.setDataSource(fileDescriptor) | ||
| } catch (e: Exception) { | ||
| throw IOException(e.validMessage()) | ||
| } | ||
| } | ||
|
|
||
| override fun readFrame(buffer: ByteBuffer): Int { | ||
| return mediaExtractor.readSampleData(buffer, 0) | ||
| } | ||
|
|
||
| override fun advance(): Boolean { | ||
| return mediaExtractor.advance() | ||
| } | ||
|
|
||
| override fun getTimeStamp(): Long { | ||
| return mediaExtractor.sampleTime | ||
| } | ||
|
|
||
| override fun getSleepTime(ts: Long): Long { | ||
| val extractorTs = getTimeStamp() | ||
| accumulativeTs += extractorTs - lastExtractorTs | ||
| lastExtractorTs = getTimeStamp() | ||
| sleepTime = if (accumulativeTs > ts) (accumulativeTs - ts) / 1000 else 0 | ||
| return sleepTime | ||
| } | ||
|
|
||
| override fun seekTo(time: Long) { | ||
| mediaExtractor.seekTo(time, MediaExtractor.SEEK_TO_PREVIOUS_SYNC) | ||
| lastExtractorTs = getTimeStamp() | ||
| } | ||
|
|
||
| override fun release() { | ||
| mediaExtractor.release() | ||
| } | ||
|
|
||
| override fun getVideoInfo(): VideoInfo { | ||
| val format = this.format ?: throw IOException("Extractor track not selected") | ||
| val width = format.getIntegerSafe(MediaFormat.KEY_WIDTH) ?: throw IOException("Width info is required") | ||
| val height = format.getIntegerSafe(MediaFormat.KEY_HEIGHT) ?: throw IOException("Height info is required") | ||
| val duration = format.getLongSafe(MediaFormat.KEY_DURATION) ?: throw IOException("Duration info is required") | ||
| val fps = format.getIntegerSafe(MediaFormat.KEY_FRAME_RATE) ?: 30 | ||
| return VideoInfo(width, height, fps, duration) | ||
| } | ||
|
|
||
| override fun getAudioInfo(): AudioInfo { | ||
| val format = this.format ?: throw IOException("Extractor track not selected") | ||
| val sampleRate = format.getIntegerSafe(MediaFormat.KEY_SAMPLE_RATE) ?: throw IOException("Channels info is required") | ||
| val channels = format.getIntegerSafe(MediaFormat.KEY_CHANNEL_COUNT) ?: throw IOException("SampleRate info is required") | ||
| val duration = format.getLongSafe(MediaFormat.KEY_DURATION) ?: throw IOException("Duration info is required") | ||
| return AudioInfo(sampleRate, channels, duration) | ||
| } | ||
|
|
||
| override fun getFormat(): MediaFormat { | ||
| return format ?: throw IOException("Extractor track not selected") | ||
| } | ||
|
|
||
| private fun selectTrack(type: String): String { | ||
| for (i in 0 until mediaExtractor.trackCount) { | ||
| val format = mediaExtractor.getTrackFormat(i) | ||
| val mime = format.getString(MediaFormat.KEY_MIME) ?: continue | ||
| if (mime.startsWith(type, ignoreCase = true)) { | ||
| mediaExtractor.selectTrack(i) | ||
| this.format = format | ||
| return mime | ||
| } | ||
| } | ||
| throw IOException("track not found") | ||
| } | ||
|
|
||
| private fun reset() { | ||
| sleepTime = 0 | ||
| accumulativeTs = 0 | ||
| lastExtractorTs = 0 | ||
| format = null | ||
| } | ||
| } |
Oops, something went wrong.