summaryrefslogtreecommitdiff
path: root/fpcsrc/tests/test/trhlp40.pp
blob: ec7965328a3dbd70c6f8212899436c19e7966515 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
{ %NORUN }

{ a helper may introduce an enumerator }
program trhlp40;

{$ifdef fpc}
  {$mode delphi}
{$endif}

type
  TContainer = record
    Contents: array[0..5] of Integer;
    procedure Init;
  end;
  PContainer = ^TContainer;

  TContainerEnum = class
  private
    fIndex: Integer;
    fContainer: PContainer;
  public
    constructor Create(aContainer: PContainer);
    function GetCurrent: Integer;
    function MoveNext: Boolean;
    property Current: Integer read GetCurrent;
  end;

  TContainerHelper = record helper for TContainer
    function GetEnumerator: TContainerEnum;
  end;

{ TContainer }

procedure TContainer.Init;
var
  i: Integer;
begin
  for i := Low(Contents) to High(Contents) do
    Contents[i] := i;
end;

{ TContainerHelper }

function TContainerHelper.GetEnumerator: TContainerEnum;
begin
  Result := TContainerEnum.Create(@Self);
end;

{ TContainerEnum }

constructor TContainerEnum.Create(aContainer: PContainer);
begin
  fContainer := aContainer;
  fIndex := Low(fContainer^.Contents) - 1;
end;

function TContainerEnum.GetCurrent: Integer;
begin
  Result := fContainer^.Contents[fIndex];
end;

function TContainerEnum.MoveNext: Boolean;
begin
  Inc(fIndex);
  Result := fIndex <= High(fContainer^.Contents);
end;

var
  cont: TContainer;
  i: Integer;
begin
  cont.Init;
  for i in cont do ;
end.