Tool_Box/ctrltest.py at master · BitCrane/Tool_Box · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#! /usr/bin/env python3

# Running on Debian (sudo may be required):
#   apt-get update
#   apt-get install python-pip
#   pip install --pre pyusb
#   ./ctrltest.py

import sys
import getopt
from hfusb import usbctrl

def main(argv):
  #
  # usage
  #
  usage  = "usage: ctrltest.py\n"
  usage += "    -f <module>,<fan>,<speed>   set fan speed\n"
  usage += "    -n <name>                   set device name\n"
  usage += "    -o <0/1>                    turn device power off/on\n"
  usage += "    -r [<module>]               reboot into app\n"
  usage += "    -R [<module>]               reboot into loader\n"
  usage += "    -v <module>,<die>,<mvolts>  set die voltage (mV)\n"
  usage += "    -c                          core overview\n"
  usage += "    -e <core>,[<persist>]       enable core\n"
  usage += "    -d <core>,[<persist>]       disable core\n"
  usage += "    -C [<persist>]              clear core map (enable all cores)\n"
  usage += "    -x <core>                   core status\n"
  usage += "    -y <die>                    die status\n"
  usage += "    -z <asic>                   asic status\n"
  #
  # get opt
  #
  try:
    opts, args = getopt.getopt(argv,"hf:n:o:r:R:ce:d:C:x:y:z:")
  except getopt.GetoptError:
    print (usage)
    sys.exit(2)
  #
  # query device
  #
  dev = usbctrl.HFCtrlDevice()
  print (dev.info())
  print (dev.status())
  config = dev.config()
  print (config)
  print (dev.name())
  for module in range(config.modules):
    print (dev.version(module))
    print (dev.serial(module))
    print (dev.power(module))
    print (dev.fan(module))
  #
  # parse args
  #
  for opt, arg in opts:
    if   opt == '-h':
      print (usage)
      sys.exit()
    elif opt == '-f':
      module, fan, speed = arg.split(',')
      dev.fan_set(module, fan, speed)
    elif opt == '-n':
      name = arg
      dev.name_set(name)
    elif opt == '-o':
      pass
    elif opt == '-r': # app
      module = arg
      dev.reboot(module, 0x0000)
    elif opt == '-R': # loader
      module = arg
      dev.reboot(module, 0x0001)
    elif opt == '-f':
      module, die, mvolts = arg.split(',')
      dev.voltage_set(module, die, mvolts)
    elif opt == '-c':
      print (dev.core_overview())
    elif opt == '-e':
      if ',' in arg:
        core, persist = arg.split(',')
      else:
        core = arg
        persist = 0
      dev.core_enable(core, persist)
    elif opt == '-d':
      if ',' in arg:
        core, persist = arg.split(',')
      else:
        core = arg
        persist = 0
      dev.core_disable(core, persist)
    elif opt == '-C':
      persist = arg
      dev.core_clear(persist)
    elif opt == '-x':
      core = arg
      print (dev.core_status(core))
    elif opt == '-y':
      die = arg
      t = dev.core_die_status(die)
      print (t)
      print (t.core(0))
      print (t.core(1))
      print (t.core(2))
    elif opt == '-z':
      asic = arg
      print (dev.core_asic_status(asic))

if __name__ == "__main__":
   main(sys.argv[1:])