You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
240 lines
7.5 KiB
240 lines
7.5 KiB
/*==LICENSE==* |
|
|
|
CyanWorlds.com Engine - MMOG client, server and tools |
|
Copyright (C) 2011 Cyan Worlds, Inc. |
|
|
|
This program is free software: you can redistribute it and/or modify |
|
it under the terms of the GNU General Public License as published by |
|
the Free Software Foundation, either version 3 of the License, or |
|
(at your option) any later version. |
|
|
|
This program is distributed in the hope that it will be useful, |
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
GNU General Public License for more details. |
|
|
|
You should have received a copy of the GNU General Public License |
|
along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
|
|
Additional permissions under GNU GPL version 3 section 7 |
|
|
|
If you modify this Program, or any covered work, by linking or |
|
combining it with any of RAD Game Tools Bink SDK, Autodesk 3ds Max SDK, |
|
NVIDIA PhysX SDK, Microsoft DirectX SDK, OpenSSL library, Independent |
|
JPEG Group JPEG library, Microsoft Windows Media SDK, or Apple QuickTime SDK |
|
(or a modified version of those libraries), |
|
containing parts covered by the terms of the Bink SDK EULA, 3ds Max EULA, |
|
PhysX SDK EULA, DirectX SDK EULA, OpenSSL and SSLeay licenses, IJG |
|
JPEG Library README, Windows Media SDK EULA, or QuickTime SDK EULA, the |
|
licensors of this Program grant you additional |
|
permission to convey the resulting work. Corresponding Source for a |
|
non-source form of such a combination shall include the source code for |
|
the parts of OpenSSL and IJG JPEG Library used as well as that of the covered |
|
work. |
|
|
|
You can contact Cyan Worlds, Inc. by email legal@cyan.com |
|
or by snail mail at: |
|
Cyan Worlds, Inc. |
|
14617 N Newport Hwy |
|
Mead, WA 99021 |
|
|
|
*==LICENSE==*/ |
|
////////////////////////////////////////////////////////////////////////////// |
|
// |
|
// plRawPageAccessor - Dangerous little class that lets you take a |
|
// plRegistryPageNode and load the objects in raw (i.e. |
|
// as block memory buffers). |
|
// This should NOT be used in any normal app, only |
|
// utility apps that don't want to load objects in |
|
// normally (which basically means if you're not mcn, |
|
// don't use this!) |
|
// |
|
//// Why We're Bad /////////////////////////////////////////////////////////// |
|
// |
|
// To store all the raw buffers, we stuff them as pointers into the keys |
|
// themselves. This is Way Bad(tm) because those pointers are expecting |
|
// hsKeyedObjects, and what we're giving them certainly ain't those. |
|
// This is why it's only safe to use this class in a very small, controlled |
|
// environment, one where we know the keys won't be accessed in a normal |
|
// fashion so we know nobody will try to use our pointers in a bad way. |
|
// |
|
// Also assumes the current global resManager is a plRawResManager! |
|
// |
|
////////////////////////////////////////////////////////////////////////////// |
|
|
|
#include "hsTypes.h" |
|
#include "hsStream.h" |
|
#include "plRawPageAccessor.h" |
|
#include "plRawResManager.h" |
|
#include "plRawKeyedObject.h" |
|
|
|
#include "hsTemplates.h" |
|
#include "../pnKeyedObject/plKeyImp.h" |
|
#include "../plResMgr/plRegistryNode.h" |
|
#include "../plResMgr/plRegistryHelpers.h" |
|
#include "../plResMgr/plRegistrySource.h" |
|
|
|
|
|
//// Constructor/Destructor ////////////////////////////////////////////////// |
|
|
|
plRawPageAccessor::plRawPageAccessor( plRegistryPageNode *source, hsBool read ) |
|
{ |
|
fSource = source; |
|
if( read ) |
|
ReadFromSource(); |
|
} |
|
|
|
plRawPageAccessor::~plRawPageAccessor() |
|
{ |
|
Release(); |
|
} |
|
|
|
//// Iterators /////////////////////////////////////////////////////////////// |
|
|
|
class plRawReaderIter : public plRegistryKeyIterator |
|
{ |
|
public: |
|
|
|
virtual hsBool EatKey( plKey key ) |
|
{ |
|
plRawResManager *mgr = (plRawResManager *)hsgResMgr::ResMgr(); |
|
|
|
UInt32 len; |
|
plKeyImp *imp = (plKeyImp *)key; |
|
UInt8 *buffer = mgr->ReadObjectBuffer( imp, len ); |
|
|
|
// This will also set the object ptr in the key |
|
plRawKeyedObject *obj = new plRawKeyedObject( key, len, buffer ); |
|
delete [] buffer; // rawKeyedObject keeps a copy |
|
|
|
return true; |
|
} |
|
}; |
|
|
|
class plRawWriterIter : public plRegistryKeyIterator |
|
{ |
|
hsStream *fStream; |
|
|
|
public: |
|
|
|
plRawWriterIter( hsStream *stream ) : fStream( stream ) {} |
|
|
|
virtual hsBool EatKey( plKey key ) |
|
{ |
|
plRawResManager *mgr = (plRawResManager *)hsgResMgr::ResMgr(); |
|
|
|
plRawKeyedObject *obj = (plRawKeyedObject *)key->ObjectIsLoaded(); |
|
if( obj == nil ) |
|
{ |
|
// Mark the key as not written |
|
plRawKeyedObject::MarkAsEmpty( key ); |
|
return true; |
|
} |
|
|
|
obj->Write( fStream ); |
|
return true; |
|
} |
|
}; |
|
|
|
class plRawReleaseIter : public plRegistryKeyIterator |
|
{ |
|
public: |
|
|
|
virtual hsBool EatKey( plKey key ) |
|
{ |
|
plRawKeyedObject *obj = (plRawKeyedObject *)key->ObjectIsLoaded(); |
|
delete obj; |
|
|
|
return true; |
|
} |
|
}; |
|
|
|
|
|
//// Various Functions /////////////////////////////////////////////////////// |
|
|
|
void plRawPageAccessor::ReadFromSource( void ) |
|
{ |
|
if( !fSource->IsLoaded() ) |
|
fSource->LoadKeysFromSource(); |
|
|
|
plRawReaderIter iter; |
|
fSource->IterateKeys( &iter ); |
|
} |
|
|
|
void plRawPageAccessor::WriteToSource( void ) |
|
{ |
|
if( fSource->GetSource() == nil ) |
|
{ |
|
hsAssert( false, "Unable to write accessor to disk; no source defined!" ); |
|
return; |
|
} |
|
|
|
// Write out objects first |
|
hsStream *stream = fSource->GetSource()->OpenDataStream( fSource, true ); |
|
if( stream == nil ) |
|
return; |
|
|
|
plRawWriterIter writer( stream ); |
|
fSource->IterateKeys( &writer ); |
|
|
|
fSource->GetSource()->CloseDataStream( fSource ); |
|
|
|
// Now write out the keys |
|
fSource->WriteKeysToSource(); |
|
} |
|
|
|
void plRawPageAccessor::Release( void ) |
|
{ |
|
plRawReleaseIter iter; |
|
fSource->IterateKeys( &iter ); |
|
|
|
fSource->ClearKeyLists(); |
|
} |
|
|
|
void plRawPageAccessor::AddCopy( const plKey &origKey ) |
|
{ |
|
plRawResManager *mgr = (plRawResManager *)hsgResMgr::ResMgr(); |
|
plKey newKey; |
|
|
|
|
|
// Get the source object |
|
plRawKeyedObject *srcObj = (plRawKeyedObject *)origKey->ObjectIsLoaded(); |
|
|
|
// Construct a new uoid |
|
plUoid newUoid( fSource->GetPageInfo().GetLocation(), |
|
origKey->GetUoid().GetClassType(), |
|
origKey->GetUoid().GetObjectName() ); |
|
|
|
// Does it already exist? |
|
newKey = mgr->FindKey( newUoid ); |
|
if( newKey != nil ) |
|
{ |
|
// Yup, gotta get rid of old object (if there is one) |
|
plRawKeyedObject *obj = (plRawKeyedObject *)newKey->ObjectIsLoaded(); |
|
delete obj; |
|
} |
|
else |
|
{ |
|
// Nope, gotta create key first |
|
newKey = mgr->NewBlankKey( newUoid ); |
|
} |
|
|
|
// Force the key's uoid to the right uoid, now that it's in the right page |
|
( (plKeyImp *)newKey )->SetUoid( origKey->GetUoid() ); |
|
|
|
// Assign a new buffer to the key |
|
if( srcObj != nil ) |
|
{ |
|
// Will set obj pointer in key |
|
plRawKeyedObject *obj = new plRawKeyedObject( newKey, srcObj->GetBufferSize(), srcObj->GetBuffer() ); |
|
} |
|
} |
|
|
|
void plRawPageAccessor::UpdateDataVersion( plRegistryPageNode *from ) |
|
{ |
|
plPageInfo &orig = from->GetPageInfo(); |
|
|
|
fSource->GetPageInfo().SetVersion( orig.GetMajorVersion(), orig.GetMinorVersion() ); |
|
fSource->GetPageInfo().SetReleaseVersion( orig.GetReleaseVersion() ); |
|
} |
|
|
|
|