c# - 绘制具有颜色渐变的矩阵 "Spectrogram"

Question

在使用 STFT(短时傅立叶变换)后，输出是一个表示 3d 图的矩阵，就像 (A[X, Y] = M) A 是输出矩阵，X 是时间 ，Y为频率，第三维M为振幅，由像素颜色的强度表示，如下图:

Spectrogram 2

如何使用 C# 中的图片绘制具有渐变颜色的输出矩阵 A？
是否有包含 C# 频谱图控件的库？

---

更新:
在对给定的算法进行一些修改后，我可以绘制频谱图，除了第一种颜色更改为黑色之外，我没有更改调色板，但我不知道为什么它很褪色!

这个代表一个声音的说法

Bye Bye

Bye Bye Spectrogram

这是一个纯正弦波，所以它的频率几乎始终相同

Pure sine wave Spectrogram

输出被接受，它代表了预期的输入信号的频率，但我认为有一种方法可以使频谱图与示例中的一样好，你能看看我的代码并提出修改建议吗？

这是事件处理程序:

private void SpectrogramButton_Click(object sender, EventArgs e)
{
    Complex[][] SpectrogramData = Fourier_Transform.STFT(/*signal:*/ samples,  /*windowSize:*/ 512, /*hopSize:*/ 512);
    SpectrogramBox.Image = Spectrogram.DrawSpectrogram(SpectrogramData, /*Interpolation Factor:*/ 1000, /*Height:*/ 256);
}

这是我修改后的绘图功能:

public static Bitmap DrawSpectrogram(Complex[][] Data, int InterpolationFactor, int Height)
{
    // target size:
    Size sz = new Size(Data.GetLength(0), Height);
    Bitmap bmp = new Bitmap(sz.Width, sz.Height);

    // the data array:
    //double[,] data = new double[222, 222];

    // step sizes:
    float stepX = 1f * sz.Width / Data.GetLength(0);
    float stepY = 1f * sz.Height / Data[0].GetLength(0);

    // create a few stop colors:
    List<Color> baseColors = new List<Color>();  // create a color list
    baseColors.Add(Color.Black);
    baseColors.Add(Color.LightSkyBlue);
    baseColors.Add(Color.LightGreen);
    baseColors.Add(Color.Yellow);
    baseColors.Add(Color.Orange);
    baseColors.Add(Color.Red);


    // and the interpolate a larger number of grdient colors:
    List<Color> colors = interpolateColors(baseColors, InterpolationFactor);

    // a few boring test data
    //Random rnd = new Random(1);
    //for (int x = 0; x < data.GetLength(0); x++)
    //    for (int y = 0; y < data.GetLength(1); y++)
    //    {
    //        //data[x, y] = rnd.Next((int)(300 + Math.Sin(x * y / 999) * 200)) +
    //        //                rnd.Next(x + y + 111);
    //        data[x, y] = 0;
    //    }

    // now draw the data:
    float Max = Complex.Max(Data);
    using (Graphics G = Graphics.FromImage(bmp))
        for (int x = 0; x < Data.GetLength(0); x++)
            for (int y = 0; y < Data[0].GetLength(0); y++)
            {
                int Val = (int)Math.Ceiling((Data[x][y].Magnitude / Max) * (InterpolationFactor - 1));
                using (SolidBrush brush = new SolidBrush(colors[(int)Val]))
                    G.FillRectangle(brush, x * stepX, (Data[0].GetLength(0) - y) * stepY, stepX, stepY);
            }

    // and display the result
    return bmp;
}

我不太理解您在回答中提到的log 东西，很抱歉我的知识太少。

---

更新:
这是将 log10 添加到幅度后的输出(忽略负值):

1. This one of "Bye bye" from before:

1. A Shotgun Blast:

1. A Music Box:

我认为这个输出是可以接受的，它与我一开始带来的示例不同，但我认为它更好。

Answer 1

不，据我所知没有开箱即用的控件。当然，您可能还可以购买外部图书馆，但是嘘，您不能在 SO 上询问有关它们的信息..

理论上您可以为此使用滥用 Chart 控件。但由于 DataPoints 是相当昂贵的对象，或者至少比它们看起来更昂贵，这似乎不可取。

相反，您可以自己简单地将图形绘制到 Bitmap 中。

• 第一步是确定颜色渐变。查看interpolateColors function here举个例子!

• 然后您只需使用 floats 对步长和像素大小对数据进行双重循环，并在那里执行 Graphics.FillRectangle。

这是一个简单的例子，使用GDI+创建一个Bitmap和一个Winforms PictureBox来显示。它不会向图形添加任何轴并完全填充它。

它首先创建一些样本数据和一个具有 1000 颜色的渐变。然后它绘制成一个 Bitmap 并显示结果:

private void button6_Click(object sender, EventArgs e)
{
    // target size:
    Size sz = pictureBox1.ClientSize;
    Bitmap bmp = new Bitmap(sz.Width, sz.Height);

    // the data array:
    double[,] data = new double[222, 222];

    // step sizes:
    float stepX = 1f * sz.Width / data.GetLength(0);
    float stepY = 1f * sz.Height / data.GetLength(1);

    // create a few stop colors:
    List<Color> baseColors = new List<Color>();  // create a color list
    baseColors.Add(Color.RoyalBlue);
    baseColors.Add(Color.LightSkyBlue);
    baseColors.Add(Color.LightGreen);
    baseColors.Add(Color.Yellow);
    baseColors.Add(Color.Orange);
    baseColors.Add(Color.Red);
    // and the interpolate a larger number of grdient colors:
    List<Color> colors = interpolateColors(baseColors, 1000);

    // a few boring test data
    Random rnd = new Random(1);
    for (int x = 0; x < data.GetLength(0); x++)
    for (int y = 0; y < data.GetLength(1); y++)
    {
        data[x, y] = rnd.Next( (int) (300 + Math.Sin(x * y / 999) * 200 )) +
                        rnd.Next(  x +  y + 111);
    }

    // now draw the data:
    using (Graphics G = Graphics.FromImage(bmp))
    for (int x = 0; x < data.GetLength(0); x++)
        for (int y = 0; y < data.GetLength(1); y++)
        {
            using (SolidBrush brush = new SolidBrush(colors[(int)data[x, y]]))
                G.FillRectangle(brush, x * stepX, y * stepY, stepX, stepY);
        }

    // and display the result
    pictureBox1.Image = bmp;
}

这是链接中的函数:

List<Color> interpolateColors(List<Color> stopColors, int count)
{
    SortedDictionary<float, Color> gradient = new SortedDictionary<float, Color>();
    for (int i = 0; i < stopColors.Count; i++)
        gradient.Add(1f * i / (stopColors.Count - 1), stopColors[i]);
    List<Color> ColorList = new List<Color>();

    using (Bitmap bmp = new Bitmap(count, 1))
    using (Graphics G = Graphics.FromImage(bmp))
    {
        Rectangle bmpCRect = new Rectangle(Point.Empty, bmp.Size);
        LinearGradientBrush br = new LinearGradientBrush
                                (bmpCRect, Color.Empty, Color.Empty, 0, false);
        ColorBlend cb = new ColorBlend();
        cb.Positions = new float[gradient.Count];
        for (int i = 0; i < gradient.Count; i++)
            cb.Positions[i] = gradient.ElementAt(i).Key;
        cb.Colors = gradient.Values.ToArray();
        br.InterpolationColors = cb;
        G.FillRectangle(br, bmpCRect);
        for (int i = 0; i < count; i++) ColorList.Add(bmp.GetPixel(i, 0));
        br.Dispose();
    }
    return ColorList;
}

您可能希望绘制带有标签等的坐标轴。您可以使用 Graphics.DrawString 或 TextRenderer.DrawText 来完成此操作。在绘图区域周围留出足够的空间即可!

我使用转换为 int 的数据值作为颜色表的直接指针。

根据您的数据，您需要缩小它们甚至使用对数转换。您的第一张图片显示从 100 到 20k 的对数比例，第二张看起来是从 0 到 100 的线性。

如果您向我们展示您的数据结构，我们可以进一步提示您如何调整代码以使用它。