summaryrefslogtreecommitdiffstats
path: root/src/elements/Gradient.js
blob: 7116fc892f3e72cbd2d2c77e7e783730f451799d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import {
  extend,
  nodeOrNew,
  register,
  wrapWithAttrCheck
} from '../utils/adopter.js'
import { registerMethods } from '../utils/methods.js'
import Box from '../types/Box.js'
import Container from './Container.js'
import Stop from './Stop.js'
import baseFind from '../modules/core/selector.js'
import * as gradiented from '../modules/core/gradiented.js'

export default class Gradient extends Container {

  constructor ( type, attrs ) {

    super(
      nodeOrNew( type + 'Gradient', typeof type === 'string' ? null : type ),
      attrs
    )

  }

  // Add a color stop
  stop ( offset, color, opacity ) {

    return this.put( new Stop() ).update( offset, color, opacity )

  }

  // Update gradient
  update ( block ) {

    // remove all stops
    this.clear()

    // invoke passed block
    if ( typeof block === 'function' ) {

      block.call( this, this )

    }

    return this

  }

  // Return the fill id
  url () {

    return 'url(#' + this.id() + ')'

  }

  // Alias string convertion to fill
  toString () {

    return this.url()

  }

  // custom attr to handle transform
  attr ( a, b, c ) {

    if ( a === 'transform' ) a = 'gradientTransform'
    return super.attr( a, b, c )

  }

  targets () {

    return baseFind( 'svg [fill*="' + this.id() + '"]' )

  }

  bbox () {

    return new Box()

  }

}

extend( Gradient, gradiented )

registerMethods( {
  Container: {
    // Create gradient element in defs
    gradient: wrapWithAttrCheck( function ( type, block ) {

      return this.defs().gradient( type, block )

    } )
  },
  // define gradient
  Defs: {
    gradient: wrapWithAttrCheck( function ( type, block ) {

      return this.put( new Gradient( type ) ).update( block )

    } )
  }
} )

register( Gradient )