build/scripts/build_pln_header.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

#!/usr/bin/env python

import sys
import os


def BuildPlnHeader():
    if len(sys.argv) < 2:
        print >>sys.stderr, "Usage: build_pln_header.py <absolute/path/to/OutFile>"
        sys.exit(1)

    print >>sys.stdout, "Build Pln Header..."
    outPath = sys.argv[1]
    tmpPath = outPath + '.tmp'
    tmpFile = open(tmpPath, 'w')

    tmpFile.write('#include <library/cpp/sse/sse.h>\n')
    tmpFile.write('#include <kernel/relevfml/relev_fml.h>\n')
    for path in sys.argv[2:]:
        name = os.path.basename(path).split(".")[0]  # name without extensions
        tmpFile.write('\nextern SRelevanceFormula fml{0};\n'.format(name))
        tmpFile.write('float {0}(const float* f);\n'.format(name))
        tmpFile.write('void {0}SSE(const float* const* factors, float* result);\n'.format(name))
    tmpFile.close()
    try:
        os.remove(outPath)
    except:
        pass
    try:
        os.rename(tmpPath, outPath)
    except:
        print >>sys.stdout, 'Error: Failed to rename ' + tmpPath + ' to ' + outPath

if __name__ == '__main__':
    BuildPlnHeader()