C ++加快了地图访问速度

时间:2015-01-08 13:13:49

标签: c++ performance

我定义了以下地图:

class xy_angle {
public:
    int x;
    int y;
    int angle;

    xy_angle(int x, int y, int angle) :x(x), y(y), angle(angle){};

};

class xy_angleComparator {
public:
    bool operator () (const xy_angle &a, const xy_angle &b) const {
        if (a.x != b.x)
            return a.x < b.x;
        else if (a.y != b.y)
            return a.y < b.y;
        else if (a.angle != b.angle)
            return a.angle < b.angle;
        else
            return false;
    }
};

std::map<xy_angle, std::pair<int, int>, xy_angleComparator> transformed_coordinates_lut_;

我在初始化包含它的类时填写它:

//creating LUTs
int half_patch_size=48;
for (int x_start = -half_patch_size; x_start <= half_patch_size; x_start++){
    for (int y_start = -half_patch_size; y_start <= half_patch_size; y_start++){
        for (int angle = -314; angle < 314; angle++){
            float angle_f = (float)angle / 100.f;
            double cos_theta = cos(angle_f);
            double sin_theta = sin(angle_f);

            int x_tranformed = (int)(((float)x_start)*cos_theta - ((float)y_start)*sin_theta);
            int y_tranformed = (int)(((float)x_start)*sin_theta + ((float)y_start)*cos_theta);

            if (x_tranformed > half_patch_size)
                x_tranformed = half_patch_size;

            if (x_tranformed < -half_patch_size)
                x_tranformed = -half_patch_size;

            if (y_tranformed > half_patch_size)
                y_tranformed = half_patch_size;

            if (y_tranformed < -half_patch_size)
                y_tranformed = -half_patch_size;

            transformed_coordinates_lut_[xy_angle(x_start, y_start, angle)] = std::pair<int, int>(x_tranformed, y_tranformed);
        }
    }
}

我使用以下代码访问它:

int ax2 = transformed_coordinates_lut_[xy_angle(ax, ay, theta)].first;
int ay2 = transformed_coordinates_lut_[xy_angle(ax, ay, theta)].second;

我使用大量随机密钥测量了地图的访问运行时间,这非常疯狂。它完全支配使用它的函数的运行时间。

有什么方法可以加快速度吗?

谢谢!

吉尔。

2 个答案:

答案 0 :(得分:3)

您可以改为使用三维数组:f[x_start][y_start][angle]。它会占用相同(或更少)的空间,因为无论如何你都拥有所有可能的键。当然,您也可以使用适当的索引模拟具有平面矢量的三维数组。这种方法可以保证您不断查找时间。

答案 1 :(得分:1)

无论您使用哪个容器,此代码都很糟糕:

int ax2 = transformed_coordinates_lut_[xy_angle(ax, ay, theta)].first;
int ay2 = transformed_coordinates_lut_[xy_angle(ax, ay, theta)].second;

您正在进行两次相同的查找!绝对缓存结果:

auto& a2 = transformed_coordinates_lut_[xy_angle(ax, ay, theta)];
int ax2 = a2.first;
int ay2 = a2.second;

现在,就加快工作而言。最少的前期工作就是分出一个不同的关联容器类型:

using MapType = std::unordered_map<xy_angle,
                                   std::pair<int, int>,
                                   xy_angle_hash>; // implement this hash

这将为您提供O(1)次查找,而不是O(lg N)您目前在代码std::map中看到的class TransformMap { public: std::pair<int, int>& operator()(const xy_angle& ); private: // is it std::map? // or std::unordered_map? // or 3D-array or vector or ... ? }; 。但是如果你真的想花很多时间来探索这个容器,我建议你把它包起来以便控制实现:

{{1}}