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
|
#############################################################
## ##
## Copyright (c) 2003-2017 by The University of Queensland ##
## Centre for Geoscience Computing ##
## http://earth.uq.edu.au/centre-geoscience-computing ##
## ##
## Primary Business: Brisbane, Queensland, Australia ##
## Licensed under the Open Software License version 3.0 ##
## http://www.apache.org/licenses/LICENSE-2.0 ##
## ##
#############################################################
"""Defines the L{Arrow} base class"""
from .exception import raiseNotImplemented
from esys.lsm.util import Vec3
class ArrowExtractor(object):
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
):
self.tailPtMap = tailPtMap
self.vecMap = vecMap
self.tailRadiusMap = tailRadiusMap
self.headRadiusMap = headRadiusMap
self.headLengthMap = headLengthMap
self.modifierMap = modifierMap
self.lengthScale = lengthScale
def getTailPt(self, record):
return self.tailPtMap(record)
def getVec(self, record):
return self.vecMap(record)
def getHeadPt(self, record):
return \
(
Vec3(self.getTailPt(record))
+
(Vec3(self.getVec(record))*self.getLengthScale())
)
def getHeadRadius(self, record):
return self.headRadiusMap(record)
def getTailRadius(self, record):
return self.tailRadiusMap(record)
def getHeadLength(self, record):
return self.headLengthMap(record)
def getModifier(self, record):
return self.modifierMap(record)
def getLengthScale(self):
return self.lengthScale
|
