fix(behaviors): fix optimize viewport transform does not hide edge arrow (#6217)

Co-authored-by: antv <antv@antfin.com>

packages/g6/__tests__/demos/behavior-optimize-viewport-transform.ts

@@ -20,6 +20,7 @@ export const behaviorOptimizeViewportTransform: TestCase = async (context) => {
       style: {
         labelFontSize: 8,
         labelText: (datum) => datum.id,
+        startArrow: true,
       },
     },
     behaviors: ['drag-canvas', 'zoom-canvas', 'scroll-canvas', 'optimize-viewport-transform'],

packages/g6/src/behaviors/optimize-viewport-transform.ts

@@ -59,22 +59,16 @@ export class OptimizeViewportTransform extends BaseBehavior<OptimizeViewportTran
     this.bindEvents();
   }
 
-  private filterShapes = (shapes: DisplayObject[], classnames?: string[]) => {
-    return shapes.filter((shape) => shape.className && !classnames?.includes(shape.className));
-  };
-
   private setElementsVisibility = (
     elements: DisplayObject[],
     visibility: BaseStyleProps['visibility'],
     excludedClassnames?: string[],
   ) => {
     elements.forEach((element) => {
-      setVisibility(
-        element,
-        visibility,
-        false,
-        (shape) => !!shape.className && !excludedClassnames?.includes(shape.className),
-      );
+      setVisibility(element, visibility, false, (shape) => {
+        if (!shape.className) return true;
+        return !excludedClassnames?.includes(shape.className);
+      });
     });
   };