3dtiles totalB3dmFileNumbers.py

"""
此模块用于，获取某个3dtiles模型的所有.b3dm文件数。
"""
import os


def getPathFilesInDirectory(directory, pathFiles=[]):
    """
    此函数用于，获取directory目录下，所有文件的绝对路径，
    并存放在pathFiles里。
    directory下可以同时有目录和文件。 
    如果在调用此函数时，未对pathFiles传参，
    则pathFiles列表只在函数定义时创建，分配固定起始地址的存储空间。
    """
    filesFolders = os.listdir(directory)
    # print(filesFolders)
    for f in filesFolders:
        pathFileOrFolder = os.path.join(directory, f)
        if os.path.isfile(pathFileOrFolder):
            # print(pathFileOrFolder)
            f = pathFileOrFolder
            if f[f.rfind(".") + 1:] == "b3dm":
                pathFiles.append(pathFileOrFolder)
        else:
            getPathFilesInDirectory(pathFileOrFolder, pathFiles)
    return pathFiles


def main():
    """
    主函数。
    :return:
    """
    directory = r"D:\MarsGisPlatform\data\3dtiles\小学-机电合模\1"
    # getPathFilesInDirectory函数，若是在同一模块中反复使用，必须传参，以免混淆结果。
    # 如果此函数，只使用一次，可传可不传。
    # pathFiles = getPathFilesInDirectory(directory, [])
    # 上下两种形式均可。
    pathFiles = []
    getPathFilesInDirectory(directory, pathFiles)
    print(pathFiles)
    print(len(pathFiles))  # 677


if __name__ == "__main__":
    main()