src/fileio_func.h

   1 /* $Id$ */
   2
   3 /*
   4  * This file is part of OpenTTD.
   5  * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
   6  * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
   7  * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
   8  */
   9
  10 /** @file fileio_func.h Functions for Standard In/Out file operations */
  11
  12 #ifndef FILEIO_FUNC_H
  13 #define FILEIO_FUNC_H
  14
  15 #include "core/enum_type.hpp"
  16 #include "fileio_type.h"
  17
  18 void FioSeekTo(size_t pos, int mode);
  19 void FioSeekToFile(uint8 slot, size_t pos);
  20 size_t FioGetPos();
  21 const char *FioGetFilename(uint8 slot);
  22 byte FioReadByte();
  23 uint16 FioReadWord();
  24 uint32 FioReadDword();
  25 void FioCloseAll();
  26 void FioOpenFile(int slot, const char *filename, Subdirectory subdir);
  27 void FioReadBlock(void *ptr, size_t size);
  28 void FioSkipBytes(int n);
  29
  30 /**
  31  * The search paths OpenTTD could search through.
  32  * At least one of the slots has to be filled with a path.
  33  * NULL paths tell that there is no such path for the
  34  * current operating system.
  35  */
  36 extern const char *_searchpaths[NUM_SEARCHPATHS];
  37
  38 /** Iterator for all the search paths */
  39 #define FOR_ALL_SEARCHPATHS(sp) for (sp = SP_FIRST_DIR; sp < NUM_SEARCHPATHS; sp++) if (_searchpaths[sp] != NULL)
  40
  41 void FioFCloseFile(FILE *f);
  42 FILE *FioFOpenFile(const char *filename, const char *mode, Subdirectory subdir, size_t *filesize = NULL);
  43 bool FioCheckFileExists(const char *filename, Subdirectory subdir);
  44 int FioGetFullPath (char *buf, size_t buflen, Searchpath sp, Subdirectory subdir, const char *filename = NULL);
  45 char *FioFindFullPath(char *buf, size_t buflen, Subdirectory subdir, const char *filename);
  46 char *FioGetDirectory(char *buf, size_t buflen, Subdirectory subdir);
  47
  48 const char *FiosGetScreenshotDir();
  49
  50 void SanitizeFilename(char *filename);
  51
  52 char *BuildDirPath (uint n, const char *const *parts);
  53
  54 static inline char *BuildDirPath (const char *part0)
  55 {
  56         return BuildDirPath (1, &part0);
  57 }
  58
  59 static inline char *BuildDirPath (const char *part0, const char *part1)
  60 {
  61         const char *const parts [2] = { part0, part1 };
  62         return BuildDirPath (2, parts);
  63 }
  64
  65 void DeterminePaths(const char *exe);
  66 void *ReadFileToMem(const char *filename, size_t *lenp, size_t maxsize);
  67 bool FileExists(const char *filename);
  68
  69 extern const char *_personal_dir; ///< custom directory for personal settings, saves, newgrf, etc.
  70
  71 /** Helper for scanning for files with a given name */
  72 class FileScanner {
  73 protected:
  74         Subdirectory subdir; ///< The current sub directory we are searching through
  75 public:
  76         /** Destruct the proper one... */
  77         virtual ~FileScanner() {}
  78
  79         uint Scan(const char *extension, Subdirectory sd, bool tars = true, bool recursive = true);
  80         uint Scan(const char *extension, const char *directory, const char *dirend, bool recursive);
  81
  82         /**
  83          * Add a file with the given filename.
  84          * @param filename        the full path to the file to read
  85          * @param basepath_length amount of characters to chop of before to get a
  86          *                        filename relative to the search path.
  87          * @param tar_filename    the name of the tar file the file is read from.
  88          * @return true if the file is added.
  89          */
  90         virtual bool AddFile(const char *filename, size_t basepath_length, const char *tar_filename) = 0;
  91 };
  92
  93 /** Helper for scanning for files with tar as extension */
  94 class TarScanner : FileScanner {
  95         uint DoScan(Subdirectory sd);
  96 public:
  97         /** The mode of tar scanning. */
  98         enum Mode {
  99                 NONE     = 0,      ///< Scan nothing.
 100                 BASESET  = 1 << 0, ///< Scan for base sets.
 101                 NEWGRF   = 1 << 1, ///< Scan for non-base sets.
 102                 AI       = 1 << 2, ///< Scan for AIs and its libraries.
 103                 SCENARIO = 1 << 3, ///< Scan for scenarios and heightmaps.
 104                 GAME     = 1 << 4, ///< Scan for game scripts.
 105                 ALL      = BASESET | NEWGRF | AI | SCENARIO | GAME, ///< Scan for everything.
 106         };
 107
 108         /**
 109          * Scan for tar files in the given search path.
 110          * @param sd the sub directory to search in.
 111          * @return the number of found files, i.e. the number of times that
 112          *         AddFile returned true.
 113          */
 114         uint Scan (Subdirectory sd)
 115         {
 116                 return this->FileScanner::Scan (".tar", sd, false);
 117         }
 118
 119         /* virtual */ bool AddFile(const char *filename, size_t basepath_length, const char *tar_filename = NULL);
 120
 121         /** Do the scan for Tars. */
 122         static uint DoScan(TarScanner::Mode mode);
 123 };
 124
 125 DECLARE_ENUM_AS_BIT_SET(TarScanner::Mode)
 126
 127 /* Implementation of opendir/readdir/closedir for Windows */
 128 #if defined(WIN32)
 129 struct DIR;
 130
 131 struct dirent { // XXX - only d_name implemented
 132         TCHAR *d_name; // name of found file
 133         /* little hack which will point to parent DIR struct which will
 134          * save us a call to GetFileAttributes if we want information
 135          * about the file (for example in function fio_bla) */
 136         DIR *dir;
 137 };
 138
 139 DIR *opendir(const TCHAR *path);
 140 struct dirent *readdir(DIR *d);
 141 int closedir(DIR *d);
 142 #else
 143 /* Use system-supplied opendir/readdir/closedir functions */
 144 # include <sys/types.h>
 145 # include <dirent.h>
 146 #endif /* defined(WIN32) */
 147
 148 /**
 149  * A wrapper around opendir() which will convert the string from
 150  * OPENTTD encoding to that of the filesystem. For all purposes this
 151  * function behaves the same as the original opendir function
 152  * @param path string to open directory of
 153  * @return DIR pointer
 154  */
 155 static inline DIR *ttd_opendir(const char *path)
 156 {
 157         return opendir(OTTD2FS(path));
 158 }
 159
 160 #endif /* FILEIO_FUNC_H */