|
#include "vp_image_record_task.h"
|
|
namespace vp_nodes {
|
vp_image_record_task::vp_image_record_task(int channel_index,
|
std::string file_name_without_ext,
|
std::string save_dir,
|
bool auto_sub_dir,
|
bool osd,
|
vp_objects::vp_size resolution_w_h,
|
std::string host_node_name,
|
bool auto_start):
|
vp_record_task(channel_index, file_name_without_ext, save_dir, auto_sub_dir, resolution_w_h, osd, host_node_name) {
|
// start automatically when initializing
|
if (auto_start) {
|
start();
|
}
|
}
|
|
vp_image_record_task::~vp_image_record_task() {
|
stop_task();
|
}
|
|
void vp_image_record_task::record_task_run() {
|
/* Below Code Run In A Separate Thread! */
|
// get valid path
|
auto full_record_path = get_full_record_path();
|
|
while (status == vp_record_task_status::STARTED) {
|
// it is a consumer
|
cache_semaphore.wait();
|
|
auto frame_to_record = frames_to_record.front();
|
if (frame_to_record == nullptr) {
|
//dead flag
|
continue;
|
}
|
|
cv::Mat frame_data;
|
// preprocess, vp_frame_meta -> cv::Mat
|
preprocess(frame_to_record, frame_data);
|
frames_to_record.pop_front();
|
|
// write to disk
|
cv::imwrite(full_record_path, frame_data);
|
VP_DEBUG(vp_utils::string_format("[%s] [record] already written frame for `%s`", host_node_name.c_str(), get_full_record_path().c_str()));
|
/* for image recording, just saving only 1 frame and then complete
|
* refer to vp_video_record_task
|
*/
|
|
vp_record_info record_info;
|
record_info.record_type = vp_record_type::IMAGE;
|
notify_task_complete(record_info);
|
// loop end
|
}
|
}
|
|
std::string vp_image_record_task::get_file_ext() {
|
// save as jpg
|
return ".jpg";
|
}
|
}
|
