feat: smooth volume
This commit is contained in:
parent
8f5155895a
commit
75a2117602
9 changed files with 1113 additions and 1085 deletions
|
@ -215,7 +215,7 @@ const queue = player.createQueue(..., {
|
|||
// only trap youtube source
|
||||
if (source === "youtube") {
|
||||
// track here would be youtube track
|
||||
return (await playdl.stream(track.url)).stream;
|
||||
return (await playdl.stream(track.url, { discordPlayerCompatibility : true })).stream;
|
||||
// we must return readable stream or void (returning void means telling discord-player to look for default extractor)
|
||||
}
|
||||
}
|
||||
|
|
|
@ -18,7 +18,7 @@ const queue = player.createQueue(..., {
|
|||
// only trap youtube source
|
||||
if (source === "youtube") {
|
||||
// track here would be youtube track
|
||||
return (await playdl.stream(track.url)).stream;
|
||||
return (await playdl.stream(track.url, { discordPlayerCompatibility : true })).stream;
|
||||
// we must return readable stream or void (returning void means telling discord-player to look for default extractor)
|
||||
}
|
||||
}
|
||||
|
|
|
@ -60,14 +60,14 @@
|
|||
},
|
||||
"homepage": "https://discord-player.js.org",
|
||||
"dependencies": {
|
||||
"@discordjs/voice": "^0.7.5",
|
||||
"@discordjs/voice": "^0.8.0",
|
||||
"discord-ytdl-core": "^5.0.4",
|
||||
"libsodium-wrappers": "^0.7.9",
|
||||
"soundcloud-scraper": "^5.0.2",
|
||||
"spotify-url-info": "^2.2.3",
|
||||
"tiny-typed-emitter": "^2.1.0",
|
||||
"youtube-sr": "^4.1.9",
|
||||
"ytdl-core": "^4.9.1"
|
||||
"youtube-sr": "^4.1.12",
|
||||
"ytdl-core": "^4.10.0"
|
||||
},
|
||||
"devDependencies": {
|
||||
"@babel/cli": "^7.16.0",
|
||||
|
|
|
@ -153,6 +153,7 @@ class Player extends EventEmitter<PlayerEvents> {
|
|||
|
||||
const _meta = queueInitOptions.metadata;
|
||||
delete queueInitOptions["metadata"];
|
||||
queueInitOptions.volumeSmoothness ??= 0.1;
|
||||
queueInitOptions.ytdlOptions ??= this.options.ytdlOptions;
|
||||
const queue = new Queue(this, guild, queueInitOptions);
|
||||
queue.metadata = _meta;
|
||||
|
|
|
@ -702,6 +702,11 @@ class Queue<T = unknown> {
|
|||
|
||||
if (options.seek) this._streamTime = options.seek;
|
||||
this._filtersUpdate = options.filtersUpdate;
|
||||
|
||||
if (resource.volume && typeof this.options.volumeSmoothness === "number") {
|
||||
Reflect.set(resource.volume, "_smoothing", this.options.volumeSmoothness || 0);
|
||||
}
|
||||
|
||||
this.setVolume(this.options.initialVolume);
|
||||
|
||||
setTimeout(() => {
|
||||
|
|
116
src/VolumeTransformer.ts
Normal file
116
src/VolumeTransformer.ts
Normal file
|
@ -0,0 +1,116 @@
|
|||
// prism's volume transformer with smooth volume support
|
||||
|
||||
import { Transform, TransformOptions } from "stream";
|
||||
|
||||
export interface VolumeTransformerOptions extends TransformOptions {
|
||||
type?: "s16le" | "s16be" | "s32le" | "s32be";
|
||||
smoothness?: number;
|
||||
volume?: number;
|
||||
}
|
||||
|
||||
export class VolumeTransformer extends Transform {
|
||||
private _bits: number;
|
||||
private _smoothing: number;
|
||||
private _bytes: number;
|
||||
private _extremum: number;
|
||||
private _chunk: Buffer;
|
||||
public volume: number;
|
||||
private _targetVolume: number;
|
||||
constructor(options: VolumeTransformerOptions = {}) {
|
||||
super(options);
|
||||
switch (options.type) {
|
||||
case "s16le":
|
||||
this._readInt = (buffer, index) => buffer.readInt16LE(index);
|
||||
this._writeInt = (buffer, int, index) => buffer.writeInt16LE(int, index);
|
||||
this._bits = 16;
|
||||
break;
|
||||
case "s16be":
|
||||
this._readInt = (buffer, index) => buffer.readInt16BE(index);
|
||||
this._writeInt = (buffer, int, index) => buffer.writeInt16BE(int, index);
|
||||
this._bits = 16;
|
||||
break;
|
||||
case "s32le":
|
||||
this._readInt = (buffer, index) => buffer.readInt32LE(index);
|
||||
this._writeInt = (buffer, int, index) => buffer.writeInt32LE(int, index);
|
||||
this._bits = 32;
|
||||
break;
|
||||
case "s32be":
|
||||
this._readInt = (buffer, index) => buffer.readInt32BE(index);
|
||||
this._writeInt = (buffer, int, index) => buffer.writeInt32BE(int, index);
|
||||
this._bits = 32;
|
||||
break;
|
||||
default:
|
||||
throw new Error("VolumeTransformer type should be one of s16le, s16be, s32le, s32be");
|
||||
}
|
||||
this._bytes = this._bits / 8;
|
||||
this._extremum = Math.pow(2, this._bits - 1);
|
||||
this.volume = typeof options.volume === "undefined" ? 1 : options.volume;
|
||||
this._targetVolume = this.volume;
|
||||
this._chunk = Buffer.alloc(0);
|
||||
this._smoothing = options.smoothness || 0;
|
||||
}
|
||||
|
||||
_readInt(buffer: Buffer, index: number) {
|
||||
return index;
|
||||
}
|
||||
_writeInt(buffer: Buffer, int: number, index: number) {
|
||||
return index;
|
||||
}
|
||||
|
||||
_transform(chunk: Buffer, encoding: BufferEncoding, done: () => unknown) {
|
||||
if (this._smoothing > 0 && this.volume !== this._targetVolume) {
|
||||
if (this.volume < this._targetVolume) {
|
||||
this.volume = this.volume + this._smoothing >= this._targetVolume ? this._targetVolume : this.volume + this._smoothing;
|
||||
} else if (this.volume > this._targetVolume) {
|
||||
this.volume = this.volume - this._smoothing <= this._targetVolume ? this._targetVolume : this.volume - this._smoothing;
|
||||
}
|
||||
}
|
||||
|
||||
if (this.volume === 1) {
|
||||
this.push(chunk);
|
||||
return done();
|
||||
}
|
||||
|
||||
const { _bytes, _extremum } = this;
|
||||
|
||||
chunk = this._chunk = Buffer.concat([this._chunk, chunk]);
|
||||
if (chunk.length < _bytes) return done();
|
||||
|
||||
const complete = Math.floor(chunk.length / _bytes) * _bytes;
|
||||
|
||||
for (let i = 0; i < complete; i += _bytes) {
|
||||
const int = Math.min(_extremum - 1, Math.max(-_extremum, Math.floor(this.volume * this._readInt(chunk, i))));
|
||||
this._writeInt(chunk, int, i);
|
||||
}
|
||||
|
||||
this._chunk = chunk.slice(complete);
|
||||
this.push(chunk.slice(0, complete));
|
||||
return done();
|
||||
}
|
||||
|
||||
_destroy(err: Error, cb: (error: Error) => void) {
|
||||
super._destroy(err, cb);
|
||||
this._chunk = null;
|
||||
}
|
||||
|
||||
setVolume(volume: number) {
|
||||
this._targetVolume = volume;
|
||||
if (this._smoothing <= 0) this.volume = volume;
|
||||
}
|
||||
|
||||
setVolumeDecibels(db: number) {
|
||||
this.setVolume(Math.pow(10, db / 20));
|
||||
}
|
||||
|
||||
setVolumeLogarithmic(value: number) {
|
||||
this.setVolume(Math.pow(value, 1.660964));
|
||||
}
|
||||
|
||||
get volumeDecibels() {
|
||||
return Math.log10(this.volume) * 20;
|
||||
}
|
||||
|
||||
get volumeLogarithmic() {
|
||||
return Math.pow(this.volume, 1 / 1.660964);
|
||||
}
|
||||
}
|
4
src/smoothVolume.ts
Normal file
4
src/smoothVolume.ts
Normal file
|
@ -0,0 +1,4 @@
|
|||
import { VolumeTransformer } from "./VolumeTransformer";
|
||||
|
||||
// eslint-disable-next-line
|
||||
(require("prism-media") as typeof import("prism-media")).VolumeTransformer = VolumeTransformer;
|
|
@ -135,6 +135,8 @@ export interface PlayerProgressbarOptions {
|
|||
* @property {number} [bufferingTimeout=3000] Buffering timeout for the stream
|
||||
* @property {boolean} [spotifyBridge=true] If player should bridge spotify source to youtube
|
||||
* @property {boolean} [disableVolume=false] If player should disable inline volume
|
||||
* @property {boolean} [volumeSmoothness=0] The volume transition smoothness between volume changes (lower the value to get better result)
|
||||
* Setting this or leaving this empty will disable this effect. Example: `volumeSmoothness: 0.1`
|
||||
* @property {Function} [onBeforeCreateStream] Runs before creating stream
|
||||
*/
|
||||
export interface PlayerOptions {
|
||||
|
@ -148,6 +150,7 @@ export interface PlayerOptions {
|
|||
bufferingTimeout?: number;
|
||||
spotifyBridge?: boolean;
|
||||
disableVolume?: boolean;
|
||||
volumeSmoothness?: number;
|
||||
onBeforeCreateStream?: (track: Track, source: TrackSource, queue: Queue) => Promise<Readable>;
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue