------------------------------------------------------------------------------
-- Imports Multiple files or sequences from a Directory and or subdirectory
--
-- $Revision: 1.3 $
--
-- This script can take a root directory, and add all items in the directory to a Bin
-- By default it will scan the named directory and all subdirectories
--
-- Original Script "Create Bin From Directory"
-- written by : Isaac Guenard
--            : Peter Loveday
--            : Sean Konrad
--              March 08, 2006
-- modified by: Eric Westphal
--				February 2011
------------------------------------------------------------------------------

------------------------------------------------------------------------------
--                            DECLARE FUNCTIONS                             --
------------------------------------------------------------------------------

------------------------------------------------------------------------------
-- FUNCTION is_known_format
--
-- use this function to determine if an extension matches a valid clip format for Fusion
-- expects a table of fusion registry values, and a string containing an extension, including the "."
-- returns an index of -1 if its contained within the 'known_extensions' table,
-- or the index to the entry in fmt_list that matches the extension, or nil
------------------------------------------------------------------------------
local known_extensions = { ".fbx", ".dae", ".obj", ".3ds", ".dxf" }

function is_known_format(ext)
	ext = string.lower(ext)
	for j, ext2 in pairs(known_extensions) do
		if ext2 == ext then
			return -1
		end
	end

	for i, v in pairs(fmt_attrs) do
		for j, ext2 in pairs(v.REGST_MediaFormat_Extension) do
			if string.lower(ext2) == ext then
				return i
			end
		end
	end

	return nil
end
------------------------------------------------------------------------------
-- FUNCTION doInsert
--
-- used to determine if the table seq describes a clip we already know
-- if not, then we check to see if it is a loadable clip
-- if it is, then we return true
-- if the clip is not multi frame (like avi) and it has a sequence number then we 
-- also indicate that this should be added to the known table so that we can 
-- avoid adding every frame in a sequence to the bin
------------------------------------------------------------------------------
local function doInsert(seq, known)
	index = is_known_format(seq.Extension)

	if index then 
		if index < 0 then
			return true, nil
		else
			attrs = fmt_attrs[index]
			if attrs.REGB_MediaFormat_CanLoadMulti == true then
				return true, nil
 			else
				if seq.Number == nil then 
					return true, nil
				else
					-- term is the clean name for comparison
					term = seq.Path..seq.CleanName..seq.Extension

					if known[term] == nil then
						return true, term
					end
				end
			end
		end
	end
	return false, nil
end

------------------------------------------------------------------------------
-- doDirectories
--
-- function gets a path (ending in "\") and a mask (i.e. *.*)
-- recurses through the path building tables of clips, comps and settings
--
-------------------------------------------------------------------------------
function doDirectories(dir, mask)
	local path = dir..mask


	local files = readdir(path)

	if files == nil then print("   FAILED TO READ : "..string.lower(dir)) return end
	--print("   "..string.lower(dir))	

	for i, f in ipairs(files) do

		if type(f)=="table" and f.IsDir == true then
		
			if ret.Recurse == 1 then
				doDirectories(dir..f.Name.."\\", mask)
			end

		else
			if type(f) == "table" then
				seq = eyeon.parseFilename(dir..f.Name)

				-- if we have no extension, we don't want to bother
				if seq.Extension then
					isclip, isknown = doInsert(seq, known)

					if isclip == true then 
						table.insert(bin_insert, seq.FullPath) 
					end

					if isknown then known[isknown] = true end
				end
			end
		end
	end
end

------------------------------------------------------------------------------
--                                MAIN BODY                                 --
------------------------------------------------------------------------------

--------------------------------------------
-- Setup Initial Variables

bin_insert = {}

--------------------------------------------
-- get a list of known file formats from the Fusion registry
-- cache the list to a table to avoid slow repeated 
-- calls to GetAttrs()

fmt_list = fusion:GetRegList(CT_ImageFormat)
fmt_attrs = {}

for i,v in pairs(fmt_list) do
	fmt_attrs[i] = fmt_list[i]:GetAttrs()
end

known = {}
--------------------------------------------
-- Display the Ask User Dialog

ret = comp:AskUser("Path To Search For Clips", {
	{"SearchRoot", Name="root path for search", "PathBrowse"},
	{"Recurse", Name="do subdirectories", "Checkbox", Default=1, NumAcross=2}
	})

	
if ret == nil then print("cancelled") return end

--------------------------------------------
-- did they forget to provide a directory? exit.

if ret.SearchRoot == "" then print("You must provide a root directory for the script to scan.") return end

--------------------------------------------
-- set up initial directory

local dir  = ret.SearchRoot
local mask = "*.*"

--------------------------------------------
-- manually entered paths may not have a slash at the end. provide one

if string.sub(dir, -1, -1) ~= "\\" then	dir = dir .. "\\" end

--------------------------------------------
-- run the recursion function.
print()
print("Scanning Directories")
print("--------------------")

dir = composition:MapPath(dir)
doDirectories(dir, mask)

print()

-- Add footage to comp
comp:Lock()
	print("   adding "..table.getn(bin_insert).." clips.")
	for i, v in bin_insert do
		myLoader = Loader({Clip = v})
	end
comp:Unlock()
print("--------------------")
print("Scan Complete")
print()