"

 Copyright (c) 2010-2011 Jan Vrany, Jan Kurs & Marcel Hlopko,

                         SWING Research Group, Czech Technical University 

                         in Prague

 Permission is hereby granted, free of charge, to any person

 obtaining a copy of this software and associated documentation

 files (the 'Software'), to deal in the Software without

 restriction, including without limitation the rights to use,

 copy, modify, merge, publish, distribute, sublicense, and/or sell

 copies of the Software, and to permit persons to whom the

 Software is furnished to do so, subject to the following

 conditions:

 The above copyright notice and this permission notice shall be

 included in all copies or substantial portions of the Software.

 THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,

 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES

 OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT

 HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,

 WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

 FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR

 OTHER DEALINGS IN THE SOFTWARE.

"

"{ Package: 'stx:libjava' }"

Lookup subclass:#JavaLookup

	classVariableNames:'Instance InvokeRSelectors'

	poolDictionaries:''

	category:'Languages-Java-Interop'

!

Lookup subclass:#J2S

	instanceVariableNames:''

	classVariableNames:''

	poolDictionaries:''

	privateIn:JavaLookup

!

Lookup subclass:#S2J

	instanceVariableNames:''

	classVariableNames:''

	poolDictionaries:''

	privateIn:JavaLookup

!

!JavaLookup class methodsFor:'documentation'!

copyright

"

 Copyright (c) 2010-2011 Jan Vrany, Jan Kurs & Marcel Hlopko,

                         SWING Research Group, Czech Technical University 

                         in Prague

 Permission is hereby granted, free of charge, to any person

 obtaining a copy of this software and associated documentation

 files (the 'Software'), to deal in the Software without

 restriction, including without limitation the rights to use,

 copy, modify, merge, publish, distribute, sublicense, and/or sell

 copies of the Software, and to permit persons to whom the

 Software is furnished to do so, subject to the following

 conditions:

 The above copyright notice and this permission notice shall be

 included in all copies or substantial portions of the Software.

 THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,

 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES

 OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT

 HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,

 WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

 FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR

 OTHER DEALINGS IN THE SOFTWARE.

"

! !

!JavaLookup class methodsFor:'initialization'!

initialize

    "Invoked at system start or when the class is dynamically loaded."

    "/ please change as required (and remove this comment)

    InvokeRSelectors := 

        #(

            " 0" #'_INVOKEVIRTUAL_R:_:'

            " 1" #'_INVOKEVIRTUAL_R:_:_:'

            " 2" #'_INVOKEVIRTUAL_R:_:_:_:'

            " 3" #'_INVOKEVIRTUAL_R:_:_:_:_:'

            " 4" #'_INVOKEVIRTUAL_R:_:_:_:_:_:'

            " 5" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:'

            " 6" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:_:'

            " 7" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:_:_:'

            " 8" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:_:_:_:'

            " 9" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:_:_:_:_:'

            "10" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:_:_:_:_:_:'

            "11" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:_:_:_:_:_:_:'

            "12" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:_:_:_:_:_:_:_:'

            "13" #'_INVOKEVIRTUAL_R:_:_:_:_:_:_:_:_:_:_:_:_:_:_:'

            " 0" #'_INVOKEINTERFACE_R:_:'

            " 1" #'_INVOKEINTERFACE_R:_:_:'

            " 2" #'_INVOKEINTERFACE_R:_:_:_:'

            " 3" #'_INVOKEINTERFACE_R:_:_:_:_:'

            " 4" #'_INVOKEINTERFACE_R:_:_:_:_:_:'

            " 5" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:'

            " 6" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:_:'

            " 7" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:_:_:'

            " 8" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:_:_:_:'

            " 9" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:_:_:_:_:'

            "10" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:_:_:_:_:_:'

            "11" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:_:_:_:_:_:_:'

            "12" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:_:_:_:_:_:_:_:'

            "13" #'_INVOKEINTERFACE_R:_:_:_:_:_:_:_:_:_:_:_:_:_:_:'

        )

    "Modified (format): / 20-01-2014 / 13:50:29 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!JavaLookup class methodsFor:'instance creation'!

cleanup

    Instance := nil.

    "Modified: / 25-02-2011 / 14:45:03 / kursjan <kursjan@fit.cvut.cz>"

    "Created: / 19-09-2011 / 23:25:08 / Jan Kurs <kursjan@fit.cvut.cz>"

    "Modified: / 15-12-2011 / 23:05:37 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

instance

    Instance isNil ifTrue:[

        Instance := self basicNew initialize

    ].

    ^Instance

    "Modified: / 25-02-2011 / 14:45:03 / kursjan <kursjan@fit.cvut.cz>"

    "Created: / 19-09-2011 / 23:20:49 / Jan Kurs <kursjan@fit.cvut.cz>"

    "Modified: / 15-12-2011 / 23:05:29 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

new

    ^ self shouldNeverBeSent.

    "Created: / 25-02-2011 / 14:44:43 / kursjan <kursjan@fit.cvut.cz>"

    "Modified: / 29-08-2011 / 21:10:12 / Jan Kurs <kursjan@fit.cvut.cz>"

! !

!JavaLookup methodsFor:'initialization'!

initialize

    "Invoked when a new instance is created."

    s2j := S2J new.

! !

!JavaLookup methodsFor:'lookup'!

lookupMethodForSelector:selector directedTo:initialSearchClass for:aReceiver withArguments:argArrayOrNil from:sendingContext ilc: ilcCache

    "Invoked by the VM to ask me for a method to call.

     The arguments are: the selector, receiver and arguments,

     the class to start the search in (for here-, super and directed sends)

     the sending context and the inline/poly cache (instance of

     PolymorphicInlineCache). "

     "JV@2012-08-21: Following C code speeds up Java<->Smalltalk interop by

      order of 2 magnitudes. However, it breaks Java reflection. 

      This is because it does not handle package private methods correctly.

      On the other hand, PP methods are not handled correctly when not using 

      reflection anyway, so lets keep the optimization here.

      Once we will have functinal JIT compiler, we can remove it and Java

      package-private method will be supported correctly in all cases.

     "

     | sender |

%{

    OBJ method;

    method = __lookup(initialSearchClass, selector);

    if ( method ) {

        if ( ilcCache ) {

            __ilcBind(ilcCache, initialSearchClass, method, selector);

        }

        RETURN (method);

    }

%}.

    "/ Sigh, here we have to care about different code paths. The new JIT compilation scheme

    "/ sends JavaVM>>_INVOKE*R:... for sends whose methodref is not yet resolved. Thus the stack

    "/ (starting with sendingContext) may look like:

    "/ 

    "/ (0) real sending context - whoever it is

    "/ 

    "/ for resolved/interpreted sends or:

    "/ 

    "/ (0) performWith:withArguments:

    "/ (1) _INVOKEVIRTUAL_R:* / _INVOKEINTERFACE_R:*

    "/ (2) real sending context - whoever it is

    "/ 

    "/ for unresolved sends from JIT-compiled code.

    "/ In tha latter case we have to skip those 2 'implementation' contexts:

    sender := sendingContext.

    sender selector == #perform:withArguments: ifTrue:[

        sender := sender sender.

        (sender receiver == JavaVM and: [ InvokeRSelectors includes: sender selector ]) ifTrue:[

            sender := sender sender.

        ] ifFalse:[

            sender := sendingContext.

        ].

    ].

    sender programmingLanguage isSmalltalk ifTrue:[

        aReceiver class theNonMetaclass programmingLanguage isJavaLike ifTrue:[

            ^s2j lookupMethodForSelector:selector directedTo:initialSearchClass for:aReceiver withArguments:argArrayOrNil from:sendingContext ilc: ilcCache.

        ].

    ].

    sender programmingLanguage isJavaLike ifTrue:[

        initialSearchClass programmingLanguage isSmalltalk ifTrue:[

            "Java to Smalltalk send"

            ^j2s lookupMethodForSelector:selector directedTo:initialSearchClass for:aReceiver withArguments:argArrayOrNil from:sendingContext ilc: ilcCache.

        ].

        initialSearchClass programmingLanguage isJavaLike ifTrue:[

            "Java to Java send"

        ].

    ].

    ^super lookupMethodForSelector:selector directedTo:initialSearchClass for:aReceiver withArguments:argArrayOrNil from:sendingContext ilc: ilcCache.

    "Created: / 01-10-2011 / 13:18:40 / Jan Kurs <kursjan@fit.cvut.cz>"

    "Created: / 15-12-2011 / 23:11:45 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!JavaLookup::J2S methodsFor:'lookup'!

lookupMethodForSelector:selector directedTo:initialSearchClass for:receiver withArguments:argArrayOrNil from:sendingContext ilc: ilc

    | d m |

    d := JavaDescriptor readFrom: (selector readStream skipThrough: $(; backStep; yourself).

    m := self lookupMethodForSelector: selector directedTo: initialSearchClass numArguments: d numArgs.

    m notNil ifTrue: [

        m := self compileProxyWithSelector: selector descriptor: d in: receiver class calling: m.

        ilc notNil ifTrue:[ilc bindTo: m forClass: receiver class].

        ^m.

    ] ifFalse:[

        ^ nil

    ]

    "Created: / 06-09-2011 / 22:04:04 / Jan Kurs <kursjan@fit.cvut.cz>"

    "Modified: / 09-10-2011 / 22:59:18 / kursjan <kursjan@fit.cvut.cz>"

    "Created: / 19-11-2011 / 12:37:26 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

lookupMethodForSelector:jselector directedTo:initialSearchClass numArguments: nArgs

    | name cls |

    name := jselector upTo: $(.

    "Pass 1: Search for explicitly annotated method"

    cls := initialSearchClass.

    [ cls notNil ] whileTrue:[

        cls methodDictionary keysAndValuesDo:[:sel :mthd|

            | jdescriptor |

            mthd numArgs == nArgs ifTrue:[

                (jdescriptor := mthd annotationAt: #javaselector:) notNil ifTrue:[

                    jdescriptor arguments first == jselector ifTrue: [

                        ^mthd    

                    ]

                ].

            ]

        ].

        cls := cls superclass.

    ].

    "Pass 2: Search for method with matching name"

    cls := initialSearchClass.

    [ cls notNil ] whileTrue:[

        cls methodDictionary keysAndValuesDo:[:sel :mthd|

            mthd numArgs == nArgs ifTrue:[

                (sel startsWith: name) ifTrue:[

                    ^mthd

                ]

            ]

        ].

        cls := cls superclass.

    ].

    ^nil

    "Created: / 16-12-2011 / 00:00:31 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!JavaLookup::J2S methodsFor:'utilities'!

addSelector:selector withMethod:proxy toClass:class 

    ProxyMethod installProxies ifTrue:[

        Class withoutUpdatingChangesDo:[

            class addSelector:selector withMethod:proxy.

        ]

    ]

    "Modified: / 23-12-2011 / 13:09:06 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

compileProxyWithSelector:selector descriptor:descriptor in:class calling:callee 

    |compiler proxy body|

    compiler := ProxyMethodCompiler new.

    body := compiler newJavaMethodInvocation: callee.

    body descriptor:descriptor.

    proxy := compiler 

                compile:body 

                arguments:callee numArgs 

                selector: selector.

"/    JK: do nod add here

"/    JV: Why?

    self 

        addSelector:selector

        withMethod:proxy

        toClass:class.

    ^ proxy

    "Created: / 14-12-2011 / 20:48:55 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 01-01-2012 / 19:33:45 / kursjan <kursjan@fit.cvut.cz>"

    "Modified: / 24-02-2012 / 20:36:24 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!JavaLookup::S2J methodsFor:'lookup'!

lookupMethodForSelector: selector directedTo: initialSearchClass for: receiver withArguments: argArrayOrNil from: sendingContext ilc: ilc 

    "

     As a courtesy to a Smalltalker, try to map smalltalk selectors to a java ones.

     Returns a method or nil"

    | name candidates m |

    name := selector upTo: $:.

    candidates := OrderedCollection new.

    candidates := self lookupMethodsForSelector: selector in: initialSearchClass ? receiver class static: receiver isBehavior.

    candidates notEmpty ifTrue: [

        "/ If candidates contains only one method that is not Java method,

        "/ then return this method. It's either a smalltalk extension or

        "/ ambiguous method trampoline...

        (candidates size == 1 and:[ (m := candidates anElement) isJavaMethod not]) ifTrue:[ 

            ilc notNil ifTrue: [ ilc bindTo: m forClass: receiver class ].

            ^ m.

        ].

        m := self 

                compileProxyWithSelector: selector

                in: receiver class

                candidates: candidates.

        ilc notNil ifTrue: [ ilc bindTo: m forClass: receiver class ].

         "Install the proxy"

        self 

            addSelector: selector

            withMethod: m

            toClass: receiver class.

        ^ m.

    ].

     "Hmm, hmm, maybe a public field?"

    (argArrayOrNil size < 2) ifTrue: [

        | field |

        field := initialSearchClass theNonMetaclass 

                lookupFieldFor: name

                static: initialSearchClass isMetaclass

                onlyPublic: true.

        field notNil ifTrue: [

            m := self 

                    compileProxyWithSelector: selector

                    in: receiver class

                    accessing: field.

            ilc notNil ifTrue: [ ilc bindTo: m forClass: receiver class ].

             "Install the proxy"

            self 

                addSelector: selector

                withMethod: m

                toClass: receiver class.

            ^ m.

        ]

    ].

    ^ nil

    "Created: / 21-02-2011 / 13:38:55 / kursjan <kursjan@fit.cvut.cz>"

    "Modified: / 29-08-2011 / 20:38:21 / kursjan"

    "Modified: / 20-09-2011 / 00:03:48 / Jan Kurs <kursjan@fit.cvut.cz>"

    "Modified (format): / 25-09-2011 / 21:08:45 / Jan Kurs <kursjan@fit.cvut.cz>"

    "Created: / 19-11-2011 / 13:03:39 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 01-01-2012 / 19:58:59 / kursjan <kursjan@fit.cvut.cz>"

    "Modified (comment): / 02-01-2012 / 10:35:25 / kursjan <kursjan@fit.cvut.cz>"

    "Modified: / 18-11-2012 / 18:17:28 / Marcel Hlopko <hlopkmar@fel.cvut.cz>"

    "Modified: / 16-12-2012 / 13:59:55 / Marcel Hlopko <marcel.hlopko@fit.cvut.cz>"

    "Modified: / 19-03-2014 / 17:27:55 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

lookupMethodsForSelector: selector in: initialSearchClass static: static

    "Lookup all matching methods for given (Smalltalk) selector starting with

     `initialSearchClass`. If `static` is true, then search for static methods,

     otherwise for search instance methods.

     Returns a list of method candidates that match given selector.

    "

    | name nameSizePlusOne numArgs candidates  finder1 finder2  cls ifacesQ ifacesSeen |

    name := selector upTo: $:.

    nameSizePlusOne := name size + 1.

    numArgs := selector numArgs.

    candidates := OrderedCollection new.

    ifacesSeen := Set new.

    "/ Method finder to lookup extension methods in interfaces...

    finder1 := [:cls |

        cls interfaces notEmptyOrNil ifTrue:[

            ifacesQ := OrderedCollection with: (cls interfaces).

            [ ifacesQ notEmpty ] whileTrue:[

                | ifaces newIfaces extension |

                ifaces := ifacesQ removeFirst.

                extension := nil.

                ifaces do:[:iface |

                    (ifacesSeen includes: iface) ifFalse:[

                        | m |

                        ifacesSeen add: iface.

                        m := iface compiledMethodAt: selector.

                        m notNil ifTrue:[

                            extension notNil ifTrue:[

                                "/ Ambiguous, return error trampoline