/******************************************************************
Sample plugout program to send AFNI graphing commands.
*******************************************************************/
/***** Header file for communication routines *****/
#include "thd_iochan.h"
#include <math.h>
/***** Global variable determining on which system AFNI runs. *****/
/***** [default is the current system, can be changed by user] *****/
static char afni_host[128] = "." ;
static char afni_name[128] = "\0" ;
static int afni_port = 8777 ;
static int afni_verbose = 0 ; /* print out debug info? */
/***** Prototype *****/
int afni_io(void) ;
/*===================================================================
Main program:
Read command line for options
Call afni_io routine forever
=====================================================================*/
int main( int argc , char * argv[] )
{
int narg , ii ;
/***** See if the pitiful user wants help *****/
if( argc == 2 && strncmp(argv[1],"-help",5) == 0 ){
printf("Usage: plugout_graph [-host name] [-v]\n"
"This program connects to AFNI and sends graphing commands.\n"
"Options:\n"
" -host name Means to connect to AFNI running on the\n"
" computer 'name' using TCP/IP. The default is to\n"
" connect on the current host using shared memory.\n"
" -v Verbose mode.\n"
" -port pp Use TCP/IP port number 'pp'. The default is\n"
" 8099, but if two plugouts are running on the\n"
" same computer, they must use different ports.\n"
" -name sss Use the string 'sss' for the name that AFNI assigns\n"
" to this plugout. The default is something stupid.\n"
) ;
exit(0) ;
}
/***** Process command line options *****/
narg = 1 ;
while( narg < argc ){
/** -host name **/
if( strncmp(argv[narg],"-host",5) == 0 ){
narg++ ;
if( narg >= argc ){
fprintf(stderr,"** -host needs a following name!\a\n"); exit(1);
}
strcpy( afni_host , argv[narg] ) ;
narg++ ; continue ;
}
/** -name sss **/
if( strncmp(argv[narg],"-name",5) == 0 ){
narg++ ;
if( narg >= argc ){
fprintf(stderr,"** -name needs a following string!\a\n"); exit(1);
}
strcpy( afni_name , argv[narg] ) ;
narg++ ; continue ;
}
/** -v **/
if( strncmp(argv[narg],"-v",2) == 0 ){
afni_verbose = 1 ;
narg++ ; continue ;
}
/** -port pp **/
if( strncmp(argv[narg],"-port",4) == 0 ){
narg++ ;
if( narg >= argc ){
fprintf(stderr,"** -port needs a following argument!\a\n"); exit(1);
}
afni_port = strtol( argv[narg] , NULL , 10 ) ;
if( afni_port <= 1024 ){
fprintf(stderr,"** -port needs an argument > 1024!\a\n"); exit(1);
}
if( strcmp(afni_host,".") == 0 ) strcpy(afni_host,"localhost") ;
narg++ ; continue ;
}
/** Je ne sais pas **/
fprintf(stderr,"** Unrecognized option: %s\a\n",argv[narg]) ;
exit(1) ;
}
/***** Loop and check in with AFNI every 33 msec *****/
while( 1 ){
ii = afni_io() ; /* commune with AFNI */
if( ii < 0 ) exit(0) ; /* bad trip? then die */
iochan_sleep(33) ; /* perchance to dream */
}
}
/*===================================================================
This routine handles all communications with AFNI.
The only input is the global variable afni_host, which determines
on which system AFNI is running.
The output is -1 if an error occured, 0 if everything is OK.
=====================================================================*/
/***** Mode flags for determining what afni_io does.
The routine progress through 5 stages:
1) Open a control connection to AFNI;
then
2) Wait until AFNI also opens the control connection;
then
3) Send a control string to AFNI saying what kind of
information we want, wait for an acknowledgment,
close the control connection, and open a data
connection to AFNI;
then
4) Wait for AFNI to also open the data connection;
then
5) Send data to AFNI! *****/
#define AFNI_OPEN_CONTROL_MODE 1 /* 1st time thru: open control channel */
#define AFNI_WAIT_CONTROL_MODE 2 /* wait for AFNI to open control chan */
#define AFNI_OPEN_DATA_MODE 3 /* now can open data channel to AFNI */
#define AFNI_WAIT_DATA_MODE 4 /* waiting for AFNI to open data chan */
#define AFNI_CONTINUE_MODE 5 /* at last! data channel is ready! */
/***** macros to send acknowledgement strings to AFNI *****/
#define POACKSIZE 4 /* length of acknowledgement strings */
#define PO_ACK_BAD(ic) iochan_sendall( (ic) , "BAD" , POACKSIZE )
#define PO_ACK_OK(ic) iochan_sendall( (ic) , "OK!" , POACKSIZE )
#define PO_SEND(ic,str) iochan_sendall( (ic) , (str) , strlen((str))+1 )
int afni_io(void)
{
static int afni_mode = AFNI_OPEN_CONTROL_MODE ; /* status variable */
static IOCHAN * afni_ioc = NULL ; /* connection to AFNI */
int ii ;
/***************************************************************/
/***** Check to see if status is OK before we proceed. *****/
/***** (if an error occurs below, afni_mode gets set to 0) *****/
if( afni_mode <= 0 ) return -1 ;
/***********************************************************************/
/***** First time into this routine? Open control channel to AFNI *****/
if( afni_mode == AFNI_OPEN_CONTROL_MODE ){
char afni_iocname[128] ; /* will hold name of I/O channel */
/** Note that the control channel is always a
TCP/IP channel to port # 7957 on the AFNI host system **/
if( strcmp(afni_host,".") == 0 )
sprintf( afni_iocname , "tcp:%s:7957" , "localhost" ); /* make name */
else
sprintf( afni_iocname , "tcp:%s:7957" , afni_host ) ; /* make name */
afni_ioc = iochan_init( afni_iocname , "create" ) ; /* create it */
if( afni_ioc == NULL ){
fprintf(stderr,
"** Can't create control channel %s to AFNI!\n",afni_iocname) ;
afni_mode = 0 ;
return -1 ;
}
afni_mode = AFNI_WAIT_CONTROL_MODE ; /* waiting for AFNI connection */
if( afni_verbose )
fprintf(stderr,"++ AFNI control channel created\n") ;
}
/****************************************************/
/**** Check if AFNI control channel is connected ****/
if( afni_mode == AFNI_WAIT_CONTROL_MODE ){
ii = iochan_writecheck( afni_ioc , 5 ) ; /* wait at most 5 msec */
/** the iochan_*check() routines return
-1 for an error,
0 if not ready,
>0 if the I/O channel is ready. **/
if( ii < 0 ){
fprintf(stderr,"** Control channel to AFNI failed!\a\n") ;
iochan_set_cutoff(afni_ioc) ; IOCHAN_CLOSE(afni_ioc) ;
afni_mode = 0 ;
return -1 ;
} else if( ii > 0 ){
afni_mode = AFNI_OPEN_DATA_MODE ; /* prepare to send data */
if( afni_verbose )
fprintf(stderr,"++ AFNI control channel connected\n");
} else {
return 0 ; /* try again next time */
}
}
/**********************************************************/
/**** Send control data to AFNI, and open data channel ****/
if( afni_mode == AFNI_OPEN_DATA_MODE ){
char afni_iocname[128] ;
char afni_buf[256] ;
/** decide name of data channel:
use shared memory (shm:) on ".",
use TCP/IP (tcp:) on other computer systems;
* Note that the TCP/IP port number can be
anything that isn't already in use;
* Note that the shm control name (here "test_plugout")
is a string that will be converted to an IPC
key (in function string_to_key in iochan.c). **/
if( strcmp(afni_host,".") == 0 )
strcpy( afni_iocname , "shm:test_plugout:1K+1K" ) ;
else
sprintf( afni_iocname , "tcp:%s:%d" , afni_host , afni_port ) ;
/** write the command to AFNI into the buffer:
* each command ends with a newline character,
except (possibly) the last command;
* the command buffer is a C string, which ends
with an ASCII NUL character;
* PONAME means 'use this string for informative messages';
* IOCHAN means 'use this I/O channel from now on'. **/
if( afni_name[0] == '\0' ) strcpy(afni_name,"TambourineMan") ;
sprintf( afni_buf , "PONAME %s\n"
"IOCHAN %s\n"
"NO_ACK" ,
afni_name , afni_iocname ) ;
if( afni_verbose )
fprintf(stderr,"++ Sending control information to AFNI\n") ;
/** note that the ASCII NUL at the end of the buffer is sent **/
ii = iochan_sendall( afni_ioc , afni_buf , strlen(afni_buf)+1 ) ;
/** the return value is the number of bytes sent,
or -1 indicating a fatal error transpired. **/
if( ii < 0 ){
fprintf(stderr,"** Transmission of control data to AFNI failed!\a\n") ;
iochan_set_cutoff(afni_ioc) ; IOCHAN_CLOSE(afni_ioc) ;
afni_mode = 0 ;
return -1 ;
} else {
/** close control channel **/
iochan_set_cutoff(afni_ioc) ; IOCHAN_CLOSE(afni_ioc) ;
/** now open data channel to AFNI **/
afni_ioc = iochan_init( afni_iocname , "create" ) ;
if( afni_ioc == NULL ){
fprintf(stderr,
"** Can't open data channel %s to AFNI!\a\n",afni_iocname) ;
afni_mode = 0 ;
return -1 ;
} else {
afni_mode = AFNI_WAIT_DATA_MODE ;
if( afni_verbose ) fprintf(stderr,"++ AFNI data channel created\n") ;
}
}
}
/****************************************************/
/***** See if data channel is connected to AFNI *****/
if( afni_mode == AFNI_WAIT_DATA_MODE ){
ii = iochan_goodcheck( afni_ioc , 5 ) ; /* wait at most 5 msec */
if( ii < 0 ){
fprintf(stderr,
"** AFNI data channel aborted before any data was sent!\a\n") ;
iochan_set_cutoff(afni_ioc) ;IOCHAN_CLOSE(afni_ioc) ;
afni_mode = 0 ;
return -1 ;
} else if( ii > 0 ){ /* ready to go! */
afni_mode = AFNI_CONTINUE_MODE ;
if( afni_verbose ) fprintf(stderr,"++ AFNI data channel is open\n") ;
} else {
return 0 ; /* try again next time */
}
}
/**************************************************************/
/***** The "normal" state of affairs: AFNI is connected. *****/
/***** See if the user wants to drive AFNI. *****/
if( afni_mode == AFNI_CONTINUE_MODE ){
static int ncom=-1 , nn ;
static double f=2.0 ;
double y1,y2,y3 , t ;
char afni_buf[256];
switch( ncom ){
case -1: /* first time in only */
strcpy(afni_buf,"DRIVE_AFNI OPEN_GRAPH_XY bob '' 0 80 '' 3 -1.5 1.5") ;
srand48((long)time(NULL)) ;
ncom++ ;
break ;
case 801:
strcpy(afni_buf,"DRIVE_AFNI CLEAR_GRAPH_XY bob") ;
ncom = 0 ;
break ;
default:
t = 0.1*ncom ; ncom++ ;
y1 = cos(t) ;
y2 = cos(f*t) ;
y3 = y2+0.2*sin(t/7.0)+0.2*(drand48()-0.5) ;
sprintf(afni_buf,"DRIVE_AFNI ADDTO_GRAPH_XY bob %g %g %g %g",t,y1,y2,y3) ;
#if 1
f += 0.01*(drand48()-0.5) ;
#endif
break ;
}
/* send command to AFNI */
nn = strlen(afni_buf)+1 ;
ii = iochan_sendall( afni_ioc , afni_buf , nn ) ;
if( ii < 0 ){ /* send or acknowledgment failed */
char *sss ;
fprintf(stderr,"** AFNI data channel failed!\a\n") ;
sss = iochan_error_string() ;
if( sss != NULL ) fprintf(stderr,"** %s\n",sss) ;
iochan_set_cutoff(afni_ioc) ; IOCHAN_CLOSE(afni_ioc) ;
afni_mode = 0 ;
return -1 ;
} else if( ii < nn ){
char *sss ;
fprintf(stderr,"++ AFNI data channel full - can't send data!\n") ;
sss = iochan_error_string() ;
if( sss != NULL ) fprintf(stderr,"++ %s\n",sss) ;
}
return 0 ;
}
return 0 ;
}
syntax highlighted by Code2HTML, v. 0.9.1