RS-tracker/scan/rs_detect.c

432 wiersze
11 KiB
C

/*
* detect/identify radiosondes
* DFM, RS92-SGP, RS41, M10, iMet-1-AB
*/
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>
#ifdef CYGWIN
#include <fcntl.h> // cygwin: _setmode()
#include <io.h>
#endif
typedef unsigned char ui8_t;
typedef unsigned int ui32_t;
time_t rawtime;
struct tm *gmt;
int wavloaded = 0,
option_gmt = 0,
option_silent = 0,
option_b = 0;
#define DFM 2
#define RS41 3
#define RS92 4
#define M10 5
#define iMet 6
#define HEADLEN 32
#define HEADOFS 0 // HEADLEN+HEADOFS=32
//int dfm_baudrate = 2500;
char dfm_header[] = "01100101011001101010010110101010";
//int vai_baudrate = 4800;
char rs41_header[] = "00001000011011010101001110001000"
"01000100011010010100100000011111";
char rs92_header[] = "10100110011001101001"
"10100110011001101001"
"10100110011001101001"
"10100110011001101001"
"1010011001100110100110101010100110101001";
//int m10_baudrate = 9600;
char m10_header[] = "00110011001100110101100110110011";
//int imet_baudrate = 9600;
char imet_header[] = "11110000111100001111000011110000"
"11110000""10101100110010101100101010101100"
"11110000""10101100110010101100101010101100";
char buf25[HEADLEN+1] = "x";
int bufpos25 = -1;
char buf48[HEADLEN+1] = "x";
int bufpos48 = -1;
char buf96[HEADLEN+1] = "x";
int bufpos96 = -1;
int L, M;
char *bufb = NULL;
float *bufs = NULL;
/* -------------------------------------------------------------------------- */
int sample_rate = 0, bits_sample = 0, channels = 0;
int findstr(char *buff, char *str, int pos) {
int i;
for (i = 0; i < 4; i++) {
if (buff[(pos+i)%4] != str[i]) break;
}
return i;
}
int read_wav_header(FILE *fp) {
char txt[4+1] = "\0\0\0\0";
unsigned char dat[4];
int byte, p=0;
if (fread(txt, 1, 4, fp) < 4) return -1;
if (strncmp(txt, "RIFF", 4)) return -1;
if (fread(txt, 1, 4, fp) < 4) return -1;
// pos_WAVE = 8L
if (fread(txt, 1, 4, fp) < 4) return -1;
if (strncmp(txt, "WAVE", 4)) return -1;
// pos_fmt = 12L
for ( ; ; ) {
if ( (byte=fgetc(fp)) == EOF ) return -1;
txt[p % 4] = byte;
p++; if (p==4) p=0;
if (findstr(txt, "fmt ", p) == 4) break;
}
if (fread(dat, 1, 4, fp) < 4) return -1;
if (fread(dat, 1, 2, fp) < 2) return -1;
if (fread(dat, 1, 2, fp) < 2) return -1;
channels = dat[0] + (dat[1] << 8);
if (fread(dat, 1, 4, fp) < 4) return -1;
memcpy(&sample_rate, dat, 4); //sample_rate = dat[0]|(dat[1]<<8)|(dat[2]<<16)|(dat[3]<<24);
if (fread(dat, 1, 4, fp) < 4) return -1;
if (fread(dat, 1, 2, fp) < 2) return -1;
//byte = dat[0] + (dat[1] << 8);
if (fread(dat, 1, 2, fp) < 2) return -1;
bits_sample = dat[0] + (dat[1] << 8);
// pos_dat = 36L + info
for ( ; ; ) {
if ( (byte=fgetc(fp)) == EOF ) return -1;
txt[p % 4] = byte;
p++; if (p==4) p=0;
if (findstr(txt, "data", p) == 4) break;
}
if (fread(dat, 1, 4, fp) < 4) return -1;
if (!option_silent) {
fprintf(stderr, "sample_rate: %d\n", sample_rate);
fprintf(stderr, "bits : %d\n", bits_sample);
fprintf(stderr, "channels : %d\n", channels);
}
if ((bits_sample != 8) && (bits_sample != 16)) return -1;
return 0;
}
static int f32read_sample(FILE *fp, float *s) {
int i;
short b = 0;
for (i = 0; i < channels; i++) {
if (fread( &b, bits_sample/8, 1, fp) != 1) return EOF;
if (i == 0) { // i = 0: links bzw. mono
//if (bits_sample == 8) sint = b-128; // 8bit: 00..FF, centerpoint 0x80=128
//if (bits_sample == 16) sint = (short)b;
if (bits_sample == 8) { b -= 128; }
*s = b/128.0;
if (bits_sample == 16) { *s /= 256.0; }
}
}
return 0;
}
int par=1, par_alt=1;
unsigned long sample_count = 0;
int read_bits_fsk(FILE *fp, int *bit, int *len) {
float s = 0.0;
int n;
n = 0;
do{
if (f32read_sample(fp, &s) == EOF) return EOF;
if (option_b) {
bufs[sample_count%M] = s;
}
sample_count++;
par_alt = par;
par = (s >= 0.0) ? 1 : -1;
n++;
} while (par*par_alt > 0);
*len = n;
*bit = (1+par_alt)/2; // oben 1, unten -1
//*bit = (1-par_alt)/2;// sdr#<rev1381?, invers: unten 1, oben -1
return 0;
}
int read_bufb(char *bits, unsigned int mvp, int reset, float spb) {
static unsigned int rcount;
static float rbitgrenze;
double sum = 0.0;
if (reset) {
rcount = 0;
rbitgrenze = 0;
}
rbitgrenze += spb;
do {
sum += bufs[(rcount + mvp + M) % M];
rcount++;
} while (rcount < rbitgrenze); // n < spb
if (sum >= 0) *bits = '1';
else *bits = '0';
return 0;
}
int bufscmp(char *hdr, int len, unsigned int bufspos, float spb) {
int errs = 0;
int pos;
for (pos = 0; pos < len; pos += 1) {
read_bufb(bufb+pos, bufspos, pos==0, spb);
}
bufb[pos] = '\0';
while (len > 0) {
if (bufb[len-1] != hdr[len-1]) errs += 1;
len--;
}
return errs;
}
/* -------------------------------------------------------------------------- */
char inv(char c) {
if (c == '0') return '1';
if (c == '1') return '0';
return c;
}
void inc_buf(int *bufpos) {
*bufpos = (*bufpos+1) % HEADLEN;
}
int compare(char buf[], char header[], int bufpos) {
int i, j;
i = 0;
j = bufpos;
while (i < HEADLEN) {
if (j < 0) j = HEADLEN-1;
if (buf[j] != header[HEADOFS+HEADLEN-1-i]) break;
j--;
i++;
}
if (i == HEADLEN) return 1;
i = 0;
j = bufpos;
while (i < HEADLEN) {
if (j < 0) j = HEADLEN-1;
if (buf[j] != inv(header[HEADOFS+HEADLEN-1-i])) break;
j--;
i++;
}
if (i == HEADLEN) return -1;
return 0;
}
/* -------------------------------------------------------------------------- */
int main(int argc, char **argv) {
FILE *fp;
char *fpname;
int header_found, bit, len, i;
int len25, len48, len96;
int zeit = 0;
int herrs = 0;
#ifdef CYGWIN
_setmode(fileno(stdin), _O_BINARY); // _setmode(_fileno(stdin), _O_BINARY);
setbuf(stdout, NULL);
#endif
fpname = argv[0];
++argv;
while ((*argv) && (!wavloaded)) {
if ( (strcmp(*argv, "-h") == 0) || (strcmp(*argv, "--help") == 0) ) {
fprintf(stderr, "%s [-t nn] audio.wav\n", fpname);
return 0;
}
else if ( (strcmp(*argv, "-s") == 0) || (strcmp(*argv, "--silent") == 0) ) {
option_silent = 1;
}
else if ( (strcmp(*argv, "-z") == 0) || (strcmp(*argv, "--zulu") == 0) ) {
option_gmt = 1;
}
else if ( (strcmp(*argv, "-t") == 0) || (strcmp(*argv, "--time") == 0) ) {
++argv;
if (*argv) zeit = atoi(*argv);
else return -1;
}
else if ( (strcmp(*argv, "-b") == 0) ) {
option_b = 1;
}
else {
fp = fopen(*argv, "rb");
if (fp == NULL) {
fprintf(stderr, "%s konnte nicht geoeffnet werden\n", *argv);
return -1;
}
wavloaded = 1;
}
++argv;
}
if (!wavloaded) fp = stdin;
i = read_wav_header(fp);
if (i) {
fclose(fp);
return -1;
}
if (option_b) {
L = strlen(m10_header);
M = L*sample_rate/9600.0;
bufb = (char *)calloc( L+3, sizeof(char)); if (bufb == NULL) return -1;
bufs = (float *)calloc( M+17, sizeof(float)); if (bufs == NULL) return -1;
}
header_found = 0;
while (!read_bits_fsk(fp, &bit, &len) && !header_found) {
len25 = (int) (len * 2500.0 / sample_rate + 0.5);
len48 = (int) (len * 4800.0 / sample_rate + 0.5);
len96 = (int) (len * 9600.0 / sample_rate + 0.5);
for (i = 0; i < len25; i++) {
inc_buf(&bufpos25);
buf25[bufpos25] = 0x30 + bit; // Ascii
header_found = compare(buf25, dfm_header, bufpos25) * DFM;
if (header_found) goto ende;
}
for (i = 0; i < len48; i++) {
inc_buf(&bufpos48);
buf48[bufpos48] = 0x30 + bit; // Ascii
header_found = compare(buf48, rs41_header+16, bufpos48) * RS41;
if (header_found) goto ende;
header_found = compare(buf48, rs92_header, bufpos48) * RS92;
if (header_found) goto ende;
}
for (i = 0; i < len96; i++) {
inc_buf(&bufpos96);
buf96[bufpos96] = 0x30 + bit; // Ascii
header_found = compare(buf96, m10_header, bufpos96) * M10;
if (header_found) {
if (option_b) {
// rs41-preamble similar to m10-preamble; diffs:
// - iq: modulation-index rs41 < m10
// - pulse-shaping
// m10: 00110011 at 9600 bps
// rs41: 0101 at 4800 bps
// - m10 top-carrier
// - m10-header ..110(1)0110011()011.. bit shuffle
float spb = sample_rate/9600.0;
char hdr[HEADLEN+1];
unsigned long mvp = sample_count - (HEADLEN+len96-1-i)*spb - 1;
int j;
for (j = 0; j < HEADLEN; j++) hdr[j] = buf96[(bufpos96+1+j)%HEADLEN]; hdr[HEADLEN] = '\0';
herrs = bufscmp(hdr, HEADLEN, mvp, spb);
if (herrs < 3) goto ende; // threshold: max errors 2nd pass
}
else goto ende;
}
header_found = compare(buf96, imet_header+32, bufpos96) * iMet;
if (header_found) goto ende;
}
if (zeit > 0 && sample_count > zeit*sample_rate) goto ende;
}
ende:
fclose(fp);
if (option_b) {
if (bufb) { free(bufb); bufb = NULL; }
if (bufs) { free(bufs); bufs = NULL; }
}
if (!option_silent) {
printf("sample: %lu\n", sample_count);
if (zeit) printf("%ds = %d samples\n", zeit, zeit*sample_rate);
printf("found: ");
if (!header_found) printf("NO");
else {
time(&rawtime);
gmt = gmtime(&rawtime);
if (header_found < 0) printf("-");
if (header_found*header_found == DFM*DFM) printf("DFM");
if (header_found*header_found == RS41*RS41) printf("RS41");
if (header_found*header_found == RS92*RS92) printf("RS92");
if (header_found*header_found == M10*M10) {
printf("M10");
if (option_b) printf(" (%d)", herrs);
}
if (header_found*header_found == iMet*iMet) printf("iMet");
if (option_gmt) {
printf(" (%4d-%02d-%02d %02d:%02dZ) ", gmt->tm_year+1900, gmt->tm_mon,
gmt->tm_mday, gmt->tm_hour, gmt->tm_min);
}
}
printf("\n");
}
return header_found;
}