Functions split into own files
parent
a0b31cff5c
commit
79231c5864
|
@ -0,0 +1,170 @@
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <math.h>
|
||||||
|
#include "rftime.h"
|
||||||
|
#include "rfio.h"
|
||||||
|
|
||||||
|
struct spectrogram read_spectrogram(char *prefix,int isub,int nsub,double f0,double df0,int nbin)
|
||||||
|
{
|
||||||
|
int i,j,k,l,flag=0,status,msub,ibin;
|
||||||
|
char filename[128],header[256],nfd[32];
|
||||||
|
FILE *file;
|
||||||
|
struct spectrogram s;
|
||||||
|
float *z;
|
||||||
|
int nch,j0,j1;
|
||||||
|
double freq,samp_rate;
|
||||||
|
float length;
|
||||||
|
int nchan;
|
||||||
|
|
||||||
|
// Open first file to get number of channels
|
||||||
|
sprintf(filename,"%s_%06d.bin",prefix,isub);
|
||||||
|
|
||||||
|
// Open file
|
||||||
|
file=fopen(filename,"r");
|
||||||
|
if (file==NULL) {
|
||||||
|
printf("%s does not exist\n",filename);
|
||||||
|
return s;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read header
|
||||||
|
status=fread(header,sizeof(char),256,file);
|
||||||
|
status=sscanf(header,"HEADER\nUTC_START %s\nFREQ %lf Hz\nBW %lf Hz\nLENGTH %f s\nNCHAN %d\n",s.nfd0,&s.freq,&s.samp_rate,&length,&nch);
|
||||||
|
|
||||||
|
// Close file
|
||||||
|
fclose(file);
|
||||||
|
|
||||||
|
// Compute plotting channel
|
||||||
|
if (f0>0.0 && df0>0.0) {
|
||||||
|
s.nchan=(int) (df0/s.samp_rate*(float) nch);
|
||||||
|
|
||||||
|
j0=(int) ((f0-0.5*df0-s.freq+0.5*s.samp_rate)*(float) nch/s.samp_rate);
|
||||||
|
j1=(int) ((f0+0.5*df0-s.freq+0.5*s.samp_rate)*(float) nch/s.samp_rate);
|
||||||
|
|
||||||
|
if (j0<0 || j1>nch)
|
||||||
|
fprintf(stderr,"Requested frequency range out of limits\n");
|
||||||
|
} else {
|
||||||
|
s.nchan=nch;
|
||||||
|
j0=0;
|
||||||
|
j1=s.nchan;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Number of subints
|
||||||
|
s.nsub=nsub/nbin;
|
||||||
|
|
||||||
|
// Allocate
|
||||||
|
s.z=(float *) malloc(sizeof(float)*s.nchan*s.nsub);
|
||||||
|
z=(float *) malloc(sizeof(float)*nch);
|
||||||
|
s.mjd=(double *) malloc(sizeof(double)*s.nsub);
|
||||||
|
s.length=(float *) malloc(sizeof(float)*s.nsub);
|
||||||
|
|
||||||
|
// Initialize
|
||||||
|
for (j=0;j<s.nchan*s.nsub;j++)
|
||||||
|
s.z[j]=0.0;
|
||||||
|
for (j=0;j<s.nsub;j++)
|
||||||
|
s.mjd[j]=0.0;
|
||||||
|
|
||||||
|
// Loop over files
|
||||||
|
for (k=0,i=0,l=0,ibin=0;l<nsub;k++) {
|
||||||
|
// Generate filename
|
||||||
|
sprintf(filename,"%s_%06d.bin",prefix,k+isub);
|
||||||
|
|
||||||
|
// Open file
|
||||||
|
file=fopen(filename,"r");
|
||||||
|
if (file==NULL) {
|
||||||
|
printf("%s does not exist\n",filename);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
printf("opened %s\n",filename);
|
||||||
|
|
||||||
|
// Loop over contents of file
|
||||||
|
for (;l<nsub;l++,ibin++) {
|
||||||
|
// Read header
|
||||||
|
status=fread(header,sizeof(char),256,file);
|
||||||
|
if (status==0)
|
||||||
|
break;
|
||||||
|
status=sscanf(header,"HEADER\nUTC_START %s\nFREQ %lf Hz\nBW %lf Hz\nLENGTH %f s\nNCHAN %d\n",nfd,&freq,&samp_rate,&length,&nchan);
|
||||||
|
s.mjd[i]+=nfd2mjd(nfd)+0.5*length/86400.0;
|
||||||
|
s.length[i]+=length;
|
||||||
|
|
||||||
|
// Read buffer
|
||||||
|
status=fread(z,sizeof(float),nch,file);
|
||||||
|
if (status==0)
|
||||||
|
break;
|
||||||
|
|
||||||
|
// Copy
|
||||||
|
for (j=0;j<s.nchan;j++)
|
||||||
|
s.z[i+s.nsub*j]+=z[j+j0];
|
||||||
|
|
||||||
|
// Increment
|
||||||
|
if (l%nbin==nbin-1) {
|
||||||
|
// Scale
|
||||||
|
s.mjd[i]/=(float) nbin;
|
||||||
|
printf("%d %lf\n",ibin,s.mjd[i]);
|
||||||
|
|
||||||
|
for (j=0;j<s.nchan;j++)
|
||||||
|
s.z[i+s.nsub*j]/=(float) nbin;
|
||||||
|
|
||||||
|
ibin=0;
|
||||||
|
i++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close file
|
||||||
|
fclose(file);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Swap frequency range
|
||||||
|
if (f0>0.0 && df0>0.0) {
|
||||||
|
s.freq=f0;
|
||||||
|
s.samp_rate=df0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Free
|
||||||
|
free(z);
|
||||||
|
|
||||||
|
return s;
|
||||||
|
}
|
||||||
|
|
||||||
|
void write_spectrogram(struct spectrogram s,char *prefix)
|
||||||
|
{
|
||||||
|
int i,j;
|
||||||
|
FILE *file;
|
||||||
|
char header[256]="",filename[256],nfd[32];
|
||||||
|
float *z;
|
||||||
|
double mjd;
|
||||||
|
|
||||||
|
// Allocate
|
||||||
|
z=(float *) malloc(sizeof(float)*s.nchan);
|
||||||
|
|
||||||
|
// Generate filename
|
||||||
|
sprintf(filename,"%s_%06d.bin",prefix,0);
|
||||||
|
|
||||||
|
// Open file
|
||||||
|
file=fopen(filename,"w");
|
||||||
|
|
||||||
|
// Loop over subints
|
||||||
|
for (i=0;i<s.nsub;i++) {
|
||||||
|
// Date
|
||||||
|
mjd=s.mjd[i]-0.5*s.length[i]/86400.0;
|
||||||
|
mjd2nfd(mjd,nfd);
|
||||||
|
|
||||||
|
// Generate header
|
||||||
|
sprintf(header,"HEADER\nUTC_START %s\nFREQ %lf Hz\nBW %lf Hz\nLENGTH %f s\nNCHAN %d\nEND\n",nfd,s.freq,s.samp_rate,s.length[i],s.nchan);
|
||||||
|
|
||||||
|
// Copy buffer
|
||||||
|
for (j=0;j<s.nchan;j++)
|
||||||
|
z[j]=s.z[i+s.nsub*j];
|
||||||
|
|
||||||
|
// Dump contents
|
||||||
|
fwrite(header,sizeof(char),256,file);
|
||||||
|
fwrite(z,sizeof(float),s.nchan,file);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close file
|
||||||
|
fclose(file);
|
||||||
|
|
||||||
|
// Free
|
||||||
|
free(z);
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
|
@ -0,0 +1,10 @@
|
||||||
|
struct spectrogram {
|
||||||
|
int nsub,nchan;
|
||||||
|
double *mjd;
|
||||||
|
double freq,samp_rate;
|
||||||
|
float *length;
|
||||||
|
float *z;
|
||||||
|
char nfd0[32];
|
||||||
|
};
|
||||||
|
struct spectrogram read_spectrogram(char *prefix,int isub,int nsub,double f0,double df0,int nbin);
|
||||||
|
void write_spectrogram(struct spectrogram s,char *prefix);
|
|
@ -0,0 +1,91 @@
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <math.h>
|
||||||
|
#include "rftime.h"
|
||||||
|
|
||||||
|
// nfd2mjd
|
||||||
|
double nfd2mjd(char *date)
|
||||||
|
{
|
||||||
|
int year,month,day,hour,min;
|
||||||
|
double mjd,dday;
|
||||||
|
float sec;
|
||||||
|
|
||||||
|
sscanf(date,"%04d-%02d-%02dT%02d:%02d:%f",&year,&month,&day,&hour,&min,&sec);
|
||||||
|
dday=day+hour/24.0+min/1440.0+sec/86400.0;
|
||||||
|
mjd=date2mjd(year,month,dday);
|
||||||
|
|
||||||
|
return mjd;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compute Julian Day from Date
|
||||||
|
double date2mjd(int year,int month,double day)
|
||||||
|
{
|
||||||
|
int a,b;
|
||||||
|
double jd;
|
||||||
|
|
||||||
|
if (month<3) {
|
||||||
|
year--;
|
||||||
|
month+=12;
|
||||||
|
}
|
||||||
|
|
||||||
|
a=floor(year/100.);
|
||||||
|
b=2.-a+floor(a/4.);
|
||||||
|
|
||||||
|
if (year<1582) b=0;
|
||||||
|
if (year==1582 && month<10) b=0;
|
||||||
|
if (year==1852 && month==10 && day<=4) b=0;
|
||||||
|
|
||||||
|
jd=floor(365.25*(year+4716))+floor(30.6001*(month+1))+day+b-1524.5;
|
||||||
|
|
||||||
|
return jd-2400000.5;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compute Date from Julian Day
|
||||||
|
void mjd2nfd(double mjd,char *nfd)
|
||||||
|
{
|
||||||
|
double f,jd,dday;
|
||||||
|
int z,alpha,a,b,c,d,e;
|
||||||
|
int year,month,day,hour,min;
|
||||||
|
float sec,x;
|
||||||
|
|
||||||
|
jd=mjd+2400000.5;
|
||||||
|
jd+=0.5;
|
||||||
|
|
||||||
|
z=floor(jd);
|
||||||
|
f=fmod(jd,1.);
|
||||||
|
|
||||||
|
if (z<2299161)
|
||||||
|
a=z;
|
||||||
|
else {
|
||||||
|
alpha=floor((z-1867216.25)/36524.25);
|
||||||
|
a=z+1+alpha-floor(alpha/4.);
|
||||||
|
}
|
||||||
|
b=a+1524;
|
||||||
|
c=floor((b-122.1)/365.25);
|
||||||
|
d=floor(365.25*c);
|
||||||
|
e=floor((b-d)/30.6001);
|
||||||
|
|
||||||
|
dday=b-d-floor(30.6001*e)+f;
|
||||||
|
if (e<14)
|
||||||
|
month=e-1;
|
||||||
|
else
|
||||||
|
month=e-13;
|
||||||
|
|
||||||
|
if (month>2)
|
||||||
|
year=c-4716;
|
||||||
|
else
|
||||||
|
year=c-4715;
|
||||||
|
|
||||||
|
day=(int) floor(dday);
|
||||||
|
x=24.0*(dday-day);
|
||||||
|
x=3600.*fabs(x);
|
||||||
|
sec=fmod(x,60.);
|
||||||
|
x=(x-sec)/60.;
|
||||||
|
min=fmod(x,60.);
|
||||||
|
x=(x-min)/60.;
|
||||||
|
hour=x;
|
||||||
|
sec=floor(1000.0*sec)/1000.0;
|
||||||
|
|
||||||
|
sprintf(nfd,"%04d-%02d-%02dT%02d:%02d:%06.3f",year,month,day,hour,min,sec);
|
||||||
|
|
||||||
|
return;
|
||||||
|
}
|
Loading…
Reference in New Issue