static void drawCall(DiagramDrawer &drawer, GraphPoint source,
GraphPoint target, bool isConst, int arrowLen)
{
drawer.drawLine(source, target, isConst);
// Draw arrow
if(source.x > target.x)
arrowLen = -arrowLen;
drawer.drawLine(GraphPoint(target.x-arrowLen, target.y-arrowLen),
GraphPoint(target.x, target.y));
drawer.drawLine(GraphPoint(target.x-arrowLen, target.y+arrowLen),
GraphPoint(target.x, target.y));
}
void OperationDrawer::drawLifeLines(DiagramDrawer &drawer,
const std::vector<OperationClass> &classes,
std::vector<int> const &classEndY, int endy)
{
drawer.groupShapes(true, Color(0,0,0), Color(245,245,255));
endy += mCharHeight;
for(size_t i=0; i<classes.size(); i++)
{
const auto &cl = classes[i];
int x = cl.getLifelinePosX();
drawer.drawLine(GraphPoint(x, classEndY[i]), GraphPoint(x, endy));
}
drawer.groupShapes(false, Color(0,0,0), Color(245,245,255));
}
GraphSize OperationDrawer::drawOperationNoText(DiagramDrawer &drawer, GraphPoint pos,
OperationDefinition &operDef, const OperationGraph &graph,
const OperationDrawOptions &options,
std::set<const OperationDefinition*> &drawnOperations,
std::vector<DrawString> &drawStrings, bool draw, int callDepth)
{
GraphPoint startpos = pos;
// Add space between bottom of class and operation name
int starty = startpos.y+(mPad*2);
int y=starty;
size_t sourceIndex = operDef.getOperClassIndex();
int arrowLen = mCharHeight * 7 / 10;
std::vector<int> condStartPosY;
drawnOperations.insert(&operDef);
const OperationClass &cls = graph.getClass(sourceIndex);
if(callDepth == 0)
{
// Reinit all polys to initial conditions.
mLifelinePolygons.clear();
mLifelinePolygons.resize(graph.getClasses().size());
}
BlockPolygon &poly = mLifelinePolygons[sourceIndex];
poly.setup(cls.getLifelinePosX(), mPad);
if(!graph.isOperCalled(operDef))
{
// Show starting operation
operDef.setRect(GraphPoint(cls.getPosition().x, y),
GraphSize(mCharHeight*50, mCharHeight+mPad));
y += mCharHeight;
drawStrings.push_back(DrawString(GraphPoint(
cls.getPosition().x, y), operDef.getName()));
// Add space between operation name and called operations.
y += (mPad * 2);
int lineY = y + mPad*2;
if(draw)
{
drawCall(drawer, GraphPoint(cls.getPosition().x, lineY),
GraphPoint(cls.getLifelinePosX(), lineY),
operDef.isConst(), arrowLen);
}
}
for(const auto &stmt : operDef.getStatements())
{
int condOffset = poly.getDepth() * mPad;
switch(stmt->getStatementType())
{
case ST_Call:
{
OperationCall *call = stmt->getCall();
size_t targetIndex = call->getOperClassIndex();
int lineY = y + mCharHeight + mPad*2;
int sourcex = cls.getLifelinePosX();
sourcex += condOffset;
const OperationClass &targetCls = graph.getClass(targetIndex);
int targetx = targetCls.getLifelinePosX();
if(targetIndex == NO_INDEX)
{
// Handle [else]
// int len = mCharHeight*3;
// mDrawer.drawLine(GraphPoint(sourcex, lineY),
// GraphPoint(sourcex+len, lineY), call->isConst());
}
else if(targetIndex != sourceIndex)
{
if(draw)
{
drawCall(drawer, GraphPoint(sourcex, lineY),
GraphPoint(targetx, lineY), call->isConst(), arrowLen);
}
}
else
{
// Draw line back to same class
int len = mCharHeight*3;
int height = 3;
if(draw)
{
drawer.drawLine(GraphPoint(sourcex, lineY),
GraphPoint(sourcex+len, lineY), call->isConst());
drawer.drawLine(GraphPoint(sourcex+len, lineY),
GraphPoint(sourcex+len, lineY+height));
drawer.drawLine(GraphPoint(sourcex, lineY+height),
GraphPoint(sourcex+len, lineY+height), call->isConst());
}
y += height;
}
int textX = (sourceIndex < targetIndex) ? cls.getLifelinePosX() :
targetCls.getLifelinePosX();
GraphPoint callPos(textX+condOffset+mPad, y+mCharHeight);
drawStrings.push_back(DrawString(callPos, call->getName()));
call->setRect(GraphPoint(callPos.x, callPos.y-mCharHeight),
GraphSize(mCharHeight*50, mCharHeight+mPad));
y += mCharHeight*2;
// Draw child definition.
OperationDefinition *childDef = graph.getOperDefinition(*call);
if(childDef)
{
// poly.startChildBlock(condDepth, y);
if(drawnOperations.find(childDef) == drawnOperations.end())
{
poly.incDepth(y);
GraphSize childSize = drawOperationNoText(drawer,
GraphPoint(pos.x, y), *childDef, graph, options,
drawnOperations, drawStrings, draw, callDepth+1);
y += childSize.y + mPad * 2;
poly.decDepth(y);
}
else
{
// This draws a rectangle to signify that it is defined
// elsewhere in the diagram.
GraphRect rect(targetx + mPad*2, callPos.y + mPad*2,
mCharHeight+mPad, mCharHeight+mPad);
if(draw)
{
drawer.drawRect(rect);
}
y += mCharHeight+mPad;
}
// poly.endChildBlock(condDepth, y);
}
}
break;
case ST_VarRef:
{
OperationVarRef *ref = stmt->getVarRef();
size_t targetIndex = ref->getOperClassIndex();
int lineY = y + mCharHeight + mPad*2;
int sourcex = cls.getLifelinePosX();
sourcex += condOffset;
const OperationClass &targetCls = graph.getClass(targetIndex);
int targetx = targetCls.getLifelinePosX();
if(targetIndex == NO_INDEX)
{
int len = mCharHeight*3;
if(draw)
{
drawer.drawLine(GraphPoint(sourcex, lineY),
GraphPoint(sourcex+len, lineY), ref->isConst());
}
}
else if(targetIndex != sourceIndex)
{
if(draw)
{
drawCall(drawer, GraphPoint(sourcex, lineY),
GraphPoint(targetx, lineY), ref->isConst(), arrowLen);
}
}
else
{
// Draw line back to same class
int len = mCharHeight*3;
int height = 3;
if(draw)
{
drawer.drawLine(GraphPoint(sourcex, lineY),
GraphPoint(sourcex+len, lineY), ref->isConst());
drawer.drawLine(GraphPoint(sourcex+len, lineY),
GraphPoint(sourcex+len, lineY+height));
drawer.drawLine(GraphPoint(sourcex, lineY+height),
GraphPoint(sourcex+len, lineY+height), ref->isConst());
}
y += height;
}
int textX = (sourceIndex < targetIndex) ? cls.getLifelinePosX() :
targetCls.getLifelinePosX();
GraphPoint callPos(textX+condOffset+mPad, y+mCharHeight);
if(draw)
{
drawStrings.push_back(DrawString(callPos, ref->getName()));
}
ref->setRect(GraphPoint(callPos.x, callPos.y-mCharHeight),
GraphSize(mCharHeight*50, mCharHeight+mPad));
y += mCharHeight*2;
}
break;
case ST_OpenNest:
{
GraphPoint lifePos(cls.getLifelinePosX()+condOffset+
mPad, y+mCharHeight);
const OperationNestStart *cond = stmt->getNestStart();
if(draw)
{
drawStrings.push_back(DrawString(lifePos, cond->getExpr()));
}
condStartPosY.push_back(y);
y += mCharHeight*2;
poly.incDepth(y);
}
break;
case ST_CloseNest:
{
poly.decDepth(y);
}
break;
}
}
if(callDepth == 0)
{
for(auto &poly : mLifelinePolygons)
{
poly.finishBlock();
if(draw)
{
drawer.drawPoly(poly, Color(245,245,255));
}
}
}
return GraphSize(0, y - startpos.y);
}