VideoGif will now loop

1 files changed, 20 insertions, 2 deletions

diff --git a/gifed/src/videogif.rs b/gifed/src/videogif.rs
index a5475cc..d066019 100644
--- a/gifed/src/videogif.rs
+++ b/gifed/src/videogif.rs
@@ -1,15 +1,23 @@
-use crate::{block::Palette, writer::ImageBuilder, Color, EncodeError, Gif};
+use crate::{
+	block::{LoopCount, Palette},
+	writer::ImageBuilder,
+	Color, EncodeError, Gif,
+};
 
 use color_quant::NeuQuant;
 use rgb::{ComponentBytes, FromSlice};
 
 use std::convert::TryFrom;
 
+/// A Video-like GIF.
+///
+/// All images must have the same dimensions.
 pub struct VideoGif {
 	width: u16,
 	height: u16,
 	framerate: Option<u16>,
 	frames: Vec<Frame>,
+	looping: LoopCount,
 }
 
 impl VideoGif {
@@ -19,6 +27,7 @@ impl VideoGif {
 			height,
 			framerate: None,
 			frames: vec![],
+			looping: LoopCount::Forever,
 		}
 	}
 
@@ -30,6 +39,13 @@ impl VideoGif {
 		self.framerate = Some(100 / framerate);
 	}
 
+	/// Set the number of times this gif should loop. Defaults to forever.
+	///
+	/// See [LoopCount]
+	pub fn set_looping(&mut self, count: LoopCount) {
+		self.looping = count;
+	}
+
 	/// Adds a frame to the gif.
 	///
 	/// # Panic
@@ -53,10 +69,12 @@ impl VideoGif {
 
 	#[rustfmt::skip] // it was doing things i did not like
 	pub fn build(self) -> Result<Gif, EncodeError> {
-		let Self { width, height, framerate, frames } = self;
+		let Self { width, height, framerate, frames, looping } = self;
 
 		let mut gif = Gif::new(width, height);
 
+		gif.push(looping);
+
 		for Frame { image_indices, interval, palette } in frames {
 			//TODO: return error instead of defaulting to 10? or print warning?
 			// printing in a library is bad but perhaps so is assuming 10 fps?