DeviceManager.cpp

/*
-----------------------------------------------------------------------------
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

-----------------------------------------------------------------------------
*/

#include "DeviceManager.h"
#include "objects/KinectDevice.h"
#include "ScolPlugin.h"
#include "lib/common.h"

template<> DeviceManager* Singleton<DeviceManager>::instance = 0;

// Constructor and Destructor
DeviceManager::DeviceManager() 
{
}

DeviceManager::~DeviceManager()
{
  // Close device manager
  close();
}

// Get singleton instance
DeviceManager& DeviceManager::GetSingleton()
{
  assert(instance);
  return *instance;
}

DeviceManager* DeviceManager::GetSingletonPtr()
{
  return instance;
}

void DeviceManager::close()
{
  bool closeNI = (listOfKinectDevice.empty()) ? false : true;
  for (KinectDeviceMap::iterator it = listOfKinectDevice.begin(); it != listOfKinectDevice.end(); it++)
  {
    SAFE_DELETE(it->second);
  }

  listOfKinectDevice.clear();

  if (closeNI)
  {
    nite::NiTE::shutdown();
    openni::OpenNI::shutdown();
  }
}

unsigned int DeviceManager::createId()
{
  int idFound = 0;
  while (listOfKinectDevice.find(idFound) != listOfKinectDevice.end())
  {
    idFound++;
  }

  return idFound;
}

KinectDevice* DeviceManager::CreateKinectDevice()
{
  if (listOfKinectDevice.empty())
  {
    // init openNI
    if (openni::OpenNI::initialize() != openni::STATUS_OK)
    {
      MMechostr(MSKDEBUG," > OpenNI initialization failed ! : %s\n", openni::OpenNI::getExtendedError());
    }

    // init nite
    if (nite::NiTE::initialize() != nite::STATUS_OK)
    {
      MMechostr(MSKDEBUG," > NIte initialization failed ! : \n");
    }
  }

  KinectDevice* newKinectDevice = new KinectDevice(createId());

  addKinectDevice(newKinectDevice);
  return newKinectDevice;
}

void DeviceManager::DestroyKinectDevice(KinectDevice* existingKinectDevice)
{
  KinectDeviceMap::iterator iKinectDeviceSearched = listOfKinectDevice.find(existingKinectDevice->GetId());
  if (iKinectDeviceSearched != listOfKinectDevice.end())
  {
    listOfKinectDevice.erase(iKinectDeviceSearched);
    SAFE_DELETE(existingKinectDevice);
  }
  else
    MMechostr(MSKRUNTIME, "DeviceManagerManager::DestroyKinectDevice : Can't remove Kinect Device");

  if (listOfKinectDevice.empty())
  {
    nite::NiTE::shutdown();
    openni::OpenNI::shutdown();
  }
}

void DeviceManager::addKinectDevice(KinectDevice* existingKinectDevice)
{
  KinectDeviceMap::iterator iKinectDeviceSearched = listOfKinectDevice.find(existingKinectDevice->GetId());
  if (iKinectDeviceSearched == listOfKinectDevice.end())
  {
    listOfKinectDevice.insert(std::make_pair(existingKinectDevice->GetId(), existingKinectDevice));
  }
  else
    MMechostr(MSKRUNTIME, "DeviceManager::addKinectDevice : Kinect Device already Exists");
}

KinectDevice* DeviceManager::GetKinectDevice(unsigned int id)
{
  KinectDeviceMap::iterator iKinectDeviceSearched = listOfKinectDevice.find(id);
  if(iKinectDeviceSearched != listOfKinectDevice.end())
  {
    return iKinectDeviceSearched->second;
  }
  return 0;
}