mirror of
https://github.com/penpot/penpot.git
synced 2025-05-30 09:27:22 +02:00
131 lines
4.5 KiB
Clojure
131 lines
4.5 KiB
Clojure
;; This Source Code Form is subject to the terms of the Mozilla Public
|
|
;; License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
;; file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
;;
|
|
;; This Source Code Form is "Incompatible With Secondary Licenses", as
|
|
;; defined by the Mozilla Public License, v. 2.0.
|
|
;;
|
|
;; Copyright (c) 2020 UXBOX Labs SL
|
|
|
|
(ns app.worker.selection
|
|
(:require
|
|
[cljs.spec.alpha :as s]
|
|
[okulary.core :as l]
|
|
[app.common.data :as d]
|
|
[app.common.exceptions :as ex]
|
|
[app.common.geom.shapes :as gsh]
|
|
[app.common.pages :as cp]
|
|
[app.common.spec :as us]
|
|
[app.common.uuid :as uuid]
|
|
[app.util.quadtree :as qdt]
|
|
[app.worker.impl :as impl]))
|
|
|
|
(defonce state (l/atom {}))
|
|
|
|
(declare index-object)
|
|
(declare create-index)
|
|
|
|
(defmethod impl/handler :selection/initialize-index
|
|
[{:keys [file-id data] :as message}]
|
|
(letfn [(index-page [state page]
|
|
(let [id (:id page)
|
|
objects (:objects page)]
|
|
(assoc state id (create-index objects))))
|
|
|
|
(update-state [state]
|
|
(reduce index-page state (vals (:pages-index data))))]
|
|
(swap! state update-state)
|
|
nil))
|
|
|
|
(defmethod impl/handler :selection/update-index
|
|
[{:keys [page-id objects] :as message}]
|
|
(let [index (create-index objects)]
|
|
(swap! state update page-id (constantly index))
|
|
nil))
|
|
|
|
(defmethod impl/handler :selection/query
|
|
[{:keys [page-id rect frame-id include-frames? include-groups? disabled-masks] :or {include-groups? true
|
|
disabled-masks #{}} :as message}]
|
|
(when-let [index (get @state page-id)]
|
|
(let [result (-> (qdt/search index (clj->js rect))
|
|
(es6-iterator-seq))
|
|
|
|
;; Check if the shape matches the filter criteria
|
|
match-criteria?
|
|
(fn [shape]
|
|
(and (not (:hidden shape))
|
|
(not (:blocked shape))
|
|
(or (not frame-id) (= frame-id (:frame-id shape)))
|
|
(case (:type shape)
|
|
:frame include-frames?
|
|
:group include-groups?
|
|
true)))
|
|
|
|
overlaps?
|
|
(fn [shape]
|
|
(gsh/overlaps? shape rect))
|
|
|
|
overlaps-masks?
|
|
(fn [masks]
|
|
(->> masks
|
|
(some (comp not overlaps?))
|
|
not))
|
|
|
|
;; Shapes after filters of overlapping and criteria
|
|
matching-shapes
|
|
(into []
|
|
(comp (map #(unchecked-get % "data"))
|
|
(filter match-criteria?)
|
|
(filter (comp overlaps? :frame))
|
|
(filter (comp overlaps-masks? :masks))
|
|
(filter overlaps?))
|
|
result)]
|
|
|
|
(into (d/ordered-set)
|
|
(->> matching-shapes
|
|
(sort-by (comp - :z))
|
|
(map :id))))))
|
|
|
|
(defn create-mask-index
|
|
"Retrieves the mask information for an object"
|
|
[objects parents-index]
|
|
(let [retrieve-masks
|
|
(fn [id parents]
|
|
(->> parents
|
|
(map #(get objects %))
|
|
(filter #(:masked-group? %))
|
|
;; Retrieve the masking element
|
|
(mapv #(get objects (->> % :shapes first)))))]
|
|
(->> parents-index
|
|
(d/mapm retrieve-masks))))
|
|
|
|
(defn- create-index
|
|
[objects]
|
|
(let [shapes (-> objects (dissoc uuid/zero) (vals))
|
|
z-index (cp/calculate-z-index objects)
|
|
parents-index (cp/generate-child-all-parents-index objects)
|
|
masks-index (create-mask-index objects parents-index)
|
|
bounds (gsh/selection-rect shapes)
|
|
bounds #js {:x (:x bounds)
|
|
:y (:y bounds)
|
|
:width (:width bounds)
|
|
:height (:height bounds)}]
|
|
|
|
(reduce (partial index-object objects z-index parents-index masks-index)
|
|
(qdt/create bounds)
|
|
shapes)))
|
|
|
|
(defn- index-object
|
|
[objects z-index parents-index masks-index index obj]
|
|
(let [{:keys [x y width height]} (:selrect obj)
|
|
shape-bound #js {:x x :y y :width width :height height}
|
|
parents (get parents-index (:id obj))
|
|
masks (get masks-index (:id obj))
|
|
z (get z-index (:id obj))
|
|
frame (when (and (not= :frame (:type obj))
|
|
(not= (:frame-id obj) uuid/zero))
|
|
(get objects (:frame-id obj)))]
|
|
(qdt/insert index
|
|
shape-bound
|
|
(assoc obj :frame frame :masks masks :parents parents :z z))))
|
|
|