ot_tools_io/

lib.rs

/*
SPDX-License-Identifier: GPL-3.0-or-later
Copyright © 2024 Mike Robeson [dijksterhuis]
*/

//! Library crate for reading/writing data files for the [Elektron Octatrack][0].
//!
//! ```rust
//! // reading, mutating and writing a bank file
//!
//! use std::path::PathBuf;
//! use ot_tools_io::{OctatrackFileIO, BankFile};
//!
//! let path = PathBuf::from("test-data")
//!     .join("blank-project")
//!     .join("bank01.work");
//!
//! // read an editable version of the bank file
//! let mut bank = BankFile::from_data_file(&path).unwrap();
//!
//! // change active scenes on the working copy of Part 4
//! bank.parts.unsaved[3].active_scenes.scene_a = 2;
//! bank.parts.unsaved[3].active_scenes.scene_b = 6;
//!
//! // write to a new bank file
//! let outfpath = std::env::temp_dir()
//!     .join("ot-tools-io")
//!     .join("doctest")
//!     .join("main_example_1");
//!
//! # // when running in cicd env the /tmp/ot-tools-io directory doesn't exist yet
//! # let _ = std::fs::create_dir_all(outfpath.parent().unwrap());
//! bank.to_data_file(&outfpath).unwrap();
//! ```
//!
//! ## Brief Overview
//!
//! Important types and traits are re-exported or defined in the root of the crate's namespace for
//! your convenience -- everything you need to read and write Octatrack data files should be
//! available with `use ot_tools_io::*`. Less commonly used types and traits are public within their
//! specific modules and will require importing.
//! ```rust
//! // basic / common imports
//! use ot_tools_io::{SampleSettingsFile, OctatrackFileIO, HasHeaderField};
//!
//! // lower level / less common imports
//! use ot_tools_io::samples::{SAMPLES_HEADER, SAMPLES_FILE_VERSION, DEFAULT_GAIN, DEFAULT_TEMPO};
//! use ot_tools_io::slices::{Slice, Slices, SLICE_LOOP_POINT_DISABLED};
//! use ot_tools_io::parts::{Part, AudioTrackMachineParamsSetupPickup};
//! ```
//!
//! #### The `OctatrackFileIO` Trait
//! You'll usually want to import the [`OctatrackFileIO`] trait if you're reading or writing files.
//! It adds the associated functions & methods for file I/O, including for file I/O for YAML and
//! JSON files.
//! ```rust
//! // write a sample settings file to yaml and json
//!
//! use std::path::PathBuf;
//! use ot_tools_io::{SampleSettingsFile, OctatrackFileIO, HasHeaderField};
//!
//! let path = PathBuf::from("test-data")
//!     .join("samples")
//!     .join("sample.ot");
//!
//! let otfile = SampleSettingsFile::from_data_file(&path).unwrap();
//!
//! let outdir = std::env::temp_dir()
//!     .join("ot-tools-io")
//!     .join("doctest")
//!     .join("write_yaml_and_json");
//!
//! # // when running in cicd env the /tmp/ot-tools-io directory doesn't exist yet
//! # let _ = std::fs::create_dir_all(&outdir);
//! &otfile.to_yaml_file(&outdir.join("sample.yaml")).unwrap();
//! &otfile.to_json_file(&outdir.join("sample.json")).unwrap();
//! ```
//!
//! #### The `HasSomeField` Traits
//! The `Has*Field` traits add methods to a type to perform integrity checks: checksum calculation
//! and validation, header validation and file patch version validation.
//! ```rust
//! // check the header of sample settings file
//!
//! use std::path::PathBuf;
//! use ot_tools_io::{SampleSettingsFile, OctatrackFileIO, HasHeaderField};
//!
//! let path = PathBuf::from("test-data")
//!     .join("samples")
//!     .join("sample.ot");
//!
//! let otfile = SampleSettingsFile::from_data_file(&path).unwrap();
//! assert!(otfile.check_header().unwrap())
//! ```
//!
//! #### The `OtToolsIoError` Type
//! The `OtToolsIoError` type should be used to catch any errors in normal use as it implements
//! `From` for all other internal errors
//! ```rust
//! // handling errors
//!
//! use ot_tools_io::OtToolsIoError;
//! use ot_tools_io::settings::MidiChannel;
//!
//! // always errors
//! fn try_from_err() -> Result<MidiChannel, OtToolsIoError> {
//!     Ok(MidiChannel::try_from(100_i8)?)
//! }
//!
//! assert!(try_from_err().is_err());
//! assert_eq!(
//!     try_from_err().unwrap_err().to_string(),
//!     "invalid setting value: Invalid MIDI Channel value".to_string(),
//! );
//! ```
//!
//! #### `SomeFile` Types
//! Types directly related to some file used by the Octatrack are named as `SomeFile`, where
//! `Some` is the relevant file base name (`*.ot` files obviously don't have a basename we can
//! use).
//!
//! Only these `*File` types can be read from / written to the filesystem using this crate's
//! [`OctatrackFileIO`] trait methods / functions.
//!
//! | Type    | Filename Pattern | Description |
//! | ------------ | -------------------------- | ------------|
//! | [`ArrangementFile`] | `arr??.*` | data for arrangements |
//! | [`BankFile`] | `bank??.*` | data for parts and patterns |
//! | [`MarkersFile`] | `markers.*` | start trim/end trim/slices/loop points for sample slots |
//! | [`ProjectFile`] | `project.*` | project level settings; state; sample slots |
//! | [`SampleSettingsFile`] | `*.ot` | saved sample settings data, loops slices etc. |
//!
//! Read the relevant modules in this library for more detailed information on
//! the data contained in each file.
//!
//! ####
//!
//! ## Additional Details
//!
//! #### Octatrack File Relationships
//!
//! - Changing the sample loaded into a sample slot updates both the [`ProjectFile`] file (trig
//!   quantization settings, file path etc) and the [`MarkersFile`] file (trim settings, slices,
//!   loop points).
//!
//! - Slot data from [`ProjectFile`]s and [`MarkersFile`]s is written to an [`SampleSettingsFile`]
//!   file when saving sample attributes data from the Octatrack's audio editing menu.
//!
//! - Loading a sample into a project sample slot ([`ProjectFile`]s and [`MarkersFile`]s) reads
//!   any data in an [`SampleSettingsFile`] files and configures the sample slot accordingly.
//!
//! - A [`BankFile`]'s patterns and parts store zero-indexed sample slot IDs as flex
//!   and static slot references (machine data in a part and track p-lock trigs in a pattern). These
//!   references point to the relevant slot in **_both_** the [`MarkersFile`] and [`ProjectFile`]
//!   for a project.
//!
//! - [`ArrangementFile`]s store a `u8` which references a [`BankFile`]'s pattern, indicating
//!   the pattern should be played when the specific row in an arrangement is triggered.
//!   The field is zero-indexed, with the full range used for pattern references 0 (A01) ->
//!   256 (P16).
//!
//!
//! #### Octatrack Device File Modifications
//!
//! - `*.work` files are created when creating a new project
//!   `PROJECT MENU -> CHANGE PROJECT -> CREATE NEW`.
//! - `*.work` files are updated by using the `PROJECT MENU -> SYNC TO CARD` operation.
//! - `*.work` files are updated when the user performs a `PROJECT MENU -> SAVE PROJECT` operation.
//! - `*.strd` files are created/updated when the user performs a `PROJECT MENU -> SAVE PROJECT`
//!   operation.
//! - `*.strd` files are not changed by the `PROJECT -> SYNC TO CARD` operation.
//! - `arr??.strd` files can also be saved via the `ARRANGER MENU -> SAVE ARRANGEMENT` operation.
//!
//! #### Notable 'gotcha's
//!
//! ##### Sample Slot IDs
//!
//! Project Sample Slots store their slots with a **_one-indexed_** `slot_id` field.
//!
//! **_All other references to sample slots are zero-indexed_** (i.e. [`BankFile`] and
//! [`MarkersFile`] files).
//!
//! I'm using lots of italics and bold formatting here because it is super annoying but there's not
//! much i can do about it. Remember you will need to convert from one to zero indexed whenever you
//! deal with Sample Slots.
//!
//! ##### Default Loop Point Values Differ Between Types
//!
//! A 'Disabled' loop point in either a [`MarkersFile`] or a [`SampleSettingsFile`] is a
//! `0xFFFFFFFF` value, but the default loop point when creating new [`MarkersFile`] is always
//! `0_u32`. The 'Disabled' value setting is only set when a sample is loaded into a sample slot,
//! and a [`SampleSettingsFile`] is generated from that sample slot data.
//!
//! ####
// For Andrey.
//! ## Slava Ukraini
//!
//! I've worked with Ukrainian developers. What is happening to their country is abhorrent.
//!
//! [![Stand With Ukraine](https://raw.githubusercontent.com/vshymanskyy/StandWithUkraine/main/banner2-direct.svg)](https://stand-with-ukraine.pp.ua)
//! ##
//!
//! [0]: https://www.elektron.se/explore/octatrack-mkii
//!

pub mod arrangements;
pub mod banks;
pub mod errors;
pub mod markers;
pub mod parts;
pub mod patterns;
pub mod projects;
pub mod samples;
pub mod settings;
pub mod slices;
#[cfg(test)]
#[allow(dead_code)]
mod test_utils;
mod traits;

pub use crate::arrangements::ArrangementFile;
pub use crate::banks::BankFile;
pub use crate::markers::MarkersFile;
pub use crate::projects::ProjectFile;
pub use crate::samples::SampleSettingsFile;

pub use crate::traits::{
    CheckFileIntegrity, Defaults, HasChecksumField, HasFileVersionField, HasHeaderField, IsDefault,
    OctatrackFileIO,
};

use crate::markers::MarkersError;
use crate::projects::{ProjectError, ProjectParseError};
use crate::samples::SampleSettingsError;
use crate::settings::InvalidValueError;
use crate::slices::SliceError;
use std::fs::File;
use std::io::{Read, Write};
use std::path::{Path, PathBuf};
use thiserror::Error;

/// Global error variant handling. All internally used error types and variants can cast to this
/// type.
#[derive(Debug, Error)]
pub enum OtToolsIoError {
    /// File could not be found / opened etc.
    #[error("File OS Error: {source} Path={path} (std::io::Error)")]
    FileOs {
        #[source]
        source: std::io::Error,
        path: PathBuf,
    },
    /// Some other FIle I/O error
    #[error("File I/O Error: {0} (std::io::Error)")]
    FileIo(#[from] std::io::Error),
    /// Bincode could not de/serialize the data
    #[error("error during binary data decoding / encoding: {0} (bincode::Error)")]
    Bincode(#[from] bincode::Error),
    /// Can't read project string data
    #[error("error reading raw project data: {0} (std::str::Utf8Error)")]
    ProjectRead(#[from] std::str::Utf8Error),
    /// Can't parse project string data
    #[error("error parsing raw project data: {0} (ot_tools_io::projects::ProjectParseError)")]
    ProjectParse(#[from] ProjectParseError),
    /// Can't parse yaml data
    #[error("error processing yaml: {0} (serde_yml::Error)")]
    YamlParse(#[from] serde_yml::Error),
    /// Can't parse json data
    #[error("error processing json: {0} (serde_json::Error)")]
    JsonParse(#[from] serde_json::Error),
    /// [projects] specific errors
    #[error(
        "project files cannot be checked for integrity: {0} (ot_tools_io::projects::ProjectError)"
    )]
    ProjectFile(#[from] ProjectError),
    /// [samples] specific errors
    #[error("sample settings error: {0} (ot_tools_io::samples::SampleSettingsError)")]
    SampleSettings(#[from] SampleSettingsError),
    /// [markers] specific errors
    #[error("markers error: {0} (ot_tools_io::markers::MarkersErrors)")]
    Markers(#[from] MarkersError),
    /// [slices] specific errors
    #[error("slices error: {0} (ot_tools_io::slices::SlicesErrors)")]
    Slice(#[from] SliceError),
    /// [settings] error handling
    #[error("invalid setting value: {0}")]
    SettingValue(#[from] InvalidValueError),
    /// Header for some file type is invalid
    #[error("invalid header(s) for file")]
    FileHeader,
}

fn u8_bytes_to_u16(bytes: &[u8; 2]) -> u16 {
    ((bytes[0] as u16) << 8) | bytes[1] as u16
}

#[doc(hidden)]
/// Read bytes from a file at `path`.
// ```rust
// let fpath = std::path::PathBuf::from("test-data")
//     .join("blank-project")
//     .join("bank01.work");
// let r = ot_tools_io::read_bin_file(&fpath);
// assert!(r.is_ok());
// assert_eq!(r.unwrap().len(), 636113);
//```
fn read_bin_file(path: &Path) -> Result<Vec<u8>, OtToolsIoError> {
    let mut infile = File::open(path).map_err(|e| OtToolsIoError::FileOs {
        path: path.to_path_buf(),
        source: e,
    })?;
    let mut bytes: Vec<u8> = vec![];
    let _: usize = infile.read_to_end(&mut bytes)?;
    Ok(bytes)
}

#[cfg(test)]
mod read_bin_file {
    use crate::test_utils::*;
    use crate::{read_bin_file, OtToolsIoError};

    #[test]
    fn ok() -> Result<(), OtToolsIoError> {
        let path = get_blank_proj_dirpath().join("bank01.work");
        read_bin_file(&path)?;
        Ok(())
    }

    #[test]
    fn err_file_io() -> Result<(), OtToolsIoError> {
        let path = get_blank_proj_dirpath().join("NOTNTONTONTNTOTNONT");

        #[cfg(target_os = "windows")]
        assert_eq!(
            read_bin_file(&path).unwrap_err().to_string(),
            format!["File OS Error: The system cannot find the file specified. (os error 2) Path={} (std::io::Error)", path.display()].to_string(),
            "should throw a OtToolsIoError::FileOS error when file does not exist"
        );

        #[cfg(target_os = "linux")]
        assert_eq!(
            read_bin_file(&path).unwrap_err().to_string(),
            format![
                "File OS Error: No such file or directory (os error 2) Path={} (std::io::Error)",
                path.display()
            ]
            .to_string(),
            "should throw a OtToolsIoError::FileOS error when file does not exist"
        );

        Ok(())
    }
}

#[doc(hidden)]
/// Write bytes to a file at `path`.
// ```rust
// use std::env::temp_dir;
// use std::array::from_fn;
//
// let arr: [u8; 27] = from_fn(|_| 0);
//
// let fpath = temp_dir()
//    .join("ot-tools-io")
//    .join("doctest")
//    .join("write_bin_file.example");
//
// # use std::fs::create_dir_all;
// # create_dir_all(&fpath.parent().unwrap()).unwrap();
// let r = ot_tools_io::write_bin_file(&arr, &fpath);
// assert!(r.is_ok());
// assert!(fpath.exists());
// ```
fn write_bin_file(bytes: &[u8], path: &Path) -> Result<(), OtToolsIoError> {
    let mut file: File = File::create(path).map_err(|e| OtToolsIoError::FileOs {
        path: path.to_path_buf(),
        source: e,
    })?;
    file.write_all(bytes)?;
    Ok(())
}

#[cfg(test)]
mod write_bin_file {
    use crate::{write_bin_file, OtToolsIoError};
    use std::env::temp_dir;
    use std::fs::{create_dir_all, remove_file};

    #[test]
    fn ok() -> Result<(), OtToolsIoError> {
        let path = temp_dir()
            .join("ot-tools-io")
            .join("write_bin_file")
            .join("ok.bin");
        create_dir_all(path.parent().unwrap())?;
        if path.exists() {
            remove_file(&path)?;
        };
        write_bin_file(&[1, 2, 3, 4], &path)?;
        Ok(())
    }

    // should fail: attempting to write a file to an existing directory
    #[test]
    fn err_file_io() -> Result<(), OtToolsIoError> {
        let path = temp_dir().join("ot-tools-io").join("write_bin_file");
        create_dir_all(&path)?;

        #[cfg(target_os = "windows")]
        assert_eq!(
            write_bin_file(&[1, 2, 3, 4], &path)
                .unwrap_err()
                .to_string(),
            format![
                "File OS Error: Access is denied. (os error 5) Path={} (std::io::Error)",
                path.display()
            ]
            .to_string(),
            "should throw a OtToolsIoError::FileOS error when a file cannot be created"
        );

        #[cfg(target_os = "linux")]
        assert_eq!(
            write_bin_file(&[1, 2, 3, 4], &path)
                .unwrap_err()
                .to_string(),
            format![
                "File OS Error: Is a directory (os error 21) Path={} (std::io::Error)",
                path.display()
            ]
            .to_string(),
            "should throw a OtToolsIoError::FileOS error when a file cannot be created"
        );
        Ok(())
    }
}

#[doc(hidden)]
/// Read a file at `path` as a string.
// ```
// let fpath = std::path::PathBuf::from("test-data")
//     .join("blank-project")
//     .join("bank01.work");
// let r = ot_tools_io::read_bin_file(&fpath);
// assert!(r.is_ok());
// assert_eq!(r.unwrap().len(), 636113);
// ```
fn read_str_file(path: &Path) -> Result<String, OtToolsIoError> {
    let mut file = File::open(path).map_err(|e| OtToolsIoError::FileOs {
        path: path.to_path_buf(),
        source: e,
    })?;
    let mut string = String::new();
    let _ = file.read_to_string(&mut string)?;
    Ok(string)
}

#[cfg(test)]
mod read_str_file {
    use crate::test_utils::*;
    use crate::{read_str_file, OtToolsIoError};

    #[test]
    fn ok() -> Result<(), OtToolsIoError> {
        let path = get_blank_proj_dirpath().join("project.work");
        read_str_file(&path)?;
        Ok(())
    }

    #[test]
    fn err_file_io() -> Result<(), OtToolsIoError> {
        let path = get_blank_proj_dirpath().join("NOTNTONTONTNTOTNONT");

        #[cfg(target_os = "windows")]
        assert_eq!(
            read_str_file(&path).unwrap_err().to_string(),
            format!["File OS Error: The system cannot find the file specified. (os error 2) Path={} (std::io::Error)", path.display()].to_string(),
            "should throw a OtToolsIoError::FileOS error when file does not exist"
        );

        #[cfg(target_os = "linux")]
        assert_eq!(
            read_str_file(&path).unwrap_err().to_string(),
            format![
                "File OS Error: No such file or directory (os error 2) Path={} (std::io::Error)",
                path.display()
            ]
            .to_string(),
            "should throw a OtToolsIoError::FileOS error when file does not exist"
        );

        Ok(())
    }
}

#[doc(hidden)]
/// Write a string to a file at `path`.
// ```rust
// use std::env::temp_dir;
//
// let data = "abcd".to_string();
//
// let fpath = temp_dir()
//    .join("ot-tools-io")
//    .join("doctest")
//    .join("write_str_file.example");
//
// # use std::fs::create_dir_all;
// # create_dir_all(&fpath.parent().unwrap()).unwrap();
// let r = ot_tools_io::write_str_file(&data, &fpath);
// assert!(r.is_ok());
// assert!(fpath.exists());
// ```
fn write_str_file(string: &str, path: &Path) -> Result<(), OtToolsIoError> {
    let mut file: File = File::create(path).map_err(|e| OtToolsIoError::FileOs {
        path: path.to_path_buf(),
        source: e,
    })?;
    write!(file, "{string}")?;
    Ok(())
}

#[cfg(test)]
mod write_str_file {
    use crate::{write_str_file, OtToolsIoError};
    use std::env::temp_dir;
    use std::fs::{create_dir_all, remove_file};

    #[test]
    fn ok() -> Result<(), OtToolsIoError> {
        let path = temp_dir()
            .join("ot-tools-io")
            .join("write_str_file")
            .join("ok.txt");
        create_dir_all(path.parent().unwrap())?;
        if path.exists() {
            remove_file(&path)?;
        };
        write_str_file("SOMETHING", &path)?;
        Ok(())
    }

    // should fail: attempting to write a file to an existing directory
    #[test]
    fn err_file_io() -> Result<(), OtToolsIoError> {
        let path = temp_dir().join("ot-tools-io").join("write_str_file");
        create_dir_all(&path)?;

        #[cfg(target_os = "windows")]
        assert_eq!(
            write_str_file("SOMETHING", &path).unwrap_err().to_string(),
            format![
                "File OS Error: Access is denied. (os error 5) Path={} (std::io::Error)",
                path.display()
            ]
            .to_string(),
            "should throw a OtToolsIoError::FileOS error when a file cannot be created"
        );

        #[cfg(target_os = "linux")]
        assert_eq!(
            write_str_file("SOMETHING", &path).unwrap_err().to_string(),
            format![
                "File OS Error: Is a directory (os error 21) Path={} (std::io::Error)",
                path.display()
            ]
            .to_string(),
            "should throw a OtToolsIoError::FileOS error when a file cannot be created"
        );
        Ok(())
    }
}

fn loop_point_is_disabled(loop_point: u32) -> bool {
    loop_point == slices::SLICE_LOOP_POINT_DISABLED
}

fn loop_point_is_default(loop_point: u32) -> bool {
    loop_point == 0
}

fn loop_point_is_in_trim_range(loop_point: u32, trim_start: u32, trim_end: u32) -> bool {
    loop_point >= trim_start && loop_point < trim_end
}

fn check_loop_point(loop_point: u32, trim_start: u32, trim_end: u32) -> bool {
    loop_point_is_in_trim_range(loop_point, trim_start, trim_end)
        || loop_point_is_disabled(loop_point)
        || loop_point_is_default(loop_point)
}