www.pudn.com > reacTIVision-1.3.rar > DtouchFinder.cpp

/*  reacTIVision fiducial tracking framework
    DtouchFinder.cpp
    Copyright (C) 2006 Martin Kaltenbrunner 

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#include "DtouchFinder.h"

bool DtouchFinder::init(int w, int h, int sb, int db) {
	
	FiducialFinder::init(w,h,sb,db);

	if (sb!=1) {
		printf("source buffer must be grayscale\n");
		return false;
	}
		
	frecognition = new FiducialRecognition(width, height);
	return true;
}
void DtouchFinder::process(unsigned char *src, unsigned char *dest) {

	FiducialData *fdata = NULL;
	int count = frecognition->process(src,&fdata);

	// process symbols
	for(int i=0;i< count;i++) {

		//apply distortion
		if (distortion) {
			int index = fdata[i].getCentre().y*width+fdata[i].getCentre().x;
			fdata[i].setCentre(DTPoint((int)floor(dmap[index].x),(int)floor(dmap[index].y)));
		}

		FiducialObject *existing_fiducial = NULL;

		// check if we have an ID/Position conflict
		for (fiducial = fiducialList.begin(); fiducial!=fiducialList.end(); fiducial++) {

			float distance = fiducial->distance(fdata[i].getCentre().x,fdata[i].getCentre().y);

			if (fdata[i].getId()==fiducial->fiducial_id) {
				// find and match a fiducial we had last frame already ...
				if(existing_fiducial) {
					if (distancedistance(fdata[i].getCentre().x,fdata[i].getCentre().y))
						existing_fiducial = &(*fiducial);
				} else {
					existing_fiducial = &(*fiducial);
					for (int j=0;jdistance(fdata[j].getCentre().x,fdata[j].getCentre().y)fiducial_id);
				existing_fiducial = &(*fiducial);
				break;
			}
		}
		
		if  (existing_fiducial!=NULL) {
			// just update the fiducial from last frame ...
			existing_fiducial->update(fdata[i].getCentre().x,fdata[i].getCentre().y,fdata[i].getAngle());
		} else {
			// add the newly found object
			FiducialObject addFiducial(session_id, fdata[i].getId(), width, height);
			addFiducial.update(fdata[i].getCentre().x,fdata[i].getCentre().y,fdata[i].getAngle());
			fiducialList.push_back(addFiducial);
			if (mserver!=NULL) mserver->sendAddMessage(fdata[i].getId());
			if (msg_listener) {
				char add_message[16];
				sprintf(add_message,"add obj %d %ld",fdata[i].getId(),session_id);
				msg_listener->setMessage(std::string(add_message));
			}
			session_id++;
		}

		if(msg_listener)
			 msg_listener->setObject(fdata[i].getId(),(int)(fdata[i].getCentre().x),(int)(fdata[i].getCentre().y),(fdata[i].getAngle()+180/180)*M_PI);
	}

	if (tserver) sendTuioMessages();
	if (mserver) sendMidiMessages();
	drawGrid(src,dest);
	delete[] fdata;
}