(*
    BSD 2-Clause License

    Copyright (c) 2018-2020, Anton Krotov
    All rights reserved.
*)

MODULE KOS;

IMPORT BIN, WR := WRITER, LISTS, CHL := CHUNKLISTS;


CONST

    HEADER_SIZE = 36;

    SIZE_OF_DWORD = 4;


TYPE

    HEADER = RECORD

        menuet01: ARRAY 9 OF CHAR;
        ver, start, size, mem, sp, param, path: INTEGER

    END;


PROCEDURE Import* (program: BIN.PROGRAM; idata: INTEGER; VAR ImportTable: CHL.INTLIST; VAR len, libcount, size: INTEGER);
VAR
    i:   INTEGER;
    imp: BIN.IMPRT;

BEGIN
    libcount  := 0;
    imp := program.imp_list.first(BIN.IMPRT);
    WHILE imp # NIL DO
        IF imp.label = 0 THEN
            INC(libcount)
        END;
        imp := imp.next(BIN.IMPRT)
    END;

    len  := libcount * 2 + 2;
    size := (LISTS.count(program.imp_list) + len + 1) * SIZE_OF_DWORD;

    ImportTable := CHL.CreateIntList();
    FOR i := 0 TO size DIV SIZE_OF_DWORD - 1 DO
        CHL.PushInt(ImportTable, 0)
    END;

    i := 0;
    imp := program.imp_list.first(BIN.IMPRT);
    WHILE imp # NIL DO

        IF imp.label = 0 THEN
            CHL.SetInt(ImportTable, len, 0);
            INC(len);
            CHL.SetInt(ImportTable, i, idata + len * SIZE_OF_DWORD);
            INC(i);
            CHL.SetInt(ImportTable, i, imp.nameoffs + size + idata);
            INC(i)
        ELSE
            CHL.SetInt(ImportTable, len, imp.nameoffs + size + idata);
            imp.label := len * SIZE_OF_DWORD;
            INC(len)
        END;

        imp := imp.next(BIN.IMPRT)
    END;
    CHL.SetInt(ImportTable, len, 0);
    CHL.SetInt(ImportTable, i, 0);
    CHL.SetInt(ImportTable, i + 1, 0);
    INC(len);
    INC(size, CHL.Length(program._import))
END Import;


PROCEDURE write* (program: BIN.PROGRAM; FileName: ARRAY OF CHAR);

CONST

    PARAM_SIZE    = 2048;
    FileAlignment = 16;


VAR
    header: HEADER;

    base, text, data, idata, bss, offset: INTEGER;

    reloc:   BIN.RELOC;
    iproc:   BIN.IMPRT;
    L:       INTEGER;
    delta:   INTEGER;

    i: INTEGER;

    ImportTable: CHL.INTLIST;
    ILen, libcount, isize: INTEGER;

    icount, dcount, ccount: INTEGER;

    code: CHL.BYTELIST;

BEGIN
    base := 0;

    icount := CHL.Length(program._import);
    dcount := CHL.Length(program.data);
    ccount := CHL.Length(program.code);

    text  := base + HEADER_SIZE;
    data  := WR.align(text + ccount, FileAlignment);
    idata := WR.align(data + dcount, FileAlignment);

    Import(program, idata, ImportTable, ILen, libcount, isize);

    bss := WR.align(idata + isize, FileAlignment);

    header.menuet01 := "MENUET01";
    header.ver      :=  1;
    header.start    :=  text;
    header.size     :=  idata + isize - base;
    header.mem      :=  WR.align(header.size + program.stack + program.bss + PARAM_SIZE * 2 + 4096, FileAlignment);
    header.sp       :=  base + header.mem - PARAM_SIZE * 2;
    header.param    :=  header.sp;
    header.path     :=  header.param + PARAM_SIZE;

    code := program.code;
    reloc := program.rel_list.first(BIN.RELOC);
    WHILE reloc # NIL DO

        offset := reloc.offset;
        L := BIN.get32le(code, offset);
        delta := 3 - offset - text;

        CASE reloc.opcode OF

        |BIN.RIMP:
              iproc := BIN.GetIProc(program, L);
              delta := idata + iproc.label

        |BIN.RBSS:
              delta := L + bss

        |BIN.RDATA:
              delta := L + data

        |BIN.RCODE:
              delta := BIN.GetLabel(program, L) + text

        |BIN.PICDATA:
              INC(delta, L + data)

        |BIN.PICCODE:
              INC(delta, BIN.GetLabel(program, L) + text)

        |BIN.PICBSS:
              INC(delta, L + bss)

        |BIN.PICIMP:
              iproc := BIN.GetIProc(program, L);
              INC(delta, idata + iproc.label)

        |BIN.IMPTAB:
              INC(delta, idata)

        END;
        BIN.put32le(code, offset, delta);

        reloc := reloc.next(BIN.RELOC)
    END;

    WR.Create(FileName);

    FOR i := 0 TO 7 DO
        WR.WriteByte(ORD(header.menuet01[i]))
    END;

    WR.Write32LE(header.ver);
    WR.Write32LE(header.start);
    WR.Write32LE(header.size);
    WR.Write32LE(header.mem);
    WR.Write32LE(header.sp);
    WR.Write32LE(header.param);
    WR.Write32LE(header.path);

    CHL.WriteToFile(code);
    WR.Padding(FileAlignment);

    CHL.WriteToFile(program.data);
    WR.Padding(FileAlignment);

    FOR i := 0 TO ILen - 1 DO
        WR.Write32LE(CHL.GetInt(ImportTable, i))
    END;

    CHL.WriteToFile(program._import);

    WR.Close
END write;


END KOS.