ArTkDetector.h

Go to the documentation of this file.

/*
-----------------------------------------------------------------------------
This source file is part of OpenSpace3D
For the latest info, see http://www.openspace3d.com
 
Copyright (c) 2012 I-maginer
 
This program 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 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 Lesser General Public License for more details.
 
You should have received a copy of the GNU Lesser 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, or go to
http://www.gnu.org/copyleft/lesser.txt
 
-----------------------------------------------------------------------------
*/
 
#ifndef __BTK_ARTKDETECTOR_H__
#define __BTK_ARTKDETECTOR_H__
 
#include "Prerequisites.h"
#include <aruco.h>
#include <set>
#include "ArTkMarker.h"
#include "ArCameraParam.h"
#include "ArManager.h"
#include "BitmapToolkitThread.h"
 
#include <AR/ar.h>
#include <AR/arFilterTransMat.h>
#include <AR2/tracking.h>
#include <KPM/kpm.h>
 
class ArTkDetector : btopenutils::Thread
{
        public:
    ArTkDetector();
    ~ArTkDetector();
  protected:
 
        private:
    LASTDATA mLastData;
    boost::recursive_mutex mUpdateMutex;
    boost::mutex mDetectMutex;
    std::set<ArTkMarker*> mMarkerList;
    std::vector<AR2SurfaceSetT*> mSurfaceSet;
    KpmHandle* mKpmHandle;
    AR2HandleT* mAr2Handle;
    ARParamLT* mArCamParam;
    int mSkippage[1024];
    int mNumSkip;
    cv::Mat mDetectorFrame;
    cv::Mat mDetectorColorFrame;
    float mScale;
    ArCameraParam mCamparam;
    bool mReverse;
 
    bool mIsDirty;
    bool mDetectionInProgress;
 
    void BuildCameraParam(ArCameraParam &camparam);
    void InitDetector();
    void NFTDetect();
 
        public:
    bool needUpdate;
    void GoThread();
    void SetDirty();
    void AddMarker(ArMarker* marker);
    void RemoveMarker(ArMarker* marker);
  protected:
 
        private:
};
 
#endif