summaryrefslogtreecommitdiff
path: root/tropical/qt/caldrawer.py
diff options
context:
space:
mode:
Diffstat (limited to 'tropical/qt/caldrawer.py')
-rw-r--r--tropical/qt/caldrawer.py192
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)
+