comparison: src/objwriter.cxx
src/objwriter.cxx
- changeset 71
- 11f23fabf8a6
- child 72
- 03e4d9db3fd9
equal
deleted
inserted
replaced
| 70:fc257920ac00 | 71:11f23fabf8a6 |
|---|---|
| 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 #define __OBJWRITER_CXX__ | |
| 42 #include <stdio.h> | |
| 43 #include <stdlib.h> | |
| 44 #include <string.h> | |
| 45 #include "common.h" | |
| 46 #include "str.h" | |
| 47 #include "objwriter.h" | |
| 48 #include "databuffer.h" | |
| 49 #include "stringtable.h" | |
| 50 | |
| 51 #include "bots.h" | |
| 52 | |
| 53 extern bool g_GotMainLoop; | |
| 54 | |
| 55 ObjWriter::ObjWriter (str path) { | |
| 56 MainBuffer = new DataBuffer; | |
| 57 MainLoopBuffer = new DataBuffer; | |
| 58 OnEnterBuffer = new DataBuffer; | |
| 59 SwitchBuffer = NULL; // created on demand | |
| 60 numWrittenBytes = 0; | |
| 61 numWrittenReferences = 0; | |
| 62 filepath = path; | |
| 63 } | |
| 64 | |
| 65 void ObjWriter::WriteString (char* s) { | |
| 66 Write (strlen (s)); | |
| 67 for (unsigned int u = 0; u < strlen (s); u++) | |
| 68 Write ((s)[u]); | |
| 69 } | |
| 70 | |
| 71 void ObjWriter::WriteString (const char* s) { | |
| 72 WriteString (const_cast<char*> (s)); | |
| 73 } | |
| 74 | |
| 75 void ObjWriter::WriteString (str s) { | |
| 76 WriteString (s.chars()); | |
| 77 } | |
| 78 | |
| 79 void ObjWriter::WriteBuffer (DataBuffer* buf) { | |
| 80 GetCurrentBuffer()->Merge (buf); | |
| 81 } | |
| 82 | |
| 83 void ObjWriter::WriteBuffers () { | |
| 84 // If there was no mainloop defined, write a dummy one now. | |
| 85 if (!g_GotMainLoop) { | |
| 86 MainLoopBuffer->Write (DH_MAINLOOP); | |
| 87 MainLoopBuffer->Write (DH_ENDMAINLOOP); | |
| 88 } | |
| 89 | |
| 90 // Write the onenter and mainloop buffers, IN THAT ORDER | |
| 91 for (int i = 0; i < 2; i++) { | |
| 92 DataBuffer** buf = (!i) ? &OnEnterBuffer : &MainLoopBuffer; | |
| 93 WriteBuffer (*buf); | |
| 94 | |
| 95 // Clear the buffer afterwards for potential next state | |
| 96 *buf = new DataBuffer; | |
| 97 } | |
| 98 | |
| 99 // Next state definitely has no mainloop yet | |
| 100 g_GotMainLoop = false; | |
| 101 } | |
| 102 | |
| 103 // Write string table | |
| 104 void ObjWriter::WriteStringTable () { | |
| 105 unsigned int stringcount = CountStringTable (); | |
| 106 if (!stringcount) | |
| 107 return; | |
| 108 | |
| 109 // Write header | |
| 110 Write (DH_STRINGLIST); | |
| 111 Write (stringcount); | |
| 112 | |
| 113 // Write all strings | |
| 114 for (unsigned int a = 0; a < stringcount; a++) | |
| 115 WriteString (g_StringTable[a]); | |
| 116 } | |
| 117 | |
| 118 // Write main buffer to file | |
| 119 void ObjWriter::WriteToFile () { | |
| 120 fp = fopen (filepath, "w"); | |
| 121 CHECK_FILE (fp, filepath, "writing"); | |
| 122 | |
| 123 // First, resolve references | |
| 124 numWrittenReferences = 0; | |
| 125 for (unsigned int u = 0; u < MAX_MARKS; u++) { | |
| 126 ScriptMarkReference* ref = MainBuffer->refs[u]; | |
| 127 if (!ref) | |
| 128 continue; | |
| 129 | |
| 130 // Substitute the placeholder with the mark position | |
| 131 union_t<word> uni; | |
| 132 uni.val = static_cast<word> (MainBuffer->marks[ref->num]->pos); | |
| 133 for (unsigned int v = 0; v < sizeof (word); v++) | |
| 134 memset (MainBuffer->buffer + ref->pos + v, uni.b[v], 1); | |
| 135 | |
| 136 /* | |
| 137 printf ("reference %u at %d resolved to %u at %d\n", | |
| 138 u, ref->pos, ref->num, MainBuffer->marks[ref->num]->pos); | |
| 139 */ | |
| 140 numWrittenReferences++; | |
| 141 } | |
| 142 | |
| 143 // Then, dump the main buffer to the file | |
| 144 for (unsigned int x = 0; x < MainBuffer->writesize; x++) | |
| 145 WriteDataToFile<byte> (*(MainBuffer->buffer+x)); | |
| 146 | |
| 147 printf ("-- %u byte%s written to %s\n", numWrittenBytes, PLURAL (numWrittenBytes), filepath.chars()); | |
| 148 fclose (fp); | |
| 149 } | |
| 150 | |
| 151 DataBuffer* ObjWriter::GetCurrentBuffer() { | |
| 152 return SwitchBuffer ? SwitchBuffer : | |
| 153 (g_CurMode == MODE_MAINLOOP) ? MainLoopBuffer : | |
| 154 (g_CurMode == MODE_ONENTER) ? OnEnterBuffer : | |
| 155 MainBuffer; | |
| 156 } | |
| 157 | |
| 158 ScriptMark* g_ScriptMark = NULL; | |
| 159 | |
| 160 // Adds a mark | |
| 161 unsigned int ObjWriter::AddMark (str name) { | |
| 162 return GetCurrentBuffer()->AddMark (name); | |
| 163 } | |
| 164 | |
| 165 // Adds a reference | |
| 166 unsigned int ObjWriter::AddReference (unsigned int mark) { | |
| 167 DataBuffer* b = GetCurrentBuffer(); | |
| 168 return b->AddMarkReference (mark); | |
| 169 } | |
| 170 | |
| 171 // Finds a mark | |
| 172 unsigned int ObjWriter::FindMark (str name) { | |
| 173 DataBuffer* b = GetCurrentBuffer(); | |
| 174 for (unsigned int u = 0; u < MAX_MARKS; u++) { | |
| 175 if (b->marks[u] && !b->marks[u]->name.icompare (name)) | |
| 176 return u; | |
| 177 } | |
| 178 return MAX_MARKS; | |
| 179 } | |
| 180 | |
| 181 // Moves a mark to the current position | |
| 182 void ObjWriter::MoveMark (unsigned int mark) { | |
| 183 GetCurrentBuffer()->MoveMark (mark); | |
| 184 } | |
| 185 | |
| 186 // Deletes a mark | |
| 187 void ObjWriter::DeleteMark (unsigned int mark) { | |
| 188 GetCurrentBuffer()->DeleteMark (mark); | |
| 189 } | |
| 190 | |
| 191 void ObjWriter::OffsetMark (unsigned int mark, int offset) { | |
| 192 GetCurrentBuffer()->OffsetMark (mark, offset); | |
| 193 } |
