Revert "Clear replay data in replay save thread to not block the main capture thread"

This reverts commit 5db30f3fb99545524fc80f87ecc488972aec1711.

5 files changed, 8 insertions, 79 deletions

diff --git a/src/recorder/recorder.c b/src/recorder/recorder.c
index f24495c..901077c 100644
--- a/src/recorder/recorder.c
+++ b/src/recorder/recorder.c
@@ -752,9 +752,15 @@ static void recorder_poll_replay_save(gsr_recorder *self) {
         atomic_store(&self->save_replay_seconds, 0);
         const int restart_replay_request = atomic_exchange(&self->save_replay_restart_replay, GSR_RESTART_REPLAY_USE_OPTION);
         const bool restart_replay = restart_replay_request == GSR_RESTART_REPLAY_USE_OPTION ? self->settings.restart_replay_on_save : restart_replay_request == GSR_RESTART_REPLAY_ENABLE;
-        const bool replay_start_result = gsr_replay_save_start(&self->replay_save, self->video_codec_context, GSR_VIDEO_STREAM_INDEX, &self->audio_capture, &self->encoder, &self->settings, self->file_extension, self->hdr, self->video_sources, current_save_replay_seconds, restart_replay && current_save_replay_seconds == GSR_SAVE_REPLAY_SECONDS_FULL);
+        const bool replay_start_result = gsr_replay_save_start(&self->replay_save, self->video_codec_context, GSR_VIDEO_STREAM_INDEX, &self->audio_capture, &self->encoder, &self->settings, self->file_extension, self->hdr, self->video_sources, current_save_replay_seconds);
         if(!replay_start_result && self->callbacks.replay_saved)
             self->callbacks.replay_saved(NULL, self->callbacks.userdata);
+
+        if(restart_replay && current_save_replay_seconds == GSR_SAVE_REPLAY_SECONDS_FULL) {
+            pthread_mutex_lock(&self->encoder.replay_mutex);
+            gsr_replay_buffer_clear(self->encoder.replay_buffer);
+            pthread_mutex_unlock(&self->encoder.replay_mutex);
+        }
     }
 }
 
diff --git a/src/recorder/replay_save.c b/src/recorder/replay_save.c
index 55cf35d..3251616 100644
--- a/src/recorder/replay_save.c
+++ b/src/recorder/replay_save.c
@@ -23,15 +23,6 @@ static void gsr_replay_save_cleanup(gsr_replay_save *self) {
         self->cloned_replay_buffer = NULL;
     }
 
-    /*
-        This is the data that the replay buffer had before the replay was saved and nothing references it anymore,
-        so it can be free'd without holding the replay mutex. This can take a while when the replay buffer is large.
-    */
-    if(self->restarted_replay_buffer_data) {
-        gsr_replay_buffer_destroy(self->restarted_replay_buffer_data);
-        self->restarted_replay_buffer_data = NULL;
-    }
-
     if(self->audio_pts_offsets) {
         free(self->audio_pts_offsets);
         self->audio_pts_offsets = NULL;
@@ -125,7 +116,7 @@ static void* replay_save_thread(void *userdata) {
     return NULL;
 }
 
-bool gsr_replay_save_start(gsr_replay_save *self, AVCodecContext *video_codec_context, int video_stream_index, const gsr_audio_capture *audio_capture, gsr_encoder *encoder, const gsr_recorder_settings *settings, const char *file_extension, bool hdr, gsr_video_sources *video_sources, int current_save_replay_seconds, bool restart_replay) {
+bool gsr_replay_save_start(gsr_replay_save *self, AVCodecContext *video_codec_context, int video_stream_index, const gsr_audio_capture *audio_capture, gsr_encoder *encoder, const gsr_recorder_settings *settings, const char *file_extension, bool hdr, gsr_video_sources *video_sources, int current_save_replay_seconds) {
     if(self->thread_created)
         return true;
 
@@ -134,16 +125,10 @@ bool gsr_replay_save_start(gsr_replay_save *self, AVCodecContext *video_codec_co
     self->video_stream_index = video_stream_index;
     self->output_filepath[0] = '\0';
     self->success = false;
-    self->restarted_replay_buffer_data = NULL;
     atomic_store(&self->finished, 0);
 
     pthread_mutex_lock(&encoder->replay_mutex);
     self->cloned_replay_buffer = gsr_replay_buffer_clone(encoder->replay_buffer);
-    if(self->cloned_replay_buffer && restart_replay) {
-        self->restarted_replay_buffer_data = gsr_replay_buffer_take_data(encoder->replay_buffer);
-        if(!self->restarted_replay_buffer_data)
-            gsr_log(GSR_LOG_LEVEL_ERROR, "gsr_replay_save: failed to restart the replay buffer");
-    }
     pthread_mutex_unlock(&encoder->replay_mutex);
     if(!self->cloned_replay_buffer) {
         /* TODO: Return this error to mark the replay as failed */
diff --git a/src/replay_buffer/replay_buffer.c b/src/replay_buffer/replay_buffer.c
index 4958bba..2f7db54 100644
--- a/src/replay_buffer/replay_buffer.c
+++ b/src/replay_buffer/replay_buffer.c
@@ -47,10 +47,6 @@ gsr_replay_buffer* gsr_replay_buffer_clone(gsr_replay_buffer *self) {
     return self->clone(self);
 }
 
-gsr_replay_buffer* gsr_replay_buffer_take_data(gsr_replay_buffer *self) {
-    return self->take_data(self);
-}
-
 gsr_replay_buffer_iterator gsr_replay_buffer_find_packet_index_by_time_passed(gsr_replay_buffer *self, int seconds) {
     return self->find_packet_index_by_time_passed(self, seconds);
 }
diff --git a/src/replay_buffer/replay_buffer_disk.c b/src/replay_buffer/replay_buffer_disk.c
index da82df5..b61ccfa 100644
--- a/src/replay_buffer/replay_buffer_disk.c
+++ b/src/replay_buffer/replay_buffer_disk.c
@@ -254,36 +254,6 @@ static uint8_t* gsr_replay_buffer_disk_iterator_get_packet_data(gsr_replay_buffe
     return packet_data;
 }
 
-static gsr_replay_buffer* gsr_replay_buffer_disk_take_data(gsr_replay_buffer *replay_buffer) {
-    gsr_replay_buffer_disk *self = (gsr_replay_buffer_disk*)replay_buffer;
-    gsr_replay_buffer_disk *destination = calloc(1, sizeof(gsr_replay_buffer_disk));
-    if(!destination)
-        return NULL;
-
-    gsr_replay_buffer_disk_set_impl_funcs(destination);
-
-    destination->replay_buffer_time = self->replay_buffer_time;
-    destination->storage_counter = self->storage_counter;
-    destination->storage_num_bytes_written = self->storage_num_bytes_written;
-    destination->storage_fd = self->storage_fd;
-
-    for(size_t i = 0; i < self->num_files; ++i) {
-        destination->files[i] = self->files[i];
-    }
-    destination->num_files = self->num_files;
-
-    snprintf(destination->replay_directory, sizeof(destination->replay_directory), "%s", self->replay_directory);
-    /* |self| keeps writing to the same directory, so it's the one that removes it */
-    destination->owns_directory = false;
-
-    self->num_files = 0;
-    self->storage_fd = 0;
-    self->storage_num_bytes_written = 0;
-    /* |storage_counter| is not reset, to not overwrite the files that were moved to |destination| */
-
-    return (gsr_replay_buffer*)destination;
-}
-
 static gsr_replay_buffer* gsr_replay_buffer_disk_clone(gsr_replay_buffer *replay_buffer) {
     gsr_replay_buffer_disk *self = (gsr_replay_buffer_disk*)replay_buffer;
     gsr_replay_buffer_disk *destination = calloc(1, sizeof(gsr_replay_buffer_disk));
@@ -430,7 +400,6 @@ static void gsr_replay_buffer_disk_set_impl_funcs(gsr_replay_buffer_disk *self)
     self->replay_buffer.iterator_get_packet = gsr_replay_buffer_disk_iterator_get_packet;
     self->replay_buffer.iterator_get_packet_data = gsr_replay_buffer_disk_iterator_get_packet_data;
     self->replay_buffer.clone = gsr_replay_buffer_disk_clone;
-    self->replay_buffer.take_data = gsr_replay_buffer_disk_take_data;
     self->replay_buffer.find_packet_index_by_time_passed = gsr_replay_buffer_disk_find_file_index_by_time_passed;
     self->replay_buffer.find_keyframe = gsr_replay_buffer_disk_find_keyframe;
     self->replay_buffer.iterator_next = gsr_replay_buffer_disk_iterator_next;
diff --git a/src/replay_buffer/replay_buffer_ram.c b/src/replay_buffer/replay_buffer_ram.c
index ce3914b..39f31b1 100644
--- a/src/replay_buffer/replay_buffer_ram.c
+++ b/src/replay_buffer/replay_buffer_ram.c
@@ -130,32 +130,6 @@ static uint8_t* gsr_replay_buffer_ram_iterator_get_packet_data(gsr_replay_buffer
     return NULL;
 }
 
-static gsr_replay_buffer* gsr_replay_buffer_ram_take_data(gsr_replay_buffer *replay_buffer) {
-    gsr_replay_buffer_ram *self = (gsr_replay_buffer_ram*)replay_buffer;
-    gsr_av_packet_ram **packets = calloc(self->capacity_num_packets, sizeof(gsr_av_packet_ram*));
-    if(!packets)
-        return NULL;
-
-    gsr_replay_buffer_ram *destination = calloc(1, sizeof(gsr_replay_buffer_ram));
-    if(!destination) {
-        free(packets);
-        return NULL;
-    }
-
-    gsr_replay_buffer_ram_set_impl_funcs(destination);
-
-    destination->packets = self->packets;
-    destination->capacity_num_packets = self->capacity_num_packets;
-    destination->num_packets = self->num_packets;
-    destination->index = self->index;
-
-    self->packets = packets;
-    self->num_packets = 0;
-    self->index = 0;
-
-    return (gsr_replay_buffer*)destination;
-}
-
 static gsr_replay_buffer* gsr_replay_buffer_ram_clone(gsr_replay_buffer *replay_buffer) {
     gsr_replay_buffer_ram *self = (gsr_replay_buffer_ram*)replay_buffer;
     gsr_replay_buffer_ram *destination = calloc(1, sizeof(gsr_replay_buffer_ram));
@@ -242,7 +216,6 @@ static void gsr_replay_buffer_ram_set_impl_funcs(gsr_replay_buffer_ram *self) {
     self->replay_buffer.iterator_get_packet = gsr_replay_buffer_ram_iterator_get_packet;
     self->replay_buffer.iterator_get_packet_data = gsr_replay_buffer_ram_iterator_get_packet_data;
     self->replay_buffer.clone = gsr_replay_buffer_ram_clone;
-    self->replay_buffer.take_data = gsr_replay_buffer_ram_take_data;
     self->replay_buffer.find_packet_index_by_time_passed = gsr_replay_buffer_ram_find_packet_index_by_time_passed;
     self->replay_buffer.find_keyframe = gsr_replay_buffer_ram_find_keyframe;
     self->replay_buffer.iterator_next = gsr_replay_buffer_ram_iterator_next;