From 45958269eaaafc3db56c23b8ac6a6f6b197109e4 Mon Sep 17 00:00:00 2001 From: KubaPro010 Date: Tue, 31 Dec 2024 22:07:07 +0100 Subject: [PATCH] test ssb --- lib/hilbert.c | 24 +++++ lib/hilbert.h | 42 +++++++++ src/ssb_stereo_coder.c | 205 +++++++++++++++++++++++++++++++++++++++++ 3 files changed, 271 insertions(+) create mode 100644 lib/hilbert.c create mode 100644 lib/hilbert.h create mode 100644 src/ssb_stereo_coder.c diff --git a/lib/hilbert.c b/lib/hilbert.c new file mode 100644 index 0000000..b31424e --- /dev/null +++ b/lib/hilbert.c @@ -0,0 +1,24 @@ +#include "hilbert.h" + +void init_hilbert(HilbertTransformer *hilbert) { + hilbert->delay = calloc(D_SIZE, sizeof(float)); + hilbert->dptr = 0; +} + +void apply_hilbert(HilbertTransformer *hilbert, float sample, float *output_0deg, float *output_90deg) { + float hilb; + + hilbert->delay[hilbert->dptr] = sample; + hilb = 0.0f; + for(int i = 0; i < NZEROS/2; i++) { + hilb += (xcoeffs[i] * hilbert->delay[(hilbert->dptr - i*2) & (D_SIZE - 1)]); + } + + *output_0deg = hilbert->delay[(hilbert->dptr - 99) & (D_SIZE - 1)]; + *output_90deg = hilb; + hilbert->dptr = (hilbert->dptr + 1) & (D_SIZE - 1); +} + +void exit_hilbert(HilbertTransformer *hilbert) { + free(hilbert->delay); +} \ No newline at end of file diff --git a/lib/hilbert.h b/lib/hilbert.h new file mode 100644 index 0000000..920fb6d --- /dev/null +++ b/lib/hilbert.h @@ -0,0 +1,42 @@ +#include + +#define D_SIZE 256 +#define NZEROS 200 + +/* The non-zero taps of the Hilbert transformer */ +static float xcoeffs[] = { + +0.0008103736f, +0.0008457886f, +0.0009017196f, +0.0009793364f, + +0.0010798341f, +0.0012044365f, +0.0013544008f, +0.0015310235f, + +0.0017356466f, +0.0019696659f, +0.0022345404f, +0.0025318040f, + +0.0028630784f, +0.0032300896f, +0.0036346867f, +0.0040788644f, + +0.0045647903f, +0.0050948365f, +0.0056716186f, +0.0062980419f, + +0.0069773575f, +0.0077132300f, +0.0085098208f, +0.0093718901f, + +0.0103049226f, +0.0113152847f, +0.0124104218f, +0.0135991079f, + +0.0148917649f, +0.0163008758f, +0.0178415242f, +0.0195321089f, + +0.0213953037f, +0.0234593652f, +0.0257599469f, +0.0283426636f, + +0.0312667947f, +0.0346107648f, +0.0384804823f, +0.0430224431f, + +0.0484451086f, +0.0550553725f, +0.0633242001f, +0.0740128560f, + +0.0884368322f, +0.1090816773f, +0.1412745301f, +0.1988673273f, + +0.3326528346f, +0.9997730178f, -0.9997730178f, -0.3326528346f, + -0.1988673273f, -0.1412745301f, -0.1090816773f, -0.0884368322f, + -0.0740128560f, -0.0633242001f, -0.0550553725f, -0.0484451086f, + -0.0430224431f, -0.0384804823f, -0.0346107648f, -0.0312667947f, + -0.0283426636f, -0.0257599469f, -0.0234593652f, -0.0213953037f, + -0.0195321089f, -0.0178415242f, -0.0163008758f, -0.0148917649f, + -0.0135991079f, -0.0124104218f, -0.0113152847f, -0.0103049226f, + -0.0093718901f, -0.0085098208f, -0.0077132300f, -0.0069773575f, + -0.0062980419f, -0.0056716186f, -0.0050948365f, -0.0045647903f, + -0.0040788644f, -0.0036346867f, -0.0032300896f, -0.0028630784f, + -0.0025318040f, -0.0022345404f, -0.0019696659f, -0.0017356466f, + -0.0015310235f, -0.0013544008f, -0.0012044365f, -0.0010798341f, + -0.0009793364f, -0.0009017196f, -0.0008457886f, -0.0008103736f, +}; + +typedef struct { + float* delay; + int dptr; +} HilbertTransformer; + +void init_hilbert(HilbertTransformer *hilbert); +void apply_hilbert(HilbertTransformer *hilbert, float sample, float *output_0deg, float *output_90deg); +void exit_hilbert(HilbertTransformer *hilbert); \ No newline at end of file diff --git a/src/ssb_stereo_coder.c b/src/ssb_stereo_coder.c new file mode 100644 index 0000000..556c83e --- /dev/null +++ b/src/ssb_stereo_coder.c @@ -0,0 +1,205 @@ +#include +#include +#include +#include +#include +#include +#include + +#include "../lib/constants.h" +#include "../lib/oscillator.h" +#include "../lib/filters.h" +#include "../lib/hilbert.h" + +// Features +#include "features.h" + +#define SAMPLE_RATE 192000 // Don't go lower than 108 KHz, becuase it (53000*2) and (38000+15000) + +#define INPUT_DEVICE "real_real_tx_audio_input.monitor" +#define OUTPUT_DEVICE "alsa_output.platform-soc_sound.stereo-fallback" +#define BUFFER_SIZE 512 +#define CLIPPER_THRESHOLD 0.375 // Adjust this as needed + +#define MONO_VOLUME 0.45f // L+R Signal +#define PILOT_VOLUME 0.0175f // 19 KHz Pilot +#define STEREO_VOLUME 0.35f // L-R signal + +#ifdef PREEMPHASIS +#define PREEMPHASIS_TAU 0.00005 // 50 microseconds, use 0.000075 if in america +#endif + +#ifdef LPF +#define LPF_CUTOFF 15000 +#endif + +volatile sig_atomic_t to_run = 1; + +float clip(float sample) { + if (sample > CLIPPER_THRESHOLD) { + return CLIPPER_THRESHOLD; // Clip to the upper threshold + } else if (sample < -CLIPPER_THRESHOLD) { + return -CLIPPER_THRESHOLD; // Clip to the lower threshold + } else { + return sample; // No clipping + } +} + +void uninterleave(const float *input, float *left, float *right, size_t num_samples) { + // For stereo, usually it is like this: LEFT RIGHT LEFT RIGHT LEFT RIGHT so this is used to get LEFT LEFT LEFT and RIGHT RIGHT RIGHT + for (size_t i = 0; i < num_samples/2; i++) { + left[i] = input[i * 2]; + right[i] = input[i * 2 + 1]; + } +} + +static void stop(int signum) { + (void)signum; + printf("\nReceived stop signal. Cleaning up...\n"); + to_run = 0; +} + +int main() { + printf("STCode : Stereo encoder made by radio95 (with help of ChatGPT and Claude, thanks!)\n"); + // Define formats and buffer atributes + pa_sample_spec stereo_format = { + .format = PA_SAMPLE_FLOAT32NE, //Float32 NE, or Float32 Native Endian, the float in c uses the endianess of your pc, or native endian, and float is float32, and double is float64 + .channels = 2, + .rate = SAMPLE_RATE // Same sample rate makes it easy, leave the resampling to pipewire, it should know better + }; + pa_sample_spec mono_format = { + .format = PA_SAMPLE_FLOAT32NE, + .channels = 1, + .rate = SAMPLE_RATE + }; + + pa_buffer_attr input_buffer_atr = { + .maxlength = 4096, // You can lower this to 512, but this is fine, it's sub-second delay, you're probably not gonna notice unless you're looking for it + .fragsize = 2048 + }; + pa_buffer_attr output_buffer_atr = { + .maxlength = 4096, + .tlength = 2048, + .prebuf = 0 + }; + + printf("Connecting to input device... (%s)\n", INPUT_DEVICE); + + pa_simple *input_device = pa_simple_new( + NULL, + "StereoEncoder", + PA_STREAM_RECORD, + INPUT_DEVICE, + "Audio Input", + &stereo_format, + NULL, + &input_buffer_atr, + NULL + ); + if (!input_device) { + fprintf(stderr, "Error: cannot open input device.\n"); + return 1; + } + + printf("Connecting to output device... (%s)\n", OUTPUT_DEVICE); + + pa_simple *output_device = pa_simple_new( + NULL, + "StereoEncoder", + PA_STREAM_PLAYBACK, + OUTPUT_DEVICE, + "MPX", + &mono_format, + NULL, + &output_buffer_atr, + NULL + ); + if (!output_device) { + fprintf(stderr, "Error: cannot open output device.\n"); + pa_simple_free(input_device); + return 1; + } + + Oscillator pilot_osc; + init_oscillator(&pilot_osc, 19000.0, SAMPLE_RATE); // Pilot, it's there to indicate stereo and as a refrence signal with the stereo carrier + HilbertTransformer hilbert; + init_hilbert(&hilbert); +#ifdef PREEMPHASIS + Emphasis preemp_l, preemp_r; + init_emphasis(&preemp_l, PREEMPHASIS_TAU, SAMPLE_RATE); + init_emphasis(&preemp_r, PREEMPHASIS_TAU, SAMPLE_RATE); +#endif +#ifdef LPF + LowPassFilter lpf_l, lpf_r; + init_low_pass_filter(&lpf_l, LPF_CUTOFF, SAMPLE_RATE); + init_low_pass_filter(&lpf_r, LPF_CUTOFF, SAMPLE_RATE); +#endif + + signal(SIGINT, stop); + signal(SIGTERM, stop); + + float input[BUFFER_SIZE*2]; // Input from device, interleaved stereo + float left[BUFFER_SIZE+64], right[BUFFER_SIZE+64]; // Audio, same thing as in input but ininterleaved, ai told be there could be a buffer overflow here + float mpx[BUFFER_SIZE]; // MPX, this goes to the output + while (to_run) { + if (pa_simple_read(input_device, input, sizeof(input), NULL) < 0) { + fprintf(stderr, "Error reading from input device.\n"); + break; + } + uninterleave(input, left, right, BUFFER_SIZE*2); + + for (int i = 0; i < BUFFER_SIZE; i++) { + float sin38 = sinf(pilot_osc.phase*2); // Stereo carrier should be a harmonic of the pilot which is in phase, best way to generate the harmonic is to multiply the pilot's phase by two, so it is mathematically impossible for them to not be in phase + float cos38 = cosf(pilot_osc.phase*2); // Stereo carrier should be a harmonic of the pilot which is in phase, best way to generate the harmonic is to multiply the pilot's phase by two, so it is mathematically impossible for them to not be in phase + float pilot = get_oscillator_sin_sample(&pilot_osc); // This is after because if it was before then the stereo would be out of phase by one increment, so [GET STEREO] ([GET PILOT] [INCREMENT PHASE]) + float l_in = left[i]; + float r_in = right[i]; + +#ifdef PREEMPHASIS +#ifdef LPF + float lowpassed_left = apply_low_pass_filter(&lpf_l, l_in); + float lowpassed_right = apply_low_pass_filter(&lpf_r, r_in); + float preemphasized_left = apply_pre_emphasis(&preemp_l, lowpassed_left); + float preemphasized_right = apply_pre_emphasis(&preemp_r, lowpassed_right); + float current_left_input = clip(preemphasized_left); + float current_right_input = clip(preemphasized_right); +#else + float preemphasized_left = apply_pre_emphasis(&preemp_l, l_in); + float preemphasized_right = apply_pre_emphasis(&preemp_r, r_in); + float current_left_input = clip(preemphasized_left); + float current_right_input = clip(preemphasized_right); +#endif +#else +#ifdef LPF + float lowpassed_left = apply_low_pass_filter(&lpf_l, l_in); + float lowpassed_right = apply_low_pass_filter(&lpf_r, r_in); + float current_left_input = clip(lowpassed_left); + float current_right_input = clip(lowpassed_right); +#else + float current_left_input = clip(l_in); + float current_right_input = clip(r_in); +#endif +#endif + + float mono = (current_left_input + current_right_input) / 2.0f; // Stereo to Mono + + float stereo = (current_left_input - current_right_input) / 2.0f; // Also Stereo to Mono but a bit diffrent + float stereo_i, stereo_q; + apply_hilbert(&hilbert, stereo, &stereo_i, &stereo_q); + + mpx[i] = mono * MONO_VOLUME + + pilot * PILOT_VOLUME + + (stereo_i*cos38+stereo_q*sin38)*STEREO_VOLUME; + } + + if (pa_simple_write(output_device, mpx, sizeof(mpx), NULL) < 0) { + fprintf(stderr, "Error writing to output device.\n"); + break; + } + } + printf("Cleaning up...\n"); + pa_simple_free(input_device); + pa_simple_free(output_device); + exit_hilbert(&hilbert); + return 0; +}