1 /* |
|
2 * botc source code |
|
3 * Copyright (C) 2012 Santeri `Dusk` Piippo |
|
4 * All rights reserved. |
|
5 * |
|
6 * Redistribution and use in source and binary forms, with or without |
|
7 * modification, are permitted provided that the following conditions are met: |
|
8 * |
|
9 * 1. Redistributions of source code must retain the above copyright notice, |
|
10 * this list of conditions and the following disclaimer. |
|
11 * 2. Redistributions in binary form must reproduce the above copyright notice, |
|
12 * this list of conditions and the following disclaimer in the documentation |
|
13 * and/or other materials provided with the distribution. |
|
14 * 3. Neither the name of the developer nor the names of its contributors may |
|
15 * be used to endorse or promote products derived from this software without |
|
16 * specific prior written permission. |
|
17 * 4. Redistributions in any form must be accompanied by information on how to |
|
18 * obtain complete source code for the software and any accompanying |
|
19 * software that uses the software. The source code must either be included |
|
20 * in the distribution or be available for no more than the cost of |
|
21 * distribution plus a nominal fee, and must be freely redistributable |
|
22 * under reasonable conditions. For an executable file, complete source |
|
23 * code means the source code for all modules it contains. It does not |
|
24 * include source code for modules or files that typically accompany the |
|
25 * major components of the operating system on which the executable file |
|
26 * runs. |
|
27 * |
|
28 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
|
29 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
30 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
31 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE |
|
32 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
|
33 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
|
34 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
|
35 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
|
36 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
|
37 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
|
38 * POSSIBILITY OF SUCH DAMAGE. |
|
39 */ |
|
40 |
|
41 #ifndef __OBJWRITER_H__ |
|
42 #define __OBJWRITER_H__ |
|
43 |
|
44 #include <stdio.h> |
|
45 #include <typeinfo> |
|
46 #include <string.h> |
|
47 #include "common.h" |
|
48 #include "str.h" |
|
49 #include "databuffer.h" |
|
50 |
|
51 class ObjWriter { |
|
52 public: |
|
53 // ==================================================================== |
|
54 // MEMBERS |
|
55 |
|
56 // Pointer to the file we're writing to |
|
57 FILE* fp; |
|
58 |
|
59 // Path to the file we're writing to |
|
60 str filepath; |
|
61 |
|
62 // The main buffer - the contents of this is what we |
|
63 // write to file after parsing is complete |
|
64 DataBuffer* MainBuffer; |
|
65 |
|
66 // onenter buffer - the contents of the onenter{} block |
|
67 // is buffered here and is merged further at the end of state |
|
68 DataBuffer* OnEnterBuffer; |
|
69 |
|
70 // Mainloop buffer - the contents of the mainloop{} block |
|
71 // is buffered here and is merged further at the end of state |
|
72 DataBuffer* MainLoopBuffer; |
|
73 |
|
74 // Switch buffer - switch case data is recorded to this |
|
75 // buffer initially, instead of into main buffer. |
|
76 DataBuffer* SwitchBuffer; |
|
77 |
|
78 // How many bytes have we written to file? |
|
79 unsigned int numWrittenBytes; |
|
80 |
|
81 // How many references did we resolve in the main buffer? |
|
82 unsigned int numWrittenReferences; |
|
83 |
|
84 // ==================================================================== |
|
85 // METHODS |
|
86 ObjWriter (str path); |
|
87 void WriteString (char* s); |
|
88 void WriteString (const char* s); |
|
89 void WriteString (str s); |
|
90 void WriteBuffer (DataBuffer* buf); |
|
91 void WriteBuffers (); |
|
92 void WriteStringTable (); |
|
93 void WriteToFile (); |
|
94 DataBuffer* GetCurrentBuffer (); |
|
95 |
|
96 unsigned int AddMark (str name); |
|
97 unsigned int FindMark (str name); |
|
98 unsigned int AddReference (unsigned int mark); |
|
99 void MoveMark (unsigned int mark); |
|
100 void OffsetMark (unsigned int mark, int offset); |
|
101 void DeleteMark (unsigned int mark); |
|
102 template <class T> void DoWrite (const char* func, T stuff) { |
|
103 GetCurrentBuffer ()->DoWrite (func, stuff); |
|
104 } |
|
105 |
|
106 // Default to word |
|
107 void DoWrite (const char* func, word stuff) { |
|
108 DoWrite<word> (func, stuff); |
|
109 } |
|
110 |
|
111 void DoWrite (const char* func, byte stuff) { |
|
112 DoWrite<byte> (func, stuff); |
|
113 } |
|
114 |
|
115 private: |
|
116 // Write given data to file. |
|
117 template <class T> void WriteDataToFile (T stuff) { |
|
118 // One byte at a time |
|
119 union_t<T> uni; |
|
120 uni.val = stuff; |
|
121 for (unsigned int x = 0; x < sizeof (T); x++) { |
|
122 fwrite (&uni.b[x], 1, 1, fp); |
|
123 numWrittenBytes++; |
|
124 } |
|
125 } |
|
126 }; |
|
127 |
|
128 #endif // __OBJWRITER_H__ |
|