ViewVC Help
View File | Revision Log | View Changeset | Root Listing
root/Oni2/oup/current/DataAccess/Access_OniArchive.pas
(Generate patch)

Comparing oup/current/DataAccess/Access_OniArchive.pas (file contents):
Revision 113 by alloc, Sun Feb 25 17:20:22 2007 UTC vs.
Revision 321 by alloc, Wed May 6 13:47:23 2009 UTC

# Line 15 | Line 15 | type
15      FDatOpened:          Boolean;
16      FRawOpened:          Boolean;
17      FSepOpened:          Boolean;
18 +    procedure SetUnloadWhenUnused(doit: Boolean);
19    protected
20    public
21 <    property UnloadWhenUnused: Boolean Read FUnloadWhenUnused Write FUnloadWhenUnused;
21 >    property UnloadWhenUnused: Boolean Read FUnloadWhenUnused Write SetUnloadWhenUnused;
22  
23      constructor Create(DatFilename: String; ConnectionID: Integer; var Msg: TStatusMessages); override;
24      procedure Close; override;
# Line 34 | Line 35 | type
35      procedure UpdateDatFilePart(FileID, Offset, Size: Integer; Src: TStream); overload; override;
36  
37      function GetDatLinks(FileID: Integer): TDatLinkList; override;
38 +    function GetDatLink(FileID, DatOffset: Integer): TDatLink; override;
39      function GetRawList(FileID: Integer): TRawDataList; override;
40      function GetRawInfo(FileID, DatOffset: Integer): TRawDataInfo; override;
41 +    function GetRawsForType(RawType: String): TRawDataList; override;
42  
43 <    procedure LoadRawOffset(LocSep: Boolean; RawAddr, Size: Integer; target: Pointer);
43 >    procedure LoadRawOffset(LocSep: Boolean; RawAddr, Size: Integer; var target: TStream); overload;
44 >    procedure LoadRawOffset(LocSep: Boolean; RawAddr, Size: Integer; target: Pointer); overload;
45      procedure LoadRawFile(FileID, DatOffset: Integer; var Target: TStream); overload; override;
46      procedure UpdateRawFile(FileID, DatOffset: Integer; Src: TStream); overload; override;
47      procedure LoadRawFilePart(FileID, DatOffset, Offset, Size: Integer; var Target: TStream); overload; override;
# Line 50 | Line 54 | type
54   implementation
55  
56   uses
57 <  SysUtils, StrUtils, Data, Functions, RawList;
57 >  SysUtils, StrUtils, Data, Functions, RawList, DatLinks, Math;
58  
59  
60   (*
# Line 157 | Line 161 | begin
161    FBackend := DB_ONI;
162    FConnectionID := ConnectionID;
163    FChangeRights := [CR_EditDat, CR_EditRaw, CR_AppendRaw];
164 +
165 +  inherited;
166 + end;
167 +
168 +
169 +
170 +
171 + procedure TAccess_OniArchive.SetUnloadWhenUnused(doit: Boolean);
172 + begin
173 +  FUnloadWhenUnused := doit;
174 +  if FUnloadWhenUnused then
175 +  begin
176 +    if FDatOpened then
177 +    begin
178 +      FDatOpened := False;
179 +      Fdat_file.Free;
180 +    end;
181 +    if FRawOpened then
182 +    begin
183 +      FRawOpened := False;
184 +      Fraw_file.Free;
185 +    end;
186 +    if FSepOpened then
187 +    begin
188 +      FSepOpened := False;
189 +      Fsep_file.Free;
190 +    end;
191 +  end
192 +  else
193 +  begin
194 +    if not FDatOpened then
195 +    begin
196 +      Fdat_file := TFileStream.Create(FFileName, fmOpenReadWrite);
197 +      FDatOpened := True;
198 +    end;
199 +    if not FRawOpened then
200 +    begin
201 +      Fraw_file := TFileStream.Create(AnsiReplaceStr(FFileName, '.dat', '.raw'),
202 +        fmOpenReadWrite);
203 +      FRawOpened := True;
204 +    end;
205 +    if (not FSepOpened) and (FDataOS <> DOS_WIN) then
206 +    begin
207 +      Fsep_file := TFileStream.Create(AnsiReplaceStr(FFileName, '.dat', '.sep'),
208 +        fmOpenReadWrite);
209 +      FSepOpened := True;
210 +    end;
211 +  end;
212   end;
213  
214  
215  
216  
217 +
218   procedure TAccess_OniArchive.Close;
219   begin
220    if FDatOpened then
# Line 232 | Line 285 | var
285   begin
286    list := TStringList.Create;
287    list.Sorted := True;
288 +  if ext = '*' then
289 +    ext := '';
290    for i := 0 to GetFileCount - 1 do
291    begin
292      if ((Length(ext) = 0) or (Pos(Fdat_files[i].Extension, ext) > 0)) and
# Line 319 | Line 374 | begin
374    begin
375      if not Assigned(Target) then
376        Target := TMemoryStream.Create;
377 <    if not FDatOpened then
323 <      Fdat_file := TFileStream.Create(FFileName, fmOpenReadWrite);
324 <    Fdat_file.Seek(Fdat_files[fileid].DatAddr, soFromBeginning);
325 <    streampos := Target.Position;
326 <    Target.CopyFrom(Fdat_file, Fdat_files[fileid].Size);
327 <    Target.Seek(streampos, soFromBeginning);
328 <    if UnloadWhenUnused then
377 >    if GetFileInfo(FileID).Size > 0 then
378      begin
379 <      Fdat_file.Free;
380 <      FDatOpened := False;
381 <    end
382 <    else
383 <      FDatOpened := True;
379 >      if not FDatOpened then
380 >        Fdat_file := TFileStream.Create(FFileName, fmOpenReadWrite);
381 >      Fdat_file.Seek(Fdat_files[fileid].DatAddr, soFromBeginning);
382 >      streampos := Target.Position;
383 >      Target.CopyFrom(Fdat_file, Fdat_files[fileid].Size);
384 >      Target.Seek(streampos, soFromBeginning);
385 >      if UnloadWhenUnused then
386 >      begin
387 >        Fdat_file.Free;
388 >        FDatOpened := False;
389 >      end
390 >      else
391 >        FDatOpened := True;
392 >    end;
393    end;
394   end;
395  
# Line 397 | Line 455 | end;
455  
456  
457  
458 + function TAccess_OniArchive.GetDatLink(FileID, DatOffset: Integer): TDatLink;
459 + var
460 +  link: Integer;
461 + begin
462 +  Result := DatLinksManager.GetDatLink(FConnectionID, FileID, DatOffset);
463 +  LoadDatFilePart(fileid, Result.SrcOffset, 4, @link);
464 +  if link > 0 then
465 +    Result.DestID := link div 256
466 +  else
467 +    Result.DestID := -1;
468 + end;
469 +
470 +
471 + function TAccess_OniArchive.GetDatLinks(FileID: Integer): TDatLinkList;
472 + var
473 +  i: Integer;
474 +  link: Integer;
475 + begin
476 +  Result := DatLinksManager.GetDatLinks(FConnectionID, FileID);
477 +  if Length(Result) > 0 then
478 +  begin
479 +    for i := 0 to High(Result) do
480 +    begin
481 +      LoadDatFilePart(fileid, Result[i].SrcOffset, 4, @link);
482 +      if link > 0 then
483 +        Result[i].DestID := link div 256
484 +      else
485 +        Result[i].DestID := -1;
486 +    end;
487 +  end;
488 + end;
489 +
490 +
491   function TAccess_OniArchive.GetRawList(FileID: Integer): TRawDataList;
492   begin
493    Result := RawLists.GetRawList(FConnectionID, FileID);
494   end;
495  
496  
497 + function TAccess_OniArchive.GetRawsForType(RawType: String): TRawDataList;
498 + var
499 +  i, j: Integer;
500 +  dats: TStrings;
501 +  list: TRawDataList;
502 + begin
503 +  dats := nil;
504 +  dats := GetFilesList(MidStr(RawType, 1, 4), '', True, ST_IDAsc);
505 +  for i := 0 to dats.Count - 1 do
506 +  begin
507 +    list := GetRawList(StrToInt(MidStr(dats.Strings[i], 1, 5)));
508 +    for j := 0 to Length(list) - 1 do
509 +    begin
510 +      if (list[j].RawType = RawType) and (list[j].RawSize > 0) then
511 +      begin
512 +        SetLength(Result, Length(Result)+1);
513 +        Result[High(Result)] := list[j];
514 +      end;
515 +    end;
516 +  end;
517 + end;
518 +
519 +
520   function TAccess_OniArchive.GetRawInfo(FileID, DatOffset: Integer): TRawDataInfo;
521   begin
522    Result := RawLists.GetRawInfo(FConnectionID, FileID, DatOffset);
# Line 410 | Line 524 | end;
524  
525  
526  
527 <
414 < procedure TAccess_OniArchive.LoadRawOffset(LocSep: Boolean; RawAddr, Size: Integer; target: Pointer);
527 > procedure TAccess_OniArchive.LoadRawOffset(LocSep: Boolean; RawAddr, Size: Integer; var target: TStream);
528   begin
529 +  if not Assigned(Target) then
530 +    Target := TMemoryStream.Create;
531    if not LocSep then
532    begin
533      if not FRawOpened then
# Line 421 | Line 536 | begin
536      if RawAddr <= Fraw_file.Size then
537      begin
538        Fraw_file.Seek(RawAddr, soFromBeginning);
539 <      Fraw_file.Read(target^, size);
539 >      Target.CopyFrom(Fraw_file, size);
540 >      Target.Seek(0, soFromBeginning);
541      end;
542      if UnloadWhenUnused then
543      begin
# Line 439 | Line 555 | begin
555      if RawAddr <= Fsep_file.Size then
556      begin
557        Fsep_file.Seek(RawAddr, soFromBeginning);
558 <      Fsep_file.Read(target^, size);
558 >      Target.CopyFrom(Fsep_file, size);
559 >      Target.Seek(0, soFromBeginning);
560      end;
561      if UnloadWhenUnused then
562      begin
# Line 451 | Line 568 | begin
568    end;
569   end;
570  
571 + procedure TAccess_OniArchive.LoadRawOffset(LocSep: Boolean; RawAddr, Size: Integer; target: Pointer);
572 + var
573 +  data: TStream;
574 + begin
575 +  data := nil;
576 +  LoadRawOffset(LocSep, RawAddr, Size, data);
577 +  data.Read(Target^, Size);
578 +  data.Free;
579 + end;
580 +
581   procedure TAccess_OniArchive.LoadRawFile(FileID, DatOffset: Integer; var Target: TStream);
582   var
583    raw_info: TRawDataInfo;
# Line 461 | Line 588 | begin
588    if fileid < GetFileCount then
589    begin
590      raw_info := Self.GetRawInfo(FileID, DatOffset);
591 +    if raw_info.RawSize = 0 then
592 +      exit;
593      if not raw_info.LocSep then
594      begin
595        if not FRawOpened then
# Line 511 | Line 640 | begin
640          Fraw_file := TFileStream.Create(AnsiReplaceStr(FFileName, '.dat', '.raw'),
641            fmOpenReadWrite);
642        Fraw_file.Seek(raw_info.RawAddr, soFromBeginning);
643 <      Fraw_file.CopyFrom(Src, raw_info.RawSize);
643 >      Fraw_file.CopyFrom(Src, Min(raw_info.RawSize, Src.Size));
644        if UnloadWhenUnused then
645        begin
646          FRawOpened := False;
# Line 556 | Line 685 | begin
685    end;
686   end;
687  
688 +
689   procedure TAccess_OniArchive.UpdateRawFilePart(FileID, DatOffset, Offset, Size: Integer; Src: TStream);
690   var
691    raw_info: TRawDataInfo;
# Line 597 | Line 727 | begin
727   end;
728  
729   function TAccess_OniArchive.AppendRawFile(LocSep: Boolean; Src: TStream): Integer;
730 + const
731 +  EmptyBytes: Array[0..31] of Byte = (
732 +      0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0 );
733   begin
734    if not LocSep then
735    begin
736      if not FRawOpened then
737        Fraw_file := TFileStream.Create(AnsiReplaceStr(FFileName, '.dat', '.raw'),
738          fmOpenReadWrite);
739 +    if (Fraw_file.Size mod 32) > 0 then
740 +      Fraw_file.Write(EmptyBytes[0], 32 - (Fraw_file.Size mod 32));
741      Result := Fraw_file.Size;
742      Fraw_file.Seek(0, soFromEnd);
743      Fraw_file.CopyFrom(Src, Src.Size);
744 +    if (Fraw_file.Size mod 32) > 0 then
745 +      Fraw_file.Write(EmptyBytes[0], 32 - (Fraw_file.Size mod 32));
746      if UnloadWhenUnused then
747      begin
748        FRawOpened := False;
# Line 619 | Line 756 | begin
756      if not FSepOpened then
757        Fsep_file := TFileStream.Create(AnsiReplaceStr(FFileName, '.dat', '.sep'),
758          fmOpenReadWrite);
759 +    if (Fsep_file.Size mod 32) > 0 then
760 +      Fsep_file.Write(EmptyBytes[0], 32 - (Fsep_file.Size mod 32));
761      Result := Fsep_file.Size;
762      Fsep_file.Seek(0, soFromEnd);
763      Fsep_file.CopyFrom(Src, Src.Size);
764 +    if (Fsep_file.Size mod 32) > 0 then
765 +      Fsep_file.Write(EmptyBytes[0], 32 - (Fsep_file.Size mod 32));
766      if UnloadWhenUnused then
767      begin
768        FSepOpened := False;

Diff Legend

Removed lines
+ Added lines
< Changed lines (old)
> Changed lines (new)