int main(int argc, char *argv[]) {
/* thread args */
struct guppi_thread_args net_args, pfb_args, accum_args, disk_args;
guppi_thread_args_init(&net_args);
guppi_thread_args_init(&pfb_args);
guppi_thread_args_init(&accum_args);
guppi_thread_args_init(&disk_args);
net_args.output_buffer = 1;
pfb_args.input_buffer = net_args.output_buffer;
pfb_args.output_buffer = 2;
accum_args.input_buffer = pfb_args.output_buffer;
accum_args.output_buffer = 3;
disk_args.input_buffer = accum_args.output_buffer;
/* Init status shared mem */
struct guppi_status stat;
int rv = guppi_status_attach(&stat);
if (rv!=GUPPI_OK) {
fprintf(stderr, "Error connecting to guppi_status\n");
exit(1);
}
hputs(stat.buf, "BW_MODE", "low");
hputs(stat.buf, "SWVER", SWVER);
/* Init first shared data buffer */
struct guppi_databuf *gpu_input_dbuf=NULL;
gpu_input_dbuf = guppi_databuf_attach(pfb_args.input_buffer);
/* If attach fails, first try to create the databuf */
if (gpu_input_dbuf==NULL)
gpu_input_dbuf = guppi_databuf_create(24, 32*1024*1024,
pfb_args.input_buffer, GPU_INPUT_BUF);
/* If that also fails, exit */
if (gpu_input_dbuf==NULL) {
fprintf(stderr, "Error connecting to gpu_input_dbuf\n");
exit(1);
}
guppi_databuf_clear(gpu_input_dbuf);
/* Init second shared data buffer */
struct guppi_databuf *cpu_input_dbuf=NULL;
cpu_input_dbuf = guppi_databuf_attach(accum_args.input_buffer);
/* If attach fails, first try to create the databuf */
if (cpu_input_dbuf==NULL)
cpu_input_dbuf = guppi_databuf_create(24, 32*1024*1024,
accum_args.input_buffer, CPU_INPUT_BUF);
/* If that also fails, exit */
if (cpu_input_dbuf==NULL) {
fprintf(stderr, "Error connecting to cpu_input_dbuf\n");
exit(1);
}
guppi_databuf_clear(cpu_input_dbuf);
/* Init third shared data buffer */
struct guppi_databuf *disk_input_dbuf=NULL;
disk_input_dbuf = guppi_databuf_attach(disk_args.input_buffer);
/* If attach fails, first try to create the databuf */
if (disk_input_dbuf==NULL)
disk_input_dbuf = guppi_databuf_create(16, 32*1024*1024,
disk_args.input_buffer, DISK_INPUT_BUF);
/* If that also fails, exit */
if (disk_input_dbuf==NULL) {
fprintf(stderr, "Error connecting to disk_input_dbuf\n");
exit(1);
}
guppi_databuf_clear(disk_input_dbuf);
signal(SIGINT, cc);
/* Launch net thread */
pthread_t net_thread_id;
#ifdef FAKE_NET
rv = pthread_create(&net_thread_id, NULL, guppi_fake_net_thread,
(void *)&net_args);
#else
rv = pthread_create(&net_thread_id, NULL, guppi_net_thread,
(void *)&net_args);
#endif
if (rv) {
fprintf(stderr, "Error creating net thread.\n");
perror("pthread_create");
exit(1);
}
/* Launch PFB thread */
pthread_t pfb_thread_id;
rv = pthread_create(&pfb_thread_id, NULL, vegas_pfb_thread, (void *)&pfb_args);
if (rv) {
fprintf(stderr, "Error creating PFB thread.\n");
perror("pthread_create");
exit(1);
}
/* Launch accumulator thread */
pthread_t accum_thread_id;
rv = pthread_create(&accum_thread_id, NULL, guppi_accum_thread, (void *)&accum_args);
if (rv) {
fprintf(stderr, "Error creating accumulator thread.\n");
perror("pthread_create");
exit(1);
}
/* Launch RAW_DISK thread, SDFITS disk thread, or PSRFITS disk thread */
pthread_t disk_thread_id;
#ifdef RAW_DISK
rv = pthread_create(&disk_thread_id, NULL, guppi_rawdisk_thread,
(void *)&disk_args);
#elif defined NULL_DISK
rv = pthread_create(&disk_thread_id, NULL, guppi_null_thread,
(void *)&disk_args);
#elif defined EXT_DISK
rv = 0;
#elif FITS_TYPE == PSRFITS
rv = pthread_create(&disk_thread_id, NULL, guppi_psrfits_thread,
(void *)&disk_args);
#elif FITS_TYPE == SDFITS
rv = pthread_create(&disk_thread_id, NULL, guppi_sdfits_thread,
(void *)&disk_args);
#endif
if (rv) {
fprintf(stderr, "Error creating disk thread.\n");
perror("pthread_create");
exit(1);
}
/* Wait for end */
run_threads=1;
while (run_threads) {
sleep(1);
if (disk_args.finished) run_threads=0;
}
pthread_cancel(disk_thread_id);
pthread_cancel(pfb_thread_id);
pthread_cancel(accum_thread_id);
pthread_cancel(net_thread_id);
pthread_kill(disk_thread_id,SIGINT);
pthread_kill(accum_thread_id,SIGINT);
pthread_kill(pfb_thread_id,SIGINT);
pthread_kill(net_thread_id,SIGINT);
pthread_join(net_thread_id,NULL);
printf("Joined net thread\n"); fflush(stdout);
pthread_join(pfb_thread_id,NULL);
printf("Joined PFB thread\n"); fflush(stdout);
pthread_join(accum_thread_id,NULL);
printf("Joined accumulator thread\n"); fflush(stdout);
pthread_join(disk_thread_id,NULL);
printf("Joined disk thread\n"); fflush(stdout);
guppi_thread_args_destroy(&net_args);
guppi_thread_args_destroy(&pfb_args);
guppi_thread_args_destroy(&accum_args);
guppi_thread_args_destroy(&disk_args);
exit(0);
}
int main(int argc, char *argv[]) {
static struct option long_opts[] = {
{"help", 0, NULL, 'h'},
{"disk", 0, NULL, 'd'},
{"only_net",0, NULL, 'o'},
{0,0,0,0}
};
int opt, opti;
int disk=0, only_net=0;
while ((opt=getopt_long(argc,argv,"hdo",long_opts,&opti))!=-1) {
switch (opt) {
case 'd':
disk=1;
break;
case 'o':
only_net=1;
break;
default:
case 'h':
usage();
exit(0);
break;
}
}
/* Net thread args */
struct guppi_thread_args net_args;
guppi_thread_args_init(&net_args);
net_args.output_buffer = 1;
/* Init shared mem */
struct guppi_status stat;
struct guppi_databuf *dbuf=NULL;
int rv = guppi_status_attach(&stat);
if (rv!=GUPPI_OK) {
fprintf(stderr, "Error connecting to guppi_status\n");
exit(1);
}
dbuf = guppi_databuf_attach(net_args.output_buffer);
/* If attach fails, first try to create the databuf */
if (dbuf==NULL)
#ifdef NEW_GBT
dbuf = guppi_databuf_create(24, 32*1024*1024, net_args.output_buffer, CPU_INPUT_BUF);
#else
dbuf = guppi_databuf_create(24, 32*1024*1024, net_args.output_buffer);
#endif
/* If that also fails, exit */
if (dbuf==NULL) {
fprintf(stderr, "Error connecting to guppi_databuf\n");
exit(1);
}
guppi_databuf_clear(dbuf);
run=1;
signal(SIGINT, cc);
/* Launch net thread */
pthread_t net_thread_id;
rv = pthread_create(&net_thread_id, NULL, guppi_net_thread,
(void *)&net_args);
if (rv) {
fprintf(stderr, "Error creating net thread.\n");
perror("pthread_create");
exit(1);
}
/* Launch raw disk (or null) thread */
struct guppi_thread_args null_args;
guppi_thread_args_init(&null_args);
null_args.input_buffer = net_args.output_buffer;
pthread_t disk_thread_id=0;
if (only_net==0) {
if (disk)
rv = pthread_create(&disk_thread_id, NULL, guppi_rawdisk_thread,
(void *)&null_args);
else
rv = pthread_create(&disk_thread_id, NULL, guppi_null_thread,
(void *)&null_args);
if (rv) {
fprintf(stderr, "Error creating null thread.\n");
perror("pthread_create");
exit(1);
}
}
/* Wait for end */
while (run) { sleep(1); }
if (disk_thread_id) pthread_cancel(disk_thread_id);
pthread_cancel(net_thread_id);
if (disk_thread_id) pthread_kill(disk_thread_id,SIGINT);
pthread_kill(net_thread_id,SIGINT);
pthread_join(net_thread_id,NULL);
printf("Joined net thread\n"); fflush(stdout); fflush(stderr);
if (disk_thread_id) {
pthread_join(disk_thread_id,NULL);
printf("Joined disk thread\n"); fflush(stdout); fflush(stderr);
}
guppi_thread_args_destroy(&null_args);
exit(0);
}
