neverlib 0.2.7__py3-none-any.whl → 0.2.8__py3-none-any.whl

neverlib/QA/gen_init.py

@@ -1,4 +1,5 @@
 # 自动生成指定包目录下的 __init__.py（懒加载格式：lazy_loader.attach）
+# 并且支持IDE友好的 __init__.py 版本
 
 import ast
 from pathlib import Path
@@ -43,7 +44,7 @@ def _extract_exports_from_module(py_file: Path) -> List[str]:
     return sorted(exports)
 
 
-def generate_init_for_directory(package_dir: Path) -> Path:
+def generate_init_for_directory(package_dir: Path, ide_friendly: bool = False) -> Path:
     """
     为指定目录生成懒加载版 __init__.py（覆盖写入）。
     - 仅扫描一级子模块（同级 .py 文件），忽略以下划线开头的模块与 __init__.py 本身。
@@ -55,6 +56,10 @@ def generate_init_for_directory(package_dir: Path) -> Path:
             submodules=["m1", "m2", ...],
             submod_attrs={"m1": ["A", "B"], ...}
         )
+
+    Args:
+        package_dir: 包目录路径
+        ide_friendly: 是否生成IDE友好版本（使用TYPE_CHECKING）
     """
     package_dir = package_dir.resolve()
     if not package_dir.is_dir():
@@ -76,6 +81,24 @@ def generate_init_for_directory(package_dir: Path) -> Path:
     lines: List[str] = []
     lines.append('# This file is auto-generated. Do NOT edit manually.')
     lines.append('# Generated by neverlib.QA.gen_init')
+
+    if ide_friendly:
+        lines.append('')
+        lines.append('from typing import TYPE_CHECKING, Any')
+        lines.append('')
+        lines.append('if TYPE_CHECKING:')
+        lines.append('    # 仅在类型检查时导入，提供IDE补全支持')
+
+        # 生成 TYPE_CHECKING 导入
+        for module_name in sorted(module_to_exports.keys()):
+            exports = module_to_exports[module_name]
+            if exports:
+                imports = ', '.join(exports)
+                lines.append(f'    from .{module_name} import {imports}')
+
+        lines.append('')
+        lines.append('# 运行时使用懒加载')
+
     lines.append('from lazy_loader import attach')
     lines.append('')
     lines.append('__getattr__, __dir__, __all__ = attach(')
@@ -93,24 +116,102 @@ def generate_init_for_directory(package_dir: Path) -> Path:
     lines.append(')')
     lines.append('')
 
+    if ide_friendly:
+        # 生成显式的 __all__ 声明
+        lines.append('# 显式声明 __all__ 以便 IDE 识别')
+        lines.append('if TYPE_CHECKING:')
+        lines.append('    __all__ = [')
+
+        all_exports = []
+        for module_name in sorted(module_to_exports.keys()):
+            exports = module_to_exports[module_name]
+            for export in exports:
+                all_exports.append(f"        '{export}',")
+
+        lines.extend(all_exports)
+        lines.append('    ]')
+
     init_file = package_dir / '__init__.py'
     init_file.write_text('\n'.join(lines), encoding='utf-8')
     return init_file
 
 
+def generate_all_packages(root_dir: Path, ide_friendly: bool = False) -> List[Path]:
+    """
+    为所有子包生成 __init__.py 文件
+
+    Args:
+        root_dir: 根目录路径
+        ide_friendly: 是否生成IDE友好版本
+
+    Returns:
+        生成的文件路径列表
+    """
+    generated_files = []
+    processed_dirs = set()  # 避免重复处理
+
+    # 查找所有包含 __init__.py 的包目录
+    for init_file in root_dir.rglob('__init__.py'):
+        package_dir = init_file.parent
+
+        # 跳过根目录和已处理的目录
+        if package_dir == root_dir or package_dir in processed_dirs:
+            continue
+
+        # 检查是否包含其他Python文件（不是只有__init__.py）
+        has_other_py_files = any(
+            f.name != '__init__.py' and f.suffix == '.py'
+            for f in package_dir.iterdir()
+        )
+
+        if has_other_py_files:
+            processed_dirs.add(package_dir)
+            try:
+                init_path = generate_init_for_directory(package_dir, ide_friendly=ide_friendly)
+                generated_files.append(init_path)
+                print(f'✅ 已生成: {init_path}')
+            except Exception as e:
+                print(f'❌ 生成失败 {package_dir}: {e}')
+
+    return generated_files
+
+
 def main():
     import argparse
 
     parser = argparse.ArgumentParser(description='为指定包目录自动生成 __init__.py')
     parser.add_argument('-p', '--package-dir', dest='package_dir',
-                        default='../../neverlib/vad',
                         help='包目录路径，比如 /path/to/pkg 或 ./neverlib/utils')
+    parser.add_argument('--all', action='store_true',
+                        help='为所有子包生成 __init__.py')
+    parser.add_argument('--ide-friendly', action='store_true',
+                        help='生成IDE友好版本（使用TYPE_CHECKING）')
+    parser.add_argument('--backup', action='store_true',
+                        help='备份原始文件')
     args = parser.parse_args()
 
-    target_dir = Path(args.package_dir)
-    init_path = generate_init_for_directory(target_dir)
-    print(f'已生成: {init_path}')
-# python -m neverlib.QA.gen_init /data03/never/Desktop/neverlib/neverlib/utils
+    if args.all:
+        # 批量生成所有包
+        root_dir = Path(__file__).parent.parent  # neverlib 目录
+        generated_files = generate_all_packages(root_dir, ide_friendly=args.ide_friendly)
+    else:
+        # 生成单个包
+        if not args.package_dir:
+            parser.error('请指定包目录路径 (-p) 或使用 --all 批量生成')
+
+        target_dir = Path(args.package_dir)
+
+        if args.backup and (target_dir / '__init__.py').exists():
+            backup_path = target_dir / '__init__.py.backup'
+            (target_dir / '__init__.py').rename(backup_path)
+            print(f"已备份原始文件到: {backup_path}")
+
+        init_path = generate_init_for_directory(target_dir, ide_friendly=args.ide_friendly)
+        print(f'已生成: {init_path}')
+        if args.ide_friendly:
+            print('✅ 已生成IDE友好版本（支持代码补全和类型检查）')
+        else:
+            print('ℹ️  生成标准懒加载版本，如需IDE支持请使用 --ide-friendly 参数')
 
 
 if __name__ == '__main__':

neverlib/__init__.py

@@ -37,6 +37,13 @@ try:
 except Exception:
     __version__ = "0.1.2"  # 如果出错, 使用默认版本号
 
+from typing import TYPE_CHECKING, Any
+
+
+if TYPE_CHECKING:
+    from .utils import get_path_list
+    from .filter import HPFilter
+    from .audio_aug import volume_norm
 
 # 懒加载子包，减少初始导入开销
 from lazy_loader import attach
@@ -45,4 +52,16 @@ __getattr__, __dir__, __all__ = attach(
     __name__,
     submodules=["audio_aug", "data_analyze", "filter", "metrics", "utils", "vad", ],
     # 只导出子模块，不直接导出函数
+    submod_attrs={
+        "utils": ["get_path_list"],
+        "filter": ["HPFilter"],
+        "audio_aug": ["volume_norm"],
+    }
 )
+
+if TYPE_CHECKING:
+    __all__ = [
+        "get_path_list",
+        "HPFilter",
+        "volume_norm",
+    ]

neverlib/audio_aug/__init__.py

@@ -1,11 +1,27 @@
 # This file is auto-generated. Do NOT edit manually.
 # Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .HarmonicDistortion import apply_harmonic_distortion, apply_pedalboard_distortion
+    from .TFMask import FreqMask, TimeMask
+    from .audio_aug import add_reverb, get_snr_use_vad, limiter, measure_loudness, snr_aug_Interpolation, snr_aug_changeClean, snr_aug_changeNoise, snr_aug_vad_Interpolation, snr_diff_changeClean, snr_diff_changeNoise, volume_aug, volume_aug_dbrms, volume_aug_linmax, volume_aug_lufs, volume_convert, volume_norm
+    from .clip_aug import clipping_aug
+    from .coder_aug import aac_aug_save, amr_nb_aug, amr_wb_aug, flac_aug, flac_encode_save, mp3_aug, opus_aug_save, vorbis_aug
+    from .coder_aug2 import apply_codec_distortion, check_codec_available
+    from .loss_packet_aug import simulate_packet_loss_vectorized
+    from .quant_aug import apply_mulaw_quantization, apply_uniform_quantization
+
+# 运行时使用懒加载
 from lazy_loader import attach
 
 __getattr__, __dir__, __all__ = attach(
     __name__,
     submodules=[
         "HarmonicDistortion",
+        "TFDrop",
         "TFMask",
         "audio_aug",
         "clip_aug",
@@ -17,10 +33,7 @@ __getattr__, __dir__, __all__ = attach(
     submod_attrs={
         "HarmonicDistortion": ['apply_harmonic_distortion', 'apply_pedalboard_distortion'],
         "TFMask": ['FreqMask', 'TimeMask'],
-        "audio_aug": ['add_reverb', 'get_snr_use_vad', 'limiter', 'measure_loudness', 'snr_aug_Interpolation',
-                      'snr_aug_changeClean', 'snr_aug_changeNoise', 'snr_aug_vad_Interpolation',
-                      'snr_diff_changeClean', 'snr_diff_changeNoise', 'volume_aug', 'volume_aug_dbrms',
-                      'volume_aug_linmax', 'volume_aug_lufs', 'volume_convert', 'volume_norm'],
+        "audio_aug": ['add_reverb', 'get_snr_use_vad', 'limiter', 'measure_loudness', 'snr_aug_Interpolation', 'snr_aug_changeClean', 'snr_aug_changeNoise', 'snr_aug_vad_Interpolation', 'snr_diff_changeClean', 'snr_diff_changeNoise', 'volume_aug', 'volume_aug_dbrms', 'volume_aug_linmax', 'volume_aug_lufs', 'volume_convert', 'volume_norm'],
         "clip_aug": ['clipping_aug'],
         "coder_aug": ['aac_aug_save', 'amr_nb_aug', 'amr_wb_aug', 'flac_aug', 'flac_encode_save', 'mp3_aug', 'opus_aug_save', 'vorbis_aug'],
         "coder_aug2": ['apply_codec_distortion', 'check_codec_available'],
@@ -28,3 +41,42 @@ __getattr__, __dir__, __all__ = attach(
         "quant_aug": ['apply_mulaw_quantization', 'apply_uniform_quantization'],
     }
 )
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'apply_harmonic_distortion',
+        'apply_pedalboard_distortion',
+        'FreqMask',
+        'TimeMask',
+        'add_reverb',
+        'get_snr_use_vad',
+        'limiter',
+        'measure_loudness',
+        'snr_aug_Interpolation',
+        'snr_aug_changeClean',
+        'snr_aug_changeNoise',
+        'snr_aug_vad_Interpolation',
+        'snr_diff_changeClean',
+        'snr_diff_changeNoise',
+        'volume_aug',
+        'volume_aug_dbrms',
+        'volume_aug_linmax',
+        'volume_aug_lufs',
+        'volume_convert',
+        'volume_norm',
+        'clipping_aug',
+        'aac_aug_save',
+        'amr_nb_aug',
+        'amr_wb_aug',
+        'flac_aug',
+        'flac_encode_save',
+        'mp3_aug',
+        'opus_aug_save',
+        'vorbis_aug',
+        'apply_codec_distortion',
+        'check_codec_available',
+        'simulate_packet_loss_vectorized',
+        'apply_mulaw_quantization',
+        'apply_uniform_quantization',
+    ]

neverlib/data_analyze/__init__.py

@@ -1,5 +1,19 @@
 # This file is auto-generated. Do NOT edit manually.
 # Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .dataset_analyzer import AudioFileInfo, DatasetAnalyzer, analyze_audio_dataset
+    from .quality_metrics import QualityAnalyzer, audio_health_check, comprehensive_quality_assessment
+    from .rms_distrubution import get_rms_vad
+    from .spectral_analysis import SpectralAnalyzer, compute_spectral_features, frequency_domain_stats
+    from .statistics import AudioStatistics, compare_datasets, quick_audio_stats
+    from .temporal_features import dB, dc_offset, max_rms_amplitude, mean_rms_amplitude, min_rms_amplitude, peak_amplitude, rms_amplitude, short_time_energy, zero_crossing_rate
+    from .visualization import AudioVisualizer, create_analysis_dashboard, plot_dataset_overview
+
+# 运行时使用懒加载
 from lazy_loader import attach
 
 __getattr__, __dir__, __all__ = attach(
@@ -23,3 +37,33 @@ __getattr__, __dir__, __all__ = attach(
         "visualization": ['AudioVisualizer', 'create_analysis_dashboard', 'plot_dataset_overview'],
     }
 )
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'AudioFileInfo',
+        'DatasetAnalyzer',
+        'analyze_audio_dataset',
+        'QualityAnalyzer',
+        'audio_health_check',
+        'comprehensive_quality_assessment',
+        'get_rms_vad',
+        'SpectralAnalyzer',
+        'compute_spectral_features',
+        'frequency_domain_stats',
+        'AudioStatistics',
+        'compare_datasets',
+        'quick_audio_stats',
+        'dB',
+        'dc_offset',
+        'max_rms_amplitude',
+        'mean_rms_amplitude',
+        'min_rms_amplitude',
+        'peak_amplitude',
+        'rms_amplitude',
+        'short_time_energy',
+        'zero_crossing_rate',
+        'AudioVisualizer',
+        'create_analysis_dashboard',
+        'plot_dataset_overview',
+    ]

neverlib/filter/__init__.py

@@ -1,5 +1,15 @@
 # This file is auto-generated. Do NOT edit manually.
 # Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .biquad import BiquadFilter
+    from .common import HPFilter, HPFilter_torch, LPFilter
+    from .core import EQFilter, EQ_test, eq_process, eq_process_test
+
+# 运行时使用懒加载
 from lazy_loader import attach
 
 __getattr__, __dir__, __all__ = attach(
@@ -15,3 +25,16 @@ __getattr__, __dir__, __all__ = attach(
         "core": ['EQFilter', 'EQ_test', 'eq_process', 'eq_process_test'],
     }
 )
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'BiquadFilter',
+        'HPFilter',
+        'HPFilter_torch',
+        'LPFilter',
+        'EQFilter',
+        'EQ_test',
+        'eq_process',
+        'eq_process_test',
+    ]

neverlib/filter/auto_eq/__init__.py

@@ -1,5 +1,16 @@
 # This file is auto-generated. Do NOT edit manually.
 # Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .de_eq import get_filter_function, match_frequency_response, plot_spectra_comparison
+    from .freq_eq import get_freq_eq
+    from .ga_eq_advanced import EQConfig, EQOptimizer, load_config_from_yaml, main
+    from .ga_eq_basic import custom_mutate, evaluate_individual, generate_active_gene, generate_dbgain_gene, generate_fc_gene, generate_q_gene, generate_type_gene, get_combined_eq_response_db, get_magnitude_spectrum_db, get_single_filter_freq_response_db_from_coeffs, individual_creator, main_ga
+
+# 运行时使用懒加载
 from lazy_loader import attach
 
 __getattr__, __dir__, __all__ = attach(
@@ -17,3 +28,28 @@ __getattr__, __dir__, __all__ = attach(
         "ga_eq_basic": ['custom_mutate', 'evaluate_individual', 'generate_active_gene', 'generate_dbgain_gene', 'generate_fc_gene', 'generate_q_gene', 'generate_type_gene', 'get_combined_eq_response_db', 'get_magnitude_spectrum_db', 'get_single_filter_freq_response_db_from_coeffs', 'individual_creator', 'main_ga'],
     }
 )
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'get_filter_function',
+        'match_frequency_response',
+        'plot_spectra_comparison',
+        'get_freq_eq',
+        'EQConfig',
+        'EQOptimizer',
+        'load_config_from_yaml',
+        'main',
+        'custom_mutate',
+        'evaluate_individual',
+        'generate_active_gene',
+        'generate_dbgain_gene',
+        'generate_fc_gene',
+        'generate_q_gene',
+        'generate_type_gene',
+        'get_combined_eq_response_db',
+        'get_magnitude_spectrum_db',
+        'get_single_filter_freq_response_db_from_coeffs',
+        'individual_creator',
+        'main_ga',
+    ]

neverlib/metrics/__init__.py

@@ -1,5 +1,18 @@
 # This file is auto-generated. Do NOT edit manually.
 # Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .dnsmos import ComputeScore
+    from .lpc_lsp import framing, lpc_lsp_distance, lpc_to_lsp
+    from .snr import get_snr, get_snr_from_noisy, psnr, seg_snr, si_sdr
+    from .spec import lsd, mcd, sd
+    from .test_pesq import mos2pesq, pesq2mos
+    from .time import dc_offset, mean_rms_amplitude, peak_amplitude, rms_amplitude
+
+# 运行时使用懒加载
 from lazy_loader import attach
 
 __getattr__, __dir__, __all__ = attach(
@@ -13,7 +26,7 @@ __getattr__, __dir__, __all__ = attach(
         "time",
     ],
     submod_attrs={
-        "dnsmos": ['ComputeScore', 'main'],
+        "dnsmos": ['ComputeScore'],
         "lpc_lsp": ['framing', 'lpc_lsp_distance', 'lpc_to_lsp'],
         "snr": ['get_snr', 'get_snr_from_noisy', 'psnr', 'seg_snr', 'si_sdr'],
         "spec": ['lsd', 'mcd', 'sd'],
@@ -21,3 +34,26 @@ __getattr__, __dir__, __all__ = attach(
         "time": ['dc_offset', 'mean_rms_amplitude', 'peak_amplitude', 'rms_amplitude'],
     }
 )
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'ComputeScore',
+        'framing',
+        'lpc_lsp_distance',
+        'lpc_to_lsp',
+        'get_snr',
+        'get_snr_from_noisy',
+        'psnr',
+        'seg_snr',
+        'si_sdr',
+        'lsd',
+        'mcd',
+        'sd',
+        'mos2pesq',
+        'pesq2mos',
+        'dc_offset',
+        'mean_rms_amplitude',
+        'peak_amplitude',
+        'rms_amplitude',
+    ]

neverlib/tests/__init__.py

@@ -1 +1,33 @@
-# 测试包初始化文件 
+# This file is auto-generated. Do NOT edit manually.
+# Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .test_preprocess import test_NS_shape, test_pre_emphasis
+    from .test_vad import test_vad2nad
+
+# 运行时使用懒加载
+from lazy_loader import attach
+
+__getattr__, __dir__, __all__ = attach(
+    __name__,
+    submodules=[
+        "test_imports",
+        "test_preprocess",
+        "test_vad",
+    ],
+    submod_attrs={
+        "test_preprocess": ['test_NS_shape', 'test_pre_emphasis'],
+        "test_vad": ['test_vad2nad'],
+    }
+)
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'test_NS_shape',
+        'test_pre_emphasis',
+        'test_vad2nad',
+    ]

neverlib/tests/test_imports.py

@@ -8,6 +8,8 @@ Description: 测试neverlib导入功能
 import sys
 import os
 import time
+from neverlib.utils import get_path_list
+from neverlib.data_analyze.dataset_analyzer import AudioFileInfo
 
 # 确保当前目录在Python路径中，以便导入neverlib
 # sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

neverlib/utils/__init__.py

@@ -1,5 +1,19 @@
 # This file is auto-generated. Do NOT edit manually.
 # Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .audio_split import audio_split_VADfunasr, audio_split_VADsilero, audio_split_ffmpeg, audio_split_np, audio_split_pydub, audio_split_random, audio_split_sox
+    from .checkGPU import get_gpu_utilization, is_gpu_idle, monitor_gpu_utilization
+    from .lazy_expose import attach_and_expose_all
+    from .lazy_module import LazyModule
+    from .message import send_QQEmail, send_QQEmail_with_images
+    from .utils import DatasetSubfloderSplit, TrainValSplit, TrainValTestSplit, del_empty_folders, get_audio_segments, get_file_time, \
+        get_leaf_folders, get_path_list, pcm2wav, read_pcm, rename_files_and_folders, save_weight_histogram, wav2pcm, EPS
+
+# 运行时使用懒加载
 from lazy_loader import attach
 
 __getattr__, __dir__, __all__ = attach(
@@ -13,14 +27,45 @@ __getattr__, __dir__, __all__ = attach(
         "utils",
     ],
     submod_attrs={
-        "audio_split": ['audio_split_VADfunasr', 'audio_split_VADsilero', 'audio_split_ffmpeg',
-                        'audio_split_np', 'audio_split_pydub', 'audio_split_random', 'audio_split_sox'],
+        "audio_split": ['audio_split_VADfunasr', 'audio_split_VADsilero', 'audio_split_ffmpeg', 'audio_split_np', 'audio_split_pydub', 'audio_split_random', 'audio_split_sox'],
         "checkGPU": ['get_gpu_utilization', 'is_gpu_idle', 'monitor_gpu_utilization'],
         "lazy_expose": ['attach_and_expose_all'],
         "lazy_module": ['LazyModule'],
         "message": ['send_QQEmail', 'send_QQEmail_with_images'],
-        "utils": ['EPS', 'DatasetSubfloderSplit', 'TrainValSplit', 'TrainValTestSplit', 'del_empty_folders',
-                  'get_audio_segments', 'get_file_time', 'get_leaf_folders', 'get_path_list', 'pcm2wav',
-                  'rename_files_and_folders', 'save_weight_histogram', 'wav2pcm'],
+        "utils": ['DatasetSubfloderSplit', 'TrainValSplit', 'TrainValTestSplit', 'del_empty_folders', 'get_audio_segments', 'get_file_time', 'get_leaf_folders',
+                  'get_path_list', 'pcm2wav', 'read_pcm', 'rename_files_and_folders', 'save_weight_histogram', 'wav2pcm', "EPS"],
     }
 )
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'audio_split_VADfunasr',
+        'audio_split_VADsilero',
+        'audio_split_ffmpeg',
+        'audio_split_np',
+        'audio_split_pydub',
+        'audio_split_random',
+        'audio_split_sox',
+        'get_gpu_utilization',
+        'is_gpu_idle',
+        'monitor_gpu_utilization',
+        'attach_and_expose_all',
+        'LazyModule',
+        'send_QQEmail',
+        'send_QQEmail_with_images',
+        'DatasetSubfloderSplit',
+        'TrainValSplit',
+        'TrainValTestSplit',
+        'del_empty_folders',
+        'get_audio_segments',
+        'get_file_time',
+        'get_leaf_folders',
+        'get_path_list',
+        'pcm2wav',
+        'read_pcm',
+        'rename_files_and_folders',
+        'save_weight_histogram',
+        'wav2pcm',
+        'EPS',
+    ]

neverlib/utils/audio_split.py

@@ -53,6 +53,7 @@ def audio_split_sox(source_path, target_path, duration, endwith="*.wav"):
     :param duration: 分割为时长(短音频)(单位:秒)
     :param endwith: 音频格式(只支持wav)
     """
+    assert endwith == "*.wav", "只支持wav格式的音频"
     wav_path_list = get_path_list(source_path, end=endwith)
 
     for wav_path in wav_path_list:

neverlib/utils/utils.py

@@ -385,3 +385,18 @@ def save_weight_histogram(model,
             plt.hist(buffer, bins=bins)
             plt.savefig(os.path.join(save_dir, "buffer", f"{name}.png"))
             plt.close()
+
+
+def read_pcm(file_path, channels=5, sample_rate=16000, sample_width=2):
+    # Read raw binary data from the PCM file
+    with open(file_path, 'rb') as f:
+        raw_data = f.read()
+
+    # Convert binary data to numpy array of the correct dtype
+    audio_data = np.frombuffer(raw_data, dtype=np.int16)
+
+    # Reshape the data into a 2D array with shape (num_frames, channels)
+    num_frames = len(audio_data) // channels
+    audio_data = audio_data.reshape((num_frames, channels))
+
+    return audio_data

neverlib/vad/PreProcess.py

@@ -1,66 +1,66 @@
-'''
-Author: 凌逆战 | Never
-Date: 2025-02-13 20:06:07
-LastEditTime: 2025-08-16 02:07:24
-FilePath: \neverlib\vad\PreProcess.py
-Description: 
-'''
-# -*- coding:utf-8 -*-
-# Author:凌逆战 | Never
-# Date: 2024/9/14
-"""
-通过一些预处理方法, 来提高VAD的准确率
-"""
-import numpy as np
-import noisereduce as nr
-
-
-def pre_emphasis(audio_data, alpha=0.97):
-    """
-    预加重
-    """
-    # y(n)=x(n)−α⋅x(n−1)
-    emphasized_audio = np.append(audio_data[0], audio_data[1:] - alpha * audio_data[:-1])
-    return emphasized_audio
-
-
-def NS(wav, sr=16000, stationary=True, prop_decrease=1.):
-    """ 传统降噪 Doc: https://pypi.org/project/noisereduce/
-    :param wav: (xxx,) or (channels, xxx)
-    :param sr: 采样率
-    :param stationary: 平稳降噪还是非平稳降噪
-    :param prop_decrease: 0~1, 降噪噪声百分比
-    :return:
-    """
-    if stationary:
-        # 平稳噪声抑制 stationary=True
-        reduced_noise = nr.reduce_noise(y=wav, sr=sr, stationary=True,
-                                        prop_decrease=prop_decrease,  # 降噪噪声的比例
-                                        )
-    else:
-        # 非平稳噪声抑制 stationary=False
-        reduced_noise = nr.reduce_noise(y=wav, sr=sr, stationary=False,
-                                        prop_decrease=prop_decrease,
-                                        )
-    return reduced_noise
-
-
-def NS_test():
-    import soundfile as sf
-    sr = 16000
-    wav_path = "../../data/vad_example.wav"
-    wav, wav_sr = sf.read(wav_path, always_2d=False, dtype="float32")
-    wav_NS = NS(wav, sr=sr, stationary=True, prop_decrease=0.6)
-    sf.write("../../wav_data/000_short_NS.wav", wav_NS, samplerate=sr)
-
-    # 绘制降噪后的频谱图
-    import matplotlib.pyplot as plt
-    plt.subplot(211)
-    plt.specgram(wav, Fs=sr, scale_by_freq=True, sides='default', cmap="jet")
-    plt.subplot(212)
-    plt.specgram(wav_NS, Fs=sr, scale_by_freq=True, sides='default', cmap="jet")
-    plt.show()
-
-
-if __name__ == "__main__":
-    NS_test()
+'''
+Author: 凌逆战 | Never
+Date: 2025-02-13 20:06:07
+LastEditTime: 2025-08-16 02:07:24
+FilePath: \\neverlib\\vad\\PreProcess.py
+Description: 
+'''
+# -*- coding:utf-8 -*-
+# Author:凌逆战 | Never
+# Date: 2024/9/14
+"""
+通过一些预处理方法, 来提高VAD的准确率
+"""
+import numpy as np
+import noisereduce as nr
+
+
+def pre_emphasis(audio_data, alpha=0.97):
+    """
+    预加重
+    """
+    # y(n)=x(n)−α⋅x(n−1)
+    emphasized_audio = np.append(audio_data[0], audio_data[1:] - alpha * audio_data[:-1])
+    return emphasized_audio
+
+
+def NS(wav, sr=16000, stationary=True, prop_decrease=1.):
+    """ 传统降噪 Doc: https://pypi.org/project/noisereduce/
+    :param wav: (xxx,) or (channels, xxx)
+    :param sr: 采样率
+    :param stationary: 平稳降噪还是非平稳降噪
+    :param prop_decrease: 0~1, 降噪噪声百分比
+    :return:
+    """
+    if stationary:
+        # 平稳噪声抑制 stationary=True
+        reduced_noise = nr.reduce_noise(y=wav, sr=sr, stationary=True,
+                                        prop_decrease=prop_decrease,  # 降噪噪声的比例
+                                        )
+    else:
+        # 非平稳噪声抑制 stationary=False
+        reduced_noise = nr.reduce_noise(y=wav, sr=sr, stationary=False,
+                                        prop_decrease=prop_decrease,
+                                        )
+    return reduced_noise
+
+
+def NS_test():
+    import soundfile as sf
+    sr = 16000
+    wav_path = "../../data/vad_example.wav"
+    wav, wav_sr = sf.read(wav_path, always_2d=False, dtype="float32")
+    wav_NS = NS(wav, sr=sr, stationary=True, prop_decrease=0.6)
+    sf.write("../../wav_data/000_short_NS.wav", wav_NS, samplerate=sr)
+
+    # 绘制降噪后的频谱图
+    import matplotlib.pyplot as plt
+    plt.subplot(211)
+    plt.specgram(wav, Fs=sr, scale_by_freq=True, sides='default', cmap="jet")
+    plt.subplot(212)
+    plt.specgram(wav_NS, Fs=sr, scale_by_freq=True, sides='default', cmap="jet")
+    plt.show()
+
+
+if __name__ == "__main__":
+    NS_test()

neverlib/vad/__init__.py

@@ -1,5 +1,23 @@
 # This file is auto-generated. Do NOT edit manually.
 # Generated by neverlib.QA.gen_init
+
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    # 仅在类型检查时导入，提供IDE补全支持
+    from .PreProcess import NS, NS_test, pre_emphasis
+    from .VAD_Energy import EnergyVad_C
+    from .VAD_Silero import Silero_VAD_C
+    from .VAD_WebRTC import WebRTC_VAD_C
+    from .VAD_funasr import FunASR_VAD_C
+    from .VAD_statistics import Statistics_VAD
+    from .VAD_vadlib import Vadlib_C
+    from .VAD_whisper import Whisper_VAD_C
+    from .class_get_speech import getSpeech
+    from .class_vad import VADClass
+    from .utils import from_vadArray_to_vadEndpoint, vad2nad, vad_smooth
+
+# 运行时使用懒加载
 from lazy_loader import attach
 
 __getattr__, __dir__, __all__ = attach(
@@ -31,3 +49,23 @@ __getattr__, __dir__, __all__ = attach(
         "utils": ['from_vadArray_to_vadEndpoint', 'vad2nad', 'vad_smooth'],
     }
 )
+
+# 显式声明 __all__ 以便 IDE 识别
+if TYPE_CHECKING:
+    __all__ = [
+        'NS',
+        'NS_test',
+        'pre_emphasis',
+        'EnergyVad_C',
+        'Silero_VAD_C',
+        'WebRTC_VAD_C',
+        'FunASR_VAD_C',
+        'Statistics_VAD',
+        'Vadlib_C',
+        'Whisper_VAD_C',
+        'getSpeech',
+        'VADClass',
+        'from_vadArray_to_vadEndpoint',
+        'vad2nad',
+        'vad_smooth',
+    ]

{neverlib-0.2.7.dist-info → neverlib-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: neverlib
-Version: 0.2.7
+Version: 0.2.8
 Summary: A successful sign for python setup
 Author-email: "Never.Ling" <1786088386@qq.com>
 License: MIT

{neverlib-0.2.7.dist-info → neverlib-0.2.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-neverlib/__init__.py,sha256=kqm4-34anitzcQfir4sU9rMLeICDcl9TKEY3gNoFLqg,1622
+neverlib/__init__.py,sha256=Kq_ZWgw8gFiqQIN2gPzXkgti_zXUi2DYx0CD-88ExZU,2048
 neverlib/.claude/settings.local.json,sha256=nBol-eXbiZ3FMr1a7e6ls6CO9nwz5f3quPjEQzMkTyQ,126
 neverlib/.history/__init___20250805234212.py,sha256=RuVTeJiZfStZ12eMlEfTZ8ZjiC5RBaIQFht-bzMWdTs,1158
 neverlib/.history/__init___20250904102635.py,sha256=7G8jWnR8puUua7yvH63_AKM6rWGq0IvwgrincgrEs58,1081
@@ -417,13 +417,13 @@ neverlib/Docs/vad/VAD_statistics.ipynb,sha256=wT0k_91YUpLLIt6LZMKrjvJLk4kz8g-fZd
 neverlib/Docs/vad/VAD_tenVAD.ipynb,sha256=w8oOBVeabW44Fw7TF6CSxIqa3u1gGydoMuWSxE7slaA,351893
 neverlib/Docs/vad/VAD_vadlib.ipynb,sha256=m55oKqOGeMOSgmQlbMkOR6zVh5_Hn6SIocHi6mOtg1k,300998
 neverlib/Docs/vad/VAD_whisper.ipynb,sha256=P78NqxCfwbbEJYQsqA1NV8vpxQxWNxy2lDX85zO_yD4,364207
-neverlib/QA/gen_init.py,sha256=KJz5Thqo3rUDIVTPLmRN_iCtn1w_bMstSJHQo8DDqzc,4369
+neverlib/QA/gen_init.py,sha256=9QysSB2vk1OR9FmuuKcFtShtiJeCTnvvg6RBlxOUcQU,8127
 neverlib/QA/get_fun.py,sha256=-UP2FOm3jB_138ntpbsW0eG1S0tJ0d90NqI6f95ryYg,541
 neverlib/QA/html2markdown.py,sha256=ZiXvpwcJ_QBhIR6aetaMOmktWHtd_F7Ed07Deo3qM4c,669
 neverlib/audio_aug/HarmonicDistortion.py,sha256=MqbfcdWJLtl_WGloQYDa6fIB6QCgukYdjoxsj8X03kk,2706
 neverlib/audio_aug/TFDrop.py,sha256=HDLf5qfqMT73-dxKder0v0AT7b0FYsSlexbbh3sL2iM,987
 neverlib/audio_aug/TFMask.py,sha256=Fakx5YTj88bZEdj_w3aH9Z9AhEGzGq01TZcnjF4v4BU,1769
-neverlib/audio_aug/__init__.py,sha256=sPtnnOxjo9QyEX3bMaP_IoJEFI5vRSWd5fN-GXfluR8,1349
+neverlib/audio_aug/__init__.py,sha256=5mBg5p8aDHvOPlRSmR-ojGhtxaJTFCmCZUltaj2RPAs,3313
 neverlib/audio_aug/audio_aug.py,sha256=5Eqtx_v2RLtNEwh1gTyfqSHBirwz1JCgDrjgt4WYBlY,16674
 neverlib/audio_aug/clip_aug.py,sha256=xZi89AtdOgUpX7W4x4_UiZFeK-oCWktueoWSEAY19hU,1041
 neverlib/audio_aug/coder_aug.py,sha256=1JtNadfUVbOw8Xg3vpp0cTl87HHuDedHZ3zbUZ2fWW0,8347
@@ -438,7 +438,7 @@ neverlib/data/white.wav,sha256=JJ3aaPx8v7WakYU7U--m0kO1qUkDSMOrOrPKi6RqAA0,32508
 neverlib/data/white_EQ.wav,sha256=UN1W0hH_4JInnye9vIsQlagVXM-rnJPd1j7A_RIia8c,325086
 neverlib/data/white_matched.wav,sha256=h9vD-2q1eKw9s8pVUoIeXXjeKEC-VRiNABCY-CYcUQ0,115514
 neverlib/data_analyze/README.md,sha256=jqPmD--kinoTqlcpu51teospu9p_ijjtjLdFGTMnBc8,6051
-neverlib/data_analyze/__init__.py,sha256=PTobY-7N40PzfZiZ5-2JaU_VUVktb5xw6HgD0s-eXEg,1138
+neverlib/data_analyze/__init__.py,sha256=3Gl_o2s9OYmy1r7UZLxKZGwwOXJYuQ__F1RM578lvAI,2781
 neverlib/data_analyze/dataset_analyzer.py,sha256=JawpMrahP2YD9mX6kIGJrBRtC7sqYJJdY1R5pGBUDDY,21902
 neverlib/data_analyze/quality_metrics.py,sha256=wOivyQ4JXdyiq1STLCSdBDtoW8Kju-Zc1RZMyPNmNjY,11726
 neverlib/data_analyze/rms_distrubution.py,sha256=EpK9s7WJLtJHXd1xD8SvuiyPVTguouYVHyS7Lha2Qag,1526
@@ -447,17 +447,17 @@ neverlib/data_analyze/statistics.py,sha256=9b9S_7eBoklNfz9cRAs8oWlRnDSjuark4SwTr
 neverlib/data_analyze/temporal_features.py,sha256=qrxA0Uh-DxRPbuILQle66AAgv4SgwiLoPTIR9Q2_Jz4,3065
 neverlib/data_analyze/visualization.py,sha256=kaJBfh1ixlGeVfr4yNRfZlzTvfhbdnIW4LOhbGqwYKM,16439
 neverlib/filter/README.md,sha256=X1VQRnSxzjkeR_16ylbAqRbYLAlf4c5stkHzmWlCzlo,3801
-neverlib/filter/__init__.py,sha256=7787a3wwVkSdeuDEuEeU6Q1CLg1wp6r8lPqURkpV_zY,450
+neverlib/filter/__init__.py,sha256=oQJLBqxGpC1T-DmHbcSceY9Ir31VXqknzoRNQRAGChI,1019
 neverlib/filter/biquad.py,sha256=EEGAqGwTQGG5O49Q855hb475WVX0oOu3zq-Gn_ppIto,1438
 neverlib/filter/common.py,sha256=9FD3VKwRpe_IdQujt2AJjlMJ_WR-YNEY8OmhOQUDDO8,1953
 neverlib/filter/core.py,sha256=qr-0IGEEpiSUFgo_uVOXJLLZeaSDFlALzg_b8mEivrY,13091
 neverlib/filter/auto_eq/README.md,sha256=-qcqLxIQPo31YfzABHK-_tcqksLdOTHz8ZG3izz46Ug,4968
-neverlib/filter/auto_eq/__init__.py,sha256=LQkDdkHYbwusic8xkP1a6MKswm96INlxYgkizqvmsBY,849
+neverlib/filter/auto_eq/__init__.py,sha256=RWgAXvBWBDboBxzlPBQJWwQLPctniIAZBXZRQAC_QFw,2190
 neverlib/filter/auto_eq/de_eq.py,sha256=XlbFWDPOdGHI97fbU9wXd4h3Aiosv1T5k_e-VEUcDMQ,16411
 neverlib/filter/auto_eq/freq_eq.py,sha256=1b5hnoqUy1zxVd2WSNqhpokAgJo0xTRwR5AyVPaMr3E,3494
 neverlib/filter/auto_eq/ga_eq_advanced.py,sha256=qu9J5ugWeEAcXboV9Fv1g60bMOkIebw9ZwYT_zQq5pI,23458
 neverlib/filter/auto_eq/ga_eq_basic.py,sha256=gbPqJcimrThzhCjgWkg4DuWpeswRRqxNKMC4I07rp6g,15431
-neverlib/metrics/__init__.py,sha256=HAnkorpoxDQjxT5G1SpvPkkYmoGDkzoqF7EsqeG6ITg,693
+neverlib/metrics/__init__.py,sha256=NCE7SOoA2UXu4aVmiF0oWvvdS9RaqR9tSY7PZjzDNew,1628
 neverlib/metrics/dnsmos.py,sha256=xfrKxYR9mMDkr2HQ7AT_7V-TtdRpy2MybjsJ2AuLXus,4590
 neverlib/metrics/lpc_lsp.py,sha256=OXKfillGKo55M2aa7LzyvDGcjtOxdOJquWgqC1yV4Ho,3710
 neverlib/metrics/snr.py,sha256=l8cYOUuEYeKc5LzQ3q5n4-wql3PcuJfEYcsjBAEQt5w,6360
@@ -478,18 +478,18 @@ neverlib/metrics/pesq_c/pesqio.c,sha256=S8PLwBfx0oHHVLfZ1jUsGbwwMniFhHNNV8X6bcUx
 neverlib/metrics/pesq_c/pesqmain.c,sha256=NaANjEDJrCFwJBbXErX51ZKtE86SHKdrXLhS0P3TbT0,27322
 neverlib/metrics/pesq_c/pesqmod.c,sha256=9stYhCLUShFAd6Zc85nQn57RUKDbS2Q8MrhQFB_-ySA,52999
 neverlib/metrics/pesq_c/pesqpar.h,sha256=XZkwXcwycL8ekK6LQy_PT9Ih2K0zirSPHCMhGKyJExs,20986
-neverlib/tests/__init__.py,sha256=PITL4btOVji1rexFP35A20v2gcLqc6vmGWiF-_5ywnI,27
-neverlib/tests/test_imports.py,sha256=aEEjlz54XtKk-N7krFGvIFz_0j-H90qKKx4nPoZfbT8,439
+neverlib/tests/__init__.py,sha256=kCV9FZWa7j9g_i6bqb-hEq7pvmWmDz2EMnGX15n7YmE,811
+neverlib/tests/test_imports.py,sha256=3LGZaHZLAz2oJq2ewfRbnWNeAKikHih6d9CIdyqhb0E,547
 neverlib/tests/test_preprocess.py,sha256=GHj2PvlG8chJlV-zpRfy8IYlTXzGkn5heO9ipy6Xe20,1003
 neverlib/tests/test_vad.py,sha256=TjBuB_qPooflrjkkvoy19HySdBo-MiVeb5yTmLGBJvM,415
-neverlib/utils/__init__.py,sha256=1WO9Q_X9fUWUF_ZaK_ALQNzHpWqletxIJfLoI7FOyCQ,1090
-neverlib/utils/audio_split.py,sha256=8ehyecES9Ih7f85ZyBiOAycrduCQOzaHl1yifsCMjDM,12306
+neverlib/utils/__init__.py,sha256=A969zG3BBrSVtoZFsm60PolCaH0BI0PflUUubqfVT8M,2727
+neverlib/utils/audio_split.py,sha256=tDWw36FXMIVOZ08PC9KQTYB3Zz_doFcP1ugYdpLkQjQ,12368
 neverlib/utils/checkGPU.py,sha256=OkztJZJxncqSSIttxISpLSgfqa9xjXOaHIuMlRNwX-4,4393
 neverlib/utils/lazy_expose.py,sha256=3h1LidRoab_ylv574-dOBjnJpOKWgLFgXpTWrvqce9E,1027
 neverlib/utils/lazy_module.py,sha256=JdEdAYnbHmolB-hktrcuOlzR4Ee5jgu6V0XjDNqioVM,3995
 neverlib/utils/message.py,sha256=XV7m0yUsR7Zq-olS-vL1oUEatSgh-Am8pyB_QPLzQok,4121
-neverlib/utils/utils.py,sha256=7aPSwrPvkVAPARb5brGikV5C4DVlS71fujTrpq8DnjI,14658
-neverlib/vad/PreProcess.py,sha256=syd-s4TiQeqvLeQqQRzTzSqeNuMdWnZEIywX24H4Vro,2108
+neverlib/utils/utils.py,sha256=rsKSOuo__ISdmTe1AQhNqFD8ENHl8A60aDEoGUTiBlE,15179
+neverlib/vad/PreProcess.py,sha256=oib4iRfsUXp5U67pKzJF8k53rqjo8NopK_9LW4O8Mqk,2045
 neverlib/vad/README.md,sha256=3aFnGmz3uhg-kQVOGei40Ei3Lx5P71r2rk0Xl8-JZXI,1490
 neverlib/vad/VAD_Energy.py,sha256=ErsiDxn4mH_EOiFBRcoBBRqTUsFfEsvEvRELrdZflkw,2135
 neverlib/vad/VAD_Silero.py,sha256=L9_eO94VQcO3TBdOWypJxnjMl5ZZBYTkL5DR_6APfs4,2324
@@ -498,13 +498,13 @@ neverlib/vad/VAD_funasr.py,sha256=oJqU3VOFT-KXmFV2g8kvT64nHIo5-PxcB4tdb-Ap3Bw,15
 neverlib/vad/VAD_statistics.py,sha256=o310LKoMECBHO6LcqU3vb-uirF6PM2fHbaRautT2w-w,16663
 neverlib/vad/VAD_vadlib.py,sha256=5reZ7blu0BsqqO9WJn8YPIvPQVn8VcFPcJnH1scgEzQ,2228
 neverlib/vad/VAD_whisper.py,sha256=GhBFV-EIS865c-rj2RvM4IUKKG8ZsoD9LCTwrx1Fx2c,1792
-neverlib/vad/__init__.py,sha256=MgQXDgeqFZN9MqLrYn0hS9Dz5OqY0nTjHW5fQoZ5LGk,967
+neverlib/vad/__init__.py,sha256=UflLGnBIUyqIoIuGLThF4vHeKD6pPRMy3mRyXMiUCAk,2035
 neverlib/vad/class_get_speech.py,sha256=s1n24I4P4aTFaQSiyXrh_xAmOChuFZ-wWgrYEE6eCHs,2074
 neverlib/vad/class_vad.py,sha256=ao_20-vyzcI8CxEcvcA_-HoRPTiPEtfuKor2_YWEbz0,5212
 neverlib/vad/img.png,sha256=eLlaUgDnJNyleWZwW8sNv3eXaeaGoXwhFUZCCDJh9u8,179778
 neverlib/vad/utils.py,sha256=rEWEzGv2E_iTHIdQxY361iEW4Hq9MRDwkMBQMTA9mGw,3049
-neverlib-0.2.7.dist-info/licenses/LICENSE,sha256=h-U7vhFdQhieyaNCiA_utrRNMk1rjfrBAUJVodgkfsw,1096
-neverlib-0.2.7.dist-info/METADATA,sha256=d3h2QWvLLjEILDa65gPYVj1PNAcI0grCTfKn6WU4j9E,2811
-neverlib-0.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-neverlib-0.2.7.dist-info/top_level.txt,sha256=QqwYFuDiY_iFTz0Kx8av6zKCP_s4M5NiMTNsIXOy8Po,9
-neverlib-0.2.7.dist-info/RECORD,,
+neverlib-0.2.8.dist-info/licenses/LICENSE,sha256=h-U7vhFdQhieyaNCiA_utrRNMk1rjfrBAUJVodgkfsw,1096
+neverlib-0.2.8.dist-info/METADATA,sha256=8PeHX6no4CgyTvP8c44NjGA4mclVM_22uKbafKoJO9w,2811
+neverlib-0.2.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+neverlib-0.2.8.dist-info/top_level.txt,sha256=QqwYFuDiY_iFTz0Kx8av6zKCP_s4M5NiMTNsIXOy8Po,9
+neverlib-0.2.8.dist-info/RECORD,,