-
Notifications
You must be signed in to change notification settings - Fork 56
/
setup.py
118 lines (100 loc) · 3.94 KB
/
setup.py
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
# -*- coding: utf-8 -*-
import os
import sys
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
from os.path import join as path_join
print "Remember: Edit at first ffmpegpath within setup.py"
##################################################################################
try:
import numpy.distutils.misc_util as nd
with_numpy=True
except:
with_numpy=False
sys.stderr.write("NumPy does not seems to be installed on your system.\n")
sys.stderr.write("You may still use PyFFmpeg, but audio support and NumPy bridge are disabled.\n")
##################################################################################
## Try to locate source if necessary
if sys.platform in [ 'win32', 'win64' ] :
ffmpegpath = r'c:\ffmpeg'
for x in [ r'..\ffmpeg', r'c:\ffmpeg-static', r'c:\ffmpeg' ]:
try:
os.stat(x)
ffmpegpath = x
except:
pass
extra_compiler_args=["-static-libgcc"]
else:
ffmpegpath = '/opt/codecs'
for x in [ os.environ["HOME"]+'build/ffmpeg', '/opt/local', '/usr/local/ffmpeg', '/opt/codecs', '/opt/ffmpeg']:
try:
os.stat(x)
ffmpegpath = x
except:
pass
extra_compiler_args=[]
##################################################################################
# Try to resove
# static dependencies resolution by looking into pkgconfig files
def static_resolver(libs):
deps = []
for lib in libs:
try:
pc = open(path_join(ffmpegpath, 'lib', 'pkgconfig', 'lib' + lib + '.pc'))
except IOError:
continue
# we only need line starting with 'Libs:'
l = filter(lambda x: x.startswith('Libs:'), pc).pop().strip()
# we only need after '-lmylib' and one entry for library
d = l.split(lib, 1).pop().split()
# remove '-l'
d = map(lambda x: x[2:], d)
# empty list means no deps
if len(d): deps += d
# Unique list
result = list(libs)
map(lambda x: x not in result and result.append(x), deps)
return result
libinc = [ path_join(ffmpegpath, 'lib') ]
libs = [ 'avformat', 'avcodec', 'avutil', 'swscale' ]
if sys.platform in [ 'win32', 'win64' ] :
libs = static_resolver(libs)
libinc += [ r'/mingw/lib' ] # it seems some people require this
incdir = path_join(ffmpegpath, 'include')
else:
incdir = [ path_join(ffmpegpath, 'include'), "/usr/include" , "./include" ]
if (with_numpy):
incdir = incdir + list(nd.get_numpy_include_dirs())
##################################################################################
if with_numpy:
ext_modules=[ Extension('pyffmpeg', [ 'pyffmpeg.pyx' ],
include_dirs = incdir,
library_dirs = libinc,
libraries = libs,
extra_compile_args=extra_compiler_args),
Extension('audioqueue', [ 'audioqueue.pyx' ],
include_dirs = incdir,
library_dirs = libinc,
libraries = libs,
extra_compile_args=extra_compiler_args),
Extension('pyffmpeg_numpybindings', [ 'pyffmpeg_numpybindings.pyx' ],
include_dirs = incdir,
library_dirs = libinc,
libraries = libs,
extra_compile_args=extra_compiler_args)
]
else:
ext_modules=[ Extension('pyffmpeg', [ 'pyffmpeg.pyx' ],
include_dirs = incdir,
library_dirs = libinc,
libraries = libs,
extra_compile_args=extra_compiler_args)
]
##################################################################################
setup(
name = 'pyffmpeg',
cmdclass = {'build_ext': build_ext},
version = "2.3",
ext_modules = ext_modules
)