count_lines.py

#!/usr/bin/env python3

# Copyright 1996-2024 Cyberbotics Ltd.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Display the number of lines of codes in this repository for different programming languages."""

import glob
import os


def count_lines(filename):
    f = open(filename, 'rb')
    lines = 0
    buf_size = 1024 * 1024
    read_f = f.raw.read
    buf = read_f(buf_size)
    while buf:
        lines += buf.count(b'\n')
        buf = read_f(buf_size)
    return lines


def count_files(extensions, name):
    line_counter = 0
    file_counter = 0
    if isinstance(extensions, str):
        extensions = [extensions]
    for extension in extensions:
        files = glob.glob(os.path.join('..', '..', '**', '*.' + extension), recursive=True)
        file_counter += len(files)
        for f in files:
            line_counter += count_lines(f)
    print((name + ':').ljust(12) + str(line_counter).ljust(6) + ' lines of codes in ' + str(file_counter).ljust(4) + ' files')


count_files(['cpp', 'hpp'], 'C++')
count_files(['c', 'h'], 'C')
count_files(['py'], 'Python')
count_files('js', 'JavaScript')
count_files('java', 'Java')
count_files('m', 'MATLAB')
count_files('vert', 'GLSL')