Functions split into own files

rfio.c

@@ -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;
+}

rfio.h

@@ -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);

rftime.c

@@ -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;
+}

rftime.h

@@ -0,0 +1,3 @@
+double nfd2mjd(char *date);
+double date2mjd(int year,int month,double day);
+void mjd2nfd(double mjd,char *nfd);