Turn on polygon smoothing, in an attempt to get even smoother in the edges.

[pitch] / glpitch.cpp

diff --git a/glpitch.cpp b/glpitch.cpp
index a3c301051046fe2466b3bb5a7a45a234a905703d..e47e977f2b8743960b776e28c13f6a77947b574c 100644 (file)
--- a/glpitch.cpp
+++ b/glpitch.cpp
@@ -9,11 +9,46 @@
 #define MIN_X -40
 #define MAX_X 5
 
-double find_pos(double freq)
+double find_linear_pos(double freq)
 {
        return 12.0 * log2(freq / BASE_PITCH) - 3.0;
 }
 
+#if COMPRESS_PITCH_DISPLAY
+// bend the scale a bit to get more precision around the interesting points
+double magnifying_glass(double x)
+{
+       double sigma = PITCH_COMPRESSION_SIGMA;
+       double div = 1.0 / sqrt(2.0 * sigma * sigma);
+       return
+               erf((x - find_linear_pos(notes[0].freq)) * div) +
+               erf((x - find_linear_pos(notes[1].freq)) * div) +
+               erf((x - find_linear_pos(notes[2].freq)) * div) +
+               erf((x - find_linear_pos(notes[3].freq)) * div) +
+               erf((x - find_linear_pos(notes[4].freq)) * div) +
+               erf((x - find_linear_pos(notes[5].freq)) * div);
+}
+
+// like nonlinear_func, but f(MIN_X) = MIN_X and f(MAX_X) = MAX_X
+double normalized_magnifying_glass(double x)
+{
+       double y = magnifying_glass(x);
+       double min_y = magnifying_glass(MIN_X);
+       double max_y = magnifying_glass(MAX_X);
+       return (y - min_y) * (MAX_X - MIN_X) / (max_y - min_y) + MIN_X;
+}
+#endif /* COMPRESS_PITCH_DISPLAY */
+
+double find_display_pos(double freq)
+{
+       double linear_pos = find_linear_pos(freq);
+#if COMPRESS_PITCH_DISPLAY
+       return normalized_magnifying_glass(linear_pos);
+#else
+       return linear_pos;
+#endif
+}
+
 int main(void)
 {
        PitchDetector pd(SAMPLE_RATE, FFT_LENGTH, PAD_FACTOR, OVERLAP);
@@ -63,9 +98,11 @@ int main(void)
                glEnd();
 
                if (peak.second - log10(FFT_LENGTH) >= 0.0) {
-                       double peak_pos = find_pos(peak.first);
+                       double peak_pos = find_display_pos(peak.first);
 
                        glBlendFunc(GL_SRC_ALPHA, GL_ONE);
+                       glEnable(GL_POLYGON_SMOOTH);
+
                        glBegin(GL_QUADS);      
                        glColor4f(0.0, 1.0f, 0.0f, 0.2f);
                        glVertex2f(peak_pos - 0.1, 0.0f);
@@ -73,13 +110,15 @@ int main(void)
                        glVertex2f(peak_pos + 0.1, 1.0f);
                        glVertex2f(peak_pos + 0.1, 0.0f);
                        glEnd();
+
+                       glDisable(GL_POLYGON_SMOOTH);
                }
 
                glColor4f(1.0f, 1.0f, 1.0f, 1.0f);
                glBegin(GL_LINES);
 
                for (int i = 0; i < 6; ++i) {
-                       double pos = find_pos(notes[i].freq);
+                       double pos = find_display_pos(notes[i].freq);
                        glVertex2f(pos, 0.0f);
                        glVertex2f(pos, 1.0f);
                }