Skip to content

Commit 33b2b21

Browse files
molpopgenmolpopgen
committed
cargo check passes
1 parent beece1a commit 33b2b21

File tree

6 files changed

+147
-85
lines changed

6 files changed

+147
-85
lines changed

‎src/sys/individual_table.rs‎

Lines changed: 44 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use std::ptr::NonNull;
22

3+
use super::bindings::tsk_size_t;
34
use super::flags::IndividualFlags;
45
use super::newtypes::IndividualId;
56

@@ -76,18 +77,46 @@ impl IndividualTable {
7677

7778
raw_metadata_getter_for_tables!(IndividualId);
7879

79-
pub fn location(&self, row: IndividualId) -> Option<&[super::newtypes::Location]> {
80-
assert!(
81-
(self.as_ref().num_rows == 0 && self.as_ref().location_length == 0)
82-
|| (!self.as_ref().location.is_null() && !self.as_ref().location_offset.is_null())
83-
);
80+
pub fn location_column(&self) -> &[super::newtypes::Location] {
8481
unsafe {
85-
super::tsk_ragged_column_access(
86-
row,
87-
self.as_ref().location,
88-
self.as_ref().num_rows,
82+
std::slice::from_raw_parts(
83+
self.as_ref().location.cast::<super::newtypes::Location>(),
84+
self.as_ref().location_length as usize,
85+
)
86+
}
87+
}
88+
89+
fn location_offset_column_raw(&self) -> &[tsk_size_t] {
90+
unsafe {
91+
std::slice::from_raw_parts(
8992
self.as_ref().location_offset,
90-
self.as_ref().location_length,
93+
self.as_ref().num_rows as usize,
94+
)
95+
}
96+
}
97+
98+
pub fn location(&self, row: IndividualId) -> Option<&[super::newtypes::Location]> {
99+
super::tsk_ragged_column_access(
100+
row,
101+
self.location_column(),
102+
self.location_offset_column_raw(),
103+
)
104+
}
105+
106+
fn parents_column(&self) -> &[IndividualId] {
107+
unsafe {
108+
std::slice::from_raw_parts(
109+
self.as_ref().parents.cast::<IndividualId>(),
110+
self.as_ref().parents_length as usize,
111+
)
112+
}
113+
}
114+
115+
fn parents_offset_column_raw(&self) -> &[tsk_size_t] {
116+
unsafe {
117+
std::slice::from_raw_parts(
118+
self.as_ref().parents_offset,
119+
self.as_ref().num_rows as usize,
91120
)
92121
}
93122
}
@@ -97,15 +126,11 @@ impl IndividualTable {
97126
(self.as_ref().num_rows == 0 && self.as_ref().parents_length == 0)
98127
|| (!self.as_ref().parents.is_null() && !self.as_ref().location_offset.is_null())
99128
);
100-
unsafe {
101-
super::tsk_ragged_column_access(
102-
row,
103-
self.as_ref().parents,
104-
self.as_ref().num_rows,
105-
self.as_ref().parents_offset,
106-
self.as_ref().parents_length,
107-
)
108-
}
129+
super::tsk_ragged_column_access(
130+
row,
131+
self.parents_column(),
132+
self.parents_offset_column_raw(),
133+
)
109134
}
110135
}
111136

‎src/sys/macros.rs‎

Lines changed: 20 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -295,21 +295,30 @@ macro_rules! safe_tsk_column_access {
295295

296296
macro_rules! raw_metadata_getter_for_tables {
297297
($idtype: ty) => {
298-
pub fn raw_metadata<I: Into<$idtype>>(&self, row: I) -> Option<&[u8]> {
299-
assert!(
300-
(self.as_ref().num_rows == 0 && self.as_ref().metadata_length == 0)
301-
|| (!self.as_ref().metadata.is_null()
302-
&& !self.as_ref().metadata_offset.is_null())
303-
);
298+
fn metadata_column(&self) -> &[u8] {
299+
unsafe {
300+
std::slice::from_raw_parts(
301+
self.as_ref().metadata.cast::<u8>(),
302+
self.as_ref().metadata_length as usize,
303+
)
304+
}
305+
}
306+
307+
fn metadata_offset_raw(&self) -> &[super::bindings::tsk_size_t] {
304308
unsafe {
305-
$crate::sys::tsk_ragged_column_access::<'_, u8, $idtype, _, _>(
306-
row.into(),
307-
self.as_ref().metadata,
308-
self.as_ref().num_rows,
309+
std::slice::from_raw_parts(
309310
self.as_ref().metadata_offset,
310-
self.as_ref().metadata_length,
311+
self.as_ref().num_rows as usize,
311312
)
312313
}
313314
}
315+
316+
pub fn raw_metadata<I: Into<$idtype>>(&self, row: I) -> Option<&[u8]> {
317+
$crate::sys::tsk_ragged_column_access(
318+
row.into(),
319+
self.metadata_column(),
320+
self.metadata_offset_raw(),
321+
)
322+
}
314323
};
315324
}

‎src/sys/mod.rs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ fn tsk_ragged_column_access_detail<'a, T>(
163163
}
164164
}
165165

166-
fn tsk_ragged_column_access<'a, O, R: Into<bindings::tsk_id_t>, L: Into<bindings::tsk_size_t>>(
166+
fn tsk_ragged_column_access<'a, O, R: Into<bindings::tsk_id_t>>(
167167
row: R,
168168
column: &'a [O],
169169
raw_offset: &'a [bindings::tsk_size_t],

‎src/sys/mutation_table.rs‎

Lines changed: 21 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ use super::bindings::tsk_mutation_table_add_row;
1010
use super::bindings::tsk_mutation_table_clear;
1111
use super::bindings::tsk_mutation_table_init;
1212
use super::bindings::tsk_mutation_table_t;
13+
use super::bindings::tsk_size_t;
1314
use super::tskbox::TskBox;
1415
use super::TskitError;
1516

@@ -100,24 +101,31 @@ impl MutationTable {
100101

101102
raw_metadata_getter_for_tables!(MutationId);
102103

103-
pub fn derived_state(&self, row: MutationId) -> Option<&[u8]> {
104-
assert!(
105-
(self.as_ref().num_rows == 0 && self.as_ref().derived_state_length == 0)
106-
|| (!self.as_ref().derived_state.is_null()
107-
&& !self.as_ref().derived_state_offset.is_null())
108-
);
109-
// SAFETY: either both columns are empty or both pointers at not NULL,
110-
// in which case the correct lengths are from the low-level objects
104+
fn derived_state_column(&self) -> &[u8] {
111105
unsafe {
112-
super::tsk_ragged_column_access(
113-
row,
114-
self.as_ref().derived_state,
115-
self.as_ref().num_rows,
106+
std::slice::from_raw_parts(
107+
self.as_ref().derived_state.cast::<u8>(),
108+
self.as_ref().derived_state_length as usize,
109+
)
110+
}
111+
}
112+
113+
fn derived_state_offset_raw(&self) -> &[tsk_size_t] {
114+
unsafe {
115+
std::slice::from_raw_parts(
116116
self.as_ref().derived_state_offset,
117-
self.as_ref().derived_state_length,
117+
self.as_ref().num_rows as usize,
118118
)
119119
}
120120
}
121+
122+
pub fn derived_state(&self, row: MutationId) -> Option<&[u8]> {
123+
super::tsk_ragged_column_access(
124+
row,
125+
self.derived_state_column(),
126+
self.derived_state_offset_raw(),
127+
)
128+
}
121129
}
122130

123131
impl Default for MutationTable {

‎src/sys/provenance_table.rs‎

Lines changed: 40 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -58,47 +58,57 @@ impl ProvenanceTable {
5858
Ok(rv)
5959
}
6060

61-
pub fn timestamp(&self, row: ProvenanceId) -> Option<&str> {
62-
assert!(
63-
(self.as_ref().num_rows != 0 && self.as_ref().timestamp_length != 0)
64-
|| (!self.as_ref().timestamp.is_null()
65-
&& !self.as_ref().timestamp_offset.is_null())
66-
);
61+
fn timestamp_column(&self) -> &[u8] {
62+
unsafe {
63+
std::slice::from_raw_parts(
64+
self.as_ref().timestamp.cast::<u8>(),
65+
self.as_ref().timestamp_length as usize,
66+
)
67+
}
68+
}
6769

68-
// SAFETY: the previous assert checks the safety
69-
// requirements
70-
let timestamp_slice = unsafe {
71-
super::tsk_ragged_column_access(
72-
row,
73-
self.as_ref().timestamp,
74-
self.as_ref().num_rows,
70+
fn timestamp_offset_column_raw(&self) -> &[tsk_size_t] {
71+
unsafe {
72+
std::slice::from_raw_parts(
7573
self.as_ref().timestamp_offset,
76-
self.as_ref().timestamp_length,
74+
self.as_ref().num_rows as usize,
7775
)
78-
};
76+
}
77+
}
78+
79+
pub fn timestamp(&self, row: ProvenanceId) -> Option<&str> {
80+
let timestamp_slice = super::tsk_ragged_column_access(
81+
row,
82+
self.timestamp_column(),
83+
self.timestamp_offset_column_raw(),
84+
);
7985
match timestamp_slice {
8086
Some(tstamp) => std::str::from_utf8(tstamp).ok(),
8187
None => None,
8288
}
8389
}
8490

91+
fn record_column(&self) -> &[u8] {
92+
unsafe {
93+
std::slice::from_raw_parts(
94+
self.as_ref().record.cast::<u8>(),
95+
self.as_ref().record_length as usize,
96+
)
97+
}
98+
}
99+
100+
fn record_offset_column_raw(&self) -> &[tsk_size_t] {
101+
unsafe {
102+
std::slice::from_raw_parts(self.as_ref().record_offset, self.as_ref().num_rows as usize)
103+
}
104+
}
105+
85106
pub fn record(&self, row: ProvenanceId) -> Option<&str> {
86-
assert!(
87-
(self.as_ref().num_rows != 0 && self.as_ref().record_length != 0)
88-
|| (!self.as_ref().record.is_null() && !self.as_ref().record_offset.is_null())
107+
let record_slice = super::tsk_ragged_column_access(
108+
row,
109+
self.record_column(),
110+
self.record_offset_column_raw(),
89111
);
90-
91-
// SAFETY: the previous assert checks the safety
92-
// requirements
93-
let record_slice = unsafe {
94-
super::tsk_ragged_column_access(
95-
row,
96-
self.as_ref().record,
97-
self.as_ref().num_rows,
98-
self.as_ref().record_offset,
99-
self.as_ref().record_length,
100-
)
101-
};
102112
match record_slice {
103113
Some(rec) => std::str::from_utf8(rec).ok(),
104114
None => None,

‎src/sys/site_table.rs‎

Lines changed: 21 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
use std::ptr::NonNull;
22

3+
use super::bindings::tsk_size_t;
34
use super::newtypes::Position;
45
use super::newtypes::SiteId;
56

@@ -76,22 +77,31 @@ impl SiteTable {
7677

7778
raw_metadata_getter_for_tables!(SiteId);
7879

79-
pub fn ancestral_state(&self, row: SiteId) -> Option<&[u8]> {
80-
assert!(
81-
(self.as_ref().num_rows == 0 && self.as_ref().ancestral_state_length == 0)
82-
|| (!self.as_ref().ancestral_state.is_null()
83-
&& !self.as_ref().ancestral_state_offset.is_null())
84-
);
80+
fn ancestral_state_column(&self) -> &[u8] {
8581
unsafe {
86-
super::tsk_ragged_column_access(
87-
row,
88-
self.as_ref().ancestral_state,
89-
self.as_ref().num_rows,
82+
std::slice::from_raw_parts(
83+
self.as_ref().ancestral_state.cast::<u8>(),
84+
self.as_ref().ancestral_state_length as usize,
85+
)
86+
}
87+
}
88+
89+
fn ancestral_state_offset_raw(&self) -> &[tsk_size_t] {
90+
unsafe {
91+
std::slice::from_raw_parts(
9092
self.as_ref().ancestral_state_offset,
91-
self.as_ref().ancestral_state_length,
93+
self.as_ref().num_rows as usize,
9294
)
9395
}
9496
}
97+
98+
pub fn ancestral_state(&self, row: SiteId) -> Option<&[u8]> {
99+
super::tsk_ragged_column_access(
100+
row,
101+
self.ancestral_state_column(),
102+
self.ancestral_state_offset_raw(),
103+
)
104+
}
95105
}
96106

97107
impl Default for SiteTable {

0 commit comments

Comments
 (0)