Added AIFetchInventoryFolder and more.
Added a new statemachine AIFetchInventoryFolder, which can be used to fetch the contents of a folder by name or UUID. Also added AIEvent (and one event, AIEvent::LLInventoryModel_mIsAgentInvUsable_true, which is needed for AIFetchInventoryFolder). Fixed LLInventoryModel::sBackgroundFetchActive to correctly reflect whether or not LLInventoryModel::backgroundFetch is added to gIdleCallbacks. Avoid duplicated entries in sFetchQueue. Reset sFullFetchStarted in LLInventoryModel::stopBackgroundFetch to allow for a renewed full fetch when some single-folder fetch stops it. Added AIStateMachine::mQueued to make calling 'cont()' more robust: calling cont() / idle() / cont() on a row would otherwise add a statemachine twice to the active list, which would cause a crash when it's killed.
This commit is contained in:
Aleric Inglewood
193
indra/newview/statemachine/aifetchinventoryfolder.cpp
Normal file
193
indra/newview/statemachine/aifetchinventoryfolder.cpp
Normal file
@@ -0,0 +1,193 @@
|
||||
/**
|
||||
* @file aifetchinventoryfolder.cpp
|
||||
* @brief Implementation of AIFetchInventoryFolder
|
||||
*
|
||||
* Copyright (c) 2011, Aleric Inglewood.
|
||||
*
|
||||
* 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 2 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/>.
|
||||
*
|
||||
* There are special exceptions to the terms and conditions of the GPL as
|
||||
* it is applied to this Source Code. View the full text of the exception
|
||||
* in the file doc/FLOSS-exception.txt in this software distribution.
|
||||
*
|
||||
* CHANGELOG
|
||||
* and additional copyright holders.
|
||||
*
|
||||
* 19/05/2011
|
||||
* Initial version, written by Aleric Inglewood @ SL
|
||||
*/
|
||||
|
||||
#include "linden_common.h"
|
||||
#include "aifetchinventoryfolder.h"
|
||||
#include "aievent.h"
|
||||
#include "llagent.h"
|
||||
|
||||
enum fetchinventoryfolder_state_type {
|
||||
AIFetchInventoryFolder_checkFolderExists = AIStateMachine::max_state,
|
||||
AIFetchInventoryFolder_fetchDescendents,
|
||||
AIFetchInventoryFolder_folderCompleted
|
||||
};
|
||||
|
||||
class AIInventoryFetchDescendentsObserver : public LLInventoryFetchDescendentsObserver {
|
||||
public:
|
||||
AIInventoryFetchDescendentsObserver(AIStateMachine* statemachine, LLUUID const& folder);
|
||||
~AIInventoryFetchDescendentsObserver() { gInventory.removeObserver(this); }
|
||||
|
||||
protected:
|
||||
/*virtual*/ void done()
|
||||
{
|
||||
mStateMachine->set_state(AIFetchInventoryFolder_folderCompleted);
|
||||
delete this;
|
||||
}
|
||||
|
||||
private:
|
||||
AIStateMachine* mStateMachine;
|
||||
};
|
||||
|
||||
AIInventoryFetchDescendentsObserver::AIInventoryFetchDescendentsObserver(AIStateMachine* statemachine, LLUUID const& folder) : mStateMachine(statemachine)
|
||||
{
|
||||
mStateMachine->idle();
|
||||
folder_ref_t folders(1, folder);
|
||||
fetchDescendents(folders);
|
||||
gInventory.addObserver(this);
|
||||
if (isEverythingComplete())
|
||||
done();
|
||||
}
|
||||
|
||||
void AIFetchInventoryFolder::fetch(std::string const& foldername, bool create, bool fetch_contents)
|
||||
{
|
||||
fetch(gAgent.getInventoryRootID(), foldername, create, fetch_contents);
|
||||
}
|
||||
|
||||
char const* AIFetchInventoryFolder::state_str_impl(state_type run_state) const
|
||||
{
|
||||
switch(run_state)
|
||||
{
|
||||
AI_CASE_RETURN(AIFetchInventoryFolder_checkFolderExists);
|
||||
AI_CASE_RETURN(AIFetchInventoryFolder_fetchDescendents);
|
||||
AI_CASE_RETURN(AIFetchInventoryFolder_folderCompleted);
|
||||
}
|
||||
return "UNKNOWN STATE";
|
||||
}
|
||||
|
||||
void AIFetchInventoryFolder::initialize_impl(void)
|
||||
{
|
||||
mExists = false;
|
||||
mCreated = false;
|
||||
mNeedNotifyObservers = false;
|
||||
set_state(AIFetchInventoryFolder_checkFolderExists);
|
||||
if (!gInventory.isInventoryUsable())
|
||||
AIEvent::Register(AIEvent::LLInventoryModel_mIsAgentInvUsable_true, this);
|
||||
}
|
||||
|
||||
void AIFetchInventoryFolder::multiplex_impl(void)
|
||||
{
|
||||
switch (mRunState)
|
||||
{
|
||||
case AIFetchInventoryFolder_checkFolderExists:
|
||||
{
|
||||
// If LLInventoryModel_mIsAgentInvUsable_true then this should be and stay true forever.
|
||||
llassert(gInventory.isInventoryUsable());
|
||||
if (mParentFolder.isNull())
|
||||
mParentFolder = gAgent.getInventoryRootID();
|
||||
if (mFolderUUID.isNull() || !gInventory.getCategory(mFolderUUID)) // Is the UUID unknown, or doesn't exist?
|
||||
{
|
||||
// Set this to null here in case we abort.
|
||||
mFolderUUID.setNull();
|
||||
if (mFolderName.empty())
|
||||
{
|
||||
// We can only find a folder by name, or create it, if we know it's name.
|
||||
llwarns << "Unknown folder ID " << mFolderUUID << llendl;
|
||||
abort();
|
||||
break;
|
||||
}
|
||||
// Check if the parent exists.
|
||||
if (mParentFolder != gAgent.getInventoryRootID() && !gInventory.getCategory(mParentFolder))
|
||||
{
|
||||
llwarns << "Unknown parent folder ID " << mParentFolder << llendl;
|
||||
abort();
|
||||
break;
|
||||
}
|
||||
// Look up UUID by name.
|
||||
LLInventoryModel::cat_array_t* categories;
|
||||
gInventory.getDirectDescendentsOf(mParentFolder, categories);
|
||||
for (S32 i = 0; i < categories->getLength(); ++i)
|
||||
{
|
||||
LLPointer<LLViewerInventoryCategory> const& category(categories->get(i));
|
||||
if (category->getName() == mFolderName)
|
||||
{
|
||||
mFolderUUID = category->getUUID();
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (mFolderUUID.isNull()) // Does the folder exist?
|
||||
{
|
||||
if (!mCreate)
|
||||
{
|
||||
// We're done.
|
||||
finish();
|
||||
break;
|
||||
}
|
||||
// Create the folder.
|
||||
mFolderUUID = gInventory.createNewCategory(mParentFolder, LLAssetType::AT_NONE, mFolderName);
|
||||
llassert_always(!mFolderUUID.isNull());
|
||||
Dout(dc::statemachine, "Created folder \"" << mFolderName << "\".");
|
||||
mNeedNotifyObservers = true;
|
||||
}
|
||||
mCreated = true;
|
||||
}
|
||||
// mFolderUUID is now valid.
|
||||
mExists = true;
|
||||
if (!mFetchContents || // No request to fetch contents.
|
||||
LLInventoryModel::isEverythingFetched()) // No need to fetch contents.
|
||||
{
|
||||
// We're done.
|
||||
finish();
|
||||
break;
|
||||
}
|
||||
set_state(AIFetchInventoryFolder_fetchDescendents);
|
||||
/*Fall-through*/
|
||||
}
|
||||
case AIFetchInventoryFolder_fetchDescendents:
|
||||
{
|
||||
// This sets the state to AIFetchInventoryFolder_folderCompleted once the folder is complete.
|
||||
new AIInventoryFetchDescendentsObserver(this, mFolderUUID);
|
||||
break;
|
||||
}
|
||||
case AIFetchInventoryFolder_folderCompleted:
|
||||
{
|
||||
// Does it still exist?
|
||||
if (!gInventory.getCategory(mFolderUUID))
|
||||
{
|
||||
// Assume the folder was deleted in the meantime.
|
||||
abort();
|
||||
break;
|
||||
}
|
||||
llassert(gInventory.isCategoryComplete(mFolderUUID));
|
||||
// The folder is complete!
|
||||
finish();
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void AIFetchInventoryFolder::abort_impl(void)
|
||||
{
|
||||
}
|
||||
|
||||
void AIFetchInventoryFolder::finish_impl(void)
|
||||
{
|
||||
if (mNeedNotifyObservers)
|
||||
gInventory.notifyObservers();
|
||||
}
|
||||
Reference in New Issue
Block a user