fix(qt_display): fix fps unstable problem, fix the size of BMLabel

Change-Id: I51806b9672bb121e7c692b3446d6ebb61eb65ab6
sophgo · Nov 25, 2024 · a42be1b · a42be1b
1 parent 9ff32f1
commit a42be1b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/element/tools/qt_display/include/BMLabel.h b/element/tools/qt_display/include/BMLabel.h
@@ -25,8 +25,7 @@ namespace qt_display {
 class BMLabel : public QLabel {
   Q_OBJECT
  public:
-  explicit BMLabel(QWidget* parent = nullptr,
-                   Qt::WindowFlags f = Qt::WindowFlags());
+  explicit BMLabel(QWidget* parent, int width, int height);
   ~BMLabel();
 
   void show_img(std::shared_ptr<bm_image> bmimg_ptr);

diff --git a/element/tools/qt_display/src/BMLabel.cc b/element/tools/qt_display/src/BMLabel.cc
@@ -13,7 +13,8 @@ namespace sophon_stream {
 namespace element {
 namespace qt_display {
 
-BMLabel::BMLabel(QWidget* parent, Qt::WindowFlags f) {
+BMLabel::BMLabel(QWidget* parent, int width, int height) :QLabel(parent){
+  setFixedSize(width, height);
   connect(this, &BMLabel::show_signals, this, &BMLabel::show_pixmap);
 
 }

diff --git a/element/tools/qt_display/src/qt_display.cc b/element/tools/qt_display/src/qt_display.cc
@@ -35,16 +35,20 @@ int QtDisplay::qt_func() {
 
   qwidget_ptr->setGeometry(0, 0, screen_width, screen_height);
   layout = new QGridLayout(qwidget_ptr);
-  qwidget_ptr->setLayout(layout);
+
+  int label_width = screen_width/cols;
+  int label_height = screen_height/rows;
+
   for (int row = 0; row < rows; row++) {
     for (int col = 0; col < cols; col++) {
       std::shared_ptr<BMLabel> label_ptr =
-          std::make_shared<BMLabel>(qwidget_ptr);
+          std::make_shared<BMLabel>(qwidget_ptr, label_width, label_height);
       layout->addWidget(label_ptr.get(), row, col);
       label_vec.push_back(label_ptr);
     }
   }
 
+  qwidget_ptr->setLayout(layout);
   qwidget_ptr->show();
 
   return qapp->exec();