1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351 |
- {
- This file is part of the Free Component Library (FCL)
- Copyright (c) 1999-2007 by the Free Pascal development team
- Some modifications (c) 2007 by Martin Schreiber
- See the file COPYING.FPC, included in this distribution,
- for details about the copyright.
- This program 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.
- **********************************************************************}
- {$IFDEF FPC}
- {$mode objfpc}
- {$H+}
- {$ENDIF}
- {
- TMemDataset : In-memory dataset.
- - Has possibility to copy Structure/Data from other dataset.
- - Can load/save to/from stream.
- Ideas taken from THKMemTab Component by Harri Kasulke - Hamburg/Germany
- E-mail: [email protected]
- }
- unit memds;
- interface
- uses
- sysutils, classes, db, types;
- const
- // Stream Markers.
- MarkerSize = SizeOf(Integer);
- smEOF = 0;
- smFieldDefs = 1;
- smData = 2;
- type
- {$IFNDEF FPC}
- {$i memdsdelphi.inc} // should set ptrint is longint|intptr
- // & trecordbuffer ( if <2009)
- {$ENDIF}
- MDSError=class(Exception);
- { TMemDataset }
- TMemDataset=class(TDataSet)
- private
- type
- TMDSBlobList = class(TFPList)
- public
- procedure Clear; reintroduce;
- end;
- var
- FOpenStream : TStream;
- FFileName : String;
- FFileModified : Boolean;
- FStream: TMemoryStream;
- FRecInfoOffset: integer;
- FRecCount: integer;
- FRecSize: integer;
- FCurrRecNo: integer;
- FIsOpen: boolean;
- FTableIsCreated: boolean;
- FFilterBuffer: TRecordBuffer;
- ffieldoffsets: PInteger;
- ffieldsizes: PInteger;
- FBlobs: TMDSBlobList;
- function GetRecordBufferPointer(p:TRecordBuffer; Pos:Integer):TRecordBuffer;
- function GetIntegerPointer(p:PInteger; Pos:Integer):PInteger;
- procedure calcrecordlayout;
- function MDSGetRecordOffset(ARecNo: integer): longint;
- function MDSGetFieldOffset(FieldNo: integer): integer;
- function MDSGetBufferSize(FieldNo: integer): integer;
- function MDSGetActiveBuffer(out Buffer: TRecordBuffer): Boolean;
- procedure MDSReadRecord(Buffer:TRecordBuffer;ARecNo:Integer);
- procedure MDSWriteRecord(Buffer:TRecordBuffer;ARecNo:Integer);
- procedure MDSAppendRecord(Buffer:TRecordBuffer);
- function MDSFilterRecord(Buffer:TRecordBuffer): Boolean;
- function MDSLocateRecord(const KeyFields: string; const KeyValues: Variant; Options: TLocateOptions; out ARecNo: integer): Boolean;
- protected
- // Mandatory
- function AllocRecordBuffer: TRecordBuffer; override;
- procedure FreeRecordBuffer(var Buffer: TRecordBuffer); override;
- procedure GetBookmarkData(Buffer: TRecordBuffer; Data: Pointer); override;
- function GetBookmarkFlag(Buffer: TRecordBuffer): TBookmarkFlag; override;
- function GetFieldData(Field: TField; Buffer: Pointer): Boolean; override;
- function GetRecord(Buffer: TRecordBuffer; GetMode: TGetMode; DoCheck: Boolean): TGetResult; override;
- function GetRecordSize: Word; override;
- procedure InternalAddRecord(Buffer: Pointer; DoAppend: Boolean); override;
- procedure InternalClose; override;
- procedure InternalDelete; override;
- procedure InternalFirst; override;
- procedure InternalGotoBookmark(ABookmark: Pointer); override;
- procedure InternalInitFieldDefs; override;
- procedure InternalInitRecord(Buffer: TRecordBuffer); override;
- procedure ClearCalcFields(Buffer: TRecordBuffer); override;
- procedure InternalLast; override;
- procedure InternalOpen; override;
- procedure InternalPost; override;
- procedure InternalSetToRecord(Buffer: TRecordBuffer); override;
- function IsCursorOpen: Boolean; override;
- procedure SetBookmarkFlag(Buffer: TRecordBuffer; Value: TBookmarkFlag); override;
- procedure SetBookmarkData(Buffer: TRecordBuffer; Data: Pointer); override;
- procedure SetFieldData(Field: TField; Buffer: Pointer); override;
- // Optional.
- function GetRecordCount: Integer; override;
- procedure SetRecNo(Value: Integer); override;
- function GetRecNo: Integer; override;
- // Own.
- procedure SetFilterText(AValue: string); //silently drops filter
- Procedure RaiseError(Fmt : String; Args : Array of const);
- Procedure CheckMarker(F : TStream; Marker : Integer);
- Procedure WriteMarker(F : TStream; Marker : Integer);
- Procedure ReadFieldDefsFromStream(F : TStream);
- Procedure SaveFieldDefsToStream(F : TStream);
- // These should be overridden if you want to load more data.
- // E.g. index defs.
- Procedure LoadDataFromStream(F : TStream); virtual;
- // If SaveData=False, a size 0 block should be written.
- Procedure SaveDataToStream(F : TStream; SaveData : Boolean); virtual;
- public
- constructor Create(AOwner:TComponent); override;
- destructor Destroy; override;
- function BookmarkValid(ABookmark: TBookmark): Boolean; override;
- function CompareBookmarks(Bookmark1, Bookmark2: TBookmark): Longint; override;
- function CreateBlobStream(Field: TField; Mode: TBlobStreamMode): TStream; override;
- function Locate(const KeyFields: string; const KeyValues: Variant; Options: TLocateOptions): boolean; override;
- function Lookup(const KeyFields: string; const KeyValues: Variant; const ResultFields: string): Variant; override;
- procedure CreateTable;
- Function DataSize : Integer;
- Procedure Clear(ClearDefs : Boolean);{$IFNDEF FPC} overload; {$ENDIF}
- Procedure Clear;{$IFNDEF FPC} overload; {$ENDIF}
- Procedure SaveToFile(AFileName : String);{$IFNDEF FPC} overload; {$ENDIF}
- Procedure SaveToFile(AFileName : String; SaveData : Boolean);{$IFNDEF FPC} overload; {$ENDIF}
- Procedure SaveToStream(F : TStream); {$IFNDEF FPC} overload; {$ENDIF}
- Procedure SaveToStream(F : TStream; SaveData : Boolean);{$IFNDEF FPC} overload; {$ENDIF}
- Procedure LoadFromStream(F : TStream);
- Procedure LoadFromFile(AFileName : String);
- Procedure CopyFromDataset(DataSet : TDataSet); {$IFNDEF FPC} overload; {$ENDIF}
- Procedure CopyFromDataset(DataSet : TDataSet; CopyData : Boolean); {$IFNDEF FPC} overload; {$ENDIF}
- Property FileModified : Boolean Read FFileModified;
- // TMemDataset does not implement Filter. Please use OnFilter instead.
- Property Filter; unimplemented;
- published
- Property FileName : String Read FFileName Write FFileName;
- property Filtered;
- Property Active;
- Property FieldDefs;
- property BeforeOpen;
- property AfterOpen;
- property BeforeClose;
- property AfterClose;
- property BeforeInsert;
- property AfterInsert;
- property BeforeEdit;
- property AfterEdit;
- property BeforePost;
- property AfterPost;
- property BeforeCancel;
- property AfterCancel;
- property BeforeDelete;
- property AfterDelete;
- property BeforeScroll;
- property AfterScroll;
- property OnDeleteError;
- property OnEditError;
- property OnNewRecord;
- property OnPostError;
- property OnFilterRecord;
- end;
- implementation
- uses
- DBConst, Variants, FmtBCD;
- ResourceString
- SErrFieldTypeNotSupported = 'Fieldtype of Field "%s" not supported.';
- SErrBookMarkNotFound = 'Bookmark %d not found.';
- SErrInvalidDataStream = 'Error in data stream at position %d';
- SErrInvalidMarkerAtPos = 'Wrong data stream marker at position %d. Got %d, expected %d';
- SErrNoFileName = 'Filename must not be empty.';
- type
- TMDSRecInfo=record
- Bookmark: Longint;
- BookmarkFlag: TBookmarkFlag;
- end;
- PRecInfo=^TMDSRecInfo;
- TMDSBlobField = record
- Buffer: Pointer; // pointer to memory allocated for Blob data
- Size: PtrInt; // size of Blob data
- end;
- { TMDSBlobStream }
- TMDSBlobStream = class(TStream)
- private
- FField : TBlobField;
- FDataSet : TMemDataset;
- FBlobField : TMDSBlobField;
- FPosition : PtrInt;
- FModified : boolean;
- procedure AllocBlobField(NewSize: PtrInt);
- procedure FreeBlobField;
- public
- constructor Create(Field: TField; Mode: TBlobStreamMode);
- destructor Destroy; override;
- function Seek(const Offset: int64; Origin: TSeekOrigin): int64; override;
- function Read(var Buffer; Count: Longint): Longint; override;
- function Write(const Buffer; Count: Longint): Longint; override;
- end;
- Const
- SizeRecInfo = SizeOf(TMDSRecInfo);
- procedure unsetfieldisnull(nullmask: pbyte; const x: integer);
- begin
- inc(nullmask,(x shr 3));
- nullmask^:= nullmask^ or (1 shl (x and 7));
- end;
- procedure setfieldisnull(nullmask: pbyte; const x: integer);
- begin
- inc(nullmask,(x shr 3));
- nullmask^:= nullmask^ and Not (1 shl (x and 7));
- end;
- function getfieldisnull(nullmask: pbyte; const x: integer): boolean;
- begin
- inc(nullmask,(x shr 3));
- result:= nullmask^ and (1 shl (x and 7)) = 0;
- end;
- { ---------------------------------------------------------------------
- Stream functions
- ---------------------------------------------------------------------}
- Function ReadInteger(S : TStream) : Integer;
- begin
- S.ReadBuffer(Result,SizeOf(Result));
- end;
- Function ReadString(S : TStream) : String;
- Var
- L : Integer;
- begin
- L:=ReadInteger(S);
- Setlength(Result,L);
- If (L<>0) then
- S.ReadBuffer(Result[1],L);
- end;
- Procedure WriteInteger(S : TStream; Value : Integer);
- begin
- S.WriteBuffer(Value,SizeOf(Value));
- end;
- Procedure WriteString(S : TStream; Value : String);
- Var
- L : Integer;
- begin
- L:=Length(Value);
- WriteInteger(S,Length(Value));
- If (L<>0) then
- S.WriteBuffer(Value[1],L);
- end;
- { TMDSBlobStream }
- constructor TMDSBlobStream.Create(Field: TField; Mode: TBlobStreamMode);
- begin
- FField := Field as TBlobField;
- FDataSet := Field.DataSet as TMemDataset;
- if not Field.GetData(@FBlobField) then // IsNull
- begin
- FBlobField.Buffer := nil;
- FBlobField.Size := 0;
- end;
- if Mode = bmWrite then
- // release existing Blob
- FreeBlobField;
- end;
- destructor TMDSBlobStream.Destroy;
- begin
- if FModified then
- begin
- if FBlobField.Size = 0 then // Empty blob = IsNull
- FField.SetData(nil)
- else
- FField.SetData(@FBlobField);
- end;
- inherited;
- end;
- procedure TMDSBlobStream.FreeBlobField;
- begin
- FDataSet.FBlobs.Remove(FBlobField.Buffer);
- FreeMem(FBlobField.Buffer, FBlobField.Size);
- FBlobField.Buffer := nil;
- FBlobField.Size := 0;
- FModified := True;
- end;
- procedure TMDSBlobStream.AllocBlobField(NewSize: PtrInt);
- begin
- FDataSet.FBlobs.Remove(FBlobField.Buffer);
- ReAllocMem(FBlobField.Buffer, NewSize);
- FDataSet.FBlobs.Add(FBlobField.Buffer);
- FModified := True;
- end;
- function TMDSBlobStream.Seek(const Offset: int64; Origin: TSeekOrigin): int64;
- begin
- Case Origin of
- soBeginning : FPosition := Offset;
- soEnd : FPosition := FBlobField.Size + Offset;
- soCurrent : FPosition := FPosition + Offset;
- end;
- Result := FPosition;
- end;
- function TMDSBlobStream.Read(var Buffer; Count: Longint): Longint;
- var p: Pointer;
- begin
- if FPosition + Count > FBlobField.Size then
- Count := FBlobField.Size - FPosition;
- p := FBlobField.Buffer + FPosition;
- Move(p^, Buffer, Count);
- Inc(FPosition, Count);
- Result := Count;
- end;
- function TMDSBlobStream.Write(const Buffer; Count: Longint): Longint;
- var p: Pointer;
- begin
- AllocBlobField(FPosition+Count);
- p := FBlobField.Buffer + FPosition;
- Move(Buffer, p^, Count);
- Inc(FBlobField.Size, Count);
- Inc(FPosition, Count);
- Result := Count;
- end;
- { TMemDataset.TMDSBlobList }
- procedure TMemDataset.TMDSBlobList.Clear;
- var i: integer;
- begin
- for i:=0 to Count-1 do FreeMem(Items[i]);
- inherited Clear;
- end;
- { ---------------------------------------------------------------------
- TMemDataset
- ---------------------------------------------------------------------}
- constructor TMemDataset.Create(AOwner:TComponent);
- begin
- inherited Create(AOwner);
- FStream:=TMemoryStream.Create;
- FRecCount:=0;
- FRecSize:=0;
- FRecInfoOffset:=0;
- FCurrRecNo:=-1;
- BookmarkSize := sizeof(Longint);
- FBlobs := TMDSBlobList.Create;
- end;
- destructor TMemDataset.Destroy;
- begin
- // FStream.Free;
- FreeMem(FFieldOffsets);
- FreeMem(FFieldSizes);
- FBlobs.Clear;
- FBlobs.Free;
- inherited Destroy;
- FStream.Free;
- end;
- function TMemDataset.BookmarkValid(ABookmark: TBookmark): Boolean;
- var
- ReqBookmark: integer;
- begin
- Result := False;
- if ABookMark=nil then exit;
- ReqBookmark:=PInteger(ABookmark)^;
- Result := (ReqBookmark>=0) and (ReqBookmark<FRecCount);
- end;
- function TMemDataset.CompareBookmarks(Bookmark1, Bookmark2: TBookmark): Longint;
- const r: array[Boolean, Boolean] of ShortInt = ((2,-1),(1,0));
- begin
- Result := r[Bookmark1=nil, Bookmark2=nil];
- if Result = 2 then
- Result := PInteger(Bookmark1)^ - PInteger(Bookmark2)^;
- end;
- function TMemDataset.CreateBlobStream(Field: TField; Mode: TBlobStreamMode
- ): TStream;
- begin
- // Blobs are not saved to stream/file !
- if Mode = bmWrite then
- begin
- if not (State in [dsEdit, dsInsert, dsFilter, dsCalcFields]) then
- DatabaseErrorFmt(SNotEditing, [Name], Self);
- if Field.ReadOnly and not (State in [dsSetKey, dsFilter]) then
- DatabaseErrorFmt(SReadOnlyField, [Field.DisplayName]);
- end;
- Result := TMDSBlobStream.Create(Field, Mode);
- end;
- function TMemDataset.MDSGetRecordOffset(ARecNo: integer): longint;
- begin
- Result:=FRecSize*ARecNo
- end;
- function TMemDataset.MDSGetFieldOffset(FieldNo: integer): integer;
- begin
- Result:= getIntegerPointer(ffieldoffsets, fieldno-1)^;
- end;
- procedure TMemDataset.RaiseError(Fmt: String; Args: array of const);
- begin
- Raise MDSError.CreateFmt(Fmt,Args);
- end;
- function TMemDataset.MDSGetBufferSize(FieldNo: integer): integer;
- var
- FD: TFieldDef;
- begin
- FD := FieldDefs.Items[FieldNo-1];
- case FD.DataType of
- ftString : Result:=FD.Size*FD.CharSize+1;
- ftGuid: result:=FD.Size+1;
- ftFixedChar:result:=FD.Size*FD.CharSize+1;
- ftBoolean: result:=SizeOf(Wordbool);
- ftCurrency,
- ftFloat: result:=SizeOf(Double);
- ftBCD: result:=SizeOf(currency);
- ftLargeInt: result:=SizeOf(int64);
- ftSmallInt: result:=SizeOf(SmallInt);
- ftWord,
- ftAutoInc,
- ftInteger: result:=SizeOf(longint);
- ftDateTime,
- ftTime,
- ftDate: result:=SizeOf(TDateTime);
- ftFmtBCD: result:=SizeOf(TBCD);
- ftWideString, ftFixedWideChar:
- result:=(FD.Size+1)*SizeOf(WideChar);
- ftBytes: result := FD.Size;
- ftVarBytes: result := FD.Size + SizeOf(Word);
- ftBlob, ftMemo, ftWideMemo:
- result := SizeOf(TMDSBlobField);
- ftLongWord: Result := SizeOf(LongWord);
- ftShortInt: Result := SizeOf(ShortInt);
- ftByte: Result := SizeOf(Byte);
- ftExtended: Result := SizeOf(Extended);
- ftSingle : Result := SizeOf(Single);
- else
- RaiseError(SErrFieldTypeNotSupported,[FD.Name]);
- end;
- {$IFDEF FPC_REQUIRES_PROPER_ALIGNMENT}
- Result:=Align(Result,4);
- {$ENDIF}
- end;
- function TMemDataset.MDSGetActiveBuffer(out Buffer: TRecordBuffer): Boolean;
- begin
- case State of
- dsEdit,
- dsInsert:
- Buffer:=ActiveBuffer;
- dsFilter:
- Buffer:=FFilterBuffer;
- dsCalcFields:
- Buffer:=CalcBuffer;
- else
- if IsEmpty then
- Buffer:=nil
- else
- Buffer:=ActiveBuffer;
- end;
- Result := Buffer<>nil;
- end;
- procedure TMemDataset.MDSReadRecord(Buffer:TRecordBuffer;ARecNo:Integer); //Reads a Rec from Stream in Buffer
- begin
- FStream.Position:=MDSGetRecordOffset(ARecNo);
- FStream.ReadBuffer(Buffer^, FRecSize);
- end;
- procedure TMemDataset.MDSWriteRecord(Buffer:TRecordBuffer;ARecNo:Integer); //Writes a Rec from Buffer to Stream
- begin
- FStream.Position:=MDSGetRecordOffset(ARecNo);
- FStream.WriteBuffer(Buffer^, FRecSize);
- FFileModified:=True;
- end;
- procedure TMemDataset.MDSAppendRecord(Buffer:TRecordBuffer); //Appends a Rec (from Buffer) to Stream
- begin
- FStream.Position:=MDSGetRecordOffset(FRecCount);
- FStream.WriteBuffer(Buffer^, FRecSize);
- FFileModified:=True;
- end;
- //Abstract Overrides
- function TMemDataset.AllocRecordBuffer: TRecordBuffer;
- begin
- GetMem(Result, FRecSize+CalcFieldsSize);
- end;
- procedure TMemDataset.FreeRecordBuffer (var Buffer: TRecordBuffer);
- begin
- FreeMem(Buffer);
- end;
- procedure TMemDataset.InternalInitRecord(Buffer: TRecordBuffer);
- begin
- FillChar(Buffer^,FRecSize,0);
- end;
- procedure TMemDataset.ClearCalcFields(Buffer: TRecordBuffer);
- begin
- FillChar(Buffer[RecordSize], CalcFieldsSize, 0);
- end;
- procedure TMemDataset.InternalDelete;
- Var
- TS : TMemoryStream;
- begin
- if (FCurrRecNo<0) or (FCurrRecNo>=FRecCount) then
- Exit;
- // Very inefficient. We should simply move the last part closer to the beginning in
- // The FStream.
- TS:=TMemoryStream.Create;
- Try
- if FCurrRecNo>0 then
- begin
- FStream.Position:=MDSGetRecordOffset(0); //Delete Rec
- if FCurrRecNo<FRecCount-1 then
- begin
- TS.CopyFrom(FStream, MDSGetRecordOffset(FCurrRecNo)-MDSGetRecordOffset(0));
- FStream.Position:=MDSGetRecordOffset(FCurrRecNo+1);
- TS.CopyFrom(FStream,(MDSGetRecordOffset(FRecCount))-MDSGetRecordOffset(FCurrRecNo+1));
- end
- else
- TS.CopyFrom(FStream,MDSGetRecordOffset(FRecCount-1));
- end
- else
- begin //Delete first Rec
- FStream.Position:=MDSGetRecordOffset(FCurrRecNo+1);
- TS.CopyFrom(FStream,(MDSGetRecordOffset(FRecCount))-MDSGetRecordOffset(FCurrRecNo+1));
- end;
- FStream.LoadFromStream(TS);
- Dec(FRecCount);
- if FRecCount=0 then
- FCurrRecNo:=-1
- else
- if FCurrRecNo>=FRecCount then FCurrRecNo:=FRecCount-1;
- Finally
- TS.Free;
- end;
- FFileModified:=True;
- end;
- procedure TMemDataset.InternalInitFieldDefs;
- begin
- If (FOpenStream<>Nil) then
- ReadFieldDefsFromStream(FOpenStream);
- end;
- procedure TMemDataset.CheckMarker(F: TStream; Marker: Integer);
- Var
- I,P : Integer;
- begin
- P:=F.Position;
- If F.Read(I,MarkerSize)<>MarkerSize then
- RaiseError(SErrInvalidDataStream,[P])
- else
- if (I<>Marker) then
- RaiseError(SErrInvalidMarkerAtPos,[P,I,Marker]);
- end;
- procedure TMemDataset.ReadFieldDefsFromStream(F : TStream);
- Var
- I,ACount : Integer;
- FN : String;
- FS : Integer;
- B : Boolean;
- FT : TFieldType;
- begin
- CheckMarker(F,smFieldDefs);
- FieldDefs.Clear;
- ACount:=ReadInteger(F);
- For I:=1 to ACount do
- begin
- FN:=ReadString(F);
- FS:=ReadInteger(F);
- FT:=TFieldType(ReadInteger(F));
- B:=ReadInteger(F)<>0;
- TFieldDef.Create(FieldDefs,FN,ft,FS,B,I);
- end;
- FTableIsCreated:=False;
- end;
- procedure TMemDataset.InternalFirst;
- begin
- FCurrRecNo:=-1;
- end;
- procedure TMemDataset.InternalLast;
- begin
- FCurrRecNo:=FRecCount;
- end;
- procedure TMemDataset.InternalOpen;
- begin
- If (FFileName<>'') and FileExists(FFileName) then
- FOpenStream:=TFileStream.Create(FFileName,fmOpenRead);
- Try
- InternalInitFieldDefs;
- if DefaultFields then
- CreateFields;
- BindFields(True); // BindFields computes CalcFieldsSize
- if not FTableIsCreated then
- CreateTable;
- FCurrRecNo:=-1;
- If (FOpenStream<>Nil) then
- begin
- LoadDataFromStream(FOpenStream);
- CheckMarker(FOpenStream,smEOF);
- end;
- Finally
- FreeAndNil(FOpenStream);
- end;
- FIsOpen:=True;
- end;
- procedure TMemDataset.LoadDataFromStream(F: TStream);
- Var
- Size : Integer;
- begin
- CheckMarker(F,smData);
- Size:=ReadInteger(F);
- FBlobs.Clear;
- FStream.Clear;
- FStream.CopyFrom(F,Size);
- FRecCount:=Size div FRecSize;
- FCurrRecNo:=-1;
- end;
- procedure TMemDataset.LoadFromStream(F: TStream);
- begin
- Close;
- ReadFieldDefsFromStream(F);
- CreateTable;
- LoadDataFromStream(F);
- CheckMarker(F,smEOF);
- FFileModified:=False;
- end;
- procedure TMemDataset.LoadFromFile(AFileName: String);
- Var
- F : TFileStream;
- begin
- F:=TFileStream.Create(AFileName,fmOpenRead);
- Try
- LoadFromStream(F);
- Finally
- F.Free;
- end;
- end;
- procedure TMemDataset.SaveToFile(AFileName: String);
- begin
- SaveToFile(AFileName,True);
- end;
- procedure TMemDataset.SaveToFile(AFileName: String; SaveData: Boolean);
- Var
- F : TFileStream;
- begin
- If (AFileName='') then
- RaiseError(SErrNoFileName,[]);
- F:=TFileStream.Create(AFileName,fmCreate);
- try
- SaveToStream(F,SaveData);
- Finally
- F.Free;
- end;
- end;
- procedure TMemDataset.WriteMarker(F: TStream; Marker: Integer);
- begin
- Writeinteger(F,Marker);
- end;
- procedure TMemDataset.SaveToStream(F: TStream);
- begin
- SaveToStream(F,True);
- end;
- procedure TMemDataset.SaveToStream(F: TStream; SaveData: Boolean);
- begin
- SaveFieldDefsToStream(F);
- If SaveData then
- SaveDataToStream(F,SaveData);
- WriteMarker(F,smEOF);
- end;
- procedure TMemDataset.SaveFieldDefsToStream(F: TStream);
- Var
- I : Integer;
- FD : TFieldDef;
- begin
- WriteMarker(F,smFieldDefs);
- WriteInteger(F,FieldDefs.Count);
- For I:=1 to FieldDefs.Count do
- begin
- FD:=FieldDefs[I-1];
- WriteString(F,FD.Name);
- WriteInteger(F,FD.Size);
- WriteInteger(F,Ord(FD.DataType));
- WriteInteger(F,Ord(FD.Required));
- end;
- end;
- procedure TMemDataset.SaveDataToStream(F: TStream; SaveData: Boolean);
- begin
- if SaveData then
- begin
- WriteMarker(F,smData);
- WriteInteger(F,FStream.Size);
- FStream.Position:=0;
- F.CopyFrom(FStream,FStream.Size);
- FFileModified:=False;
- end
- else
- begin
- WriteMarker(F,smData);
- WriteInteger(F,0);
- end;
- end;
- procedure TMemDataset.InternalClose;
- begin
- if (FFileModified) and (FFileName<>'') then begin
- SaveToFile(FFileName,True);
- end;
- FIsOpen:=False;
- FFileModified:=False;
- // BindFields(False);
- if DefaultFields then
- DestroyFields;
- end;
- procedure TMemDataset.InternalPost;
- begin
- CheckActive;
- if not (State in [dsEdit, dsInsert]) then
- Exit;
- inherited InternalPost;
- if (State=dsEdit) then
- MDSWriteRecord(ActiveBuffer, FCurrRecNo)
- else
- InternalAddRecord(ActiveBuffer,True);
- end;
- function TMemDataset.IsCursorOpen: Boolean;
- begin
- Result:=FIsOpen;
- end;
- function TMemDataset.GetRecord(Buffer: TRecordBuffer; GetMode: TGetMode; DoCheck: Boolean): TGetResult;
- var
- Accepted: Boolean;
- begin
- Result:=grOk;
- Accepted:=False;
- if (FRecCount<1) then
- begin
- Result:=grEOF;
- exit;
- end;
- repeat
- case GetMode of
- gmCurrent:
- if (FCurrRecNo>=FRecCount) or (FCurrRecNo<0) then
- Result:=grError;
- gmNext:
- if (FCurrRecNo<FRecCount-1) then
- Inc(FCurrRecNo)
- else
- Result:=grEOF;
- gmPrior:
- if (FCurrRecNo>0) then
- Dec(FCurrRecNo)
- else
- result:=grBOF;
- end;
- if result=grOK then
- begin
- MDSReadRecord(Buffer, FCurrRecNo);
- PRecInfo(Buffer+FRecInfoOffset)^.Bookmark:=FCurrRecNo;
- PRecInfo(Buffer+FRecInfoOffset)^.BookmarkFlag:=bfCurrent;
- GetCalcFields(Buffer);
- if (Filtered) then
- Accepted:=MDSFilterRecord(Buffer) //Filtering
- else
- Accepted:=True;
- if (GetMode=gmCurrent) and not Accepted then
- result:=grError;
- end;
- until (result<>grOK) or Accepted;
- end;
- function TMemDataset.GetFieldData(Field: TField; Buffer: Pointer): Boolean;
- var
- SrcBuffer: TRecordBuffer;
- I: integer;
- begin
- I:= Field.FieldNo - 1;
- result := MDSGetActiveBuffer(SrcBuffer);
- if not result then Exit;
- if I >= 0 then
- begin
- result := not getfieldisnull(pointer(srcbuffer),I);
- if result and assigned(Buffer) then
- Move(GetRecordBufferPointer(SrcBuffer, GetIntegerPointer(ffieldoffsets,I)^)^, Buffer^, GetIntegerPointer(FFieldSizes, I)^);
- end
- else // Calculated, Lookup
- begin
- Inc(SrcBuffer, RecordSize + Field.Offset);
- result := Boolean(SrcBuffer[0]);
- if result and assigned(Buffer) then
- Move(SrcBuffer[1], Buffer^, Field.DataSize);
- end;
- end;
- procedure TMemDataset.SetFieldData(Field: TField; Buffer: Pointer);
- var
- DestBuffer: TRecordBuffer;
- I,J: integer;
- begin
- I:= Field.FieldNo - 1;
- if not MDSGetActiveBuffer(DestBuffer) then Exit;
- if I >= 0 then
- begin
- if State in [dsEdit, dsInsert, dsNewValue] then
- Field.Validate(Buffer);
- if Buffer = nil then
- setfieldisnull(pointer(DestBuffer),I)
- else
- begin
- unsetfieldisnull(pointer(DestBuffer),I);
- J:=GetIntegerPointer(FFieldSizes, I)^;
- if Field.DataType=ftString then
- Dec(J); // Do not move terminating 0, which is in the size.
- Move(Buffer^, GetRecordBufferPointer(DestBuffer, getIntegerPointer(FFieldOffsets, I)^)^, J);
- end;
- end
- else // Calculated, Lookup
- begin
- Inc(DestBuffer, RecordSize + Field.Offset);
- Boolean(DestBuffer[0]) := Buffer <> nil;
- if assigned(Buffer) then
- Move(Buffer^, DestBuffer[1], Field.DataSize);
- end;
- if not (State in [dsCalcFields, dsFilter, dsNewValue]) then
- DataEvent(deFieldChange, PtrInt(Field));
- end;
- function TMemDataset.GetRecordSize: Word;
- begin
- Result:= FRecSize;
- end;
- procedure TMemDataset.InternalGotoBookmark(ABookmark: Pointer);
- var
- ReqBookmark: integer;
- begin
- ReqBookmark:=PInteger(ABookmark)^;
- if (ReqBookmark>=0) and (ReqBookmark<FRecCount) then
- FCurrRecNo:=ReqBookmark
- else
- RaiseError(SErrBookMarkNotFound,[ReqBookmark]);
- end;
- procedure TMemDataset.InternalSetToRecord(Buffer: TRecordBuffer);
- var
- ReqBookmark: integer;
- begin
- ReqBookmark:=PRecInfo(Buffer+FRecInfoOffset)^.Bookmark;
- InternalGotoBookmark (@ReqBookmark);
- end;
- function TMemDataset.GetBookmarkFlag(Buffer: TRecordBuffer): TBookmarkFlag;
- begin
- Result:=PRecInfo(Buffer+FRecInfoOffset)^.BookmarkFlag;
- end;
- procedure TMemDataset.SetBookmarkFlag(Buffer: TRecordBuffer; Value: TBookmarkFlag);
- begin
- PRecInfo(Buffer+FRecInfoOffset)^.BookmarkFlag := Value;
- end;
- procedure TMemDataset.GetBookmarkData(Buffer: TRecordBuffer; Data: Pointer);
- begin
- if Data<>nil then
- PInteger(Data)^:=PRecInfo(Buffer+FRecInfoOffset)^.Bookmark;
- end;
- procedure TMemDataset.SetBookmarkData(Buffer: TRecordBuffer; Data: Pointer);
- begin
- if Data<>nil then
- PRecInfo(Buffer+FRecInfoOffset)^.Bookmark:=PInteger(Data)^
- else
- PRecInfo(Buffer+FRecInfoOffset)^.Bookmark:=0;
- end;
- function TMemDataset.MDSFilterRecord(Buffer: TRecordBuffer): Boolean;
- var
- SaveState: TDatasetState;
- begin
- Result:=True;
- if not Assigned(OnFilterRecord) then
- Exit;
- SaveState:=SetTempState(dsFilter);
- Try
- FFilterBuffer:=Buffer;
- OnFilterRecord(Self,Result);
- Finally
- RestoreState(SaveState);
- end;
- end;
- function TMemDataset.DataSize: Integer;
- begin
- Result:=FStream.Size;
- end;
- procedure TMemDataset.Clear;
- begin
- Clear(True);
- end;
- procedure TMemDataset.Clear(ClearDefs : Boolean);
- begin
- FBlobs.Clear;
- FStream.Clear;
- FRecCount:=0;
- FCurrRecNo:=-1;
- if Active then
- Resync([]);
- If ClearDefs then
- begin
- Close;
- FieldDefs.Clear;
- FTableIsCreated:=False;
- end;
- end;
- procedure TMemDataset.calcrecordlayout;
- var
- i,Count,aSize : integer;
- begin
- Count := FieldDefs.Count;
- // Avoid mem-leak if CreateTable is called twice
- FreeMem(FFieldOffsets);
- Freemem(FFieldSizes);
- {$IFDEF FPC}
- FFieldOffsets:=getmem(Count*sizeof(integer));
- FFieldSizes:=getmem(Count*sizeof(integer));
- {$ELSE}
- getmem(FFieldOffsets, Count*sizeof(integer));
- getmem(FFieldSizes, Count*sizeof(integer));
- {$ENDIF}
- FRecSize:= (Count+7) div 8; //null mask
- {$IFDEF FPC_REQUIRES_PROPER_ALIGNMENT}
- FRecSize:=Align(FRecSize,4);
- {$ENDIF}
- for i:= 0 to Count-1 do
- begin
- GetIntegerPointer(FFieldOffsets, i)^ := FRecSize;
- aSize:=MDSGetBufferSize(i+1);
- GetIntegerPointer(FFieldSizes, i)^ := aSize;
- FRecSize:= FRecSize+aSize;
- end;
- FRecInfoOffset:=FRecSize;
- FRecSize:=FRecSize+SizeRecInfo;
- end;
- procedure TMemDataset.CreateTable;
- begin
- CheckInactive;
- Clear(False);
- calcrecordlayout;
- FTableIsCreated:=True;
- end;
- procedure TMemDataset.InternalAddRecord(Buffer: Pointer; DoAppend: Boolean);
- begin
- MDSAppendRecord(ActiveBuffer);
- InternalLast;
- Inc(FRecCount);
- end;
- procedure TMemDataset.SetRecNo(Value: Integer);
- begin
- CheckBrowseMode;
- if (Value>=1) and (Value<=FRecCount) then
- begin
- FCurrRecNo:=Value-1;
- Resync([]);
- DoAfterScroll;
- end;
- end;
- function TMemDataset.GetRecNo: Integer;
- begin
- UpdateCursorPos;
- if (FCurrRecNo<0) or (FRecCount=0) or (State=dsInsert) then
- Result:=0
- else
- Result:=FCurrRecNo+1;
- end;
- function TMemDataset.GetRecordCount: Integer;
- begin
- CheckActive;
- Result:=FRecCount;
- end;
- procedure TMemDataset.CopyFromDataset(DataSet: TDataSet);
- begin
- CopyFromDataset(Dataset,True);
- end;
- procedure TMemDataset.CopyFromDataset(DataSet: TDataSet; CopyData: Boolean);
- Var
- I : Integer;
- F,F1,F2 : TField;
- L1,L2 : TList;
- N : String;
- OriginalPosition: TBookMark;
- begin
- Clear(True);
- // NOT from FieldDefs. The data may not be available in buffers !!
- For I:=0 to Dataset.FieldCount-1 do
- begin
- F:=Dataset.Fields[I];
- TFieldDef.Create(FieldDefs,F.FieldName,F.DataType,F.Size,F.Required,F.FieldNo);
- end;
- CreateTable;
- If CopyData then
- begin
- Open;
- L1:=TList.Create;
- Try
- L2:=TList.Create;
- Try
- For I:=0 to FieldDefs.Count-1 do
- begin
- N:=FieldDefs[I].Name;
- F1:=FieldByName(N);
- F2:=DataSet.FieldByName(N);
- L1.Add(F1);
- L2.Add(F2);
- end;
- DisableControls;
- Dataset.DisableControls;
- OriginalPosition:=Dataset.GetBookmark;
- Try
- Dataset.Open;
- Dataset.First; //make sure we copy from the beginning
- While not Dataset.EOF do
- begin
- Append;
- For I:=0 to L1.Count-1 do
- begin
- F1:=TField(L1[i]);
- F2:=TField(L2[I]);
- if F2.IsNull then
- F1.Clear
- else
- Case F1.DataType of
- ftFixedChar,
- ftString : F1.AsString:=F2.AsString;
- ftBoolean : F1.AsBoolean:=F2.AsBoolean;
- ftFloat : F1.AsFloat:=F2.AsFloat;
- ftLargeInt : F1.AsLargeInt:=F2.AsLargeInt;
- ftSmallInt,
- ftInteger,
- ftShortInt,
- ftByte : F1.AsInteger:=F2.AsInteger;
- ftDate : F1.AsDateTime:=F2.AsDateTime;
- ftTime : F1.AsDateTime:=F2.AsDateTime;
- ftDateTime : F1.AsDateTime:=F2.AsDateTime;
- ftLongWord : F1.AsLongWord:=F2.AsLongWord;
- ftExtended : F1.AsExtended:=F2.AsExtended;
- ftSingle : F1.AsSingle:=F2.AsSingle;
- else F1.AsString:=F2.AsString;
- end;
- end;
- Try
- Post;
- except
- Cancel;
- Raise;
- end;
- Dataset.Next;
- end;
- Finally
- DataSet.GotoBookmark(OriginalPosition); //Return to original record
- Dataset.EnableControls;
- EnableControls;
- end;
- finally
- L2.Free;
- end;
- finally
- l1.Free;
- end;
- end;
- end;
- function TMemDataset.GetRecordBufferPointer(p:TRecordBuffer; Pos:Integer):TRecordBuffer;
- begin
- Result:=p;
- inc(Result, Pos);
- end;
- function TMemDataset.GetIntegerPointer(p:PInteger; Pos:Integer):PInteger;
- begin
- Result:=p;
- inc(Result, Pos);
- end;
- function TMemDataset.MDSLocateRecord(const KeyFields: string; const KeyValues: Variant;
- Options: TLocateOptions; out ARecNo: integer): Boolean;
- var
- SaveState: TDataSetState;
- lstKeyFields: TList;
- Matched: boolean;
- AKeyValues: variant;
- i: integer;
- AField: TField;
- s1,s2: UTF8String;
- begin
- Result := false;
- SaveState := SetTempState(dsFilter);
- FFilterBuffer := TempBuffer;
- lstKeyFields := TList.Create;
- try
- GetFieldList(lstKeyFields, KeyFields);
- if VarArrayDimCount(KeyValues) = 0 then
- begin
- Matched := lstKeyFields.Count = 1;
- AKeyValues := VarArrayOf([KeyValues]);
- end
- else if VarArrayDimCount(KeyValues) = 1 then
- begin
- Matched := VarArrayHighBound(KeyValues,1) + 1 = lstKeyFields.Count;
- AKeyValues := KeyValues;
- end
- else
- Matched := false;
- if Matched then
- begin
- ARecNo:=0;
- while ARecNo<FRecCount do
- begin
- MDSReadRecord(FFilterBuffer, ARecNo);
- if Filtered then
- Result:=MDSFilterRecord(FFilterBuffer)
- else
- Result:=true;
- // compare field by field
- i:=0;
- while Result and (i<lstKeyFields.Count) do
- begin
- AField := TField(lstKeyFields[i]);
- // string fields
- if AField.DataType in [ftString, ftFixedChar] then
- begin
- if TStringField(AField).CodePage=CP_UTF8 then
- begin
- s1 := AField.AsUTF8String;
- s2 := UTF8Encode(VarToUnicodeStr(AKeyValues[i]));
- end
- else
- begin
- s1 := AField.AsString;
- s2 := VarToStr(AKeyValues[i]);
- end;
- if loPartialKey in Options then
- s1 := copy(s1, 1, length(s2));
- if loCaseInsensitive in Options then
- Result := AnsiCompareText(s1, s2)=0
- else
- Result := s1=s2;
- end
- // all other fields
- else
- Result := AField.Value=AKeyValues[i];
- inc(i);
- end;
- if Result then
- break;
- inc(ARecNo);
- end;
- end;
- finally
- lstKeyFields.Free;
- RestoreState(SaveState);
- end;
- end;
- procedure TMemDataset.SetFilterText(AValue: string);
- begin
- // Just do nothing; filter is not implemented
- end;
- function TMemDataset.Locate(const KeyFields: string; const KeyValues: Variant;
- Options: TLocateOptions): boolean;
- var
- ARecNo: integer;
- begin
- // Call inherited to make sure the dataset is bi-directional
- Result := inherited;
- CheckActive;
- Result:=MDSLocateRecord(KeyFields, KeyValues, Options, ARecNo);
- if Result then begin
- // TODO: generate scroll events if matched record is found
- FCurrRecNo:=ARecNo;
- Resync([]);
- end;
- end;
- function TMemDataset.Lookup(const KeyFields: string; const KeyValues: Variant;
- const ResultFields: string): Variant;
- var
- ARecNo: integer;
- SaveState: TDataSetState;
- begin
- if MDSLocateRecord(KeyFields, KeyValues, [], ARecNo) then
- begin
- SaveState := SetTempState(dsCalcFields);
- try
- // FFilterBuffer contains found record
- CalculateFields(FFilterBuffer); // CalcBuffer is set to FFilterBuffer
- Result:=FieldValues[ResultFields];
- finally
- RestoreState(SaveState);
- end;
- end
- else
- Result:=Null;
- end;
- end.
|