Graphite/libraries/rawkit/src/tiff/mod.rs

pub mod file;
pub mod tags;
mod types;
pub mod values;

use file::TiffRead;
use num_enum::{FromPrimitive, IntoPrimitive};
use std::fmt::Display;
use std::io::{Read, Seek};
use tags::Tag;
use thiserror::Error;

#[derive(Copy, Clone, Debug, PartialEq, Eq, FromPrimitive, IntoPrimitive)]
#[repr(u16)]
pub enum TagId {
	ImageWidth = 0x100,
	ImageLength = 0x101,
	BitsPerSample = 0x102,
	Compression = 0x103,
	PhotometricInterpretation = 0x104,
	Make = 0x10f,
	Model = 0x110,
	StripOffsets = 0x111,
	Orientation = 0x112,
	SamplesPerPixel = 0x115,
	RowsPerStrip = 0x116,
	StripByteCounts = 0x117,
	SubIfd = 0x14a,
	JpegOffset = 0x201,
	JpegLength = 0x202,
	SonyToneCurve = 0x7010,
	BlackLevel = 0x7310,
	WhiteBalanceRggbLevels = 0x7313,
	CfaPatternDim = 0x828d,
	CfaPattern = 0x828e,
	ColorMatrix1 = 0xc621,
	ColorMatrix2 = 0xc622,

	#[num_enum(catch_all)]
	Unknown(u16),
}

#[repr(u16)]
#[derive(Copy, Clone, Debug, PartialEq, Eq, FromPrimitive, IntoPrimitive)]
pub enum IfdTagType {
	Byte = 1,
	Ascii = 2,
	Short = 3,
	Long = 4,
	Rational = 5,
	SByte = 6,
	Undefined = 7,
	SShort = 8,
	SLong = 9,
	SRational = 10,
	Float = 11,
	Double = 12,

	#[num_enum(catch_all)]
	Unknown(u16),
}

#[derive(Copy, Clone, Debug)]
pub struct IfdEntry {
	tag: TagId,
	the_type: IfdTagType,
	count: u32,
	value: u32,
}

#[derive(Clone, Debug)]
pub struct Ifd {
	current_ifd_offset: u32,
	ifd_entries: Vec<IfdEntry>,
	next_ifd_offset: Option<u32>,
}

impl Ifd {
	pub fn new_first_ifd<R: Read + Seek>(file: &mut TiffRead<R>) -> Result<Self, TiffError> {
		file.seek_from_start(4)?;
		let current_ifd_offset = file.read_u32()?;
		Ifd::new_from_offset(file, current_ifd_offset)
	}

	pub fn new_from_offset<R: Read + Seek>(file: &mut TiffRead<R>, offset: u32) -> Result<Self, TiffError> {
		if offset == 0 {
			return Err(TiffError::InvalidOffset);
		}

		file.seek_from_start(offset)?;
		let num = file.read_u16()?;

		let mut ifd_entries = Vec::with_capacity(num.into());
		for _ in 0..num {
			let tag = file.read_u16()?.into();
			let the_type = file.read_u16()?.into();
			let count = file.read_u32()?;
			let value = file.read_u32()?;

			ifd_entries.push(IfdEntry { tag, the_type, count, value });
		}

		let next_ifd_offset = file.read_u32()?;
		let next_ifd_offset = if next_ifd_offset == 0 { None } else { Some(next_ifd_offset) };

		Ok(Ifd {
			current_ifd_offset: offset,
			ifd_entries,
			next_ifd_offset,
		})
	}

	fn _next_ifd<R: Read + Seek>(&self, file: &mut TiffRead<R>) -> Result<Self, TiffError> {
		Ifd::new_from_offset(file, self.next_ifd_offset.unwrap_or(0))
	}

	pub fn ifd_entries(&self) -> &[IfdEntry] {
		&self.ifd_entries
	}

	pub fn iter(&self) -> impl Iterator<Item = &IfdEntry> {
		self.ifd_entries.iter()
	}

	pub fn get_value<T: Tag, R: Read + Seek>(&self, file: &mut TiffRead<R>) -> Result<T::Output, TiffError> {
		T::get(self, file)
	}
}

impl Display for Ifd {
	fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
		f.write_str("IFD offset: ")?;
		self.current_ifd_offset.fmt(f)?;
		f.write_str("\n")?;

		for ifd_entry in self.ifd_entries() {
			f.write_fmt(format_args!(
				"|- Tag: {:x?}, Type: {:?}, Count: {}, Value: {:x}\n",
				ifd_entry.tag, ifd_entry.the_type, ifd_entry.count, ifd_entry.value
			))?;
		}

		f.write_str("Next IFD offset: ")?;
		if let Some(offset) = self.next_ifd_offset {
			offset.fmt(f)?;
		} else {
			f.write_str("None")?;
		}
		f.write_str("\n")?;

		Ok(())
	}
}

#[derive(Error, Debug)]
pub enum TiffError {
	#[error("The value was invalid")]
	InvalidValue,
	#[error("The type was invalid")]
	InvalidType,
	#[error("The count was invalid")]
	InvalidCount,
	#[error("The tag was missing")]
	MissingTag,
	#[error("The offset was invalid or zero")]
	InvalidOffset,
	#[error("An error occurred when converting integer from one type to another")]
	ConversionError(#[from] std::num::TryFromIntError),
	#[error("An IO Error ocurred")]
	IoError(#[from] std::io::Error),
}