Skip to content
Snippets Groups Projects
Select Git revision
  • f1adb477e526cde0fff42b998fc7a72acce11f88
  • main default protected
  • experiment-step-sharing
  • experiment-color-coded-graph
  • experiment-separated-graph
5 results

utils.py

Blame
  • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    spi_switch_Unb2c.py 7.79 KiB
    """
    Copyright 2021 Stichting Nederlandse Wetenschappelijk Onderzoek Instituten,
    ASTRON Netherlands Institute for Radio Astronomy
    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.
    
    
    Created: 2021-05-19 by Leon Hiemstra, edited by Gijs
    file used to read and write to switch registers
    
    """
    
    import time
    import sys
    import spidev
    
    DEBUG=False
    # We only have SPI bus 0 available to us on the Pi
    bus = 0
    
    #Device is the chip select pin. Set to 0 or 1, depending on the connections
    device = 0
    
    # Enable SPI
    spi = spidev.SpiDev()
    
    # Open a connection to a specific bus and device (chip select pin)
    spi.open(bus, device)
    
    # Set SPI speed and mode
    spi.max_speed_hz = 1000000
    #spi.max_speed_hz = 50000
    spi.mode = 1
    
    
    cmd_normal_read = 0x60
    cmd_normal_write = 0x61
    
    
    def read_register(addr):
        cmd =  cmd_normal_read
        if 0:
            spi.writebytes([cmd,addr])
            ret = spi.readbytes(2)
        else:
            ret = spi.xfer2([cmd,addr,0, 0, 0, 0])
        if DEBUG:
            stri = 'read_register  0x{:0>2x} = 0x{:0>2x}, 0x{:0>2x}'.format(addr, ret[2], ret[3])
            print(stri)
        return ret[2:]
    
    def write_register(addr, data):
        cmd =  cmd_normal_write
        if 0:
            spi.writebytes([cmd, addr, data])
            ret = spi.readbytes(2)
        else:
            ret = spi.xfer2([cmd,addr,data])
        if DEBUG:
            stri = 'write_register 0x{:0>2x} = 0x{:0>2x}'.format(addr, data)
            print(stri)
    
    def read_switch(page, addr, pr_stri = True):
        stri = '<< read switch from page: 0x{0:0>2x}, address: 0x{1:0>2x}'.format(page, addr)
        ret = spi.xfer2([cmd_normal_write, 0xff, page])
        ret = spi.xfer2([cmd_normal_read, addr, 0, 0, 0, 0])
        ret = read_register(0xfe)
        if (ret[2] & 0xf0) == 0xa0:
            ret = read_register(0xf0)
            ret.reverse()
            if pr_stri:
                stri += " data 0x"
                for byte in ret:
                    stri += "{:0>2x}".format(byte) 
                print(stri)
        else:
            print("read error")
        return ret
    
    def write_switch_bytes(page, addr, data, pr_stri = True):
        stri = '> write switch from page: 0x{0:0>2x}, address: 0x{1:0>2x} data 0x'.format(page, addr)
        for byte_cnt in range(len(data)):
            add_stri = "{0:0>2x}".format(data[-1-byte_cnt])
            stri += add_stri
        if pr_stri:
            print(stri)
        read_register(0xfe)
        ret = spi.xfer2([cmd_normal_write, 0xff, page])
        wr_bytes = [cmd_normal_write, addr]
        wr_bytes.extend(data)
        ret = spi.xfer2(wr_bytes)
        ret = read_register(0xfe)[2]
        if ret != 0:
            print("write error, not enough words written")
        if DEBUG:
            read_register(0xfe)
            read_register(0xfe)
            read_register(addr)
    
    def read_link_status(ports=16):
        print("links status register")
        ret = read_switch(0x01,0x00, pr_stri=False)
        stri = "|15 |14 |13 |12 |11 |10 | 9 | 8 | 7 | 6 | 5 | 4 | 3 | 2 | 1 | 0 |"
        print(stri)
        stri = "| "
        for cnt_port in range(16):
            if ret[3]  & (0x8000 >> cnt_port) :
                stri += "U | "
            else:
                stri += "  | "
        print(stri)
        for cnt in range(ports):
            stri = "Port status phy nr {} ".format(cnt)
            ret = read_switch(0x01,0x20+cnt, pr_stri = False)
            if ret[1] & 0x01:
                stri += "link up "
            else:
                stri += "link down "
            if ret[1] & 0x02:
                stri += "dupplex "
            else:
                stri += "simplex "
            if ret[1] & 0x4:
                stri += " 100M "
            elif ret[1] & 0x8:
                stri += "1000M "
            else:
                stri += "  10M "
            ret = read_switch(0x10+cnt,0x28, pr_stri = False)
            if ret[1] & 0x01:
                stri += "SGMII  "
            else:
                stri += "SERDES "
            if ret[2] & 0x08:
                stri += "Tx: Er "
            if ret[2] & 0x04:
                stri += "Rx: Er "
            if ret[2] & 0x40:
                stri += "Rx FIFO Er "
            print(stri)
    # Read phy registister status
        for cnt in range(4):
            ret = read_switch(0x80+cnt,0x02, pr_stri = False)
            stri = " Phy status ch {} ".format(cnt)
            if ret[1] & 0x20 :
                stri += "AN_complete "
            if ret[1] & 0x04:
                stri += "link up "
                if ret[1] & 0x10:
                    stri += "remote fault "
    #            print(stri)
                ret = read_switch(0x80+cnt,0x14, pr_stri = False)
                stri += " link status ch{} ".format(cnt)
                if ret[2] & 0x10:
                    stri += "remote status is good "
                else:
                    stri += "remote status is NOK "
                if ret[2] & 0x20:
                    stri += "local status is good "
                else:
                    stri += "local status is NOK "
                if ret[2] & 0x40:
                    stri += "local master"
                else:
                    stri += "local slave"
            else:
                stri += " No link "
            print(stri)
            if 0:
                write_switch_bytes(0x80+cnt, 0x3c, [0x35, 0x08], pr_stri = False)
                ret = read_switch(0x80+cnt,0x3e)
        if 0:
            print("Drop packet count register")
            for cnt in range(16):
    #        read_switch(0x41,0x80+2*cnt)
                read_switch(0x0,0x0 + cnt)
            print("Port State Override")
            for cnt in range(16):
        #        read_switch(0x41,0x80+2*cnt)
                read_switch(0x0,0x60 + cnt)
            read_switch(0x0,0x10)
            read_switch(0x0,0x20)
        if 1:
            print("Receive count register")
            for cnt in range(16):
                # Set to receive packet count
                write_switch_bytes(0x10+cnt, 0x20, [0xD0, 0x09], pr_stri = False) 
            time.sleep(0.5)
            for cnt in range(16):
                read_switch(0x10+cnt,0x2e)
            
    
    
    
    if len(sys.argv) < 2:
        print(sys.argv)
        print("write and read led register")
        write_switch_bytes(0x00, 0x24, [0x20, 0x02]) #LSB first
        read_switch(0x00,0x24)
        print("write and read jumbo register")
        write_switch_bytes(0x40, 0x01, [0xff, 0xff, 0x00, 0x00])
        read_switch(0x40,0x01)
        print("strap resistors")
        read_switch(0x01,0x70)
        read_link_status(4)
    elif sys.argv[1] == "stat":
        read_link_status(16)
    elif sys.argv[1] == "set":
        # Extra setting for the switch, not needed, bonus settings
        if 0:
            print("write and read led register")
            write_switch_bytes(0x00, 0x24, [0x20, 0x02]) #LSB first
            read_switch(0x00,0x24)
            print("write and read jumbo register")
            write_switch_bytes(0x40, 0x01, [0xff, 0xff, 0x00, 0x00])
            read_switch(0x40,0x01)
            print("strap resistors")
            read_switch(0x01,0x70)
        # required setting for the switch
        if 1:
    #        for ch_cnt in range(16):
    #           print("write and read SGMII register CH0, fifo size max")
    #           write_switch_bytes(0x10+ch_cnt, 0x24, [0x44, 0x00])
    #           read_switch(0x10+ch_cnt,0x24)
            speed_100Mbit = False
            speed_1000Mbit = False
            for ch_cnt in range(4):
                print("Set PHY ch 0 and read back to 10 Mbit 0x01 100Mbit 0x21")
                write_switch_bytes(0x80 + ch_cnt, 0x00, [(0x00 | (speed_1000Mbit << 6)), (0x01 | (speed_100Mbit << 5))])
                write_switch_bytes(0x00, 0x60 + ch_cnt, [(0x83 | (speed_100Mbit << 2) | (speed_1000Mbit << 3))]) # fix PHY ports to 10 Mbit
    # Set the FPGA links to disable Auto negatiation
        for cnt in [15, 13, 11, 9]: # only ETH0 interface
            write_switch_bytes(0x00, 0x60+cnt, [0x8B]) #Fix FPGA links
    else:
        print("spi_switch_Unb2c stat for status")
        print("spi_switch_Unb2c set  to set registers")
    
    spi.close()