source: branches/Void/Generators/UZ80Generator.pas

unit UZ80Generator;

{$mode Delphi}{$H+}

interface

uses
  Classes, SysUtils, UModel, StrUtils, UOutputGenerator;

type
  TZ80Generator = class(TOutputGenerator)
  private
    function GenerateVariableValue(Value: TVariableValue): string;
  public
    procedure Generate(Model: TModel); override;
    procedure GenerateModule(Module: TModule);
  end;

implementation

{ TZ80Generator }

function TZ80Generator.GenerateVariableValue(Value: TVariableValue): string;
begin
  case Value.ValueType of
    vtVariable: if Assigned(Value.VariableDef) then Result := Value.VariableDef.Name
      else Result := '';
    vtFloat: Result := FloatToStr(Value.FloatConstant);
    vtNumber: Result := IntToStr(Value.NumberConstant);
    vtString: Result := '''' + StringReplace(Value.StringConstant, '''', '''''', [rfReplaceAll]) + '''';
    vtChar: Result := '''' + Value.CharConstant + '''';
  end;
end;

procedure TZ80Generator.Generate(Model: TModel);
begin
  inherited;
  GenerateModule(Model.Module);
end;

procedure TZ80Generator.GenerateModule(Module: TModule);
var
  I, P: Integer;
  ParameterText: string;
  Row: string;
begin
  with Module do begin
    // Prepare output
    Output.Clear;

    // var section
    for I := 0 to Variables.Count - 1 do
      with Variables[I] do
        Output.Add(Name + ':  DB ' + VarType.Name + ';');

    // Code block
    Output.Add('begin:');
    Inc(IndentCount);
    for I := 0 to BeginEnd.Commands.Count - 1 do
      with BeginEnd.Commands[I] do begin
        if Name = 'Assignment' then Output.Add('  LD ' +
          Parameters[0].VariableDef.Name + ', ' +
          GenerateVariableValue(TVariableValue(Parameters[1])))
        else begin
          if Parameters.Count > 0 then begin
            ParameterText := '';
            for P := 0 to Parameters.Count - 1 do
              Output.Add('  PUSH ' + GenerateVariableValue(Parameters[P]));
          end;
          Output.Add('  CALL ' + Name);
        end;
      end;
    Dec(IndentCount);
    Output.Add('HALT');
  end;
end;

end.