swf-parser 0.14.0

use nom::number::complete::{le_u16 as parse_le_u16, le_u32 as parse_le_u32, le_u8 as parse_u8};
use nom::IResult as NomResult;
use swf_types as swf;

pub fn sound_rate_from_code(sound_rate_code: u8) -> Result<swf::SoundRate, ()> {
  match sound_rate_code {
    0 => Ok(swf::SoundRate::SoundRate5500),
    1 => Ok(swf::SoundRate::SoundRate11000),
    2 => Ok(swf::SoundRate::SoundRate22000),
    3 => Ok(swf::SoundRate::SoundRate44000),
    _ => Err(()),
  }
}

pub fn audio_coding_format_from_code(audio_codec_code: u8) -> Result<swf::AudioCodingFormat, ()> {
  match audio_codec_code {
    0 => Ok(swf::AudioCodingFormat::UncompressedNativeEndian),
    1 => Ok(swf::AudioCodingFormat::Adpcm),
    2 => Ok(swf::AudioCodingFormat::Mp3),
    3 => Ok(swf::AudioCodingFormat::UncompressedLittleEndian),
    4 => Ok(swf::AudioCodingFormat::Nellymoser16),
    5 => Ok(swf::AudioCodingFormat::Nellymoser8),
    6 => Ok(swf::AudioCodingFormat::Nellymoser),
    11 => Ok(swf::AudioCodingFormat::Speex),
    _ => Err(()),
  }
}

// TODO: Move this to a method on `swf::AudioCodingFormat`
pub fn is_uncompressed_audio_coding_format(format: swf::AudioCodingFormat) -> bool {
  matches!(format, swf::AudioCodingFormat::UncompressedNativeEndian | swf::AudioCodingFormat::UncompressedLittleEndian)
}

pub fn parse_sound_info(input: &[u8]) -> NomResult<&[u8], swf::SoundInfo> {
  use nom::combinator::cond;
  use nom::multi::count;

  let (input, flags) = parse_u8(input)?;
  #[allow(clippy::identity_op)]
  let has_in_point = (flags & (1 << 0)) != 0;
  let has_out_point = (flags & (1 << 1)) != 0;
  let has_loops = (flags & (1 << 2)) != 0;
  let has_envelope = (flags & (1 << 3)) != 0;
  let sync_no_multiple = (flags & (1 << 4)) != 0;
  let sync_stop = (flags & (1 << 5)) != 0;
  // Bits [6, 7] are reserved
  let (input, in_point) = cond(has_in_point, parse_le_u32)(input)?;
  let (input, out_point) = cond(has_out_point, parse_le_u32)(input)?;
  let (input, loop_count) = cond(has_loops, parse_le_u16)(input)?;
  let (input, envelope_records) = if has_envelope {
    let (input, record_count) = parse_u8(input)?;
    let (input, envelope_records) = count(parse_sound_envelope, usize::from(record_count))(input)?;
    (input, Some(envelope_records))
  } else {
    (input, None)
  };

  Ok((
    input,
    swf::SoundInfo {
      sync_stop,
      sync_no_multiple,
      in_point,
      out_point,
      loop_count,
      envelope_records,
    },
  ))
}

pub fn parse_sound_envelope(input: &[u8]) -> NomResult<&[u8], swf::SoundEnvelope> {
  let (input, pos44) = parse_le_u32(input)?;
  let (input, left_level) = parse_le_u16(input)?;
  let (input, right_level) = parse_le_u16(input)?;
  Ok((
    input,
    swf::SoundEnvelope {
      pos44,
      left_level,
      right_level,
    },
  ))
}