12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811 |
- //-----------------------------------------------------------------------------
- // Copyright (c) 2013 GarageGames, LLC
- //
- // Permission is hereby granted, free of charge, to any person obtaining a copy
- // of this software and associated documentation files (the "Software"), to
- // deal in the Software without restriction, including without limitation the
- // rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
- // sell copies of the Software, and to permit persons to whom the Software is
- // furnished to do so, subject to the following conditions:
- //
- // The above copyright notice and this permission notice shall be included in
- // all copies or substantial portions of the Software.
- //
- // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
- // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
- // IN THE SOFTWARE.
- //-----------------------------------------------------------------------------
- #include "platform/platform.h"
- #include "platform/platformTLS.h"
- #include "platform/threads/thread.h"
- #include "console/console.h"
- #include "console/consoleInternal.h"
- #include "console/consoleObject.h"
- #include "io/fileStream.h"
- #include "io/resource/resourceManager.h"
- #include "console/ast.h"
- #include "collection/findIterator.h"
- #include "console/consoleTypes.h"
- #include "debug/telnetDebugger.h"
- #include "sim/simBase.h"
- #include "console/compiler.h"
- #include "string/stringStack.h"
- #include "component/dynamicConsoleMethodComponent.h"
- #include "memory/safeDelete.h"
- #include <stdarg.h>
- #ifndef _HASHTABLE_H
- #include "collection/hashTable.h"
- #endif
- static Mutex* sLogMutex;
- extern StringStack STR;
- ExprEvalState gEvalState;
- StmtNode *statementList;
- ConsoleConstructor *ConsoleConstructor::first = NULL;
- bool gWarnUndefinedScriptVariables;
- static char scratchBuffer[4096];
- CON_DECLARE_PARSER(CMD);
- // TO-DO: Console debugger stuff to be cleaned up later
- static S32 dbgGetCurrentFrame(void)
- {
- return gEvalState.stack.size() - 1;
- }
- static const char * prependDollar ( const char * name )
- {
- if(name[0] != '$')
- {
- S32 len = dStrlen(name);
- AssertFatal(len < sizeof(scratchBuffer)-2, "CONSOLE: name too long");
- scratchBuffer[0] = '$';
- dMemcpy(scratchBuffer + 1, name, len + 1);
- name = scratchBuffer;
- }
- return name;
- }
- static const char * prependPercent ( const char * name )
- {
- if(name[0] != '%')
- {
- S32 len = dStrlen(name);
- AssertFatal(len < sizeof(scratchBuffer)-2, "CONSOLE: name too long");
- scratchBuffer[0] = '%';
- dMemcpy(scratchBuffer + 1, name, len + 1);
- name = scratchBuffer;
- }
- return name;
- }
- //--------------------------------------
- void ConsoleConstructor::init(const char *cName, const char *fName, const char *usg, S32 minArgs, S32 maxArgs)
- {
- mina = minArgs;
- maxa = maxArgs;
- funcName = fName;
- usage = usg;
- className = cName;
- sc = 0; fc = 0; vc = 0; bc = 0; ic = 0;
- group = false;
- next = first;
- ns = false;
- first = this;
- }
- void ConsoleConstructor::setup()
- {
- for(ConsoleConstructor *walk = first; walk; walk = walk->next)
- {
- if(walk->sc)
- Con::addCommand(walk->className, walk->funcName, walk->sc, walk->usage, walk->mina, walk->maxa);
- else if(walk->ic)
- Con::addCommand(walk->className, walk->funcName, walk->ic, walk->usage, walk->mina, walk->maxa);
- else if(walk->fc)
- Con::addCommand(walk->className, walk->funcName, walk->fc, walk->usage, walk->mina, walk->maxa);
- else if(walk->vc)
- Con::addCommand(walk->className, walk->funcName, walk->vc, walk->usage, walk->mina, walk->maxa);
- else if(walk->bc)
- Con::addCommand(walk->className, walk->funcName, walk->bc, walk->usage, walk->mina, walk->maxa);
- else if(walk->group)
- Con::markCommandGroup(walk->className, walk->funcName, walk->usage);
- else if(walk->overload)
- Con::addOverload(walk->className, walk->funcName, walk->usage);
- else if(walk->ns)
- {
- Namespace* ns = Namespace::find(StringTable->insert(walk->className));
- if( ns )
- ns->mUsage = walk->usage;
- }
- else
- AssertFatal(false, "Found a ConsoleConstructor with an indeterminate type!");
- }
- }
- ConsoleConstructor::ConsoleConstructor(const char *className, const char *funcName, StringCallback sfunc, const char *usage, S32 minArgs, S32 maxArgs)
- {
- init(className, funcName, usage, minArgs, maxArgs);
- sc = sfunc;
- }
- ConsoleConstructor::ConsoleConstructor(const char *className, const char *funcName, IntCallback ifunc, const char *usage, S32 minArgs, S32 maxArgs)
- {
- init(className, funcName, usage, minArgs, maxArgs);
- ic = ifunc;
- }
- ConsoleConstructor::ConsoleConstructor(const char *className, const char *funcName, FloatCallback ffunc, const char *usage, S32 minArgs, S32 maxArgs)
- {
- init(className, funcName, usage, minArgs, maxArgs);
- fc = ffunc;
- }
- ConsoleConstructor::ConsoleConstructor(const char *className, const char *funcName, VoidCallback vfunc, const char *usage, S32 minArgs, S32 maxArgs)
- {
- init(className, funcName, usage, minArgs, maxArgs);
- vc = vfunc;
- }
- ConsoleConstructor::ConsoleConstructor(const char *className, const char *funcName, BoolCallback bfunc, const char *usage, S32 minArgs, S32 maxArgs)
- {
- init(className, funcName, usage, minArgs, maxArgs);
- bc = bfunc;
- }
- ConsoleConstructor::ConsoleConstructor(const char* className, const char* groupName, const char* aUsage)
- {
- init(className, groupName, usage, -1, -2);
- group = true;
- // Somewhere, the entry list is getting flipped, partially.
- // so we have to do tricks to deal with making sure usage
- // is properly populated.
- // This is probably redundant.
- static char * lastUsage = NULL;
- if(aUsage)
- lastUsage = (char *)aUsage;
- usage = lastUsage;
- }
- ConsoleConstructor::ConsoleConstructor(const char* className, const char* usage)
- {
- init(className, NULL, usage, -1, -2);
- ns = true;
- }
- // We comment out the implementation of the Con namespace when doxygenizing because
- // otherwise Doxygen decides to ignore our docs in console.h
- #ifndef DOXYGENIZING
- namespace Con
- {
- static Vector<ConsumerCallback> gConsumers(__FILE__, __LINE__);
- static DataChunker consoleLogChunker;
- static Vector<ConsoleLogEntry> consoleLog(__FILE__, __LINE__);
- static bool consoleLogLocked;
- static bool logBufferEnabled=true;
- static S32 printLevel = 10;
- static FileStream consoleLogFile;
- static const char *defLogFileName = "console.log";
- static S32 consoleLogMode = 0;
- static bool active = false;
- static bool newLogFile;
- static const char *logFileName;
- static const int MaxCompletionBufferSize = 4096;
- static char completionBuffer[MaxCompletionBufferSize];
- static char tabBuffer[MaxCompletionBufferSize] = {0};
- static SimObjectPtr<SimObject> tabObject;
- static U32 completionBaseStart;
- static U32 completionBaseLen;
- #ifdef TORQUE_MULTITHREAD
- static S32 gMainThreadID = -1;
- #endif
- /// Current script file name and root, these are registered as
- /// console variables.
- /// @{
- ///
- StringTableEntry gCurrentFile;
- StringTableEntry gCurrentRoot;
- /// @}
- ConsoleFunctionGroupBegin( Clipboard, "Miscellaneous functions to control the clipboard and clear the console.");
- ConsoleFunction( cls, void, 1, 1, "() Use the cls function to clear the console output.\n"
- "@return No return value")
- {
- if(consoleLogLocked)
- return;
- consoleLogChunker.freeBlocks();
- consoleLog.setSize(0);
- };
- ConsoleFunction( getClipboard, const char*, 1, 1, "() Use the getClipboard function to get the contents of the GUI clipboard.\n"
- "@return Returns a string equal to the current contents of the copy the clipboard, or a NULL strain if the copy clipboard is empty.\n"
- "@sa setClipboard")
- {
- return Platform::getClipboard();
- };
- ConsoleFunction( setClipboard, bool, 2, 2, "( string ) Use the setClipboard function to Set value on clipboard to string.\n"
- "@param string The new value to place in the GUI clipboard.\n"
- "@return Returns true if successful, false otherwise.\n"
- "@sa getClipoard")
- {
- return Platform::setClipboard(argv[1]);
- };
- ConsoleFunctionGroupEnd( Clipboard );
- void init()
- {
- AssertFatal(active == false, "Con::init should only be called once.");
- // Set up general init values.
- active = true;
- logFileName = NULL;
- newLogFile = true;
- gWarnUndefinedScriptVariables = false;
- sLogMutex = new Mutex;
- #ifdef TORQUE_MULTITHREAD
- // Note the main thread ID.
- gMainThreadID = ThreadManager::getCurrentThreadId();
- #endif
- // Initialize subsystems.
- Namespace::init();
- ConsoleConstructor::setup();
- // Set up the parser(s)
- CON_ADD_PARSER(CMD, "cs", true); // TorqueScript
- // Variables
- setVariable("Con::prompt", "% ");
- addVariable("Con::logBufferEnabled", TypeBool, &logBufferEnabled);
- addVariable("Con::printLevel", TypeS32, &printLevel);
- addVariable("Con::warnUndefinedVariables", TypeBool, &gWarnUndefinedScriptVariables);
- // Current script file name and root
- Con::addVariable( "Con::File", TypeString, &gCurrentFile );
- Con::addVariable( "Con::Root", TypeString, &gCurrentRoot );
- // Setup the console types.
- ConsoleBaseType::initialize();
- // And finally, the ACR...
- AbstractClassRep::initialize();
- }
- //--------------------------------------
- void shutdown()
- {
- AssertFatal(active == true, "Con::shutdown should only be called once.");
- active = false;
- consoleLogFile.close();
- Namespace::shutdown();
- SAFE_DELETE( sLogMutex );
- }
- bool isActive()
- {
- return active;
- }
- bool isMainThread()
- {
- #ifdef TORQUE_MULTITHREAD
- return ThreadManager::isCurrentThread(gMainThreadID);
- #else
- // If we're single threaded we're always in the main thread.
- return true;
- #endif
- }
- //--------------------------------------
- void getLockLog(ConsoleLogEntry *&log, U32 &size)
- {
- consoleLogLocked = true;
- log = consoleLog.address();
- size = consoleLog.size();
- }
- void unlockLog()
- {
- consoleLogLocked = false;
- }
- U32 tabComplete(char* inputBuffer, U32 cursorPos, U32 maxResultLength, bool forwardTab)
- {
- // Check for null input.
- if (!inputBuffer[0])
- {
- return cursorPos;
- }
- // Cap the max result length.
- if (maxResultLength > MaxCompletionBufferSize)
- {
- maxResultLength = MaxCompletionBufferSize;
- }
- // See if this is the same partial text as last checked.
- if (dStrcmp(tabBuffer, inputBuffer))
- {
- // If not...
- // Save it for checking next time.
- dStrcpy(tabBuffer, inputBuffer);
- // Scan backward from the cursor position to find the base to complete from.
- S32 p = cursorPos;
- while ((p > 0) && (inputBuffer[p - 1] != ' ') && (inputBuffer[p - 1] != '.') && (inputBuffer[p - 1] != '('))
- {
- p--;
- }
- completionBaseStart = p;
- completionBaseLen = cursorPos - p;
- // Is this function being invoked on an object?
- if (inputBuffer[p - 1] == '.')
- {
- // If so...
- if (p <= 1)
- {
- // Bail if no object identifier.
- return cursorPos;
- }
- // Find the object identifier.
- S32 objLast = --p;
- while ((p > 0) && (inputBuffer[p - 1] != ' ') && (inputBuffer[p - 1] != '('))
- {
- p--;
- }
- if (objLast == p)
- {
- // Bail if no object identifier.
- return cursorPos;
- }
- // Look up the object identifier.
- dStrncpy(completionBuffer, inputBuffer + p, objLast - p);
- completionBuffer[objLast - p] = 0;
- tabObject = Sim::findObject(completionBuffer);
- if (!bool(tabObject))
- {
- // Bail if not found.
- return cursorPos;
- }
- }
- else
- {
- // Not invoked on an object; we'll use the global namespace.
- tabObject = 0;
- }
- }
- // Chop off the input text at the cursor position.
- inputBuffer[cursorPos] = 0;
- // Try to find a completion in the appropriate namespace.
- const char *newText;
- if (bool(tabObject))
- {
- newText = tabObject->tabComplete(inputBuffer + completionBaseStart, completionBaseLen, forwardTab);
- }
- else
- {
- // In the global namespace, we can complete on global vars as well as functions.
- if (inputBuffer[completionBaseStart] == '$')
- {
- newText = gEvalState.globalVars.tabComplete(inputBuffer + completionBaseStart, completionBaseLen, forwardTab);
- }
- else
- {
- newText = Namespace::global()->tabComplete(inputBuffer + completionBaseStart, completionBaseLen, forwardTab);
- }
- }
- if (newText)
- {
- // If we got something, append it to the input text.
- S32 len = dStrlen(newText);
- if (len + completionBaseStart > maxResultLength)
- {
- len = maxResultLength - completionBaseStart;
- }
- dStrncpy(inputBuffer + completionBaseStart, newText, len);
- inputBuffer[completionBaseStart + len] = 0;
- // And set the cursor after it.
- cursorPos = completionBaseStart + len;
- }
- // Save the modified input buffer for checking next time.
- dStrcpy(tabBuffer, inputBuffer);
- // Return the new (maybe) cursor position.
- return cursorPos;
- }
- //------------------------------------------------------------------------------
- static void log(const char *string)
- {
- // Lock.
- MutexHandle mutex;
- if( sLogMutex )
- mutex.lock( sLogMutex, true );
- // Bail if we ain't logging.
- if (!consoleLogMode)
- {
- return;
- }
- // In mode 1, we open, append, close on each log write.
- if ((consoleLogMode & 0x3) == 1)
- {
- consoleLogFile.open(defLogFileName, FileStream::ReadWrite);
- }
- // Write to the log if its status is hunky-dory.
- if ((consoleLogFile.getStatus() == Stream::Ok) || (consoleLogFile.getStatus() == Stream::EOS))
- {
- consoleLogFile.setPosition(consoleLogFile.getStreamSize());
- // If this is the first write...
- if (newLogFile)
- {
- // Make a header.
- Platform::LocalTime lt;
- Platform::getLocalTime(lt);
- char buffer[128];
- dSprintf(buffer, sizeof(buffer), "//-------------------------- %d/%d/%d -- %02d:%02d:%02d -----\r\n",
- lt.month + 1,
- lt.monthday,
- lt.year + 1900,
- lt.hour,
- lt.min,
- lt.sec);
- consoleLogFile.write(dStrlen(buffer), buffer);
- newLogFile = false;
- if (consoleLogMode & 0x4)
- {
- // Dump anything that has been printed to the console so far.
- consoleLogMode -= 0x4;
- U32 size, line;
- ConsoleLogEntry *log;
- getLockLog(log, size);
- for (line = 0; line < size; line++)
- {
- consoleLogFile.write(dStrlen(log[line].mString), log[line].mString);
- consoleLogFile.write(2, "\r\n");
- }
- unlockLog();
- }
- }
- // Now write what we came here to write.
- consoleLogFile.write(dStrlen(string), string);
- consoleLogFile.write(2, "\r\n");
- }
- if ((consoleLogMode & 0x3) == 1)
- {
- consoleLogFile.close();
- }
- }
- //------------------------------------------------------------------------------
- #if defined( _MSC_VER )
- #include <windows.h>
- static void _outputDebugString(char* pString)
- {
- // Format string.
- char* pBuffer = pString;
- S32 stringLength = dStrlen(pString);
- pBuffer += stringLength;
- *pBuffer++ = '\r';
- *pBuffer++ = '\n';
- *pBuffer = '\0';
- stringLength = strlen(pString) + 1;
- wchar_t *wstr = new wchar_t[stringLength];
- dMemset( wstr, 0, stringLength );
- // Convert to wide string.
- Con::MultiByteToWideChar( CP_ACP, NULL, pString, -1, wstr, stringLength );
- // Output string.
- Con::OutputDebugStringW( wstr );
- delete [] wstr;
- }
- #endif
- //------------------------------------------------------------------------------
- static void _printf(ConsoleLogEntry::Level level, ConsoleLogEntry::Type type, const char* fmt)
- {
- Con::active = false;
- char buffer[4096];
- U32 offset = 0;
- if(gEvalState.traceOn && gEvalState.stack.size())
- {
- offset = gEvalState.stack.size() * 3;
- for(U32 i = 0; i < offset; i++)
- buffer[i] = ' ';
- }
- dSprintf(buffer + offset, sizeof(buffer) - offset, "%s", fmt);
- for(U32 i = 0; i < (U32)gConsumers.size(); i++)
- gConsumers[i](level, buffer);
- Platform::cprintf(buffer);
- if(logBufferEnabled || consoleLogMode)
- {
- char *pos = buffer;
- while(*pos)
- {
- if(*pos == '\t')
- *pos = '^';
- pos++;
- }
- pos = buffer;
- for(;;)
- {
- char *eofPos = dStrchr(pos, '\n');
- if(eofPos)
- *eofPos = 0;
- log(pos);
- if(logBufferEnabled && !consoleLogLocked)
- {
- ConsoleLogEntry entry;
- entry.mLevel = level;
- entry.mType = type;
- entry.mString = (const char *)consoleLogChunker.alloc(dStrlen(pos) + 1);
- dStrcpy(const_cast<char*>(entry.mString), pos);
- consoleLog.push_back(entry);
- }
- if(!eofPos)
- break;
- pos = eofPos + 1;
- }
- }
- Con::active = true;
- #if defined( _MSC_VER )
- _outputDebugString( buffer );
- #endif
- }
- //------------------------------------------------------------------------------
- class ConPrinfThreadedEvent : public SimEvent
- {
- ConsoleLogEntry::Level mLevel;
- ConsoleLogEntry::Type mType;
- char *mBuf;
- public:
- ConPrinfThreadedEvent(ConsoleLogEntry::Level level = ConsoleLogEntry::Normal, ConsoleLogEntry::Type type = ConsoleLogEntry::General, const char *buf = NULL)
- {
- mLevel = level;
- mType = type;
- if(buf)
- {
- mBuf = (char*)dMalloc(dStrlen(buf)+1);
- dMemcpy((void*)mBuf, (void*)buf, dStrlen(buf));
- mBuf[dStrlen(buf)] = 0;
- }
- else
- mBuf = NULL;
- }
- ~ConPrinfThreadedEvent()
- {
- SAFE_FREE(mBuf);
- }
- virtual void process(SimObject *object)
- {
- if(mBuf)
- {
- switch(mLevel)
- {
- case ConsoleLogEntry::Normal :
- Con::printf(mBuf);
- break;
- case ConsoleLogEntry::Warning :
- Con::warnf(mType, mBuf);
- break;
- case ConsoleLogEntry::Error :
- Con::errorf(mType, mBuf);
- break;
- case ConsoleLogEntry::NUM_CLASS :
- Con::errorf("Unhandled case NUM_CLASS");
- break;
-
- }
- }
- }
- };
- //------------------------------------------------------------------------------
- void printf(const char* fmt,...)
- {
- va_list argptr;
- va_start(argptr, fmt);
- char buf[8192];
- dVsprintf(buf, sizeof(buf), fmt, argptr);
- if(!isMainThread())
- Sim::postEvent(Sim::getRootGroup(), new ConPrinfThreadedEvent(ConsoleLogEntry::Normal, ConsoleLogEntry::General, buf), Sim::getTargetTime());
- else
- _printf(ConsoleLogEntry::Normal, ConsoleLogEntry::General, buf);
- va_end(argptr);
- }
- void warnf(ConsoleLogEntry::Type type, const char* fmt,...)
- {
- va_list argptr;
- va_start(argptr, fmt);
- char buf[8192];
- dVsprintf(buf, sizeof(buf), fmt, argptr);
- if(!isMainThread())
- Sim::postEvent(Sim::getRootGroup(), new ConPrinfThreadedEvent(ConsoleLogEntry::Warning, type, buf), Sim::getTargetTime());
- else
- _printf(ConsoleLogEntry::Warning, type, buf);
- va_end(argptr);
- }
- void errorf(ConsoleLogEntry::Type type, const char* fmt,...)
- {
- va_list argptr;
- va_start(argptr, fmt);
- char buf[8192];
- dVsprintf(buf, sizeof(buf), fmt, argptr);
- if(!isMainThread())
- Sim::postEvent(Sim::getRootGroup(), new ConPrinfThreadedEvent(ConsoleLogEntry::Error, type, buf), Sim::getTargetTime());
- else
- _printf(ConsoleLogEntry::Error, type, buf);
- va_end(argptr);
- }
- void warnf(const char* fmt,...)
- {
- va_list argptr;
- va_start(argptr, fmt);
- char buf[8192];
- dVsprintf(buf, sizeof(buf), fmt, argptr);
- if(!isMainThread())
- Sim::postEvent(Sim::getRootGroup(), new ConPrinfThreadedEvent(ConsoleLogEntry::Warning, ConsoleLogEntry::General, buf), Sim::getTargetTime());
- else
- _printf(ConsoleLogEntry::Warning, ConsoleLogEntry::General, buf);
- va_end(argptr);
- }
- void errorf(const char* fmt,...)
- {
- va_list argptr;
- va_start(argptr, fmt);
- char buf[8192];
- dVsprintf(buf, sizeof(buf), fmt, argptr);
- if(!isMainThread())
- Sim::postEvent(Sim::getRootGroup(), new ConPrinfThreadedEvent(ConsoleLogEntry::Error, ConsoleLogEntry::General, buf), Sim::getTargetTime());
- else
- _printf(ConsoleLogEntry::Error, ConsoleLogEntry::General, buf);
- va_end(argptr);
- }
- //---------------------------------------------------------------------------
- void setVariable(const char *name, const char *value)
- {
- name = prependDollar(name);
- gEvalState.globalVars.setVariable(StringTable->insert(name), value);
- }
- void setLocalVariable(const char *name, const char *value)
- {
- name = prependPercent(name);
- gEvalState.stack.last()->setVariable(StringTable->insert(name), value);
- }
- void setBoolVariable(const char *varName, bool value)
- {
- setVariable(varName, value ? "1" : "0");
- }
- void setIntVariable(const char *varName, S32 value)
- {
- char scratchBuffer[32];
- dSprintf(scratchBuffer, sizeof(scratchBuffer), "%d", value);
- setVariable(varName, scratchBuffer);
- }
- void setFloatVariable(const char *varName, F32 value)
- {
- char scratchBuffer[32];
- dSprintf(scratchBuffer, sizeof(scratchBuffer), "%.9g", value);
- setVariable(varName, scratchBuffer);
- }
- //---------------------------------------------------------------------------
- void addConsumer(ConsumerCallback consumer)
- {
- gConsumers.push_back(consumer);
- }
- // dhc - found this empty -- trying what I think is a reasonable impl.
- void removeConsumer(ConsumerCallback consumer)
- {
- for(U32 i = 0; i < (U32)gConsumers.size(); i++)
- if (gConsumers[i] == consumer)
- { // remove it from the list.
- gConsumers.erase(i);
- break;
- }
- }
- void stripColorChars(char* line)
- {
- char* c = line;
- char cp = *c;
- while (cp)
- {
- if (cp < 18)
- {
- // Could be a color control character; let's take a closer look.
- if ((cp != 8) && (cp != 9) && (cp != 10) && (cp != 13))
- {
- // Yep... copy following chars forward over this.
- char* cprime = c;
- char cpp;
- do
- {
- cpp = *++cprime;
- *(cprime - 1) = cpp;
- }
- while (cpp);
- // Back up 1 so we'll check this position again post-copy.
- c--;
- }
- }
- cp = *++c;
- }
- }
- const char *getVariable(const char *name)
- {
- // get the field info from the object..
- if(name[0] != '$' && dStrchr(name, '.') && !isFunction(name))
- {
- S32 len = dStrlen(name);
- AssertFatal(len < sizeof(scratchBuffer)-1, "Sim::getVariable - name too long");
- dMemcpy(scratchBuffer, name, len+1);
- char * token = dStrtok(scratchBuffer, ".");
- SimObject * obj = Sim::findObject(token);
- if(!obj)
- return("");
- token = dStrtok(0, ".\0");
- if(!token)
- return("");
- while(token != NULL)
- {
- const char * val = obj->getDataField(StringTable->insert(token), 0);
- if(!val)
- return("");
- token = dStrtok(0, ".\0");
- if(token)
- {
- obj = Sim::findObject(token);
- if(!obj)
- return("");
- }
- else
- return(val);
- }
- }
- name = prependDollar(name);
- return gEvalState.globalVars.getVariable(StringTable->insert(name));
- }
- const char *getLocalVariable(const char *name)
- {
- name = prependPercent(name);
- return gEvalState.stack.last()->getVariable(StringTable->insert(name));
- }
- bool getBoolVariable(const char *varName, bool def)
- {
- const char *value = getVariable(varName);
- return *value ? dAtob(value) : def;
- }
- S32 getIntVariable(const char *varName, S32 def)
- {
- const char *value = getVariable(varName);
- return *value ? dAtoi(value) : def;
- }
- F32 getFloatVariable(const char *varName, F32 def)
- {
- const char *value = getVariable(varName);
- return *value ? dAtof(value) : def;
- }
- //---------------------------------------------------------------------------
- bool addVariable(const char *name, S32 t, void *dp)
- {
- gEvalState.globalVars.addVariable(name, t, dp);
- return true;
- }
- bool removeVariable(const char *name)
- {
- name = StringTable->lookup(prependDollar(name));
- return name!=0 && gEvalState.globalVars.removeVariable(name);
- }
- //---------------------------------------------------------------------------
- void addCommand(const char *nsName, const char *name,StringCallback cb, const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace *ns = lookupNamespace(nsName);
- ns->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void addCommand(const char *nsName, const char *name,VoidCallback cb, const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace *ns = lookupNamespace(nsName);
- ns->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void addCommand(const char *nsName, const char *name,IntCallback cb, const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace *ns = lookupNamespace(nsName);
- ns->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void addCommand(const char *nsName, const char *name,FloatCallback cb, const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace *ns = lookupNamespace(nsName);
- ns->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void addCommand(const char *nsName, const char *name,BoolCallback cb, const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace *ns = lookupNamespace(nsName);
- ns->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void markCommandGroup(const char * nsName, const char *name, const char* usage)
- {
- Namespace *ns = lookupNamespace(nsName);
- ns->markGroup(name,usage);
- }
- void beginCommandGroup(const char * nsName, const char *name, const char* usage)
- {
- markCommandGroup(nsName, name, usage);
- }
- void endCommandGroup(const char * nsName, const char *name)
- {
- markCommandGroup(nsName, name, NULL);
- }
- void addOverload(const char * nsName, const char * name, const char * altUsage)
- {
- Namespace *ns = lookupNamespace(nsName);
- ns->addOverload(name,altUsage);
- }
- void addCommand(const char *name,StringCallback cb,const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace::global()->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void addCommand(const char *name,VoidCallback cb,const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace::global()->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void addCommand(const char *name,IntCallback cb,const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace::global()->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void addCommand(const char *name,FloatCallback cb,const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace::global()->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- void addCommand(const char *name,BoolCallback cb,const char *usage, S32 minArgs, S32 maxArgs)
- {
- Namespace::global()->addCommand(StringTable->insert(name), cb, usage, minArgs, maxArgs);
- }
- const char *evaluate(const char* string, bool echo, const char *fileName)
- {
- if (echo)
- Con::printf("%s%s", getVariable( "$Con::Prompt" ), string);
- if(fileName)
- fileName = StringTable->insert(fileName);
- CodeBlock *newCodeBlock = new CodeBlock();
- return newCodeBlock->compileExec(fileName, string, false, fileName ? -1 : 0);
- }
- //------------------------------------------------------------------------------
- const char *evaluatef(const char* string, ...)
- {
- const char * result = NULL;
- char * buffer = new char[4096];
- if (buffer != NULL)
- {
- va_list args;
- va_start(args, string);
- dVsprintf(buffer, 4096, string, args);
- va_end (args);
- CodeBlock *newCodeBlock = new CodeBlock();
- result = newCodeBlock->compileExec(NULL, buffer, false, 0);
- delete [] buffer;
- buffer = NULL;
- }
- return result;
- }
- const char *execute(S32 argc, const char *argv[])
- {
- #ifdef TORQUE_MULTITHREAD
- if(isMainThread())
- {
- #endif
- Namespace::Entry *ent;
- StringTableEntry funcName = StringTable->insert(argv[0]);
- ent = Namespace::global()->lookup(funcName);
- if(!ent)
- {
- warnf(ConsoleLogEntry::Script, "%s: Unknown command.", argv[0]);
- // Clean up arg buffers, if any.
- STR.clearFunctionOffset();
- return "";
- }
- const char *ret = ent->execute(argc, argv, &gEvalState);
- // Reset the function offset so the stack
- // doesn't continue to grow unnecessarily
- STR.clearFunctionOffset();
- return ret;
- #ifdef TORQUE_MULTITHREAD
- }
- else
- {
- SimConsoleThreadExecCallback cb;
- SimConsoleThreadExecEvent *evt = new SimConsoleThreadExecEvent(argc, argv, false, &cb);
- Sim::postEvent(Sim::getRootGroup(), evt, Sim::getCurrentTime());
-
- return cb.waitForResult();
- }
- #endif
- }
- //------------------------------------------------------------------------------
- const char *execute(SimObject *object, S32 argc, const char *argv[],bool thisCallOnly)
- {
- static char idBuf[16];
- if(argc < 2)
- return "";
- // [neo, 10/05/2007 - #3010]
- // Make sure we don't get recursive calls, respect the flag!
- // Should we be calling handlesMethod() first?
- if( !thisCallOnly )
- {
- DynamicConsoleMethodComponent *com = dynamic_cast<DynamicConsoleMethodComponent *>(object);
- if(com)
- com->callMethodArgList(argc, argv, false);
- }
-
- if(object->getNamespace())
- {
- StringTableEntry funcName = StringTable->insert(argv[0]);
- Namespace::Entry *ent = object->getNamespace()->lookup(funcName);
- if(ent == NULL)
- {
- //warnf(ConsoleLogEntry::Script, "%s: undefined for object '%s' - id %d", funcName, object->getName(), object->getId());
- // Clean up arg buffers, if any.
- STR.clearFunctionOffset();
- return "";
- }
- // Twiddle %this argument
- const char *oldArg1 = argv[1];
- dSprintf(idBuf, sizeof(idBuf), "%d", object->getId());
- argv[1] = idBuf;
- object->pushScriptCallbackGuard();
- SimObject *save = gEvalState.thisObject;
- gEvalState.thisObject = object;
- const char *ret = ent->execute(argc, argv, &gEvalState);
- gEvalState.thisObject = save;
- object->popScriptCallbackGuard();
- // Twiddle it back
- argv[1] = oldArg1;
- // Reset the function offset so the stack
- // doesn't continue to grow unnecessarily
- STR.clearFunctionOffset();
- return ret;
- }
- warnf(ConsoleLogEntry::Script, "Con::execute - %d has no namespace: %s", object->getId(), argv[0]);
- return "";
- }
- const char *executef(SimObject *object, S32 argc, ...)
- {
- const char *argv[128];
- va_list args;
- va_start(args, argc);
- for(S32 i = 0; i < argc; i++)
- argv[i+1] = va_arg(args, const char *);
- va_end(args);
- argv[0] = argv[1];
- argc++;
- return execute(object, argc, argv);
- }
- //------------------------------------------------------------------------------
- const char *executef(S32 argc, ...)
- {
- const char *argv[128];
- va_list args;
- va_start(args, argc);
- for(S32 i = 0; i < argc; i++)
- argv[i] = va_arg(args, const char *);
- va_end(args);
- return execute(argc, argv);
- }
- //------------------------------------------------------------------------------
- bool isFunction(const char *fn)
- {
- const char *string = StringTable->lookup(fn);
- if(!string)
- return false;
- else
- return Namespace::global()->lookup(string) != NULL;
- }
- //------------------------------------------------------------------------------
- void setLogMode(S32 newMode)
- {
- if ((newMode & 0x3) != (consoleLogMode & 0x3))
- {
- if (newMode && !consoleLogMode)
- {
- // Enabling logging when it was previously disabled.
- newLogFile = true;
- }
- if ((consoleLogMode & 0x3) == 2)
- {
- // Changing away from mode 2, must close logfile.
- consoleLogFile.close();
- }
- else if ((newMode & 0x3) == 2)
- {
- // Starting mode 2, must open logfile.
- consoleLogFile.open(defLogFileName, FileStream::Write);
- }
- consoleLogMode = newMode;
- }
- }
- Namespace *lookupNamespace(const char *ns)
- {
- if(!ns)
- return Namespace::global();
- return Namespace::find(StringTable->insert(ns));
- }
- bool linkNamespaces(const char *parent, const char *child)
- {
- Namespace *pns = lookupNamespace(parent);
- Namespace *cns = lookupNamespace(child);
- if(pns && cns)
- return cns->classLinkTo(pns);
- return false;
- }
- bool unlinkNamespaces(const char *parent, const char *child)
- {
- Namespace *pns = lookupNamespace(parent);
- Namespace *cns = lookupNamespace(child);
- if(pns && cns)
- return cns->unlinkClass(pns);
- return false;
- }
- bool classLinkNamespaces(Namespace *parent, Namespace *child)
- {
- if(parent && child)
- return child->classLinkTo(parent);
- return false;
- }
- void setData(S32 type, void *dptr, S32 index, S32 argc, const char **argv, EnumTable *tbl, BitSet32 flag)
- {
- ConsoleBaseType *cbt = ConsoleBaseType::getType(type);
- AssertFatal(cbt, "Con::setData - could not resolve type ID!");
- cbt->setData((void *) (((const char *)dptr) + index * cbt->getTypeSize()),argc, argv, tbl, flag);
- }
- const char *getData(S32 type, void *dptr, S32 index, EnumTable *tbl, BitSet32 flag)
- {
- ConsoleBaseType *cbt = ConsoleBaseType::getType(type);
- AssertFatal(cbt, "Con::getData - could not resolve type ID!");
- return cbt->getData((void *) (((const char *)dptr) + index * cbt->getTypeSize()), tbl, flag);
- }
- //------------------------------------------------------------------------------
- StringTableEntry getModNameFromPath(const char *path)
- {
- if(path == NULL || *path == 0)
- return NULL;
- char buf[1024];
- buf[0] = 0;
- if(path[0] == '/' || path[1] == ':')
- {
- // It's an absolute path
- const StringTableEntry exePath = Platform::getMainDotCsDir();
- U32 len = dStrlen(exePath);
- if(dStrnicmp(exePath, path, len) == 0)
- {
- const char *ptr = path + len + 1;
- const char *slash = dStrchr(ptr, '/');
- if(slash)
- {
- dStrncpy(buf, ptr, slash - ptr);
- buf[slash - ptr] = 0;
- }
- else
- return NULL;
- }
- else
- return NULL;
- }
- else
- {
- const char *slash = dStrchr(path, '/');
- if(slash)
- {
- dStrncpy(buf, path, slash - path);
- buf[slash - path] = 0;
- }
- else
- return NULL;
- }
- return StringTable->insert(buf);
- }
- //-----------------------------------------------------------------------------
- typedef HashMap<StringTableEntry, StringTableEntry> typePathExpandoMap;
- static typePathExpandoMap PathExpandos;
- //-----------------------------------------------------------------------------
- void addPathExpando( const char* pExpandoName, const char* pPath )
- {
- // Sanity!
- AssertFatal( pExpandoName != NULL, "Expando name cannot be NULL." );
- AssertFatal( pPath != NULL, "Expando path cannot be NULL." );
- // Fetch expando name.
- StringTableEntry expandoName = StringTable->insert( pExpandoName );
- // Fetch the length of the path.
- S32 pathLength = dStrlen(pPath);
- char pathBuffer[1024];
- // Sanity!
- if ( pathLength == 0 || pathLength >= sizeof(pathBuffer) )
- {
- Con::warnf( "Cannot add path expando '%s' with path '%s' as the path is an invalid length.", pExpandoName, pPath );
- return;
- }
- // Strip repeat slashes.
- if ( !Con::stripRepeatSlashes(pathBuffer, pPath, sizeof(pathBuffer) ) )
- {
- Con::warnf( "Cannot add path expando '%s' with path '%s' as the path is an invalid length.", pExpandoName, pPath );
- return;
- }
- // Fetch new path length.
- pathLength = dStrlen(pathBuffer);
- // Sanity!
- if ( pathLength == 0 )
- {
- Con::warnf( "Cannot add path expando '%s' with path '%s' as the path is an invalid length.", pExpandoName, pPath );
- return;
- }
- // Remove any terminating slash.
- if (pathBuffer[pathLength-1] == '/')
- pathBuffer[pathLength-1] = 0;
- // Fetch expanded path.
- StringTableEntry expandedPath = StringTable->insert( pathBuffer );
- // Info.
- #if defined(TORQUE_DEBUG)
- Con::printf("Adding path expando of '%s' as '%s'.", expandoName, expandedPath );
- #endif
- // Find any existing path expando.
- typePathExpandoMap::iterator expandoItr = PathExpandos.find( pExpandoName );
- // Does the expando exist?
- if( expandoItr != PathExpandos.end() )
- {
- // Yes, so modify the path.
- expandoItr->value = expandedPath;
- return;
- }
- // Insert expando.
- PathExpandos.insert( expandoName, expandedPath );
- }
- //-----------------------------------------------------------------------------
- StringTableEntry getPathExpando( const char* pExpandoName )
- {
- // Sanity!
- AssertFatal( pExpandoName != NULL, "Expando name cannot be NULL." );
- // Fetch expando name.
- StringTableEntry expandoName = StringTable->insert( pExpandoName );
- // Find any existing path expando.
- typePathExpandoMap::iterator expandoItr = PathExpandos.find( expandoName );
- // Does the expando exist?
- if( expandoItr != PathExpandos.end() )
- {
- // Yes, so return it.
- return expandoItr->value;
- }
- // Not found.
- return NULL;
- }
- //-----------------------------------------------------------------------------
- void removePathExpando( const char* pExpandoName )
- {
- // Sanity!
- AssertFatal( pExpandoName != NULL, "Expando name cannot be NULL." );
- // Fetch expando name.
- StringTableEntry expandoName = StringTable->insert( pExpandoName );
- // Find any existing path expando.
- typePathExpandoMap::iterator expandoItr = PathExpandos.find( expandoName );
- // Does the expando exist?
- if ( expandoItr == PathExpandos.end() )
- {
- // No, so warn.
- #if defined(TORQUE_DEBUG)
- Con::warnf("Removing path expando of '%s' but it does not exist.", expandoName );
- #endif
- return;
- }
- // Info.
- #if defined(TORQUE_DEBUG)
- Con::printf("Removing path expando of '%s' as '%s'.", expandoName, expandoItr->value );
- #endif
- // Remove expando.
- PathExpandos.erase( expandoItr );
- }
- //-----------------------------------------------------------------------------
- bool isPathExpando( const char* pExpandoName )
- {
- // Sanity!
- AssertFatal( pExpandoName != NULL, "Expando name cannot be NULL." );
- // Fetch expando name.
- StringTableEntry expandoName = StringTable->insert( pExpandoName );
- return PathExpandos.contains( expandoName );
- }
- //-----------------------------------------------------------------------------
- U32 getPathExpandoCount( void )
- {
- return PathExpandos.size();
- }
- //-----------------------------------------------------------------------------
- StringTableEntry getPathExpandoKey( U32 expandoIndex )
- {
- // Finish if index is out of range.
- if ( expandoIndex >= PathExpandos.size() )
- return NULL;
- // Find indexed iterator.
- typePathExpandoMap::iterator expandoItr = PathExpandos.begin();
- while( expandoIndex > 0 ) { ++expandoItr; --expandoIndex; }
- return expandoItr->key;
- }
- //-----------------------------------------------------------------------------
- StringTableEntry getPathExpandoValue( U32 expandoIndex )
- {
- // Finish if index is out of range.
- if ( expandoIndex >= PathExpandos.size() )
- return NULL;
- // Find indexed iterator.
- typePathExpandoMap::iterator expandoItr = PathExpandos.begin();
- while( expandoIndex > 0 ) { ++expandoItr; --expandoIndex; }
- return expandoItr->value;
- }
- //-----------------------------------------------------------------------------
- bool expandPath( char* pDstPath, U32 size, const char* pSrcPath, const char* pWorkingDirectoryHint, const bool ensureTrailingSlash )
- {
- char pathBuffer[2048];
- const char* pSrc = pSrcPath;
- char* pSlash;
- // Fetch leading character.
- const char leadingToken = *pSrc;
- // Fetch following token.
- const char followingToken = leadingToken != 0 ? pSrc[1] : 0;
- // Expando.
- if ( leadingToken == '^' )
- {
- // Initial prefix search.
- const char* pPrefixSrc = pSrc+1;
- char* pPrefixDst = pathBuffer;
- // Search for end of expando.
- while( *pPrefixSrc != '/' && *pPrefixSrc != 0 )
- {
- // Copy prefix character.
- *pPrefixDst++ = *pPrefixSrc++;
- }
- // Yes, so terminate the expando string.
- *pPrefixDst = 0;
- // Fetch the expando path.
- StringTableEntry expandoPath = getPathExpando(pathBuffer);
-
- // Does the expando exist?
- if( expandoPath == NULL )
- {
- // No, so error.
- Con::errorf("expandPath() : Could not find path expando '%s' for path '%s'.", pathBuffer, pSrcPath );
- // Are we ensuring the trailing slash?
- if ( ensureTrailingSlash )
- {
- // Yes, so ensure it.
- Con::ensureTrailingSlash( pDstPath, pSrcPath );
- }
- else
- {
- // No, so just use the source path.
- dStrcpy( pDstPath, pSrcPath );
- }
- return false;
- }
- // Skip the expando and the following slash.
- pSrc += dStrlen(pathBuffer) + 1;
- // Format the output path.
- dSprintf( pathBuffer, sizeof(pathBuffer), "%s/%s", expandoPath, pSrc );
- // Are we ensuring the trailing slash?
- if ( ensureTrailingSlash )
- {
- // Yes, so ensure it.
- Con::ensureTrailingSlash( pathBuffer, pathBuffer );
- }
- // Strip repeat slashes.
- Con::stripRepeatSlashes( pDstPath, pathBuffer, size );
- return true;
- }
- // Script-Relative.
- if ( leadingToken == '.' )
- {
- // Fetch the code-block file-path.
- const StringTableEntry codeblockFullPath = CodeBlock::getCurrentCodeBlockFullPath();
- // Do we have a code block full path?
- if( codeblockFullPath == NULL )
- {
- // No, so error.
- Con::errorf("expandPath() : Could not find relative path from code-block for path '%s'.", pSrcPath );
- // Are we ensuring the trailing slash?
- if ( ensureTrailingSlash )
- {
- // Yes, so ensure it.
- Con::ensureTrailingSlash( pDstPath, pSrcPath );
- }
- else
- {
- // No, so just use the source path.
- dStrcpy( pDstPath, pSrcPath );
- }
- return false;
- }
- // Yes, so use it as the prefix.
- dStrcpy(pathBuffer, codeblockFullPath );
- // Find the final slash in the code-block.
- pSlash = dStrrchr(pathBuffer, '/');
- // Is this a parent directory token?
- if ( followingToken == '.' )
- {
- // Yes, so terminate after the slash so we include it.
- pSlash[1] = 0;
- }
- else
- {
- // No, it's a current directory token so terminate at the slash so we don't include it.
- pSlash[0] = 0;
- // Skip the current directory token.
- pSrc++;
- }
-
- // Format the output path.
- dSprintf( pathBuffer, sizeof(pathBuffer), "%s/%s", pathBuffer, pSrc );
- // Are we ensuring the trailing slash?
- if ( ensureTrailingSlash )
- {
- // Yes, so ensure it.
- Con::ensureTrailingSlash( pathBuffer, pathBuffer );
- }
- // Strip repeat slashes.
- Con::stripRepeatSlashes( pDstPath, pathBuffer, size );
- return true;
- }
- // All else.
- Platform::makeFullPathName( pSrcPath, pathBuffer, sizeof(pathBuffer), pWorkingDirectoryHint );
- // Are we ensuring the trailing slash?
- if ( ensureTrailingSlash )
- {
- // Yes, so ensure it.
- Con::ensureTrailingSlash( pathBuffer, pathBuffer );
- }
- // Strip repeat slashes.
- Con::stripRepeatSlashes( pDstPath, pathBuffer, size );
- return true;
- }
- //-----------------------------------------------------------------------------
- bool isBasePath( const char* SrcPath, const char* pBasePath )
- {
- char expandBuffer[1024];
- Con::expandPath( expandBuffer, sizeof(expandBuffer), SrcPath );
- return dStrnicmp(pBasePath, expandBuffer, dStrlen( pBasePath ) ) == 0;
- }
- //-----------------------------------------------------------------------------
- void collapsePath( char* pDstPath, U32 size, const char* pSrcPath, const char* pWorkingDirectoryHint )
- {
- // Check path against expandos. If there are multiple matches, choose the
- // expando that produces the shortest relative path.
- char pathBuffer[2048];
- // Fetch expando count.
- const U32 expandoCount = getPathExpandoCount();
- // Iterate expandos.
- U32 expandoRelativePathLength = U32_MAX;
- for( U32 expandoIndex = 0; expandoIndex < expandoCount; ++expandoIndex )
- {
- // Fetch expando value (path).
- StringTableEntry expandoValue = getPathExpandoValue( expandoIndex );
- // Skip if not the base path.
- if ( !isBasePath( pSrcPath, expandoValue ) )
- continue;
- // Fetch path relative to expando path.
- StringTableEntry relativePath = Platform::makeRelativePathName( pSrcPath, expandoValue );
- // If the relative path is simply a period
- if ( relativePath[0] == '.' )
- relativePath++;
- if ( dStrlen(relativePath) > expandoRelativePathLength )
- {
- // This expando covers less of the path than any previous one found.
- // We will keep the previous one.
- continue;
- }
- // Keep track of the relative path length
- expandoRelativePathLength = dStrlen(relativePath);
- // Fetch expando key (name).
- StringTableEntry expandoName = getPathExpandoKey( expandoIndex );
- // Format against expando.
- dSprintf( pathBuffer, sizeof(pathBuffer), "^%s/%s", expandoName, relativePath );
- }
- // Check if we've found a suitable expando
- if ( expandoRelativePathLength != U32_MAX )
- {
- // Strip repeat slashes.
- Con::stripRepeatSlashes( pDstPath, pathBuffer, size );
- return;
- }
- // Fetch the working directory.
- StringTableEntry workingDirectory = pWorkingDirectoryHint != NULL ? pWorkingDirectoryHint : Platform::getCurrentDirectory();
- // Fetch path relative to current directory.
- StringTableEntry relativePath = Platform::makeRelativePathName( pSrcPath, workingDirectory );
- // If the relative path is simply a period
- if ( relativePath[0] == '.' && relativePath[1] != '.' )
- relativePath++;
- // Format against expando.
- dSprintf( pathBuffer, sizeof(pathBuffer), "%s/%s", workingDirectory, relativePath );
- // Strip repeat slashes.
- Con::stripRepeatSlashes( pDstPath, pathBuffer, size );
- }
- //-----------------------------------------------------------------------------
- void ensureTrailingSlash( char* pDstPath, const char* pSrcPath )
- {
- // Copy to target.
- dStrcpy( pDstPath, pSrcPath );
- // Find trailing character index.
- S32 trailIndex = dStrlen(pDstPath);
- // Ignore if empty string.
- if ( trailIndex == 0 )
- return;
- // Finish if the trailing slash already exists.
- if ( pDstPath[trailIndex-1] == '/' )
- return;
- // Add trailing slash.
- pDstPath[trailIndex++] = '/';
- pDstPath[trailIndex] = 0;
- }
- //-----------------------------------------------------------------------------
- bool stripRepeatSlashes( char* pDstPath, const char* pSrcPath, S32 dstSize )
- {
- // Note original destination.
- char* pOriginalDst = pDstPath;
- // Reset last source character.
- char lastSrcChar = 0;
- // Search source...
- while ( dstSize > 0 )
- {
- // Fetch characters.
- const char srcChar = *pSrcPath++;
- // Do we have a repeat slash?
- if ( srcChar == '/' && lastSrcChar == '/' )
- {
- // Yes, so skip it.
- continue;
- }
- // No, so copy character.
- *pDstPath++ = srcChar;
- // Finish if end of source.
- if ( srcChar == 0 )
- return true;
- // Reduce room left in destination.
- dstSize--;
- // Set last character.
- lastSrcChar = srcChar;
- }
- // Terminate the destination string as we ran out of room.
- *pOriginalDst = 0;
- // Fail!
- return false;
- }
- //-----------------------------------------------------------------------------
- ConsoleFunction(expandPath, const char*, 2, 2, "(string path) - Expands an expando or relative path into a full path.")
- {
- char* ret = Con::getReturnBuffer( 1024 );
- expandPath(ret, 1024, argv[1]);
- return ret;
- }
- //-----------------------------------------------------------------------------
- ConsoleFunction(collapsePath, const char*, 2, 2, "(string path) - Collapses a path into either an expando path or a relative path.")
- {
- char* ret = Con::getReturnBuffer( 1024 );
- Con::collapsePath(ret, 1024, argv[1]);
- return ret;
- }
- //-----------------------------------------------------------------------------
- ConsoleFunction(addPathExpando, void, 3, 3, "(string expando, string path) - Adds the expando to the path. If it already exists then it is replaced.")
- {
- addPathExpando(argv[1], argv[2]);
- }
- //-----------------------------------------------------------------------------
- ConsoleFunction(removePathExpando, void, 2, 2, "(string expando) - Removes the specified path expando.")
- {
- removePathExpando(argv[1]);
- }
- //-----------------------------------------------------------------------------
- ConsoleFunction(isPathExpando, bool, 2, 2, "(string expando) - Checks whether the specified path expando is current set or not.")
- {
- return isPathExpando(argv[1]);
- }
- //-----------------------------------------------------------------------------
- ConsoleFunction(getPathExpandoCount, S32, 1, 1, "() - Gets the expando path count.")
- {
- return getPathExpandoCount();
- }
- //-----------------------------------------------------------------------------
- ConsoleFunction(getPathExpandoKey, const char*, 2, 2, "(int expandoIndex) - Gets the path expando key (the expando name) at the specified index.")
- {
- // Fetch expando index.
- const S32 expandoIndex = dAtoi(argv[1]);
- // Is the expando index in range?
- if ( expandoIndex < 0 || expandoIndex >= (S32)getPathExpandoCount() )
- {
- // No, so warn.
- Con::warnf("getPathExpandoKey() - Expando index of '%d' is out of bounds. Current expando count is '%d'.",
- expandoIndex,
- getPathExpandoCount() );
- return StringTable->EmptyString;
- }
- // Fetch path expando key.
- return getPathExpandoKey( expandoIndex );
- }
- //-----------------------------------------------------------------------------
- ConsoleFunction(getPathExpandoValue, const char*, 2, 2, "(int expandoIndex) - Gets the path expando value (the expando path) at the specified index.")
- {
- // Fetch expando index.
- const S32 expandoIndex = dAtoi(argv[1]);
- // Is the expando index in range?
- if ( expandoIndex < 0 || expandoIndex >= (S32)getPathExpandoCount() )
- {
- // No, so warn.
- Con::warnf("getPathExpandoValue() - Expando index of '%d' is out of bounds. Current expando count is '%d'.",
- expandoIndex,
- getPathExpandoCount() );
- return StringTable->EmptyString;
- }
- // Fetch path expando value.
- return getPathExpandoValue( expandoIndex );
- }
- } // end of Console namespace
- #endif
|