//----------------------------------------------------------------------------------------------------------------------
void ResetAndAssert(string fullPath, int texType)
{
StreamingImageSequencePlugin.UnloadImageAndNotify(fullPath);
ImageLoader.GetImageDataInto(fullPath, texType, out ImageData readResult);
Assert.AreEqual(C.READ_STATUS_UNAVAILABLE, readResult.ReadStatus, "ResetAndAssert");
Assert.AreEqual(0, StreamingImageSequencePlugin.GetUsedImagesMemory());
}
static void EditorApplicationManager_PlayModeStateChanged(PlayModeStateChange state)
{
ImageLoader.SetImageLoadTaskHandler(null);
if (PlayModeStateChange.EnteredEditMode != state)
{
return;
}
InitImageLoaderForEditMode();
StreamingImageSequencePlugin.ResetImageLoadOrder();
}
private static void ShowImageLoadOrder()
{
StringBuilder sb = new StringBuilder();
for (int imageType = 0; imageType < StreamingImageSequenceConstants.MAX_IMAGE_TYPES; ++imageType)
{
int latestRequestFrame = StreamingImageSequencePlugin.GetImageLoadOrder(imageType);
sb.AppendLine($"IMAGE_TYPE: {imageType.ToString()}, order: {latestRequestFrame}");
sb.AppendLine();
}
sb.AppendLine("Current Frame: " + ImageLoader.GetCurrentFrame());
Debug.Log(sb.ToString());
}
private static void ShowLoadedImages()
{
StringBuilder sb = new StringBuilder();
for (int imageType = 0; imageType < StreamingImageSequenceConstants.MAX_IMAGE_TYPES; ++imageType)
{
sb.AppendLine("IMAGE_TYPE: " + imageType.ToString());
List <string> loadedTextures = new List <string>();
StreamingImageSequencePlugin.ListLoadedImages(imageType, (fileName) => {
loadedTextures.Add(fileName);
});
foreach (var fileName in loadedTextures)
{
ImageLoader.GetImageDataInto(fileName, imageType, out ImageData readResult);
sb.Append(" ");
sb.Append(fileName);
sb.Append(". Status: " + readResult.ReadStatus);
sb.Append(", Size: (" + readResult.Width + ", " + readResult.Height);
sb.AppendLine(") ");
}
sb.AppendLine("----------------------------------------------------------------");
sb.AppendLine();
sb.AppendLine();
}
sb.AppendLine("Preview Textures: ");
IDictionary <string, PreviewTexture> previewTextures = PreviewTextureFactory.GetPreviewTextures();
foreach (var kvp in previewTextures)
{
sb.Append(" ");
sb.AppendLine(kvp.Key);
}
Debug.Log(sb.ToString());
}
//----------------------------------------------------------------------------------------------------------------------
internal static IEnumerator UpdateRenderCacheCoroutine(PlayableDirector director, RenderCachePlayableAsset renderCachePlayableAsset)
{
Assert.IsNotNull(director);
Assert.IsNotNull(renderCachePlayableAsset);
TimelineClipSISData timelineClipSISData = renderCachePlayableAsset.GetBoundTimelineClipSISData();
if (null == timelineClipSISData)
{
EditorUtility.DisplayDialog("Streaming Image Sequence",
"RenderCachePlayableAsset is not ready",
"Ok");
yield break;
}
TrackAsset track = renderCachePlayableAsset.GetBoundTimelineClipSISData().GetOwner().parentTrack;
BaseRenderCapturer renderCapturer = director.GetGenericBinding(track) as BaseRenderCapturer;
if (null == renderCapturer)
{
EditorUtility.DisplayDialog("Streaming Image Sequence",
"Please bind an appropriate RenderCapturer component to the track.",
"Ok");
yield break;
}
//begin capture
bool canCapture = renderCapturer.BeginCapture();
if (!canCapture)
{
EditorUtility.DisplayDialog("Streaming Image Sequence",
renderCapturer.GetLastErrorMessage(),
"Ok");
yield break;
}
//Check output folder
string outputFolder = renderCachePlayableAsset.GetFolder();
if (string.IsNullOrEmpty(outputFolder) || !Directory.Exists(outputFolder))
{
EditorUtility.DisplayDialog("Streaming Image Sequence",
"Invalid output folder",
"Ok");
yield break;
}
Texture capturerTex = renderCapturer.GetInternalTexture();
//Show progress in game view
GameObject progressGo = new GameObject("Blitter");
LegacyTextureBlitter blitter = progressGo.AddComponent <LegacyTextureBlitter>();
blitter.SetTexture(capturerTex);
blitter.SetCameraDepth(int.MaxValue);
TimelineClip timelineClip = timelineClipSISData.GetOwner();
double nextDirectorTime = timelineClip.start;
double timePerFrame = 1.0f / track.timelineAsset.editorSettings.fps;
int fileCounter = 0;
int numFiles = (int)Math.Ceiling(timelineClip.duration / timePerFrame) + 1;
int numDigits = MathUtility.GetNumDigits(numFiles);
string prefix = $"{timelineClip.displayName}_";
List <string> imageFileNames = new List <string>(numFiles);
//Store old files that has the same pattern
string[] existingFiles = Directory.GetFiles(outputFolder, $"*.png");
HashSet <string> filesToDelete = new HashSet <string>(existingFiles);
bool cancelled = false;
while (nextDirectorTime <= timelineClip.end && !cancelled)
{
string fileName = $"{prefix}{fileCounter.ToString($"D{numDigits}")}.png";
string outputFilePath = Path.Combine(outputFolder, fileName);
SISPlayableFrame playableFrame = timelineClipSISData.GetPlayableFrame(fileCounter);
bool captureFrame = (!timelineClipSISData.AreFrameMarkersRequested() || //if markers are not requested, capture
!File.Exists(outputFilePath) || //if file doesn't exist, capture
(null != playableFrame && playableFrame.IsUsed() && !playableFrame.IsLocked())
);
if (filesToDelete.Contains(outputFilePath))
{
filesToDelete.Remove(outputFilePath);
}
imageFileNames.Add(fileName);
if (captureFrame)
{
SetDirectorTime(director, nextDirectorTime);
//Need at least two frames in order to wait for the TimelineWindow to be updated ?
yield return(null);
yield return(null);
yield return(null);
//Unload texture because it may be overwritten
StreamingImageSequencePlugin.UnloadImageAndNotify(outputFilePath);
renderCapturer.CaptureToFile(outputFilePath);
}
nextDirectorTime += timePerFrame;
++fileCounter;
cancelled = EditorUtility.DisplayCancelableProgressBar(
"StreamingImageSequence", "Caching render results", ((float)fileCounter / numFiles));
}
if (!cancelled)
{
renderCachePlayableAsset.SetImageFileNames(imageFileNames);
//Delete old files
if (AssetDatabase.IsValidFolder(outputFolder))
{
foreach (string oldFile in filesToDelete)
{
AssetDatabase.DeleteAsset(oldFile);
}
}
else
{
foreach (string oldFile in filesToDelete)
{
File.Delete(oldFile);
}
}
}
//Notify
FolderContentsChangedNotifier.GetInstance().Notify(outputFolder);
//Cleanup
EditorUtility.ClearProgressBar();
renderCapturer.EndCapture();
ObjectUtility.Destroy(progressGo);
AssetDatabase.Refresh();
yield return(null);
}
private static void ShowUsedImageMemory()
{
Debug.Log($"Used memory for images: {StreamingImageSequencePlugin.GetUsedImagesMemory().ToString()} MB");
}
public IEnumerator UpdateRenderCachePNGInStreamingAssets()
{
PlayableDirector director = EditorUtilityTest.NewSceneWithDirector();
TimelineClip clip = EditorUtilityTest.CreateTestRenderCacheTimelineClip(director);
TimelineAsset timelineAsset = director.playableAsset as TimelineAsset;
RenderCachePlayableAsset renderCachePlayableAsset = clip.asset as RenderCachePlayableAsset;
RenderCacheTrack track = clip.GetParentTrack() as RenderCacheTrack;
Assert.IsNotNull(timelineAsset);
Assert.IsNotNull(renderCachePlayableAsset);
Assert.IsNotNull(track);
Assert.IsNotNull(Camera.main);
yield return(null);
clip.duration = (1.0f / timelineAsset.editorSettings.GetFPS());
const string OUTPUT_FOLDER = "Asset/StreamingAssets/RenderCachePNGForTestRunner";
Directory.CreateDirectory(OUTPUT_FOLDER);
renderCachePlayableAsset.SetFolder(OUTPUT_FOLDER);
GameObject cameraRenderCapturerGO = new GameObject();
CameraRenderCapturer cameraRenderCapturer = cameraRenderCapturerGO.AddComponent <CameraRenderCapturer>();
cameraRenderCapturer.SetCamera(Camera.main);
director.SetGenericBinding(track, cameraRenderCapturer);
yield return(null);
//Update RenderCache
EditorCoroutineUtility.StartCoroutineOwnerless(
RenderCachePlayableAssetInspector.UpdateRenderCacheCoroutine(director, renderCachePlayableAsset)
);
//A hack to wait until the coroutine is finished
const float TIMEOUT_SEC = 3.0f;
Assert.IsTrue(Directory.Exists(OUTPUT_FOLDER));
float prevTime = Time.realtimeSinceStartup;
while (Time.realtimeSinceStartup - prevTime < TIMEOUT_SEC)
{
yield return(null);
}
string imageFilePath = renderCachePlayableAsset.GetImageFilePath(0);
Assert.IsTrue(File.Exists(imageFilePath));
ImageLoader.RequestLoadFullImage(imageFilePath);
//Another hack to wait until the load is finished
prevTime = Time.realtimeSinceStartup;
while (Time.realtimeSinceStartup - prevTime < TIMEOUT_SEC)
{
yield return(null);
}
ImageLoader.GetImageDataInto(imageFilePath, StreamingImageSequenceConstants.IMAGE_TYPE_FULL, out ImageData imageData);
Assert.AreEqual(StreamingImageSequenceConstants.READ_STATUS_SUCCESS, imageData.ReadStatus);
yield return(null);
//cleanup
StreamingImageSequencePlugin.UnloadAllImages();
bool folderDeleted = FileUtility.DeleteFilesAndFolders(OUTPUT_FOLDER);
Assert.IsTrue(folderDeleted);
EditorUtilityTest.DestroyTestTimelineAssets(clip);
yield return(null);
}
internal void SetMaxImagesMemoryMB(int value) {
m_maxImagesMemoryMB = value;
StreamingImageSequencePlugin.SetMaxImagesMemory(m_maxImagesMemoryMB);
}
static void SISUserSettings_OnLoadRuntime() {
SISUserSettings userSettings = GetInstance();
StreamingImageSequencePlugin.SetMaxImagesMemory(userSettings.GetMaxImagesMemoryMB());
}
//----------------------------------------------------------------------------------------------------------------------
internal static IEnumerator UpdateRenderCacheCoroutine(PlayableDirector director, RenderCachePlayableAsset renderCachePlayableAsset)
{
Assert.IsNotNull(director);
Assert.IsNotNull(renderCachePlayableAsset);
PlayableFrameClipData clipData = renderCachePlayableAsset.GetBoundClipData();
if (null == clipData)
{
EditorUtility.DisplayDialog("Streaming Image Sequence",
"RenderCachePlayableAsset is not ready",
"Ok");
yield break;
}
TrackAsset track = renderCachePlayableAsset.GetBoundClipData().GetOwner().GetParentTrack();
BaseRenderCapturer renderCapturer = director.GetGenericBinding(track) as BaseRenderCapturer;
if (null == renderCapturer)
{
EditorUtility.DisplayDialog("Streaming Image Sequence",
"Please bind an appropriate RenderCapturer component to the track.",
"Ok");
yield break;
}
//Check output folder
string outputFolder = renderCachePlayableAsset.GetFolder();
if (string.IsNullOrEmpty(outputFolder) || !Directory.Exists(outputFolder))
{
EditorUtility.DisplayDialog("Streaming Image Sequence",
"Invalid output folder",
"Ok");
yield break;
}
//Check if we can capture
bool canCapture = renderCapturer.CanCaptureV();
if (!canCapture)
{
EditorUtility.DisplayDialog("Streaming Image Sequence",
renderCapturer.GetLastErrorMessage(),
"Ok");
yield break;
}
//begin capture
IEnumerator beginCapture = renderCapturer.BeginCaptureV();
while (beginCapture.MoveNext())
{
yield return(beginCapture.Current);
}
//Show progress in game view
Texture capturerTex = renderCapturer.GetInternalTexture();
RenderCachePlayableAssetEditorConfig editorConfig = renderCachePlayableAsset.GetEditorConfig();
BaseTextureBlitter blitter = CreateBlitter(capturerTex);
Material blitToScreenMat = renderCapturer.GetOrCreateBlitToScreenEditorMaterialV();
if (!blitToScreenMat.IsNullRef())
{
blitToScreenMat.SetColor(m_bgColorProperty, editorConfig.GetUpdateBGColor());
blitter.SetBlitMaterial(blitToScreenMat);
}
GameObject blitterGO = blitter.gameObject;
TimelineClip timelineClip = clipData.GetOwner();
double timePerFrame = 1.0f / track.timelineAsset.editorSettings.GetFPS();
//initial calculation of loop vars
bool captureAllFrames = editorConfig.GetCaptureAllFrames();
int fileCounter = 0;
int numFiles = (int)Math.Ceiling(timelineClip.duration / timePerFrame) + 1;
int numDigits = MathUtility.GetNumDigits(numFiles);
if (!captureAllFrames)
{
fileCounter = editorConfig.GetCaptureStartFrame();
numFiles = (editorConfig.GetCaptureEndFrame() - fileCounter) + 1;
if (numFiles <= 0)
{
EditorUtility.DisplayDialog("Streaming Image Sequence", "Invalid Start/End Frame Settings", "Ok");
yield break;
}
}
int captureStartFrame = fileCounter;
string prefix = $"{timelineClip.displayName}_";
List <WatchedFileInfo> imageFiles = new List <WatchedFileInfo>(numFiles);
//Store old files that has the same pattern
string[] existingFiles = Directory.GetFiles(outputFolder, $"*.png");
HashSet <string> filesToDelete = new HashSet <string>(existingFiles);
RenderCacheOutputFormat outputFormat = renderCachePlayableAsset.GetOutputFormat();
string outputExt = null;
switch (outputFormat)
{
case RenderCacheOutputFormat.EXR: outputExt = "exr"; break;
default: outputExt = "png"; break;;
}
bool cancelled = false;
while (!cancelled)
{
//Always recalculate from start to avoid floating point errors
double directorTime = timelineClip.start + (fileCounter * timePerFrame);
if (directorTime > timelineClip.end)
{
break;
}
if (!captureAllFrames && fileCounter > editorConfig.GetCaptureEndFrame())
{
break;
}
string fileName = $"{prefix}{fileCounter.ToString($"D{numDigits}")}.{outputExt}";
string outputFilePath = Path.Combine(outputFolder, fileName);
SISPlayableFrame playableFrame = clipData.GetPlayableFrame(fileCounter);
bool captureFrame = (!clipData.AreFrameMarkersRequested() || //if markers are not requested, capture
!File.Exists(outputFilePath) || //if file doesn't exist, capture
(null != playableFrame && playableFrame.IsUsed() && !playableFrame.IsLocked())
);
if (filesToDelete.Contains(outputFilePath))
{
filesToDelete.Remove(outputFilePath);
}
if (captureFrame)
{
SetDirectorTime(director, directorTime);
//Need at least two frames in order to wait for the TimelineWindow to be updated ?
yield return(null);
yield return(null);
yield return(null);
//Unload texture because it may be overwritten
StreamingImageSequencePlugin.UnloadImageAndNotify(outputFilePath);
renderCapturer.CaptureToFile(outputFilePath, outputFormat);
}
Assert.IsTrue(File.Exists(outputFilePath));
FileInfo fileInfo = new FileInfo(outputFilePath);
imageFiles.Add(new WatchedFileInfo(fileName, fileInfo.Length));
++fileCounter;
cancelled = EditorUtility.DisplayCancelableProgressBar(
"StreamingImageSequence", "Caching render results", ((float)(fileCounter - captureStartFrame) / numFiles));
}
if (!cancelled)
{
//Delete old files
if (AssetDatabase.IsValidFolder(outputFolder))
{
foreach (string oldFile in filesToDelete)
{
AssetDatabase.DeleteAsset(oldFile);
}
}
else
{
foreach (string oldFile in filesToDelete)
{
File.Delete(oldFile);
}
}
}
//Notify
FolderContentsChangedNotifier.GetInstance().Notify(outputFolder);
//Cleanup
EditorUtility.ClearProgressBar();
renderCapturer.EndCaptureV();
ObjectUtility.Destroy(blitterGO);
AssetDatabase.Refresh();
renderCachePlayableAsset.Reload();;
yield return(null);
}
//----------------------------------------------------------------------------------------------------------------------
private void ResetThreadsAndPlugin()
{
ThreadManager.Reset();
StreamingImageSequencePlugin.ResetPlugin();
}
//----------------------------------------------------------------------------------------------------------------------
internal static void ResetPluginAndTasks()
{
ThreadManager.Reset();
StreamingImageSequencePlugin.ResetPlugin();
EditorApplicationManager.Reset();
}
