KinectDevice.h

/*
-----------------------------------------------------------------------------
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 __KINECTDEVICE_H__
#define __KINECTDEVICE_H__

#include "lib/common.h"
#include "core/Thread.h"
#include "objects/KinectUser.h"

class KinectDevice : public Thread, public openni::OpenNI::DeviceConnectedListener,
                                                                        public openni::OpenNI::DeviceDisconnectedListener,
                                                                        public openni::OpenNI::DeviceStateChangedListener
{
  public:
    KinectDevice(unsigned int id);
    ~KinectDevice();

    // Get context instance
    // Init and close device manager
    openni::Device& GetDevice();
    bool IsConnected();
    void GoThread();

    // Get an existing user
    KinectUser* CreateKinectUser();
    void DestroyKinectUser(KinectUser* existingKinectUser);
    
    unsigned int GetId();
    int GetExistingAvailableId();
    void AddExistingId(unsigned int id);
    void RemoveExistingId(unsigned int id);
    void GetValidUsersMap(KinectUser** tUsers);
    KinectUser* GetUser(int userId);
    
    Depth* GetDepthGenerator();
    Image* GetImageGenerator();
    User* GetUserGenerator();

    void GetGeneratorsSize(int &width, int &height);
    void SetGeneratorsSize(int width, int height);

    void SetSkeletonSmoothing(float value);
    float GetSkeletonSmoothing();
    
    void SetMirrorMode(bool state);
    bool GetMirrorMode();

    void SetDistCutoff(int dist);
    int GetDistCutoff();
    void SetAngCutoff(float ang);
    float GetAngCutoff();
    
    void GetHandsPixels(PtrObjBitmap Bcolor, PtrObjBitmap Balpha);
    void GetDepthBuffer(PtrObjBitmap Bcolor);
    void GetImageBuffer(PtrObjBitmap Bcolor);
    void GetImageGreyBuffer(PtrObjBitmap Bcolor);
    void GetValidUserPixels(PtrObjBitmap Bcolor, PtrObjBitmap Balpha);
    void GetValidUserPixelsRGB(PtrObjBitmap Bcolor, PtrObjBitmap Balpha);
    
    bool GetSklMutexLock();
    void ReleaseSklMutex();
  public:

  private:
    // List of detected users
    KinectUserList listOfUsers;
    boost::mutex uListMutex;
    boost::mutex updateMutex;
    boost::mutex updateHandsMutex;
    boost::mutex uFingersUpdate;
    boost::mutex uGeneratorsCreate;
    boost::mutex uSklMutex;
    OpenNIUserList listOfAvalaibleUsers;

    int mId;
    bool bConnected;
    bool bMirror;
    float msklSmoothing;
    int iDistCutoff;
    float fAngCutoff;

    // Device object
    openni::Device NiDevice;

    cv::Size mframeSize;
    cv::Mat mfingersBgr;

    Depth* mDepthGenerator;
    Image* mImageGenerator;
    User* mUserGenerator;
    //Gesture* mGestureGenerator;   // $AS: Add gesture and hands generators
    //Hands* mHandsGenerator;
    
    bool CreateAndStartGenerators();
    void StopAndDestroyGenerators();
    void DetectUsers();
    void DetectHands();

    virtual void onDeviceStateChanged(const openni::DeviceInfo* pInfo, openni::DeviceState state);
          virtual void onDeviceConnected(const openni::DeviceInfo* pInfo);
          virtual void onDeviceDisconnected(const openni::DeviceInfo* pInfo);

  protected:
    KinectDevice();
};

#endif