"

 COPYRIGHT (c) 2010 by Jan Vrany, SWING Research Group. CTU in Prague

	      All Rights Reserved

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:libtool' }"

"{ NameSpace: Tools }"

CodeViewService subclass:#BreakpointService

	instanceVariableNames:'breakpoints currentMethod mode'

	classVariableNames:''

	poolDictionaries:''

	category:'Interface-CodeView'

!

!BreakpointService class methodsFor:'documentation'!

copyright

"

 COPYRIGHT (c) 2010 by Jan Vrany, SWING Research Group. CTU in Prague

	      All Rights Reserved

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.

"

! !

!BreakpointService methodsFor:'accessing'!

breakpoints

    ^ breakpoints

! !

!BreakpointService methodsFor:'change & update'!

update: aspect with: param from: sender

    aspect == #visibility ifTrue:[^ self].

    aspect == #sizeOfView ifTrue:[^ self].

    aspect == #classHolder ifTrue:[^ self].

    aspect == #languageHolder ifTrue:[^ self].

    sender == codeView modifiedChannel ifTrue:[^ self].

    (aspect == #methodHolder or:[sender == codeView methodHolder]) ifTrue:[

	self updateCurrentMethod.

    ].

    super update: aspect with: param from: sender

    "Created: / 06-07-2011 / 15:21:08 / cg"

!

updateBreakPointsFor:aMethod

    |methodsBreakPoints|

    aMethod notNil ifTrue:[

	mode == #vm ifTrue:[

	    (aMethod breakpointTable ? #()) do:[:each|

		each class == Breakpoint ifTrue:[

		    methodsBreakPoints add: each.

		].

	    ].

	] ifFalse:[

	aMethod literalsDo:[:eachLiteral |

	    eachLiteral class == Breakpoint ifTrue:[

		methodsBreakPoints isNil ifTrue:[

		    methodsBreakPoints := OrderedCollection new.

		].

		methodsBreakPoints add:eachLiteral copy.

	    ].

	].

	].

    ].

    breakpoints := methodsBreakPoints.

    currentMethod := aMethod.

    "Created: / 06-07-2011 / 15:24:09 / cg"

    "Modified: / 06-07-2011 / 17:32:54 / jv"

    "Modified: / 15-04-2013 / 12:33:48 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

updateCurrentMethod

    |method realMethod|

    method := realMethod := codeView methodHolder value.

    (method notNil and:[method mclass isNil]) ifTrue:[

	"/ hack: ouch - was wrapped in the meantime;

	"/ hurry up and update. Should be done elsewhere (in codeView)

	realMethod := MethodWithBreakpoints allInstances detect:[:m | m originalMethod == method and:[m mclass notNil]] ifNone:nil.

	realMethod isNil ifTrue:[

	    realMethod := WrappedMethod allInstances detect:[:m | m originalMethod == method and:[m mclass notNil]] ifNone:nil.

	].

    ].

    realMethod ~~ currentMethod ifTrue:[

	"/ codeView methodHolder setValue:realMethod.

	self updateBreakPointsFor:realMethod.

	gutterView invalidate.

    ].

! !

!BreakpointService methodsFor:'event handling'!

buttonPress:button x:x y:y in:view

    |lineNr|

    codeView methodHolder value isNil ifTrue:[

	^ false

    ].

    view == gutterView ifTrue:[

	button == 1 ifTrue:[

	    lineNr := textView yVisibleToLineNr:y.

	    lineNr notNil ifTrue:[ self setOrToggleBreakpointAtLine:lineNr ].

	    ^ true.

	].

	button == 3 ifTrue:[

	    ^ true.

	]

    ].

    ^ false

    "Created: / 17-06-2011 / 13:05:22 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 28-06-2011 / 08:31:39 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 19-09-2011 / 14:41:00 / cg"

!

linesDeletedFrom: start to: end

    breakpoints isEmptyOrNil ifTrue:[^self].

    self moveBreakpointsAfterLine: start - 1 by: (end - start + 1) negated

    "Created: / 06-07-2011 / 17:16:27 / jv"

!

linesInsertedFrom: start to: end

    breakpoints isEmptyOrNil ifTrue:[^self].

    self moveBreakpointsAfterLine: start - 1 by: (end - start + 1)

    "Created: / 06-07-2011 / 17:16:36 / jv"

! !

!BreakpointService methodsFor:'initialization'!

initialize

    super initialize.

    breakpoints := OrderedCollection new.

    mode := (ConfigurableFeatures includesFeature:#VMBreakpointSupport)

		ifTrue:[#vm]

		ifFalse:[#compiled].

    "Created: / 17-06-2011 / 13:49:12 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 15-04-2013 / 12:28:50 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!BreakpointService methodsFor:'private'!

breakpointAtLine:line

    |pos|

    breakpoints isNil ifTrue:[^ nil].

    pos := textView characterPositionOfLine:line col:1.

    ^ breakpoints detect:[:each | each position = pos ] ifNone:[ nil ]

    "Modified: / 17-06-2011 / 13:59:17 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified (format): / 05-07-2011 / 21:33:23 / cg"

!

moveBreakpointsAfterLine:line by: delta

    |pos |

    breakpoints do:[:bpnt|

	bpnt line >= line ifTrue:[

	    pos := textView characterPositionOfLine:bpnt line + delta col:1.

	    bpnt position:pos line:(bpnt line + delta).

	]

    ].

    "/gutterView redrawLinesFrom: line.

    "Created: / 17-06-2011 / 13:45:22 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Created: / 06-07-2011 / 17:26:30 / jv"

    "Modified: / 02-08-2012 / 09:27:10 / cg"

!

recompile

    "recompile the current method for changed breakpoints"

    |oldMethod newMethod compiler class selector|

    oldMethod := codeView methodHolder value.

    (oldMethod notNil and:[oldMethod hasPrimitiveCode not]) ifTrue:[

        "/ be careful: if the text has been edited/modified, do not compile

        textView modified ifTrue:[

            self breakPoint: #cg.

            self breakPoint: #jv.

            ^self.

        ] ifFalse:[

            "/ prepare to get reachable bpts

            breakpoints do:[:bp | bp isReached:false].

            class := oldMethod mclass.

            class isNil ifTrue:[

                class := codeView classHolder value.

                class isNil ifTrue:[

                    self breakPoint:#jv.

                    Dialog warn:'oops - lost the methods''s class'.

                    ^ self.

                ]

            ].

            selector := oldMethod selector.

            Class withoutUpdatingChangesDo:[

                compiler := ByteCodeCompilerWithBreakpointSupport new.

                compiler breakpoints:breakpoints.

                compiler methodClass:MethodWithBreakpoints.

                newMethod := compiler

                            compile:oldMethod source

                            forClass:class

                            inCategory:oldMethod category

                            notifying:nil

                            install:false

                            skipIfSame:false

                            silent:true

                            foldConstants:true

                            ifFail:[ Transcript showCR:'BreakpointService: failed to recompile for breakpoint' ].

                selector isNil ifTrue:[

                    "/ May happen as the selector is not stored in the method but

                    "/ searches through method's mclass methodDictionary.

                    "/ Following should be save as breakpoint is not installed when

                    "/ the code is modified...

                    selector := compiler selector.

                ].

                oldMethod isWrapped ifTrue:[

                    "/ update the wrapped method - do not install

                    newMethod originalMethod: oldMethod originalMethod.

                    oldMethod replaceOriginalMethodWith:newMethod.

                ] ifFalse:[

                    "/ install

                    newMethod originalMethod: oldMethod.

                    (class primAddSelector: selector withMethod:newMethod) ifFalse:[

                        oldMethod mclass:class.

                        self breakPoint: #cg.

                        self breakPoint: #jv.

                        ^ self

                    ].

                ].

                breakpoints := breakpoints

                                select:[:bp |

"/                                    bp isReached ifFalse:[ 

"/                                        "/ Transcript show:'remove unreached:'; showCR:bp 

"/                                    ].

                                    bp isReached

                                ].

                "/ must update breakpoints BEFORE the following, because it leads to a change

                "/ notification, which may clear the breakpoints collection!!

                codeView methodHolder value:newMethod.

                oldMethod mclass isNil ifTrue:[

                    "/ although this is not strictly true, not doing this

                    "/ would confuse a lot of other tools (such as the browser)

                    oldMethod mclass:class.

                ].

                class changed:#methodTrap with:selector. "/ tell browsers

                Smalltalk changed:#methodTrap with:(MethodTrapChangeNotificationParameter changeClass:class changeSelector:selector).

            ].

        ]

    ]

    "Created: / 05-07-2011 / 21:33:13 / cg"

    "Modified: / 18-07-2012 / 10:53:22 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 02-08-2012 / 09:35:41 / cg"

!

reinstall

    | vmbreakpoints i |

    vmbreakpoints := Array new: breakpoints size * 2.

    i := 1.

    breakpoints do:[:breakpoint|

	vmbreakpoints at: i     put: breakpoint line.

	vmbreakpoints at: i + 1 put: breakpoint.

	i := i + 2.

    ].

    currentMethod breakpointTable: vmbreakpoints

    "Created: / 15-04-2013 / 12:32:15 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

setOrToggleBreakpointAtLine:line

    |pos bpnt prepareFullBreakSupport|

    "/ if true, setting a single breakpoint in a method will create

    "/ a whole set of invisible (and disabled) breakpoints in that method,

    "/ one for each line.

    "/ These can later be enabled in the debugger

    "/ (otherwise, the debugger's behavior is stupid, as it cannot recompile a method

    "/ to set additional breakpoints).

    "/ We accept the additional overhead, as we are in debug mode anyway.

    "/ prepareFullBreakSupport := false.

    prepareFullBreakSupport := true.

    textView reallyModified ifTrue:[

        "/ leads to ugly behavior (method no longer found), if we allow

        "/ this...

          Dialog warn:'Please accept first (cannot set breakpoint while text is modified)'.

        ^ self

    ].

    pos := textView characterPositionOfLine:line col:1.

    bpnt := self breakpointAtLine:line.

    bpnt isNil ifTrue:[

        "/ no breakpoint there - add as required

        (self canCreateOrToggleBreakpointAtLine:line) ifTrue:[

            prepareFullBreakSupport ifTrue:[

                "/ add a (disabled) breakpoint for every source line. This

                "/ allows for breakpoints to be enabled/disabled in the debugger...

                1 to:textView numberOfLines do:[:eachLine |

                    |oldBPnt eachPos otherBpnt|

                    oldBPnt := self breakpointAtLine:eachLine.

                    oldBPnt isNil ifTrue:[

                        eachPos := textView characterPositionOfLine:eachLine col:1.

                        breakpoints isNil ifTrue:[ breakpoints := OrderedCollection new].

                        breakpoints add:((otherBpnt := Breakpoint new) position:eachPos line:eachLine).

                        eachLine == line ifTrue:[

                            bpnt := otherBpnt.

                        ] ifFalse:[

                            otherBpnt beInvisible.

                        ]

                    ].

                ].

            ] ifFalse:[

                breakpoints add:((bpnt := Breakpoint new) position:pos line:line).

            ].

            Display shiftDown ifTrue:[

                "/ trace

                bpnt beTracepoint

            ].

            self assert: breakpoints notEmptyOrNil.

            self recompile.

        ] ifFalse:[

            codeView topView class == DebugView ifTrue:[

                Dialog warn:'Sorry, in an active method, I can only add new breakpoints in an already breakpointed method.

(i.e. a method stopped at a method breakpoint or one which already has statement breakpoints)

The reason is that the method needs to be recompiled for the breakpoint, which would not affect the currently executed method.'.

                "/ Dialog warn:'Sorry, can only add a new breakpoint in a wrapped method which has not yet started.'.

            ] ifFalse:[

                Dialog warn:'Sorry, cannot add a new breakpoint here.'.

            ].

        ]

    ] ifFalse:[

        "/ breakpoint already there - just enable/disable

        Display shiftDown ifTrue:[

            bpnt toggleTracing

        ] ifFalse:[

            bpnt toggle.

        ].

        currentMethod mclass isNil ifTrue:[

            "/ hack: ouch - was wrapped in the meantime;

            "/ hurry up and update. Should be done elsewhere (in codeView)

            self updateCurrentMethod.

        ].

        Smalltalk changed:#methodTrap with:(MethodTrapChangeNotificationParameter changeClass:currentMethod mclass changeSelector:currentMethod selector).

    ].

    gutterView redrawLine:line.

    "Created: / 17-06-2011 / 13:45:22 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 27-07-2011 / 13:27:55 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 02-08-2012 / 09:26:38 / cg"

! !

!BreakpointService methodsFor:'queries'!

canCreateOrToggleBreakpointAtLine:lineOrNilForAnywhere

    |pos bpnt topView|

    textView reallyModified ifTrue:[

	^ false

    ].

    "/ can always toggle existing breakpoints...

    lineOrNilForAnywhere notNil ifTrue:[

	pos := textView characterPositionOfLine:lineOrNilForAnywhere col:1.

	bpnt := self breakpointAtLine:lineOrNilForAnywhere.

	bpnt notNil ifTrue:[

	    ^ true.

	]

    ].

    ((topView := codeView topView) class == DebugView) ifTrue:[

	"/ can only create new breakpoints in the debugger,

	"/ iff we are in a wrapped method's prolog

	topView selectedContextIsWrapped ifTrue:[

	    topView selectedContext lineNumber == 1 ifTrue:[

		^ true

	    ].

	].

	^ false.

    ].

    ^  true.

!

hasBreakpoints

    ^ breakpoints notEmptyOrNil

! !

!BreakpointService methodsFor:'redrawing'!

drawLine:lineNo in:view atX:x y:y width:w height:h from:startCol to:endColOrNil with:fg and:bg

    "Called by both gutterView and textView (well, not yet) to

     allow services to draw custom things on text view.

     Ask JV what the args means if unsure (I'm lazy to document

     them, now it is just an experiment...)"

    |mthd bpnt icon dx dy|

    (mthd := codeView methodHolder value) isNil ifTrue:[

	^ self

    ].

    mthd mclass isNil ifTrue:[

	"/ hack: ouch - was wrapped in the meantime;

	^ self. "/ wait for the real update

	"/ hurry up and update. Should be done elsewhere (in codeView)

	self updateCurrentMethod.

    ].

   view == gutterView ifTrue:[

	bpnt := self breakpointAtLine:lineNo.

	bpnt isNil ifTrue:[

	    ^ self

	].

	icon := bpnt icon.

	icon isNil ifTrue:[

	    ^ self

	].

	dx := ((w - icon width) / 2) rounded.

	dy := ((h - icon height) / 2) rounded.

	icon

	    displayOn:view

	    x:x + dx

	    y:y - h + dy + 4. "TODO: Magic constant"

    ].

    "Created: / 17-06-2011 / 13:52:52 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified (format): / 05-07-2011 / 22:14:33 / cg"