#!/usr/bin/env python3

import numpy as np
import csv

# --- Configuration ---
fs = 16000          # Sampling frequency in Hz
duration = 0.1      # Duration in seconds
signal_freq = 1000.0 # Frequency of the sine/cosine wave in Hz (adjustable)

# --- Generate Data ---
num_samples = fs * duration
t = np.arange(num_samples) / fs                     # Time vector: 0 to 0.9999375 s
sine_vals = np.sin(2 * np.pi * signal_freq * t)     # Sine wave samples
cosine_vals = np.cos(2 * np.pi * signal_freq * t)   # Cosine wave samples

# Stack columns for easier CSV writing
data = np.column_stack((t, sine_vals, cosine_vals))

# --- Write to CSV ---
output_file = "sine_cosine_16kHz.csv"
with open(output_file, 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(['time', 'sine', 'cosine'])      # Header
    writer.writerows(data)                          # Data rows

print(f"✅ Successfully created '{output_file}' with {len(t)} samples.")