File: arrow.py

package info (click to toggle)
esys-particle 2.3.5%2Bdfsg2-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 13,132 kB
  • sloc: cpp: 81,480; python: 5,872; makefile: 1,259; sh: 313; perl: 225
file content (76 lines) | stat: -rw-r--r-- 2,543 bytes parent folder | download | duplicates (4)
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
#############################################################
##                                                         ##
## 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 Arrow(object):
    """
    Objects of this class represent arrows with conical head
    and cylindrical tail.
    """
    def __init__(self, tailPt, headPt, tailRadius, headRadius, headLength):
        """
        Initialises arrow.
        """
        self.headPt     = tuple(headPt)
        self.tailPt     = tuple(tailPt)
        self.tailRadius = tailRadius
        self.headRadius = headRadius
        self.headLength = headLength

    def getTailPt(self):
        """
        Returns the coordinate of the tail of this arrow.
        @return: Tail coordinate of this arrow.
        """
        return self.tailPt

    def getHeadPt(self):
        """
        Returns the coordinate of the pointy head of this arrow.
        @return: Head coordinate of this arrow.
        """
        return self.headPt

    def getHeadRadius(self):
        """
        Returns the max radius of the head of this arrow.
        @return: Radius of base of this arrow's conical head.
        """
        return self.headRadius

    def getTailRadius(self):
        """
        Returns the radius of the cylindrical tail of this arrow.
        @return: Radius of this arrow's cylindrical tail.
        """
        return self.tailRadius

    def getHeadLength(self):
        """
        Returns the height of the cone which forms the head of this arrow.
        @return: Height of cone which forms the head of this arrow.
        """
        return self.headLength

    def getLength(self):
        """
        Returns the distance between this arrows head-point and tail-point.
        @rtype: float
        @return: length of this arrow from head-point to tail-point.
        """
        return (Vec3(self.getHeadPt())-Vec3(self.getTailPt())).norm()