lapacke 0.3.0

The package provides wrappers for LAPACKE (C).
Documentation 
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

import re

argument_re = re.compile("\s*(\w+): ([^,\)]+)([,\)\s]*)")
name_re = re.compile("\s*pub fn (?:LAPACKE_)?(\w+[a-z0-9])(_?)")
return_re = re.compile("(?:\s*->\s*([^;]+))?")

class Function(object):
    def __init__(self, name, args, ret):
        self.name = name
        self.args = args
        self.ret = ret

    @staticmethod
    def parse(line):
        name, line = pull_name(line)
        if name is None:
            return None

        assert(line[0] == '(')
        line = line[1:]
        args = []
        while True:
            arg, aty, line = pull_argument(line)
            if arg is None:
                break
            if arg == "matrix_layout":
                arg = "layout"
            args.append((arg, aty))
            line = line.strip()

        ret, line = pull_return(line)

        return Function(name, args, ret)

def pull_argument(s):
    match = argument_re.match(s)
    if match is None:
        return None, None, s
    return match.group(1), match.group(2), s[match.end(3):]

def pull_name(s):
    match = name_re.match(s)
    assert(match is not None)
    return match.group(1), s[match.end(2):]

def pull_return(s):
    match = return_re.match(s)
    if match is None:
        return None, s
    return match.group(1), s[match.end(1):]

def read_functions(path):
    lines = []
    with open(path) as file:
        append = False
        for line in file:
            if line == 'extern "C" {\n':
                append = True
            elif line == '}\n':
                append = False
            elif append:
                lines.append(line)
    return ''.join(lines)