source: GTP/trunk/App/Demos/Illum/DepthOfField/GetFilePath.h @ 760

/* 
GetFilePath.h

A convenient way to search the NVSDK for media files
*/


#ifndef H_GETFILEPATH_H
#define H_GETFILEPATH_H

#include <windows.h>
#include <TCHAR.H>
#include <string>
#include "shared/NV_Common.h"
#include "shared/NV_Error.h"
#include "shared/nvFileSearch.h"

typedef std::basic_string<TCHAR> tstring; 

namespace GetFilePath
{
        static tstring strStartPath;

        typedef tstring (*GetFilePathFunction)(const tstring &, bool bVerbose );
        
        // This variable allows us to recursively search for media
        class NVMediaSearch : public NVFileSearch
        {
                public:
                tstring m_strPath;
                virtual bool FileFoundCallback(const WIN32_FIND_DATA& FindData, const tstring& strDir)
                {
                        UNREFERENCED_PARAMETER(FindData);
                        m_strPath = strDir;
                        
                        return false;
                }
        };


        static tstring GetModulePath() { return strStartPath; }
        static void        SetModulePath(const tstring &strPath)
        {
                tstring::size_type Pos = strPath.find_last_of(_T("\\"), strPath.size());
                if (Pos != strPath.npos)
                        strStartPath = strPath.substr(0, Pos);
                else
                        strStartPath = _T(".");
        }

        static void             SetDefaultModulePath()
        {
                DWORD ret;
                TCHAR buf[MAX_PATH];
                ret = GetModuleFileName( NULL, buf, MAX_PATH );         // get name for current module
                if( ret == 0 )
                {
                        FMsg(TEXT("SetDefaultModulePath() GetModuleFileName() failed!\n"));
                        assert( false );
                }

                SetModulePath( buf );
        }


        
        static tstring GetFullPath(const tstring &strRelativePath)
        {
                TCHAR buf[MAX_PATH];
                TCHAR *pAdr;
                GetFullPathName(strRelativePath.data(), MAX_PATH, buf, &pAdr);
                return buf;
        }

        //a helper class to save and restore the currentDirectory
        class DirectorySaver
        {
        private:
                TCHAR savedDirectory[MAX_PATH];
        public:
                DirectorySaver( )
                {
                        // Save current directory
                        GetCurrentDirectory(MAX_PATH, savedDirectory);
                }
                ~DirectorySaver( )
                {
                        // return to previous directory
                SetCurrentDirectory(this->savedDirectory);
                }
        };

    // Recursively searchs the given path until it finds the file. Returns "" if 
    // file can't be found
    static tstring FindMediaFile(const tstring &strFilename, const tstring &mediaPath, bool bVerbose = false )
    {
        WIN32_FIND_DATA findInfo;
        HANDLE hFind;
        tstring result;
    
        
                //save and auto restore the current working directory
                DirectorySaver whenIGoOutOfScopeTheCurrentWorkingDirectoryWillBeRestored;
                
        if (!SetCurrentDirectory(mediaPath.data()))
                {
                        // DWORD tmp2 = GetLastError();
                        if( bVerbose )
                        {       
                                FMsg(TEXT("FindMediaFile Couldn't SetCurrentDirectory to [%s].  Returning empty string\n"), mediaPath.c_str() );
                        }
            return _T("");
                }

        // check if file is in current directory
        FILE *fp;
        fp = _tfopen(strFilename.data(), _T("r"));
        if (fp)
        {
            fclose(fp);
            return mediaPath + _T("\\") + strFilename;
        }
                else
                {       
                        if( bVerbose )
                        {
                                // report where the file is NOT
                                FMsg(TEXT("FindMediaFile: File [%s] is not in %s\n"), strFilename.c_str(), mediaPath.c_str() );
                        }
                }
    
        // if file not in current directory, search for all directories
        // and search inside them until the file is found
        hFind = FindFirstFile( _T( "*.*" ) , &findInfo);
        if (hFind == INVALID_HANDLE_VALUE)
            return _T("");
        
        result = _T("");
        do
        {
            if (findInfo.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
            {
                // Make sure we don't try to delete the '.' or '..' directories
                if ((_tcscmp(findInfo.cFileName, _T("..")) == 0) || (_tcscmp(findInfo.cFileName, _T(".")) == 0))
                    continue;
                
                                // Add directory to the media path
                                // Keep same original file name
                result = FindMediaFile(strFilename, mediaPath + _T("\\") + findInfo.cFileName, bVerbose );
                if (result != _T(""))
                    break;                    
            }
        } while (FindNextFile(hFind, &findInfo));
    
        FindClose(hFind);
        
        return result;
    }

        //////////////////////////////////////////////////////////////////////////////
        // Gets the full path to the requested file by searching in the MEDIA directory.
    // (note: for a pointer to a 1-parameter version of this function, see below.)
        static tstring GetFilePath(const tstring& strFileName, bool bVerbose )
        {
        // Check if strFilename already points to a file
        FILE *fp;
        fp = _tfopen(strFileName.data(), _T("r"));
        if (fp)
        {
            fclose(fp);
            return strFileName;
        }

                // You should call SetModulePath before using GetFilePath()
                // If not, it will set the module path for you

                tstring strMediaPath = GetModulePath();
                if( strMediaPath.empty() == true )
                {
                        SetDefaultModulePath();
                        strMediaPath = GetModulePath();
                }

                // Find the last occurrence of '\' or '/'
                // This has the effect of backing up 4 directories, implying a direct dependence on
                //  the location of the calling .exe in order to start looking for media in the right
                //  place.  This is bad, so a more general search for the first subdirectory \MEDIA 
                //  should be put in place
                // TODO : see above
        strMediaPath = strMediaPath.substr(0, strMediaPath.find_last_of(_T("\\/")));
            strMediaPath = strMediaPath.substr(0, strMediaPath.find_last_of(_T("\\/")));
            strMediaPath = strMediaPath.substr(0, strMediaPath.find_last_of(_T("\\/")));
            strMediaPath = strMediaPath.substr(0, strMediaPath.find_last_of(_T("\\/")));
        strMediaPath += _T("\\MEDIA");
                        
                tstring result;
        result = FindMediaFile(strFileName, strMediaPath, bVerbose);
           
        if (result != _T(""))
            return result;

        //////////////////// for local shaders /////////////////////////
                strMediaPath = GetModulePath();
        strMediaPath += _T("\\Shaders");

        result = FindMediaFile(strFileName, strMediaPath, bVerbose);
           
        if (result != _T(""))
            return result;

        //////////////////// for local ../shaders /////////////////////////
                strMediaPath = GetModulePath();
        strMediaPath = strMediaPath.substr(0, strMediaPath.find_last_of(_T("\\/")));
        strMediaPath += _T("\\Shaders");

        result = FindMediaFile(strFileName, strMediaPath, bVerbose);
           
        if (result != _T(""))
            return result;

                // If prog gets to here, the find has failed.
                // Return the input file name so other apps can report the failure
                //  to find the file.
                if( bVerbose )
                        FMsg(TEXT("GetFilePath() Couldn't find : %s\n"), strFileName.c_str() );

                return strFileName;
        };

    // Use these wrapper functions if you need to pass a pointer [callback] 
    // to a 1-parameter version of the GetFilePath() function:
    static tstring GetFilePath(const tstring& strFileName) {
        return GetFilePath(strFileName, false);
    }
        static tstring GetFilePathVerbose(const tstring& strFileName) {
        return GetFilePath(strFileName, true);
    }

};       // namespace GetFilePath


#endif