1
0

boxAutoGroup.cpp 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. #include "cdrapp.h"
  2. #include <algorithm>
  3. #include <map>
  4. #include <vector>
  5. #include <iostream>
  6. #include <fstream>
  7. #include <chrono>
  8. #define GET_BOUNDING_BOX(box) \
  9. GetBoundingBox(&(box).x, &(box).y, &(box).w, &(box).h, false)
  10. #define ZERO_4PC 0, 0, 0, 0
  11. typedef struct {
  12. double x; // 左下角 x 坐标
  13. double y; // 左下角 y 坐标
  14. double w; // 宽度
  15. double h; // 高度
  16. } BoundingBox;
  17. double get_bounding_box_area(BoundingBox box) { return box.w * box.h; }
  18. // 自定义比较函数,按 BoundingBox 的面积大小进行排序
  19. bool compare_bounding_boxes(const std::pair<int, BoundingBox> &a,
  20. const std::pair<int, BoundingBox> &b) {
  21. return get_bounding_box_area(a.second) > get_bounding_box_area(b.second);
  22. }
  23. // 查找父节点
  24. int find(std::vector<int>& parent, int x) {
  25. if (parent[x] != x) {
  26. parent[x] = find(parent, parent[x]);
  27. }
  28. return parent[x];
  29. }
  30. // 合并两个集合
  31. void unionSet(std::vector<int>& parent, int x, int y) {
  32. int xroot = find(parent, x);
  33. int yroot = find(parent, y);
  34. parent[xroot] = yroot;
  35. }
  36. // 检查两个矩形是否重叠
  37. bool isOverlapped(const BoundingBox &a, const BoundingBox &b) {
  38. return a.x < b.x + b.w && a.x + a.w > b.x && a.y < b.y + b.h && a.y + a.h > b.y;
  39. } // 函数使用AABB(Axis-Aligned Bounding Box)碰撞检测
  40. bool BBox_DrawRectangle(corel *cdr) {
  41. BoundingBox box;
  42. auto sr = cdr->ActiveSelectionRange; // 获得选择范围
  43. auto al = cdr->ActiveLayer; // 获得当前层
  44. if (!sr || !al) return false;
  45. // CorelDRAW Shapes 物件 Item 编号从1开始
  46. for (auto i = 0; i != sr->Count; i++) {
  47. sr->Shapes->Item[i + 1]->GET_BOUNDING_BOX(box); // 获得Shapes的BoundingBox,赋值到box
  48. al->CreateRectangle2(box.x, box.y, box.w, box.h, ZERO_4PC); // 使用BoundingBox box 创建一个矩形
  49. }
  50. return true;
  51. }
  52. // 快速分组重叠的区域, 使用算法"Union-Find" 算法。这个算法可以有效地处理这种并集问题。
  53. // 算法的时间复杂度为 O(n^2),其中 n 是矩形的数量。如果矩形数量较多,可以考虑使用更高效的算法,
  54. // 例如使用四叉树(Quadtree)或者区间树(Interval Tree)等数据结构来加速计算。
  55. bool Box_AutoGroup(corel *cdr) {
  56. BoundingBox box;
  57. auto sr = cdr->ActiveSelectionRange; // 获得选择范围
  58. auto al = cdr->ActiveLayer; // 获得当前层
  59. if (!sr || !al) return false;
  60. auto start = std::chrono::high_resolution_clock::now(); // 开始时间
  61. BeginOpt(cdr);
  62. std::vector<BoundingBox> boxes;
  63. std::vector<int> parent;
  64. // CorelDRAW Shapes 物件 Item 编号从1开始
  65. for (auto i = 0; i != sr->Count; i++) {
  66. sr->Shapes->Item[i + 1]->GET_BOUNDING_BOX(box);
  67. boxes.push_back(box);
  68. parent.push_back(i);
  69. }
  70. // 实现 Union-Find 算法来合并重叠的区域
  71. for (int i = 0; i < boxes.size(); i++) {
  72. for (int j = i + 1; j < boxes.size(); j++) {
  73. if (isOverlapped(boxes[i], boxes[j])) {
  74. unionSet(parent, i, j);
  75. }
  76. }
  77. }
  78. // 输出分组结果到文件
  79. std::ofstream output_file("D:\\group.txt");
  80. if (output_file.is_open()) {
  81. std::map<int, std::vector<int>> groups;
  82. for (int i = 0; i < parent.size(); i++) {
  83. int root = find(parent, i);
  84. groups[root].push_back(i + 1); // CorelDRAW Shapes 物件 Item 编号从1开始
  85. }
  86. for (const auto& group : groups) {
  87. output_file << "Group: ";
  88. for (int index : group.second) {
  89. output_file << index << " ";
  90. }
  91. output_file << std::endl;
  92. }
  93. auto end = std::chrono::high_resolution_clock::now(); // 结束时间
  94. // 计算持续时间
  95. std::chrono::duration<double> duration = end - start;
  96. output_file << "Execution time: " << duration.count() << " seconds\n";
  97. output_file.close();
  98. }
  99. // 输出分组结果
  100. std::map<int, std::vector<int>> groups;
  101. for (int i = 0; i < parent.size(); i++) {
  102. int root = find(parent, i);
  103. groups[root].push_back(i + 1); // CorelDRAW Shapes 物件 Item 编号从1开始
  104. }
  105. auto srgp = cdr->CreateShapeRange();
  106. auto srs = cdr->CreateShapeRange();
  107. cdr->ActiveDocument->ClearSelection();
  108. // 原来 没有取消选择 最初速度
  109. // Execution time: 63.0305 seconds
  110. // srgp->GET_BOUNDING_BOX(box);
  111. // al->CreateRectangle2(box.x, box.y, box.w, box.h, ZERO_4PC); // 使用边界 创建一个矩形
  112. // box边界 转左上和右下坐标 box.x, box.y + box.h, box.x + box.w, box.y
  113. // auto sh = cdr->ActivePage->SelectShapesFromRectangle(box.x, box.y + box.h, box.x + box.w, box.y, false);
  114. // sh->Group();
  115. // 使用 SelectShapesFromRectangle 框选的形状进行群组
  116. // Execution time: 2.44753 seconds
  117. // cdr->ActiveDocument->ClearSelection(); // 使用取消选择
  118. // Execution time: 1.7432 seconds
  119. // srgp->CreateSelection();
  120. // cdr->ActiveSelectionRange->Group();
  121. // Execution time: 1.87662 seconds
  122. // 分组分别进行群组
  123. for (const auto& group : groups) {
  124. for (int index : group.second)
  125. srgp->Add(sr->Shapes->Item[index]);
  126. if(sr->Count >1)
  127. srs->Add(srgp->Group());
  128. else
  129. srs->AddRange(srgp);
  130. srgp->RemoveAll();
  131. }
  132. srs->CreateSelection();
  133. EndOpt(cdr);
  134. return true;
  135. }