nitf-rs 0.3.1

//! Data Extension segment subheader definition
use std::fmt::Display;
use std::io::{Read, Seek, Write};
use std::str::FromStr;

use crate::headers::NitfSegmentHeader;
use crate::types::{ExtendedSubheader, NitfField, Security};
use crate::{NitfError, NitfResult};

/// Metadata for Data Extension Segment
#[derive(Clone, Debug, Eq, PartialEq, Ord, PartialOrd)]
pub struct DataExtensionHeader {
    /// File Part Type
    pub de: NitfField<DE>,
    /// Unique DES Type Identifier
    pub desid: NitfField<String>,
    /// Check on this registration
    /// Version of the Data Definition
    pub desver: NitfField<u8>,
    //// Security information
    pub security: Security,
    /// Overflowed Header Type
    pub desoflw: NitfField<OverflowedHeaderType>,
    /// Data Item Overflowed
    pub desitem: NitfField<u16>,
    /// DES User-defined Subheader Length
    pub desshl: NitfField<u16>,
    /// User-defined Subheader Fields
    pub desshf: ExtendedSubheader,
}
impl Default for DataExtensionHeader {
    fn default() -> Self {
        Self {
            de: NitfField::init(2u8, "DE"),
            desid: NitfField::init(25u8, "DESID"),
            desver: NitfField::init(2u8, "DESVER"),
            security: Security::default(),
            desoflw: NitfField::init(6u8, "DESOFLW"),
            desitem: NitfField::init(3u8, "DESITEM"),
            desshl: NitfField::init(4u8, "DESSHL"),
            desshf: ExtendedSubheader::init("DESSHF"),
        }
    }
}
impl NitfSegmentHeader for DataExtensionHeader {
    fn read(&mut self, reader: &mut (impl Read + Seek)) -> NitfResult<()> {
        self.de.read(reader)?;
        self.desid.read(reader)?;
        self.desver.read(reader)?;
        self.security.read(reader)?;
        if self.desid.val.trim() == "TRE_OVERFLOW" {
            self.desoflw.read(reader)?;
            self.desitem.read(reader)?;
        }
        self.desshl.read(reader)?;
        if self.desshl.val != 0 {
            self.desshf.read(reader, self.desshl.val as usize)?;
        }
        Ok(())
    }
    fn write(&self, writer: &mut (impl Write + Seek)) -> NitfResult<usize> {
        let mut bytes_written = 0;
        bytes_written += self.de.write(writer)?;
        bytes_written += self.desid.write(writer)?;
        bytes_written += self.desver.write(writer)?;
        bytes_written += self.security.write(writer)?;
        if self.desid.val.trim() == "TRE_OVERFLOW" {
            bytes_written += self.desoflw.write(writer)?;
            bytes_written += self.desitem.write(writer)?;
        }
        bytes_written += self.desshl.write(writer)?;
        if self.desshl.val != 0 {
            bytes_written += self.desshf.write(writer)?;
        }
        Ok(bytes_written)
    }
    fn length(&self) -> usize {
        let mut length: usize = 0;
        length += self.de.length;
        length += self.desid.length;
        length += self.desver.length;
        length += self.security.length();
        if self.desid.val.trim() == "TRE_OVERFLOW" {
            length += self.desoflw.length;
            length += self.desitem.length;
        }
        length += self.desshl.length;
        length += self.desshf.size();
        length
    }
}
#[derive(Default, Copy, Clone, Debug, Eq, PartialEq, Ord, PartialOrd)]
pub enum DE {
    #[default]
    DE,
}
impl FromStr for DE {
    type Err = NitfError;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "DE" => Ok(Self::default()),
            _ => Err(NitfError::ParseError("DE".to_string())),
        }
    }
}
impl Display for DE {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "DE")
    }
}

/// Selection of which header/subheader this extension corresponds to
#[derive(Debug, Default, Copy, Clone, Eq, PartialEq, Ord, PartialOrd)]
pub enum OverflowedHeaderType {
    #[default]
    /// Image subheader extended subheader data overflow
    IXSHD,
    /// Graphic subheader extended subheader data overflow
    SXSHD,
    /// Text subheader extended subheader data overflow
    TXSHD,
    /// Header user defined header overflow
    UDHD,
    /// Image subheader user defined image data overflow
    UDID,
}
impl Display for DataExtensionHeader {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let mut out_str = String::default();
        out_str += format!("{}, ", self.de).as_ref();
        out_str += format!("{}, ", self.desid).as_ref();
        out_str += format!("{}, ", self.desver).as_ref();
        out_str += format!("SECURITY: [{}], ", self.security).as_ref();
        out_str += format!("{}, ", self.desoflw).as_ref();
        out_str += format!("{}, ", self.desitem).as_ref();
        out_str += format!("{}, ", self.desshl).as_ref();
        out_str += format!("{}", self.desshf).as_ref();
        write!(f, "Data Extension Segment: [{out_str}]")
    }
}
impl FromStr for OverflowedHeaderType {
    type Err = NitfError;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "IXSHD" => Ok(Self::IXSHD),
            "SXSHD" => Ok(Self::SXSHD),
            "TXSHD" => Ok(Self::TXSHD),
            "UDHD" => Ok(Self::UDHD),
            "UDID" => Ok(Self::UDID),
            _ => Err(NitfError::ParseError("OverflowedHeaderType".to_string())),
        }
    }
}
impl Display for OverflowedHeaderType {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::IXSHD => write!(f, "IXSHD"),
            Self::SXSHD => write!(f, "SXSHD"),
            Self::TXSHD => write!(f, "TXSHD"),
            Self::UDHD => write!(f, "UDHD"),
            Self::UDID => write!(f, "UDID"),
        }
    }
}