player.rs 56 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706
  1. use byteorder::{LittleEndian, ReadBytesExt};
  2. use futures;
  3. use futures::{future, Async, Future, Poll, Stream};
  4. use std;
  5. use std::borrow::Cow;
  6. use std::cmp::max;
  7. use std::io::{Read, Result, Seek, SeekFrom};
  8. use std::mem;
  9. use std::thread;
  10. use std::time::{Duration, Instant};
  11. use crate::config::{Bitrate, PlayerConfig};
  12. use librespot_core::session::Session;
  13. use librespot_core::spotify_id::SpotifyId;
  14. use librespot_core::util::SeqGenerator;
  15. use crate::audio::{AudioDecrypt, AudioFile, StreamLoaderController};
  16. use crate::audio::{VorbisDecoder, VorbisPacket};
  17. use crate::audio::{
  18. READ_AHEAD_BEFORE_PLAYBACK_ROUNDTRIPS, READ_AHEAD_BEFORE_PLAYBACK_SECONDS,
  19. READ_AHEAD_DURING_PLAYBACK_ROUNDTRIPS, READ_AHEAD_DURING_PLAYBACK_SECONDS,
  20. };
  21. use crate::audio_backend::Sink;
  22. use crate::metadata::{AudioItem, FileFormat};
  23. use crate::mixer::AudioFilter;
  24. const PRELOAD_NEXT_TRACK_BEFORE_END_DURATION_MS: u32 = 30000;
  25. pub struct Player {
  26. commands: Option<futures::sync::mpsc::UnboundedSender<PlayerCommand>>,
  27. thread_handle: Option<thread::JoinHandle<()>>,
  28. play_request_id_generator: SeqGenerator<u64>,
  29. }
  30. #[derive(PartialEq, Debug, Clone, Copy)]
  31. pub enum SinkStatus {
  32. Running,
  33. Closed,
  34. TemporarilyClosed,
  35. }
  36. pub type SinkEventCallback = Box<dyn Fn(SinkStatus) + Send>;
  37. struct PlayerInternal {
  38. session: Session,
  39. config: PlayerConfig,
  40. commands: futures::sync::mpsc::UnboundedReceiver<PlayerCommand>,
  41. state: PlayerState,
  42. preload: PlayerPreload,
  43. sink: Box<dyn Sink>,
  44. sink_status: SinkStatus,
  45. sink_event_callback: Option<SinkEventCallback>,
  46. audio_filter: Option<Box<dyn AudioFilter + Send>>,
  47. event_senders: Vec<futures::sync::mpsc::UnboundedSender<PlayerEvent>>,
  48. }
  49. enum PlayerCommand {
  50. Load {
  51. track_id: SpotifyId,
  52. play_request_id: u64,
  53. play: bool,
  54. position_ms: u32,
  55. },
  56. Preload {
  57. track_id: SpotifyId,
  58. },
  59. Play,
  60. Pause,
  61. Stop,
  62. Seek(u32),
  63. AddEventSender(futures::sync::mpsc::UnboundedSender<PlayerEvent>),
  64. SetSinkEventCallback(Option<SinkEventCallback>),
  65. EmitVolumeSetEvent(u16),
  66. }
  67. #[derive(Debug, Clone)]
  68. pub enum PlayerEvent {
  69. // Fired when the player is stopped (e.g. by issuing a "stop" command to the player).
  70. Stopped {
  71. play_request_id: u64,
  72. track_id: SpotifyId,
  73. },
  74. // The player started working on playback of a track while it was in a stopped state.
  75. // This is always immediately followed up by a "Loading" or "Playing" event.
  76. Started {
  77. play_request_id: u64,
  78. track_id: SpotifyId,
  79. position_ms: u32,
  80. },
  81. // Same as started but in the case that the player already had a track loaded.
  82. // The player was either playing the loaded track or it was paused.
  83. Changed {
  84. old_track_id: SpotifyId,
  85. new_track_id: SpotifyId,
  86. },
  87. // The player is delayed by loading a track.
  88. Loading {
  89. play_request_id: u64,
  90. track_id: SpotifyId,
  91. position_ms: u32,
  92. },
  93. // The player is playing a track.
  94. // This event is issued at the start of playback of whenever the position must be communicated
  95. // because it is out of sync. This includes:
  96. // start of a track
  97. // un-pausing
  98. // after a seek
  99. // after a buffer-underrun
  100. Playing {
  101. play_request_id: u64,
  102. track_id: SpotifyId,
  103. position_ms: u32,
  104. duration_ms: u32,
  105. },
  106. // The player entered a paused state.
  107. Paused {
  108. play_request_id: u64,
  109. track_id: SpotifyId,
  110. position_ms: u32,
  111. duration_ms: u32,
  112. },
  113. // The player thinks it's a good idea to issue a preload command for the next track now.
  114. // This event is intended for use within spirc.
  115. TimeToPreloadNextTrack {
  116. play_request_id: u64,
  117. track_id: SpotifyId,
  118. },
  119. // The player reached the end of a track.
  120. // This event is intended for use within spirc. Spirc will respond by issuing another command
  121. // which will trigger another event (e.g. Changed or Stopped)
  122. EndOfTrack {
  123. play_request_id: u64,
  124. track_id: SpotifyId,
  125. },
  126. // The player was unable to load the requested track.
  127. Unavailable {
  128. play_request_id: u64,
  129. track_id: SpotifyId,
  130. },
  131. // The mixer volume was set to a new level.
  132. VolumeSet {
  133. volume: u16,
  134. },
  135. }
  136. impl PlayerEvent {
  137. pub fn get_play_request_id(&self) -> Option<u64> {
  138. use PlayerEvent::*;
  139. match self {
  140. Loading {
  141. play_request_id, ..
  142. }
  143. | Unavailable {
  144. play_request_id, ..
  145. }
  146. | Started {
  147. play_request_id, ..
  148. }
  149. | Playing {
  150. play_request_id, ..
  151. }
  152. | TimeToPreloadNextTrack {
  153. play_request_id, ..
  154. }
  155. | EndOfTrack {
  156. play_request_id, ..
  157. }
  158. | Paused {
  159. play_request_id, ..
  160. }
  161. | Stopped {
  162. play_request_id, ..
  163. } => Some(*play_request_id),
  164. Changed { .. } | VolumeSet { .. } => None,
  165. }
  166. }
  167. }
  168. pub type PlayerEventChannel = futures::sync::mpsc::UnboundedReceiver<PlayerEvent>;
  169. #[derive(Clone, Copy, Debug)]
  170. struct NormalisationData {
  171. track_gain_db: f32,
  172. track_peak: f32,
  173. album_gain_db: f32,
  174. album_peak: f32,
  175. }
  176. impl NormalisationData {
  177. fn parse_from_file<T: Read + Seek>(mut file: T) -> Result<NormalisationData> {
  178. const SPOTIFY_NORMALIZATION_HEADER_START_OFFSET: u64 = 144;
  179. file.seek(SeekFrom::Start(SPOTIFY_NORMALIZATION_HEADER_START_OFFSET))
  180. .unwrap();
  181. let track_gain_db = file.read_f32::<LittleEndian>().unwrap();
  182. let track_peak = file.read_f32::<LittleEndian>().unwrap();
  183. let album_gain_db = file.read_f32::<LittleEndian>().unwrap();
  184. let album_peak = file.read_f32::<LittleEndian>().unwrap();
  185. let r = NormalisationData {
  186. track_gain_db: track_gain_db,
  187. track_peak: track_peak,
  188. album_gain_db: album_gain_db,
  189. album_peak: album_peak,
  190. };
  191. Ok(r)
  192. }
  193. fn get_factor(config: &PlayerConfig, data: NormalisationData) -> f32 {
  194. let mut normalisation_factor = f32::powf(
  195. 10.0,
  196. (data.track_gain_db + config.normalisation_pregain) / 20.0,
  197. );
  198. if normalisation_factor * data.track_peak > 1.0 {
  199. warn!("Reducing normalisation factor to prevent clipping. Please add negative pregain to avoid.");
  200. normalisation_factor = 1.0 / data.track_peak;
  201. }
  202. debug!("Normalisation Data: {:?}", data);
  203. debug!("Applied normalisation factor: {}", normalisation_factor);
  204. normalisation_factor
  205. }
  206. }
  207. impl Player {
  208. pub fn new<F>(
  209. config: PlayerConfig,
  210. session: Session,
  211. audio_filter: Option<Box<dyn AudioFilter + Send>>,
  212. sink_builder: F,
  213. ) -> (Player, PlayerEventChannel)
  214. where
  215. F: FnOnce() -> Box<dyn Sink> + Send + 'static,
  216. {
  217. let (cmd_tx, cmd_rx) = futures::sync::mpsc::unbounded();
  218. let (event_sender, event_receiver) = futures::sync::mpsc::unbounded();
  219. let handle = thread::spawn(move || {
  220. debug!("new Player[{}]", session.session_id());
  221. let internal = PlayerInternal {
  222. session: session,
  223. config: config,
  224. commands: cmd_rx,
  225. state: PlayerState::Stopped,
  226. preload: PlayerPreload::None,
  227. sink: sink_builder(),
  228. sink_status: SinkStatus::Closed,
  229. sink_event_callback: None,
  230. audio_filter: audio_filter,
  231. event_senders: [event_sender].to_vec(),
  232. };
  233. // While PlayerInternal is written as a future, it still contains blocking code.
  234. // It must be run by using wait() in a dedicated thread.
  235. let _ = internal.wait();
  236. debug!("PlayerInternal thread finished.");
  237. });
  238. (
  239. Player {
  240. commands: Some(cmd_tx),
  241. thread_handle: Some(handle),
  242. play_request_id_generator: SeqGenerator::new(0),
  243. },
  244. event_receiver,
  245. )
  246. }
  247. fn command(&self, cmd: PlayerCommand) {
  248. self.commands.as_ref().unwrap().unbounded_send(cmd).unwrap();
  249. }
  250. pub fn load(&mut self, track_id: SpotifyId, start_playing: bool, position_ms: u32) -> u64 {
  251. let play_request_id = self.play_request_id_generator.get();
  252. self.command(PlayerCommand::Load {
  253. track_id,
  254. play_request_id,
  255. play: start_playing,
  256. position_ms,
  257. });
  258. play_request_id
  259. }
  260. pub fn preload(&self, track_id: SpotifyId) {
  261. self.command(PlayerCommand::Preload { track_id });
  262. }
  263. pub fn play(&self) {
  264. self.command(PlayerCommand::Play)
  265. }
  266. pub fn pause(&self) {
  267. self.command(PlayerCommand::Pause)
  268. }
  269. pub fn stop(&self) {
  270. self.command(PlayerCommand::Stop)
  271. }
  272. pub fn seek(&self, position_ms: u32) {
  273. self.command(PlayerCommand::Seek(position_ms));
  274. }
  275. pub fn get_player_event_channel(&self) -> PlayerEventChannel {
  276. let (event_sender, event_receiver) = futures::sync::mpsc::unbounded();
  277. self.command(PlayerCommand::AddEventSender(event_sender));
  278. event_receiver
  279. }
  280. pub fn get_end_of_track_future(&self) -> Box<dyn Future<Item = (), Error = ()>> {
  281. let result = self
  282. .get_player_event_channel()
  283. .filter(|event| match event {
  284. PlayerEvent::EndOfTrack { .. } | PlayerEvent::Stopped { .. } => true,
  285. _ => false,
  286. })
  287. .into_future()
  288. .map_err(|_| ())
  289. .map(|_| ());
  290. Box::new(result)
  291. }
  292. pub fn set_sink_event_callback(&self, callback: Option<SinkEventCallback>) {
  293. self.command(PlayerCommand::SetSinkEventCallback(callback));
  294. }
  295. pub fn emit_volume_set_event(&self, volume: u16) {
  296. self.command(PlayerCommand::EmitVolumeSetEvent(volume));
  297. }
  298. }
  299. impl Drop for Player {
  300. fn drop(&mut self) {
  301. debug!("Shutting down player thread ...");
  302. self.commands = None;
  303. if let Some(handle) = self.thread_handle.take() {
  304. match handle.join() {
  305. Ok(_) => (),
  306. Err(_) => error!("Player thread panicked!"),
  307. }
  308. }
  309. }
  310. }
  311. struct PlayerLoadedTrackData {
  312. decoder: Decoder,
  313. normalisation_factor: f32,
  314. stream_loader_controller: StreamLoaderController,
  315. bytes_per_second: usize,
  316. duration_ms: u32,
  317. stream_position_pcm: u64,
  318. }
  319. enum PlayerPreload {
  320. None,
  321. Loading {
  322. track_id: SpotifyId,
  323. loader: Box<dyn Future<Item = PlayerLoadedTrackData, Error = ()>>,
  324. },
  325. Ready {
  326. track_id: SpotifyId,
  327. loaded_track: PlayerLoadedTrackData,
  328. },
  329. }
  330. type Decoder = VorbisDecoder<Subfile<AudioDecrypt<AudioFile>>>;
  331. enum PlayerState {
  332. Stopped,
  333. Loading {
  334. track_id: SpotifyId,
  335. play_request_id: u64,
  336. start_playback: bool,
  337. loader: Box<dyn Future<Item = PlayerLoadedTrackData, Error = ()>>,
  338. },
  339. Paused {
  340. track_id: SpotifyId,
  341. play_request_id: u64,
  342. decoder: Decoder,
  343. normalisation_factor: f32,
  344. stream_loader_controller: StreamLoaderController,
  345. bytes_per_second: usize,
  346. duration_ms: u32,
  347. stream_position_pcm: u64,
  348. suggested_to_preload_next_track: bool,
  349. },
  350. Playing {
  351. track_id: SpotifyId,
  352. play_request_id: u64,
  353. decoder: Decoder,
  354. normalisation_factor: f32,
  355. stream_loader_controller: StreamLoaderController,
  356. bytes_per_second: usize,
  357. duration_ms: u32,
  358. stream_position_pcm: u64,
  359. reported_nominal_start_time: Option<Instant>,
  360. suggested_to_preload_next_track: bool,
  361. },
  362. EndOfTrack {
  363. track_id: SpotifyId,
  364. play_request_id: u64,
  365. loaded_track: PlayerLoadedTrackData,
  366. },
  367. Invalid,
  368. }
  369. impl PlayerState {
  370. fn is_playing(&self) -> bool {
  371. use self::PlayerState::*;
  372. match *self {
  373. Stopped | EndOfTrack { .. } | Paused { .. } | Loading { .. } => false,
  374. Playing { .. } => true,
  375. Invalid => panic!("invalid state"),
  376. }
  377. }
  378. #[allow(dead_code)]
  379. fn is_stopped(&self) -> bool {
  380. use self::PlayerState::*;
  381. match *self {
  382. Stopped => true,
  383. _ => false,
  384. }
  385. }
  386. fn is_loading(&self) -> bool {
  387. use self::PlayerState::*;
  388. match *self {
  389. Loading { .. } => true,
  390. _ => false,
  391. }
  392. }
  393. fn decoder(&mut self) -> Option<&mut Decoder> {
  394. use self::PlayerState::*;
  395. match *self {
  396. Stopped | EndOfTrack { .. } | Loading { .. } => None,
  397. Paused {
  398. ref mut decoder, ..
  399. }
  400. | Playing {
  401. ref mut decoder, ..
  402. } => Some(decoder),
  403. Invalid => panic!("invalid state"),
  404. }
  405. }
  406. fn stream_loader_controller(&mut self) -> Option<&mut StreamLoaderController> {
  407. use self::PlayerState::*;
  408. match *self {
  409. Stopped | EndOfTrack { .. } | Loading { .. } => None,
  410. Paused {
  411. ref mut stream_loader_controller,
  412. ..
  413. }
  414. | Playing {
  415. ref mut stream_loader_controller,
  416. ..
  417. } => Some(stream_loader_controller),
  418. Invalid => panic!("invalid state"),
  419. }
  420. }
  421. fn playing_to_end_of_track(&mut self) {
  422. use self::PlayerState::*;
  423. match mem::replace(self, Invalid) {
  424. Playing {
  425. track_id,
  426. play_request_id,
  427. decoder,
  428. duration_ms,
  429. bytes_per_second,
  430. normalisation_factor,
  431. stream_loader_controller,
  432. stream_position_pcm,
  433. ..
  434. } => {
  435. *self = EndOfTrack {
  436. track_id,
  437. play_request_id,
  438. loaded_track: PlayerLoadedTrackData {
  439. decoder,
  440. duration_ms,
  441. bytes_per_second,
  442. normalisation_factor,
  443. stream_loader_controller,
  444. stream_position_pcm,
  445. },
  446. };
  447. }
  448. _ => panic!("Called playing_to_end_of_track in non-playing state."),
  449. }
  450. }
  451. fn paused_to_playing(&mut self) {
  452. use self::PlayerState::*;
  453. match ::std::mem::replace(self, Invalid) {
  454. Paused {
  455. track_id,
  456. play_request_id,
  457. decoder,
  458. normalisation_factor,
  459. stream_loader_controller,
  460. duration_ms,
  461. bytes_per_second,
  462. stream_position_pcm,
  463. suggested_to_preload_next_track,
  464. } => {
  465. *self = Playing {
  466. track_id,
  467. play_request_id,
  468. decoder,
  469. normalisation_factor,
  470. stream_loader_controller,
  471. duration_ms,
  472. bytes_per_second,
  473. stream_position_pcm,
  474. reported_nominal_start_time: None,
  475. suggested_to_preload_next_track,
  476. };
  477. }
  478. _ => panic!("invalid state"),
  479. }
  480. }
  481. fn playing_to_paused(&mut self) {
  482. use self::PlayerState::*;
  483. match ::std::mem::replace(self, Invalid) {
  484. Playing {
  485. track_id,
  486. play_request_id,
  487. decoder,
  488. normalisation_factor,
  489. stream_loader_controller,
  490. duration_ms,
  491. bytes_per_second,
  492. stream_position_pcm,
  493. reported_nominal_start_time: _,
  494. suggested_to_preload_next_track,
  495. } => {
  496. *self = Paused {
  497. track_id,
  498. play_request_id,
  499. decoder,
  500. normalisation_factor,
  501. stream_loader_controller,
  502. duration_ms,
  503. bytes_per_second,
  504. stream_position_pcm,
  505. suggested_to_preload_next_track,
  506. };
  507. }
  508. _ => panic!("invalid state"),
  509. }
  510. }
  511. }
  512. struct PlayerTrackLoader {
  513. session: Session,
  514. config: PlayerConfig,
  515. }
  516. impl PlayerTrackLoader {
  517. fn find_available_alternative<'a>(&self, audio: &'a AudioItem) -> Option<Cow<'a, AudioItem>> {
  518. if audio.available {
  519. Some(Cow::Borrowed(audio))
  520. } else {
  521. if let Some(alternatives) = &audio.alternatives {
  522. let alternatives = alternatives
  523. .iter()
  524. .map(|alt_id| AudioItem::get_audio_item(&self.session, *alt_id));
  525. let alternatives = future::join_all(alternatives).wait().unwrap();
  526. alternatives
  527. .into_iter()
  528. .find(|alt| alt.available)
  529. .map(Cow::Owned)
  530. } else {
  531. None
  532. }
  533. }
  534. }
  535. fn stream_data_rate(&self, format: FileFormat) -> usize {
  536. match format {
  537. FileFormat::OGG_VORBIS_96 => 12 * 1024,
  538. FileFormat::OGG_VORBIS_160 => 20 * 1024,
  539. FileFormat::OGG_VORBIS_320 => 40 * 1024,
  540. FileFormat::MP3_256 => 32 * 1024,
  541. FileFormat::MP3_320 => 40 * 1024,
  542. FileFormat::MP3_160 => 20 * 1024,
  543. FileFormat::MP3_96 => 12 * 1024,
  544. FileFormat::MP3_160_ENC => 20 * 1024,
  545. FileFormat::MP4_128_DUAL => 16 * 1024,
  546. FileFormat::OTHER3 => 40 * 1024, // better some high guess than nothing
  547. FileFormat::AAC_160 => 20 * 1024,
  548. FileFormat::AAC_320 => 40 * 1024,
  549. FileFormat::MP4_128 => 16 * 1024,
  550. FileFormat::OTHER5 => 40 * 1024, // better some high guess than nothing
  551. }
  552. }
  553. fn load_track(&self, spotify_id: SpotifyId, position_ms: u32) -> Option<PlayerLoadedTrackData> {
  554. let audio = match AudioItem::get_audio_item(&self.session, spotify_id).wait() {
  555. Ok(audio) => audio,
  556. Err(_) => {
  557. error!("Unable to load audio item.");
  558. return None;
  559. }
  560. };
  561. info!("Loading <{}> with Spotify URI <{}>", audio.name, audio.uri);
  562. let audio = match self.find_available_alternative(&audio) {
  563. Some(audio) => audio,
  564. None => {
  565. warn!("<{}> is not available", audio.uri);
  566. return None;
  567. }
  568. };
  569. assert!(audio.duration >= 0);
  570. let duration_ms = audio.duration as u32;
  571. // (Most) podcasts seem to support only 96 bit Vorbis, so fall back to it
  572. let formats = match self.config.bitrate {
  573. Bitrate::Bitrate96 => [
  574. FileFormat::OGG_VORBIS_96,
  575. FileFormat::OGG_VORBIS_160,
  576. FileFormat::OGG_VORBIS_320,
  577. ],
  578. Bitrate::Bitrate160 => [
  579. FileFormat::OGG_VORBIS_160,
  580. FileFormat::OGG_VORBIS_96,
  581. FileFormat::OGG_VORBIS_320,
  582. ],
  583. Bitrate::Bitrate320 => [
  584. FileFormat::OGG_VORBIS_320,
  585. FileFormat::OGG_VORBIS_160,
  586. FileFormat::OGG_VORBIS_96,
  587. ],
  588. };
  589. let format = formats
  590. .iter()
  591. .find(|format| audio.files.contains_key(format))
  592. .unwrap();
  593. let file_id = match audio.files.get(&format) {
  594. Some(&file_id) => file_id,
  595. None => {
  596. warn!("<{}> in not available in format {:?}", audio.name, format);
  597. return None;
  598. }
  599. };
  600. let bytes_per_second = self.stream_data_rate(*format);
  601. let play_from_beginning = position_ms == 0;
  602. let key = self.session.audio_key().request(spotify_id, file_id);
  603. let encrypted_file = AudioFile::open(
  604. &self.session,
  605. file_id,
  606. bytes_per_second,
  607. play_from_beginning,
  608. );
  609. let encrypted_file = match encrypted_file.wait() {
  610. Ok(encrypted_file) => encrypted_file,
  611. Err(_) => {
  612. error!("Unable to load encrypted file.");
  613. return None;
  614. }
  615. };
  616. let mut stream_loader_controller = encrypted_file.get_stream_loader_controller();
  617. if play_from_beginning {
  618. // No need to seek -> we stream from the beginning
  619. stream_loader_controller.set_stream_mode();
  620. } else {
  621. // we need to seek -> we set stream mode after the initial seek.
  622. stream_loader_controller.set_random_access_mode();
  623. }
  624. let key = match key.wait() {
  625. Ok(key) => key,
  626. Err(_) => {
  627. error!("Unable to load decryption key");
  628. return None;
  629. }
  630. };
  631. let mut decrypted_file = AudioDecrypt::new(key, encrypted_file);
  632. let normalisation_factor = match NormalisationData::parse_from_file(&mut decrypted_file) {
  633. Ok(normalisation_data) => {
  634. NormalisationData::get_factor(&self.config, normalisation_data)
  635. }
  636. Err(_) => {
  637. warn!("Unable to extract normalisation data, using default value.");
  638. 1.0 as f32
  639. }
  640. };
  641. let audio_file = Subfile::new(decrypted_file, 0xa7);
  642. let mut decoder = VorbisDecoder::new(audio_file).unwrap();
  643. if position_ms != 0 {
  644. match decoder.seek(position_ms as i64) {
  645. Ok(_) => (),
  646. Err(err) => error!("Vorbis error: {:?}", err),
  647. }
  648. stream_loader_controller.set_stream_mode();
  649. }
  650. let stream_position_pcm = PlayerInternal::position_ms_to_pcm(position_ms);
  651. info!("<{}> ({} ms) loaded", audio.name, audio.duration);
  652. Some(PlayerLoadedTrackData {
  653. decoder,
  654. normalisation_factor,
  655. stream_loader_controller,
  656. bytes_per_second,
  657. duration_ms,
  658. stream_position_pcm,
  659. })
  660. }
  661. }
  662. impl Future for PlayerInternal {
  663. type Item = ();
  664. type Error = ();
  665. fn poll(&mut self) -> Poll<(), ()> {
  666. // While this is written as a future, it still contains blocking code.
  667. // It must be run on its own thread.
  668. loop {
  669. let mut all_futures_completed_or_not_ready = true;
  670. // process commands that were sent to us
  671. let cmd = match self.commands.poll() {
  672. Ok(Async::Ready(None)) => return Ok(Async::Ready(())), // client has disconnected - shut down.
  673. Ok(Async::Ready(Some(cmd))) => {
  674. all_futures_completed_or_not_ready = false;
  675. Some(cmd)
  676. }
  677. Ok(Async::NotReady) => None,
  678. Err(_) => None,
  679. };
  680. if let Some(cmd) = cmd {
  681. self.handle_command(cmd);
  682. }
  683. // Handle loading of a new track to play
  684. if let PlayerState::Loading {
  685. ref mut loader,
  686. track_id,
  687. start_playback,
  688. play_request_id,
  689. } = self.state
  690. {
  691. match loader.poll() {
  692. Ok(Async::Ready(loaded_track)) => {
  693. self.start_playback(
  694. track_id,
  695. play_request_id,
  696. loaded_track,
  697. start_playback,
  698. );
  699. if let PlayerState::Loading { .. } = self.state {
  700. panic!("The state wasn't changed by start_playback()");
  701. }
  702. }
  703. Ok(Async::NotReady) => (),
  704. Err(_) => {
  705. warn!("Unable to load <{:?}>\nSkipping to next track", track_id);
  706. assert!(self.state.is_loading());
  707. self.send_event(PlayerEvent::EndOfTrack {
  708. track_id,
  709. play_request_id,
  710. })
  711. }
  712. }
  713. }
  714. // handle pending preload requests.
  715. if let PlayerPreload::Loading {
  716. ref mut loader,
  717. track_id,
  718. } = self.preload
  719. {
  720. match loader.poll() {
  721. Ok(Async::Ready(loaded_track)) => {
  722. self.preload = PlayerPreload::Ready {
  723. track_id,
  724. loaded_track,
  725. };
  726. }
  727. Ok(Async::NotReady) => (),
  728. Err(_) => {
  729. debug!("Unable to preload {:?}", track_id);
  730. self.preload = PlayerPreload::None;
  731. // Let Spirc know that the track was unavailable.
  732. if let PlayerState::Playing {
  733. play_request_id, ..
  734. }
  735. | PlayerState::Paused {
  736. play_request_id, ..
  737. } = self.state
  738. {
  739. self.send_event(PlayerEvent::Unavailable {
  740. track_id,
  741. play_request_id,
  742. });
  743. }
  744. }
  745. }
  746. }
  747. if self.state.is_playing() {
  748. self.ensure_sink_running();
  749. if let PlayerState::Playing {
  750. track_id,
  751. play_request_id,
  752. ref mut decoder,
  753. normalisation_factor,
  754. ref mut stream_position_pcm,
  755. ref mut reported_nominal_start_time,
  756. duration_ms,
  757. ..
  758. } = self.state
  759. {
  760. let packet = decoder.next_packet().expect("Vorbis error");
  761. if let Some(ref packet) = packet {
  762. *stream_position_pcm =
  763. *stream_position_pcm + (packet.data().len() / 2) as u64;
  764. let stream_position_millis = Self::position_pcm_to_ms(*stream_position_pcm);
  765. let notify_about_position = match *reported_nominal_start_time {
  766. None => true,
  767. Some(reported_nominal_start_time) => {
  768. // only notify if we're behind. If we're ahead it's probably due to a buffer of the backend and we;re actually in time.
  769. let lag = (Instant::now() - reported_nominal_start_time).as_millis()
  770. as i64
  771. - stream_position_millis as i64;
  772. if lag > 1000 {
  773. true
  774. } else {
  775. false
  776. }
  777. }
  778. };
  779. if notify_about_position {
  780. *reported_nominal_start_time = Some(
  781. Instant::now()
  782. - Duration::from_millis(stream_position_millis as u64),
  783. );
  784. self.send_event(PlayerEvent::Playing {
  785. track_id,
  786. play_request_id,
  787. position_ms: stream_position_millis as u32,
  788. duration_ms,
  789. });
  790. }
  791. }
  792. self.handle_packet(packet, normalisation_factor);
  793. } else {
  794. unreachable!();
  795. };
  796. }
  797. if let PlayerState::Playing {
  798. track_id,
  799. play_request_id,
  800. duration_ms,
  801. stream_position_pcm,
  802. ref mut stream_loader_controller,
  803. ref mut suggested_to_preload_next_track,
  804. ..
  805. }
  806. | PlayerState::Paused {
  807. track_id,
  808. play_request_id,
  809. duration_ms,
  810. stream_position_pcm,
  811. ref mut stream_loader_controller,
  812. ref mut suggested_to_preload_next_track,
  813. ..
  814. } = self.state
  815. {
  816. if (!*suggested_to_preload_next_track)
  817. && ((duration_ms as i64 - Self::position_pcm_to_ms(stream_position_pcm) as i64)
  818. < PRELOAD_NEXT_TRACK_BEFORE_END_DURATION_MS as i64)
  819. && stream_loader_controller.range_to_end_available()
  820. {
  821. *suggested_to_preload_next_track = true;
  822. self.send_event(PlayerEvent::TimeToPreloadNextTrack {
  823. track_id,
  824. play_request_id,
  825. });
  826. }
  827. }
  828. if self.session.is_invalid() {
  829. return Ok(Async::Ready(()));
  830. }
  831. if (!self.state.is_playing()) && all_futures_completed_or_not_ready {
  832. return Ok(Async::NotReady);
  833. }
  834. }
  835. }
  836. }
  837. impl PlayerInternal {
  838. fn position_pcm_to_ms(position_pcm: u64) -> u32 {
  839. (position_pcm * 10 / 441) as u32
  840. }
  841. fn position_ms_to_pcm(position_ms: u32) -> u64 {
  842. position_ms as u64 * 441 / 10
  843. }
  844. fn ensure_sink_running(&mut self) {
  845. if self.sink_status != SinkStatus::Running {
  846. trace!("== Starting sink ==");
  847. if let Some(callback) = &mut self.sink_event_callback {
  848. callback(SinkStatus::Running);
  849. }
  850. match self.sink.start() {
  851. Ok(()) => self.sink_status = SinkStatus::Running,
  852. Err(err) => error!("Could not start audio: {}", err),
  853. }
  854. }
  855. }
  856. fn ensure_sink_stopped(&mut self, temporarily: bool) {
  857. match self.sink_status {
  858. SinkStatus::Running => {
  859. trace!("== Stopping sink ==");
  860. self.sink.stop().unwrap();
  861. self.sink_status = if temporarily {
  862. SinkStatus::TemporarilyClosed
  863. } else {
  864. SinkStatus::Closed
  865. };
  866. if let Some(callback) = &mut self.sink_event_callback {
  867. callback(self.sink_status);
  868. }
  869. }
  870. SinkStatus::TemporarilyClosed => {
  871. if !temporarily {
  872. self.sink_status = SinkStatus::Closed;
  873. if let Some(callback) = &mut self.sink_event_callback {
  874. callback(SinkStatus::Closed);
  875. }
  876. }
  877. }
  878. SinkStatus::Closed => (),
  879. }
  880. }
  881. fn handle_player_stop(&mut self) {
  882. match self.state {
  883. PlayerState::Playing {
  884. track_id,
  885. play_request_id,
  886. ..
  887. }
  888. | PlayerState::Paused {
  889. track_id,
  890. play_request_id,
  891. ..
  892. }
  893. | PlayerState::EndOfTrack {
  894. track_id,
  895. play_request_id,
  896. ..
  897. }
  898. | PlayerState::Loading {
  899. track_id,
  900. play_request_id,
  901. ..
  902. } => {
  903. self.ensure_sink_stopped(false);
  904. self.send_event(PlayerEvent::Stopped {
  905. track_id,
  906. play_request_id,
  907. });
  908. self.state = PlayerState::Stopped;
  909. }
  910. PlayerState::Stopped => (),
  911. PlayerState::Invalid => panic!("invalid state"),
  912. }
  913. }
  914. fn handle_play(&mut self) {
  915. if let PlayerState::Paused {
  916. track_id,
  917. play_request_id,
  918. stream_position_pcm,
  919. duration_ms,
  920. ..
  921. } = self.state
  922. {
  923. self.state.paused_to_playing();
  924. let position_ms = Self::position_pcm_to_ms(stream_position_pcm);
  925. self.send_event(PlayerEvent::Playing {
  926. track_id,
  927. play_request_id,
  928. position_ms,
  929. duration_ms,
  930. });
  931. self.ensure_sink_running();
  932. } else {
  933. warn!("Player::play called from invalid state");
  934. }
  935. }
  936. fn handle_pause(&mut self) {
  937. if let PlayerState::Playing {
  938. track_id,
  939. play_request_id,
  940. stream_position_pcm,
  941. duration_ms,
  942. ..
  943. } = self.state
  944. {
  945. self.state.playing_to_paused();
  946. self.ensure_sink_stopped(false);
  947. let position_ms = Self::position_pcm_to_ms(stream_position_pcm);
  948. self.send_event(PlayerEvent::Paused {
  949. track_id,
  950. play_request_id,
  951. position_ms,
  952. duration_ms,
  953. });
  954. } else {
  955. warn!("Player::pause called from invalid state");
  956. }
  957. }
  958. fn handle_packet(&mut self, packet: Option<VorbisPacket>, normalisation_factor: f32) {
  959. match packet {
  960. Some(mut packet) => {
  961. if packet.data().len() > 0 {
  962. if let Some(ref editor) = self.audio_filter {
  963. editor.modify_stream(&mut packet.data_mut())
  964. };
  965. if self.config.normalisation && normalisation_factor != 1.0 {
  966. for x in packet.data_mut().iter_mut() {
  967. *x = (*x as f32 * normalisation_factor) as i16;
  968. }
  969. }
  970. if let Err(err) = self.sink.write(&packet.data()) {
  971. error!("Could not write audio: {}", err);
  972. self.ensure_sink_stopped(false);
  973. }
  974. }
  975. }
  976. None => {
  977. self.state.playing_to_end_of_track();
  978. if let PlayerState::EndOfTrack {
  979. track_id,
  980. play_request_id,
  981. ..
  982. } = self.state
  983. {
  984. self.send_event(PlayerEvent::EndOfTrack {
  985. track_id,
  986. play_request_id,
  987. })
  988. } else {
  989. unreachable!();
  990. }
  991. }
  992. }
  993. }
  994. fn start_playback(
  995. &mut self,
  996. track_id: SpotifyId,
  997. play_request_id: u64,
  998. loaded_track: PlayerLoadedTrackData,
  999. start_playback: bool,
  1000. ) {
  1001. let position_ms = Self::position_pcm_to_ms(loaded_track.stream_position_pcm);
  1002. if start_playback {
  1003. self.ensure_sink_running();
  1004. self.send_event(PlayerEvent::Playing {
  1005. track_id,
  1006. play_request_id,
  1007. position_ms,
  1008. duration_ms: loaded_track.duration_ms,
  1009. });
  1010. self.state = PlayerState::Playing {
  1011. track_id: track_id,
  1012. play_request_id: play_request_id,
  1013. decoder: loaded_track.decoder,
  1014. normalisation_factor: loaded_track.normalisation_factor,
  1015. stream_loader_controller: loaded_track.stream_loader_controller,
  1016. duration_ms: loaded_track.duration_ms,
  1017. bytes_per_second: loaded_track.bytes_per_second,
  1018. stream_position_pcm: loaded_track.stream_position_pcm,
  1019. reported_nominal_start_time: Some(
  1020. Instant::now() - Duration::from_millis(position_ms as u64),
  1021. ),
  1022. suggested_to_preload_next_track: false,
  1023. };
  1024. } else {
  1025. self.ensure_sink_stopped(false);
  1026. self.state = PlayerState::Paused {
  1027. track_id: track_id,
  1028. play_request_id: play_request_id,
  1029. decoder: loaded_track.decoder,
  1030. normalisation_factor: loaded_track.normalisation_factor,
  1031. stream_loader_controller: loaded_track.stream_loader_controller,
  1032. duration_ms: loaded_track.duration_ms,
  1033. bytes_per_second: loaded_track.bytes_per_second,
  1034. stream_position_pcm: loaded_track.stream_position_pcm,
  1035. suggested_to_preload_next_track: false,
  1036. };
  1037. self.send_event(PlayerEvent::Paused {
  1038. track_id,
  1039. play_request_id,
  1040. position_ms,
  1041. duration_ms: loaded_track.duration_ms,
  1042. });
  1043. }
  1044. }
  1045. fn handle_command_load(
  1046. &mut self,
  1047. track_id: SpotifyId,
  1048. play_request_id: u64,
  1049. play: bool,
  1050. position_ms: u32,
  1051. ) {
  1052. if !self.config.gapless {
  1053. self.ensure_sink_stopped(play);
  1054. }
  1055. // emit the correct player event
  1056. match self.state {
  1057. PlayerState::Playing {
  1058. track_id: old_track_id,
  1059. ..
  1060. }
  1061. | PlayerState::Paused {
  1062. track_id: old_track_id,
  1063. ..
  1064. }
  1065. | PlayerState::EndOfTrack {
  1066. track_id: old_track_id,
  1067. ..
  1068. }
  1069. | PlayerState::Loading {
  1070. track_id: old_track_id,
  1071. ..
  1072. } => self.send_event(PlayerEvent::Changed {
  1073. old_track_id: old_track_id,
  1074. new_track_id: track_id,
  1075. }),
  1076. PlayerState::Stopped => self.send_event(PlayerEvent::Started {
  1077. track_id,
  1078. play_request_id,
  1079. position_ms,
  1080. }),
  1081. PlayerState::Invalid { .. } => panic!("Player is in an invalid state."),
  1082. }
  1083. // Now we check at different positions whether we already have a pre-loaded version
  1084. // of this track somewhere. If so, use it and return.
  1085. // Check if there's a matching loaded track in the EndOfTrack player state.
  1086. // This is the case if we're repeating the same track again.
  1087. if let PlayerState::EndOfTrack {
  1088. track_id: previous_track_id,
  1089. ..
  1090. } = self.state
  1091. {
  1092. if previous_track_id == track_id {
  1093. let mut loaded_track = match mem::replace(&mut self.state, PlayerState::Invalid) {
  1094. PlayerState::EndOfTrack { loaded_track, .. } => loaded_track,
  1095. _ => unreachable!(),
  1096. };
  1097. if Self::position_ms_to_pcm(position_ms) != loaded_track.stream_position_pcm {
  1098. loaded_track
  1099. .stream_loader_controller
  1100. .set_random_access_mode();
  1101. let _ = loaded_track.decoder.seek(position_ms as i64); // This may be blocking.
  1102. // But most likely the track is fully
  1103. // loaded already because we played
  1104. // to the end of it.
  1105. loaded_track.stream_loader_controller.set_stream_mode();
  1106. loaded_track.stream_position_pcm = Self::position_ms_to_pcm(position_ms);
  1107. }
  1108. self.preload = PlayerPreload::None;
  1109. self.start_playback(track_id, play_request_id, loaded_track, play);
  1110. if let PlayerState::Invalid = self.state {
  1111. panic!("start_playback() hasn't set a valid player state.");
  1112. }
  1113. return;
  1114. }
  1115. }
  1116. // Check if we are already playing the track. If so, just do a seek and update our info.
  1117. if let PlayerState::Playing {
  1118. track_id: current_track_id,
  1119. ref mut stream_position_pcm,
  1120. ref mut decoder,
  1121. ref mut stream_loader_controller,
  1122. ..
  1123. }
  1124. | PlayerState::Paused {
  1125. track_id: current_track_id,
  1126. ref mut stream_position_pcm,
  1127. ref mut decoder,
  1128. ref mut stream_loader_controller,
  1129. ..
  1130. } = self.state
  1131. {
  1132. if current_track_id == track_id {
  1133. // we can use the current decoder. Ensure it's at the correct position.
  1134. if Self::position_ms_to_pcm(position_ms) != *stream_position_pcm {
  1135. stream_loader_controller.set_random_access_mode();
  1136. let _ = decoder.seek(position_ms as i64); // This may be blocking.
  1137. stream_loader_controller.set_stream_mode();
  1138. *stream_position_pcm = Self::position_ms_to_pcm(position_ms);
  1139. }
  1140. // Move the info from the current state into a PlayerLoadedTrackData so we can use
  1141. // the usual code path to start playback.
  1142. let old_state = mem::replace(&mut self.state, PlayerState::Invalid);
  1143. if let PlayerState::Playing {
  1144. stream_position_pcm,
  1145. decoder,
  1146. stream_loader_controller,
  1147. bytes_per_second,
  1148. duration_ms,
  1149. normalisation_factor,
  1150. ..
  1151. }
  1152. | PlayerState::Paused {
  1153. stream_position_pcm,
  1154. decoder,
  1155. stream_loader_controller,
  1156. bytes_per_second,
  1157. duration_ms,
  1158. normalisation_factor,
  1159. ..
  1160. } = old_state
  1161. {
  1162. let loaded_track = PlayerLoadedTrackData {
  1163. decoder,
  1164. normalisation_factor,
  1165. stream_loader_controller,
  1166. bytes_per_second,
  1167. duration_ms,
  1168. stream_position_pcm,
  1169. };
  1170. self.preload = PlayerPreload::None;
  1171. self.start_playback(track_id, play_request_id, loaded_track, play);
  1172. if let PlayerState::Invalid = self.state {
  1173. panic!("start_playback() hasn't set a valid player state.");
  1174. }
  1175. return;
  1176. } else {
  1177. unreachable!();
  1178. }
  1179. }
  1180. }
  1181. // Check if the requested track has been preloaded already. If so use the preloaded data.
  1182. if let PlayerPreload::Ready {
  1183. track_id: loaded_track_id,
  1184. ..
  1185. } = self.preload
  1186. {
  1187. if track_id == loaded_track_id {
  1188. let preload = std::mem::replace(&mut self.preload, PlayerPreload::None);
  1189. if let PlayerPreload::Ready {
  1190. track_id,
  1191. mut loaded_track,
  1192. } = preload
  1193. {
  1194. if Self::position_ms_to_pcm(position_ms) != loaded_track.stream_position_pcm {
  1195. loaded_track
  1196. .stream_loader_controller
  1197. .set_random_access_mode();
  1198. let _ = loaded_track.decoder.seek(position_ms as i64); // This may be blocking
  1199. loaded_track.stream_loader_controller.set_stream_mode();
  1200. }
  1201. self.start_playback(track_id, play_request_id, loaded_track, play);
  1202. return;
  1203. } else {
  1204. unreachable!();
  1205. }
  1206. }
  1207. }
  1208. // We need to load the track - either from scratch or by completing a preload.
  1209. // In any case we go into a Loading state to load the track.
  1210. self.ensure_sink_stopped(play);
  1211. self.send_event(PlayerEvent::Loading {
  1212. track_id,
  1213. play_request_id,
  1214. position_ms,
  1215. });
  1216. // Try to extract a pending loader from the preloading mechanism
  1217. let loader = if let PlayerPreload::Loading {
  1218. track_id: loaded_track_id,
  1219. ..
  1220. } = self.preload
  1221. {
  1222. if (track_id == loaded_track_id) && (position_ms == 0) {
  1223. let mut preload = PlayerPreload::None;
  1224. std::mem::swap(&mut preload, &mut self.preload);
  1225. if let PlayerPreload::Loading { loader, .. } = preload {
  1226. Some(loader)
  1227. } else {
  1228. None
  1229. }
  1230. } else {
  1231. None
  1232. }
  1233. } else {
  1234. None
  1235. };
  1236. self.preload = PlayerPreload::None;
  1237. // If we don't have a loader yet, create one from scratch.
  1238. let loader = loader
  1239. .or_else(|| Some(self.load_track(track_id, position_ms)))
  1240. .unwrap();
  1241. // Set ourselves to a loading state.
  1242. self.state = PlayerState::Loading {
  1243. track_id,
  1244. play_request_id,
  1245. start_playback: play,
  1246. loader,
  1247. };
  1248. }
  1249. fn handle_command_preload(&mut self, track_id: SpotifyId) {
  1250. debug!("Preloading track");
  1251. let mut preload_track = true;
  1252. // check whether the track is already loaded somewhere or being loaded.
  1253. if let PlayerPreload::Loading {
  1254. track_id: currently_loading,
  1255. ..
  1256. }
  1257. | PlayerPreload::Ready {
  1258. track_id: currently_loading,
  1259. ..
  1260. } = self.preload
  1261. {
  1262. if currently_loading == track_id {
  1263. // we're already preloading the requested track.
  1264. preload_track = false;
  1265. } else {
  1266. // we're preloading something else - cancel it.
  1267. self.preload = PlayerPreload::None;
  1268. }
  1269. }
  1270. if let PlayerState::Playing {
  1271. track_id: current_track_id,
  1272. ..
  1273. }
  1274. | PlayerState::Paused {
  1275. track_id: current_track_id,
  1276. ..
  1277. }
  1278. | PlayerState::EndOfTrack {
  1279. track_id: current_track_id,
  1280. ..
  1281. } = self.state
  1282. {
  1283. if current_track_id == track_id {
  1284. // we already have the requested track loaded.
  1285. preload_track = false;
  1286. }
  1287. }
  1288. // schedule the preload of the current track if desired.
  1289. if preload_track {
  1290. let loader = self.load_track(track_id, 0);
  1291. self.preload = PlayerPreload::Loading { track_id, loader }
  1292. }
  1293. }
  1294. fn handle_command_seek(&mut self, position_ms: u32) {
  1295. if let Some(stream_loader_controller) = self.state.stream_loader_controller() {
  1296. stream_loader_controller.set_random_access_mode();
  1297. }
  1298. if let Some(decoder) = self.state.decoder() {
  1299. match decoder.seek(position_ms as i64) {
  1300. Ok(_) => {
  1301. if let PlayerState::Playing {
  1302. ref mut stream_position_pcm,
  1303. ..
  1304. }
  1305. | PlayerState::Paused {
  1306. ref mut stream_position_pcm,
  1307. ..
  1308. } = self.state
  1309. {
  1310. *stream_position_pcm = Self::position_ms_to_pcm(position_ms);
  1311. }
  1312. }
  1313. Err(err) => error!("Vorbis error: {:?}", err),
  1314. }
  1315. } else {
  1316. warn!("Player::seek called from invalid state");
  1317. }
  1318. // If we're playing, ensure, that we have enough data leaded to avoid a buffer underrun.
  1319. if let Some(stream_loader_controller) = self.state.stream_loader_controller() {
  1320. stream_loader_controller.set_stream_mode();
  1321. }
  1322. // ensure we have a bit of a buffer of downloaded data
  1323. self.preload_data_before_playback();
  1324. if let PlayerState::Playing {
  1325. track_id,
  1326. play_request_id,
  1327. ref mut reported_nominal_start_time,
  1328. duration_ms,
  1329. ..
  1330. } = self.state
  1331. {
  1332. *reported_nominal_start_time =
  1333. Some(Instant::now() - Duration::from_millis(position_ms as u64));
  1334. self.send_event(PlayerEvent::Playing {
  1335. track_id,
  1336. play_request_id,
  1337. position_ms,
  1338. duration_ms,
  1339. });
  1340. }
  1341. if let PlayerState::Paused {
  1342. track_id,
  1343. play_request_id,
  1344. duration_ms,
  1345. ..
  1346. } = self.state
  1347. {
  1348. self.send_event(PlayerEvent::Paused {
  1349. track_id,
  1350. play_request_id,
  1351. position_ms,
  1352. duration_ms,
  1353. });
  1354. }
  1355. }
  1356. fn handle_command(&mut self, cmd: PlayerCommand) {
  1357. debug!("command={:?}", cmd);
  1358. match cmd {
  1359. PlayerCommand::Load {
  1360. track_id,
  1361. play_request_id,
  1362. play,
  1363. position_ms,
  1364. } => self.handle_command_load(track_id, play_request_id, play, position_ms),
  1365. PlayerCommand::Preload { track_id } => self.handle_command_preload(track_id),
  1366. PlayerCommand::Seek(position_ms) => self.handle_command_seek(position_ms),
  1367. PlayerCommand::Play => self.handle_play(),
  1368. PlayerCommand::Pause => self.handle_pause(),
  1369. PlayerCommand::Stop => self.handle_player_stop(),
  1370. PlayerCommand::AddEventSender(sender) => self.event_senders.push(sender),
  1371. PlayerCommand::SetSinkEventCallback(callback) => self.sink_event_callback = callback,
  1372. PlayerCommand::EmitVolumeSetEvent(volume) => {
  1373. self.send_event(PlayerEvent::VolumeSet { volume })
  1374. }
  1375. }
  1376. }
  1377. fn send_event(&mut self, event: PlayerEvent) {
  1378. let mut index = 0;
  1379. while index < self.event_senders.len() {
  1380. match self.event_senders[index].unbounded_send(event.clone()) {
  1381. Ok(_) => index += 1,
  1382. Err(_) => {
  1383. self.event_senders.remove(index);
  1384. }
  1385. }
  1386. }
  1387. }
  1388. fn load_track(
  1389. &self,
  1390. spotify_id: SpotifyId,
  1391. position_ms: u32,
  1392. ) -> Box<dyn Future<Item = PlayerLoadedTrackData, Error = ()>> {
  1393. // This method creates a future that returns the loaded stream and associated info.
  1394. // Ideally all work should be done using asynchronous code. However, seek() on the
  1395. // audio stream is implemented in a blocking fashion. Thus, we can't turn it into future
  1396. // easily. Instead we spawn a thread to do the work and return a one-shot channel as the
  1397. // future to work with.
  1398. let loader = PlayerTrackLoader {
  1399. session: self.session.clone(),
  1400. config: self.config.clone(),
  1401. };
  1402. let (result_tx, result_rx) = futures::sync::oneshot::channel();
  1403. std::thread::spawn(move || {
  1404. loader
  1405. .load_track(spotify_id, position_ms)
  1406. .and_then(move |data| {
  1407. let _ = result_tx.send(data);
  1408. Some(())
  1409. });
  1410. });
  1411. Box::new(result_rx.map_err(|_| ()))
  1412. }
  1413. fn preload_data_before_playback(&mut self) {
  1414. if let PlayerState::Playing {
  1415. bytes_per_second,
  1416. ref mut stream_loader_controller,
  1417. ..
  1418. } = self.state
  1419. {
  1420. // Request our read ahead range
  1421. let request_data_length = max(
  1422. (READ_AHEAD_DURING_PLAYBACK_ROUNDTRIPS
  1423. * (0.001 * stream_loader_controller.ping_time_ms() as f64)
  1424. * bytes_per_second as f64) as usize,
  1425. (READ_AHEAD_DURING_PLAYBACK_SECONDS * bytes_per_second as f64) as usize,
  1426. );
  1427. stream_loader_controller.fetch_next(request_data_length);
  1428. // Request the part we want to wait for blocking. This effecively means we wait for the previous request to partially complete.
  1429. let wait_for_data_length = max(
  1430. (READ_AHEAD_BEFORE_PLAYBACK_ROUNDTRIPS
  1431. * (0.001 * stream_loader_controller.ping_time_ms() as f64)
  1432. * bytes_per_second as f64) as usize,
  1433. (READ_AHEAD_BEFORE_PLAYBACK_SECONDS * bytes_per_second as f64) as usize,
  1434. );
  1435. stream_loader_controller.fetch_next_blocking(wait_for_data_length);
  1436. }
  1437. }
  1438. }
  1439. impl Drop for PlayerInternal {
  1440. fn drop(&mut self) {
  1441. debug!("drop PlayerInternal[{}]", self.session.session_id());
  1442. }
  1443. }
  1444. impl ::std::fmt::Debug for PlayerCommand {
  1445. fn fmt(&self, f: &mut ::std::fmt::Formatter) -> ::std::fmt::Result {
  1446. match *self {
  1447. PlayerCommand::Load {
  1448. track_id,
  1449. play,
  1450. position_ms,
  1451. ..
  1452. } => f
  1453. .debug_tuple("Load")
  1454. .field(&track_id)
  1455. .field(&play)
  1456. .field(&position_ms)
  1457. .finish(),
  1458. PlayerCommand::Preload { track_id } => {
  1459. f.debug_tuple("Preload").field(&track_id).finish()
  1460. }
  1461. PlayerCommand::Play => f.debug_tuple("Play").finish(),
  1462. PlayerCommand::Pause => f.debug_tuple("Pause").finish(),
  1463. PlayerCommand::Stop => f.debug_tuple("Stop").finish(),
  1464. PlayerCommand::Seek(position) => f.debug_tuple("Seek").field(&position).finish(),
  1465. PlayerCommand::AddEventSender(_) => f.debug_tuple("AddEventSender").finish(),
  1466. PlayerCommand::SetSinkEventCallback(_) => {
  1467. f.debug_tuple("SetSinkEventCallback").finish()
  1468. }
  1469. PlayerCommand::EmitVolumeSetEvent(volume) => {
  1470. f.debug_tuple("VolumeSet").field(&volume).finish()
  1471. }
  1472. }
  1473. }
  1474. }
  1475. impl ::std::fmt::Debug for PlayerState {
  1476. fn fmt(&self, f: &mut ::std::fmt::Formatter<'_>) -> ::std::fmt::Result {
  1477. use PlayerState::*;
  1478. match *self {
  1479. Stopped => f.debug_struct("Stopped").finish(),
  1480. Loading {
  1481. track_id,
  1482. play_request_id,
  1483. ..
  1484. } => f
  1485. .debug_struct("Loading")
  1486. .field("track_id", &track_id)
  1487. .field("play_request_id", &play_request_id)
  1488. .finish(),
  1489. Paused {
  1490. track_id,
  1491. play_request_id,
  1492. ..
  1493. } => f
  1494. .debug_struct("Paused")
  1495. .field("track_id", &track_id)
  1496. .field("play_request_id", &play_request_id)
  1497. .finish(),
  1498. Playing {
  1499. track_id,
  1500. play_request_id,
  1501. ..
  1502. } => f
  1503. .debug_struct("Playing")
  1504. .field("track_id", &track_id)
  1505. .field("play_request_id", &play_request_id)
  1506. .finish(),
  1507. EndOfTrack {
  1508. track_id,
  1509. play_request_id,
  1510. ..
  1511. } => f
  1512. .debug_struct("EndOfTrack")
  1513. .field("track_id", &track_id)
  1514. .field("play_request_id", &play_request_id)
  1515. .finish(),
  1516. Invalid => f.debug_struct("Invalid").finish(),
  1517. }
  1518. }
  1519. }
  1520. struct Subfile<T: Read + Seek> {
  1521. stream: T,
  1522. offset: u64,
  1523. }
  1524. impl<T: Read + Seek> Subfile<T> {
  1525. pub fn new(mut stream: T, offset: u64) -> Subfile<T> {
  1526. stream.seek(SeekFrom::Start(offset)).unwrap();
  1527. Subfile {
  1528. stream: stream,
  1529. offset: offset,
  1530. }
  1531. }
  1532. }
  1533. impl<T: Read + Seek> Read for Subfile<T> {
  1534. fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
  1535. self.stream.read(buf)
  1536. }
  1537. }
  1538. impl<T: Read + Seek> Seek for Subfile<T> {
  1539. fn seek(&mut self, mut pos: SeekFrom) -> Result<u64> {
  1540. pos = match pos {
  1541. SeekFrom::Start(offset) => SeekFrom::Start(offset + self.offset),
  1542. x => x,
  1543. };
  1544. let newpos = self.stream.seek(pos)?;
  1545. if newpos > self.offset {
  1546. Ok(newpos - self.offset)
  1547. } else {
  1548. Ok(0)
  1549. }
  1550. }
  1551. }