| /* |
| * main.cpp - test |
| * |
| * Copyright (c) 2014 Intel Corporation |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| * Author: Wind Yuan <[email protected]> |
| * Author: Yinhang Liu <[email protected]> |
| * Author: Wei Zong <[email protected]> |
| */ |
| |
| #include "device_manager.h" |
| #include "uvc_device.h" |
| #include "fake_v4l2_device.h" |
| #include "x3a_analyzer_simple.h" |
| #include "analyzer_loader.h" |
| #include "smart_analyzer_loader.h" |
| #if HAVE_IA_AIQ |
| #include "isp/atomisp_device.h" |
| #include "isp/isp_controller.h" |
| #include "isp/isp_image_processor.h" |
| #include "isp/isp_poll_thread.h" |
| #include "isp/x3a_analyzer_aiq.h" |
| #include "x3a_analyze_tuner.h" |
| #include "dynamic_analyzer_loader.h" |
| #include "isp/hybrid_analyzer_loader.h" |
| #endif |
| #if HAVE_LIBCL |
| #include "ocl/cl_3a_image_processor.h" |
| #include "ocl/cl_post_image_processor.h" |
| #include "ocl/cl_csc_image_processor.h" |
| #include "ocl/cl_tnr_handler.h" |
| #endif |
| #if HAVE_LIBDRM |
| #include "drm_display.h" |
| #endif |
| #include "fake_poll_thread.h" |
| #include "image_file_handle.h" |
| #include <base/xcam_3a_types.h> |
| #include <unistd.h> |
| #include <signal.h> |
| #include <stdlib.h> |
| #include <string> |
| #include <getopt.h> |
| #include "test_common.h" |
| |
| using namespace XCam; |
| |
| #define IMX185_WDR_CPF "/etc/atomisp/imx185_wdr.cpf" |
| |
| static Mutex g_mutex; |
| static Cond g_cond; |
| static bool g_stop = false; |
| |
| class MainDeviceManager |
| : public DeviceManager |
| { |
| public: |
| MainDeviceManager () |
| : _save_file (false) |
| , _interval (1) |
| , _frame_width (0) |
| , _frame_height (0) |
| , _frame_count (0) |
| , _frame_save (0) |
| , _enable_display (false) |
| { |
| #if HAVE_LIBDRM |
| _display = DrmDisplay::instance (); |
| #endif |
| XCAM_OBJ_PROFILING_INIT; |
| } |
| |
| ~MainDeviceManager () { |
| _file_handle.close (); |
| } |
| |
| void enable_save_file (bool enable) { |
| _save_file = enable; |
| } |
| |
| void set_interval (uint32_t inteval) { |
| _interval = inteval; |
| } |
| |
| void set_frame_width (uint32_t frame_width) { |
| _frame_width = frame_width; |
| } |
| |
| void set_frame_height (uint32_t frame_height) { |
| _frame_height = frame_height; |
| } |
| |
| void set_frame_save (uint32_t frame_save) { |
| _frame_save = frame_save; |
| } |
| |
| void enable_display(bool value) { |
| _enable_display = value; |
| } |
| |
| #if HAVE_LIBDRM |
| void set_display_mode(DrmDisplayMode mode) { |
| _display->set_display_mode (mode); |
| } |
| #endif |
| |
| protected: |
| virtual void handle_message (const SmartPtr<XCamMessage> &msg); |
| virtual void handle_buffer (const SmartPtr<VideoBuffer> &buf); |
| |
| int display_buf (const SmartPtr<VideoBuffer> &buf); |
| |
| private: |
| void open_file (); |
| |
| private: |
| bool _save_file; |
| uint32_t _interval; |
| uint32_t _frame_width; |
| uint32_t _frame_height; |
| uint32_t _frame_count; |
| uint32_t _frame_save; |
| bool _enable_display; |
| ImageFileHandle _file_handle; |
| #if HAVE_LIBDRM |
| SmartPtr<DrmDisplay> _display; |
| #endif |
| XCAM_OBJ_PROFILING_DEFINES; |
| }; |
| |
| void |
| MainDeviceManager::handle_message (const SmartPtr<XCamMessage> &msg) |
| { |
| XCAM_UNUSED (msg); |
| } |
| |
| void |
| MainDeviceManager::handle_buffer (const SmartPtr<VideoBuffer> &buf) |
| { |
| if (!buf.ptr ()) { |
| XCAM_LOG_WARNING ("video buffer is null, handle buffer failed."); |
| return; |
| } |
| |
| FPS_CALCULATION (fps_buf, 30); |
| XCAM_OBJ_PROFILING_START; |
| |
| if (_enable_display) |
| display_buf (buf); |
| |
| XCAM_OBJ_PROFILING_END("main_dev_manager_display", XCAM_OBJ_DUR_FRAME_NUM); |
| |
| if (!_save_file) |
| return ; |
| |
| if ((_frame_count++ % _interval) != 0) |
| return; |
| |
| if ((_frame_save != 0) && (_frame_count > _frame_save)) { |
| SmartLock locker (g_mutex); |
| g_stop = true; |
| g_cond.broadcast (); |
| return; |
| } |
| |
| open_file (); |
| |
| if (!_file_handle.is_valid ()) { |
| XCAM_LOG_ERROR ("open file failed"); |
| return; |
| } |
| _file_handle.write_buf (buf); |
| } |
| |
| int |
| MainDeviceManager::display_buf (const SmartPtr<VideoBuffer> &data) |
| { |
| #if HAVE_LIBDRM |
| XCamReturn ret = XCAM_RETURN_NO_ERROR; |
| SmartPtr<VideoBuffer> buf = data; |
| const VideoBufferInfo & frame_info = buf->get_video_info (); |
| struct v4l2_rect rect = { 0, 0, frame_info.width, frame_info.height}; |
| |
| if (!_display->is_render_inited ()) { |
| ret = _display->render_init (0, 0, this->_frame_width, this->_frame_height, |
| frame_info.format, &rect); |
| CHECK (ret, "display failed on render_init"); |
| } |
| ret = _display->render_setup_frame_buffer (buf); |
| CHECK (ret, "display failed on framebuf set"); |
| ret = _display->render_buffer (buf); |
| CHECK (ret, "display failed on rendering"); |
| #else |
| XCAM_UNUSED (data); |
| #endif |
| |
| return 0; |
| } |
| |
| |
| void |
| MainDeviceManager::open_file () |
| { |
| if (_file_handle.is_valid () && (_frame_save == 0)) |
| return; |
| |
| std::string file_name = DEFAULT_SAVE_FILE_NAME; |
| |
| if (_frame_save != 0) { |
| file_name += std::to_string(_frame_count); |
| } |
| file_name += ".raw"; |
| |
| if (_file_handle.open (file_name.c_str (), "wb") != XCAM_RETURN_NO_ERROR) { |
| XCAM_LOG_WARNING ("create file(%s) failed", file_name.c_str ()); |
| } |
| } |
| |
| #define V4L2_CAPTURE_MODE_STILL 0x2000 |
| #define V4L2_CAPTURE_MODE_VIDEO 0x4000 |
| #define V4L2_CAPTURE_MODE_PREVIEW 0x8000 |
| |
| typedef enum { |
| AnalyzerTypeSimple = 0, |
| AnalyzerTypeAiqTuner, |
| AnalyzerTypeDynamic, |
| AnalyzerTypeHybrid, |
| } AnalyzerType; |
| |
| void dev_stop_handler(int sig) |
| { |
| XCAM_UNUSED (sig); |
| |
| SmartLock locker (g_mutex); |
| g_stop = true; |
| g_cond.broadcast (); |
| |
| //exit(0); |
| } |
| |
| void print_help (const char *bin_name) |
| { |
| printf ("Usage: %s [-a analyzer]\n" |
| "Configurations:\n" |
| "\t -a analyzer specify a analyzer\n" |
| "\t select from [simple" |
| #if HAVE_IA_AIQ |
| ", aiq" |
| #if HAVE_LIBCL |
| ", dynamic, hybrid" |
| #endif |
| #endif |
| "], default is [simple]\n" |
| "\t -m mem_type specify video memory type\n" |
| "\t mem_type select from [dma, mmap], default is [mmap]\n" |
| "\t -s save file to %s\n" |
| "\t -n interval save file on every [interval] frame\n" |
| "\t -f pixel_fmt specify output pixel format\n" |
| "\t pixel_fmt select from [NV12, YUYV, BA10, BA12], default is [NV12]\n" |
| "\t -W image_width specify image width, default is [1920]\n" |
| "\t -H image_height specify image height, default is [1080]\n" |
| "\t -d cap_mode specify capture mode\n" |
| "\t cap_mode select from [video, still], default is [video]\n" |
| "\t -i frame_save specify the frame count to save, default is 0 which means endless\n" |
| "\t -p preview on enable local display, need root privilege\n" |
| "\t --usb specify node for usb camera device, enables capture path through USB camera \n" |
| "\t specify [/dev/video4, /dev/video5] depending on which node USB camera is attached\n" |
| "\t -e display_mode preview mode\n" |
| "\t select from [primary, overlay], default is [primary]\n" |
| "\t --sync set analyzer in sync mode\n" |
| "\t -r raw_input specify the path of raw image as fake source instead of live camera\n" |
| "\t -h help\n" |
| #if HAVE_LIBCL |
| "CL features:\n" |
| "\t -c process image with cl kernel\n" |
| #if HAVE_IA_AIQ |
| "\t -b brightness specify brightness level\n" |
| "\t brightness level select from [0, 256], default is [128]\n" |
| #endif |
| "\t --capture specify the capture stage of image\n" |
| "\t capture_stage select from [bayer, tonemapping], default is [tonemapping]\n" |
| "\t --tnr specify temporal noise reduction type, default is tnr off\n" |
| "\t only support [yuv]\n" |
| "\t --tnr-level specify tnr level\n" |
| "\t --wdr-mode specify wdr mode. select from [gaussian, haleq]\n" |
| "\t --enable-bnr enable bayer noise reduction\n" |
| "\t --defog-mode specify defog mode\n" |
| "\t select from [disabled, retinex, dcp], default is [disabled]\n" |
| "\t --wavelet-mode specify wavelet denoise mode, default is off\n" |
| "\t select from [0:disable, 1:Hat Y, 2:Hat UV, 3:Haar Y, 4:Haar UV, 5:Haar YUV, 6:Haar Bayes Shrink]\n" |
| "\t --3d-denoise specify 3D Denoise mode\n" |
| "\t select from [disabled, yuv, uv], default is [disabled]\n" |
| "\t --enable-wireframe enable wire frame\n" |
| "\t --pipeline specify pipe mode\n" |
| "\t select from [basic, advance, extreme], default is [basic]\n" |
| "\t --disable-post disable cl post image processor\n" |
| #endif |
| , bin_name |
| , DEFAULT_SAVE_FILE_NAME); |
| } |
| |
| int main (int argc, char *argv[]) |
| { |
| XCamReturn ret = XCAM_RETURN_NO_ERROR; |
| SmartPtr<V4l2Device> device; |
| #if HAVE_IA_AIQ |
| SmartPtr<V4l2SubDevice> event_device; |
| SmartPtr<IspController> isp_controller; |
| SmartPtr<ImageProcessor> isp_processor; |
| #endif |
| SmartPtr<X3aAnalyzer> analyzer; |
| SmartPtr<AnalyzerLoader> loader; |
| AnalyzerType analyzer_type = AnalyzerTypeSimple; |
| |
| #if HAVE_LIBCL |
| bool have_cl_processor = false; |
| SmartPtr<SmartAnalyzer> smart_analyzer; |
| bool have_cl_post_processor = true; |
| SmartPtr<CL3aImageProcessor> cl_processor; |
| SmartPtr<CLPostImageProcessor> cl_post_processor; |
| uint32_t tnr_type = CL_TNR_DISABLE; |
| uint32_t denoise_type = 0; |
| uint8_t tnr_level = 0; |
| CL3aImageProcessor::PipelineProfile pipeline_mode = CL3aImageProcessor::BasicPipelineProfile; |
| CL3aImageProcessor::CaptureStage capture_stage = CL3aImageProcessor::TonemappingStage; |
| CL3aImageProcessor::CLTonemappingMode wdr_mode = CL3aImageProcessor::WDRdisabled; |
| |
| #if HAVE_IA_AIQ |
| int32_t brightness_level = 128; |
| #endif |
| uint32_t defog_type = 0; |
| CLWaveletBasis wavelet_mode = CL_WAVELET_DISABLED; |
| uint32_t wavelet_channel = CL_IMAGE_CHANNEL_UV; |
| bool wavelet_bayes_shrink = false; |
| uint32_t denoise_3d_mode = 0; |
| uint8_t denoise_3d_ref_count = 3; |
| bool wireframe_type = false; |
| bool image_warp_type = false; |
| #endif |
| |
| bool need_display = false; |
| #if HAVE_LIBDRM |
| DrmDisplayMode display_mode = DRM_DISPLAY_MODE_PRIMARY; |
| #endif |
| enum v4l2_memory v4l2_mem_type = V4L2_MEMORY_MMAP; |
| const char *bin_name = argv[0]; |
| uint32_t capture_mode = V4L2_CAPTURE_MODE_VIDEO; |
| uint32_t pixel_format = V4L2_PIX_FMT_NV12; |
| |
| bool have_usbcam = 0; |
| std::string usb_device_name; |
| bool sync_mode = false; |
| bool save_file = false; |
| uint32_t interval_frames = 1; |
| uint32_t save_frames = 0; |
| uint32_t frame_rate; |
| uint32_t frame_width = 1920; |
| uint32_t frame_height = 1080; |
| std::string path_to_fake; |
| |
| int opt; |
| const char *short_opts = "sca:n:m:f:W:H:d:b:pi:e:r:h"; |
| const struct option long_opts[] = { |
| {"tnr", required_argument, NULL, 'T'}, |
| {"tnr-level", required_argument, NULL, 'L'}, |
| {"wdr-mode", required_argument, NULL, 'w'}, |
| {"enable-bnr", no_argument, NULL, 'B'}, |
| {"defog-mode", required_argument, NULL, 'X'}, |
| {"wavelet-mode", required_argument, NULL, 'V'}, |
| {"3d-denoise", required_argument, NULL, 'N'}, |
| {"enable-wireframe", no_argument, NULL, 'F'}, |
| {"enable-warp", no_argument, NULL, 'A'}, |
| {"usb", required_argument, NULL, 'U'}, |
| {"sync", no_argument, NULL, 'Y'}, |
| {"capture", required_argument, NULL, 'C'}, |
| {"pipeline", required_argument, NULL, 'P'}, |
| {"disable-post", no_argument, NULL, 'O'}, |
| {0, 0, 0, 0}, |
| }; |
| |
| while ((opt = getopt_long(argc, argv, short_opts, long_opts, NULL)) != -1) { |
| switch (opt) { |
| case 'a': { |
| XCAM_ASSERT (optarg); |
| if (!strcmp (optarg, "simple")) |
| analyzer_type = AnalyzerTypeSimple; |
| #if HAVE_IA_AIQ |
| else if (!strcmp (optarg, "aiq")) |
| analyzer_type = AnalyzerTypeAiqTuner; |
| #if HAVE_LIBCL |
| else if (!strcmp (optarg, "dynamic")) |
| analyzer_type = AnalyzerTypeDynamic; |
| else if (!strcmp (optarg, "hybrid")) |
| analyzer_type = AnalyzerTypeHybrid; |
| #endif |
| #endif |
| else { |
| print_help (bin_name); |
| return -1; |
| } |
| break; |
| } |
| |
| case 'm': { |
| XCAM_ASSERT (optarg); |
| if (!strcmp (optarg, "dma")) |
| v4l2_mem_type = V4L2_MEMORY_DMABUF; |
| else if (!strcmp (optarg, "mmap")) |
| v4l2_mem_type = V4L2_MEMORY_MMAP; |
| else |
| print_help (bin_name); |
| break; |
| } |
| |
| case 's': |
| save_file = true; |
| break; |
| case 'n': |
| XCAM_ASSERT (optarg); |
| interval_frames = atoi(optarg); |
| break; |
| case 'i': |
| XCAM_ASSERT (optarg); |
| save_frames = atoi(optarg); |
| break; |
| case 'f': |
| XCAM_ASSERT (optarg); |
| CHECK_EXP ((strlen(optarg) == 4), "invalid pixel format\n"); |
| pixel_format = v4l2_fourcc ((unsigned)optarg[0], |
| (unsigned)optarg[1], |
| (unsigned)optarg[2], |
| (unsigned)optarg[3]); |
| break; |
| case 'd': |
| XCAM_ASSERT (optarg); |
| if (!strcmp (optarg, "still")) |
| capture_mode = V4L2_CAPTURE_MODE_STILL; |
| else if (!strcmp (optarg, "video")) |
| capture_mode = V4L2_CAPTURE_MODE_VIDEO; |
| else { |
| print_help (bin_name); |
| return -1; |
| } |
| break; |
| case 'U': |
| XCAM_ASSERT (optarg); |
| have_usbcam = true; |
| usb_device_name = optarg; |
| XCAM_LOG_DEBUG("using USB camera plugged in at node: %s", XCAM_STR(usb_device_name.c_str())); |
| break; |
| case 'W': |
| XCAM_ASSERT (optarg); |
| frame_width = atoi(optarg); |
| break; |
| case 'H': |
| XCAM_ASSERT (optarg); |
| frame_height = atoi(optarg); |
| break; |
| case 'e': { |
| #if HAVE_LIBDRM |
| XCAM_ASSERT (optarg); |
| if (!strcmp (optarg, "primary")) |
| display_mode = DRM_DISPLAY_MODE_PRIMARY; |
| else if (!strcmp (optarg, "overlay")) |
| display_mode = DRM_DISPLAY_MODE_OVERLAY; |
| else { |
| print_help (bin_name); |
| return -1; |
| } |
| #else |
| XCAM_LOG_WARNING ("preview is not supported"); |
| #endif |
| break; |
| } |
| |
| case 'Y': |
| sync_mode = true; |
| break; |
| #if HAVE_LIBCL |
| case 'c': |
| have_cl_processor = true; |
| break; |
| #if HAVE_IA_AIQ |
| case 'b': |
| XCAM_ASSERT (optarg); |
| brightness_level = atoi(optarg); |
| if(brightness_level < 0 || brightness_level > 256) { |
| print_help (bin_name); |
| return -1; |
| } |
| break; |
| #endif |
| |
| case 'B': { |
| denoise_type |= XCAM_DENOISE_TYPE_BNR; |
| break; |
| } |
| case 'X': { |
| XCAM_ASSERT (optarg); |
| defog_type = true; |
| if (!strcmp (optarg, "disabled")) |
| defog_type = CLPostImageProcessor::DefogDisabled; |
| else if (!strcmp (optarg, "retinex")) |
| defog_type = CLPostImageProcessor::DefogRetinex; |
| else if (!strcmp (optarg, "dcp")) |
| defog_type = CLPostImageProcessor::DefogDarkChannelPrior; |
| else { |
| print_help (bin_name); |
| return -1; |
| } |
| break; |
| } |
| case 'V': { |
| XCAM_ASSERT (optarg); |
| if (atoi(optarg) < 0 || atoi(optarg) > 255) { |
| print_help (bin_name); |
| return -1; |
| } |
| if (atoi(optarg) == 1) { |
| wavelet_mode = CL_WAVELET_HAT; |
| wavelet_channel = CL_IMAGE_CHANNEL_Y; |
| } else if (atoi(optarg) == 2) { |
| wavelet_mode = CL_WAVELET_HAT; |
| wavelet_channel = CL_IMAGE_CHANNEL_UV; |
| } else if (atoi(optarg) == 3) { |
| wavelet_mode = CL_WAVELET_HAAR; |
| wavelet_channel = CL_IMAGE_CHANNEL_Y; |
| } else if (atoi(optarg) == 4) { |
| wavelet_mode = CL_WAVELET_HAAR; |
| wavelet_channel = CL_IMAGE_CHANNEL_UV; |
| } else if (atoi(optarg) == 5) { |
| wavelet_mode = CL_WAVELET_HAAR; |
| wavelet_channel = CL_IMAGE_CHANNEL_UV | CL_IMAGE_CHANNEL_Y; |
| } else if (atoi(optarg) == 6) { |
| wavelet_mode = CL_WAVELET_HAAR; |
| wavelet_channel = CL_IMAGE_CHANNEL_UV | CL_IMAGE_CHANNEL_Y; |
| wavelet_bayes_shrink = true; |
| } else { |
| wavelet_mode = CL_WAVELET_DISABLED; |
| } |
| break; |
| } |
| case 'N': { |
| XCAM_ASSERT (optarg); |
| denoise_3d_mode = true; |
| if (!strcmp (optarg, "disabled")) |
| denoise_3d_mode = CLPostImageProcessor::Denoise3DDisabled; |
| else if (!strcmp (optarg, "yuv")) |
| denoise_3d_mode = CLPostImageProcessor::Denoise3DYuv; |
| else if (!strcmp (optarg, "uv")) |
| denoise_3d_mode = CLPostImageProcessor::Denoise3DUV; |
| else { |
| print_help (bin_name); |
| return -1; |
| } |
| break; |
| } |
| case 'F': { |
| wireframe_type = true; |
| break; |
| } |
| case 'A': { |
| image_warp_type = true; |
| break; |
| } |
| case 'T': { |
| XCAM_ASSERT (optarg); |
| if (!strcasecmp (optarg, "yuv")) |
| tnr_type = CL_TNR_TYPE_YUV; |
| else { |
| printf ("--tnr only support <yuv>, <%s> is not supported\n", optarg); |
| print_help (bin_name); |
| return -1; |
| } |
| break; |
| } |
| case 'L': { |
| XCAM_ASSERT (optarg); |
| if (atoi(optarg) < 0 || atoi(optarg) > 255) { |
| print_help (bin_name); |
| return -1; |
| } |
| tnr_level = atoi(optarg); |
| break; |
| } |
| case 'w': { |
| XCAM_ASSERT (optarg); |
| if (!strcasecmp (optarg, "gaussian")) |
| wdr_mode = CL3aImageProcessor::Gaussian; |
| else if (!strcasecmp (optarg, "haleq")) |
| wdr_mode = CL3aImageProcessor::Haleq; |
| |
| pixel_format = V4L2_PIX_FMT_SGRBG12; |
| setenv ("AIQ_CPF_PATH", IMX185_WDR_CPF, 1); |
| break; |
| } |
| case 'P': { |
| XCAM_ASSERT (optarg); |
| if (!strcasecmp (optarg, "basic")) |
| pipeline_mode = CL3aImageProcessor::BasicPipelineProfile; |
| else if (!strcasecmp (optarg, "advance")) |
| pipeline_mode = CL3aImageProcessor::AdvancedPipelineProfile; |
| else if (!strcasecmp (optarg, "extreme")) |
| pipeline_mode = CL3aImageProcessor::ExtremePipelineProfile; |
| else { |
| print_help (bin_name); |
| return -1; |
| } |
| break; |
| } |
| case 'C': { |
| XCAM_ASSERT (optarg); |
| if (!strcmp (optarg, "bayer")) |
| capture_stage = CL3aImageProcessor::BasicbayerStage; |
| break; |
| } |
| case 'O': { |
| have_cl_post_processor = false; |
| break; |
| } |
| #endif |
| case 'r': { |
| XCAM_ASSERT (optarg); |
| XCAM_LOG_INFO ("use raw image %s as input source", optarg); |
| path_to_fake = optarg; |
| break; |
| } |
| case 'p': { |
| #if HAVE_LIBDRM |
| need_display = true; |
| #else |
| XCAM_LOG_WARNING ("preview is not supported, disable preview now"); |
| need_display = false; |
| #endif |
| break; |
| } |
| case 'h': |
| print_help (bin_name); |
| return 0; |
| |
| default: |
| print_help (bin_name); |
| return -1; |
| } |
| } |
| |
| SmartPtr<MainDeviceManager> device_manager = new MainDeviceManager (); |
| device_manager->enable_save_file (save_file); |
| device_manager->set_interval (interval_frames); |
| device_manager->set_frame_save (save_frames); |
| device_manager->set_frame_width (frame_width); |
| device_manager->set_frame_height (frame_height); |
| |
| if (!device.ptr ()) { |
| if (path_to_fake.c_str ()) { |
| device = new FakeV4l2Device (); |
| } else if (have_usbcam) { |
| device = new UVCDevice (usb_device_name.c_str ()); |
| } |
| #if HAVE_IA_AIQ |
| else { |
| if (capture_mode == V4L2_CAPTURE_MODE_STILL) |
| device = new AtomispDevice (CAPTURE_DEVICE_STILL); |
| else if (capture_mode == V4L2_CAPTURE_MODE_VIDEO) |
| device = new AtomispDevice (CAPTURE_DEVICE_VIDEO); |
| else |
| device = new AtomispDevice (DEFAULT_CAPTURE_DEVICE); |
| } |
| #endif |
| } |
| |
| #if HAVE_IA_AIQ |
| if (!isp_controller.ptr ()) |
| isp_controller = new IspController (device); |
| #endif |
| |
| switch (analyzer_type) { |
| case AnalyzerTypeSimple: |
| analyzer = new X3aAnalyzerSimple (); |
| break; |
| #if HAVE_IA_AIQ |
| case AnalyzerTypeAiqTuner: { |
| SmartPtr<X3aAnalyzer> aiq_analyzer = new X3aAnalyzerAiq (isp_controller, DEFAULT_CPF_FILE); |
| SmartPtr<X3aAnalyzeTuner> tuner_analyzer = new X3aAnalyzeTuner (); |
| XCAM_ASSERT (aiq_analyzer.ptr () && tuner_analyzer.ptr ()); |
| tuner_analyzer->set_analyzer (aiq_analyzer); |
| analyzer = tuner_analyzer; |
| break; |
| } |
| #if HAVE_LIBCL |
| case AnalyzerTypeDynamic: { |
| const char *path_of_3a = DEFAULT_DYNAMIC_3A_LIB; |
| SmartPtr<DynamicAnalyzerLoader> dynamic_loader = new DynamicAnalyzerLoader (path_of_3a); |
| loader = dynamic_loader.dynamic_cast_ptr<AnalyzerLoader> (); |
| analyzer = dynamic_loader->load_analyzer (loader); |
| CHECK_EXP (analyzer.ptr (), "load dynamic 3a lib(%s) failed", path_of_3a); |
| break; |
| } |
| case AnalyzerTypeHybrid: { |
| const char *path_of_3a = DEFAULT_HYBRID_3A_LIB; |
| SmartPtr<HybridAnalyzerLoader> hybrid_loader = new HybridAnalyzerLoader (path_of_3a); |
| hybrid_loader->set_cpf_path (DEFAULT_CPF_FILE); |
| hybrid_loader->set_isp_controller (isp_controller); |
| loader = hybrid_loader.dynamic_cast_ptr<AnalyzerLoader> (); |
| analyzer = hybrid_loader->load_analyzer (loader); |
| CHECK_EXP (analyzer.ptr (), "load hybrid 3a lib(%s) failed", path_of_3a); |
| break; |
| } |
| #endif |
| #endif |
| default: |
| print_help (bin_name); |
| return -1; |
| } |
| XCAM_ASSERT (analyzer.ptr ()); |
| analyzer->set_sync_mode (sync_mode); |
| |
| #if HAVE_LIBCL |
| SmartHandlerList smart_handlers = SmartAnalyzerLoader::load_smart_handlers (DEFAULT_SMART_ANALYSIS_LIB_DIR); |
| if (!smart_handlers.empty ()) { |
| smart_analyzer = new SmartAnalyzer (); |
| if (smart_analyzer.ptr ()) { |
| SmartHandlerList::iterator i_handler = smart_handlers.begin (); |
| for (; i_handler != smart_handlers.end (); ++i_handler) |
| { |
| XCAM_ASSERT ((*i_handler).ptr ()); |
| smart_analyzer->add_handler (*i_handler); |
| } |
| } else { |
| XCAM_LOG_WARNING ("load smart analyzer(%s) failed, please check.", DEFAULT_SMART_ANALYSIS_LIB_DIR); |
| } |
| } |
| |
| if (smart_analyzer.ptr ()) { |
| if (smart_analyzer->prepare_handlers () != XCAM_RETURN_NO_ERROR) { |
| XCAM_LOG_WARNING ("analyzer(%s) prepare handlers failed", smart_analyzer->get_name ()); |
| } |
| device_manager->set_smart_analyzer (smart_analyzer); |
| } |
| #endif |
| |
| signal(SIGINT, dev_stop_handler); |
| |
| device->set_sensor_id (0); |
| device->set_capture_mode (capture_mode); |
| //device->set_mem_type (V4L2_MEMORY_DMABUF); |
| device->set_mem_type (v4l2_mem_type); |
| device->set_buffer_count (8); |
| if (pixel_format == V4L2_PIX_FMT_SGRBG12) { |
| frame_rate = 30; |
| device->set_framerate (frame_rate, 1); |
| } |
| #if HAVE_LIBCL |
| else { |
| frame_rate = 25; |
| device->set_framerate (frame_rate, 1); |
| if(wdr_mode != CL3aImageProcessor::WDRdisabled) { |
| XCAM_LOG_WARNING("Tonemapping is only applicable under BA12 format. Disable tonemapping automatically."); |
| wdr_mode = CL3aImageProcessor::WDRdisabled; |
| } |
| } |
| #endif |
| ret = device->open (); |
| CHECK (ret, "device(%s) open failed", device->get_device_name()); |
| ret = device->set_format (frame_width, frame_height, pixel_format, V4L2_FIELD_NONE, frame_width * 2); |
| CHECK (ret, "device(%s) set format failed", device->get_device_name()); |
| |
| #if HAVE_IA_AIQ |
| if (!event_device.ptr ()) |
| event_device = new V4l2SubDevice (DEFAULT_EVENT_DEVICE); |
| ret = event_device->open (); |
| if (ret == XCAM_RETURN_NO_ERROR) { |
| CHECK (ret, "event device(%s) open failed", event_device->get_device_name()); |
| int event = V4L2_EVENT_ATOMISP_3A_STATS_READY; |
| ret = event_device->subscribe_event (event); |
| CHECK_CONTINUE ( |
| ret, |
| "device(%s) subscribe event(%d) failed", |
| event_device->get_device_name(), event); |
| event = V4L2_EVENT_FRAME_SYNC; |
| ret = event_device->subscribe_event (event); |
| CHECK_CONTINUE ( |
| ret, |
| "device(%s) subscribe event(%d) failed", |
| event_device->get_device_name(), event); |
| |
| device_manager->set_event_device (event_device); |
| } |
| #endif |
| |
| device_manager->set_capture_device (device); |
| if (analyzer.ptr()) |
| device_manager->set_3a_analyzer (analyzer); |
| |
| #if HAVE_IA_AIQ |
| #if HAVE_LIBCL |
| if (have_cl_processor) |
| isp_processor = new IspExposureImageProcessor (isp_controller); |
| else |
| #endif |
| isp_processor = new IspImageProcessor (isp_controller); |
| |
| XCAM_ASSERT (isp_processor.ptr ()); |
| device_manager->add_image_processor (isp_processor); |
| #endif |
| #if HAVE_LIBCL |
| if (have_cl_processor) { |
| cl_processor = new CL3aImageProcessor (); |
| cl_processor->set_stats_callback(device_manager); |
| cl_processor->set_denoise (denoise_type); |
| cl_processor->set_capture_stage (capture_stage); |
| |
| cl_processor->set_tonemapping (wdr_mode); |
| if (wdr_mode != CL3aImageProcessor::WDRdisabled) { |
| cl_processor->set_gamma (false); |
| cl_processor->set_3a_stats_bits (12); |
| } |
| |
| cl_processor->set_tnr (tnr_type, tnr_level); |
| cl_processor->set_profile (pipeline_mode); |
| #if HAVE_IA_AIQ |
| analyzer->set_parameter_brightness((brightness_level - 128) / 128.0); |
| #endif |
| device_manager->add_image_processor (cl_processor); |
| } |
| |
| if (have_cl_post_processor) { |
| cl_post_processor = new CLPostImageProcessor (); |
| |
| cl_post_processor->set_stats_callback (device_manager); |
| cl_post_processor->set_defog_mode ((CLPostImageProcessor::CLDefogMode)defog_type); |
| cl_post_processor->set_wavelet (wavelet_mode, wavelet_channel, wavelet_bayes_shrink); |
| cl_post_processor->set_3ddenoise_mode ((CLPostImageProcessor::CL3DDenoiseMode) denoise_3d_mode, denoise_3d_ref_count); |
| |
| cl_post_processor->set_wireframe (wireframe_type); |
| cl_post_processor->set_image_warp (image_warp_type); |
| if (smart_analyzer.ptr () && (wireframe_type || image_warp_type)) { |
| cl_post_processor->set_scaler (true); |
| cl_post_processor->set_scaler_factor (640.0 / frame_width); |
| } |
| |
| if (need_display) { |
| need_display = false; |
| XCAM_LOG_WARNING ("CLVideoBuffer doesn't support local preview, disable local preview now"); |
| } |
| |
| if (need_display) { |
| #if HAVE_LIBDRM |
| if (DrmDisplay::set_preview (need_display)) { |
| device_manager->set_display_mode (display_mode); |
| cl_post_processor->set_output_format (V4L2_PIX_FMT_XBGR32); |
| } else { |
| need_display = false; |
| XCAM_LOG_WARNING ("set preview failed, disable local preview now"); |
| } |
| #else |
| XCAM_LOG_WARNING ("preview is not supported, disable preview now"); |
| need_display = false; |
| #endif |
| } |
| device_manager->enable_display (need_display); |
| |
| device_manager->add_image_processor (cl_post_processor); |
| } |
| #endif |
| |
| SmartPtr<PollThread> poll_thread; |
| if (have_usbcam) { |
| poll_thread = new PollThread (); |
| } else if (path_to_fake.c_str ()) { |
| poll_thread = new FakePollThread (path_to_fake.c_str ()); |
| } |
| #if HAVE_IA_AIQ |
| else { |
| SmartPtr<IspPollThread> isp_poll_thread = new IspPollThread (); |
| isp_poll_thread->set_isp_controller (isp_controller); |
| poll_thread = isp_poll_thread; |
| } |
| #endif |
| device_manager->set_poll_thread (poll_thread); |
| |
| ret = device_manager->start (); |
| CHECK (ret, "device manager start failed"); |
| |
| #if HAVE_LIBCL |
| // hard code exposure range and max gain for imx185 WDR |
| if (wdr_mode != CL3aImageProcessor::WDRdisabled) { |
| if (frame_rate == 30) |
| analyzer->set_ae_exposure_time_range (80 * 1110 * 1000 / 37125, 1120 * 1110 * 1000 / 37125); |
| else |
| analyzer->set_ae_exposure_time_range (80 * 1320 * 1000 / 37125, 1120 * 1320 * 1000 / 37125); |
| analyzer->set_ae_max_analog_gain (3.98); // 12dB |
| } |
| #endif |
| |
| // wait for interruption |
| { |
| SmartLock locker (g_mutex); |
| while (!g_stop) |
| g_cond.wait (g_mutex); |
| } |
| |
| ret = device_manager->stop(); |
| CHECK_CONTINUE (ret, "device manager stop failed"); |
| device->close (); |
| #if HAVE_IA_AIQ |
| event_device->close (); |
| #endif |
| |
| return 0; |
| } |
