use noodles_vcf::variant::record::AlternateBases;
pub use crate::pbs::helixmtdb::Record;
use noodles_vcf::variant::record_buf::info::field;
impl Record {
pub fn from_vcf_allele(
record: &noodles_vcf::variant::RecordBuf,
allele_no: usize,
) -> Result<Self, anyhow::Error> {
let chrom = record.reference_sequence_name().to_string();
let pos: usize = record
.variant_start()
.expect("Telomeric breakends not supported")
.get();
let pos = i32::try_from(pos)?;
let ref_allele = record.reference_bases().to_string();
let alt_allele = record
.alternate_bases()
.iter()
.nth(allele_no)
.ok_or_else(|| anyhow::anyhow!("no such allele: {}", allele_no))??
.to_string();
let num_total =
if let Some(Some(field::Value::Integer(num_total))) = record.info().get("AN") {
*num_total
} else {
anyhow::bail!("missing INFO/AN in HelixMtDb record")
};
let num_het =
if let Some(Some(field::Value::Integer(num_het))) = record.info().get("AC_het") {
*num_het
} else {
anyhow::bail!("missing INFO/AC in HelixMtDb record")
};
let num_hom =
if let Some(Some(field::Value::Integer(num_hom))) = record.info().get("AC_hom") {
*num_hom
} else {
anyhow::bail!("missing INFO/AC_hom in HelixMtDb record")
};
let feature_type =
if let Some(Some(field::Value::String(feature))) = record.info().get("FEATURE") {
feature.to_string()
} else {
anyhow::bail!("missing INFO/FEATURE in HelixMtDb record")
};
let gene_name =
if let Some(Some(field::Value::String(gene_name))) = record.info().get("GENE") {
gene_name.to_string()
} else {
anyhow::bail!("missing INFO/GENE in HelixMtDb record")
};
Ok(Record {
chrom,
pos,
ref_allele,
alt_allele,
num_total,
num_het,
num_hom,
feature_type,
gene_name,
})
}
}