File: camera.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 (98 lines) | stat: -rw-r--r-- 3,208 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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#############################################################
##                                                         ##
## 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              ##
##                                                         ##
#############################################################

from __future__     import division
from   esys.lsm.vis import core
from   .object       import LightSource

class Camera(core.Camera):
    def __init__(self, posn, lookAt, angle=70.0):
        core.Camera.__init__(self)
        self.posn      = posn
        self.lookAt    = lookAt
        self.angle     = angle
        self.type      = "orthographic"
        self.right     = (-1.0,0.0,0.0) # right-handed coordinate system
        self.up        = ( 0.0,1.0,0.0)
        self.direction = ( 0.0,0.0,0.1)
        self.setZoom(1.0)

    def getPosn(self):
        return self.posn

    def setPosn(self, posn):
        self.posn = posn

    def getLookAt(self):
        return self.lookAt

    def setLookAt(self, lookAt):
        self.lookAt = lookAt

    def setDirection(self, dir):
        self.direction = dir

    def setZoom(self, factor):
        self.zoomFactor = factor

    def getZoom(self):
        return self.zoomFactor

    def setAspect(self, sizeTuple):
        ratio = abs(float(sizeTuple[0])/float(sizeTuple[1]))
        self.right = (-ratio,0.0,0.0) # right handed coordinate system
        self.up    = (0.0,1.0,0.0)

    def getUp(self):
        return core.Vec3(self.up)*(2.05/self.zoomFactor)

    def getRight(self):
        return core.Vec3(self.right)*(2.05/self.zoomFactor)

    def writeSdl(self, f):
        f.write("camera {\n")
        f.write(str(self.type))
        f.write("\nlocation <")
        f.write(str.join(",", (list(map(str,self.posn)))))
        f.write(">\ndirection <")
        f.write(str.join(",", (list(map(str,self.direction)))))
        #f.write(">\nangle ")
        #f.write(str(self.angle))
        f.write(">\nup <")
        f.write(str.join(",", (list(map(str, self.getUp())))))
        f.write(">\nright <")
        f.write(str.join(",", (list(map(str, self.getRight())))))
        f.write(
          ">\nlook_at <"
          +
          str.join(",", (list(map(str,self.lookAt))))
          +
          ">"
        )
        f.write("\n}")

class LightedCamera(Camera):
    def __init__(self, posn, lookAt, angle=70.0):
        Camera.__init__(self, posn, lookAt, angle)
        self.lightSource = LightSource(posn=self.getPosn())

    def setLightSource(self, lightSource):
        self.lightSource = lightSource
        self.lightSource.setPosn(self.getPosn())

    def setPosn(self, posn):
        Camera.setPosn(self, posn)
        self.lightSource.setPosn(posn)

    def writeSdl(self, f):
        Camera.writeSdl(self, f)
        self.lightSource.writeSdl(f)