Merging r45748 through r45791 from trunk into soc-2011-tomato
[blender.git] / build_files / buildbot / slave_pack.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20
21 # Runs on buildbot slave, creating a release package using the build
22 # system and zipping it into buildbot_upload.zip. This is then uploaded
23 # to the master in the next buildbot step.
24
25 import os
26 import subprocess
27 import sys
28 import zipfile
29
30 # get builder name
31 if len(sys.argv) < 2:
32     sys.stderr.write("Not enough arguments, expecting builder name\n")
33     sys.exit(1)
34
35 builder = sys.argv[1]
36 branch = ''
37
38 if len(sys.argv) >= 3:
39     branch = sys.argv[2]
40
41 # scons does own packaging
42 if builder.find('scons') != -1:
43     os.chdir('../blender')
44     scons_options = ['BF_QUICK=slnt', 'BUILDBOT_BRANCH=' + branch, 'buildslave', 'BF_FANCY=False']
45
46     if builder.find('linux') != -1:
47         buildbot_dir = os.path.dirname(os.path.realpath(__file__))
48         config_dir = os.path.join(buildbot_dir, 'config')
49         build_dir = os.path.join('..', 'build', builder)
50         install_dir = os.path.join('..', 'install', builder)
51
52         scons_options += ['WITH_BF_NOBLENDER=True', 'WITH_BF_PLAYER=False',
53             'BF_BUILDDIR=' + build_dir,
54             'BF_INSTALLDIR=' + install_dir,
55             'WITHOUT_BF_INSTALL=True']
56
57         config = None
58
59         if builder.endswith('linux_x86_64_scons'):
60             config = 'user-config-x86_64.py'
61         elif builder.endswith('linux_i386_scons'):
62             config = 'user-config-x86_64.py'
63
64         if config is not None:
65             config_fpath = os.path.join(config_dir, config)
66             scons_options.append('BF_CONFIG=' + config_fpath)
67
68         blender = os.path.join(install_dir, 'blender')
69         blenderplayer = os.path.join(install_dir, 'blenderplayer')
70         subprocess.call(['strip', '--strip-all', blender, blenderplayer])
71
72         retcode = subprocess.call(['python', 'scons/scons.py'] + scons_options)
73         sys.exit(retcode)
74     else:
75         if builder.find('win') != -1:
76             bitness = '32'
77
78             if builder.find('win64') != -1:
79                 bitness = '64'
80
81             scons_options.append('BF_BITNESS=' + bitness)
82             scons_options.append('WITH_BF_CYCLES_CUDA_BINARIES=True')
83             scons_options.append('BF_CYCLES_CUDA_NVCC=nvcc.exe')
84
85         retcode = subprocess.call(['python', 'scons/scons.py'] + scons_options)
86         sys.exit(retcode)
87
88 # clean release directory if it already exists
89 release_dir = 'release'
90
91 if os.path.exists(release_dir):
92     for f in os.listdir(release_dir):
93         if os.path.isfile(os.path.join(release_dir, f)):
94             os.remove(os.path.join(release_dir, f))
95
96 # create release package
97 try:
98     subprocess.call(['make', 'package_archive'])
99 except Exception, ex:
100     sys.stderr.write('Make package release failed' + str(ex) + '\n')
101     sys.exit(1)
102
103 # find release directory, must exist this time
104 if not os.path.exists(release_dir):
105     sys.stderr.write("Failed to find release directory %r.\n" % release_dir)
106     sys.exit(1)
107
108 # find release package
109 file = None
110 filepath = None
111
112 for f in os.listdir(release_dir):
113     rf = os.path.join(release_dir, f)
114     if os.path.isfile(rf) and f.startswith('blender'):
115         file = f
116         filepath = rf
117
118 if not file:
119     sys.stderr.write("Failed to find release package.\n")
120     sys.exit(1)
121
122 # create zip file
123 try:
124     upload_zip = "buildbot_upload.zip"
125     if os.path.exists(upload_zip):
126         os.remove(upload_zip)
127     z = zipfile.ZipFile(upload_zip, "w", compression=zipfile.ZIP_STORED)
128     z.write(filepath, arcname=file)
129     z.close()
130 except Exception, ex:
131     sys.stderr.write('Create buildbot_upload.zip failed' + str(ex) + '\n')
132     sys.exit(1)