chippymu 0.1.0__py3-none-any.whl

chippymu/channelgen.py

@@ -0,0 +1,85 @@
+"""
+按通道生成音G
+"""
+
+import numpy as np
+
+from chippymu.utils import basic_wave_gen, basic_drum_gen, note_to_freq
+
+from numpy import ndarray
+from chippymu.configs import BasicParams
+from chippymu.models import DrumType, WaveType
+
+
+def generate_wave(
+    *, melody: list[tuple[float, int, float]], wave_type: WaveType, params: BasicParams
+) -> ndarray:
+    """
+    生成指定波形和旋律的通道
+
+    - 参数：
+        - melody: 旋律列表，每个元素为 (frequency, duration, amplitude)
+        - wave_type: 波形类型
+        - params: 基本参数
+
+    - 返回：
+        - 通道数据
+    """
+    melody_beat: list[tuple[float, int, float]] = [
+        (start * params.beat_duration, note, dur * params.beat_duration)
+        for start, note, dur in melody
+    ]
+
+    melody_audio = np.zeros(params.whole_duration)
+
+    for start, note, dur in melody_beat:
+        wave = basic_wave_gen(
+            wave_type=wave_type,
+            frequency=note_to_freq(note=note),
+            duration=dur,
+            sample_rate=params.sample_rate,
+        )
+        start_idx = int(start * params.sample_rate)
+        end_idx = start_idx + len(wave)
+        melody_audio[start_idx:end_idx] += wave
+
+    return melody_audio
+
+
+def generate_drums(
+    *, drums: list[tuple[float, str, float]], params: BasicParams
+) -> ndarray:
+    """
+    生成鼓声
+
+    - 参数：
+        - drums: 鼓声列表，每个元素是一个元组，包含起始时间，鼓声类型，持续时间
+        - params: 基础参数
+
+    - 返回：
+        - 通道数据
+    """
+    drums_beat: list[tuple[float, str, float]] = [
+        (start * params.beat_duration, note, dur * params.beat_duration)
+        for start, note, dur in drums
+    ]
+
+    drums_audio = np.zeros(params.whole_duration)
+
+    for start, note, dur in drums_beat:
+        try:
+            drum_type = DrumType(note)
+        except ValueError:
+            raise ValueError(f"无效的鼓类型名称: {note}")
+
+        wave = basic_drum_gen(
+            drum_type=drum_type,
+            duration=dur,
+            amplitude=1.0,
+            sample_rate=params.sample_rate,
+        )
+        start_idx = int(start * params.sample_rate)
+        end_idx = start_idx + len(wave)
+        drums_audio[start_idx:end_idx] += wave
+
+    return drums_audio

chippymu/configs.py

@@ -0,0 +1,28 @@
+"""
+基础参数的配置文件
+"""
+
+
+class BasicParams:
+    def __init__(self, *, sample_rate: int = 16000, bpm: int = 90, length: float):
+        # 定义采样率
+        self.sample_rate: int = sample_rate
+        # 定义BPM
+        self.bpm: int = bpm
+        # 定义音频总拍数
+        self.length: float = length
+
+    @property
+    def beat_duration(self) -> float:
+        """
+        计算每拍的时长
+        """
+        return 60.0 / self.bpm
+
+    @property
+    def whole_duration(self) -> int:
+        """
+        计算音频矩阵总长
+        """
+        return int(self.length * self.beat_duration * self.sample_rate)
+        

chippymu/models.py

@@ -0,0 +1,27 @@
+"""
+定义一些基础类型
+"""
+
+from enum import Enum
+
+
+class WaveType(str, Enum):
+    """
+    几种波形
+    """
+
+    SINE = "sine"
+    SQUARE = "square"
+    TRIANGLE = "triangle"
+    SAWTOOTH = "sawtooth"
+    NOISE = "noise"
+
+
+class DrumType(str, Enum):
+    """
+    几种鼓声
+    """
+
+    KICK = "kick"
+    SNARE = "snare"
+    HIHAT = "hihat"

chippymu/sound.py

@@ -0,0 +1,34 @@
+"""
+后处理和音频播放
+"""
+
+import sounddevice as sd # type: ignore
+import numpy as np
+
+from numpy import ndarray
+
+from chippymu.configs import BasicParams
+
+
+def post_processing(channels: list[ndarray], volumes: list[float]) -> ndarray:
+    """
+    后处理过程，包括音量控制、混音、裁剪、量化
+    """
+    if len(channels) != len(volumes):
+        raise ValueError("channels和volumes长度不一致")
+
+    mix = np.zeros_like(channels[0])
+    for i in range(len(channels)):
+        mix += channels[i] * volumes[i]
+
+    mix = np.clip(mix, -1, 1)
+
+    mix_8bit = np.round(mix * 127).astype(np.int8)
+
+    return mix_8bit
+
+
+def play(data: ndarray, params: BasicParams):
+    sd.play(data=data, samplerate=params.sample_rate)
+    sd.wait()
+    

chippymu/utils.py

@@ -0,0 +1,168 @@
+"""
+基础工具
+"""
+
+import numpy as np
+
+from chippymu.models import DrumType, WaveType
+from numpy import ndarray
+
+
+def note_to_freq(*, note: int) -> float:
+    """
+    将 MIDI 音符编号转换为频率 (Hz)
+
+    - 参数
+        - note: MIDI 音符编号
+
+    - 返回
+        - 频率 (Hz)
+    """
+    return 440 * 2 ** ((note - 69) / 12)
+
+
+def basic_wave_gen(
+    *,
+    wave_type: WaveType,
+    frequency: float,
+    duration: float,
+    amplitude: float = 1.0,
+    duty_cycle: float = 0.5,
+    sample_rate: int = 16000,
+) -> ndarray:
+    """
+    生成指定类型的波形数组。
+
+    - 参数:
+        - wave_type: 波形的类型，可以是 "sine"、"square"、"triangle"、"sawtooth" 或 "noise"。
+        - frequency: 波形的频率，单位为赫兹（Hz）。
+        - duration: 波形的持续时间，单位为秒（s）。
+        - amplitude: 波形的振幅，范围从 0 到 1。
+        - duty_cycle:  Square 波形的占空比，范围从 0 到 1。
+        - sample_rate: 采样率，单位为赫兹（Hz），默认为16000，使用时可能从Params中获取。
+
+    - 返回:
+        - 一个包含波形数据的 NumPy 数组。
+    """
+    t: ndarray = np.linspace(
+        start=0, stop=duration, num=int(sample_rate * duration), endpoint=False
+    )
+    phase: ndarray = (t * frequency) % 1
+
+    match wave_type:
+        case WaveType.SINE:
+            return amplitude * np.sin(2 * np.pi * frequency * t)
+        case WaveType.SQUARE:
+            return amplitude * np.where(phase < duty_cycle, 1, -1)
+        case WaveType.TRIANGLE:
+            return amplitude * np.where(phase < 0.5, -1 + 4 * phase, 3 - 4 * phase)
+        case WaveType.SAWTOOTH:
+            return amplitude * (2 * phase - 1)
+        case WaveType.NOISE:
+            return amplitude * (2 * np.random.rand(len(t)) - 1)
+        case _:
+            raise ValueError("未知波形类型")
+
+
+def generate_kick(
+    *,
+    duration: float,
+    amplitude: float = 1.0,
+    sample_rate: int = 16000,
+) -> ndarray:
+    """
+    生成底鼓声音，低频正弦波带衰减。
+
+    - 参数：
+        - duration: 持续时间，单位秒
+        - frequency: 频率，单位赫兹
+        - amplitude: 振幅，范围0~1
+        - sample_rate: 采样率，默认16000
+
+    - 返回：
+        - 音频数据，ndarray
+    """
+    frequency: int = 50
+    t: ndarray = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
+    envelope: ndarray = np.exp(-10 * t)  # 衰减
+    return amplitude * envelope * np.sin(2 * np.pi * frequency * t)
+
+
+def generate_snare(
+    *, duration: float, amplitude: float = 1.0, sample_rate: int = 16000
+) -> ndarray:
+    """
+    生成军鼓声音，噪声带衰减。
+
+    - 参数：
+        - duration: 持续时间，单位秒
+        - amplitude: 振幅，范围0~1
+        - sample_rate: 采样率，默认16000
+
+    - 返回：
+        - 音频数据，ndarray
+    """
+    t: ndarray = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
+    noise: ndarray = 2 * np.random.rand(len(t)) - 1
+    envelope: ndarray = np.exp(-20 * t)
+    return amplitude * envelope * noise
+
+
+def generate_hihat(
+    *, duration: float, amplitude: float = 1.0, sample_rate: int = 16000
+) -> ndarray:
+    """
+    生成踩镲声音，短噪声带快速衰减。
+
+    - 参数：
+        - duration: 持续时间，单位秒
+        - amplitude: 音量
+        - sample_rate: 采样率，默认16000
+
+    - 返回：
+        - 噪声, ndarray
+    """
+    t: ndarray = np.linspace(0, duration, int(sample_rate * duration), endpoint=False)
+    noise: ndarray = 2 * np.random.rand(len(t)) - 1
+    envelope: ndarray = np.exp(-50 * t)
+    return amplitude * envelope * noise
+
+
+def basic_drum_gen(
+    *,
+    drum_type: DrumType,
+    frequency: int = 50,
+    duration: float,
+    amplitude: float = 1.0,
+    sample_rate: int = 16000,
+) -> ndarray:
+    """
+    生成基本鼓声
+
+    - 参数：
+        - drum_type: 鼓声类型
+        - frequency: 频率，单位赫兹
+        - duration: 持续时间，单位秒
+        - amplitude: 振幅，范围0~1
+        - sample_rate: 采样率，单位赫兹
+
+    - 返回：
+        - 音频数据，ndarray
+    """
+    match drum_type:
+        case DrumType.KICK:
+            return generate_kick(
+                duration=duration,
+                amplitude=amplitude,
+                sample_rate=sample_rate,
+            )
+        case DrumType.SNARE:
+            return generate_snare(
+                duration=duration, amplitude=amplitude, sample_rate=sample_rate
+            )
+        case DrumType.HIHAT:
+            return generate_hihat(
+                duration=duration, amplitude=amplitude, sample_rate=sample_rate
+            )
+        case _:
+            raise ValueError("未知鼓类型")

chippymu-0.1.0.dist-info/METADATA

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: chippymu
+Version: 0.1.0
+Summary: 一个用numpy实现的8bit音乐生成器
+Author-email: wolido <270262953@qq.com>
+Requires-Python: >=3.13
+Requires-Dist: numpy>=2.3.0
+Requires-Dist: sounddevice>=0.5.2

chippymu-0.1.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+chippymu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+chippymu/channelgen.py,sha256=56CdyMFMA9ToInqbU46hi6OIpJumOkBNS_nZWDp066I,2314
+chippymu/configs.py,sha256=2lAIvQhW010vJxvnc_8WEgIbdsNe0dZjqQxSWXpM_0E,654
+chippymu/models.py,sha256=3PUjKHONggFbm6A34Z6q_yR3jYn7CTl6HSvr6394ACU,352
+chippymu/sound.py,sha256=sKvYZKxoL8G3hqmBvfqYmtW3qUoFjTh1X7uX_KyJn_k,764
+chippymu/utils.py,sha256=QLk-32XpcnPfFwHocsIn5pXvKdizFcJUVhh7c6AZImc,4799
+chippymu-0.1.0.dist-info/METADATA,sha256=Eg07ehGCHswGrOCLMHOoynxLDOnyUz8999xOwENjaSU,230
+chippymu-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+chippymu-0.1.0.dist-info/entry_points.txt,sha256=Gg86qeAd6CivtSYQqWRqgS9tsClmZNEApTdGov6ssrg,43
+chippymu-0.1.0.dist-info/RECORD,,

chippymu-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

chippymu-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+chippymu = chippymu:main