colorthreshold.cpp

来自「barcode readers [ from Image]」· C++ 代码 · 共 107 行

//
// ColorThreshold
//   The simplest color threshold I could think of.
//
// Copyright (C) 2006 by Jon A. Webb (Contact via GMail; username is jonawebb)
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// 
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
// 
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//

#include "ColorThreshold.h"

#include "Debug.h"
#include "GrayImageIter.h"
#include "Image.h"
#include "ImageSize.h"
#include "Replicate.h"
#include "RGBImageIter.h"

#include <e32std.h>
#include <fbs.h>

using namespace Core;

namespace Algorithm
{

	EXPORT_C CColorThreshold* CColorThreshold::NewL(int r, int g, int b, bool passHigh)
	{	
		return new (ELeave) CColorThreshold(r, g, b, passHigh);	
	}

	CColorThreshold::~CColorThreshold()
	{
	}

	CColorThreshold::CColorThreshold(int r, int g, int b, bool passHigh) :
		ibEmpty(true),
		ibPassHigh(passHigh),
		iThreshR(r),
		iThreshG(g),
		iThreshB(b)
	{
	}

	CImage CColorThreshold::FrontL() 
	{
		if (ibEmpty) {
			User::Leave(KErrGeneral);
			return CImage();
		}
		ibEmpty = true;
		return iImage;
	}


	void CColorThreshold::PushL(CImage image)
	{
		IRgbImageIter rRgb(image);
		if (ibPassHigh) {
			for (rRgb.Reset(); !rRgb.End(); rRgb.NextRow()) {
				for (; !rRgb.REnd(); rRgb++) {
					if (rRgb()[R] < iThreshR ||
						rRgb()[G] < iThreshG ||
						rRgb()[B] < iThreshB) {
							rRgb()[R] = 0;
							rRgb()[G] = 0;
							rRgb()[B] = 0;
						}
				}
			}
		} else {
			for (rRgb.Reset(); !rRgb.End(); rRgb.NextRow()) {
				for (; !rRgb.REnd(); rRgb++) {
					if (rRgb()[R] > iThreshR ||
						rRgb()[G] > iThreshG ||
						rRgb()[B] > iThreshB) {
							rRgb()[R] = 255;
							rRgb()[G] = 255;
							rRgb()[B] = 255;
						}
				}
			}
		}
		iImage = image;
		CDebug::ShowImage(iImage);
		ibEmpty = false;
	}

	bool CColorThreshold::Empty() const
	{
		return ibEmpty;
	}

};