src/shared/WADEDITOR.pas

   1 unit WADEDITOR;
   2
   3 {$DEFINE SFS_DWFAD_DEBUG}
   4
   5 interface
   6
   7 uses
   8   sfs, xstreams;
   9
  10
  11 type
  12   SArray = array of ShortString;
  13
  14   TWADEditor_1 = class(TObject)
  15   private
  16     fFileName: string; // empty: not opened
  17     fIter: TSFSFileList;
  18
  19     function getIsOpen (): Boolean;
  20
  21    public
  22     constructor Create();
  23     destructor Destroy(); override;
  24
  25     procedure FreeWAD();
  26
  27     function ReadFile (FileName: string): Boolean;
  28     function ReadMemory (Data: Pointer; Len: LongWord): Boolean;
  29     function GetResource (Section, Resource: string; var pData: Pointer; var Len: Integer): Boolean;
  30     function GetResourcesList (Section: string): SArray;
  31
  32     property isOpen: Boolean read getIsOpen;
  33   end;
  34
  35 {
  36 const
  37   DFWAD_NOERROR                = 0;
  38   DFWAD_ERROR_WADNOTFOUND      = -1;
  39   DFWAD_ERROR_CANTOPENWAD      = -2;
  40   DFWAD_ERROR_RESOURCENOTFOUND = -3;
  41   DFWAD_ERROR_FILENOTWAD       = -4;
  42   DFWAD_ERROR_WADNOTLOADED     = -5;
  43   DFWAD_ERROR_READRESOURCE     = -6;
  44   DFWAD_ERROR_READWAD          = -7;
  45   DFWAD_ERROR_WRONGVERSION     = -8;
  46 }
  47
  48
  49 procedure g_ProcessResourceStr (ResourceStr: String; var FileName, SectionName, ResourceName: String); overload;
  50 procedure g_ProcessResourceStr (ResourceStr: String; FileName, SectionName, ResourceName: PString); overload;
  51
  52
  53 implementation
  54
  55 uses
  56   SysUtils, Classes, BinEditor, e_log, g_options;
  57
  58
  59 procedure g_ProcessResourceStr (ResourceStr: String; var FileName, SectionName, ResourceName: String);
  60 var
  61   a, i: Integer;
  62
  63 begin
  64   //e_WriteLog(Format('g_ProcessResourceStr0: [%s]', [ResourceStr]), MSG_NOTIFY);
  65   for i := Length(ResourceStr) downto 1 do
  66     if ResourceStr[i] = ':' then
  67       Break;
  68
  69   FileName := Copy(ResourceStr, 1, i-1);
  70
  71   for a := i+1 to Length(ResourceStr) do
  72     if (ResourceStr[a] = '\') or (ResourceStr[a] = '/') then Break;
  73
  74   ResourceName := Copy(ResourceStr, a+1, Length(ResourceStr)-Abs(a));
  75   SectionName := Copy(ResourceStr, i+1, Length(ResourceStr)-Length(ResourceName)-Length(FileName)-2);
  76 end;
  77
  78
  79 procedure g_ProcessResourceStr (ResourceStr: AnsiString; FileName, SectionName, ResourceName: PAnsiString);
  80 var
  81   a, i, l1, l2: Integer;
  82
  83 begin
  84   //e_WriteLog(Format('g_ProcessResourceStr1: [%s]', [ResourceStr]), MSG_NOTIFY);
  85   for i := Length(ResourceStr) downto 1 do
  86     if ResourceStr[i] = ':' then
  87       Break;
  88
  89   if FileName <> nil then
  90     begin
  91       FileName^ := Copy(ResourceStr, 1, i-1);
  92       l1 := Length(FileName^);
  93     end
  94   else
  95     l1 := 0;
  96
  97   for a := i+1 to Length(ResourceStr) do
  98     if (ResourceStr[a] = '\') or (ResourceStr[a] = '/') then Break;
  99
 100   if ResourceName <> nil then
 101     begin
 102       ResourceName^ := Copy(ResourceStr, a+1, Length(ResourceStr)-Abs(a));
 103       l2 := Length(ResourceName^);
 104     end
 105   else
 106     l2 := 0;
 107
 108   if SectionName <> nil then
 109     SectionName^ := Copy(ResourceStr, i+1, Length(ResourceStr)-l2-l1-2);
 110 end;
 111
 112
 113 { TWADEditor_1 }
 114 constructor TWADEditor_1.Create();
 115 begin
 116   fFileName := '';
 117 end;
 118
 119
 120 destructor TWADEditor_1.Destroy();
 121 begin
 122   FreeWAD();
 123   inherited;
 124 end;
 125
 126
 127 function TWADEditor_1.getIsOpen (): Boolean;
 128 begin
 129   result := (fFileName <> '');
 130 end;
 131
 132
 133 procedure TWADEditor_1.FreeWAD();
 134 begin
 135   if fIter <> nil then FreeAndNil(fIter);
 136   //if fFileName <> '' then e_WriteLog(Format('TWADEditor_1.ReadFile: [%s] closed', [fFileName]), MSG_NOTIFY);
 137   fFileName := '';
 138 end;
 139
 140
 141 function removeExt (s: string): string;
 142 var
 143   i: Integer;
 144 begin
 145   i := length(s)+1;
 146   while (i > 1) and (s[i-1] <> '.') and (s[i-1] <> '/') do Dec(i);
 147   if (i > 1) and (s[i-1] = '.') then
 148   begin
 149     //writeln('[', s, '] -> [', Copy(s, 1, i-2), ']');
 150     s := Copy(s, 1, i-2);
 151   end;
 152   result := s;
 153 end;
 154
 155 function TWADEditor_1.GetResource (Section, Resource: string; var pData: Pointer; var Len: Integer): Boolean;
 156 var
 157   f: Integer;
 158   fi: TSFSFileInfo;
 159   fs: TStream;
 160   fpp: Pointer;
 161   //fn: string;
 162 begin
 163   Result := False;
 164   if not isOpen or (fIter = nil) then Exit;
 165   if length(Resource) = 0 then Exit; // just in case
 166   if (length(Section) <> 0) and (Section[length(Section)] <> '/') then Section := Section+'/';
 167   // backwards, due to possible similar names and such
 168   for f := fIter.Count-1 downto 0 do
 169   begin
 170     fi := fIter.Files[f];
 171     if fi = nil then continue;
 172     //e_WriteLog(Format('DFWAD: searching for [%s : %s] in [%s]; current is [%s : %s]', [Section, Resource, fFileName, fi.path, fi.name]), MSG_NOTIFY);
 173     if {SFSStrEqu}SFSDFPathEqu(fi.path, Section) and SFSStrEqu(removeExt(fi.name), Resource) then
 174     begin
 175       // i found her!
 176       //fn := fFileName+'::'+fi.path+fi.name;
 177       //fs := SFSFileOpen(fn);
 178       try
 179         fs := fIter.volume.OpenFileByIndex(f);
 180       except
 181         fs := nil;
 182       end;
 183       if fs = nil then
 184       begin
 185         e_WriteLog(Format('DFWAD: can''t open file [%s%s] in [%s]', [Section, Resource, fFileName]), MSG_WARNING);
 186         break;
 187       end;
 188       Len := Integer(fs.size);
 189       GetMem(pData, Len);
 190       fpp := pData;
 191       try
 192         fs.ReadBuffer(pData^, Len);
 193         fpp := nil;
 194       finally
 195         if fpp <> nil then
 196         begin
 197           FreeMem(fpp);
 198           pData := nil;
 199           Len := 0;
 200         end;
 201         fs.Free;
 202       end;
 203       result := true;
 204       {$IFDEF SFS_DWFAD_DEBUG}
 205       if gSFSDebug then
 206         e_WriteLog(Format('DFWAD: file [%s%s] FOUND in [%s]; size is %d bytes', [Section, Resource, fFileName, Len]), MSG_NOTIFY);
 207       {$ENDIF}
 208       exit;
 209     end;
 210   end;
 211   e_WriteLog(Format('DFWAD: file [%s%s] not found in [%s]', [Section, Resource, fFileName]), MSG_WARNING);
 212 end;
 213
 214
 215 function TWADEditor_1.GetResourcesList (Section: string): SArray;
 216 var
 217   f: Integer;
 218   fi: TSFSFileInfo;
 219 begin
 220   Result := nil;
 221   if not isOpen or (fIter = nil) then Exit;
 222   if (length(Section) <> 0) and (Section[length(Section)] <> '/') then Section := Section+'/';
 223   for f := 0 to fIter.Count-1 do
 224   begin
 225     fi := fIter.Files[f];
 226     if fi = nil then continue;
 227     if length(fi.name) = 0 then continue;
 228     if {SFSStrEqu}SFSDFPathEqu(fi.path, Section) then
 229     begin
 230       SetLength(result, Length(result)+1);
 231       result[high(result)] := removeExt(fi.name);
 232     end;
 233   end;
 234 end;
 235
 236
 237 function TWADEditor_1.ReadFile (FileName: string): Boolean;
 238 var
 239   rfn, path: string;
 240   //f: Integer;
 241   //fi: TSFSFileInfo;
 242 begin
 243   Result := False;
 244   //e_WriteLog(Format('TWADEditor_1.ReadFile: [%s]', [FileName]), MSG_NOTIFY);
 245   FreeWAD();
 246   path := ExtractFilePath(FileName);
 247   rfn := ExtractFileName(FileName);
 248   if not sfsFindFileCI(path, rfn) then
 249   begin
 250     //{if gSFSDebug then} e_WriteLog(Format('TWADEditor_1.ReadFile: error looking for [%s] [%s]', [path, ExtractFileName(FileName)]), MSG_NOTIFY);
 251     if SFSStrEqu(ExtractFileExt(FileName), '.wad') then
 252     begin
 253       rfn := ChangeFileExt(ExtractFileName(FileName), '.pk3');
 254       //{if gSFSDebug then} e_WriteLog(Format('  looking for [%s] [%s]', [path, rfn]), MSG_NOTIFY);
 255       if not sfsFindFileCI(path, rfn) then
 256       begin
 257         //{if gSFSDebug then} e_WriteLog(Format('  looking for [%s] [%s]', [path, rfn]), MSG_NOTIFY);
 258         rfn := ChangeFileExt(ExtractFileName(FileName), '.zip');
 259         if not sfsFindFileCI(path, rfn) then exit;
 260       end;
 261     end
 262     else
 263     begin
 264       exit;
 265     end;
 266     //{if gSFSDebug then} e_WriteLog(Format('TWADEditor_1.ReadFile: FOUND [%s]', [rfn]), MSG_NOTIFY);
 267   end
 268   else
 269   begin
 270     //if rfn <> ExtractFileName(FileName) then e_WriteLog(Format('TWADEditor_1.ReadFile: FOUND [%s]', [rfn]), MSG_NOTIFY);
 271   end;
 272   {$IFDEF SFS_DWFAD_DEBUG}
 273   if gSFSDebug then e_WriteLog(Format('TWADEditor_1.ReadFile: FOUND [%s]', [rfn]), MSG_NOTIFY);
 274   {$ENDIF}
 275   // cache this wad
 276   rfn := path+rfn;
 277   try
 278     if gSFSFastMode then
 279     begin
 280       if not SFSAddDataFile(rfn, true) then exit;
 281     end
 282     else
 283     begin
 284       if not SFSAddDataFileTemp(rfn, true) then exit;
 285     end;
 286   except
 287     exit;
 288   end;
 289   fIter := SFSFileList(rfn);
 290   if fIter = nil then Exit;
 291   fFileName := rfn;
 292   {$IFDEF SFS_DWFAD_DEBUG}
 293   if gSFSDebug then
 294     e_WriteLog(Format('TWADEditor_1.ReadFile: [%s] opened', [fFileName]), MSG_NOTIFY);
 295   {$ENDIF}
 296   {
 297     for f := 0 to fIter.Count-1 do
 298     begin
 299       fi := fIter.Files[f];
 300       if fi = nil then continue;
 301       e_WriteLog(Format('[%s]: [%s : %s] %u', [fFileName, fi.path, fi.name, fi.size]), MSG_NOTIFY);
 302     end;
 303   }
 304   Result := True;
 305 end;
 306
 307
 308 var
 309   uniqueCounter: Integer = 0;
 310
 311 function TWADEditor_1.ReadMemory (Data: Pointer; Len: LongWord): Boolean;
 312 var
 313   fn: string;
 314   st: TStream = nil;
 315   //f: Integer;
 316   //fi: TSFSFileInfo;
 317 begin
 318   Result := False;
 319   FreeWAD();
 320   if (Data = nil) or (Len = 0) then
 321   begin
 322     e_WriteLog('TWADEditor_1.ReadMemory: EMPTY SUBWAD!', MSG_WARNING);
 323     Exit;
 324   end;
 325
 326   fn := Format(' -- memwad %d -- ', [uniqueCounter]);
 327   Inc(uniqueCounter);
 328   {$IFDEF SFS_DWFAD_DEBUG}
 329     e_WriteLog(Format('TWADEditor_1.ReadMemory: [%s]', [fn]), MSG_NOTIFY);
 330   {$ENDIF}
 331
 332   try
 333     st := TSFSMemoryStreamRO.Create(Data, Len);
 334     if not SFSAddSubDataFile(fn, st, true) then
 335     begin
 336       st.Free;
 337       Exit;
 338     end;
 339   except
 340     st.Free;
 341     Exit;
 342   end;
 343
 344   fIter := SFSFileList(fn);
 345   if fIter = nil then Exit;
 346
 347   fFileName := fn;
 348   {$IFDEF SFS_DWFAD_DEBUG}
 349     e_WriteLog(Format('TWADEditor_1.ReadMemory: [%s] opened', [fFileName]), MSG_NOTIFY);
 350   {$ENDIF}
 351
 352   {
 353   for f := 0 to fIter.Count-1 do
 354   begin
 355     fi := fIter.Files[f];
 356     if fi = nil then continue;
 357     st := fIter.volume.OpenFileByIndex(f);
 358     if st = nil then
 359     begin
 360       e_WriteLog(Format('[%s]: [%s : %s] CAN''T OPEN', [fFileName, fi.path, fi.name]), MSG_NOTIFY);
 361     end
 362     else
 363     begin
 364       e_WriteLog(Format('[%s]: [%s : %s] %u', [fFileName, fi.path, fi.name, st.size]), MSG_NOTIFY);
 365       st.Free;
 366     end;
 367   end;
 368   //fIter.volume.OpenFileByIndex(0);
 369   }
 370
 371   Result := True;
 372 end;
 373
 374
 375 begin
 376   sfsDiskDirs := '<exedir>/data'; //FIXME
 377 end.