FilesExpand file tree

 main

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

114 lines (95 loc) · 3.66 KB

 main

/

setup.py

Copy path

Top

File metadata and controls

• Code
• Blame

114 lines (95 loc) · 3.66 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

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

import os

import platform

import sys

import ssl

import urllib.request

from pathlib import Path

def get_os_name():

system = platform.system()

if system == "Windows":

return "Windows 64-bit" if platform.machine().endswith('64') else "Windows 32-bit"

elif system == "Darwin":

return "macOS"

elif system == "Linux":

return "Linux"

elif system == "FreeBSD":

return "FreeBSD"

elif system.startswith("CYGWIN"):

return "Cygwin"

else:

return "Other"

def download_file(url, output_filename):

total_downloaded = 0

total_truncated = 0 # only print download progress every 2MB to avoid spamming logs

def report_progress(block_num, block_size, total_size):

nonlocal total_downloaded

nonlocal total_truncated

total_downloaded += block_size

if total_downloaded // (1024 * 1024) > total_truncated:

total_truncated = total_downloaded // (1024 * 1024)

if total_truncated % 2 == 0:

print(f"\rDownloaded {total_downloaded // (1024 * 1024)} MB", end="")

try:

ssl._create_default_https_context = ssl._create_stdlib_context

urllib.request.urlretrieve(url, output_filename, reporthook=report_progress)

print(f"\nDownloaded {output_filename}")

return True

except Exception as e:

print(f"\nFailed to download {output_filename}")

print(f"Error: {str(e)}")

sys.exit(1)

def check_os(os_name):

print("\nChecking System")

print("===============\n")

print(f" Operating System : {os_name}")

supported = {"macOS", "Linux"}

if os_name not in supported:

print("Unsupported operating system")

sys.exit(1)

def download_dawn(os_name):

print("\nDownload Dawn Library")

print("=====================\n")

outfile_map = {

"macOS": "third_party/lib/libwebgpu_dawn.dylib",

"Linux": "third_party/lib/libwebgpu_dawn.so",

}

url_map = {

"macOS": "https://github.com/austinvhuang/dawn-artifacts/releases/download/0.2.0/libwebgpu_dawn.dylib",

"Linux": "https://github.com/austinvhuang/dawn-artifacts/releases/download/0.2.0/libwebgpu_dawn.so",

}

outfile = outfile_map.get(os_name)

url = url_map.get(os_name)

if not outfile or not url:

print(f"No download information for {os_name}")

sys.exit(1)

print(f" URL : {url}")

print(f" Download File : {outfile}\n")

print(" Downloading ...\n")

if Path(outfile).exists():

print(f" File {outfile} already exists, skipping.")

sys.exit(0)

Path(outfile).parent.mkdir(parents=True, exist_ok=True)

download_file(url, outfile)

def setup_env(os_name):

print("\nEnvironment Setup")

print("=================\n")

current_dir = os.getcwd()

lib_dir = os.path.join(current_dir, "third_party", "lib")

if os_name == "macOS":

print(" Before running the program, run the following command or add it to your shell profile:")

print(f" export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:{lib_dir}")

with open("source", "w") as f:

f.write(f"export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:{lib_dir}\n")

if os_name == "Linux":

print(" Before running the program, run the following command or add it to your shell profile:")

print(f" export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:{lib_dir}")

with open("source", "w") as f:

f.write(f"export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:{lib_dir}\n")

def main():

os_name = get_os_name()

check_os(os_name)

download_dawn(os_name)

setup_env(os_name)

print()

if __name__ == "__main__":

main()