Mercurial > touhou
annotate pytouhou/utils/bitstream.pyx @ 182:20843875ad8f
(Hopefully) use difficulty as it should.
The difficulty[0] (also called rank) varies from 0 to 32 and affects
various parts of the game. The difficulty now impact those parts,
but how it is modified during the gameplay is not clear.
Such changes to the difficulty are not handled yet.
[0] http://en.touhouwiki.net/wiki/Embodiment_of_Scarlet_Devil/Gameplay#Rank
| author | Thibaut Girka <thib@sitedethib.com> |
|---|---|
| date | Tue, 25 Oct 2011 01:29:40 +0200 |
| parents | fab7ad2f0d8b |
| children | b5c7369abd7c |
| rev | line source |
|---|---|
|
52
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
1 # -*- encoding: utf-8 -*- |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
2 ## |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
3 ## Copyright (C) 2011 Thibaut Girka <thib@sitedethib.com> |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
4 ## |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
5 ## This program is free software; you can redistribute it and/or modify |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
6 ## it under the terms of the GNU General Public License as published |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
7 ## by the Free Software Foundation; version 3 only. |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
8 ## |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
9 ## This program is distributed in the hope that it will be useful, |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
10 ## but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
11 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
12 ## GNU General Public License for more details. |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
13 ## |
|
ab826bc29aa2
Add some documentation, GPLv3 headers, README and COPYING file.
Thibaut Girka <thib@sitedethib.com>
parents:
0
diff
changeset
|
14 |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
15 cdef class BitStream: |
|
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
16 cdef public io |
|
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
17 cdef public int bits |
|
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
18 cdef public int byte |
|
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
19 |
|
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
20 def __init__(BitStream self, io): |
| 0 | 21 self.io = io |
| 22 self.bits = 0 | |
| 23 self.byte = 0 | |
| 24 | |
| 25 | |
| 97 | 26 def __enter__(self): |
| 27 return self | |
| 28 | |
| 29 | |
| 30 def __exit__(self, type, value, traceback): | |
| 31 return self.io.__exit__(type, value, traceback) | |
| 32 | |
| 33 | |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
34 def seek(BitStream self, offset, whence=0): |
| 0 | 35 self.io.seek(offset, whence) |
| 36 self.byte = 0 | |
| 37 self.bits = 0 | |
| 38 | |
| 39 | |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
40 def tell(BitStream self): |
| 0 | 41 return self.io.tell() |
| 42 | |
| 43 | |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
44 def tell2(BitStream self): |
| 0 | 45 return self.io.tell(), self.bits |
| 46 | |
| 47 | |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
48 cpdef unsigned char read_bit(BitStream self): |
| 0 | 49 if not self.bits: |
| 50 self.byte = ord(self.io.read(1)) | |
| 51 self.bits = 8 | |
| 52 self.bits -= 1 | |
| 53 return (self.byte >> self.bits) & 0x01 | |
| 54 | |
| 55 | |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
56 def read(BitStream self, nb_bits): |
|
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
57 cdef unsigned int value |
| 0 | 58 value = 0 |
| 59 for i in range(nb_bits - 1, -1, -1): | |
| 60 value |= self.read_bit() << i | |
| 61 return value | |
| 62 | |
| 63 | |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
64 cpdef write_bit(BitStream self, bit): |
| 0 | 65 if self.bits == 8: |
| 66 self.io.write(chr(self.byte)) | |
| 67 self.bits = 0 | |
| 68 self.byte = 0 | |
| 69 self.byte &= ~(1 << (7 - self.bits)) | |
| 70 self.byte |= bit << (7 - self.bits) | |
| 71 self.bits += 1 | |
| 72 | |
| 73 | |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
74 def write(BitStream self, bits, nb_bits): |
| 0 | 75 for i in range(nb_bits): |
| 76 self.write_bit(bits >> (nb_bits - 1 - i) & 0x01) | |
| 77 | |
| 78 | |
|
131
fab7ad2f0d8b
Use Cython, improve performances!
Thibaut Girka <thib@sitedethib.com>
parents:
97
diff
changeset
|
79 def flush(BitStream self): |
| 0 | 80 self.io.write(chr(self.byte)) |
| 81 self.bits = 0 | |
| 82 self.byte = 0 | |
| 83 self.io.flush() | |
| 84 |