simple_putdata.c

Go to the documentation of this file.

#include "edtinc.h"

#include <string.h>
#include <stdlib.h>
#ifndef _NT_
#include <unistd.h>
#endif

#define NUMBUFS 4   /* Four is the optimal number of buffers for EDT boards */
int numbufs = NUMBUFS ;
int buffers_inuse ;


void
usage(char *err)
{
    puts(err);
    puts("usage: putdata") ;
    puts("\t-u <unit>      - unit number") ;
    puts("\t-c <channel>   - channel number") ;
    puts("\t-n <num>       - set number of ring buffers") ;
    puts("\t-b             - set byteswap on") ;
    puts("\t-s <size>      - specifies buffer size in bytes") ;
    puts("\t-k <size>      - specifies buffer size in kbytes") ;
    puts("\t-l <loops>     - specifies number of loops") ;
    puts("\t-i <infile>    - specifies input filename (\"-i -\" uses stdandard input");
    puts("\t-h             - prints this message") ;

    exit(1);
}

void init_buf(u_char *buf, int size) ;

int
main(int argc, char **argv)
{
    EdtDev *edt_p ;
    int unit ;
    int loop ;
        int loopcount ;
        int byteswap=0;
        double dtime ;
    unsigned int size ;
    u_char *buf ;
    char *infile ;
    int ifd = -1 ;
        int lstcnt ;
        int curcnt ;
        int timeouts ;
        int channel = 0;
        int fill;
        int done = 0;
        int total_buffers = 0;
        double num_bytes;
    /* 
     * set defaults 
     */
    unit = 0 ;
    infile = NULL ;
    /* default to 100 writes of 1 Megabyte */
    size = 1024*1024 ;
    loopcount = 100 ;
    timeouts = 0 ;

    --argc;
    ++argv;
    while (argc && argv[0][0] == '-')
    {
                switch (argv[0][1])
                {
                case 'u':
                        ++argv;
                        --argc;
                        if (argc < 1) 
            {
                        usage("Error: option 'u' requires a numeric argument\n");
                }
                if ((argv[0][0] >= '0') && (argv[0][0] <= '9'))
            {
                unit = atoi(argv[0]);
            }
            else 
            {
                        usage("Error: option 'u' requires a numeric argument\n");
                }
                break;

                case 'b':
                        byteswap = 1;
                        break ;
                case 's':
                        ++argv;
                        --argc;
                        size = strtol(argv[0],0,0);
                        break ;
                case 'k':
                        ++argv;
                        --argc;
                        size = atoi(argv[0]) * 1024;
                        break ;
                case 'n':
                        ++argv;
                        --argc;
            if (argc < 1) 
            {
                        usage("Error: option 'n' requires a numeric argument\n");
                }
                if ((argv[0][0] >= '0') && (argv[0][0] <= '9'))
            {
                numbufs = atoi(argv[0]) ;
                            if (numbufs < 2)
                            {
                                    printf("numbufs should be at least two to take\n") ;
                                    printf("advantage of ring buffer mode\n") ;
                                    exit(1) ;
                            }
            }
            else 
            {
                        usage("Error: option 'n' requires a numeric argument\n");
                }
                break;
                        
                case 'l':
                        ++argv;
                        --argc;
                        if (argc < 1) 
            {
                        usage("Error: option 'l' requires a numeric argument\n");
                }
                if ((argv[0][0] >= '0') && (argv[0][0] <= '9'))
            {
                loopcount = atoi(argv[0]) ;
            }
            else 
            {
                        usage("Error: option 'l' requires a numeric argument\n");
                }
                break;

                case 'i':
                        ++argv ;
                        --argc ;
                        infile = strdup(argv[0]) ;
                        if (strcmp(infile, "-") == 0)
                                ifd = fileno(stdin) ;
                        else
                        {
#ifdef _NT_
                                ifd =   (int)CreateFile(
                                        infile,
                                        GENERIC_READ | GENERIC_WRITE ,
                                        FILE_SHARE_READ | FILE_SHARE_WRITE,
                                        NULL,
                                        OPEN_EXISTING,
                                        FILE_ATTRIBUTE_NORMAL,
                                        NULL);
                                if (ifd == (int)INVALID_HANDLE_VALUE)
                                {
                                        fprintf(stderr, "failed open %s\n",infile) ;
                                        exit(1) ;
                                }
#else
                                if ((ifd = open(infile, 0666)) < 0)
                                {
                                        printf("create of %s failed\n", infile) ;
                                        perror("infile") ;
                                        exit(1) ;
                                }
#endif
                        }
                        break ;
                case 'c':
                        ++argv ;
                        --argc ;
                        if (argc < 1) 
            {
                        usage("Error: option 'c' requires a numeric argument\n");
                }
                if ((argv[0][0] >= '0') && (argv[0][0] <= '9'))
            {
                channel = atoi(argv[0]);
            }
            else 
            {
                        usage("Error: option 'c' requires a numeric argument\n");
                }
                break;

                case 'h':
                case '?':
            usage("help");
            break;

                default:
                        usage("unknown option") ;
                }
                --argc;
                ++argv;
    }

    if (infile )
            printf("writing buffers until end of file of %d bytes from unit %d channel %d with %d bufs\n",
                 size, unit, channel, numbufs) ;
    else
            printf("writing %d buffers of %d bytes from unit %d channel %d with %d bufs\n",
                loopcount, size, unit, channel, numbufs) ;

    if ((edt_p = edt_open_channel(EDT_INTERFACE, unit,channel)) == NULL)
    {
                perror("edt_open") ;
                exit(1) ;
    }

    /*
     * Have edt_configure_ring_buffers allocate the ring buffers by
     * passing the last argument as NULL.  Otherwise pass in an array
     * of pointers to the buffers.
     */
        
    if (edt_configure_ring_buffers(edt_p, size, numbufs, EDT_WRITE, NULL) == -1)
    {
        perror("edt_configure_ring_buffers") ;
        exit(1) ;
    }

        (void)edt_dtime();
#ifdef PCD
        if (byteswap)
            pcd_set_byteswap(edt_p, 1) ;
#endif

    buffers_inuse = 0 ;
        curcnt = edt_timeouts(edt_p) ;
        lstcnt = curcnt ;
    done = 0;
    if (infile)
        loopcount = 0;

    for(loop = 0 ; (loop < loopcount || loopcount == 0) && !done ; loop++)
    {
                /*
                 * Initialize and start ring buffers until they're all in use,
                 * i.e. don't wait for a buffer until they're all being used.
                 */
                if (buffers_inuse >= numbufs)
                {
                        /* All ring buffers are in use; wait for the next one to finish */
                        edt_wait_for_buffers(edt_p, 1) ;
                        --buffers_inuse ;
                        curcnt = edt_timeouts(edt_p) ;
                        if (lstcnt != curcnt) 
                        {
                                putchar('x') ;
                                timeouts++ ;
                                lstcnt = curcnt ;
                                edt_reset_ring_buffers(edt_p,0) ;
                                edt_flush_fifo(edt_p) ;
                                edt_start_buffers(edt_p, 0) ; /* start the transfers in free running mode */
                        }
                        else putchar('.') ;
                        fflush(stdout) ;
                }

                /*
                 * Copy data into each ring buffer, then start DMA.
                 * Data could come from a file, a tape drive, etc,
                 * and is written to the external device connected to
                 * the EDT board continuously via the ring buffers.
                 */
                for(fill=buffers_inuse; fill<numbufs; fill++)
                {
                buf = edt_next_writebuf(edt_p) ;

                if (infile)
                {
                        unsigned long Length ;
#ifdef _NT_             
                        ReadFile((void *)ifd, buf, size, &Length, NULL) ;
#else
                        Length = read(ifd, buf, size) ;
#endif
                        buf += size;
                        if (Length != size)
                                {
                                /* change size on last if not full buf */
                                printf("end of file\n") ;
                                done = 1;
                                }
                }
                else 
                        init_buf(buf, size);
                }


                if (done == 0) 
                {
                        /* Start the transfers on this buffer */
                        edt_start_buffers(edt_p, numbufs - buffers_inuse); 
                        total_buffers = total_buffers + (numbufs - buffers_inuse);
                        buffers_inuse = numbufs;
                }
    }

    /* Now wait for remaining buffers to finish */
    while (buffers_inuse)
    {
                --buffers_inuse ;
                edt_wait_for_buffers(edt_p, 1) ;

                putchar('.') ;
                fflush(stdout) ;
    }

        dtime = edt_dtime() ;
        num_bytes = (double)total_buffers * (double)size;
        printf("\ntotalbuffers %d %f bytes/sec %d timeouts\n",
                total_buffers, num_bytes / dtime,timeouts) ;
        printf("\nput %f bytes in %f seconds for %f bytes/second\n",
                            num_bytes,
                            dtime,
                            num_bytes/dtime);
        edt_close(edt_p) ;
        exit(0) ;
        return(0) ;
}

/*
 * initialize buffer with 16-bit incremented data
 * could replace with read of data from file, other device, or ???
 */
void
init_buf(u_char *buf, int size)
{
    u_short word = 0;
    u_short *p, *endp;
    static int count = 0 ;

    if (++count > numbufs)
        return ;

    endp = (u_short *) (buf + size);
    p = (u_short *) buf;

    while (p < endp)
    {
        *p++ = word++;
    }
}

---