-
Notifications
You must be signed in to change notification settings - Fork 31
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Now caches frames in the background, and when a frame is requested it checks if it has already has been cached(with AMaZE demosaic), and if it has, it gives it, if not it does a simple bilinear debayer for quickness.
- Loading branch information
Showing
8 changed files
with
271 additions
and
59 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,125 @@ | ||
| /* Yeas, we have another background thread */ | ||
| #include <stdio.h> | ||
| #include <stdint.h> | ||
| #include <stdlib.h> | ||
| #include <pthread.h> | ||
|
|
||
| #include "video_mlv.h" | ||
| #include "../debayer/debayer.h" | ||
|
|
||
| #define MIN(X, Y) (((X) < (Y)) ? (X) : (Y)) | ||
|
|
||
| /* Hmmmm, did anyone need 2 ways of doing this? */ | ||
|
|
||
| /* What I call MegaBytes is actually MebiBytes! I'm so upset to find that out :( */ | ||
| void setMlvRawCacheLimitMegaBytes(mlvObject_t * video, uint64_t megaByteLimit) | ||
| { | ||
| uint64_t frame_size = getMlvWidth(video) * getMlvHeight(video) * sizeof(uint16_t) * 3; | ||
| uint64_t bytes_limit = megaByteLimit * (1 << 20); | ||
|
|
||
| video->cache_limit_mb = megaByteLimit; | ||
| video->cache_limit_bytes = bytes_limit; | ||
|
|
||
| /* Protection against zero division, cuz that causes "Floating point exception: 8"... | ||
| * ...LOL there's not even a floating point in sight */ | ||
| if (frame_size != 0 && video->is_active) | ||
| { | ||
| uint64_t frame_limit = (uint64_t)bytes_limit / (uint64_t)frame_size; | ||
| video->cache_limit_frames = frame_limit; | ||
|
|
||
| printf("\nEnough memory allowed to cache %i frames (%i MiB)\n\n", (int)frame_limit, (int)megaByteLimit); | ||
|
|
||
| /* Begin updating cached frames */ | ||
| if (!video->is_caching) | ||
| { | ||
| pthread_create(&video->cache_thread, NULL, (void *)cache_mlv_frames, (void *)video); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| void setMlvRawCacheLimitFrames(mlvObject_t * video, uint64_t frameLimit) | ||
| { | ||
| uint64_t frame_size = getMlvWidth(video) * getMlvHeight(video) * sizeof(uint16_t) * 3; | ||
|
|
||
| /* Do only if clip is loaded */ | ||
| if (frame_size != 0 && video->is_active) | ||
| { | ||
| uint64_t bytes_limit = frame_size * frameLimit; | ||
| uint64_t mbyte_limit = bytes_limit / (1 << 20); | ||
|
|
||
| video->cache_limit_bytes = bytes_limit; | ||
| video->cache_limit_mb = mbyte_limit; | ||
| video->cache_limit_frames = frameLimit; | ||
|
|
||
| /* Begin updating cached frames */ | ||
| if (!video->is_caching) | ||
| { | ||
| /* cache on bg thread */ | ||
| pthread_create(&video->cache_thread, NULL, (void *)cache_mlv_frames, (void *)video); | ||
| } | ||
| } | ||
| } | ||
|
|
||
| /* Will run in background, caching all frames until it is done, | ||
| * and will be called again on a change */ | ||
| /* TODO: add removing old/un-needed frames ability */ | ||
| void cache_mlv_frames(mlvObject_t * video) | ||
| { | ||
| int width = getMlvWidth(video); | ||
| int height = getMlvHeight(video); | ||
| int threads = getMlvCpuCores(video) / 2 + 1; | ||
| int cache_frames = MIN((int)video->cache_limit_frames, (int)video->frames); | ||
| size_t frame_size_rgb = width * height * sizeof(uint16_t) * 3; | ||
|
|
||
| float * raw_frame = malloc( getMlvWidth(video) * getMlvHeight(video) * sizeof(float) ); | ||
|
|
||
| video->is_caching = 1; | ||
|
|
||
| printf("\nTotal frames %i, Cache limit frames: %i\n\n", (int)video->frames, (int)video->cache_limit_frames); | ||
|
|
||
| /* Cache until done */ | ||
| for (int frame_index = 0; frame_index < cache_frames; ++frame_index) | ||
| { | ||
| /* Only debayer if frame is not already cached and has not been requested to stop */ | ||
| if (!video->cached_frames[frame_index] && !video->stop_caching) | ||
| { | ||
| video->rgb_raw_frames[frame_index] = (uint16_t *)malloc( frame_size_rgb ); | ||
|
|
||
| /* debayer_type 1, we want to cache AMaZE frames */ | ||
| get_mlv_raw_frame_debayered(video, frame_index, raw_frame, video->rgb_raw_frames[frame_index], 1); | ||
|
|
||
| video->cached_frames[frame_index] = 1; | ||
|
|
||
| printf("Debayered frame %i/%i has been cached.\n", frame_index, cache_frames); | ||
| } | ||
| } | ||
|
|
||
| free(raw_frame); | ||
|
|
||
| video->is_caching = 0; | ||
| } | ||
|
|
||
| /* Gets a freshly debayered frame every time ( temp memory should be Width * Height * sizeof(float) ) */ | ||
| void get_mlv_raw_frame_debayered( mlvObject_t * video, | ||
| int frame_index, | ||
| float * temp_memory, | ||
| uint16_t * output_frame, | ||
| int debayer_type ) /* 0=bilinear 1=amaze */ | ||
| { | ||
| int width = getMlvWidth(video); | ||
| int height = getMlvHeight(video); | ||
|
|
||
| /* Get the raw data in B&W */ | ||
| getMlvRawFrameFloat(video, frame_index, temp_memory); | ||
|
|
||
| if (debayer_type) | ||
| { | ||
| /* Debayer AMAZEly - using some of the cores (why this fixes the PHAT pink stripes? ...how would i know) */ | ||
| debayerAmaze(output_frame, temp_memory, width, height, getMlvCpuCores(video) / 2 + 1); | ||
| } | ||
| else | ||
| { | ||
| /* Debayer quickly (bilinearly) */ | ||
| debayerBasic(output_frame, temp_memory, width, height, 1); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.