From da1149ec566fe421acb1f94e94720c32e4c35534 Mon Sep 17 00:00:00 2001
From: Muaz Ahmad <mahmad2000@protonmail.com>
Date: Mon, 23 Oct 2023 16:29:19 +0500
Subject: [PATCH] prep to handle segmentation

---
 src/muxer/hls/handler.rs     | 17 +++++++++++++++--
 src/muxer/hls/mp4/mod.rs     | 10 ++++++----
 src/muxer/hls/mp4/samples.rs |  9 +++++++++
 src/muxer/hls/segments.rs    |  8 ++++++++
 4 files changed, 38 insertions(+), 6 deletions(-)
 create mode 100644 src/muxer/hls/segments.rs

diff --git a/src/muxer/hls/handler.rs b/src/muxer/hls/handler.rs
index d0094c4..c4d346b 100644
--- a/src/muxer/hls/handler.rs
+++ b/src/muxer/hls/handler.rs
@@ -1,5 +1,7 @@
 use std::error::Error;
 use std::sync::{mpsc, Arc};
+use std::time;
+use std::thread;
 
 use crate::util;
 use crate::muxer::hls::mp4;
@@ -8,14 +10,25 @@ pub struct HLSHandler {
     muxer: mp4::MP4Muxer,
     err_in: mpsc::Sender<Box<dyn Error + Send + Sync>>,
     args: util::HLSArgs,
+    curr_segment_idx: usize,
 }
 
 impl HLSHandler {
-    pub fn data_loop(&self) {
+    pub fn data_loop(&mut self) {
+        self.muxer.spawn_read_loops();
+        loop {
+            if self.muxer.v_samples.segment_ready(self.args.segment_time as usize) {
+                todo!();
+            }
+            if self.muxer.a_samples.segment_ready(self.args.segment_time as usize) {
+                todo!();
+            }
+            thread::sleep(time::Duration::from_millis(100))
+        }
         todo!();
     }
 }
 
 pub fn spawn(v: mpsc::Receiver<util::NALUPacket>, a: mpsc::Receiver<util::NALUPacket>, err_in: mpsc::Sender<Box<dyn Error + Send + Sync>>, metadata: Arc<util::Metadata>, args: util::HLSArgs) -> Result<HLSHandler, Box<dyn Error>> {
-    Ok(HLSHandler {muxer: mp4::new_muxer(v, a, metadata, err_in.clone())?, err_in: err_in, args: args})
+    Ok(HLSHandler {muxer: mp4::new_muxer(v, a, metadata, err_in.clone())?, err_in: err_in, args: args, curr_segment_idx: 0})
 }
diff --git a/src/muxer/hls/mp4/mod.rs b/src/muxer/hls/mp4/mod.rs
index b7b6dba..b504690 100644
--- a/src/muxer/hls/mp4/mod.rs
+++ b/src/muxer/hls/mp4/mod.rs
@@ -9,9 +9,9 @@ use crate::util;
 
 pub struct MP4Muxer {
     v: Option<mpsc::Receiver<util::NALUPacket>>,
-    v_samples: Arc<samples::SampleQueue>,
+    pub v_samples: Arc<samples::SampleQueue>,
     a: Option<mpsc::Receiver<util::NALUPacket>>,
-    a_samples: Arc<samples::SampleQueue>,
+    pub a_samples: Arc<samples::SampleQueue>,
     metadata: Arc<util::Metadata>,
     err_in: mpsc::Sender<Box<dyn Error + Send + Sync>>,
 }
@@ -20,15 +20,17 @@ pub fn new_muxer(v: mpsc::Receiver<util::NALUPacket>, a: mpsc::Receiver<util::NA
     let mut muxer = MP4Muxer {
         v: Some(v),
         a: Some(a),
-        metadata: metadata,
+        metadata: metadata.clone(),
         err_in: err_in,
         v_samples: Arc::new(samples::SampleQueue {
             queue: Mutex::new(Vec::new()),
             default_duration: 1000,
+            samples_per_sec:  metadata.video.framerate as usize,
         }),
         a_samples: Arc::new(samples::SampleQueue {
             queue: Mutex::new(Vec::new()),
-            default_duration: 960,
+            default_duration: metadata.audio.samplerate * 20 / 1000,
+            samples_per_sec: 50,
         }),
     };
     muxer.gen_init()?;
diff --git a/src/muxer/hls/mp4/samples.rs b/src/muxer/hls/mp4/samples.rs
index 1b8e447..b8ae431 100644
--- a/src/muxer/hls/mp4/samples.rs
+++ b/src/muxer/hls/mp4/samples.rs
@@ -9,6 +9,7 @@ pub struct Sample {
 pub struct SampleQueue {
     pub queue: Mutex<Vec<Sample>>,
     pub default_duration: u32,
+    pub samples_per_sec: usize,
 }
 
 impl SampleQueue {
@@ -19,4 +20,12 @@ impl SampleQueue {
             flags: flags,
         })
     }
+
+    pub fn segment_ready(&self, secs: usize) -> bool {
+        return self.queue.lock().unwrap().len() >= self.samples_per_sec * secs;
+    }
+
+    pub fn get_segment(&self, secs: usize) -> Vec<Sample> {
+        return self.queue.lock().unwrap().drain(..self.samples_per_sec * secs).collect();
+    }
 }
diff --git a/src/muxer/hls/segments.rs b/src/muxer/hls/segments.rs
new file mode 100644
index 0000000..5e2b81d
--- /dev/null
+++ b/src/muxer/hls/segments.rs
@@ -0,0 +1,8 @@
+use std::time;
+use crate::muxer::hls::mp4::samples;
+
+pub struct Segment {
+    segment_idx: usize,
+    segment_video: Vec<samples::Sample>,
+    segment_audio: Vec<samples::Sample>,
+}