Skip to content

Commit

Permalink
feat: Add to_vcf_string method for bcf::Record (#443)
Browse files Browse the repository at this point in the history
* Add bcf record to_vcf_string method and test

* Reformat as Display trait

* Revert to Record::to_vcf_string method

Added return value checking on vcf_format, which also suggests that the
methods should return a Result and shound't be implemented directly as
a Display trait.

---------

Co-authored-by: Johannes Köster <[email protected]>
  • Loading branch information
ctsa and johanneskoester authored Nov 12, 2024
1 parent 9bda5f7 commit 489c0d7
Show file tree
Hide file tree
Showing 2 changed files with 60 additions and 0 deletions.
58 changes: 58 additions & 0 deletions src/bcf/record.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1088,6 +1088,38 @@ impl Record {
}
"".to_owned()
}

/// Convert to VCF String
///
/// Intended for debug only. Use Writer for efficient VCF output.
///
pub fn to_vcf_string(&self) -> Result<String> {
let mut buf = htslib::kstring_t {
l: 0,
m: 0,
s: ptr::null_mut(),
};
let ret = unsafe { htslib::vcf_format(self.header().inner, self.inner, &mut buf) };

if ret < 0 {
if !buf.s.is_null() {
unsafe {
libc::free(buf.s as *mut libc::c_void);
}
}
return Err(Error::BcfToString);
}

let vcf_str = unsafe {
let vcf_str = String::from(ffi::CStr::from_ptr(buf.s).to_str().unwrap());
if !buf.s.is_null() {
libc::free(buf.s as *mut libc::c_void);
}
vcf_str
};

Ok(vcf_str)
}
}

impl Clone for Record {
Expand Down Expand Up @@ -1714,4 +1746,30 @@ mod tests {
assert!(!record.has_filter(&bar));
assert!(record.has_filter("PASS".as_bytes()));
}

#[test]
fn test_record_to_vcf_string_err() {
let tmp = NamedTempFile::new().unwrap();
let path = tmp.path();
let header = Header::new();
let vcf = Writer::from_path(path, &header, true, Format::Vcf).unwrap();
let record = vcf.empty_record();
assert!(record.to_vcf_string().is_err());
}

#[test]
fn test_record_to_vcf_string() {
let tmp = NamedTempFile::new().unwrap();
let path = tmp.path();
let mut header = Header::new();
header.push_record(b"##contig=<ID=chr1,length=1000>");
header.push_record(br#"##FILTER=<ID=foo,Description="sample is a foo fighter">"#);
let vcf = Writer::from_path(path, &header, true, Format::Vcf).unwrap();
let mut record = vcf.empty_record();
record.push_filter("foo".as_bytes()).unwrap();
assert_eq!(
record.to_vcf_string().unwrap(),
"chr1\t1\t.\t.\t.\t0\tfoo\t.\n"
);
}
}
2 changes: 2 additions & 0 deletions src/errors.rs
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,8 @@ pub enum Error {
BcfSetValues,
#[error("failed to remove alleles in BCF/VCF record")]
BcfRemoveAlleles,
#[error("failed to render BCF record as string")]
BcfToString,

#[error("invalid compression level {level}")]
BgzfInvalidCompressionLevel { level: i8 },
Expand Down

0 comments on commit 489c0d7

Please sign in to comment.