summaryrefslogtreecommitdiff
path: root/scripts/code_tools/fix_whitespace.py
blob: 7a88265479493cbbc899ba7be3c9632a32d7c3b7 (plain)
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
#!/usr/bin/env python
"""\

This script replaces tab characters with spaces in source code files.

$LicenseInfo:firstyear=2024&license=viewerlgpl$
Second Life Viewer Source Code
Copyright (C) 2024, Linden Research, Inc.

This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation;
version 2.1 of the License only.

This library 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
Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA
$/LicenseInfo$
"""

import argparse
import os

def convert_tabs_to_spaces(file_path, tab_stop):
    """Convert tabs in a file to spaces, considering tab stops."""
    with open(file_path, 'r') as file:
        lines = file.readlines()

    # Skip files with no tabs
    if not any('\t' in line for line in lines):
        return

    new_lines = []
    for line in lines:
        # Remove trailing spaces
        line = line.rstrip()
        new_line = ''
        column = 0  # Track the column index for calculating tab stops
        for char in line:
            if char == '\t':
                # Calculate spaces needed to reach the next tab stop
                spaces_needed = tab_stop - (column % tab_stop)
                new_line += ' ' * spaces_needed
                column += spaces_needed
            else:
                new_line += char
                column += 1

        new_lines.append(new_line + '\n')

    with open(file_path, 'w', newline='\n') as file:
        file.writelines(new_lines)

def process_directory(directory, extensions, tab_stop):
    """Recursively process files in directory, considering tab stops."""
    extensions = tuple(extensions)
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extensions):
                file_path = os.path.join(root, file)
                print(f"Processing {file_path}")
                convert_tabs_to_spaces(file_path, tab_stop)

def main():
    parser = argparse.ArgumentParser(description='Convert tabs to spaces in files, considering tab stops.')
    parser.add_argument('-e', '--extensions', type=str, default='c,cpp,h,hpp,inl,py,glsl,cmake', help='Comma-separated list of file extensions to process (default: "c,cpp,h,hpp,inl,py,glsl,cmake")')
    parser.add_argument('-t', '--tabstop', type=int, default=4, help='Tab stop size (default: 4)')
    parser.add_argument('-d', '--directory', type=str, required=True, help='Directory to process')

    args = parser.parse_args()

    extensions = args.extensions.split(',')
    # Add a dot prefix to each extension if not present
    extensions = [ext if ext.startswith('.') else f".{ext}" for ext in extensions]

    process_directory(args.directory, extensions, args.tabstop)
    print("Processing completed.")

if __name__ == "__main__":
    main()