OpenNi2Device.h

Go to the documentation of this file.

#ifndef  __JBXL_OPENNI2_DEV_H_
#define  __JBXL_OPENNI2_DEV_H_
 
#ifdef  ENABLE_OPENNI2
 
#ifdef WIN32
//#pragma warning(disable:4996)
#pragma warning(disable:4005)       // INT8_MIN etc マクロの再定義
#endif
 
#include  <OpenNI.h>
#include  <NiTE.h>
 
#include  "NiDevice.h"
#include  "common++.h"
#include  "tools++.h"
#include  "buffer.h"
 
//
namespace jbxl {
 
 
#define   OPENNI2_DEVICE_PS1080     "PS1080"
#define   OPENNI2_DEVICE_KINECT     "Kinect"
#define   OPENNI2_DEVICE_ONIFILE    "oni File"
 
//
#define   OPENNI2_STREAM_DATA_NUM   3
 
#define   OPENNI2_DEPTH_FLAG        0x01
#define   OPENNI2_IMAGE_FLAG        0x02
//#define   OPENNI2_AUDIO_FLAG      0x04
 
#define   OPENNI2_DEPTH_INDEX       0
#define   OPENNI2_IMAGE_INDEX       1
//#define   OPENNI2_AUDIO_INDEX     2
 
 
class  COpenNi2Device
{
public:
    COpenNi2Device(void);
    virtual ~COpenNi2Device(void) { free();}
 
    ONI_EnuDev  m_enu_dev;
 
    int      m_state;
    BOOL     m_has_camera;
    Buffer   m_err_mesg;
    Buffer   m_dev_name;
 
    int      m_xsize;
    int      m_ysize;
    int      m_nfps;
 
public:
    BOOL    init(BOOL use_image);
    BOOL    init(char* fname, BOOL use_image);
    void    free(void);
 
    // Generators
    BOOL    create_Context(char* fname=NULL);
    BOOL    create_Recorder(char* fname, BOOL use_image);
 
    BOOL    create_Image(void);
    BOOL    create_Depth(void);
    BOOL    create_User(void);
 
    void    delete_Context(void);
    void    delete_Recorder(void);
 
    void    delete_Image(void);
    void    delete_Depth(void);
    void    delete_User(void);
 
public:
    void    wait_StreamData(void);
 
    BOOL    set_ImageData(void); 
    BOOL    set_DepthData(void);
    BOOL    set_SceneData(void);
 
    uByte*  get_ImageData(void) { if (imageFrame.isValid()) return (uByte*)imageFrame.getData(); else return NULL;}
    uByte*  get_DepthData(void) { if (depthFrame.isValid()) return (uByte*)depthFrame.getData(); else return NULL;}
    uByte*  get_SceneData(void) { if (userFrame.isValid())  return (uByte*)userFrame.getUserMap().getPixels(); else return NULL;}
 
public:
    openni::Device*         context;
 
    openni::VideoStream*    image;
    openni::VideoStream*    depth;
//  openni::VideoStream*    audio;
 
    openni::VideoFrameRef   imageFrame;
    openni::VideoFrameRef   depthFrame;
//  openni::VideoFrameRef   audioFrame;
 
    openni::VideoMode       imageMode;
    openni::VideoMode       depthMode;
 
    openni::Recorder*       recorder;
 
    // NiTE
    nite::UserTracker*      user;
    nite::UserTrackerFrameRef userFrame;
 
private:
    unsigned char   m_stream_flg;
    int             m_stream_num;
    BOOL            m_has_image_data;
    BOOL            m_has_depth_data;
//  BOOL            m_has_audio_data;
 
    openni::VideoStream*    m_streams[OPENNI2_STREAM_DATA_NUM];
 
public:
    //
    BOOL     setup_Tracking(int profile, double smooth);
 
    // File
    BOOL     start_Recorde(char* file_name, BOOL use_image);
    void     stop_Recorde(void);
 
public:
    // USB
    BOOL    open_USB_Device (void)  { return FALSE;}
    void    close_USB_Device(void)  {}
    void    set_LED_Color (int col) {}
    void    set_Tilt_Motor(int ang) {}
};
 
 
 
//
 
Buffer  initialize_OpenNI2(void);
void    shutdown_OpenNI2(void);
 
Buffer  initialize_NiTE2(void);
void    shutdown_NiTE2(void);
 
 
}       // namespace
 
 
#endif  // ENABLE_OPENNI2
 
#endif