Subversion Repositories usb_fpga_2_14

/*%

   fx3demo -- Demonstrates common features of the FX3

   Copyright (C) 2009-2017 ZTEX GmbH.

   http://www.ztex.de

   Licensed under the Apache License, Version 2.0 (the "License");

   you may not use this file except in compliance with the License.

   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software

   distributed under the License is distributed on an "AS IS" BASIS,

   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

   See the License for the specific language governing permissions and

   limitations under the License.

%*/

import java.io.*;

import java.util.*;

import java.nio.*;

import org.usb4java.*;

import ztex.*;

// *****************************************************************************

// ******* ParameterException **************************************************

// *****************************************************************************

// Exception the prints a help message

class ParameterException extends Exception {

    public final static String helpMsg = new String (

                "Parameters:\n"+

                "    -d <number>       Device Number (default: 0)\n" +

                "    -f                Force uploads\n" +

                "    -p                Print bus info\n" +

                "    -h                This help" );

    public ParameterException (String msg) {

        super( msg + "\n" + helpMsg );

    }

}

// *****************************************************************************

// ******* Fx3Demo *************************************************************

// *****************************************************************************

class Fx3Demo extends Ztex1v1 {

// ******* Debug ***************************************************************

// constructor

    public Fx3Demo ( ZtexDevice1 pDev ) throws UsbException {

        super ( pDev );

    }

// ******* echo ****************************************************************

// writes a string to Endpoint 4, reads it back from Endpoint 2 and writes the output to System.out

    public void echo ( String input ) throws UsbException {

        int i = bulkWrite(0x04, allocateByteBuffer(input.getBytes()) , 1000);

        if ( i<0 ) throw new UsbException("Error sending data: " + LibUsb.strError(i));

        System.out.println("Send "+i+" bytes: `"+input+"'" );

        try {

            Thread.sleep( 10 );

        }

            catch ( InterruptedException e ) {

        }

        ByteBuffer buffer = BufferUtils.allocateByteBuffer(1024);

        i = bulkRead(0x82, buffer, 1000);

        if ( i<0 ) throw new UsbException("Error receiving data: " + LibUsb.strError(i));

        else if (i==0) System.out.println("Read "+0+" bytes" );

        else {

            byte[] buf = new byte[i];

            buffer.get(buf);

            System.out.println("Read "+i+" bytes: `"+new String(buf,0,i)+"'" );

        }

    }

// ******* main ****************************************************************

    public static void main (String args[]) {

        int devNum = 0;

        boolean force = false;

        if ( ! System.getProperty("os.name").equalsIgnoreCase("linux") ) {

            Runtime.getRuntime().addShutdownHook(new Thread() {

                public void run() {

                    Scanner s=new Scanner(System.in);

                    System.out.println("Press <enter> to continue ...");

                    s.nextLine();

                }

            });

        }

        try {

// Scan the USB. This also creates and initializes a new USB context.

            ZtexScanBus1 bus = new ZtexScanBus1( ZtexDevice1.ztexVendorId, ZtexDevice1.ztexProductId, true, false, 1);

// scan the command line arguments

            for (int i=0; i<args.length; i++ ) {

                if ( args[i].equals("-d") ) {

                    i++;

                    try {

                        if (i>=args.length) throw new Exception();

                        devNum = Integer.parseInt( args[i] );

                    }

                    catch (Exception e) {

                        throw new ParameterException("Device number expected after -d");

                    }

                }

                else if ( args[i].equals("-f") ) {

                    force = true;

                }

                else if ( args[i].equals("-p") ) {

                    bus.printBus(System.out);

                    System.exit(0);

                }

                else if ( args[i].equals("-p") ) {

                    bus.printBus(System.out);

                    System.exit(0);

                }

                else if ( args[i].equals("-h") ) {

                    System.err.println(ParameterException.helpMsg);

                    System.exit(0);

                }

                else throw new ParameterException("Invalid Parameter: "+args[i]);

            }

// create the main class            

            if ( bus.numberOfDevices() <= 0) {

                System.err.println("No devices found");

                System.exit(0);

            }

            Fx3Demo ztex = new Fx3Demo ( bus.device(devNum) );

            bus.unref();

// upload the firmware if necessary

            if ( force || ! ztex.valid() || ! ztex.dev().productString().equals("Demo for FX3 Boards")  ) {

                System.out.println("Trying to overwrite firmware in RAM. This only works if a board specific firmware with reset support is running.\n"+

                                   "If uploading firmware fails please restart board with disabled firmware" );

                System.out.println("Firmware upload time: " + ztex.uploadFirmware( "fx3demo.img", force ) + " ms");

            }

            else {

                ztex.resetDevice(false);        // sync data toggles

            }

            ztex.debug2PrintNextLogMessages(System.out);

// claim interface 0

            ztex.trySetConfiguration ( 1 );

            ztex.claimInterface ( 0 );

// print log

            ztex.debug2PrintNextLogMessages(System.out);

// test flash

            if ( ztex.flashEnabled() ) {

                System.out.println("SPI Flash");

                System.out.println("Sector size: "+ztex.flashSectorSize());

                System.out.println("Sectors: "+ztex.flashSectors());

//              ztex.printSpiState();

                byte[] buf0 = new byte[ztex.flashSectorSize()];

                byte[] buf1 = new byte[ztex.flashSectorSize()];

                byte[] buf2 = new byte[ztex.flashSectorSize()];

                for (int i=0; i<ztex.flashSectorSize(); i++ ) {

                    buf1[i] = (byte)((i % 251) & 255);

                    buf2[i] = (byte) 255;

                }

                try {

                    System.out.println("Reading backup of sector 7");

                    ztex.flashReadSector(7, 1, buf0);

                    System.out.println("Writing sector 7");

                    ztex.flashWriteSector(7, 1, buf1);

                    System.out.println("Reading sector 7");

                    ztex.flashReadSector(7, 1, buf2);

                    boolean errors = false;

                    for (int i=0; i<ztex.flashSectorSize(); i++ ) {

                        if ( buf1[i] != buf2[i] ) {

                            errors = true;

                            System.out.println("Flash error at " + i + ":  " + (buf1[i] & 255) + " " + (buf2[i] & 255)+"        ");

                        }

                    }

                    if ( ! errors )  System.out.println("No Flash errors");

                    System.out.println("Writing backup of sector 7");

                    ztex.flashWriteSector(7, 1, buf0);

                }

                catch (Exception e) {

                    System.out.println("Flash test failed: " + e.getLocalizedMessage() );

                    ztex.printSpiState();

                }

            }

            else {

                System.out.println("No SPI Flash found or SPI Flash disabled due to reboot");

            }

            ztex.debug2PrintNextLogMessages(System.out);

// speed test

            try {

                ZtexEventHandler eventHandler = new ZtexEventHandler(ztex);

                eventHandler.start();

                ZtexUsbReader bulkReader = new ZtexUsbReader( ztex, 1, false, 16, 256*1024 );

                bulkReader.start(-1);

                long oldByteCount = 0;

                for ( int i=0; i<5; i++ ) {

                    System.out.print("1s speed test: ");

                    try { Thread.sleep(1000); } catch ( InterruptedException e) { }

                    System.out.println(Math.round((bulkReader.byteCount()-oldByteCount)/1000000.0) + " MByte/s");

                    oldByteCount = bulkReader.byteCount();

                }

                if ( !bulkReader.cancelWait(10000) ) System.err.println("Unable to cancel reading");

                if ( !eventHandler.terminate() ) System.err.println("Unable to terminate event handler");

            }

            catch ( Exception e ) {

                System.out.println("Speed test failed: " + e.getLocalizedMessage() );

            }

            ztex.debug2PrintNextLogMessages(System.out);

// read string from stdin and write it to USB device

            try {

                String str = "";

                BufferedReader reader = new BufferedReader( new InputStreamReader( System.in ) );

                while ( ! str.equals("quit") ) {

                    System.out.print("Enter a string or `quit' to exit the program: ");

                    str = reader.readLine();

                    if ( ! str.equals("") ) ztex.echo(str);

                    ztex.debug2PrintNextLogMessages(System.out);

                    System.out.println("");

                }

            }

            catch ( Exception e ) {

                System.out.println("ucecho test failed: " + e.getLocalizedMessage() );

            }

            ztex.debug2PrintNextLogMessages(System.out);

            ztex.dispose();  // this also releases claimed interfaces

        }

        catch (Exception e) {

            System.out.println("Error: "+e.getLocalizedMessage() );

        }

   }

}