318 lines
		
	
	
		
			6.8 KiB
		
	
	
	
		
			Factor
		
	
	
			
		
		
	
	
			318 lines
		
	
	
		
			6.8 KiB
		
	
	
	
		
			Factor
		
	
	
! Copyright (C) 2010 Slava Pestov.
 | 
						|
USING: accessors combinators fry kernel locals namespaces
 | 
						|
game.models.half-edge euler.b-rep sequences typed math
 | 
						|
math.vectors ;
 | 
						|
IN: euler.operators
 | 
						|
 | 
						|
ERROR: edges-not-incident ;
 | 
						|
 | 
						|
: assert-incident ( e1 e2 -- )
 | 
						|
    incident? [ edges-not-incident ] unless ;
 | 
						|
 | 
						|
ERROR: should-not-be-equal obj1 obj2 ;
 | 
						|
 | 
						|
: assert-not= ( obj1 obj2 -- )
 | 
						|
    2dup eq? [ should-not-be-equal ] [ 2drop ] if ;
 | 
						|
 | 
						|
ERROR: edges-in-different-faces ;
 | 
						|
 | 
						|
: assert-same-face ( e1 e2 -- )
 | 
						|
    same-face? [ edges-in-different-faces ] unless ;
 | 
						|
 | 
						|
ERROR: edges-in-same-face ;
 | 
						|
 | 
						|
: assert-different-faces ( e1 e2 -- )
 | 
						|
    same-face? [ edges-in-same-face ] when ;
 | 
						|
 | 
						|
: assert-isolated-component ( edge -- )
 | 
						|
    [ [ opposite-edge>> ] [ next-edge>> ] bi assert= ]
 | 
						|
    [ dup opposite-edge>> assert-same-face ]
 | 
						|
    bi ;
 | 
						|
 | 
						|
ERROR: not-a-base-face face ;
 | 
						|
 | 
						|
: assert-base-face ( face -- )
 | 
						|
    dup base-face? [ drop ] [ not-a-base-face ] if ;
 | 
						|
 | 
						|
ERROR: has-rings face ;
 | 
						|
 | 
						|
: assert-no-rings ( face -- )
 | 
						|
    dup next-ring>> [ has-rings ] [ drop ] if ;
 | 
						|
 | 
						|
: assert-ring-of ( ring face -- )
 | 
						|
    [ base-face>> ] dip assert= ;
 | 
						|
 | 
						|
: with-b-rep ( b-rep quot -- )
 | 
						|
    [ b-rep ] dip with-variable ; inline
 | 
						|
 | 
						|
: make-b-rep ( quot -- b-rep )
 | 
						|
    <b-rep> [ swap with-b-rep ] [ finish-b-rep ] [ ] tri ; inline
 | 
						|
 | 
						|
<PRIVATE
 | 
						|
 | 
						|
:: make-loop ( vertex face -- edge )
 | 
						|
    b-rep get new-edge :> edge
 | 
						|
    vertex edge vertex<<
 | 
						|
    edge edge next-edge<<
 | 
						|
    face edge face<<
 | 
						|
 | 
						|
    edge ;
 | 
						|
 | 
						|
: make-loop-face ( vertex -- edge )
 | 
						|
    b-rep get new-face
 | 
						|
    dup >>base-face
 | 
						|
    make-loop ;
 | 
						|
 | 
						|
:: make-edge ( vertex next-edge -- edge )
 | 
						|
    b-rep get new-edge :> edge
 | 
						|
    vertex edge vertex<<
 | 
						|
    next-edge edge next-edge<<
 | 
						|
    next-edge face>> edge face<<
 | 
						|
 | 
						|
    edge ;
 | 
						|
 | 
						|
: opposite-edges ( e1 e2 -- )
 | 
						|
    [ opposite-edge<< ] [ swap opposite-edge<< ] 2bi ;
 | 
						|
 | 
						|
PRIVATE>
 | 
						|
 | 
						|
MIXIN: point
 | 
						|
INSTANCE: sequence point
 | 
						|
INSTANCE: number point
 | 
						|
 | 
						|
TYPED:: make-vefs ( pos1: point pos2: point -- edge: b-edge )
 | 
						|
    b-rep get :> b-rep
 | 
						|
 | 
						|
    pos1 b-rep new-vertex :> v1
 | 
						|
    v1 make-loop-face :> e1
 | 
						|
 | 
						|
    pos2 b-rep new-vertex :> v2
 | 
						|
    v2 e1 make-edge :> e2
 | 
						|
 | 
						|
    e2 e1 next-edge<<
 | 
						|
    e1 e2 opposite-edges
 | 
						|
 | 
						|
    e2 ;
 | 
						|
 | 
						|
TYPED:: make-ev-one ( edge: b-edge point: point -- edge: b-edge )
 | 
						|
    point b-rep get new-vertex :> v
 | 
						|
    v edge make-edge :> e1'
 | 
						|
 | 
						|
    edge vertex>> e1' make-edge :> e2'
 | 
						|
 | 
						|
    e2' edge face-cw next-edge<<
 | 
						|
    e1' e2' opposite-edges
 | 
						|
 | 
						|
    e1' ;
 | 
						|
 | 
						|
<PRIVATE
 | 
						|
 | 
						|
:: subdivide-vertex-cycle ( e1 e2 v -- )
 | 
						|
    e1 e2 eq? [
 | 
						|
        v e1 vertex<<
 | 
						|
        e1 vertex-cw e2 v subdivide-vertex-cycle
 | 
						|
    ] unless ;
 | 
						|
 | 
						|
:: (make-ev) ( e1 e2 point -- edge )
 | 
						|
    e1 e2 assert-incident
 | 
						|
 | 
						|
    point b-rep get new-vertex :> v'
 | 
						|
    v' e2 make-edge :> e1'
 | 
						|
 | 
						|
    e1 vertex>> :> v
 | 
						|
 | 
						|
    v e1 make-edge :> e2'
 | 
						|
 | 
						|
    e1 e2 v' subdivide-vertex-cycle
 | 
						|
 | 
						|
    e1 face-cw :> e1p
 | 
						|
    e2 face-cw :> e2p
 | 
						|
    e1 opposite-edge>> :> e1m
 | 
						|
 | 
						|
    e1m e1p assert-not=
 | 
						|
 | 
						|
    e1' e2p next-edge<<
 | 
						|
    e2' e1p next-edge<<
 | 
						|
 | 
						|
    e1' e2' opposite-edges
 | 
						|
 | 
						|
    e1' ;
 | 
						|
 | 
						|
PRIVATE>
 | 
						|
 | 
						|
TYPED:: make-ev ( e1: b-edge e2: b-edge point: point -- edge: b-edge )
 | 
						|
    e1 e2 eq?
 | 
						|
    [ e1 point make-ev-one ] [ e1 e2 point (make-ev) ] if ;
 | 
						|
 | 
						|
<PRIVATE
 | 
						|
 | 
						|
: subdivide-edge-cycle ( face e1 e2 -- )
 | 
						|
    2dup eq? [ 3drop ] [
 | 
						|
        [ drop face<< ]
 | 
						|
        [ [ next-edge>> ] dip subdivide-edge-cycle ] 3bi
 | 
						|
    ] if ;
 | 
						|
 | 
						|
PRIVATE>
 | 
						|
 | 
						|
TYPED:: make-ef ( e1: b-edge e2: b-edge -- edge: b-edge )
 | 
						|
    e1 e2 assert-same-face
 | 
						|
 | 
						|
    e2 vertex>> make-loop-face :> e1'
 | 
						|
    e1 vertex>> e2 make-edge :> e2'
 | 
						|
    e1' e2' opposite-edges
 | 
						|
 | 
						|
    e1 face-cw :> e1p
 | 
						|
 | 
						|
    e1 e2 eq? [
 | 
						|
        e2 face-cw :> e2p
 | 
						|
 | 
						|
        e1' face>> e1 e2 subdivide-edge-cycle
 | 
						|
 | 
						|
        e1' e2p next-edge<<
 | 
						|
        e1 e1' next-edge<<
 | 
						|
    ] unless
 | 
						|
 | 
						|
    e2' e1p next-edge<<
 | 
						|
    e1' ;
 | 
						|
 | 
						|
TYPED:: make-e-kill-r ( edge-ring: b-edge edge-face: b-edge -- edge: b-edge )
 | 
						|
    edge-ring face>> :> ring
 | 
						|
    edge-face face>> :> face
 | 
						|
    ring face assert-ring-of
 | 
						|
 | 
						|
    edge-ring [ face >>face drop ] each-face-edge
 | 
						|
 | 
						|
    edge-ring vertex>> edge-face make-edge :> e1
 | 
						|
    edge-face vertex>> edge-ring make-edge :> e2
 | 
						|
 | 
						|
    ring face delete-ring
 | 
						|
    ring b-rep get delete-face
 | 
						|
 | 
						|
    e2 edge-face face-cw next-edge<<
 | 
						|
    e1 edge-ring face-cw next-edge<<
 | 
						|
 | 
						|
    e1 e2 opposite-edges
 | 
						|
 | 
						|
    e1 ;
 | 
						|
 | 
						|
TYPED:: make-f-kill-rh ( edge-ring: b-edge -- )
 | 
						|
    edge-ring face>> :> ring
 | 
						|
    ring base-face>> :> base-face
 | 
						|
    ring base-face delete-ring
 | 
						|
    ring ring base-face<< ;
 | 
						|
 | 
						|
TYPED:: kill-vefs ( edge: b-edge -- )
 | 
						|
    edge assert-isolated-component
 | 
						|
 | 
						|
    b-rep get :> b-rep
 | 
						|
    edge dup opposite-edge>> :> ( e2 e1 )
 | 
						|
 | 
						|
    e1 vertex>> :> v1
 | 
						|
    e2 vertex>> :> v2
 | 
						|
 | 
						|
    e1 face>> b-rep delete-face
 | 
						|
 | 
						|
    e1 b-rep delete-edge
 | 
						|
    e2 b-rep delete-edge
 | 
						|
    v1 b-rep delete-vertex
 | 
						|
    v2 b-rep delete-vertex ;
 | 
						|
 | 
						|
TYPED:: kill-ev ( edge: b-edge -- )
 | 
						|
    b-rep get :> b-rep
 | 
						|
 | 
						|
    edge vertex>> :> v
 | 
						|
    edge opposite-edge>> :> edge'
 | 
						|
    edge' vertex>> :> v'
 | 
						|
 | 
						|
    edge [ v' >>vertex drop ] each-vertex-edge
 | 
						|
 | 
						|
    edge face-cw :> edgep
 | 
						|
    edge' face-cw :> edge'p
 | 
						|
 | 
						|
    edge next-edge>> edgep next-edge<<
 | 
						|
    edge' next-edge>> edge'p next-edge<<
 | 
						|
 | 
						|
    v b-rep delete-vertex
 | 
						|
    edge b-rep delete-edge
 | 
						|
    edge' b-rep delete-edge ;
 | 
						|
 | 
						|
TYPED:: kill-ef ( edge: b-edge -- )
 | 
						|
    b-rep get :> b-rep
 | 
						|
 | 
						|
    edge :> e1
 | 
						|
    edge opposite-edge>> :> e2
 | 
						|
 | 
						|
    e1 e2 assert-different-faces
 | 
						|
 | 
						|
    e1 face-cw :> e1p
 | 
						|
    e2 face-cw :> e2p
 | 
						|
 | 
						|
    e1 face>> :> f1
 | 
						|
    e2 face>> :> f2
 | 
						|
 | 
						|
    e1 [ f2 >>face drop ] each-face-edge
 | 
						|
    f1 b-rep delete-face
 | 
						|
 | 
						|
    e1 e2 incident? [ 
 | 
						|
        e2 next-edge>> e2p next-edge<<
 | 
						|
 | 
						|
    ] [
 | 
						|
        e2 next-edge>> e1p next-edge<<
 | 
						|
        e1 next-edge>> e2p next-edge<<
 | 
						|
    ] if
 | 
						|
 | 
						|
    e1 b-rep delete-edge
 | 
						|
    e2 b-rep delete-edge ;
 | 
						|
 | 
						|
TYPED:: kill-e-make-r ( edge: b-edge -- edge-ring: b-edge )
 | 
						|
    b-rep get :> b-rep
 | 
						|
 | 
						|
    edge opposite-edge>> :> edge'
 | 
						|
    edge' next-edge>> :> edge-ring
 | 
						|
    edge-ring opposite-edge>> :> edge-ring'
 | 
						|
 | 
						|
    edge edge' assert-same-face
 | 
						|
    edge edge-ring assert-same-face
 | 
						|
    edge edge-ring' assert-different-faces
 | 
						|
 | 
						|
    b-rep new-face :> ring
 | 
						|
 | 
						|
    ring edge face>> base-face>> add-ring
 | 
						|
    ring edge' edge subdivide-edge-cycle
 | 
						|
 | 
						|
    edge b-rep delete-edge
 | 
						|
    edge' b-rep delete-edge
 | 
						|
 | 
						|
    edge-ring ;
 | 
						|
 | 
						|
TYPED:: kill-f-make-rh ( edge-face: b-edge edge-base-face: b-edge -- )
 | 
						|
    edge-face face>> :> face
 | 
						|
    edge-base-face face>> :> base-face
 | 
						|
 | 
						|
    face assert-base-face
 | 
						|
    base-face assert-base-face
 | 
						|
    edge-face edge-base-face assert-different-faces
 | 
						|
 | 
						|
    face base-face add-ring ;
 | 
						|
 | 
						|
TYPED: move-v ( edge: b-edge point: point -- )
 | 
						|
    swap vertex>> position<< ;
 | 
						|
 | 
						|
TYPED: move-e ( edge: b-edge offset: point -- )
 | 
						|
    [ dup opposite-edge>> ] dip
 | 
						|
    '[ vertex>> [ _ v+ ] change-position drop ] bi@ ;
 | 
						|
 | 
						|
TYPED: move-f ( edge: b-edge offset: point -- )
 | 
						|
    '[ vertex>> [ _ v+ ] change-position drop ] each-face-edge ;
 | 
						|
 | 
						|
TYPED: sharp-e ( edge: b-edge sharp?: boolean -- )
 | 
						|
    >>sharpness drop ;
 | 
						|
 | 
						|
TYPED: sharp-f ( edge: b-edge sharp?: boolean -- )
 | 
						|
    '[ _ sharp-e ] each-face-edge ;
 | 
						|
 | 
						|
TYPED: sharp-v ( edge: b-edge sharp?: boolean -- )
 | 
						|
    '[ _ sharp-e ] each-vertex-edge ;
 | 
						|
 | 
						|
TYPED: material-f ( edge: b-edge material -- ) 2drop ;
 |