scopes: Add include/scope.inc.

[sigrok-firmware-fx2lafw.git] / hantek_6022be.c

diff --git a/hantek_6022be.c b/hantek_6022be.c
index 93665b8d41879ec3a5fc022ad37e8c38b7d3cb32..69ba9586235e79d42c6e268bf123eb8e5d9ee9ba 100644 (file)
--- a/hantek_6022be.c
+++ b/hantek_6022be.c
@@ -26,6 +26,10 @@
 
 #define SET_ANALOG_MODE()
 
+#define SET_COUPLING(x)
+
+#define SET_CALIBRATION_PULSE(x)
+
 /* Toggle the 1kHz calibration pin, only accurate up to ca. 8MHz. */
 #define TOGGLE_CALIBRATION_PIN() PA7 = !PA7
 
@@ -33,6 +37,8 @@
 #define LED_GREEN() PC0 = 1; PC1 = 0;
 #define LED_RED()   PC0 = 0; PC1 = 1;
 
+#define TIMER2_VAL 500
+
 /* CTLx pin index (IFCLK, ADC clock input). */
 #define CTL_BIT 2
 
@@ -137,6 +143,25 @@ static BOOL set_voltage(BYTE channel, BYTE val)
        return TRUE;
 }
 
+/**
+ * Each LSB in the nibble of the byte controls the coupling per channel.
+ *
+ * Setting PE3 disables AC coupling capacitor on CH0.
+ * Setting PE0 disables AC coupling capacitor on CH1.
+ */
+static void set_coupling(BYTE coupling_cfg)
+{
+       if (coupling_cfg & 0x01)
+               IOE |= 0x08;
+       else
+               IOE &= ~0x08;
+
+       if (coupling_cfg & 0x10)
+               IOE |= 0x01;
+       else
+               IOE &= ~0x01;
+}
+
 static BOOL set_numchannels(BYTE numchannels)
 {
        if (numchannels == 1 || numchannels == 2) {
@@ -320,6 +345,30 @@ static BOOL set_samplerate(BYTE rate)
        return TRUE;
 }
 
+static BOOL set_calibration_pulse(BYTE fs)
+{
+       switch (fs) {
+       case 0:         // 100Hz
+               RCAP2L = -10000 & 0xff;
+               RCAP2H = (-10000 & 0xff00) >> 8;
+               return TRUE;
+       case 1:         // 1kHz
+               RCAP2L = -1000 & 0xff;
+               RCAP2H = (-1000 & 0xff00) >> 8;
+               return TRUE;
+       case 10:        // 1kHz
+               RCAP2L = (BYTE)(-100 & 0xff);
+               RCAP2H = 0xff;
+               return TRUE;
+       case 50:        // 50kHz
+               RCAP2L = (BYTE)(-20 & 0xff);
+               RCAP2H = 0xff;
+               return TRUE;
+       default:
+               return FALSE;
+       }
+}
+
 /* Set *alt_ifc to the current alt interface for ifc. */
 BOOL handle_get_interface(BYTE ifc, BYTE *alt_ifc)
 {
@@ -367,7 +416,7 @@ BOOL handle_vendorcommand(BYTE cmd)
        ledcounter = 1000;
 
        /* Clear EP0BCH/L for each valid command. */
-       if (cmd >= 0xe0 && cmd <= 0xe4) {
+       if (cmd >= 0xe0 && cmd <= 0xe6) {
                EP0BCH = 0;
                EP0BCL = 0;
                while (EP0CS & bmEPBUSY);
@@ -388,6 +437,12 @@ BOOL handle_vendorcommand(BYTE cmd)
        case 0xe4:
                set_numchannels(EP0BUF[0]);
                return TRUE;
+       case 0xe5:
+               SET_COUPLING(EP0BUF[0]);
+               return TRUE;
+       case 0xe6:
+               SET_CALIBRATION_PULSE(EP0BUF[0]);
+               return TRUE;
        }
 
        return FALSE; /* Not handled by handlers. */
@@ -435,8 +490,8 @@ static void main(void)
        EA = 1;
 
        /* Init timer2. */
-       RCAP2L = -500 & 0xff;
-       RCAP2H = (-500 & 0xff00) >> 8;
+       RCAP2L = -TIMER2_VAL & 0xff;
+       RCAP2H = (-TIMER2_VAL & 0xff00) >> 8;
        T2CON = 0;
        ET2 = 1;
        TR2 = 1;