diff options
Diffstat (limited to 'tropical/qt/caldrawer.py')
| -rw-r--r-- | tropical/qt/caldrawer.py | 192 |
1 files changed, 192 insertions, 0 deletions
diff --git a/tropical/qt/caldrawer.py b/tropical/qt/caldrawer.py new file mode 100644 index 0000000..7e6b3fc --- /dev/null +++ b/tropical/qt/caldrawer.py @@ -0,0 +1,192 @@ + +from PyQt6.QtWidgets import QGraphicsScene, QGraphicsView, QSizePolicy +from PyQt6 import uic, QtGui, QtCore +from PyQt6.QtCore import Qt, QRect +import math +#https://forum.qt.io/topic/93327/how-can-i-use-qpainter-to-paint-on-qgraphicsview/5 +#https://www.pythonguis.com/tutorials/pyqt6-bitmap-graphics/#qpainter + +# to solve the fit problem: https://stackoverflow.com/questions/61886358/qgraphicsview-fitinview-not-working-as-expected + +class CalQGraphicsView(QGraphicsView): + def __init__(self, scene): + super().__init__(None) + self.setScene(scene) + + def resizeEvent(self, event): + self.fitInView(self.sceneRect(), Qt.AspectRatioMode.IgnoreAspectRatio) + + +class CalDrawerScene(QGraphicsScene): + def __init__(self, calState): + self.gridWidth=2 + self.daysLabelBG="#dddddd" + self.eventsLabelBG="#36e364" + super().__init__(None) + self.showWeekends=True + self.daysRect=list() + self.eventsRect=list() + self.calState=calState + self.daysNames=["Monday","Tuesday","Wednesday","Thursday","Friday","Saturday","Sunday"] + + def drawForeground(self, painter, rect): + origXF, origYF, widthF, heightF = rect.getRect() + origXI, origYI, widthI, heightI = (int(origXF),int(origYF),int(widthF),int(heightF)) + self.defaultBrush=painter.brush() + self.daysRect.clear() + self.eventsRect.clear() + daysNamesH=self.drawDaysName(painter, origXI, origYI, widthI, heightI) + self.drawGrid(painter, origXI, origYI+daysNamesH, widthI, heightI-daysNamesH) + self.drawDaysLabel(painter) + self.drawEvents(painter) + + def drawDaysName(self, painter, x, y, width, height): + # Init Pen + pen=QtGui.QPen() + pen.setWidth(self.gridWidth) + pen.setJoinStyle(Qt.PenJoinStyle.MiterJoin) + po=int(self.gridWidth/2) # Pen offset + painter.setPen(pen) + # Init Brush + painter.setBrush(Qt.BrushStyle.NoBrush) + # Setup dimensions + weekLength=7 + font=painter.font() + metric=QtGui.QFontMetrics(font); + labelH=metric.boundingRect("".join(self.daysNames)).height() + margin=0 + offsetY=int(labelH/2)+int(labelH/4)+margin + w=int((width-po*2)/weekLength) + h=int(labelH) + for i in range(0, 7): + labelW=metric.boundingRect(self.daysNames[i]).width() + offsetX=int(w/2-labelW/2) +# painter.drawText(x+offsetX+w*i,y+offsetY,self.daysNames[i]) + painter.drawText(x+w*i,y,w,h,Qt.AlignmentFlag.AlignHCenter,self.daysNames[i]) + rect=QRect(x+po+w*i,y+po,w,h) # Draw grid + painter.drawRect(rect) + return labelH + + def drawGrid(self,painter, x, y, width, height): + # Init Pen + pen=QtGui.QPen() + pen.setWidth(self.gridWidth) + pen.setJoinStyle(Qt.PenJoinStyle.MiterJoin) + po=int(self.gridWidth/2) # Pen offset + painter.setPen(pen) + # Init Brush + painter.setBrush(Qt.BrushStyle.NoBrush) + # Setup dimensions + weekLength=7 + if not self.showWeekends: + weekLength=5 + daysCount=weekLength*6 + + + # brush=QtGui.QBrush() + # brush.setColor(QtGui.QColor("#0000FF")) + # brush.setStyle(Qt.BrushStyle.SolidPattern) + + + w=int((width-po*2)/weekLength) + h=int((height-po*2)/6) + row=0 + col=0 + self.daysRect.clear() + while row*weekLength+col<daysCount: + rect=QRect(x+po+w*col,y+po+row*h,w,h) # Draw grid + rectInside=QRect(rect.x()+po,rect.y()+po,rect.width()-po*2,rect.height()-po*2) # Area within grid (within stroke) + self.daysRect.append(rectInside) # Store this region for later drawing + # if col==0 and row == 0: + # painter.setBrush(brush) + # painter.setPen(Qt.PenStyle.NoPen) + # painter.drawRect(QRect(b.x(),b.y(),b.width(),b.height())) + # painter.setPen(self.gridPen) + # painter.setBrush(self.defaultBrush) + painter.drawRect(rect) + col+=1 + if col==weekLength: + col=0 + row+=1 + + def drawDaysLabel(self,painter): + # Init Pen + pen=QtGui.QPen() + pen.setWidth(self.gridWidth) + pen.setJoinStyle(Qt.PenJoinStyle.MiterJoin) + po=int(self.gridWidth/2) # Pen offset + # Init Brush + brush=QtGui.QBrush() + brush.setColor(QtGui.QColor(self.daysLabelBG)) + brush.setStyle(Qt.BrushStyle.SolidPattern) + painter.setBrush(brush) + # Init various things + days=self.calState.getMonthDays() + font=painter.font() + metric=QtGui.QFontMetrics(font); + labelH=metric.boundingRect("1234567890").height() + margin=0 + offsetY=int(labelH/2)+int(labelH/4)+margin + # Draw labels + for i in range(0,len(self.daysRect)): + r=self.daysRect[i] + d=days[i] + dayLabel=str(d[2]) + labelW=metric.boundingRect(dayLabel).width() + offsetX=int(r.width()/2-labelW/2) + painter.setPen(Qt.PenStyle.NoPen) + painter.drawRect(r.x(),r.y(),r.width(),labelH) # Remember r is within grid stroke + painter.setPen(pen) +# painter.drawText(r.x()+offsetX,r.y()+offsetY,dayLabel) + painter.drawText(r,Qt.AlignmentFlag.AlignHCenter,dayLabel) + self.eventsRect.append(QRect(r.x(),r.y()+labelH+margin,r.width(),r.height()-(labelH+margin))) + + def drawEvents(self,painter): + # Init Pen + pen=QtGui.QPen() + pen.setWidth(self.gridWidth) + pen.setJoinStyle(Qt.PenJoinStyle.MiterJoin) + po=int(self.gridWidth/2) # Pen offset + # Init Brush + brush=QtGui.QBrush() + brush.setColor(QtGui.QColor(self.eventsLabelBG)) + brush.setStyle(Qt.BrushStyle.SolidPattern) + painter.setBrush(brush) + # Init various things + days=self.calState.getMonthDays() + font=painter.font() + metric=QtGui.QFontMetrics(font); + labelH=metric.height() + margin=0 + colMark=5 + colMarkPadding=2 + offsetY=int(labelH/2)+int(labelH/4)+margin + # Draw + po=int(self.gridWidth/2) # Pen offset + for r in self.eventsRect: + offsetX=colMark+colMarkPadding + painter.drawText(r.x()+po+offsetX,r.y(),r.width()-offsetX-po*2,r.height(),0,"event testddddddddddddd") + painter.setPen(Qt.PenStyle.NoPen) + painter.drawRect(r.x(),r.y(),colMark,labelH) # Remember r is within grid stroke + painter.setPen(pen) + +class CalDrawer(): + + def __init__(self, layout, calState): + self.gs=CalDrawerScene(calState) + self.gv=CalQGraphicsView(self.gs) + # Setup propertion + spLeft=QSizePolicy(QSizePolicy.Policy.Preferred,QSizePolicy.Policy.Preferred); + spLeft.setHorizontalStretch(3); + self.gv.setSizePolicy(spLeft); + + # if jean: + # spLeft=QSizePolicy(QSizePolicy.Policy.Preferred,QSizePolicy.Policy.Preferred); + # spLeft.setHorizontalStretch(2); + # self.gv.setSizePolicy(spLeft); + # else: + # spLeft=QSizePolicy(QSizePolicy.Policy.Preferred,QSizePolicy.Policy.Preferred); + # spLeft.setHorizontalStretch(1); + # self.gv.setSizePolicy(spLeft); + layout.addWidget(self.gv) + |
