FilesExpand file tree

master

/

metal2string.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

96 lines (74 loc) · 3.65 KB

master

/

metal2string.py

Copy path

Top

File metadata and controls

• Code
• Blame

96 lines (74 loc) · 3.65 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

#!/usr/bin/python

# Modification (c)Ronan LE MEILLAT

# Based on

# Copyright (C) 2019 Matej Gomboc https://github.com/MatejGomboc/Galaxy-Simulator

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version. This program is distributed in the

# hope that it will be useful, but WITHOUT ANY WARRANTY; without even the

# implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

# See the GNU Affero General Public License for more details. You should

# have received a copy of the GNU Affero General Public License along with

# this program. If not, see <https://www.gnu.org/licenses/>.

import os, sys, argparse, re

def comment_remover(text):

def replacer(match):

s = match.group(0)

if s.startswith('/'):

return " " # note: a space and not an empty string

else:

return s

pattern = re.compile(

r'//.*?$|/\*.*?\*/|\'(?:\\.|[^\\\'])*\'|"(?:\\.|[^\\"])*"',

re.DOTALL | re.MULTILINE

)

return re.sub(pattern, replacer, text)

if __name__ == "__main__":

parser = argparse.ArgumentParser()

parser.add_argument("-v", "--verbose", help="increase output verbosity", action="store_true")

parser.add_argument("-c", "--compact", help="remove all comments and empty lines", action="store_true")

parser.add_argument("inputFileName", help="name of input OpenCL file with source (*.cl, *.clh)", type=str)

parser.add_argument("outputFileName", help="name of output C/C++ file with source as C char string (*.c, *.cpp, *.h)", type=str)

args = parser.parse_args()

if not args.inputFileName.endswith(".metal") :

sys.exit("invalid input file extension: " + args.inputFileName)

if not args.outputFileName.endswith(".h") :

sys.exit("invalid output file extension: " + args.outputFileName)

if args.verbose:

print("input file: " + args.inputFileName)

print("output file: " + args.outputFileName)

if not args.compact:

with open(args.inputFileName, "r") as file:

stringSource = "\n".join([line.rstrip() for line in file]) # remove trailing whitespaces

else:

stringSource = open(args.inputFileName, "r").read()

if args.verbose:

print("\noriginal OpenCL source:")

print(stringSource)

if args.compact:

stringSource = comment_remover(stringSource) # remove all comments

stringSource = stringSource.replace("\t", " ") # replace '\t' with single space

stringSource = stringSource.replace(" ", " ") # replace triple space with single space

stringSource = stringSource.replace(" ", " ") # replace double space with single space

# remove spaces around each operator and bracket

for char in "!\"%&/()=?*+-|,.;:^{}~\'[]<>":

stringSource = stringSource.replace(char + " ", char)

stringSource = stringSource.replace(" " + char, char)

temp = [line.strip() for line in stringSource.split("\n")] # remove leading and trailing whitespaces

stringSource = "\n".join([line for line in temp if line]) # remove empty lines

if not args.compact:

stringSource = stringSource.replace("\\", "\\\\") # replace '\\' in comments

stringSource = stringSource.replace("\"", "\\\"") # replace '\"' in comments

stringSource = stringSource.replace("\'", "\\\'") # replace '\'' in comments

stringSource = stringSource.replace("\n", "\\n\"\n\"") # replace '\n'

stringSource = "const char* metal_src_" + os.path.splitext(args.inputFileName)[0] + " =\n\"" + stringSource

if args.compact:

stringSource += "\";\n"

else:

stringSource += "\\n\";\n"

if args.verbose:

print("\nOpenCL source as C char string:")

print(stringSource)

open(args.outputFileName, "w").write(stringSource)