vtkICCapturingSourceTest1.cxx

Go to the documentation of this file.

/*=Plus=header=begin======================================================
Program: Plus
Copyright (c) Laboratory for Percutaneous Surgery. All rights reserved.
See License.txt for details.
=========================================================Plus=header=end*/

#include "PlusConfigure.h"
#include "vtkCallbackCommand.h"
#include "vtkCommand.h"
#include "vtkPlusICCapturingSource.h"
#include "vtkImageData.h"
#include "vtkImageViewer2.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkRenderer.h"
#include "vtkSmartPointer.h"
#include "vtksys/CommandLineArguments.hxx"

class vtkMyCallback : public vtkCommand
{
public:
  static vtkMyCallback *New() {return new vtkMyCallback;}
  virtual void Execute(vtkObject *caller, unsigned long, void*)
  {
    viewer->Render();

    //update the timer so it will trigger again
    iren->CreateTimer(VTKI_TIMER_UPDATE);
  }
  vtkImageViewer2* viewer;
  vtkRenderWindowInteractor *iren;
};

int main(int argc, char **argv)
{
  bool printHelp(false); 
  bool renderingOff(false);
  bool listDevices=false;
  std::string deviceName("DFG/USB2-lt");
  std::string videoNorm("NTSC_M");
  std::string videoFormat("Y800 (640x480)");
  std::string inputChannel("01 Video: SVideo");

  vtksys::CommandLineArguments args;
  args.Initialize(argc, argv);

  int verboseLevel = vtkPlusLogger::LOG_LEVEL_UNDEFINED;

  args.AddArgument("--help", vtksys::CommandLineArguments::NO_ARGUMENT, &printHelp, "Print this help.");  
  args.AddArgument("--list-devices", vtksys::CommandLineArguments::NO_ARGUMENT, &listDevices, "Show the list of available devices, norms, formats, and frame sizes and exit");
  args.AddArgument("--rendering-off", vtksys::CommandLineArguments::NO_ARGUMENT, &renderingOff, "Run test without rendering.");  
  args.AddArgument("--device-name", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &deviceName, "IC Capturing device name (Default: DFG/USB2-lt)." );
  args.AddArgument("--video-norm", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &videoNorm, "IC Capturing device video norm (Default: NTSC_M)." );
  args.AddArgument("--video-format", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &videoFormat, "IC Capturing device video format (Default: Y800 (640x480))." );
  args.AddArgument("--input-channel", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &inputChannel, "IC Capturing device input channel (Default: 01 Video: SVideo)." );
  args.AddArgument("--verbose", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &verboseLevel, "Verbose level (1=error only, 2=warning, 3=info, 4=debug, 5=trace)");  

  if ( !args.Parse() )
  {
    std::cerr << "Problem parsing arguments" << std::endl;
    std::cout << "\n\nvtkPlusICCapturingSourceTest1 help:" << args.GetHelp() << std::endl;
    exit(EXIT_FAILURE);
  }

  vtkPlusLogger::Instance()->SetLogLevel(verboseLevel);

  if ( printHelp ) 
  {
    std::cout << "\n\nvtkPlusICCapturingSourceTest1 help:" << args.GetHelp() << std::endl;
    exit(EXIT_SUCCESS); 
  }

  vtkSmartPointer<vtkPlusICCapturingSource> frameGrabber = vtkSmartPointer<vtkPlusICCapturingSource>::New();

  if (listDevices)
  {
    frameGrabber->LogListOfCaptureDevices();
    exit(EXIT_SUCCESS);
  }

  frameGrabber->SetDeviceName(deviceName.c_str()); 
  frameGrabber->SetVideoNorm(videoNorm.c_str()); 
  frameGrabber->SetVideoFormat(videoFormat.c_str()); 
  frameGrabber->SetInputChannel(inputChannel.c_str()); 

  frameGrabber->CreateDefaultOutputChannel();

  LOG_INFO("Initialize..."); 
  frameGrabber->Connect(); 

  if ( frameGrabber->GetConnected() )
  {
    LOG_INFO("Start recording..."); 
    frameGrabber->StartRecording(); 
  }
  else
  {
    frameGrabber->Disconnect();
    LOG_ERROR( "Unable to connect to IC capture device: " << deviceName ); 
    exit(EXIT_FAILURE); 
  }

  if (renderingOff)
  {
    LOG_INFO("No need for rendering, stop the device..."); 
    frameGrabber->StopRecording(); 
    frameGrabber->Disconnect();
    LOG_INFO("Exit successfully"); 
    exit(EXIT_SUCCESS); 
  }

  vtkSmartPointer<vtkImageViewer2> viewer = vtkSmartPointer<vtkImageViewer2>::New();
  viewer->SetColorWindow(255);
  viewer->SetColorLevel(100.5);
  viewer->SetSize(640,480); 

  viewer->SetInputConnection(frameGrabber->GetOutputPort());

  //Create the interactor that handles the event loop
  vtkSmartPointer<vtkRenderWindowInteractor> iren = vtkSmartPointer<vtkRenderWindowInteractor>::New();
  iren->SetRenderWindow(viewer->GetRenderWindow());
  viewer->SetupInteractor(iren);

  viewer->Render(); 

  //establish timer event and create timer
  vtkSmartPointer<vtkMyCallback> call = vtkSmartPointer<vtkMyCallback>::New();
  call->viewer=viewer;
  call->iren=iren;
  iren->AddObserver(vtkCommand::TimerEvent, call);
  iren->CreateTimer(VTKI_TIMER_FIRST);    //VTKI_TIMER_FIRST = 0

  // iren must be initialized so that it can handle events
  iren->Initialize();
  // start the event loop (exit with 'q' key)
  iren->Start();

  frameGrabber->StopRecording(); 
  frameGrabber->Disconnect();

  LOG_INFO("Exit successfully"); 
  return EXIT_SUCCESS; 
}