hantek_6022be.c

   1 /*
   2  * This file is part of the sigrok-firmware-fx2lafw project.
   3  *
   4  * Copyright (C) 2009 Ubixum, Inc.
   5  * Copyright (C) 2015 Jochen Hoenicke
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include <fx2macros.h>
  22 #include <fx2ints.h>
  23 #include <autovector.h>
  24 #include <delay.h>
  25 #include <setupdat.h>
  26
  27 #define SET_ANALOG_MODE()
  28
  29 /* Toggle the 1kHz calibration pin, only accurate up to ca. 8MHz. */
  30 #define TOGGLE_CALIBRATION_PIN() PA7 = !PA7
  31
  32 #define LED_CLEAR() PC0 = 1; PC1 = 1;
  33 #define LED_GREEN() PC0 = 1; PC1 = 0;
  34 #define LED_RED()   PC0 = 0; PC1 = 1;
  35
  36 /* Change to support as many interfaces as you need. */
  37 static BYTE altiface = 0;
  38
  39 static volatile WORD ledcounter = 0;
  40
  41 static volatile __bit dosud = FALSE;
  42 static volatile __bit dosuspend = FALSE;
  43
  44 extern __code BYTE highspd_dscr;
  45 extern __code BYTE fullspd_dscr;
  46
  47 void resume_isr(void) __interrupt RESUME_ISR
  48 {
  49         CLEAR_RESUME();
  50 }
  51
  52 void sudav_isr(void) __interrupt SUDAV_ISR
  53 {
  54         dosud = TRUE;
  55         CLEAR_SUDAV();
  56 }
  57
  58 void usbreset_isr(void) __interrupt USBRESET_ISR
  59 {
  60         handle_hispeed(FALSE);
  61         CLEAR_USBRESET();
  62 }
  63
  64 void hispeed_isr(void) __interrupt HISPEED_ISR
  65 {
  66         handle_hispeed(TRUE);
  67         CLEAR_HISPEED();
  68 }
  69
  70 void suspend_isr(void) __interrupt SUSPEND_ISR
  71 {
  72         dosuspend = TRUE;
  73         CLEAR_SUSPEND();
  74 }
  75
  76 void timer2_isr(void) __interrupt TF2_ISR
  77 {
  78         TOGGLE_CALIBRATION_PIN();
  79
  80         if (ledcounter && (--ledcounter == 0))
  81                 LED_CLEAR();
  82
  83         TF2 = 0;
  84 }
  85
  86 /*
  87  * This sets three bits for each channel, one channel at a time.
  88  * For channel 0 we want to set bits 5, 6 & 7
  89  * For channel 1 we want to set bits 2, 3 & 4
  90  *
  91  * We convert the input values that are strange due to original
  92  * firmware code into the value of the three bits as follows:
  93  *
  94  * val -> bits
  95  * 1  -> 010b
  96  * 2  -> 001b
  97  * 5  -> 000b
  98  * 10 -> 011b
  99  *
 100  * The third bit is always zero since there are only four outputs connected
 101  * in the serial selector chip.
 102  *
 103  * The multiplication of the converted value by 0x24 sets the relevant bits in
 104  * both channels and then we mask it out to only affect the channel currently
 105  * requested.
 106  */
 107 static BOOL set_voltage(BYTE channel, BYTE val)
 108 {
 109         BYTE bits, mask;
 110
 111         switch (val) {
 112         case 1:
 113                 bits = 0x24 * 2;
 114                 break;
 115         case 2:
 116                 bits = 0x24 * 1;
 117                 break;
 118         case 5:
 119                 bits = 0x24 * 0;
 120                 break;
 121         case 10:
 122                 bits = 0x24 * 3;
 123                 break;
 124         default:
 125                 return FALSE;
 126         }
 127
 128         mask = (channel) ? 0xe0 : 0x1c;
 129         IOC = (IOC & ~mask) | (bits & mask);
 130
 131         return TRUE;
 132 }
 133
 134 static BOOL set_numchannels(BYTE numchannels)
 135 {
 136         if (numchannels == 1 || numchannels == 2) {
 137                 BYTE fifocfg = 7 + numchannels;
 138                 EP2FIFOCFG = fifocfg;
 139                 EP6FIFOCFG = fifocfg;
 140                 return TRUE;
 141         }
 142
 143         return FALSE;
 144 }
 145
 146 static void clear_fifo(void)
 147 {
 148         GPIFABORT = 0xff;
 149         SYNCDELAY3;
 150         FIFORESET = 0x80;
 151         SYNCDELAY3;
 152         FIFORESET = 0x82;
 153         SYNCDELAY3;
 154         FIFORESET = 0x86;
 155         SYNCDELAY3;
 156         FIFORESET = 0;
 157 }
 158
 159 static void stop_sampling(void)
 160 {
 161         GPIFABORT = 0xff;
 162         SYNCDELAY3;
 163         INPKTEND = (altiface == 0) ? 6 : 2;
 164 }
 165
 166 static void start_sampling(void)
 167 {
 168         int i;
 169
 170         SET_ANALOG_MODE();
 171
 172         clear_fifo();
 173
 174         for (i = 0; i < 1000; i++);
 175
 176         while (!(GPIFTRIG & 0x80))
 177                 ;
 178
 179         SYNCDELAY3;
 180         GPIFTCB1 = 0x28;
 181         SYNCDELAY3;
 182         GPIFTCB0 = 0;
 183         GPIFTRIG = (altiface == 0) ? 6 : 4;
 184
 185         /* Set green LED, don't clear LED afterwards (ledcounter = 0). */
 186         LED_GREEN();
 187         ledcounter = 0;
 188 }
 189
 190 static void select_interface(BYTE alt)
 191 {
 192         const BYTE *pPacketSize = \
 193                 ((USBCS & bmHSM) ? &highspd_dscr : &fullspd_dscr)
 194                 + (9 + (16 * alt) + 9 + 4);
 195
 196         altiface = alt;
 197
 198         if (alt == 0) {
 199                 /* Bulk on EP6. */
 200                 EP2CFG = 0x00;
 201                 EP6CFG = 0xe0;
 202                 EP6GPIFFLGSEL = 1;
 203                 EP6AUTOINLENL = pPacketSize[0];
 204                 EP6AUTOINLENH = pPacketSize[1];
 205         } else {
 206                 /* Iso on EP2. */
 207                 EP2CFG = 0xd8;
 208                 EP6CFG = 0x00;
 209                 EP2GPIFFLGSEL = 1;
 210                 EP2AUTOINLENL = pPacketSize[0];
 211                 EP2AUTOINLENH = pPacketSize[1] & 0x7;
 212                 EP2ISOINPKTS = (pPacketSize[1] >> 3) + 1;
 213         }
 214 }
 215
 216 static const struct samplerate_info {
 217         BYTE rate;
 218         BYTE wait0;
 219         BYTE wait1;
 220         BYTE opc0;
 221         BYTE opc1;
 222         BYTE out0;
 223         BYTE ifcfg;
 224 } samplerates[] = {
 225         { 48, 0x80,   0, 3, 0, 0x00, 0xea },
 226         { 30, 0x80,   0, 3, 0, 0x00, 0xaa },
 227         { 24,    1,   0, 2, 1, 0x40, 0xca },
 228         { 16,    1,   1, 2, 0, 0x40, 0xca },
 229         { 12,    2,   1, 2, 0, 0x40, 0xca },
 230         {  8,    3,   2, 2, 0, 0x40, 0xca },
 231         {  4,    6,   5, 2, 0, 0x40, 0xca },
 232         {  2,   12,  11, 2, 0, 0x40, 0xca },
 233         {  1,   24,  23, 2, 0, 0x40, 0xca },
 234         { 50,   48,  47, 2, 0, 0x40, 0xca },
 235         { 20,  120, 119, 2, 0, 0x40, 0xca },
 236         { 10,  240, 239, 2, 0, 0x40, 0xca },
 237 };
 238
 239 static BOOL set_samplerate(BYTE rate)
 240 {
 241         BYTE i = 0;
 242
 243         while (samplerates[i].rate != rate) {
 244                 i++;
 245                 if (i == sizeof(samplerates) / sizeof(samplerates[0]))
 246                         return FALSE;
 247         }
 248
 249         IFCONFIG = samplerates[i].ifcfg;
 250
 251         AUTOPTRSETUP = 7;
 252         AUTOPTRH2 = 0xE4; /* 0xE400: GPIF waveform descriptor 0. */
 253         AUTOPTRL2 = 0x00;
 254
 255         /*
 256          * The program for low-speed, e.g. 1 MHz, is:
 257          * wait 24, CTLx=0, FIFO
 258          * wait 23, CTLx=1
 259          * jump 0, CTLx=1
 260          *
 261          * The program for 24 MHz is:
 262          * wait 1, CTLx=0, FIFO
 263          * jump 0, CTLx=1
 264          *
 265          * The program for 30/48 MHz is:
 266          * jump 0, CTLx=Z, FIFO, LOOP
 267          *
 268          * (CTLx is device-dependent, could be e.g. CTL0 or CTL2.)
 269          */
 270
 271         /* LENGTH / BRANCH 0-7 */
 272         EXTAUTODAT2 = samplerates[i].wait0;
 273         EXTAUTODAT2 = samplerates[i].wait1;
 274         EXTAUTODAT2 = 1;
 275         EXTAUTODAT2 = 0;
 276         EXTAUTODAT2 = 0;
 277         EXTAUTODAT2 = 0;
 278         EXTAUTODAT2 = 0;
 279         EXTAUTODAT2 = 0;
 280
 281         /* OPCODE 0-7 */
 282         EXTAUTODAT2 = samplerates[i].opc0;
 283         EXTAUTODAT2 = samplerates[i].opc1;
 284         EXTAUTODAT2 = 1; /* DATA=0 DP=1 */
 285         EXTAUTODAT2 = 0;
 286         EXTAUTODAT2 = 0;
 287         EXTAUTODAT2 = 0;
 288         EXTAUTODAT2 = 0;
 289         EXTAUTODAT2 = 0;
 290
 291         /* OUTPUT 0-7 */
 292         EXTAUTODAT2 = samplerates[i].out0;
 293         EXTAUTODAT2 = 0x44; /* OE2=1, CTL2=1 */
 294         EXTAUTODAT2 = 0x44; /* OE2=1, CTL2=1 */
 295         EXTAUTODAT2 = 0;
 296         EXTAUTODAT2 = 0;
 297         EXTAUTODAT2 = 0;
 298         EXTAUTODAT2 = 0;
 299         EXTAUTODAT2 = 0;
 300
 301         /* LOGIC FUNCTION 0-7 */
 302         EXTAUTODAT2 = 0;
 303         EXTAUTODAT2 = 0;
 304         EXTAUTODAT2 = 0;
 305         EXTAUTODAT2 = 0;
 306         EXTAUTODAT2 = 0;
 307         EXTAUTODAT2 = 0;
 308         EXTAUTODAT2 = 0;
 309         EXTAUTODAT2 = 0;
 310
 311         for (i = 0; i < 96; i++)
 312                 EXTAUTODAT2 = 0;
 313
 314         return TRUE;
 315 }
 316
 317 /* Set *alt_ifc to the current alt interface for ifc. */
 318 BOOL handle_get_interface(BYTE ifc, BYTE *alt_ifc)
 319 {
 320         (void)ifc;
 321
 322         *alt_ifc = altiface;
 323
 324         return TRUE;
 325 }
 326
 327 /*
 328  * Return TRUE if you set the interface requested.
 329  *
 330  * Note: This function should reconfigure and reset the endpoints
 331  * according to the interface descriptors you provided.
 332  */
 333 BOOL handle_set_interface(BYTE ifc,BYTE alt_ifc)
 334 {
 335         if (ifc == 0)
 336                 select_interface(alt_ifc);
 337
 338         return TRUE;
 339 }
 340
 341 BYTE handle_get_configuration(void)
 342 {
 343         /* We only support configuration 0. */
 344         return 0;
 345 }
 346
 347 BOOL handle_set_configuration(BYTE cfg)
 348 {
 349         /* We only support configuration 0. */
 350         (void)cfg;
 351
 352         return TRUE;
 353 }
 354
 355 BOOL handle_vendorcommand(BYTE cmd)
 356 {
 357         stop_sampling();
 358
 359         /* Set red LED, clear after timeout. */
 360         LED_RED();
 361         ledcounter = 1000;
 362
 363         /* Clear EP0BCH/L for each valid command. */
 364         if (cmd >= 0xe0 && cmd <= 0xe4) {
 365                 EP0BCH = 0;
 366                 EP0BCL = 0;
 367                 while (EP0CS & bmEPBUSY);
 368         }
 369
 370         switch (cmd) {
 371         case 0xe0:
 372         case 0xe1:
 373                 set_voltage(cmd - 0xe0, EP0BUF[0]);
 374                 return TRUE;
 375         case 0xe2:
 376                 set_samplerate(EP0BUF[0]);
 377                 return TRUE;
 378         case 0xe3:
 379                 if (EP0BUF[0] == 1)
 380                         start_sampling();
 381                 return TRUE;
 382         case 0xe4:
 383                 set_numchannels(EP0BUF[0]);
 384                 return TRUE;
 385         }
 386
 387         return FALSE; /* Not handled by handlers. */
 388 }
 389
 390 static void init(void)
 391 {
 392         EP4CFG = 0;
 393         EP8CFG = 0;
 394
 395         SET_ANALOG_MODE();
 396
 397         /* In idle mode tristate all outputs. */
 398         GPIFIDLECTL = 0x00; /* Don't enable CTL0-5 outputs. */
 399         GPIFCTLCFG = 0x80; /* TRICTL=1. CTL0-2: CMOS outputs, tri-statable. */
 400         GPIFWFSELECT = 0x00;
 401         GPIFREADYSTAT = 0x00;
 402
 403         stop_sampling();
 404
 405         set_voltage(0, 1);
 406         set_voltage(1, 1);
 407         set_samplerate(1);
 408         set_numchannels(2);
 409         select_interface(0);
 410 }
 411
 412 static void main(void)
 413 {
 414         /* Save energy. */
 415         SETCPUFREQ(CLK_12M);
 416
 417         init();
 418
 419         /* Set up interrupts. */
 420         USE_USB_INTS();
 421
 422         ENABLE_SUDAV();
 423         ENABLE_USBRESET();
 424         ENABLE_HISPEED();
 425         ENABLE_SUSPEND();
 426         ENABLE_RESUME();
 427
 428         /* Global (8051) interrupt enable. */
 429         EA = 1;
 430
 431         /* Init timer2. */
 432         RCAP2L = -500 & 0xff;
 433         RCAP2H = (-500 & 0xff00) >> 8;
 434         T2CON = 0;
 435         ET2 = 1;
 436         TR2 = 1;
 437
 438         RENUMERATE();
 439
 440         PORTECFG = 0;
 441         PORTCCFG = 0;
 442         PORTACFG = 0;
 443         OEE = 0xff;
 444         OEC = 0xff;
 445         OEA = 0xff;
 446
 447         while (TRUE) {
 448                 if (dosud) {
 449                         dosud = FALSE;
 450                         handle_setupdata();
 451                 }
 452
 453                 if (dosuspend) {
 454                         dosuspend = FALSE;
 455                         do {
 456                                 /* Make sure ext wakeups are cleared. */
 457                                 WAKEUPCS |= bmWU | bmWU2;
 458                                 SUSPEND = 1;
 459                                 PCON |= 1;
 460                                 __asm
 461                                 nop
 462                                 nop
 463                                 nop
 464                                 nop
 465                                 nop
 466                                 nop
 467                                 nop
 468                                 __endasm;
 469                         } while (!remote_wakeup_allowed && REMOTE_WAKEUP());
 470
 471                         /* Resume (TRM 6.4). */
 472                         if (REMOTE_WAKEUP()) {
 473                                 delay(5);
 474                                 USBCS |= bmSIGRESUME;
 475                                 delay(15);
 476                                 USBCS &= ~bmSIGRESUME;
 477                         }
 478                 }
 479         }
 480 }