/home/rex/ebltrunk/tools/libeblearntools/include/pascal_dataset.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 PASCAL_DATASET_H_
#define PASCAL_DATASET_H_

#include "dataset.h"
#include "xml_utils.h"
#include "pascal_xml.h"

namespace ebl {
  
  template <class Tdata> class pascal_dataset : public dataset<Tdata> {
  public:

    // constructors

    pascal_dataset(const char *name, const char *inroot = NULL,
                   bool ignore_diff = true, bool ignore_trunc = false,
                   bool ignore_occluded = false,
                   const char *annotations = NULL,
                   const char *ignore_path = NULL,
                   bool ignore_bbox = false);

    virtual ~pascal_dataset();

    // data

    virtual bool extract();
    virtual void extract_statistics();
    virtual void set_min_aspect_ratio(float minar);
    virtual void set_max_aspect_ratio(float maxar);
    virtual void set_minborders(idxdim &d);
    virtual void set_max_jitter_match(float match);

  protected:

#ifdef __BOOST__ // disable some derived methods if BOOST not available
#ifdef __XML__ // disable some derived methods if XML not available
    
    virtual bool included(const string &class_name, uint difficult,
                          uint truncated, uint occluded);
    
    // data
    
    virtual intg count_samples();
    virtual void count_sample(Node::NodeList &olist);
    
    // internal methods

    virtual bool process_xml(const string &xmlfile);
    virtual bool process_objects(const vector<object*> &objs,
                                 int height, int width,
                                 const string &image_fullname,
                                 const string &image_filename,
                                 const rect<int> *cropr = NULL);
    virtual void process_image(midx<Tdata> &img, const rect<int> &r,
                               string &obj_class, uint difficult,
                               const string &image_filename,
                               pair<int,int> *center,
                               const rect<int> *visr = NULL,
                               const rect<int> *cropr = NULL);
    virtual void load_data(const string &fname);
    
#endif /* __XML__ */
#endif /* __BOOST__ */

    virtual void compute_random_jitter();
    virtual void remove_jitter_matches(const std::vector<object*> &objs,
                                       uint iobj, float max_match);
    virtual void write_statistics(string &xml, ofstream &fp);
    
  protected:
    // "difficult" samples /////////////////////////////////////////
    std::vector<object*> objects; 
    uint iobj; 
    bool        ignore_difficult;       
    bool        ignore_bbox;            
    intg        total_difficult;        
    bool        ignore_truncated;       
    intg        total_truncated;        
    bool        ignore_occluded;        
    intg        total_occluded;         
    intg        total_ignored;          
    float       min_aspect_ratio;
    float       max_aspect_ratio;
    idxdim      minborders;
    idxdim      indims;
    uint        input_height;
    uint        input_width;
    // directories /////////////////////////////////////////////////
    string      annroot;        
    string      imgroot;        
    string      ignore_root;    
    // base class members to be used ///////////////////////////////
    using dataset<Tdata>::load_img;
    using dataset<Tdata>::usepose;
    using dataset<Tdata>::useparts;
    using dataset<Tdata>::usepartsonly;
    using dataset<Tdata>::allocated;
    using dataset<Tdata>::total_samples;
    using dataset<Tdata>::inroot;
    using dataset<Tdata>::display_extraction;
    using dataset<Tdata>::display_result;
    using dataset<Tdata>::outdims;
    using dataset<Tdata>::full;
    using dataset<Tdata>::add_data;
    using dataset<Tdata>::print_stats;
    using dataset<Tdata>::data_cnt;
    using dataset<Tdata>::extension;
    using dataset<Tdata>::scale_mode;
    using dataset<Tdata>::exclude;
    using dataset<Tdata>::include;
    using dataset<Tdata>::force_label;
    using dataset<Tdata>::mindims;
    using dataset<Tdata>::xtimer;
    using dataset<Tdata>::processed_cnt;
    using dataset<Tdata>::minvisibility;
    using dataset<Tdata>::save_mode;
    using dataset<Tdata>::outdir;
    // jitter //////////////////////////////////////////////////////
    using dataset<Tdata>::random_jitter; 
    float max_jitter_match; 
  };

} // end namespace ebl

#include "pascal_dataset.hpp"

#endif /* PASCAL_DATASET_H_ */