=== SCons ===
[blender.git] / tools / btools.py
1 import os
2 import os.path
3 import SCons.Options
4
5 import SCons.Variables
6 try:
7         import subprocess
8 except ImportError:
9         pass
10 import string
11 import glob
12 import shutil
13 import sys
14
15 Variables = SCons.Variables
16 BoolVariable = SCons.Variables.BoolVariable
17
18 def print_arguments(args, bc):
19         if len(args):
20                 for k,v in args.iteritems():
21                         print '\t'+bc.OKBLUE+k+bc.ENDC+' = '+bc.OKGREEN + v + bc.ENDC
22         else:
23                 print '\t'+bc.WARNING+'No  command-line arguments given'+bc.ENDC
24
25 def validate_arguments(args, bc):
26         opts_list = [
27                         'WITH_BF_PYTHON', 'BF_PYTHON', 'BF_PYTHON_VERSION', 'BF_PYTHON_INC', 'BF_PYTHON_BINARY', 'BF_PYTHON_LIB', 'BF_PYTHON_LIBPATH', 'BF_PYTHON_LINKFLAGS', 'WITH_BF_STATICPYTHON', 'BF_PYTHON_LIB_STATIC',
28                         'WITH_BF_OPENAL', 'BF_OPENAL', 'BF_OPENAL_INC', 'BF_OPENAL_LIB', 'BF_OPENAL_LIBPATH', 'WITH_BF_STATICOPENAL', 'BF_OPENAL_LIB_STATIC',
29                         'WITH_BF_SDL', 'BF_SDL', 'BF_SDL_INC', 'BF_SDL_LIB', 'BF_SDL_LIBPATH',
30                         'BF_PTHREADS', 'BF_PTHREADS_INC', 'BF_PTHREADS_LIB', 'BF_PTHREADS_LIBPATH',
31                         'WITH_BF_FMOD',
32                         'WITH_BF_OPENEXR', 'BF_OPENEXR', 'BF_OPENEXR_INC', 'BF_OPENEXR_LIB', 'BF_OPENEXR_LIBPATH', 'WITH_BF_STATICOPENEXR', 'BF_OPENEXR_LIB_STATIC',
33                         'WITH_BF_DDS',
34                         'WITH_BF_FFMPEG', 'BF_FFMPEG_LIB','BF_FFMPEG_EXTRA', 'BF_FFMPEG',  'BF_FFMPEG_INC',
35                         'WITH_BF_OGG', 'BF_OGG', 'BF_OGG_LIB',
36                         'WITH_BF_JPEG', 'BF_JPEG', 'BF_JPEG_INC', 'BF_JPEG_LIB', 'BF_JPEG_LIBPATH',
37                         'WITH_BF_PNG', 'BF_PNG', 'BF_PNG_INC', 'BF_PNG_LIB', 'BF_PNG_LIBPATH',
38                         'BF_TIFF', 'BF_TIFF_INC',
39                         'WITH_BF_ZLIB', 'BF_ZLIB', 'BF_ZLIB_INC', 'BF_ZLIB_LIB', 'BF_ZLIB_LIBPATH',
40                         'WITH_BF_INTERNATIONAL',
41                         'BF_GETTEXT', 'BF_GETTEXT_INC', 'BF_GETTEXT_LIB', 'BF_GETTEXT_LIBPATH',
42                         'WITH_BF_ICONV', 'BF_ICONV', 'BF_ICONV_INC', 'BF_ICONV_LIB', 'BF_ICONV_LIBPATH',
43                         'WITH_BF_ODE', 'BF_ODE', 'BF_ODE_INC', 'BF_ODE_LIB',
44                         'WITH_BF_GAMEENGINE', 'WITH_BF_BULLET', 'BF_BULLET', 'BF_BULLET_INC', 'BF_BULLET_LIB',
45                         'BF_SOLID', 'BF_SOLID_INC', 'BF_WINTAB', 'BF_WINTAB_INC',
46                         'WITH_BF_YAFRAY',
47                         'WITH_BF_FREETYPE', 'BF_FREETYPE', 'BF_FREETYPE_INC', 'BF_FREETYPE_LIB', 'BF_FREETYPE_LIBPATH',
48                         'WITH_BF_QUICKTIME', 'BF_QUICKTIME', 'BF_QUICKTIME_INC', 'BF_QUICKTIME_LIB', 'BF_QUICKTIME_LIBPATH',
49                         'WITH_BF_STATICOPENGL', 'BF_OPENGL', 'BF_OPENGL_INC', 'BF_OPENGL_LIB', 'BF_OPENGL_LIBPATH', 'BF_OPENGL_LIB_STATIC', 'BF_OPENGL_LINKFLAGS',
50                         'WITH_BF_FTGL', 'BF_FTGL', 'BF_FTGL_INC', 'BF_FTGL_LIB',
51                         'WITH_BF_PLAYER',
52                         'WITH_BF_NOBLENDER',
53                         'WITH_BF_BINRELOC',
54                         'CFLAGS', 'CCFLAGS', 'CXXFLAGS', 'CPPFLAGS',
55                         'REL_CFLAGS', 'REL_CCFLAGS', 'REL_CXXFLAGS',
56                         'BF_PROFILE_FLAGS', 'BF_PROFILE_FLAGS', 'BF_PROFILE_CXXFLAGS',
57                         'BF_DEBUG_CFLAGS', 'BF_DEBUG_CCFLAGS', 'BF_DEBUG_CXXFLAGS',
58                         'C_WARN', 'CC_WARN', 'CXX_WARN',
59                         'LLIBS', 'PLATFORM_LINKFLAGS',
60                         'LCGDIR',
61                         'BF_CXX', 'WITH_BF_STATICCXX', 'BF_CXX_LIB_STATIC'
62                         'WITH_BF_VERSE', 'BF_VERSE_INCLUDE',
63                         'VERSE_BUILD_BINARY', 'VERSE_BUILD_DIR', 'VERSE_REGEN_PROTO',
64                         'BF_TWEAK_MODE', 'BF_SPLIT_SRC',
65                         'WITHOUT_BF_INSTALL',
66                         'WITH_BF_OPENMP',
67                         'WITHOUT_BF_INSTALL',
68                         'BF_FANCY', 'BF_QUIET',
69                         'BF_X264_CONFIG',
70                         'BF_XVIDCORE_CONFIG',
71                         'WITH_BF_DOCS',
72                         'BF_NUMJOBS',
73                         ]
74
75         arg_list = ['BF_DEBUG', 'BF_QUIET', 'BF_CROSS', 'BF_UPDATE',
76                         'BF_INSTALLDIR', 'BF_TOOLSET', 'BF_BINNAME',
77                         'BF_BUILDDIR', 'BF_FANCY', 'BF_QUICK', 'BF_PROFILE',
78                         'BF_BSC', 'BF_CONFIG',
79                         'BF_PRIORITYLIST', 'BF_BUILDINFO','CC', 'CXX', 'BF_QUICKDEBUG',
80                         'BF_LISTDEBUG', 'LCGDIR', 'BF_X264_CONFIG', 'BF_XVIDCORE_CONFIG',
81                         'BF_DOCDIR']
82
83         all_list = opts_list + arg_list
84         okdict = {}
85
86         for k,v in args.iteritems():
87                 if k in all_list:
88                         okdict[k] = v
89                 else:
90                         print '\t'+bc.WARNING+'Invalid argument: '+bc.ENDC+k+'='+v
91
92         return okdict
93
94 def print_targets(targs, bc):
95         if len(targs)>0:
96                 for t in targs:
97                         print '\t'+bc.OKBLUE+t+bc.ENDC
98         else:
99                 print '\t'+bc.WARNING+'No targets given, using '+bc.ENDC+bc.OKGREEN+'default'+bc.ENDC
100
101 def validate_targets(targs, bc):
102         valid_list = ['.', 'blender', 'blenderstatic', 'blenderplayer', 'webplugin',
103                         'blendernogame', 'blenderstaticnogame', 'blenderlite', 'release',
104                         'everything', 'clean', 'install-bin', 'install', 'nsis']
105         oklist = []
106         for t in targs:
107                 if t in valid_list:
108                         oklist.append(t)
109                 else:
110                         print '\t'+bc.WARNING+'Invalid target: '+bc.ENDC+t
111         return oklist
112
113 class ourSpawn:
114         def ourspawn(self, sh, escape, cmd, args, env):
115                 newargs = string.join(args[1:], ' ')
116                 cmdline = cmd + " " + newargs
117                 startupinfo = subprocess.STARTUPINFO()
118                 startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
119                 proc = subprocess.Popen(cmdline, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
120                         stderr=subprocess.PIPE, startupinfo=startupinfo, shell = False)
121                 data, err = proc.communicate()
122                 rv = proc.wait()
123                 if rv:
124                         print "====="
125                         print err
126                         print "====="
127                 return rv
128
129 def SetupSpawn( env ):
130         buf = ourSpawn()
131         buf.ourenv = env
132         env['SPAWN'] = buf.ourspawn
133
134
135 def read_opts(cfg, args):
136         localopts = Variables.Variables(cfg, args)
137         localopts.AddVariables(
138                 ('VERSE_BUILD_BINARY', 'Build a release or debug binary.', 'release'),
139                 ('VERSE_BUILD_DIR', 'Target directory for intermediate files.', "${BF_BUILDDIR}/extern/verse"),
140                 ('VERSE_REGEN_PROTO', 'Whether to regenerate the protocol files', 'yes'),
141                 (BoolVariable('WITH_BF_VERSE', 'Use VERSE if true', False)),
142                 ('BF_VERSE_INCLUDE', 'verse include dir', '/usr/include'),
143                 ('LCGDIR', 'location of cvs lib dir'),
144                 ('VERSE_BUILD_BINARY', 'Build a release or debug binary.', 'release'),
145                 ('VERSE_BUILD_DIR', 'Target directory for intermediate files.', "${BF_BUILDDIR}/extern/verse"),
146                 ('VERSE_REGEN_PROTO', 'Whether to regenerate the protocol files', 'yes'),
147                 ('BF_DEBUG_LIBS', 'list of libraries to build with debug symbols'),
148
149                 (BoolVariable('WITH_BF_PYTHON', 'Compile with python', True)),
150                 ('BF_PYTHON', 'base path for python', ''),
151                 ('BF_PYTHON_VERSION', 'Python version to use', ''),
152                 ('BF_PYTHON_INC', 'include path for Python headers', ''),
153                 ('BF_PYTHON_BINARY', 'Path to the Python interpreter', ''),
154                 ('BF_PYTHON_LIB', 'Python library', ''),
155                 ('BF_PYTHON_LIB_STATIC', 'Python static libraries', ''),
156                 ('BF_PYTHON_LIBPATH', 'Library path', ''),
157                 ('BF_PYTHON_LINKFLAGS', 'Python link flags', ''),
158                 (BoolVariable('WITH_BF_STATICPYTHON', 'Staticly link to python', False)),
159
160                 (BoolVariable('BF_NO_ELBEEM', 'Disable Fluid Sim', False)),
161                 (BoolVariable('WITH_BF_YAFRAY', 'Enable Yafray', True)),
162
163                 (BoolVariable('WITH_BF_OPENAL', 'Use OpenAL if true', False)),
164                 ('BF_OPENAL', 'base path for OpenAL', ''),
165                 ('BF_OPENAL_INC', 'include path for python headers', ''),
166                 ('BF_OPENAL_LIB', 'Path to OpenAL library', ''),
167                 ('BF_OPENAL_LIB_STATIC', 'Path to OpenAL static library', ''),
168                 ('BF_OPENAL_LIBPATH', 'Path to OpenAL library', ''),
169                 (BoolVariable('WITH_BF_STATICOPENAL', 'Staticly link to openal', False)),
170
171                 (BoolVariable('WITH_BF_SDL', 'Use SDL if true', False)),
172                 ('BF_SDL', 'SDL base path', ''),
173                 ('BF_SDL_INC', 'SDL include path', ''),  #$(shell $(BF_SDL)/bin/sdl-config --cflags)
174                 ('BF_SDL_LIB', 'SDL library', ''),        #$(shell $(BF_SDL)/bin/sdl-config --libs) -lSDL_mixer
175                 ('BF_SDL_LIBPATH', 'SDL library path', ''),
176
177                 ('BF_PTHREADS', 'Pthreads base path', ''),
178                 ('BF_PTHREADS_INC', 'Pthreads include path', ''),
179                 ('BF_PTHREADS_LIB', 'Pthreads library', ''),
180                 ('BF_PTHREADS_LIBPATH', 'Pthreads library path', ''),
181
182                 (BoolVariable('WITH_BF_FMOD', 'Use FMOD if true', False)),
183                 #  BF_FMOD = $(LCGDIR)/fmod
184
185                 (BoolVariable('WITH_BF_OPENEXR', 'Use OPENEXR if true', True)),
186                 (BoolVariable('WITH_BF_STATICOPENEXR', 'Staticly link to OpenEXR', False)),
187                 ('BF_OPENEXR', 'OPENEXR base path', ''),
188                 ('BF_OPENEXR_INC', 'OPENEXR include path', ''),
189                 ('BF_OPENEXR_LIB', 'OPENEXR library', ''),
190                 ('BF_OPENEXR_LIBPATH', 'OPENEXR library path', ''),
191                 ('BF_OPENEXR_LIB_STATIC', 'OPENEXR static library', ''),
192
193                 (BoolVariable('WITH_BF_DDS', 'Use DDS if true', True)),
194
195                 (BoolVariable('WITH_BF_FFMPEG', 'Use FFMPEG if true', False)),
196                 ('BF_FFMPEG', 'FFMPEG base path', ''),
197                 ('BF_FFMPEG_LIB', 'FFMPEG library', ''),
198                 ('BF_FFMPEG_EXTRA', 'FFMPEG flags that must be preserved', ''),
199
200                 ('BF_FFMPEG_INC', 'FFMPEG includes', ''),
201                 ('BF_FFMPEG_LIBPATH', 'FFMPEG library path', ''),
202                 
203                 (BoolVariable('WITH_BF_OGG', 'Use OGG, THEORA, VORBIS in FFMPEG if true',
204                                         False)),
205                 ('BF_OGG', 'OGG base path', ''),
206                 ('BF_OGG_LIB', 'OGG library', ''),
207
208                 (BoolVariable('WITH_BF_JPEG', 'Use JPEG if true', True)),
209                 ('BF_JPEG', 'JPEG base path', ''),
210                 ('BF_JPEG_INC', 'JPEG include path', ''),
211                 ('BF_JPEG_LIB', 'JPEG library', ''),
212                 ('BF_JPEG_LIBPATH', 'JPEG library path', ''),
213
214                 (BoolVariable('WITH_BF_OPENJPEG', 'Use OPENJPEG if true', False)),
215                 ('BF_OPENJPEG', 'OPENJPEG base path', ''),
216                 ('BF_OPENJPEG_INC', 'OPENJPEG include path', ''),
217                 ('BF_OPENJPEG_LIB', 'OPENJPEG library', ''),
218                 ('BF_OPENJPEG_LIBPATH', 'OPENJPEG library path', ''),
219
220                 (BoolVariable('WITH_BF_REDCODE', 'Use REDCODE if true', False)),
221                 ('BF_REDCODE', 'REDCODE base path', ''),
222                 ('BF_REDCODE_INC', 'REDCODE include path', ''),
223                 ('BF_REDCODE_LIB', 'REDCODE library', ''),
224                 ('BF_REDCODE_LIBPATH', 'REDCODE library path', ''),
225
226                 (BoolVariable('WITH_BF_PNG', 'Use PNG if true', True)),
227                 ('BF_PNG', 'PNG base path', ''),
228                 ('BF_PNG_INC', 'PNG include path', ''),
229                 ('BF_PNG_LIB', 'PNG library', ''),
230                 ('BF_PNG_LIBPATH', 'PNG library path', ''),
231
232                 ('BF_TIFF', 'TIFF base path', ''),
233                 ('BF_TIFF_INC', 'TIFF include path', ''),
234
235                 (BoolVariable('WITH_BF_ZLIB', 'Use ZLib if true', True)),
236                 ('BF_ZLIB', 'ZLib base path', ''),
237                 ('BF_ZLIB_INC', 'ZLib include path', ''),
238                 ('BF_ZLIB_LIB', 'ZLib library', ''),
239                 ('BF_ZLIB_LIBPATH', 'ZLib library path', ''),
240
241                 (BoolVariable('WITH_BF_INTERNATIONAL', 'Use Gettext and Freetype if true', True)),
242
243                 ('BF_GETTEXT', 'gettext base path', ''),
244                 ('BF_GETTEXT_INC', 'gettext include path', ''),
245                 ('BF_GETTEXT_LIB', 'gettext library', ''),
246                 ('BF_GETTEXT_LIBPATH', 'gettext library path', ''),
247                 
248                 (BoolVariable('WITH_BF_ICONV', 'Use iconv if true', True)),
249                 ('BF_ICONV', 'iconv base path', ''),
250                 ('BF_ICONV_INC', 'iconv include path', ''),
251                 ('BF_ICONV_LIB', 'iconv library', ''),
252                 ('BF_ICONV_LIBPATH', 'iconv library path', ''),
253                 
254                 (BoolVariable('WITH_BF_GAMEENGINE', 'Build with gameengine' , True)),
255
256                 (BoolVariable('WITH_BF_ODE', 'Use ODE if true', True)),
257                 ('BF_ODE', 'ODE base path', ''),
258                 ('BF_ODE_INC', 'ODE include path' , ''),
259                 ('BF_ODE_LIB', 'ODE library', ''),
260
261                 (BoolVariable('WITH_BF_BULLET', 'Use Bullet if true', True)),
262                 ('BF_BULLET', 'Bullet base dir', ''),
263                 ('BF_BULLET_INC', 'Bullet include path', ''),
264                 ('BF_BULLET_LIB', 'Bullet library', ''),
265                 
266                 ('BF_SOLID', 'Solid base dir', '#/extern/solid'),
267                 ('BF_SOLID_INC', 'Solid include path', ''),
268                 ('BF_WINTAB', 'WinTab base dir', ''),
269                 ('BF_WINTAB_INC', 'WinTab include dir', ''),
270                 ('BF_CXX', 'c++ base path for libstdc++, only used when static linking', ''),
271                 (BoolVariable('WITH_BF_STATICCXX', 'static link to stdc++', False)),
272                 ('BF_CXX_LIB_STATIC', 'static library path for stdc++', ''),
273 ##
274 ##WITH_BF_NSPR = True
275 ##BF_NSPR = $(LCGDIR)/nspr
276 ##BF_NSPR_INC = -I$(BF_NSPR)/include -I$(BF_NSPR)/include/nspr
277 ##BF_NSPR_LIB = 
278 ### Uncomment the following line to use Mozilla inplace of netscape
279 ##CPPFLAGS += -DMOZ_NOT_NET
280 ### Location of MOZILLA/Netscape header files...
281 ##BF_MOZILLA = $(LCGDIR)/mozilla
282 ##BF_MOZILLA_INC = -I$(BF_MOZILLA)/include/mozilla/nspr -I$(BF_MOZILLA)/include/mozilla -I$(BF_MOZILLA)/include/mozilla/xpcom -I$(BF_MOZILLA)/include/mozilla/idl
283 ##BF_MOZILLA_LIB = 
284 ### Will fall back to look in BF_MOZILLA_INC/nspr and BF_MOZILLA_LIB
285 ### if this is not set.
286 ##
287 ### Be paranoid regarding library creation (do not update archives)
288 ##BF_PARANOID = True
289 ##
290 ### enable freetype2 support for text objects
291                 (BoolVariable('WITH_BF_FREETYPE', 'Use FreeType2 if true', False)),
292                 ('BF_FREETYPE', 'Freetype base path', ''),
293                 ('BF_FREETYPE_INC', 'Freetype include path', ''),
294                 ('BF_FREETYPE_LIB', 'Freetype library', ''),
295                 ('BF_FREETYPE_LIBPATH', 'Freetype library path', ''),
296
297                 (BoolVariable('WITH_BF_OPENMP', 'Use OpenMP if true', False)),
298
299                 (BoolVariable('WITH_BF_QUICKTIME', 'Use QuickTime if true', False)),
300                 ('BF_QUICKTIME', 'QuickTime base path', ''),
301                 ('BF_QUICKTIME_INC', 'QuickTime include path', ''),
302                 ('BF_QUICKTIME_LIB', 'QuickTime library', ''),
303                 ('BF_QUICKTIME_LIBPATH', 'QuickTime library path', ''),
304
305                 (BoolVariable('WITH_BF_STATICOPENGL', 'Use MESA if true', True)),
306                 ('BF_OPENGL', 'OpenGL base path', ''),
307                 ('BF_OPENGL_INC', 'OpenGL include path', ''),
308                 ('BF_OPENGL_LIB', 'OpenGL libraries', ''),
309                 ('BF_OPENGL_LIBPATH', 'OpenGL library path', ''),
310                 ('BF_OPENGL_LIB_STATIC', 'OpenGL static libraries', ''),
311                 ('BF_OPENGL_LINKFLAGS', 'OpenGL link flags', ''),
312                 
313                 (BoolVariable('WITH_BF_FTGL', 'Use FTGL if true', True)),
314                 ('BF_FTGL', 'FTGL base path', ''),
315                 ('BF_FTGL_INC', 'FTGL include path', ''),
316                 ('BF_FTGL_LIB', 'FTGL libraries', ''),
317
318                 (BoolVariable('WITH_BF_PLAYER', 'Build blenderplayer if true', False)),
319                 (BoolVariable('WITH_BF_NOBLENDER', 'Do not build blender if true', False)),
320
321                 ('CFLAGS', 'C only flags', ''),
322                 ('CCFLAGS', 'Generic C and C++ flags', ''),
323                 ('CXXFLAGS', 'C++ only flags', ''),
324                 ('CPPFLAGS', 'Defines', ''),
325                 ('REL_CFLAGS', 'C only release flags', ''),
326                 ('REL_CCFLAGS', 'Generic C and C++ release flags', ''),
327                 ('REL_CXXFLAGS', 'C++ only release flags', ''),
328
329                 ('C_WARN', 'C warning flags', ''),
330                 ('CC_WARN', 'Generic C and C++ warning flags', ''),
331                 ('CXX_WARN', 'C++ only warning flags', ''),
332
333                 ('LLIBS', 'Platform libs', ''),
334                 ('PLATFORM_LINKFLAGS', 'Platform linkflags', ''),
335
336                 (BoolVariable('BF_PROFILE', 'Add profiling information if true', False)),
337                 ('BF_PROFILE_CFLAGS', 'C only profiling flags', ''),
338                 ('BF_PROFILE_CCFLAGS', 'C and C++ profiling flags', ''),
339                 ('BF_PROFILE_CXXFLAGS', 'C++ only profiling flags', ''),
340
341                 (BoolVariable('BF_DEBUG', 'Add debug flags if true', False)),
342                 ('BF_DEBUG_CFLAGS', 'C only debug flags', ''),
343                 ('BF_DEBUG_CCFLAGS', 'C and C++ debug flags', ''),
344                 ('BF_DEBUG_CXXFLAGS', 'C++ only debug flags', ''),
345
346                 (BoolVariable('BF_BSC', 'Create .bsc files (msvc only)', True)),
347
348                 ('BF_BUILDDIR', 'Build dir', ''),
349                 ('BF_INSTALLDIR', 'Installation dir', ''),
350                 ('BF_DOCDIR', 'Dir where BPy documentation will be created', ''),
351
352                 ('CC', 'C compiler to use', ''),
353                 ('CXX', 'C++ compiler to use', ''),
354
355                 (BoolVariable('BF_BUILDINFO', 'Buildtime in splash if true', True)),
356
357                 (BoolVariable('BF_TWEAK_MODE', 'Enable tweak mode if true', False)),
358                 (BoolVariable('BF_SPLIT_SRC', 'Split src lib into several chunks if true', False)),
359                 (BoolVariable('WITHOUT_BF_INSTALL', 'dont install if true', False)),
360                 (BoolVariable('BF_FANCY', 'Enable fancy output if true', True)),
361                 (BoolVariable('BF_QUIET', 'Enable silent output if true', True)),
362                 (BoolVariable('WITH_BF_BINRELOC', 'Enable relocatable binary (linux only)', False)),
363
364                 ('BF_X264_CONFIG', 'configuration flags for x264', ''),
365                 ('BF_XVIDCORE_CONFIG', 'configuration flags for xvidcore', ''),
366                 (BoolVariable('WITH_BF_DOCS', 'Generate API documentation', False)),
367                 
368                 ('BF_CONFIG', 'SCons python config file used to set default options', 'user_config.py'),
369                 ('BF_NUMJOBS', 'Number of build processes to spawn', '1')
370
371         ) # end of opts.AddOptions()
372
373         return localopts
374
375 def NSIS_print(target, source, env):
376         return "Creating NSIS installer for Blender 3D"
377
378 def NSIS_Installer(target=None, source=None, env=None):
379
380         if env['OURPLATFORM'] != 'win32-vc' and env['OURPLATFORM'] != 'win32-mingw':
381                 print "NSIS installer is only available on Windows."
382                 Exit()
383                 
384         start_dir = os.getcwd()
385         rel_dir = start_dir + "\\release\\windows\\installer\\"
386         install_base_dir = start_dir + "\\"
387         
388         if not os.path.exists(install_base_dir+env['BF_INSTALLDIR']+'/plugins/include'):
389                 os.mkdir(install_base_dir+env['BF_INSTALLDIR']+'/plugins/include')
390                 
391         for f in glob.glob('source/blender/blenpluginapi/*.h'):
392                 shutil.copy(f,install_base_dir+env['BF_INSTALLDIR']+'/plugins/include')
393
394         shutil.copy('source/blender/blenpluginapi/plugin.def',install_base_dir+env['BF_INSTALLDIR']+'/plugins/include/')
395         
396         os.chdir("release")
397         v = open("VERSION")
398         version = v.read()[:-1] 
399         shortver = version.split('.')[0] + version.split('.')[1]
400         v.close()
401
402         #### change to suit install dir ####
403         inst_dir = install_base_dir + env['BF_INSTALLDIR']
404         
405         os.chdir("windows/installer")
406
407         ns = open("00.sconsblender.nsi","r")
408
409         ns_cnt = str(ns.read())
410         ns.close()
411
412         # do root
413         rootlist = []
414         rootdir = os.listdir(inst_dir+"\\")
415         for rootitem in rootdir:
416                 if os.path.isdir(inst_dir+"\\"+ rootitem) == 0:
417                         rootlist.append("File \"" + os.path.normpath(inst_dir) + "\\" + rootitem+"\"")
418         rootstring = string.join(rootlist, "\n  ")
419         rootstring += "\n\n"
420         ns_cnt = string.replace(ns_cnt, "[ROOTDIRCONTS]", rootstring)
421
422         # do delete items
423         delrootlist = []
424         for rootitem in rootdir:
425                 if os.path.isdir(inst_dir + rootitem) == 0:
426                         delrootlist.append("Delete $INSTDIR\\" + rootitem)
427         delrootstring = string.join(delrootlist, "\n ")
428         delrootstring += "\n"
429         ns_cnt = string.replace(ns_cnt, "[DELROOTDIRCONTS]", delrootstring)
430
431         # do scripts
432         scriptlist = []
433         scriptpath = "%s%s" % (inst_dir, "\\.blender\\scripts")
434         scriptdir = os.listdir(scriptpath)
435         for scriptitem in scriptdir:
436                 scriptfile = "%s\\%s" % (scriptpath, scriptitem)
437                 if os.path.isdir(scriptfile) == 0:
438                         scriptfile = os.path.normpath(scriptfile)
439                         scriptlist.append("File \"%s\"" % scriptfile)
440         scriptstring = string.join(scriptlist, "\n  ")
441         scriptstring += "\n\n"
442         ns_cnt = string.replace(ns_cnt, "[SCRIPTCONTS]", scriptstring)
443
444         # do scripts\bpymodules
445         bpymodlist = []
446         bpymodpath = "%s%s" % (inst_dir, "\\.blender\\scripts\\bpymodules")
447         bpymoddir = os.listdir(bpymodpath)
448
449         for bpymoditem in bpymoddir:
450                 bpymodfile = "%s\\%s" % (bpymodpath, bpymoditem)
451                 if os.path.isdir(bpymodfile) == 0:
452                         bpymodfile = os.path.normpath(bpymodfile)
453                         bpymodlist.append("File \"%s\"" % bpymodfile)
454         bpymodstring = string.join(bpymodlist, "\n  ")
455         bpymodstring += "\n\n"
456         ns_cnt = string.replace(ns_cnt, "[SCRIPTMODCONTS]", bpymodstring)
457
458         # do scripts\bpymodules\colladaimex
459         colladalist = []
460         bpymodpath = "%s%s" % (inst_dir, "\\.blender\\scripts\\bpymodules\\ColladaImEx")
461         bpymoddir = os.listdir(bpymodpath)
462
463         for bpymoditem in bpymoddir:
464                 bpymodfile = "%s\\%s" % (bpymodpath, bpymoditem)
465                 if os.path.isdir(bpymodfile) == 0:
466                         bpymodfile=os.path.normpath(bpymodfile)
467                         colladalist.append("File \"%s\"" % bpymodfile)
468         bpymodstring = string.join(colladalist, "\n  ")
469         bpymodstring += "\n\n"
470         ns_cnt = string.replace(ns_cnt, "[SCRIPTMODCOLLADACONT]", bpymodstring)
471
472         # do scripts\bpydata
473         bpydatalist = []
474         bpydatapath = "%s%s" % (inst_dir, "\\.blender\\scripts\\bpydata")
475         bpydatadir = os.listdir(bpydatapath)
476         for bpydataitem in bpydatadir:
477                 bpydatafile = "%s\\%s" % (bpydatapath, bpydataitem)
478                 if os.path.isdir(bpydatafile) == 0:
479                         bpydatalist.append("File \"%s\"" % bpydatafile)
480         bpydatastring = string.join(bpydatalist, "\n  ")
481         bpydatastring += "\n\n"
482         ns_cnt = string.replace(ns_cnt, "[SCRIPTDATACONTS]", bpydatastring)
483
484         # do plugins\include
485         plugincludelist = []
486         plugincludepath = "%s%s" % (inst_dir, "\\plugins\\include")
487         plugincludedir = os.listdir(plugincludepath)
488         for plugincludeitem in plugincludedir:
489                 plugincludefile = "%s\\%s" % (plugincludepath, plugincludeitem)
490                 if os.path.isdir(plugincludefile) == 0:
491                         if plugincludefile.find('.h') or plugincludefile.find('.DEF'):
492                                 plugincludefile = os.path.normpath(plugincludefile)
493                                 plugincludelist.append("File \"%s\"" % plugincludefile)
494         plugincludestring = string.join(plugincludelist, "\n  ")
495         plugincludestring += "\n\n"
496         ns_cnt = string.replace(ns_cnt, "[PLUGINCONTS]", plugincludestring)
497
498         # do scripts\bpydata\config
499         cfglist = []
500         cfgpath = "%s%s" % (inst_dir, "\\.blender\\scripts\\bpydata\\config")
501         cfgdir = os.listdir(cfgpath)
502         for cfgitem in cfgdir:
503                 cfgfile = "%s\\%s" % (cfgpath, cfgitem)
504                 if os.path.isdir(cfgfile) == 0:
505                         cfglist.append("File \"%s\"" % cfgfile)
506         cfgstring = string.join(cfglist, "\n  ")
507         cfgstring += "\n\n"
508         ns_cnt = string.replace(ns_cnt, "[SCRIPTDATACFGCONTS]", cfgstring)
509
510         # do dotblender
511         dotblendlist = []
512         dotblenddir = os.listdir(inst_dir+"\\.blender")
513         for dotblenditem in dotblenddir:
514                 if os.path.isdir(inst_dir + "\\.blender\\" + dotblenditem) == 0:
515                         dotblendlist.append("File \"" + os.path.normpath(inst_dir) + "\\.blender\\" +
516                         dotblenditem+"\"")
517         dotblendstring = string.join(dotblendlist, "\n  ")
518         dotblendstring += "\n\n"
519         ns_cnt = string.replace(ns_cnt, "[DOTBLENDERCONTS]", dotblendstring)
520
521         # do language files
522         langlist = []
523         langfiles = []
524         langdir = os.listdir(inst_dir + "\\.blender\\locale")
525         for langitem in langdir:
526                 if os.path.isdir(inst_dir + "\\.blender\\locale\\" + langitem) == 1:
527                         langfiles.append("SetOutPath $BLENDERHOME\\.blender\\locale\\" + langitem + "\\LC_MESSAGES")
528                         langfiles.append("File \"" + os.path.normpath(inst_dir) + "\\.blender\\locale\\"
529                                         + langitem + "\\LC_MESSAGES\\blender.mo\"")
530         langstring = string.join(langfiles, "\n  ")
531         langstring += "\n\n"
532         ns_cnt = string.replace(ns_cnt, "[LANGUAGECONTS]", langstring)
533
534         # var replacements
535         ns_cnt = string.replace(ns_cnt, "DISTDIR", os.path.normpath(inst_dir+"\\"))
536         ns_cnt = string.replace(ns_cnt, "SHORTVER", shortver)
537         ns_cnt = string.replace(ns_cnt, "VERSION", version)
538         ns_cnt = string.replace(ns_cnt, "RELDIR", os.path.normpath(rel_dir))
539
540         tmpnsi = os.path.normpath(install_base_dir+os.sep+env['BF_BUILDDIR']+os.sep+"00.blender_tmp.nsi")
541         new_nsis = open(tmpnsi, 'w')
542         new_nsis.write(ns_cnt)
543         new_nsis.close()
544
545         os.chdir(start_dir)
546
547         cmdline = "makensis " + "\""+tmpnsi+"\""
548
549         startupinfo = subprocess.STARTUPINFO()
550         startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
551         proc = subprocess.Popen(cmdline, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
552                 stderr=subprocess.PIPE, startupinfo=startupinfo, shell = True)
553         data, err = proc.communicate()
554         rv = proc.wait()
555
556         if rv != 0:
557                 print
558                 print data.strip().split("\n")[-1]
559         return rv
560