vtkAndorVideoSourceTest.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 "vtkContextScene.h"
#include "vtkContextView.h"
#include "vtkFloatArray.h"

#include "vtkImageData.h"
#include "vtkImageViewer.h"

#include "vtkInformation.h"
#include "vtkInformationVector.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkRenderer.h"
#include "vtkSmartPointer.h"
#include "vtkPlusAndorVideoSource.h"
#include "vtkPlusDataSource.h"
#include "igtlOSUtil.h" // for Sleep

#include "vtkXMLUtilities.h"
#include "vtksys/CommandLineArguments.hxx"
#include <stdlib.h>

class vtkMyCallback : public vtkCommand
{
public:
  static vtkMyCallback* New()
  {
    return new vtkMyCallback;
  }

  virtual void Execute(vtkObject* caller, unsigned long, void*)
  {
    m_Viewer->Render();

    //update the timer so it will trigger again
    m_Interactor->CreateTimer(VTKI_TIMER_UPDATE);
  }

  vtkRenderWindowInteractor* m_Interactor;
  vtkImageViewer* m_Viewer;

private:

  vtkMyCallback()
  {
    m_Interactor = NULL;
    m_Viewer = NULL;
  }
};


int main(int argc, char* argv[])
{
  int left;
  left = -1;
  int startX_01 = left < 0 ? 0 : left;
  left = 10;
  int startX_02 = left < 0 ? 0 : left;

  bool printHelp(false);
  bool renderingOff(false);

  std::string inputConfigFileName;
  std::string outputFileName("AndorCameraTest.mha");
  std::string flatCorrectionPath;
  std::string biasDarkCorrectionPath;

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

  int verboseLevel = vtkPlusLogger::LOG_LEVEL_INFO;

  args.AddArgument("--help", vtksys::CommandLineArguments::NO_ARGUMENT, &printHelp, "Print this help.");
  args.AddArgument("--config-file", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &inputConfigFileName, "Config file containing the device configuration.");
  args.AddArgument("--rendering-off", vtksys::CommandLineArguments::NO_ARGUMENT, &renderingOff, "Run test without rendering.");
  args.AddArgument("--output-seq-file", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &outputFileName, "Filename of the output video buffer sequence metafile (Default: AndorCameraTest.mha)");
  args.AddArgument("--verbose", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &verboseLevel, "Verbose level 1=error only, 2=warning, 3=info, 4=debug, 5=trace)");
  args.AddArgument("--flat-correction-image", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &flatCorrectionPath, "Image file (.png) containing the flat field correction.");
  args.AddArgument("--bias-correction-image", vtksys::CommandLineArguments::EQUAL_ARGUMENT, &biasDarkCorrectionPath, "Image file (.png) containing the combined bias+dark and dark correction.");

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


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

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


  vtkSmartPointer< vtkPlusAndorVideoSource > andorCamDevice = vtkSmartPointer< vtkPlusAndorVideoSource >::New();
  andorCamDevice->SetDeviceId("BLICamera");


  // Read config file
  LOG_DEBUG("Reading config file...");
  vtkSmartPointer<vtkXMLDataElement> configRootElement = vtkSmartPointer<vtkXMLDataElement>::New();

  if(PlusXmlUtils::ReadDeviceSetConfigurationFromFile(configRootElement, inputConfigFileName.c_str()) == PLUS_FAIL)
  {
    LOG_ERROR("Unable to read configuration from file " << inputConfigFileName.c_str());
    return EXIT_FAILURE;
  }

  andorCamDevice->ReadConfiguration(configRootElement);

  if(andorCamDevice->Connect() != PLUS_SUCCESS)
  {
    LOG_ERROR("Unable to connect to Andor camera");
    exit(EXIT_FAILURE);
  }

  LOG_INFO(andorCamDevice->GetSdkVersion());
  vtkIndent indent;
  andorCamDevice->PrintSelf(std::cout, indent);

  andorCamDevice->PrepareAcquisition();
  andorCamDevice->StartBLIFrameAcquisition(-1, -1, -1, 2.0);
  andorCamDevice->WaitForAcquisitionWithTimeout(100.0);
  andorCamDevice->StartGrayscaleFrameAcquisition(-1, -1, -1, 2.0);
  andorCamDevice->WaitForAcquisitionWithTimeout(100.0);
  andorCamDevice->StartRecording();

  if(renderingOff)
  {
    vtkPlusChannel* raw(nullptr);
    if(andorCamDevice->GetOutputChannelByName(raw, "BLIRawStream") != PLUS_SUCCESS)
    {
      LOG_ERROR("Unable to locate the channel with Id=\"BLIRawStream\". Check config file.");
      return EXIT_FAILURE;
    }

    vtkPlusChannel* rectified(nullptr);
    if(andorCamDevice->GetOutputChannelByName(rectified, "BLICorrectedStream") != PLUS_SUCCESS)
    {
      LOG_WARNING("Unable to locate the channel with Id=\"BLICorrectedStream\". Additional mode will not be used.");
    }

    LOG_DEBUG("Rendering disabled. Wait for just a few seconds to acquire data before exiting");
    andorCamDevice->StartBLIFrameAcquisition(-1, -1, -1, 2.0);
    andorCamDevice->WaitForAcquisitionWithTimeout(100.0);
    andorCamDevice->StartGrayscaleFrameAcquisition(-1, -1, -1, 2.0);
    andorCamDevice->WaitForAcquisitionWithTimeout(100.0);
    andorCamDevice->StartBLIFrameAcquisition(-1, -1, -1, 2.0);
    andorCamDevice->WaitForAcquisitionWithTimeout(100.0);
    andorCamDevice->StartGrayscaleFrameAcquisition(-1, -1, -1, 2.0);
    andorCamDevice->WaitForAcquisitionWithTimeout(100.0);

    vtkPlusDataSource* bSource(nullptr);
    raw->GetVideoSource(bSource);
    bSource->WriteToSequenceFile(outputFileName.c_str());

    if(rectified)
    {
      vtkPlusDataSource* rfSource(nullptr);
      rectified->GetVideoSource(rfSource);
      rfSource->WriteToSequenceFile((outputFileName + "_a.mha").c_str());
    }
  }
  else
  {
    // Show the live optical image in a VTK renderer window

    vtkSmartPointer<vtkImageViewer> viewer = vtkSmartPointer<vtkImageViewer>::New();
    viewer->SetInputConnection(andorCamDevice->GetOutputPort());   //set image to the render and window
    viewer->SetColorWindow(256);
    viewer->SetColorLevel(400);
    viewer->SetZSlice(0);
    viewer->SetSize(1024, 1024);

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

    viewer->Render(); //must be called after iren and viewer are linked or there will be problems

    // Establish timer event and create timer to update the live image
    vtkSmartPointer<vtkMyCallback> call = vtkSmartPointer<vtkMyCallback>::New();
    call->m_Interactor = iren;
    call->m_Viewer = viewer;
    iren->AddObserver(vtkCommand::TimerEvent, call);
    iren->CreateTimer(VTKI_TIMER_FIRST);

    //iren must be initialized so that it can handle events
    iren->Initialize();
    iren->Start();
  }

  andorCamDevice->Disconnect();

  //update and write configuration
  andorCamDevice->WriteConfiguration(configRootElement);
  std::string outConfigFilename = inputConfigFileName + "-updated.xml";
  bool success = vtkXMLUtilities::WriteElementToFile(configRootElement, outConfigFilename.c_str());
  if(!success)
  {
    LOG_ERROR("Unable to write configuration to: " << outConfigFilename + ".xml");
  }
  else
  {
    LOG_INFO("Configuration file written to: " << outConfigFilename + ".xml");
  }

  return EXIT_SUCCESS;
}