ITKVideo/itkVideoViewerExample.cxx at master · johan-andruejol/ITKVideo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#include "iostream"

#include "itkVideoViewer.h"
#include "itkVideoFileReader.h"

int main (int argv, char **argc)
{
  typedef itk::Image<unsigned char, 2>   ImageType;
  itk::VideoViewer<ImageType>::Pointer viewer = itk::VideoViewer<ImageType>::New();
  itk::VideoViewer<ImageType>::Pointer viewer2 = itk::VideoViewer<ImageType>::New();
  itk::VideoFileReader< ImageType >::Pointer reader = itk::VideoFileReader< ImageType >::New();
  reader->SetFileName("C:/projects/ITK-Vid-A2D2/Data/inde-circulation.avi");
  viewer->SetInput(reader->GetOutput());
  viewer2->SetInput(reader->GetOutput());
  viewer2->SetWindowName("Viewer2");
  viewer2->UseOpenCV(false);

  reader->LoadVideo();
  int FrameTotal = reader->GetFrameTotal();
  for (int i =0; i<FrameTotal; i++)
    {
    try
      {
      viewer->Update();
      viewer2->Update();
      }
    catch (itk::ExceptionObject &e)
      {
      reader->Print(std::cout);
      std::cerr<<e.GetFile()<<std::endl;
      std::cerr<<e.GetLine()<<std::endl;
      std::cerr<<e.GetLocation()<<std::endl;
      std::cerr<<e.GetNameOfClass()<<std::endl;
      std::cerr<<e.GetDescription()<<std::endl;
      std::cin>>i;
      return EXIT_FAILURE;
      }
    if ( i == static_cast<int>(100) )
      {
      viewer->Close();
      }
    if ( i == 50 )
      {
      viewer2->Close();
      }
    reader->KeepReading();
    }
  std::cin>>FrameTotal;
  return EXIT_SUCCESS;
}