重构仅因一个函数调用而异的代码块
本文关键字:函数调用 代码 一个 重构 | 更新日期: 2023-09-27 18:32:46
我有类似的公式来计算哈特利变换。唯一的区别是输入函数 - sin, cos, exp 在以下代码行中:
Math.Exp((double)tau)
Math.Sin((double)tau)
Math.Cos((double)tau)
如何转义几乎相同的代码片段并缩短以下片段中的代码?
private void CountHartley(ref double [] arr, string function)
{
int N = arr.Length;
if (function == "exp")
{
for (int nu = 0, tau = 0; ((nu < N) && (tau < N)); nu++, tau++)
{
arr[nu] = 1 / (double)N *
Math.Exp((double)tau) *
(Math.Sin(2 * Math.PI * nu * tau / (double) N) +
Math.Cos(2 * Math.PI * nu * tau / (double) N));
}
}
else if (function == "sin")
{
for (int nu = 0, tau = 0; ((nu < N) && (tau < N)); nu++, tau++)
{
arr[nu] = 1 / (double)N *
Math.Sin((double)tau) *
(Math.Sin(2 * Math.PI * nu * tau / (double)N) +
Math.Cos(2 * Math.PI * nu * tau / (double)N));
}
}
else
{
for (int nu = 0, tau = 0; ((nu < N) && (tau < N)); nu++, tau++)
{
arr[nu] = 1 / (double)N *
Math.Cos((double)tau) *
(Math.Sin(2 * Math.PI * nu * tau / (double)N) +
Math.Cos(2 * Math.PI * nu * tau / (double)N));
}
}
}
您可以直接传入函数,而不是传入带有要使用的函数的string
。
您可以使用 Func<T、TResult> 委托,如下所示:
private void CountHartley(ref double [] arr, Func<double, double> function)
{
int N = arr.Length;
for (int nu = 0, tau = 0; ((nu < N) && (tau < N)); nu++, tau++)
{
arr[nu] = 1 / (double)N *
function((double)tau) *
(Math.Sin(2 * Math.PI * nu * tau / (double) N) +
Math.Cos(2 * Math.PI * nu * tau / (double) N));
}
}
用法:
var result = CountHartley(arr, Math.Cos);
将 Math.* 作为参数传递给 Func<double, double> function
参数怎么样?