# main_test.py

from module_pulse import generate_pulse_signal

def main():
    print("Генератор импульсного сигнала (вариант 4)\n")

    ampl = float(input("Амплитуда: "))
    pulse_len = int(input("Длительность импульса (отсчёты): "))
    period = int(input("Период (импульс+пауза, отсчёты): "))
    n_samples = int(input("Число отсчётов: "))
    filename = input("Имя файла для записи (например signal.txt): ").strip()

    signal = generate_pulse_signal(ampl, pulse_len, period, n_samples, filename)

    print("\nГотово.")
    print(f"Сигнал записан в файл: {filename}")
    print("Первые 30 отсчётов (или меньше):")
    print(signal[:30])

    # чтобы было видно структуру по периодам:
    print("\nСигнал по строкам (индекс: значение) для первых 30:")
    for i in range(min(30, len(signal))):
        print(i, ":", signal[i])


if __name__ == "__main__":
    main()