1 files changed, 439 insertions, 0 deletions
diff --git a/gr-wxgui/lib/oscope_guts.cc b/gr-wxgui/lib/oscope_guts.cc
new file mode 100644
index 0000000000..3d1f085a6f
--- /dev/null
+++ b/gr-wxgui/lib/oscope_guts.cc
@@ -0,0 +1,439 @@
+/* -*- c++ -*- */
+/*
+ * Copyright 2003,2005,2013 Free Software Foundation, Inc.
+ *
+ * This file is part of GNU Radio
+ *
+ * GNU Radio is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3, or (at your option)
+ * any later version.
+ *
+ * GNU Radio is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with GNU Radio; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 51 Franklin Street,
+ * Boston, MA 02110-1301, USA.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <wxgui/oscope_guts.h>
+#include <stdexcept>
+#include <stdio.h>
+#include <algorithm>
+#include <unistd.h>
+#include <math.h>
+#include <assert.h>
+
+namespace gr {
+  namespace wxgui {
+    
+    /*
+     * Bad performance if it's large, and flaky triggering if it's too small
+     */
+    static const int OUTPUT_RECORD_SIZE = 1024;  // Must be power of 2
+
+    /*
+     * For (slow-updated) STRIPCHART triggering, we make the record
+     * size larger, since we potentially want to be able to "see"
+     * hours of data. This works as long as the update rates to a
+     * STRIPCHART are low, which they generally are--that's rather
+     * what a stripchart is all about!
+     */
+    static const int SCHART_MULT = 8;
+
+
+    static inline int
+    wrap_bi(int buffer_index, int mx)                // wrap buffer index
+    {
+      return buffer_index & (mx - 1);
+    }
+
+    static inline int
+    incr_bi(int buffer_index, int mx)                // increment buffer index
+    {
+      return wrap_bi(buffer_index + 1, mx);
+    }
+
+    static inline int
+    decr_bi(int buffer_index, int mx)                // decrement buffer index
+    {
+      return wrap_bi(buffer_index - 1, mx);
+    }
+
+    oscope_guts::oscope_guts(double sample_rate, gr_msg_queue_sptr msgq)
+      : d_nchannels(1),
+    d_msgq(msgq),
+    d_trigger_mode(TRIG_MODE_AUTO),
+    d_trigger_slope(TRIG_SLOPE_POS),
+    d_trigger_channel(0),
+    d_sample_rate(sample_rate),
+    d_update_rate(20),
+    d_trigger_level(0),
+    d_obi(0),
+    d_state(HOLD_OFF),
+    d_decimator_count(0),
+    d_decimator_count_init(1),
+    d_hold_off_count(0),
+    d_hold_off_count_init(OUTPUT_RECORD_SIZE/2-1),
+    d_pre_trigger_count(0),
+    d_post_trigger_count(0),
+    d_post_trigger_count_init(OUTPUT_RECORD_SIZE/2)
+    {
+      for(int i = 0; i < MAX_CHANNELS; i++)
+        d_buffer[i] = 0;
+
+      for(int i = 0; i < MAX_CHANNELS; i++) {
+        d_buffer[i] = new float[OUTPUT_RECORD_SIZE*SCHART_MULT];
+        for(int j = 0; j < OUTPUT_RECORD_SIZE*SCHART_MULT; j++)
+          d_buffer[i][j] = 0.0;
+      }
+
+      // be sure buffer is full before first write
+      enter_hold_off();
+      update_rate_or_decimation_changed();
+    }
+
+    oscope_guts::~oscope_guts()
+    {
+      for(int i = 0; i < MAX_CHANNELS; i++)
+        delete [] d_buffer[i];
+    }
+
+    // MANIPULATORS
+    void
+    oscope_guts::process_sample(const float *channel_data)
+    {
+      d_decimator_count--;
+      if(d_decimator_count > 0)
+        return;
+
+      d_decimator_count = d_decimator_count_init;
+
+      if(d_trigger_mode != TRIG_MODE_STRIPCHART) {
+        for(int i = 0; i < d_nchannels; i++)
+          d_buffer[i][d_obi] = channel_data[i];  // copy data into buffer
+
+        switch(d_state) {
+        case HOLD_OFF:
+          d_hold_off_count--;
+          if(d_hold_off_count <= 0)
+            enter_look_for_trigger ();
+          break;
+
+        case LOOK_FOR_TRIGGER:
+          if(found_trigger())
+            enter_post_trigger();
+          break;
+
+        case POST_TRIGGER:
+          d_post_trigger_count--;
+          if(d_post_trigger_count <= 0) {
+            write_output_records();
+            enter_hold_off();
+          }
+          break;
+
+        default:
+          assert(0);
+        }
+
+        d_obi = incr_bi(d_obi, OUTPUT_RECORD_SIZE);
+      }
+      else {
+        for(int i = 0; i < d_nchannels; i++) {
+          for(int j = (OUTPUT_RECORD_SIZE*SCHART_MULT)-1; j > 0; j--) {
+            d_buffer[i][j] = d_buffer[i][j-1];
+          }
+          d_buffer[i][0] = channel_data[i];
+        }
+        d_trigger_off = 0;
+        write_output_records();
+      }
+    }
+
+    /*
+     * Functions called on state entry
+     */
+
+    void
+    oscope_guts::enter_hold_off()
+    {
+      d_state = HOLD_OFF;
+      d_hold_off_count = d_hold_off_count_init;
+    }
+
+    void
+    oscope_guts::enter_look_for_trigger()
+    {
+      d_pre_trigger_count = 0;
+      d_state = LOOK_FOR_TRIGGER;
+    }
+
+    void
+    oscope_guts::enter_post_trigger()
+    {
+      d_state = POST_TRIGGER;
+      d_post_trigger_count = d_post_trigger_count_init;
+      //ensure that the trigger offset is no more than than half a sample
+      if(d_trigger_off > .5)
+        d_trigger_off -= 1;
+      else
+        d_post_trigger_count--;
+    }
+
+    // ----------------------------------------------------------------
+    // returns true if trigger found
+
+    bool
+    oscope_guts::found_trigger()
+    {
+      int mx = d_trigger_mode == TRIG_MODE_STRIPCHART ? OUTPUT_RECORD_SIZE*SCHART_MULT :
+        OUTPUT_RECORD_SIZE;
+
+      float prev_sample = d_buffer[d_trigger_channel][decr_bi(d_obi, mx)];
+      float new_sample = d_buffer[d_trigger_channel][d_obi];
+
+      switch(d_trigger_mode) {
+
+      case TRIG_MODE_AUTO: //too many samples without a trigger
+        d_pre_trigger_count++;
+        if(d_pre_trigger_count > OUTPUT_RECORD_SIZE/2)
+          return true;
+
+      case TRIG_MODE_NORM: //look for trigger
+        switch(d_trigger_slope) {
+
+        case TRIG_SLOPE_POS: //trigger point in pos slope?
+          if(new_sample < d_trigger_level || prev_sample >= d_trigger_level)
+            return false;
+          break;
+
+        case TRIG_SLOPE_NEG: //trigger point in neg slope?
+          if(new_sample > d_trigger_level || prev_sample <= d_trigger_level)
+            return false;
+          break;
+        }
+
+        //calculate the trigger offset in % sample
+        d_trigger_off = (d_trigger_level - prev_sample)/(new_sample - prev_sample);
+        return true;
+
+      case TRIG_MODE_FREE: //free run mode, always trigger
+        d_trigger_off = 0;
+        return true;
+
+      default:
+        assert(0);
+        return false;
+      }
+    }
+
+    // ----------------------------------------------------------------
+    // write output records (duh!)
+
+    void
+    oscope_guts::write_output_records()
+    {
+      int mx;
+
+      mx = d_trigger_mode == TRIG_MODE_STRIPCHART ?
+        OUTPUT_RECORD_SIZE*SCHART_MULT : OUTPUT_RECORD_SIZE;
+
+      // if the output queue if full, drop the data like its hot.
+      if(d_msgq->full_p())
+        return;
+      // Build a message to hold the output records
+      gr_message_sptr msg =
+        gr_make_message(0,                           // msg type
+                        d_nchannels,                 // arg1 for other side
+                        mx,                          // arg2 for other side
+                        ((d_nchannels * mx) + 1) * sizeof(float)); // sizeof payload
+
+      float *out = (float *)msg->msg();        // get pointer to raw message buffer
+
+      for(int ch = 0; ch < d_nchannels; ch++) {
+        // note that d_obi + 1 points at the oldest sample in the buffer
+        for(int i = 0; i < mx; i++) {
+          out[i] = d_buffer[ch][wrap_bi(d_obi + 1 + i, mx)];
+        }
+        out += mx;
+      }
+      //Set the last sample as the trigger offset:
+      //  The non gl scope sink will not look at this last sample.
+      //  The gl scope sink will use this last sample as an offset.
+      out[0] = d_trigger_off;
+      d_msgq->handle(msg);                // send the msg
+    }
+
+    // ----------------------------------------------------------------
+
+    bool
+    oscope_guts::set_update_rate(double update_rate)
+    {
+      d_update_rate = std::min(std::max (1./10., update_rate), d_sample_rate);
+      update_rate_or_decimation_changed();
+      return true;
+    }
+
+    bool
+    oscope_guts::set_decimation_count(int decimator_count)
+    {
+      decimator_count = std::max(1, decimator_count);
+      d_decimator_count_init = decimator_count;
+      update_rate_or_decimation_changed();
+      return true;
+    }
+
+    bool
+    oscope_guts::set_sample_rate(double sample_rate)
+    {
+      d_sample_rate = sample_rate;
+      return set_update_rate(update_rate());
+    }
+
+    void
+    oscope_guts::update_rate_or_decimation_changed()
+    {
+      d_hold_off_count_init =
+        (int)rint(d_sample_rate / d_update_rate / d_decimator_count_init);
+    }
+
+    bool
+    oscope_guts::set_trigger_channel(int channel)
+    {
+      if(channel >= 0 && channel < d_nchannels) {
+        d_trigger_channel = channel;
+        trigger_changed ();
+        return true;
+      }
+
+      return false;
+    }
+
+    bool
+    oscope_guts::set_trigger_mode(trigger_mode mode)
+    {
+      d_trigger_mode = mode;
+      trigger_changed();
+      return true;
+    }
+
+    bool
+    oscope_guts::set_trigger_slope(trigger_slope slope)
+    {
+      d_trigger_slope = slope;
+      trigger_changed();
+      return true;
+    }
+
+    bool
+    oscope_guts::set_trigger_level(double trigger_level)
+    {
+      d_trigger_level = trigger_level;
+      trigger_changed();
+      return true;
+    }
+
+    bool
+    oscope_guts::set_trigger_level_auto()
+    {
+      // find the level 1/2 way between the min and the max
+
+      float min_v = d_buffer[d_trigger_channel][0];
+      float max_v = d_buffer[d_trigger_channel][0];
+
+      for(int i = 1; i < OUTPUT_RECORD_SIZE; i++) {
+        min_v = std::min (min_v, d_buffer[d_trigger_channel][i]);
+        max_v = std::max (max_v, d_buffer[d_trigger_channel][i]);
+      }
+      return set_trigger_level((min_v + max_v) * 0.5);
+    }
+
+    bool
+    oscope_guts::set_num_channels(int nchannels)
+    {
+      if(nchannels > 0 && nchannels <= MAX_CHANNELS) {
+        d_nchannels = nchannels;
+        return true;
+      }
+      return false;
+    }
+
+    void
+    oscope_guts::trigger_changed()
+    {
+      enter_look_for_trigger();
+    }
+
+    // ACCESSORS
+
+    int
+    oscope_guts::num_channels() const
+    {
+      return d_nchannels;
+    }
+
+    double
+    oscope_guts::sample_rate() const
+    {
+      return d_sample_rate;
+    }
+
+    double
+    oscope_guts::update_rate() const
+    {
+      return d_update_rate;
+    }
+
+    int
+    oscope_guts::get_decimation_count() const
+    {
+      return d_decimator_count_init;
+    }
+
+    int
+    oscope_guts::get_trigger_channel() const
+    {
+      return d_trigger_channel;
+    }
+
+    trigger_mode
+    oscope_guts::get_trigger_mode() const
+    {
+      return d_trigger_mode;
+    }
+
+    trigger_slope
+    oscope_guts::get_trigger_slope() const
+    {
+      return d_trigger_slope;
+    }
+
+    double
+    oscope_guts::get_trigger_level() const
+    {
+      return d_trigger_level;
+    }
+
+    int
+    oscope_guts::get_samples_per_output_record() const
+    {
+      int mx;
+
+      mx = OUTPUT_RECORD_SIZE;
+      if(d_trigger_mode == TRIG_MODE_STRIPCHART) {
+        mx = OUTPUT_RECORD_SIZE*SCHART_MULT;
+      }
+      return mx;
+    }
+
+  } /* namespace wxgui */
+} /* namespace gr */