option explicit
!INC Local Scripts.EAConstants-VBScript
' Script Name: DefaultLineStyles
' Author: Geert Bellekens, Patrick Gottschalk
' Purpose: Allows to change the linestyles to their default
' Date: 27/04/2015; 03.02.2016 (LC->LV für Mode 3 korrigiert), 04.02.2016 trace (UML, Abstraction) -> Mode 3, LV
'
dim lsDirectMode, lsAutoRouteMode, lsCustomMode, lsTreeVerticalTree, lsTreeHorizontalTree, _
lsLateralHorizontalTree, lsLateralVerticalTree, lsOrthogonalSquareTree, lsOrthogonalRoundedTree
lsDirectMode = "1"
lsAutoRouteMode = "2"
lsCustomMode = "3"
lsTreeVerticalTree = "V"
lsTreeHorizontalTree = "H"
lsLateralHorizontalTree = "LH"
lsLateralVerticalTree = "LV"
lsOrthogonalSquareTree = "OS"
lsOrthogonalRoundedTree = "OR"
dim defaultStyle
dim menuDefaultLines
'*********EDIT BETWEEN HERE*************
' set here the default style to be used
defaultStyle = lsOrthogonalSquareTree
' set there the style to be used for each type of connector
function determineStyle(connector)
dim connectorType
connectorType = connector.Type
select case connectorType
case "StateFlow","ObjectFlow","InformationFlow"
determineStyle = lsOrthogonalRoundedTree
case "Generalization", "Realization", "Realisation"
determineStyle = lsTreeVerticalTree
case "UseCase","NoteLink" ' Original:"UseCase", "Dependency","NoteLink"
determineStyle = lsDirectMode
case "Dependency"
determineStyle = lsOrthogonalSquareTree'lsLateralVerticalTree
case "Abstraction" ' UML trace
determineStyle = lsTreeHorizontalTree
case else
determineStyle = defaultStyle
end select
end function
'************AND HERE****************
sub main
dim diagram
dim diagramLink
dim connector
dim dirty
dirty = false
set diagram = Repository.GetCurrentDiagram
'save the diagram first
Repository.SaveDiagram diagram.DiagramID
'then loop all diagramLinks
if not diagram is nothing then
for each diagramLink in diagram.DiagramLinks
set connector = Repository.GetConnectorByID(diagramLink.ConnectorID)
if not connector is nothing then
'set the connectorstyle
setConnectorStyle diagramLink, determineStyle(connector)
'save the diagramlink
diagramLink.Update
dirty = true
end if
next
'reload the diagram if we changed something
if dirty then
'reload the diagram to show the link style
Repository.ReloadDiagram diagram.DiagramID
end if
end if
end sub
main
'gets the diagram link object
function getdiagramLinkForConnector(connector, diagram)
dim diagramLink
set getdiagramLinkForConnector = nothing
for each diagramLink in diagram.DiagramLinks
if diagramLink.ConnectorID = connector.ConnectorID then
set getdiagramLinkForConnector = diagramLink
exit for
end if
next
end function
'actually sets the connector style
function setConnectorStyle(diagramLink, connectorStyle)
'split the style into its parts
dim styleparts
dim styleString
styleString = diagramLink.Style
styleparts = Split(styleString,";")
dim stylePart
dim mode
dim modeIndex
modeIndex = -1
dim tree
dim treeIndex
treeIndex = -1
mode = ""
tree = ""
dim i
'find if Mode and Tree are already defined
for i = 0 to Ubound(styleparts) -1
stylePart = styleparts(i)
if Instr(stylepart,"Mode=") > 0 then
modeIndex = i
elseif Instr(stylepart,"TREE=") > 0 then
treeIndex = i
end if
next
'these connectorstyles use mode=3 and the tree
if connectorStyle = lsTreeVerticalTree or _
connectorStyle = lsTreeHorizontalTree or _
connectorStyle = lsLateralHorizontalTree or _
connectorStyle = lsLateralVerticalTree or _
connectorStyle = lsOrthogonalSquareTree or _
connectorStyle = lsOrthogonalRoundedTree then
mode = "3"
tree = connectorStyle
else
mode = connectorStyle
end if
'set the mode value
if modeIndex >= 0 then
styleparts(modeIndex) = "Mode=" & mode
diagramLink.Style = join(styleparts,";")
else
diagramLink.Style = "Mode=" & mode& ";"& diagramLink.Style
end if
'set the tree value
if treeIndex >= 0 then
if len(tree) > 0 then
styleparts(treeIndex) = "TREE=" & tree
diagramLink.Style = join(styleparts,";")
else
'remove tree part
diagramLink.Style = replace(diagramLink.Style,styleparts(treeIndex)&";" , "")
end if
else
diagramLink.Style = diagramLink.Style & "TREE=" & tree & ";"
end if
end function
function getConnectorStyle(diagramLink)
'split the style
dim styleparts
styleparts = Split(diagramLink.Style,";")
dim stylePart
dim mode
dim tree
mode = ""
tree = ""
for each stylepart in styleparts
if Instr(stylepart,"Mode=") > 0 then
mode = right(stylepart, 1)
elseif Instr(stylepart,"TREE=") > 0 then
tree = replace(stylepart, "TREE=", "")
end if
next
if tree <> "" then
getConnectorStyle = tree
else
getConnectorStyle = mode
end if
end function
