Skip to content

FEAT: dummy implementaiton for a timstofdata object #41

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: develop
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 14 additions & 14 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

7 changes: 6 additions & 1 deletion src/errors.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,10 @@
use crate::io::readers::{
FrameReaderError, MetadataReaderError, QuadrupoleSettingsReaderError,
};
use crate::{io::readers::PrecursorReaderError, readers::SpectrumReaderError};
use crate::{
io::readers::PrecursorReaderError,
readers::{SpectrumReaderError, TimsTofDataError},
};

/// An error that is produced by timsrust (uses [thiserror]).
#[derive(thiserror::Error, Debug)]
Expand All @@ -20,4 +23,6 @@ pub enum TimsRustError {
#[cfg(feature = "tdf")]
#[error("{0}")]
QuadrupoleSettingsReaderError(#[from] QuadrupoleSettingsReaderError),
#[error("{0}")]
TimsTofDataError(#[from] TimsTofDataError),
}
3 changes: 3 additions & 0 deletions src/io/readers.rs
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,9 @@ mod precursor_reader;
#[cfg(feature = "tdf")]
mod quad_settings_reader;
mod spectrum_reader;
#[cfg(feature = "tdf")]
mod timstof;
mod timstof_path;

#[cfg(feature = "tdf")]
pub use frame_reader::*;
Expand All @@ -18,3 +20,4 @@ pub use precursor_reader::*;
pub use quad_settings_reader::*;
pub use spectrum_reader::*;
pub use timstof::*;
pub use timstof_path::*;
8 changes: 7 additions & 1 deletion src/io/readers/file_readers/sql_reader.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ use std::collections::HashMap;

use rusqlite::{types::FromSql, Connection};

use crate::readers::{TimsTofPathError, TimsTofPathLike};
use crate::readers::{TimsTofPath, TimsTofPathError, TimsTofPathLike};

#[derive(Debug)]
pub struct SqlReader {
Expand All @@ -19,6 +19,12 @@ pub struct SqlReader {
impl SqlReader {
pub fn open(path: impl TimsTofPathLike) -> Result<Self, SqlReaderError> {
let path = path.to_timstof_path()?;
Self::new_from_path(&path)
}

pub(crate) fn new_from_path(
path: &TimsTofPath,
) -> Result<Self, SqlReaderError> {
let connection = Connection::open(&path.tdf()?)?;
Ok(Self { connection })
}
Expand Down
65 changes: 34 additions & 31 deletions src/io/readers/frame_reader.rs
Original file line number Diff line number Diff line change
Expand Up @@ -10,12 +10,12 @@ use super::{
file_readers::{
sql_reader::{
frame_groups::SqlWindowGroup, frames::SqlFrame, ReadableSqlTable,
SqlReader, SqlReaderError,
SqlReaderError,
},
tdf_blob_reader::{TdfBlob, TdfBlobReader, TdfBlobReaderError},
},
MetadataReader, MetadataReaderError, QuadrupoleSettingsReader,
QuadrupoleSettingsReaderError, TimsTofPathLike,
QuadrupoleSettingsReaderError, TimsTofData, TimsTofDataError,
TimsTofPathLike,
};

#[derive(Debug)]
Expand All @@ -34,21 +34,28 @@ pub struct FrameReader {

impl FrameReader {
pub fn new(path: impl TimsTofPathLike) -> Result<Self, FrameReaderError> {
let compression_type =
match MetadataReader::new(&path)?.compression_type {
2 => 2,
#[cfg(feature = "timscompress")]
3 => 3,
compression_type => {
return Err(FrameReaderError::CompressionTypeError(
compression_type,
))
},
};
let mut timstofdata = TimsTofData::new(&path)?;
Self::new_from_timstofdata(&mut timstofdata)
}

let tdf_sql_reader = SqlReader::open(&path)?;
let sql_frames = SqlFrame::from_sql_reader(&tdf_sql_reader)?;
let tdf_bin_reader = TdfBlobReader::new(&path)?;
pub fn new_from_timstofdata(
timstofdata: &mut TimsTofData,
) -> Result<Self, FrameReaderError> {
let compression_type = match timstofdata.get_metadata().compression_type
{
2 => 2,
#[cfg(feature = "timscompress")]
3 => 3,
compression_type => {
return Err(FrameReaderError::CompressionTypeError(
compression_type,
))
},
};
let sql_frames =
SqlFrame::from_sql_reader(&timstofdata.get_sql_reader())?;
let tdf_bin_reader =
TdfBlobReader::new(&timstofdata.get_timstof_path())?;
#[cfg(feature = "timscompress")]
let compressed_reader = CompressedTdfBlobReader::new(
&path.as_ref().to_path_buf().join("analysis.tdf_bin"),
Expand All @@ -63,22 +70,20 @@ impl FrameReader {
};
// TODO should be refactored out to quadrupole reader
let mut window_groups = vec![0; sql_frames.len()];
let quadrupole_settings;
let mut quadrupole_settings = &vec![];
if acquisition == AcquisitionType::DIAPASEF {
for window_group in
SqlWindowGroup::from_sql_reader(&tdf_sql_reader)?
SqlWindowGroup::from_sql_reader(&timstofdata.get_sql_reader())?
{
window_groups[window_group.frame - 1] =
window_group.window_group;
}
quadrupole_settings = QuadrupoleSettingsReader::new(&path)?;
} else {
quadrupole_settings = vec![];
}
quadrupole_settings = timstofdata.get_quad_settings()?;
};
// TODO move Arc to quad settings reader?
let quadrupole_settings = quadrupole_settings
.into_iter()
.map(|x| Arc::new(x))
.iter()
.map(|x| Arc::new(x.clone()))
.collect();
let frames = (0..sql_frames.len())
.into_par_iter()
Expand Down Expand Up @@ -321,12 +326,6 @@ pub enum FrameReaderError {
TimscompressError,
#[error("{0}")]
TdfBlobReaderError(#[from] TdfBlobReaderError),
#[error("{0}")]
MetadataReaderError(#[from] MetadataReaderError),
#[error("{0}")]
FileNotFound(String),
#[error("{0}")]
SqlReaderError(#[from] SqlReaderError),
#[error("Corrupt Frame")]
CorruptFrame,
#[error("{0}")]
Expand All @@ -335,4 +334,8 @@ pub enum FrameReaderError {
IndexOutOfBounds,
#[error("Compression type {0} not understood")]
CompressionTypeError(u8),
#[error("{0}")]
TimsTofDataError(#[from] TimsTofDataError),
#[error("{0}")]
SqlReaderError(#[from] SqlReaderError),
}
6 changes: 6 additions & 0 deletions src/io/readers/metadata_reader.rs
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,12 @@ impl MetadataReader {
path: impl TimsTofPathLike,
) -> Result<Metadata, MetadataReaderError> {
let tdf_sql_reader = SqlReader::open(path)?;
Self::new_from_sql_reader(&tdf_sql_reader)
}

pub(crate) fn new_from_sql_reader(
tdf_sql_reader: &SqlReader,
) -> Result<Metadata, MetadataReaderError> {
let sql_metadata: HashMap<String, String> =
SqlMetadata::from_sql_reader(&tdf_sql_reader)?;
let compression_type =
Expand Down
Loading