/home/rex/ebltrunk/tools/libeblearngui/include/detector_gui.h

/***************************************************************************
 *   Copyright (C) 2009 by Pierre Sermanet   *
 *   pierre.sermanet@gmail.com   *
 *   All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Redistribution under a license not approved by the Open Source
 *       Initiative (http://www.opensource.org) must display the
 *       following acknowledgement in all advertising material:
 *        This product includes software developed at the Courant
 *        Institute of Mathematical Sciences (http://cims.nyu.edu).
 *     * The names of the authors may not be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL ThE AUTHORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 ***************************************************************************/

#ifndef DETECTOR_GUI_H_
#define DETECTOR_GUI_H_

#include "libidxgui.h"
#include "libeblearn.h"
#include "detector.h"
#include "ebl_arch_gui.h"
#include "utils.h"

#include <deque>

using namespace std;

namespace ebl {

  //EXPORT extern ubyte color_list[12][3];

  EXPORT void draw_bbox_parts(bbox_parts &bb, vector<string> &labels, uint h0,
                              uint w0, double dzoom);
  EXPORT void draw_bbox(bbox &bb, vector<string> &labels, uint h0, uint w0,
                        double dzoom, float transparency, bool scaled = true,
                        uint color_offset = 0, bool show_class = true,
                        bool show_conf = true);

  // detector_gui

  template <typename T, class Tstate = fstate_idx<T> >
    class EXPORT detector_gui {
  public:
    detector_gui(uint draw_extracted = 0,
                 bool show_detqueue = false, uint step = 1, uint qheight = 5,
                 uint qwidth = 5, bool show_detqueue2 = false, uint step2 = 1,
                 uint qheight2 = 5, uint qwidth2 = 5, bool show_class = true,
                 bool show_conf = true);

    virtual ~detector_gui();

    template <typename Tin>
      bboxes& display(detector<T,Tstate> &cl, idx<Tin> &img,
                      double threshold, const char *frame_name = NULL,
                      uint h0 = 0, uint w0 = 0, double dzoom = 1.0, T vmin = 0,
                      T vmax = 0, int wid = -1, const char *wname = NULL,
                      float transparency = 1.0);

    template <typename Tin>
      void display_groundtruth(detector<T,Tstate> &cl, idx<Tin> &img,
                               bboxes &groundtruth, bboxes &filtered,
                               bboxes &nonfiltered, bboxes &pos, bboxes &neg,
                               svector<midx<T> > &pp_pos,
                               svector<midx<T> > &pp_neg,
                               uint h0 = 0, uint w0 = 0, double dzoom = 1.0,
                               T vmin = 0, T vmax = 0, int wid = -1);
  
    template <typename Tin>
      static void display_minimal(idx<Tin> &img, bboxes& vb,
                                  vector<string> &labels,
                                  uint &h0, uint &w0,
                                  double dzoom = 1.0, T vmin = 0, T vmax = 0,
                                  int wid = -1, bool show_parts = false,
                                  float transparency = 1.0,
                                  bool show_class = true,
                                  bool show_conf = true,
                                  bboxes *bb2 = NULL);

    template <typename Tin>
      bboxes& display_input(detector<T,Tstate> &cl, idx<Tin> &img,
                            double threshold,
                            const char *frame_name = NULL,
                            uint h0 = 0, uint w0 = 0,
                            double dzoom = 1.0, T vmin = 0,
                            T vmax = 0,
                            int wid = -1, const char *wname = NULL,
                            float transparency = 1.0);

    void display_preprocessed(svector<midx<T> > &pp, bboxes &bbs,
                              vector<string> &labels, uint &h0, uint &w0,
                              double dzoom = 1.0, T vmin = 0, T vmax = 0,
                              uint wmax = 3000);

    template <typename Tin>
      bboxes& display_inputs_outputs(detector<T,Tstate> &cl, idx<Tin> &img,
                                     double threshold,
                                     const char *frame_name = NULL,
                                     uint h0 = 0, uint w0 = 0,
                                     double dzoom = 1.0, T vmin = 0,
                                     T vmax = 0, int wid = -1,
                                     const char *wname = NULL,
                                     T in_vmin = 0, T in_vmax = 255,
                                     float transparency = 1.0,
                                     uint wmax = 3000);

    void display_inputs(detector<T,Tstate> &cl, uint &h0, uint &w0,
                        bboxes &bb, double dzoom = 1.0, T vmin = 0, T vmax = 0,
                        float transparency = 1.0);
    void display_outputs(detector<T,Tstate> &cl, uint &h0, uint &w0,
                         bboxes &bb, double dzoom = 1.0, T vmin = 0, T vmax = 0,
                         float transparency = 1.0);

    template <typename Tin>
      bboxes& display_all(detector<T,Tstate> &cl, idx<Tin> &img,
                          double threshold, const char *frame_name = NULL,
                          uint h0 = 0, uint w0 = 0, double dzoom = 1.0,
                          T vmin = 0, T vmax = 0,
                          int wid = -1,const char *wname = NULL);

    template <typename Tin>
      void display_current(detector<T,Tstate> &cl,
                           idx<Tin> &sample,
                           int wid = -1,
                           const char *wname = NULL, double dzoom = 1.0);

    void set_mask_class(const char *name, T threshold);

    // private methods
  private:

    void update_and_display_queue(deque<idx<T> > &queue, uint step,
                                  uint qheight, uint qwidth,
                                  vector<idx<T> > &new_detections,
                                  uint detcnt, uint &h0, uint &w0,
                                  double dzoom);

    // members
  private:
    int         display_wid;    
    int         display_wid_fprop;      
    int         display_wid_gt; 
    uint        draw_extracted;
    bool        show_detqueue;  
    bool        show_detqueue2; 
    uint        step;           
    uint        step2;          
    uint        qheight;        
    uint        qwidth;         
    uint        qheight2;       
    uint        qwidth2;        
    deque<idx<T> > detqueue;    
    deque<idx<T> > detqueue2;   
    uint        detcnt;         
    string      mask_class;
    T           mask_threshold;
    bool        show_class;     
    bool        show_conf;     
  };

} // end namespace ebl

#include "detector_gui.hpp"

#endif /* DETECTOR_GUI_H_ */