516 lines
10 KiB
C
516 lines
10 KiB
C
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <math.h>
|
|
#include <qfits.h>
|
|
#include <jpeglib.h>
|
|
#include <libexif/exif-data.h>
|
|
#include <getopt.h>
|
|
|
|
struct image {
|
|
int nx,ny,nz;
|
|
float *z;
|
|
double mjd;
|
|
char nfd[32],observer[32];
|
|
int cospar,tracked;
|
|
float exptime;
|
|
};
|
|
struct image read_jpg(char *filename);
|
|
void write_fits(struct image img,char *filename);
|
|
double date2mjd(int year,int month,double day);
|
|
double nfd2mjd(char *date);
|
|
void mjd2nfd(double mjd,char *nfd);
|
|
|
|
// Read fits image
|
|
struct image read_fits(char *filename)
|
|
{
|
|
int i,j,k,l,m;
|
|
qfitsloader ql;
|
|
char key[FITS_LINESZ+1] ;
|
|
struct image img;
|
|
float s1,s2,avg,std;
|
|
|
|
// Set plane
|
|
ql.xtnum = 0;
|
|
ql.pnum = 0;
|
|
|
|
// Set loadtype
|
|
ql.ptype = PTYPE_FLOAT;
|
|
|
|
// Set filename
|
|
ql.filename=filename;
|
|
|
|
// Image size
|
|
img.nx=atoi(qfits_query_hdr(filename,"NAXIS1"));
|
|
img.ny=atoi(qfits_query_hdr(filename,"NAXIS2"));
|
|
img.nz=1;
|
|
|
|
// Initialize load
|
|
if (qfitsloader_init(&ql) != 0)
|
|
printf("Error initializing data loading\n");
|
|
|
|
// Test load
|
|
if (qfits_loadpix(&ql) != 0)
|
|
printf("Error loading actual data\n");
|
|
|
|
// Allocate image memory
|
|
img.z=(float *) malloc(sizeof(float) * img.nx*img.ny*img.nz);
|
|
|
|
// Fill z array
|
|
for (i=0,l=0,m=0;i<img.nx;i++) {
|
|
for (j=0;j<img.ny;j++) {
|
|
img.z[l]=ql.fbuf[l];
|
|
l++;
|
|
}
|
|
}
|
|
|
|
return img;
|
|
}
|
|
|
|
|
|
struct image rebin(struct image raw,int nbin)
|
|
{
|
|
int i,j,k;
|
|
int ii,jj,kk;
|
|
struct image img;
|
|
|
|
img.nx=raw.nx/nbin;
|
|
img.ny=raw.ny/nbin;
|
|
img.nz=1;
|
|
img.z=(float *) malloc(sizeof(float)*img.nx*img.ny*img.nz);
|
|
|
|
for (i=0;i<img.nx;i++) {
|
|
for (j=0;j<img.ny;j++) {
|
|
k=i+img.nx*j;
|
|
img.z[k]=0.0;
|
|
for (ii=0;ii<nbin;ii++) {
|
|
for (jj=0;jj<nbin;jj++) {
|
|
kk=ii+nbin*i+raw.nx*(jj+nbin*j);
|
|
img.z[k]+=raw.z[kk];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
img.mjd=raw.mjd;
|
|
img.cospar=raw.cospar;
|
|
img.exptime=raw.exptime;
|
|
strcpy(img.nfd,raw.nfd);
|
|
strcpy(img.observer,raw.observer);
|
|
|
|
|
|
return img;
|
|
}
|
|
|
|
void usage(void)
|
|
{
|
|
printf("jpg2fits i:t:o:d:Z:c:T:O:b:hF\n\n");
|
|
printf("-i input JPG file\n");
|
|
printf("-o output FITS file\n");
|
|
printf("-t start time (YYYY-MM-DDTHH:MM:SS.sss)\n");
|
|
printf("-d delay (in seconds)\n");
|
|
printf("-Z timezone offset (in seconds)\n");
|
|
printf("-T exposure time (in seconds)\n");
|
|
printf("-c COSPAR site number\n");
|
|
printf("-O observer name\n");
|
|
printf("-b binning factor\n");
|
|
printf("-F FITS input\n");
|
|
printf("-h this help\n");
|
|
|
|
exit(0);
|
|
|
|
return;
|
|
}
|
|
|
|
int main(int argc,char *argv[])
|
|
{
|
|
int arg;
|
|
struct image img,raw;
|
|
char infile[64],outfile[64]="",nfd[32]="2000-01-01T00:00:00";
|
|
double mjd=51544.0,delay=0.0,tz=0.0;
|
|
int cospar=0;
|
|
char observer[32]="Cees Bassa";
|
|
float exptime=10.06;
|
|
int flag=0,nbin=1,readfits=0,tracked=0;
|
|
|
|
// Decode options
|
|
if (argc>1) {
|
|
while ((arg=getopt(argc,argv,"i:t:o:d:Z:c:T:O:b:hFs"))!=-1) {
|
|
switch(arg) {
|
|
|
|
case 's':
|
|
tracked=1;
|
|
break;
|
|
|
|
case 'i':
|
|
strcpy(infile,optarg);
|
|
break;
|
|
|
|
case 'd':
|
|
delay=(double) atof(optarg);
|
|
break;
|
|
|
|
case 'b':
|
|
nbin=atoi(optarg);
|
|
break;
|
|
|
|
case 'Z':
|
|
tz=(double) atof(optarg);
|
|
break;
|
|
|
|
case 'o':
|
|
strcpy(outfile,optarg);
|
|
flag=1;
|
|
break;
|
|
|
|
case 'F':
|
|
readfits=1;
|
|
break;
|
|
|
|
case 't':
|
|
strcpy(nfd,optarg);
|
|
break;
|
|
|
|
case 'c':
|
|
cospar=atoi(optarg);
|
|
break;
|
|
|
|
case 'T':
|
|
exptime=atof(optarg);
|
|
break;
|
|
|
|
case 'O':
|
|
strcpy(observer,optarg);
|
|
break;
|
|
|
|
case 'h':
|
|
usage();
|
|
break;
|
|
|
|
default:
|
|
usage();
|
|
return 0;
|
|
|
|
}
|
|
}
|
|
} else {
|
|
usage();
|
|
}
|
|
|
|
if (infile!=NULL) {
|
|
if (nbin==1) {
|
|
if (readfits==0)
|
|
img=read_jpg(infile);
|
|
else
|
|
img=read_fits(infile);
|
|
} else {
|
|
if (readfits==0)
|
|
raw=read_jpg(infile);
|
|
else
|
|
raw=read_fits(infile);
|
|
img=rebin(raw,nbin);
|
|
}
|
|
}
|
|
|
|
// Set tracked flag
|
|
if (tracked==1)
|
|
img.tracked=1;
|
|
else
|
|
img.tracked=0;
|
|
|
|
if (nfd!=NULL) {
|
|
// Compute time
|
|
mjd=nfd2mjd(nfd);
|
|
mjd+=(delay+tz)/86400.0;
|
|
mjd2nfd(mjd,nfd);
|
|
|
|
// Into file
|
|
strcpy(img.nfd,nfd);
|
|
img.mjd=mjd;
|
|
}
|
|
|
|
if (flag==0)
|
|
sprintf(outfile,"%s.fits",img.nfd);
|
|
|
|
// Set properties
|
|
img.cospar=cospar;
|
|
img.exptime=exptime;
|
|
strcpy(img.observer,observer);
|
|
|
|
if (outfile!=NULL)
|
|
write_fits(img,outfile);
|
|
|
|
// Free
|
|
free(img.z);
|
|
if (nbin!=1)
|
|
free(raw.z);
|
|
|
|
return 0;
|
|
}
|
|
|
|
struct image read_jpg(char *filename)
|
|
{
|
|
int i=0,j,k,l,m;
|
|
unsigned long location=0;
|
|
struct image img;
|
|
struct jpeg_decompress_struct cinfo;
|
|
struct jpeg_error_mgr jerr;
|
|
JSAMPROW row_pointer[1];
|
|
unsigned char *raw_image=NULL;
|
|
FILE *file;
|
|
ExifData *ed;
|
|
ExifEntry *entry;
|
|
|
|
// Open file
|
|
file=fopen(filename,"rb");
|
|
if (!file)
|
|
perror("Error opening file");
|
|
|
|
// Get header info, decompress
|
|
cinfo.err=jpeg_std_error(&jerr);
|
|
jpeg_create_decompress(&cinfo);
|
|
jpeg_stdio_src(&cinfo,file);
|
|
jpeg_read_header(&cinfo,TRUE);
|
|
jpeg_start_decompress(&cinfo);
|
|
|
|
// Allocate memory
|
|
raw_image=(unsigned char *) malloc(cinfo.output_width*cinfo.output_height*cinfo.num_components);
|
|
|
|
// Read image, one scan at a time
|
|
row_pointer[0]=(unsigned char *) malloc(cinfo.output_width*cinfo.num_components);
|
|
while(cinfo.output_scanline<cinfo.image_height) {
|
|
jpeg_read_scanlines(&cinfo,row_pointer,1);
|
|
for(i=0;i<cinfo.image_width*cinfo.num_components;i++)
|
|
raw_image[location++]=row_pointer[0][i];
|
|
}
|
|
// wrap up decompression, destroy objects, free pointers and close open files
|
|
jpeg_finish_decompress(&cinfo);
|
|
jpeg_destroy_decompress(&cinfo);
|
|
|
|
// Copy image to image struct
|
|
img.nx=cinfo.image_width;
|
|
img.ny=cinfo.image_height;
|
|
img.nz=cinfo.num_components;
|
|
img.z=(float *) malloc(sizeof(float)*img.nx*img.ny*img.nz);
|
|
|
|
// Fill image
|
|
for (i=0;i<img.nx;i++) {
|
|
for (j=0;j<img.ny;j++) {
|
|
k=i+(img.ny-j-1)*img.nx;
|
|
img.z[k]=0.0;
|
|
for (l=0;l<img.nz;l++) {
|
|
m=img.nz*(i+img.nx*j)+l;
|
|
img.z[k]+=(float) raw_image[m];
|
|
}
|
|
img.z[k]/=3.0;
|
|
}
|
|
}
|
|
|
|
// Free allocated memory
|
|
free(row_pointer[0]);
|
|
free(raw_image);
|
|
|
|
// Close file
|
|
fclose(file);
|
|
|
|
/*
|
|
// Get exif info
|
|
ed=exif_data_new_from_file(filename);
|
|
if (!ed) {
|
|
printf("File not readable or no EXIF data in file %s\n",filename);
|
|
} else {
|
|
entry=exif_content_get_entry(ed->ifd[0],EXIF_TAG_DATE_TIME);
|
|
exif_entry_get_value(entry,img.nfd, sizeof(img.nfd));
|
|
img.nfd[4]='-';
|
|
img.nfd[7]='-';
|
|
img.nfd[10]='T';
|
|
img.nfd[20]='\0';
|
|
img.mjd=nfd2mjd(img.nfd);
|
|
}
|
|
*/
|
|
return img;
|
|
}
|
|
|
|
// Write fits file
|
|
void write_fits(struct image img,char *filename)
|
|
{
|
|
int i,j,k,l;
|
|
int *ibuf;
|
|
qfitsdumper qd;
|
|
qfits_header *qh;
|
|
char key[FITS_LINESZ+1] ;
|
|
char val[FITS_LINESZ+1] ;
|
|
char com[FITS_LINESZ+1] ;
|
|
char lin[FITS_LINESZ+1] ;
|
|
FILE *file;
|
|
|
|
// Create FITS header
|
|
qh=qfits_header_default();
|
|
|
|
// Add stuff
|
|
qfits_header_add(qh,"BITPIX","16"," ",NULL);
|
|
qfits_header_add(qh,"NAXIS","2"," ",NULL);
|
|
sprintf(val,"%i",img.nx);
|
|
qfits_header_add(qh,"NAXIS1",val," ",NULL);
|
|
sprintf(val,"%i",img.ny);
|
|
qfits_header_add(qh,"NAXIS2",val," ",NULL);
|
|
qfits_header_add(qh,"BSCALE","1.0"," ",NULL);
|
|
qfits_header_add(qh,"BZERO","0.0"," ",NULL);
|
|
qfits_header_add(qh,"DATAMAX","255.0"," ",NULL);
|
|
qfits_header_add(qh,"DATAMIN","0.0"," ",NULL);
|
|
|
|
// Astrometry keywors
|
|
sprintf(val,"%f",img.nx/2.0);
|
|
qfits_header_add(qh,"CRPIX1",val," ",NULL);
|
|
sprintf(val,"%f",img.ny/2.0);
|
|
qfits_header_add(qh,"CRPIX2",val," ",NULL);
|
|
qfits_header_add(qh,"CRVAL1","0.0"," ",NULL);
|
|
qfits_header_add(qh,"CRVAL2","0.0"," ",NULL);
|
|
qfits_header_add(qh,"CD1_1","0.0"," ",NULL);
|
|
qfits_header_add(qh,"CD1_2","0.0"," ",NULL);
|
|
qfits_header_add(qh,"CD2_1","0.0"," ",NULL);
|
|
qfits_header_add(qh,"CD2_2","0.0"," ",NULL);
|
|
qfits_header_add(qh,"CTYPE1","'RA---TAN'"," ",NULL);
|
|
qfits_header_add(qh,"CTYPE2","'DEC--TAN'"," ",NULL);
|
|
qfits_header_add(qh,"CUNIT1","'deg'"," ",NULL);
|
|
qfits_header_add(qh,"CUNIT2","'deg'"," ",NULL);
|
|
qfits_header_add(qh,"CRRES1","0.0"," ",NULL);
|
|
qfits_header_add(qh,"CRRES2","0.0"," ",NULL);
|
|
qfits_header_add(qh,"EQUINOX","2000.0"," ",NULL);
|
|
qfits_header_add(qh,"RADECSYS","ICRS"," ",NULL);
|
|
sprintf(val,"%s",img.nfd);
|
|
qfits_header_add(qh,"DATE-OBS",val," ",NULL);
|
|
sprintf(val,"%lf",img.mjd);
|
|
qfits_header_add(qh,"MJD-OBS",val," ",NULL);
|
|
sprintf(val,"%d",img.cospar);
|
|
qfits_header_add(qh,"COSPAR",val," ",NULL);
|
|
sprintf(val,"%f",img.exptime);
|
|
qfits_header_add(qh,"EXPTIME",val," ",NULL);
|
|
sprintf(val,"%s",img.observer);
|
|
qfits_header_add(qh,"OBSERVER",val," ",NULL);
|
|
sprintf(val,"%d",img.tracked);
|
|
qfits_header_add(qh,"TRACKED",val," ",NULL);
|
|
|
|
|
|
// Dump fitsheader
|
|
// qfits_header_dump(qh,stdout);
|
|
|
|
// Dump to file
|
|
file=fopen(filename,"w");
|
|
qfits_header_dump(qh,file);
|
|
fclose(file);
|
|
|
|
|
|
// Fill buffer
|
|
ibuf=malloc(img.nx*img.ny*sizeof(int));
|
|
for (i=0,l=0;i<img.nx;i++) {
|
|
for (j=img.ny-1;j>=0;j--) {
|
|
ibuf[l]=(int) img.z[l];
|
|
|
|
l++;
|
|
}
|
|
}
|
|
|
|
// Set parameters
|
|
qd.filename=filename;
|
|
qd.npix=img.nx*img.ny;
|
|
qd.ptype=PTYPE_INT;
|
|
qd.ibuf=ibuf;
|
|
qd.out_ptype=BPP_16_SIGNED;
|
|
|
|
// Dump
|
|
qfits_pixdump(&qd);
|
|
|
|
free(ibuf);
|
|
|
|
return;
|
|
}
|
|
|
|
// 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==1582 && 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;
|
|
}
|
|
|
|
// nfd2mjd
|
|
double nfd2mjd(char *date)
|
|
{
|
|
int year,month,day,hour,min;
|
|
float sec;
|
|
double mjd,dday;
|
|
|
|
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 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;
|
|
}
|