ThePain/Swiftgram
1
0
Fork 0
mirror of https://github.com/Swiftgram/Telegram-iOS.git synced 2025-08-08 08:31:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move SoftwareVideoSource to UniversalMediaPlayer

Browse Source
This commit is contained in:
Peter 2019-07-12 22:10:53 +04:00
parent 01f0c95efe
commit 464738d8a8
1 changed files with 32 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
submodules/TelegramUI/TelegramUI/SoftwareVideoSource.swift → submodules/MediaPlayer/Sources/SoftwareVideoSource.swift
View File

@ -3,7 +3,6 @@ import UIKit
import CoreMedia import CoreMedia
import SwiftSignalKit import SwiftSignalKit
import FFMpeg import FFMpeg
import UniversalMediaPlayer
private func readPacketCallback(userData: UnsafeMutableRawPointer?, buffer: UnsafeMutablePointer<UInt8>?, bufferSize: Int32) -> Int32 { private func readPacketCallback(userData: UnsafeMutableRawPointer?, buffer: UnsafeMutablePointer<UInt8>?, bufferSize: Int32) -> Int32 {
let context = Unmanaged<SoftwareVideoSource>.fromOpaque(userData!).takeUnretainedValue() let context = Unmanaged<SoftwareVideoSource>.fromOpaque(userData!).takeUnretainedValue()
@ -46,7 +45,7 @@ private final class SoftwareVideoStream {
} }
} }
final class SoftwareVideoSource { public final class SoftwareVideoSource {
private var readingError = false private var readingError = false
private var videoStream: SoftwareVideoStream? private var videoStream: SoftwareVideoStream?
private var avIoContext: FFMpegAVIOContext? private var avIoContext: FFMpegAVIOContext?
@ -55,7 +54,7 @@ final class SoftwareVideoSource {
fileprivate let fd: Int32? fileprivate let fd: Int32?
fileprivate let size: Int32 fileprivate let size: Int32
init(path: String) { public init(path: String) {
let _ = FFMpegMediaFrameSourceContextHelpers.registerFFMpegGlobals let _ = FFMpegMediaFrameSourceContextHelpers.registerFFMpegGlobals
var s = stat() var s = stat()
@ -102,10 +101,10 @@ final class SoftwareVideoSource {
let codecId = avFormatContext.codecId(atStreamIndex: streamIndex) let codecId = avFormatContext.codecId(atStreamIndex: streamIndex)
let fpsAndTimebase = avFormatContext.fpsAndTimebase(forStreamIndex: streamIndex, defaultTimeBase: CMTimeMake(1, 40000)) let fpsAndTimebase = avFormatContext.fpsAndTimebase(forStreamIndex: streamIndex, defaultTimeBase: CMTimeMake(value: 1, timescale: 40000))
let (fps, timebase) = (fpsAndTimebase.fps, fpsAndTimebase.timebase) let (fps, timebase) = (fpsAndTimebase.fps, fpsAndTimebase.timebase)
let duration = CMTimeMake(avFormatContext.duration(atStreamIndex: streamIndex), timebase.timescale) let duration = CMTimeMake(value: avFormatContext.duration(atStreamIndex: streamIndex), timescale: timebase.timescale)
let metrics = avFormatContext.metricsForStream(at: streamIndex) let metrics = avFormatContext.metricsForStream(at: streamIndex)
@ -154,14 +153,14 @@ final class SoftwareVideoSource {
if let videoStream = videoStream, Int(packet.streamIndex) == videoStream.index { if let videoStream = videoStream, Int(packet.streamIndex) == videoStream.index {
let packetPts = packet.pts let packetPts = packet.pts
let pts = CMTimeMake(packetPts, videoStream.timebase.timescale) let pts = CMTimeMake(value: packetPts, timescale: videoStream.timebase.timescale)
let dts = CMTimeMake(packet.dts, videoStream.timebase.timescale) let dts = CMTimeMake(value: packet.dts, timescale: videoStream.timebase.timescale)
let duration: CMTime let duration: CMTime
let frameDuration = packet.duration let frameDuration = packet.duration
if frameDuration != 0 { if frameDuration != 0 {
duration = CMTimeMake(frameDuration * videoStream.timebase.value, videoStream.timebase.timescale) duration = CMTimeMake(value: frameDuration * videoStream.timebase.value, timescale: videoStream.timebase.timescale)
} else { } else {
duration = videoStream.fps duration = videoStream.fps
} }
@ -193,7 +192,7 @@ final class SoftwareVideoSource {
return (frames.first, endOfStream) return (frames.first, endOfStream)
} }
func readFrame(maxPts: CMTime?) -> (MediaTrackFrame?, CGFloat, CGFloat, Bool) { public func readFrame(maxPts: CMTime?) -> (MediaTrackFrame?, CGFloat, CGFloat, Bool) {
if let videoStream = self.videoStream { if let videoStream = self.videoStream {
let (decodableFrame, loop) = self.readDecodableFrame() let (decodableFrame, loop) = self.readDecodableFrame()
if let decodableFrame = decodableFrame { if let decodableFrame = decodableFrame {
@ -209,4 +208,28 @@ final class SoftwareVideoSource {
return (nil, 0.0, 1.0, false) return (nil, 0.0, 1.0, false)
} }
} }
public func readImage() -> (UIImage?, CGFloat, CGFloat, Bool) {
if let videoStream = self.videoStream {
for _ in 0 ..< 10 {
let (decodableFrame, loop) = self.readDecodableFrame()
if let decodableFrame = decodableFrame {
if let renderedFrame = videoStream.decoder.render(frame: decodableFrame) {
return (renderedFrame, CGFloat(videoStream.rotationAngle), CGFloat(videoStream.aspect), loop)
}
}
}
return (nil, CGFloat(videoStream.rotationAngle), CGFloat(videoStream.aspect), true)
} else {
return (nil, 0.0, 1.0, false)
}
}
public func seek(timestamp: Double) {
if let stream = self.videoStream, let avFormatContext = self.avFormatContext {
let pts = CMTimeMakeWithSeconds(timestamp, preferredTimescale: stream.timebase.timescale)
avFormatContext.seekFrame(forStreamIndex: Int32(stream.index), pts: pts.value)
stream.decoder.reset()
}
}
} }