n3r4zzurr0/video-frames
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor fixes

Browse Source
This commit is contained in:
Utkarsh Verma
2022-07-08 09:08:50 +05:30
parent da9daa2841
commit 50ff515785
1 changed files with 50 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

118
index.js
View File

@@ -4,24 +4,6 @@ module.exports = async options => {
let seekResolve
let extractOffsets = false
// For minification
// Options
const _call = 'call'
const _count = 'count'
const _width = 'width'
const _height = 'height'
const _length = 'length'
const _format = 'format'
const _onLoad = 'onLoad'
const _offsets = 'offsets'
const _endTime = 'endTime'
const _duration = 'duration'
const _startTime = 'startTime'
const _onProgress = 'onProgress'
const _currentTime = 'currentTime'
const _createElement = 'createElement'
const frames = []
const isNumber = n => {
@@ -29,7 +11,7 @@ module.exports = async options => {
}
const isTimestamp = timestamp => {
return isNumber(timestamp) && +timestamp >= 0 && +timestamp <= video[_duration]
return isNumber(timestamp) && +timestamp >= 0 && +timestamp <= video.duration
}
const isPrototypeOf = (constructor, value) => {
@@ -37,13 +19,13 @@ module.exports = async options => {
}
const fallbackToDefault = (property, defaultValue) => {
options[property] = hasOwnProperty[_call](options, property) ? options[property] : defaultValue
options[property] = hasOwnProperty.call(options, property) ? options[property] : defaultValue
}
const hasOwnProperty = {}.hasOwnProperty
// Buffer Video Element
const video = document[_createElement]('video')
const video = document.createElement('video')
video.src = options.url
video.crossOrigin = 'anonymous'
video.onseeked = async () => {
@@ -53,108 +35,108 @@ module.exports = async options => {
error = true
}
while ((video[_duration] === Infinity || isNaN(video[_duration])) && video.readyState < 2) {
while ((video.duration === Infinity || isNaN(video.duration)) && video.readyState < 2) {
await new Promise(resolve => setTimeout(resolve, 100))
video[_currentTime] = 10000000 * Math.random()
video.currentTime = 10000000 * Math.random()
if (error) { break }
}
// Set currentTime to duration / 2
// (fix for correctly invoking onseeked event for currentTime = 0 if the first frame is at 0 sec.)
video[_currentTime] = video[_duration] / 2
video.currentTime = video.duration / 2
await new Promise(resolve => { seekResolve = resolve })
// Set options to default values if not set
fallbackToDefault(_format, 'image/png')
fallbackToDefault(_offsets, [])
fallbackToDefault(_startTime, 0)
fallbackToDefault(_endTime, video[_duration])
fallbackToDefault(_count, 1)
fallbackToDefault(_onLoad, false)
fallbackToDefault(_onProgress, false)
fallbackToDefault('format', 'image/png')
fallbackToDefault('offsets', [])
fallbackToDefault('startTime', 0)
fallbackToDefault('endTime', video.duration)
fallbackToDefault('count', 1)
fallbackToDefault('onLoad', false)
fallbackToDefault('onProgress', false)
// Filter out invalid offsets
if (!isPrototypeOf(Array, options[_offsets])) { options[_offsets] = [] } else {
options[_offsets] = options[_offsets].filter(offset => {
if (!isPrototypeOf(Array, options.offsets)) { options.offsets = [] } else {
options.offsets = options.offsets.filter(offset => {
return isTimestamp(offset)
})
}
if (options[_offsets][_length] !== 0) { extractOffsets = true }
if (options.offsets.length !== 0) { extractOffsets = true }
// Check if start and end times are valid
if (!isTimestamp(options[_startTime])) { options[_startTime] = 0 }
if (!isTimestamp(options.startTime)) { options.startTime = 0 }
if (!isTimestamp(options[_endTime])) { options[_endTime] = video[_duration] }
if (!isTimestamp(options.endTime)) { options.endTime = video.duration }
// Float values
options[_startTime] = +options[_startTime]
options[_endTime] = +options[_endTime]
options.startTime = +options.startTime
options.endTime = +options.endTime
if (options[_startTime] >= options[_endTime]) {
options[_startTime] = options[_endTime]
options[_count] = 1
if (options.startTime >= options.endTime) {
options.startTime = options.endTime
options.count = 1
}
// Convert count value to a positive integer (floor() or 0 if string)
options[_count] = Math.abs(~~options[_count])
if (options[_count] === 0) { options[_count] = 1 }
if (extractOffsets) { options[_count] = options[_offsets][_length] }
options.count = Math.abs(~~options.count)
if (options.count === 0) { options.count = 1 }
if (extractOffsets) { options.count = options.offsets.length }
// Starting at startTime and ending at endTime - interval
const interval = (options[_endTime] - options[_startTime]) / options[_count]
const interval = (options.endTime - options.startTime) / options.count
// Set Width and Height
let isWidthSet = hasOwnProperty[_call](options, 'width')
let isHeightSet = hasOwnProperty[_call](options, 'height')
let isWidthSet = hasOwnProperty.call(options, 'width')
let isHeightSet = hasOwnProperty.call(options, 'height')
const videoDimensionRatio = video.videoWidth / video.videoHeight
// Reset Width and Height if not valid
if (isWidthSet && !isNumber(options[_width])) { isWidthSet = false }
if (isWidthSet && !isNumber(options.width)) { isWidthSet = false }
if (isHeightSet && !isNumber(options[_height])) { isHeightSet = false }
if (isHeightSet && !isNumber(options.height)) { isHeightSet = false }
if (!isWidthSet && !isHeightSet) {
// Both Width and Height not set
options[_width] = 128 // Default Value (randomly set)
options[_height] = options[_width] / videoDimensionRatio
options.width = 128 // Default Value (randomly set)
options.height = options.width / videoDimensionRatio
} else if (isWidthSet && !isHeightSet) {
// Width set but Height not set
options[_height] = options[_width] / videoDimensionRatio
options.height = options.width / videoDimensionRatio
} else if (!isWidthSet && isHeightSet) {
// Height set but Width not set
options[_width] = options[_height] * videoDimensionRatio
options.width = options.height * videoDimensionRatio
}
// Float values
options[_width] = +options[_width]
options[_height] = +options[_height]
options.width = +options.width
options.height = +options.height
// Reset onLoad and onProgress functions if not valid
if (!isPrototypeOf(Function, options[_onLoad])) { options[_onLoad] = false }
if (!isPrototypeOf(Function, options.onLoad)) { options.onLoad = false }
if (!isPrototypeOf(Function, options[_onProgress])) { options[_onProgress] = false }
if (!isPrototypeOf(Function, options.onProgress)) { options.onProgress = false }
if (options[_onLoad]) { options[_onLoad]() }
if (options.onLoad) { options.onLoad() }
// Buffer Canvas Element
const canvas = document[_createElement]('canvas')
const canvas = document.createElement('canvas')
const context = canvas.getContext('2d')
canvas[_width] = options[_width]
canvas[_height] = options[_height]
canvas.width = options.width
canvas.height = options.height
const extract = async resolve => {
while (index < options[_count]) {
video[_currentTime] = extractOffsets ? options[_offsets][index] : options[_startTime] + index * interval
while (index < options.count) {
video.currentTime = extractOffsets ? options.offsets[index] : options.startTime + index * interval
await new Promise(resolve => { seekResolve = resolve })
context.clearRect(0, 0, canvas[_width], canvas[_height])
context.drawImage(video, 0, 0, canvas[_width], canvas[_height])
context.clearRect(0, 0, canvas.width, canvas.height)
context.drawImage(video, 0, 0, canvas.width, canvas.height)
frames.push({
offset: video[_currentTime],
image: canvas.toDataURL(options[_format])
offset: video.currentTime,
image: canvas.toDataURL(options.format)
})
index++
if (options[_onProgress]) { options[_onProgress](index, options[_count]) }
if (options.onProgress) { options.onProgress(index, options.count) }
}
resolve(frames)
}