use-add-dag-shape.ts 1.2 KB

123456789101112131415161718192021222324252627282930313233
  1. /*
  2. * Licensed to the Apache Software Foundation (ASF) under one or more
  3. * contributor license agreements. See the NOTICE file distributed with
  4. * this work for additional information regarding copyright ownership.
  5. * The ASF licenses this file to You under the Apache License, Version 2.0
  6. * (the "License"); you may not use this file except in compliance with
  7. * the License. You may obtain a copy of the License at
  8. *
  9. * http://www.apache.org/licenses/LICENSE-2.0
  10. *
  11. * Unless required by applicable law or agreed to in writing, software
  12. * distributed under the License is distributed on an "AS IS" BASIS,
  13. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. * See the License for the specific language governing permissions and
  15. * limitations under the License.
  16. */
  17. import { Graph, Cell } from '@antv/x6'
  18. import { useDagStore } from '@/store/project/dynamic/dag'
  19. export function useAddDagShape(graph: Graph) {
  20. const cells: Array<Cell> = useDagStore().getDagTasks
  21. cells.forEach(item => {
  22. if (item.shape === 'dag-edge') {
  23. cells.push((graph as Graph).addEdge(item))
  24. } else {
  25. cells.push((graph as Graph).addNode(item as any))
  26. }
  27. })
  28. ;(graph as Graph).resetCells(cells)
  29. }