pytorch

1
# - Try to find ffmpeg libraries
2
#     (libavcodec, libavformat, libavutil, libswscale)
3
# Once done this will define
4
#
5
# FFMPEG_FOUND - system has ffmpeg or libav
6
# FFMPEG_INCLUDE_DIR - the ffmpeg include directory
7
# FFMPEG_LIBRARIES - Link these to use ffmpeg
8
#
9

10
if (FFMPEG_LIBRARIES AND FFMPEG_INCLUDE_DIR)
11
  # in cache already
12
  set(FFMPEG_FOUND TRUE)
13
else (FFMPEG_LIBRARIES AND FFMPEG_INCLUDE_DIR)
14

15
  find_path(FFMPEG_AVCODEC_INCLUDE_DIR
16
    NAMES libavcodec/avcodec.h
17
    PATHS ${_FFMPEG_AVCODEC_INCLUDE_DIRS} /usr/include /usr/local/include /opt/local/include /sw/include
18
    PATH_SUFFIXES ffmpeg libav
19
  )
20

21
  find_library(FFMPEG_LIBAVCODEC
22
    NAMES avcodec
23
    PATHS ${_FFMPEG_AVCODEC_LIBRARY_DIRS} /usr/lib /usr/local/lib /opt/local/lib /sw/lib
24
  )
25

26
  find_library(FFMPEG_LIBAVFORMAT
27
    NAMES avformat
28
    PATHS ${_FFMPEG_AVFORMAT_LIBRARY_DIRS} /usr/lib /usr/local/lib /opt/local/lib /sw/lib
29
  )
30

31
  find_library(FFMPEG_LIBAVUTIL
32
    NAMES avutil
33
    PATHS ${_FFMPEG_AVUTIL_LIBRARY_DIRS} /usr/lib /usr/local/lib /opt/local/lib /sw/lib
34
  )
35

36

37
  find_library(FFMPEG_LIBSWSCALE
38
    NAMES swscale
39
    PATHS ${_FFMPEG_SWSCALE_LIBRARY_DIRS} /usr/lib /usr/local/lib /opt/local/lib /sw/lib
40
  )
41

42
  find_library(FFMPEG_LIBSWRESAMPLE
43
    NAMES swresample
44
    PATHS ${_FFMPEG_SWSCALE_LIBRARY_DIRS} /usr/lib /usr/local/lib /opt/local/lib /sw/lib
45
  )
46

47
  if (FFMPEG_LIBAVCODEC AND FFMPEG_LIBAVFORMAT)
48
    set(FFMPEG_FOUND TRUE)
49
  endif()
50

51
  if (FFMPEG_FOUND)
52
    set(FFMPEG_INCLUDE_DIR ${FFMPEG_AVCODEC_INCLUDE_DIR})
53

54
    set(FFMPEG_LIBRARIES
55
      ${FFMPEG_LIBAVCODEC}
56
      ${FFMPEG_LIBAVFORMAT}
57
      ${FFMPEG_LIBAVUTIL}
58
      ${FFMPEG_LIBSWSCALE}
59
      ${FFMPEG_LIBSWRESAMPLE}
60
    )
61

62
    if (NOT FFMPEG_FIND_QUIETLY)
63
      message(STATUS "Found FFMPEG or Libav: ${FFMPEG_LIBRARIES}, ${FFMPEG_INCLUDE_DIR}")
64
    endif (NOT FFMPEG_FIND_QUIETLY)
65
  else (FFMPEG_FOUND)
66
    if (FFMPEG_FIND_REQUIRED)
67
      message(FATAL_ERROR "Could not find libavcodec or libavformat or libavutil")
68
    endif (FFMPEG_FIND_REQUIRED)
69
  endif (FFMPEG_FOUND)
70

71
endif (FFMPEG_LIBRARIES AND FFMPEG_INCLUDE_DIR)
72