vpx_read_extract_assemble_write_compare_all.rs

mod common;

#[cfg(test)]
mod test {

    use pretty_assertions::assert_eq;
    // use rayon::prelude::*;
    use crate::common::{assert_equal_vpx, find_files};
    use std::io;
    use std::path::{Path, PathBuf};
    use testdir::testdir;

    #[test]
    #[ignore = "slow integration test that only runs on correctly set up machines"]
    fn read_extract_assemble_and_write_all() -> io::Result<()> {
        let home = dirs::home_dir().expect("no home dir");
        let folder = home.join("vpinball").join("tables");
        if !folder.exists() {
            panic!("folder does not exist: {:?}", folder);
        }
        let paths = find_files(&folder, "vpx")?;
        // testdir can not be used in non-main threads
        let dir: PathBuf = testdir!();

        // Example tables caused problems in the past:
        //
        // * Inhabiting Mars RC 4 - for animation frame
        // * DieHard_272.vpx - primitive "BM_pAirDuctGate" has a NaN value for nx
        // * Johnny Mnemonic (Williams 1995) VPW v1.0.2.vpx - animated frames that overlap with primitive names
        // * Future Spa (Bally 1979) v4.3.vpx - NaN in table setup values
        // * InvaderTable_2.260.vpx - Symbol fonts
        // * Guns N Roses (Data East 1994).vpx - contains BMP with non-255 alpha values
        let filtered: Vec<&PathBuf> = paths
            .iter()
            // .filter(|path| {
            //     let name = path.file_name().unwrap().to_str().unwrap();
            //     name.contains("Bob")
            // })
            .collect();

        // TODO why is par_iter() not faster but just consuming all cpu cores?
        filtered.iter().enumerate().try_for_each(|(n, path)| {
            println!("testing {}/{}: {:?}", n + 1, filtered.len(), path);
            let ReadAndWriteResult {
                extracted,
                test_vpx,
            } = read_and_write_vpx(&dir, path)?;
            assert_equal_vpx(path, test_vpx.clone());
            // panic!("stop");
            // if all is good we remove the test file and the extracted dir
            std::fs::remove_file(&test_vpx)?;
            std::fs::remove_dir_all(extracted)?;
            Ok(())
        })
    }

    struct ReadAndWriteResult {
        extracted: PathBuf,
        test_vpx: PathBuf,
    }

    fn read_and_write_vpx(dir: &Path, path: &Path) -> io::Result<ReadAndWriteResult> {
        let original = vpin::vpx::read(&path.to_path_buf())?;
        let extract_dir = dir.join("extracted");
        // make dir
        std::fs::create_dir_all(&extract_dir)?;
        vpin::vpx::expanded::write(&original, &extract_dir)
            .map_err(|e| io::Error::new(io::ErrorKind::Other, e))?;
        let expanded_read = vpin::vpx::expanded::read(&extract_dir)
            .map_err(|e| io::Error::new(io::ErrorKind::Other, e))?;
        // special case for comparing code
        assert_eq!(original.gamedata.code, expanded_read.gamedata.code);
        let file_name = path.file_name().unwrap();
        let test_vpx_path = dir.join(file_name);
        vpin::vpx::write(&test_vpx_path, &expanded_read)?;
        Ok(ReadAndWriteResult {
            extracted: extract_dir,
            test_vpx: test_vpx_path,
        })
    }
}