00001 // This is brl/bseg/strk/strk_info_tracker_process.h 00002 #ifndef strk_info_tracker_process_h_ 00003 #define strk_info_tracker_process_h_ 00004 //---------------------------------------------------------------------------- 00005 //: 00006 // \file 00007 // \brief tracks a correlation window across frames 00008 // \author J.L. Mundy 00009 // 00010 // \verbatim 00011 // Modifications 00012 // J.L. Mundy - August 21, 2002 - Initial version. 00013 // \endverbatim 00014 //--------------------------------------------------------------------------- 00015 #include <vtol/vtol_face_2d_sptr.h> 00016 #include <strk/strk_info_tracker_params.h> 00017 #include <strk/strk_info_tracker.h> 00018 #include <vpro/vpro_video_process.h> 00019 00020 class strk_info_tracker_process : public vpro_video_process 00021 { 00022 public: 00023 strk_info_tracker_process(strk_info_tracker_params & tp); 00024 ~strk_info_tracker_process(); 00025 virtual process_data_type get_output_type() { return TOPOLOGY; } 00026 00027 //: track to next frame 00028 virtual bool execute(); 00029 virtual bool finish(); 00030 bool set_track_output_file(vcl_string const& file_name); 00031 bool set_hist_output_file(vcl_string const& file_name); 00032 00033 //: special methods 00034 // for display 00035 vcl_vector<unsigned char> color_index(){return color_index_;} 00036 private: 00037 //members 00038 bool write_tracked_faces_; 00039 bool write_tracked_hist_; 00040 vcl_string track_file_; 00041 vcl_string hist_file_; 00042 bool failure_; 00043 bool first_frame_; 00044 strk_info_tracker tracker_; 00045 int n_verts_; 00046 int start_frame_; 00047 vcl_vector<vtol_face_2d_sptr> tracked_faces_; 00048 vcl_vector<vcl_vector<float> > tracked_hist_; 00049 vcl_vector<unsigned char> color_index_; 00050 }; 00051 00052 00053 #endif // strk_info_tracker_process_h_
1.4.4