示例#1
0
bool FCollectionManager::RemoveFromCollection(FName CollectionName, ECollectionShareType::Type ShareType, const TArray<FName>& ObjectPaths, int32* OutNumRemoved)
{
	if ( OutNumRemoved )
	{
		*OutNumRemoved = 0;
	}

	if ( !ensure(ShareType < ECollectionShareType::CST_All) )
	{
		// Bad share type
		LastError = LOCTEXT("Error_Internal", "There was an internal error.");
		return false;
	}

	FCollection** CollectionPtr = CachedCollections[ShareType].Find(CollectionName);
	if (CollectionPtr != NULL)
	{
		FCollection* Collection = *CollectionPtr;
		
		TArray<FName> RemovedAssets;
		for (int32 ObjectIdx = 0; ObjectIdx < ObjectPaths.Num(); ++ObjectIdx)
		{
			const FName& ObjectPath = ObjectPaths[ObjectIdx];

			if ( Collection->RemoveAssetFromCollection(ObjectPath) )
			{
				RemovedAssets.Add(ObjectPath);
			}
		}

		if ( RemovedAssets.Num() == 0 )
		{
			// Failed to remove, none of the objects were in the collection
			LastError = LOCTEXT("Error_NotInCollection", "None of the assets were in the collection.");
			return false;
		}
			
		if ( Collection->Save(LastError) )
		{
			// Removed and saved
			if ( OutNumRemoved )
			{
				*OutNumRemoved = RemovedAssets.Num();
			}

			FCollectionNameType CollectionNameType( Collection->GetCollectionName(), ShareType );
			AssetsRemovedEvent.Broadcast( CollectionNameType, ObjectPaths );
			return true;
		}
		else
		{
			// Removed but not saved, revert the remove
			for (int32 AssetIdx = 0; AssetIdx < RemovedAssets.Num(); ++AssetIdx)
			{
				Collection->AddAssetToCollection(RemovedAssets[AssetIdx]);
			}
			return false;
		}
	}
	else
	{
		// Collection not found
		LastError = LOCTEXT("Error_DoesntExist", "The collection doesn't exist.");
		return false;
	}
}
示例#2
0
bool FCollectionManager::AddToCollection(FName CollectionName, ECollectionShareType::Type ShareType, const TArray<FName>& ObjectPaths, int32* OutNumAdded)
{
	if ( OutNumAdded )
	{
		*OutNumAdded = 0;
	}

	if ( !ensure(ShareType < ECollectionShareType::CST_All) )
	{
		// Bad share type
		LastError = LOCTEXT("Error_Internal", "There was an internal error.");
		return false;
	}

	FCollection** CollectionPtr = CachedCollections[ShareType].Find(CollectionName);
	FCollection* Collection = NULL;
	if (CollectionPtr != NULL)
	{
		Collection = *CollectionPtr;
	}
	else
	{
		// Collection doesn't exist
		LastError = LOCTEXT("Error_DoesntExist", "The collection doesn't exist.");
		return false;
	}

	if ( ensure(Collection) )
	{
		int32 NumAdded = 0;
		for (int32 ObjectIdx = 0; ObjectIdx < ObjectPaths.Num(); ++ObjectIdx)
		{
			if ( Collection->AddAssetToCollection(ObjectPaths[ObjectIdx]) )
			{
				NumAdded++;
			}
		}

		if ( NumAdded > 0 )
		{
			if ( Collection->Save(LastError) )
			{
				// Added and saved
				if ( OutNumAdded )
				{
					*OutNumAdded = NumAdded;
				}

				FCollectionNameType CollectionNameType( Collection->GetCollectionName(), ShareType );
				AssetsAddedEvent.Broadcast( CollectionNameType, ObjectPaths );
				return true;
			}
			else
			{
				// Added but not saved, revert the add
				for (int32 ObjectIdx = 0; ObjectIdx < ObjectPaths.Num(); ++ObjectIdx)
				{
					Collection->RemoveAssetFromCollection(ObjectPaths[ObjectIdx]);
				}
				return false;
			}
		}
		else
		{
			// Failed to add, all of the objects were already in the collection
			LastError = LOCTEXT("Error_AlreadyInCollection", "All of the assets were already in the collection.");
			return false;
		}
	}
	else
	{
		// Collection was never found. Perhaps it was null in the TMap
		LastError = LOCTEXT("Error_DoesntExist", "The collection doesn't exist.");
		return false;
	}
}