← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to setup.py

Bring in the 'rabin' experiment.
Change the names and disk-strings for the various repository formats.
Make the CHK format repositories all 'rich-root' we can introduce non-rich-root later.
Make a couple other small tweaks, like copyright statements, etc.
Remove patch-delta.c, at this point, it was only a reference implementation,
as we have fully integrated the patching into pyrex, to allow nicer exception
handling.

Show diffs side-by-side

added added

removed removed

Lines of Context:
setup.py
28
28
 
29
29
class build_ext_if_possible(build_ext):
30
30
 
 
31
    user_options = build_ext.user_options + [
 
32
        ('allow-python-fallback', None,
 
33
         "When an extension cannot be built, allow falling"
 
34
         " back to the pure-python implementation.")
 
35
        ]
 
36
 
 
37
    def initialize_options(self):
 
38
        build_ext.initialize_options(self)
 
39
        self.allow_python_fallback = False
 
40
 
31
41
    def run(self):
32
42
        try:
33
43
            build_ext.run(self)
34
44
        except DistutilsPlatformError, e:
 
45
            if not self.allow_python_fallback:
 
46
                log.warn('\n  Cannot build extensions.\n'
 
47
                         '  Use "build_ext --allow-python-fallback" to use'
 
48
                         ' slower python implementations instead.\n')
 
49
                raise
35
50
            log.warn(str(e))
36
 
            log.warn('Extensions cannot be built, '
37
 
                     'will use the Python versions instead')
 
51
            log.warn('\n  Extensions cannot be built.\n'
 
52
                     '  Using the slower Python implementations instead.\n')
38
53
 
39
54
    def build_extension(self, ext):
40
55
        try:
41
56
            build_ext.build_extension(self, ext)
42
57
        except CCompilerError:
43
 
            log.warn('Building of "%s" extension failed, '
44
 
                     'will use the Python version instead' % (ext.name,))
 
58
            if not self.allow_python_fallback:
 
59
                log.warn('\n  Cannot build extension (%s).\n'
 
60
                         '  Use "build_ext --allow-python-fallback" to use'
 
61
                         ' slower python implementations instead.\n'
 
62
                         % (ext.name,))
 
63
                raise
 
64
            log.warn('\n  Building of "%s" extension failed.\n'
 
65
                     '  Using the slower Python implementation instead.'
 
66
                     % (ext.name,))
45
67
 
46
68
 
47
69
# Override the build_ext if we have Pyrex available
48
70
unavailable_files = []
49
71
 
50
72
 
51
 
def add_pyrex_extension(module_name, **kwargs):
 
73
def add_pyrex_extension(module_name, extra_source=[]):
52
74
    """Add a pyrex module to build.
53
75
 
54
76
    This will use Pyrex to auto-generate the .c file if it is available.
67
89
    # Manually honour package_dir :(
68
90
    module_name = 'bzrlib.plugins.groupcompress.' + module_name
69
91
    if have_pyrex:
70
 
        ext_modules.append(Extension(module_name, [pyrex_name]))
 
92
        source = [pyrex_name]
 
93
    elif not os.path.isfile(c_name):
 
94
        unavailable_files.append(c_name)
 
95
        return
71
96
    else:
72
 
        if not os.path.isfile(c_name):
73
 
            unavailable_files.append(c_name)
74
 
        else:
75
 
            ext_modules.append(Extension(module_name, [c_name]))
 
97
        source = [c_name]
 
98
    source.extend(extra_source)
 
99
    ext_modules.append(Extension(module_name, source,
 
100
        extra_compile_args = ['-O3']))
76
101
 
77
 
add_pyrex_extension('_groupcompress_c')
 
102
add_pyrex_extension('_groupcompress_pyx',
 
103
                    extra_source=['diff-delta.c'])
78
104
 
79
105
 
80
106
if __name__ == '__main__':