00001
00002 /***************************************************************************
00003  *  segment.cpp - Implementation of segmentation filter
00004  *                This filter can be used to draw the segmentation for a
00005  *                given object type to the Y-plane of the image
00006  *
00007  *  Created: Mon Jun 27 11:37:57 2005
00008  *  Copyright  2005-2007  Tim Niemueller [www.niemueller.de]
00009  *
00010  ****************************************************************************/
00011
00012 /*  This program is free software; you can redistribute it and/or modify
00013  *  it under the terms of the GNU General Public License as published by
00014  *  the Free Software Foundation; either version 2 of the License, or
00015  *  (at your option) any later version. A runtime exception applies to
00016  *  this software (see LICENSE.GPL_WRE file mentioned below for details).
00017  *
00018  *  This program is distributed in the hope that it will be useful,
00019  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
00020  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
00021  *  GNU Library General Public License for more details.
00022  *
00023  *  Read the full text in the LICENSE.GPL_WRE file in the doc directory.
00024  */
00025
00026 #include <models/color/colormodel.h>
00027 #include <filters/segment.h>
00028
00029 #include <fvutils/color/yuv.h>
00030 #include <cstddef>
00031 
00032 /** @class FilterSegment <filters/segment.h>
00033  * Segmentation filter.
00034  * Visually marks pixels of a given color and makes the segmentation visible.
00035  * The pixels are marked with bright colors.
00036  * @author Tim Niemueller
00037  */
00038 
00039 /** Constructor.
00040  * @param cm color model to use
00041  * @param what what to mark
00042  */
00043 FilterSegment::FilterSegment(ColorModel *cm, color_t what)
00044   : Filter("FilterSegment")
00045 {
00046   this->cm = cm;
00047   this->what = what;
00048 }
00049
00050
00051 void
00052 FilterSegment::apply()
00053 {
00054   register unsigned int h = 0;
00055   register unsigned int w = 0;
00056
00057   // y-plane
00058   register unsigned char *yp   = src[0] + (src_roi[0]->start.y * src_roi[0]->line_step) + (src_roi[0]->start.x * src_roi[0]->pixel_step);
00059   // u-plane
00060   register unsigned char *up   = YUV422_PLANAR_U_PLANE(src[0], src_roi[0]->image_width, src_roi[0]->image_height)
00061                                    + ((src_roi[0]->start.y * src_roi[0]->line_step) / 2 + (src_roi[0]->start.x * src_roi[0]->pixel_step) / 2) ;
00062   // v-plane
00063   register unsigned char *vp   = YUV422_PLANAR_V_PLANE(src[0], src_roi[0]->image_width, src_roi[0]->image_height)
00064                                    + ((src_roi[0]->start.y * src_roi[0]->line_step) / 2 + (src_roi[0]->start.x * src_roi[0]->pixel_step) / 2);
00065
00066   // destination y-plane
00067   register unsigned char *dyp  = dst + (dst_roi->start.y * dst_roi->line_step) + (dst_roi->start.x * dst_roi->pixel_step);
00068
00069   // line starts
00070   unsigned char *lyp  = yp;   // y-plane
00071   unsigned char *lup  = up;   // u-plane
00072   unsigned char *lvp  = vp;   // v-plane
00073   unsigned char *ldyp = dyp;  // destination y-plane
00074
00075   for (h = 0; (h < src_roi[0]->height) && (h < dst_roi->height); ++h) {
00076     for (w = 0; (w < src_roi[0]->width) && (w < dst_roi->width); w += 2) {
00077       if ( (cm->determine(*yp++, *up, *vp) == what) ) {
00078         *dyp++ = 255;
00079       } else {
00080         *dyp++ =   0;
00081       }
00082       if ( (cm->determine(*yp++, *up++, *vp++) == what) ) {
00083         *dyp++ = 255;
00084       } else {
00085         *dyp++ =   0;
00086       }
00087     }
00088     lyp  += src_roi[0]->line_step;
00089     lup  += src_roi[0]->line_step / 2;
00090     lvp  += src_roi[0]->line_step / 2;
00091     ldyp += dst_roi->line_step;
00092     yp    = lyp;
00093     up    = lup;
00094     vp    = lvp;
00095     dyp   = ldyp;
00096   }
00097
00098 }