-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
118 lines (100 loc) · 3.43 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
from distutils.core import setup
from distutils.extension import Extension
from Cython.Distutils import build_ext
from os.path import join as path_join
from sys import platform
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 audiosupport and numpy-bride are disabled.\n")
##
## Try to locate source if necessary
##
if platform == 'win32':
ffmpegpath = r'c:\ffmpeg-static'
for x in [ r'..\ffmpeg', r'c:\ffmpeg-static', r'c:\ffmpeg' ]:
try:
os.stat(x)
ffmpegpath = x
except:
pass
else:
ffmpegpath = '/opt/ffmpeg'
for x in [ os.environ["HOME"]+"build/ffmpeg", '/usr/local/ffmpeg', '/opt/ffmpeg' ]:
try:
os.stat(x)
ffmpegpath = x
except:
pass
#
# 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
try:
#
# compatibility with old FFMPEG ubuntu packages
#
os.stat("/usr/include/ffmpeg")
os.mkdir("include")
os.stat("/usr/include/ffmpeg/avcodec.h")
os.system("ln -s /usr/include/ffmpeg ./include/libavcodec")
os.stat("/usr/include/ffmpeg/avformat.h")
os.system("ln -s /usr/include/ffmpeg ./include/libavformat")
os.stat("/usr/include/ffmpeg/avutil.h")
os.system("ln -s /usr/include/ffmpeg ./include/libavutil")
os.stat("/usr/include/ffmpeg/swscale.h")
os.system("ln -s /usr/include/ffmpeg ./include/libswscale")
except:
pass
libs = ['avformat', 'avcodec', 'avutil', 'swscale']
incdir = [ path_join(ffmpegpath, 'include'), "/usr/include/ffmpeg" , "./include" ] + nd.get_numpy_include_dirs()
libinc = [ path_join(ffmpegpath, 'lib') ]
if platform == 'win32':
libs = static_resolver(libs + ('avutil',))
libinc += [ r'/mingw/lib' ] # why my mingw is unable to find libz2?
with_numpy=True
if with_numpy:
ext_modules=[ Extension('pylibav', [ 'pylibav.pyx' ],
include_dirs = incdir,
library_dirs = libinc,
libraries = libs),
Extension('audioq', [ 'audioq.pyx' ],
include_dirs = incdir,
library_dirs = libinc,
libraries = libs),
]
else:
ext_modules=[ Extension('pylibav', [ 'pylibav.pyx' ],
include_dirs = incdir,
library_dirs = libinc,
libraries = libs)
]
setup(
name = 'pylibav',
cmdclass = {'build_ext': build_ext},
version = "1.0",
ext_modules = ext_modules
)