Changeset 207 for trunk/src

Timestamp:

08/15/2009 08:56:54 PM (3 years ago)

Author:

guyru

Message:

highlight pie plot segments when moving over them

Location:

trunk/src

Files:

• simple_pie_plot.cpp (modified) (6 diffs)
• simple_pie_plot.h (modified) (1 diff)

trunk/src/simple_pie_plot.cpp

@@ -22 +22 @@
 #include <boost/foreach.hpp>
 #include <memory>
-#include <iostream>
 #include <cmath>
 using namespace std;
 using namespace simple_pie_plot;
 
-#define PI 3.14159265
+const double PI = 4.0 * atan(1.0);
 
 /**
@@ -59 +58 @@
                                 : wxPanel(parent, id, pos, size, style, name)
 {
+        m_highlight = -1;
         Connect(this->GetId(), wxEVT_PAINT, wxPaintEventHandler(SimplePiePlot::OnPaint));
         Connect(this->GetId(), wxEVT_SIZE, wxSizeEventHandler(SimplePiePlot::OnResize));
-        
+
+        Connect(this->GetId(), wxEVT_MOTION, wxMouseEventHandler(SimplePiePlot::OnMouseMove));
+        Connect(this->GetId(), wxEVT_LEAVE_WINDOW, wxMouseEventHandler(SimplePiePlot::OnMouseLeaveWindow));
 }
 
@@ -68 +70 @@
         wxPaintDC pdc(this);
         auto_ptr<wxGraphicsContext> dc(wxGraphicsContext::Create(pdc));
-        
+        wxBrush color_brush;
         int width, height;
+
         GetClientSize(&width, &height);
-
-        wxBrush color_brush;
-        int i = 0;
-        double item_x, item_y, item_ratio, item_height;
 
         // radius should be set so it fits exactly
         double radius = (width>height? height : width)/2.0;
-        double start_angle = 0;
-        double end_angle;
 
-        BOOST_FOREACH(double d, m_data) {
+        for (int i = 1; i<m_angles.size(); i++) {
                 // create brush
-                color_brush.SetColour(GetSegmentColor(i, false));
+                color_brush.SetColour(GetSegmentColor(i-1));
                 dc->SetBrush(color_brush);
 
-                item_ratio = m_data_total ? d/m_data_total : 0;
-                end_angle = start_angle + 2*PI* (item_ratio);
-
-                DrawPieSlice(width/2.0, height/2.0, radius, start_angle,
-                             end_angle, dc.get());
-
-                start_angle = end_angle;
-                i++;
+                DrawPieSlice(width/2.0, height/2.0, radius, m_angles[i-1],
+                             m_angles[i], dc.get());
         }
 }
@@ -105 +96 @@
 void SimplePiePlot::SetData(vector<double> d)
 {
-        //m_data.assign(d.begin(), d.end());
+        m_data.clear();
         m_data_total = 0;
         BOOST_FOREACH(double tmp, d) {
@@ -112 +103 @@
                         m_data.push_back(tmp);
         }
+
+        m_angles.clear();
+        double new_angle = 0;
+        m_angles.push_back(new_angle);
+        BOOST_FOREACH(double tmp, m_data) {
+                new_angle = new_angle + 2 * PI * (tmp/m_data_total);
+                m_angles.push_back(new_angle);
+        }
 }
 
-wxColour SimplePiePlot::GetSegmentColor(int i, bool highlight)
+wxColour SimplePiePlot::GetSegmentColor(int i)
 {
         double hue_step = 1.0/(m_data.size());
         double hue = i * hue_step;
-        double value = highlight ? 1.0 : 0.8;
+        double value = m_highlight == i ? 1.0 : 0.8;
         wxColour color;
         wxImage::RGBValue rgb;
@@ -127 +126 @@
         return color;
 }
+
+void SimplePiePlot::OnMouseMove(wxMouseEvent& event)
+{
+        //check if the move is even in the plot
+        int width, height;
+        GetClientSize(&width, &height);
+        // radius should be set so it fits exactly
+        const double radius = (width>height? height : width)/2.0;
+
+        const double dist_mouse = (width/2.0-event.m_x)*(width/2.0-event.m_x) + 
+                            (height/2.0-event.m_y)*(height/2.0-event.m_y);
+        if (dist_mouse > (radius*radius)) { 
+                ClearHighlight();
+                return;
+        }
+
+        //angles are clockwise from the x-axis
+        double angle = acos((event.m_x-width/2.0)/sqrt(dist_mouse));
+        if (event.m_y < height/2.0)
+                angle = 2*PI - angle;
+        int i;
+        for (i = 1; i<m_angles.size(); i++) {
+                if (angle<=m_angles[i])
+                        break;
+        }
+        Highlight(i-1);
+}
+
+void SimplePiePlot::OnMouseLeaveWindow(wxMouseEvent& event)
+{
+        ClearHighlight();
+}
+
+void SimplePiePlot::Highlight(int i)
+{
+        m_highlight = i;
+        Refresh();
+}
+
+void SimplePiePlot::ClearHighlight()
+{
+        if (m_highlight != -1) {
+                m_highlight = -1;
+                Refresh();
+        }
+}

trunk/src/simple_pie_plot.h

@@ -35 +35 @@
         void OnPaint(wxPaintEvent& event);
         void OnResize(wxSizeEvent& event);
+        void OnMouseLeaveWindow(wxMouseEvent& event);
+        void OnMouseMove(wxMouseEvent& event);
 private:
         /**
          * Gets the color suiting the \a i th segment.
          * \param i The segment number.
-         * \param hightlight Return color suiting for highlighting.
          */
-        wxColour GetSegmentColor(int i, bool highlight);
+        wxColour GetSegmentColor(int i);
+
+        void Highlight(int i);
+        void ClearHighlight();
 
         std::vector<double> m_data;
+        std::vector<double> m_angles;
         double m_data_total;
+        int m_highlight;
 };