/**

 * Program: ResolveMXFMedia.cs

 * 

 * This script will fix a folder renaming problem when using XDCAM discs.

 * It does this by finding the correct disc folders and updating the project

 * media pool.

 *

 *    Author: John Rofrano

 * Copyright: (c) 2008 VASST, All Rights Reserved

 *   Version: 1.1

 *  Revision: Aug. 6, 2008

**/



using System;

using System.IO;

using System.Collections;

using System.Windows.Forms;

using Sony.Vegas;



class EntryPoint

{

    Hashtable MXFFolderCache = new Hashtable();



    //

    // Change this to be the base path to search (path must end with "\")

    //

    string basePath = @"E:\Import with Sony Vegas Proxy Library\";

 

    public void FromVegas(Vegas vegas)

    {

        int processCount = 0;

        int totalOfflineFiles = 0;

        ArrayList replacedMedia = new ArrayList();



        //

        // Find all offline MXF media that are in the base path

        //

        foreach (Media media in vegas.Project.MediaPool)

        {

            string originalFilename = media.FilePath;



            if (media.IsOffline() && originalFilename.StartsWith(basePath) && originalFilename.ToUpper().EndsWith(".MXF"))

            {

                totalOfflineFiles++;



                // break up the path into pieces we can use

                string mxfPath = GetRightPath(originalFilename, 3);

                string mxfFolder = mxfPath.Substring(0, mxfPath.IndexOf('\\'));

                string mxfFile = mxfPath.Substring(mxfPath.IndexOf('\\'));



                // try and find the MXF folder

                string newFolder = FindFolder(basePath, mxfFolder);

                if (newFolder != null)

                {

                    string replacementFilename = newFolder + mxfFile;

                    Media newHDMedia = new Media(replacementFilename);

                    if (newHDMedia.IsValid())

                    {

                        try

                        {

                            media.ReplaceWith(newHDMedia);

                            processCount++;

                            replacedMedia.Add(originalFilename);

                        }

                        catch (Exception e)

                        {

                            MessageBox.Show(String.Format("Error: Could not replace [{0}] Reason: {1}", replacementFilename, e.Message));

                        }



                    }

                }

            }

        }



        // Remove the replaced files from the pool

        foreach (string replacedFilename in replacedMedia)

        {

            if (vegas.Project.MediaPool.Contains(replacedFilename))

            {

                try

                {

                    vegas.Project.MediaPool.Remove(replacedFilename);

                }

                catch (Exception e)

                {

                    MessageBox.Show(String.Format("Media: {0} could not be removed.\nError: {1}", replacedFilename, e.Message));

                }

            }

        }



        MessageBox.Show(String.Format("Processed: {0} out of {1} off-line file(s)", processCount, totalOfflineFiles), "Complete");

    }



    /// <summary>

    /// Does a depth first search of folders looking for a subfolder with the given mame

    /// </summary>

    /// <param name="startFolder">The folder to start the search</param>

    /// <param name="foldername">The folder name to search for</param>

    /// <returns></returns>

    private string FindFolder(string startFolder, string foldername)

    {

        // Since MXF disc names are unique, keep them cached to minimize searching

        if (MXFFolderCache[foldername] != null)

        {

            return (string)MXFFolderCache[foldername];

        }



        string foundFolder = null;



        // Search for the folder

        DirectoryInfo startDir = new DirectoryInfo(startFolder);

        DirectoryInfo[] childDirs = startDir.GetDirectories();



        foreach (DirectoryInfo childDirectory in childDirs)

        {

            if (childDirectory.Name.Equals(foldername))

            {

                // cache the hit

                MXFFolderCache[foldername] = childDirectory.FullName;

                foundFolder = childDirectory.FullName;

                break;

            }



            // Recurse through the child folders

            if (childDirectory.GetDirectories().Length > 0)

            {

                foundFolder = FindFolder(childDirectory.FullName, foldername);

                if (foundFolder != null)

                {

                    break;

                }

            }

        }



        return foundFolder;

    }



    /// <summary>

    /// Returns the right-most path give the number of slashes from the right

    /// </summary>

    /// <param name="basePath">The path to search</param>

    /// <param name="slashCount">The number of slashes from the right</param>

    /// <returns>The right-most part of the path from the slash to the end</returns>

    private string GetRightPath(string basePath, int slashCount)

    {

        int lastSlash = basePath.Length;

        for (int i = 0; i < slashCount; i++)

        {

            lastSlash = basePath.LastIndexOf(Path.DirectorySeparatorChar, lastSlash - 1);

        }

        lastSlash++;    // skip over slash

        return (basePath.Substring(lastSlash, basePath.Length - lastSlash));

    }



}