File: Object.py

package info (click to toggle)
esys-particle 2.1-4
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 7,284 kB
  • sloc: cpp: 77,304; python: 5,647; makefile: 1,176; sh: 10
file content (76 lines) | stat: -rw-r--r-- 2,246 bytes parent folder | download
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-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          ##
##                                                         ##
#############################################################

import string

from Color             import Colors
from esys.lsm.vis.core import Vec3

def writeVec(f, vec):
    f.write("<")
    f.write(string.join(map(str,vec), ","))
    f.write(">")

class Background(object):
    def __init__(self, color):
        self.color = color

    def writeSdl(self, f):
        f.write("background {")
        self.color.writeSdl(f)
        f.write("}")

class Plane(object):
    def __init__(self, posn, normal):
        self.posn   = Vec3(posn)
        self.normal = Vec3(normal)
        self.normal = self.normal/(self.normal.norm())
        self.pigment = None

    def setPigment(self, pigment):
        self.pigment = pigment

    def writeSdl(self, f):
        d = self.posn.dot(self.normal)
        f.write("\nplane {")
        writeVec(f, self.normal)
        f.write(",%s " % d)
        if (self.pigment != None):
            self.pigment.writeSdl(f)
        f.write("}")

class LightSource(object):
    def __init__(self, posn, color=Colors.White, type="shadowless"):
        self.posn  = posn
        self.color = color
        self.type  = type

    def getPosn(self):
        return self.posn

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

    def getColor(self):
        return self.color

    def setColor(self, color):
        self.color = color

    def writeSdl(self, f):
        f.write("light_source {\n<")
        f.write(string.join(map(str,self.getPosn()), ","))
        f.write(">\n")
        self.getColor().writeSdl(f)
        f.write("\n")
        f.write(str(self.type))
        f.write("\n}")