vlasiator/updateGpuMemoryPointerList.sh at master · fmihpc/vlasiator · GitHub

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
#!/usr/bin/env bash
# Collects all pointers passed to gpuMemoryManager
# and writes them into ./arch/gpu_base.hpp between
# //DEFINITIONS_HERE and //DEFINITIONS_END markers.

shopt -s globstar nullglob

declare -A seen

# --- Collect unique identifiers ---
for file in **/*.cpp **/*.h **/*.hpp; do
   # Skip any files inside directories named "libraries" or "submodules"
   [[ "$file" == libraries* ]] && continue
   [[ "$file" == */libraries* ]] && continue
   [[ "$file" == submodules* ]] && continue
   [[ "$file" == */submodules* ]] && continue

   while IFS= read -r match; do
      [[ -z "$match" ]] && continue
      seen["$match"]=1
   done < <(
      # Match CREATE_UNIQUE_POINTER(obj, identifier)
      grep -oP 'CREATE_UNIQUE_POINTER\s*\(\s*[A-Za-z_][A-Za-z0-9_]*\s*,\s*[A-Za-z_][A-Za-z0-9_]*\s*\)' "$file" 2>/dev/null |
      sed -E 's/.*CREATE_UNIQUE_POINTER\s*\(\s*[A-Za-z_][A-Za-z0-9_]*\s*,\s*([A-Za-z_][A-Za-z0-9_]*)\s*\).*/\1/'

      # Match CREATE_SUBPOINTERS(obj, identifier, ...)
      grep -oP 'CREATE_SUBPOINTERS\s*\(\s*[A-Za-z_][A-Za-z0-9_]*\s*,\s*[A-Za-z_][A-Za-z0-9_]*\s*,[^)]*\)' "$file" 2>/dev/null |
      sed -E 's/.*CREATE_SUBPOINTERS\s*\(\s*[A-Za-z_][A-Za-z0-9_]*\s*,\s*([A-Za-z_][A-Za-z0-9_]*)\s*,.*/\1/'

      grep -oP 'SESSION(_HOST)?_ALLOCATE\s*\(\s*[A-Za-z_][A-Za-z0-9_]*\s*,\s*[A-Za-z_][A-Za-z0-9_:<>* ]*\s*,\s*[A-Za-z_][A-Za-z0-9_]*\s*,[^)]*\)' "$file" 2>/dev/null |
      sed -E 's/.*SESSION(_HOST)?_ALLOCATE\s*\(\s*[A-Za-z_][A-Za-z0-9_]*\s*,\s*[A-Za-z_][A-Za-z0-9_:<>* ]*\s*,\s*([A-Za-z_][A-Za-z0-9_]*)\s*,.*/\2/'

   )
done


# --- Prepare the definitions content ---
definitions=$(for key in "${!seen[@]}"; do echo "$key"; done | sort)

# --- Update gpu_base.hpp ---
target="./arch/gpu_base.hpp"

if [[ ! -f "$target" ]]; then
   echo "[ERROR] $target not found."
   exit 1
fi

# Create a backup
cp "$target" "$target.bak"

# Use awk to replace content between #define DEFINITIONS_HERE / END markers
# Generate new content in a temp file
tmpfile=$(mktemp)

awk -v defs="$definitions" '
   BEGIN { split(defs, arr, "\n") }
   /^[[:space:]]*#define[[:space:]]+DEFINITIONS_HERE/ {
      print;  # print the marker exactly as it appears
      first = 1
      line = "   size_t "
      for (i in arr) {
         if (length(arr[i]) > 0) {
            if (!first) line = line ", "
            line = line arr[i]
            first = 0
         }
      }
      if (!first) print line ";"
      skip = 1
      next
   }
   /^[[:space:]]*#define[[:space:]]+DEFINITIONS_END/ {
      skip = 0
   }
   !skip
' "$target.bak" > "$tmpfile"

# Only overwrite if the content actually changed
if ! cmp -s "$tmpfile" "$target"; then
   mv "$tmpfile" "$target"
   echo "[SCAN] Updated $target"
else
   rm "$tmpfile"
fi

rm "$target.bak"