Write cue sheet to file

This commit is contained in:
Matteo Cypriani 2016-12-06 19:27:07 -05:00
parent 5973a06c1b
commit 54b5967411
2 changed files with 76 additions and 43 deletions

View File

@ -1,51 +1,30 @@
//! Module to extract the cue sheet from the NRG metadata.
use std::io::Write;
use std::fs::File;
use ::error::NrgError;
use ::metadata::metadata::NrgMetadata;
use ::metadata::cuex::NrgCuexTrack;
pub fn write_cue_sheet(image_name: &String, nm: &NrgMetadata)
/// Writes the cue sheet for `image_name` into a file.
///
/// `metadata` is the metadata extracted from `image_name` by nrgrip::metadata.
/// The output file's name is derived from `image_name`.
pub fn write_cue_sheet(image_name: &String, metadata: &NrgMetadata)
-> Result<(), NrgError> {
if nm.cuex_chunk.is_none() {
if metadata.cuex_chunk.is_none() {
return Err(NrgError::NoNrgCue);
}
let cuex_tracks = &nm.cuex_chunk.as_ref().unwrap().tracks;
let cuex_tracks = &metadata.cuex_chunk.as_ref().unwrap().tracks;
let file_name = try!(make_cue_sheet_name(image_name));
let file_name = make_cue_sheet_name(image_name);
let mut fd = try!(File::create(file_name));
println!("FILE \"{}\" RAW", image_name);
let mut index0_pos = -1;
for track in cuex_tracks {
// Ignore lead-in and lead-out areas
if track.track_number == 0 || track.track_number == 0xAA {
continue;
}
// Ignore negative positions
if track.position_sectors < 0 {
continue;
}
// Store/skip index0
if track.index_number == 0 {
index0_pos = track.position_sectors;
continue;
}
// Write track info
println!(" TRACK {:02} AUDIO", track.track_number);
// Write index0 if we stored it and it's before the current index's
// position (i.e., it indicates a pre-gap)
if index0_pos >= 0 && index0_pos < track.position_sectors {
try!(write_cue_index(0, index0_pos));
index0_pos = -1;
}
// Write current index
try!(write_cue_index(track.index_number, track.position_sectors));
}
// Write cue sheet
try!(writeln!(fd, "FILE \"{}\" RAW", image_name));
try!(write_cue_tracks(&mut fd, cuex_tracks));
Ok(())
}
@ -56,12 +35,66 @@ pub fn write_cue_sheet(image_name: &String, nm: &NrgMetadata)
/// If `image_name`'s extension is `.nrg` (case-insensitive), the cue sheet's
/// name will be the same as `image_name` with a `.cue` extension instead of
/// `.nrg`. Otherwise, the cue sheet's name will be `image_name.cue`.
fn make_cue_sheet_name(image_name: &String) -> Result<String, NrgError> {
Ok("image.cue".to_string())
fn make_cue_sheet_name(image_name: &String) -> String {
let mut name = image_name.clone();
if name.to_lowercase().ends_with(".nrg") {
let newlen = name.len() - 4;
name.truncate(newlen);
}
name.push_str(".cue");
name
}
fn write_cue_index(index: u8, position_sectors: i32)
/// Writes a list of cue tracks to `fd`.
fn write_cue_tracks(fd: &mut File, cuex_tracks: &Vec<NrgCuexTrack>)
-> Result<(), NrgError> {
let mut index0_pos = -1; // position of the last index #0 encountered
for track in cuex_tracks {
try!(write_cue_track(fd, track, &mut index0_pos));
}
Ok(())
}
/// Writes a cue track's info to `fd`.
///
/// `index0_pos` should be negative when this function is first called.
fn write_cue_track(fd: &mut File, track: &NrgCuexTrack, index0_pos: &mut i32)
-> Result<(), NrgError> {
// Ignore lead-in and lead-out areas
if track.track_number == 0 || track.track_number == 0xAA {
return Ok(());
}
// Ignore negative positions
if track.position_sectors < 0 {
return Ok(());
}
// Store/skip index0
if track.index_number == 0 {
*index0_pos = track.position_sectors;
return Ok(());
}
// Write track info
try!(writeln!(fd, " TRACK {:02} AUDIO", track.track_number));
// Write index0 if we stored it and it's before the current index's
// position (i.e., it indicates a pre-gap)
if *index0_pos >= 0 && *index0_pos < track.position_sectors {
try!(write_cue_index(fd, 0, *index0_pos));
*index0_pos = -1;
}
// Write current index
write_cue_index(fd, track.index_number, track.position_sectors)
}
/// Writes a cue index's info to `fd`.
fn write_cue_index(fd: &mut File, index: u8, position_sectors: i32)
-> Result<(), NrgError> {
assert!(position_sectors >= 0);
@ -71,8 +104,8 @@ fn write_cue_index(index: u8, position_sectors: i32)
let minutes = seconds / 60;
seconds %= 60;
println!(" INDEX {:02} {:02}:{:02}:{:02}",
index, minutes, seconds, remaining_sectors);
try!(writeln!(fd, " INDEX {:02} {:02}:{:02}:{:02}",
index, minutes, seconds, remaining_sectors));
Ok(())
}

View File

@ -6,7 +6,7 @@ use std::io::{Seek, SeekFrom};
use ::error::NrgError;
pub mod metadata;
mod cuex;
pub mod cuex;
mod daox;
mod sinf;
mod mtyp;