! Copyright (C) 2003, 2006 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
IN: prettyprint
USING: arrays generic hashtables io kernel math namespaces
sequences strings styles words ;

: declaration. ( word prop -- )
    tuck word-name word-prop [ pprint-word ] [ drop ] if ;

: declarations. ( word -- )
    {
        POSTPONE: parsing
        POSTPONE: inline
        POSTPONE: foldable
        POSTPONE: flushable
    } [ declaration. ] each-with ;

: in. ( word -- )
    <block \ IN: pprint-word word-vocabulary text block; ;

: (synopsis) ( word -- )
    dup in. dup definer pprint-word pprint-word ;

: comment. ( comment -- )
    [ H{ { font-style italic } } [ text ] with-style ] when* ;

: stack-picture ( seq -- string )
    [ [ % CHAR: \s , ] each ] "" make ;

: effect>string ( effect -- string )
    [
        "( " %
        dup first stack-picture %
        "-- " %
        second stack-picture %
        ")" %
    ] "" make ;

: stack-effect ( word -- string )
    dup "stack-effect" word-prop [ ] [
        "infer-effect" word-prop dup [
            [
                dup integer? [ object <array> ] when
                [ word-name ] map
            ] map effect>string
        ] when
    ] ?if ;

: synopsis ( word -- string )
    [
        0 margin set [
            dup (synopsis) stack-effect comment.
        ] with-pprint
    ] string-out ;

GENERIC: (see) ( word -- )

M: word (see) drop ;

: pprint-; \ ; pprint-word ;

: see-body ( quot word -- )
    <block swap pprint-elements pprint-; declarations. block; ;

M: compound (see)
    dup word-def swap see-body ;

: method. ( word class method -- )
    \ M: pprint-word
    >r pprint-word pprint-word r>
    <block pprint-elements pprint-; block; ;

M: generic (see)
    dup dup "combination" word-prop swap see-body
    dup methods [ newline first2 method. ] each-with ;

GENERIC: class. ( word -- )

: methods. ( class -- )
    #! List all methods implemented for this class.
    dup class? [
        dup implementors [
            newline
            dup in. tuck dupd "methods" word-prop hash method.
        ] each-with
    ] [
        drop
    ] if ;

M: union class.
    newline
    \ UNION: pprint-word
    dup pprint-word
    members pprint-elements pprint-; ;

M: predicate class.
    newline
    \ PREDICATE: pprint-word
    dup superclass pprint-word
    dup pprint-word
    <block
    "definition" word-prop pprint-elements
    pprint-; block; ;

M: tuple-class class.
    newline
    \ TUPLE: pprint-word
    dup pprint-word
    "slot-names" word-prop [ text ] each
    pprint-; ;

M: word class. drop ;

: see ( word -- )
    [
        dup (synopsis)
        dup (see)
        dup class.
        methods.
        newline
    ] with-pprint ;