/home/rex/ebltrunk/tools/libeblearntools/include/detection_thread.h

/***************************************************************************
 *   Copyright (C) 2010 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 DETECTION_THREAD_H_
#define DETECTION_THREAD_H_

#include <stdio.h>
#include <stdlib.h>

#include "defines_tools.h"
#include "thread.h"
#include "netconf.h"
#include "configuration.h"
#include "bbox.h"
#include "bootstrapping.h"

using namespace std;

namespace ebl {

// switch between forward only buffers or also backward
#define SFUNC fs
#define SBUF fstate_idx
#define SFUNC2(T) T,T,T,SBUF<T>
#define SFUNC3(T) T,SBUF<T>,mstate<SBUF<T> >
// backward
// #define SFUNC bbs
// #define SBUF bbstate_idx

  // A detection thread class

  template <typename Tnet>
  class detection_thread : public thread {
  public:
    detection_thread(configuration &conf, mutex *om = NULL,
                     const char *name = "",
                     const char *arg2 = NULL, bool sync = true,
                     t_chans tc = CHANS_RGB);
    ~detection_thread();

    virtual void execute();

    // thread communication ////////////////////////////////////////////////////

    virtual bool set_data(idx<ubyte> &frame, string &frame_fullname,
                          string &frame_name, uint frame_id);
    virtual bool set_data(string &frame_fullname, string &frame_name,
                          uint frame_id);
    virtual bool get_data(bboxes &bb, idx<ubyte> &frame,
                          uint &total_saved, string &frame_name,
                          uint &frame_id, svector<midx<Tnet> > &samples,
                          bboxes &bbsamples, bool &skipped);
    virtual bool available();
    virtual bool fed();
    static string get_output_directory(configuration &conf);

    static void init_detector(detector<SFUNC(Tnet)> &detector,
                              configuration &conf, string &odir);

    // internal methods ////////////////////////////////////////////////////////
  private:
    void clear_bboxes();
    void copy_bboxes(bboxes &bb);
    void copy_bootstrapping(svector<midx<Tnet> > &samples, bboxes &bb);
    void set_out_updated();
    void skip_frame();

    // private members /////////////////////////////////////////////////////////
  private:
    configuration        conf;
    const char          *arg2;
    idx<ubyte>           uframe;
    idx<Tnet>            frame;
    mutex                mutex_in;      // mutex for thread input
    mutex                mutex_out;     // mutex for thread output
    bboxes               bbs;
    bboxes::iterator     ibox;
    bool                 in_updated;    // thread input updated
    bool                 out_updated;   // thread output updated
    bool                 bavailable;    // thread is available
    bool                 bfed;  
    string               frame_name;    // name of current frame
    string               frame_fullname;
    uint                 frame_id;      
    string               outdir;        // output directory
    uint                 total_saved;
    using                thread::mout;  
    using                thread::merr;  
    t_chans              color_space;
    bool                 silent;
    svector<midx<Tnet> > returned_samples; 
    bboxes               returned_samples_bboxes;
    bootstrapping<Tnet>  boot; 
    bool                 frame_skipped; 
    bool                 frame_loaded; 

  public:
    detector<SFUNC(Tnet)>       *pdetect;
  };

} // end namespace ebl

#include "detection_thread.hpp"

#endif /* DETECTION_THREAD_H_ */