annotate gamelib/data.py @ 267:a534629f490f default tip

Fix urls
author Neil Muller <drnlmuller@gmail.com>
date Tue, 17 Mar 2020 22:39:54 +0200
parents ae6dff8ff88c
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
1 '''Simple data loader module.
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
2
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
3 Loads data files from the "data" directory shipped with a game.
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
4
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
5 Enhancing this to handle caching etc. is left as an exercise for the reader.
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
6 '''
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
7
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
8 import os
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
9
145
53277724645b Science button juggling.
Jeremy Thurgood <firxen@gmail.com>
parents: 31
diff changeset
10 from pygame import image
53277724645b Science button juggling.
Jeremy Thurgood <firxen@gmail.com>
parents: 31
diff changeset
11
53277724645b Science button juggling.
Jeremy Thurgood <firxen@gmail.com>
parents: 31
diff changeset
12
0
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
13 data_py = os.path.abspath(os.path.dirname(__file__))
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
14 data_dir = os.path.normpath(os.path.join(data_py, '..', 'data'))
154
ae6dff8ff88c Cache loaded images
Neil Muller <drnlmuller@gmail.com>
parents: 145
diff changeset
15 IMAGE_CACHE = {}
0
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
16
1
c90a6586cd66 PEP8-ify skellington (because)
Neil Muller <drnlmuller@gmail.com>
parents: 0
diff changeset
17
0
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
18 def filepath(filename):
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
19 '''Determine the path to a file in the data directory.
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
20 '''
31
51ff46f42ed2 Add pth seperator normalisation to data helpers
Neil Muller <drnlmuller@gmail.com>
parents: 1
diff changeset
21 filename = os.path.join(*filename.split('/'))
0
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
22 return os.path.join(data_dir, filename)
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
23
1
c90a6586cd66 PEP8-ify skellington (because)
Neil Muller <drnlmuller@gmail.com>
parents: 0
diff changeset
24
0
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
25 def load(filename, mode='rb'):
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
26 '''Open a file in the data directory.
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
27
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
28 "mode" is passed as the second arg to open().
d0de8832774b Import skellington-1.9 into the repo
Neil Muller <drnlmuller+bitbucket@gmail.com>
parents:
diff changeset
29 '''
31
51ff46f42ed2 Add pth seperator normalisation to data helpers
Neil Muller <drnlmuller@gmail.com>
parents: 1
diff changeset
30 return open(filepath(filename), mode)
145
53277724645b Science button juggling.
Jeremy Thurgood <firxen@gmail.com>
parents: 31
diff changeset
31
53277724645b Science button juggling.
Jeremy Thurgood <firxen@gmail.com>
parents: 31
diff changeset
32
53277724645b Science button juggling.
Jeremy Thurgood <firxen@gmail.com>
parents: 31
diff changeset
33 def load_image(filename):
154
ae6dff8ff88c Cache loaded images
Neil Muller <drnlmuller@gmail.com>
parents: 145
diff changeset
34 if filename in IMAGE_CACHE:
ae6dff8ff88c Cache loaded images
Neil Muller <drnlmuller@gmail.com>
parents: 145
diff changeset
35 return IMAGE_CACHE[filename]
ae6dff8ff88c Cache loaded images
Neil Muller <drnlmuller@gmail.com>
parents: 145
diff changeset
36 else:
ae6dff8ff88c Cache loaded images
Neil Muller <drnlmuller@gmail.com>
parents: 145
diff changeset
37 im = image.load(filepath(filename))
ae6dff8ff88c Cache loaded images
Neil Muller <drnlmuller@gmail.com>
parents: 145
diff changeset
38 IMAGE_CACHE[filename] = im
ae6dff8ff88c Cache loaded images
Neil Muller <drnlmuller@gmail.com>
parents: 145
diff changeset
39 return im