#include <iclDefaultGrabEngine.h>
#include <iclUnicapDevice.h>

namespace icl{
  DefaultGrabEngine::DefaultGrabEngine(UnicapDevice *device, bool useDMA, bool progressiveGrabMode):
    m_poDevice(device), m_iCurrBuf(0), m_bUseDMA(useDMA), m_bStarted(false), m_bProgressiveGrabMode(progressiveGrabMode){
    
    UnicapFormat fmt = m_poDevice->getCurrentUnicapFormat();
    for(int i=0;i<NBUFS;i++){
      m_oBuf[i].buffer_size = useDMA ? 0 : fmt.getBufferSize();
      m_oBuf[i].data = useDMA ? 0 :new unsigned char[fmt.getBufferSize()];
      m_oBuf[i].type = useDMA ? UNICAP_BUFFER_TYPE_SYSTEM : UNICAP_BUFFER_TYPE_USER;
      fmt.getUnicapFormat()->buffer_type = useDMA ? UNICAP_BUFFER_TYPE_SYSTEM : UNICAP_BUFFER_TYPE_USER;
    }
    device->setFormat(fmt);
  }
  DefaultGrabEngine::~DefaultGrabEngine(){
    unicap_stop_capture (m_poDevice->getUnicapHandle()); 
    for(int i=0;i<NBUFS;i++){
      if(m_oBuf[i].type == UNICAP_BUFFER_TYPE_USER && m_oBuf[i].data){
        delete [] m_oBuf[i].data;
      }
    }    
  }
  
  void DefaultGrabEngine::lockGrabber(){ 
    // no longer needed because of using double buffering
  }
  void DefaultGrabEngine::unlockGrabber(){ 
    // no longer needed because of using double buffering
  }
  
  const icl8u *DefaultGrabEngine::getCurrentFrameUnconverted(){
    
    if(m_bProgressiveGrabMode){
      if(!m_bStarted){
        printf("----1 \n");
        unicap_start_capture(m_poDevice->getUnicapHandle());
        printf("----2 \n");
        m_bStarted = true;
        printf("----3 \n");
        unicap_queue_buffer(m_poDevice->getUnicapHandle(),&m_oBuf[NEXT_IDX()]);
        printf("----4 \n");
      }
      
      printf("----5 \n");
      unicap_data_buffer_t *returned_buffer;
      static const int MAX_TRYS = 10;
      static const long WAIT_TIME = 100000;
      printf("----6 \n");
      int i=0;

      int success = 0;
      while(!success){
        if( !SUCCESS (unicap_poll_buffer(m_poDevice->getUnicapHandle(),&success))){
          ERROR_LOG("Failed to wait for the buffer to be filled! ( POLL Section)<<");
        }
        usleep(1000);
      }
      
      for(;i<MAX_TRYS;i++){
        printf("----7%d \n",i);
        usleep(1000);
        if( SUCCESS (unicap_wait_buffer (m_poDevice->getUnicapHandle(), &returned_buffer))){
          break;
        }else{
          usleep(WAIT_TIME);
        }
        printf("----8%d \n",i);
      }if(i==MAX_TRYS){
        ERROR_LOG("Failed to wait for the buffer to be filled! ( tried "<<MAX_TRYS<<" times)");        
      }
      printf("----9 \n");
      unicap_queue_buffer(m_poDevice->getUnicapHandle(),&m_oBuf[NEXT_IDX()]);
      printf("returning returned buffer is %p data is %p (10)\n",returned_buffer,returned_buffer->data);
      return returned_buffer->data;
    }else{
      if(!m_bStarted){
        unicap_start_capture(m_poDevice->getUnicapHandle());
        m_bStarted = true;
      }
      
      unicap_data_buffer_t *returned_buffer;
      static const int MAX_TRYS = 10;
      static const long WAIT_TIME = 100000;
      int i=0;
      for(;i<MAX_TRYS;i++){
        unicap_queue_buffer(m_poDevice->getUnicapHandle(),&m_oBuf[NEXT_IDX()]);
        usleep(1000);
        if( SUCCESS (unicap_wait_buffer (m_poDevice->getUnicapHandle(), &returned_buffer))){
          break;
        }else{
          usleep(WAIT_TIME);
        }
        
      }if(i==MAX_TRYS){
        ERROR_LOG("Failed to wait for the buffer to be filled! ( tried "<<MAX_TRYS<<" times)");        
      }
      
      return returned_buffer->data;
    }
  }
}