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
|
#############################################################
## ##
## Copyright (c) 2003-2011 by The University of Queensland ##
## Earth Systems Science Computational Centre (ESSCC) ##
## http://www.uq.edu.au/esscc ##
## ##
## Primary Business: Brisbane, Queensland, Australia ##
## Licensed under the Open Software License version 3.0 ##
## http://www.opensource.org/licenses/osl-3.0.php ##
## ##
#############################################################
from esys.lsm.vis import core
from Arrow import Arrow
class ArrowExtractor(core.ArrowExtractor):
def __init__(
self,
tailPtMap,
vecMap,
tailRadiusMap = lambda record: 0.10,
headRadiusMap = lambda record: 0.25,
headLengthMap = lambda record: 0.75,
modifierMap = lambda record: None,
lengthScale = 1.0
):
core.ArrowExtractor.__init__(
self,
tailPtMap,
vecMap,
tailRadiusMap,
headRadiusMap,
headLengthMap,
modifierMap,
lengthScale
)
def addActor(self, record, addTo):
"""
Adds the arrow-actor associated with the data in
C{record} to C{addTo}.
"""
glyph = \
Arrow(
tailPt = self.getTailPt(record),
headPt = self.getHeadPt(record),
tailRadius = self.getTailRadius(record),
headRadius = self.getHeadRadius(record),
headLength = self.getHeadLength(record)
)
glyph.apply(self.getModifier(record))
glyph.addActor(addTo)
|
