complete 2D cut, can add, can delete, can rename

Histogram1D.h

@@ -6,6 +6,7 @@
 //^==============================================
 //^==============================================
 class Histogram1D : public QCustomPlot{
+  Q_OBJECT
 public:
   Histogram1D(QString title, QString xLabel, int xbin, double xmin, double xmax, QWidget * parent = nullptr) : QCustomPlot(parent){
     Rebin(xbin, xmin, xmax);
@@ -78,8 +79,7 @@ public:
         QAction * a1 = menu.addAction("UnZoom");
         QAction * a2 = menu.addAction("Clear hist.");
         QAction * a3 = menu.addAction("Toggle Stat.");
-
+        QAction * a4 = menu.addAction("Rebin (clear histogram)");
-        //TODO rebin 
         //TODO fitGuass
 
         QAction *selectedAction = menu.exec(event->globalPosition().toPoint());
@@ -104,6 +104,64 @@ public:
           replot();
           usingMenu = false;
         }
+        if( selectedAction == a4 ){
+          QDialog dialog(this);
+          dialog.setWindowTitle("Rebin histogram");
+
+          QFormLayout layout(&dialog);
+
+          QLabel * info = new QLabel(&dialog);
+          info->setStyleSheet("color:red;");
+          info->setText("This will also clear histogram!!");
+          layout.addRow(info);
+
+          QStringList nameList = {"Num. Bin", "x-Min", "x-Max"};
+          QLineEdit* lineEdit[3];
+
+          for (int i = 0; i < 3; ++i) {
+              lineEdit[i] = new QLineEdit(&dialog);
+              layout.addRow(nameList[i] + " : ", lineEdit[i]);
+          }
+
+          QLabel * msg = new QLabel(&dialog);
+          msg->setStyleSheet("color:red;");
+          layout.addRow(msg);
+
+          QDialogButtonBox buttonBox(QDialogButtonBox::Ok | QDialogButtonBox::Cancel, Qt::Horizontal, &dialog);
+          layout.addRow(&buttonBox);
+
+          double number[3];
+
+          QObject::connect(&buttonBox, &QDialogButtonBox::accepted, [&]() {
+              int OKcount = 0;
+              bool conversionOk = true;
+              for( int i = 0; i < 3; i++ ){
+                number[i] = lineEdit[i]->text().toDouble(&conversionOk);
+                if( conversionOk ){
+                  OKcount++;
+                }else{
+                  msg->setText(nameList[i] + " is invalid.");
+                  return;
+                }
+              }
+
+              if( OKcount == 3 ) {
+                if( number[2] > number[1] ) {
+                  dialog.accept();
+                }else{
+                  msg->setText(nameList[2] + " is smaller than " + nameList[1]);
+                }
+              }
+          });
+          QObject::connect(&buttonBox, &QDialogButtonBox::rejected, [&]() { dialog.reject();});
+
+          if( dialog.exec() == QDialog::Accepted ){
+            Rebin((int)number[0], number[1], number[2]);
+            emit ReBinned();
+            UpdatePlot();
+          }
+
+        }
       }
     });
 
@@ -187,6 +245,9 @@ public:
     }
   }
 
+signals:
+  void ReBinned(); //ONLY for right click rebin
+
 private:
   double xMin, xMax, dX;
   int xBin;

Histogram2D.h

@@ -3,18 +3,20 @@
 
 #include "qcustomplot.h"
 
-const QList<QColor> colorCycle = { QColor(Qt::red),
+const QList<QPair<QColor, QString>> colorCycle = { {QColor(Qt::red), "Red"},
-                                   QColor(Qt::blue), 
+                                                   {QColor(Qt::blue), "Blue"},
-                                   QColor(Qt::darkGreen), 
+                                                   {QColor(Qt::darkGreen), "Dark Geen"},
-                                   QColor(Qt::darkCyan), 
+                                                   {QColor(Qt::darkCyan), "Dark Cyan"}, 
-                                   QColor(Qt::darkYellow), 
+                                                   {QColor(Qt::darkYellow), "Drak Yellow"}, 
-                                   QColor(Qt::magenta), 
+                                                   {QColor(Qt::magenta),  "Magenta"},
-                                   QColor(Qt::darkMagenta),
+                                                   {QColor(Qt::darkMagenta), "Dark Magenta"},
-                                   QColor(Qt::gray)}; 
+                                                   {QColor(Qt::gray), "Gray"}}; 
+
 
 //^==============================================
 //^==============================================
 class Histogram2D : public QCustomPlot{
+  
 public:
   Histogram2D(QString title, QString xLabel, QString yLabel, int xbin, double xmin, double xmax, int ybin, double ymin, double ymax, QWidget * parent = nullptr) : QCustomPlot(parent){
     xMin = xmin;
@@ -44,9 +46,10 @@ public:
 
     QCPColorGradient color;
     color.clearColorStops();
-    color.setColorStopAt( 0,  QColor("white" ));
+    color.setColorStopAt( 0.0, QColor("white" ));
-    color.setColorStopAt( 0.5,  QColor("blue"));
+    color.setColorStopAt( 0.3, QColor("blue"));
-    color.setColorStopAt( 1,  QColor("yellow"));
+    color.setColorStopAt( 0.4, QColor("green"));
+    color.setColorStopAt( 1.0, QColor("yellow"));
     colorMap->setGradient(color);
 
     cutList.clear();
@@ -76,11 +79,18 @@ public:
       }
     }
 
-
     rescaleAxes();
 
     usingMenu = false;
     isDrawCut = false;
+    tempCutID = -1;
+    numCut = 0;
+    lastPlottableID = -1;
+
+    line = new QCPItemLine(this);
+    line->setPen(QPen(Qt::gray, 1, Qt::DashLine));
+    line->setVisible(false);
+
 
     connect(this, &QCustomPlot::mouseMove, this, [=](QMouseEvent *event){
       double x = xAxis->pixelToCoord(event->pos().x());
@@ -92,8 +102,12 @@ public:
       QString coordinates = QString("X: %1, Y: %2, Z: %3").arg(x).arg(y).arg(z);
       QToolTip::showText(event->globalPosition().toPoint(), coordinates, this);
 
-      //TODO, when drawing cut, show dashhed line
+      //when drawing cut, show dashhed line
-
+      if( isDrawCut && tempCut.size() > 0  ){ 
+        line->end->setCoords(x,y); 
+        line->setVisible(true); 
+        replot();
+      }
 
     });
 
@@ -103,14 +117,19 @@ public:
       }
       if (event->button() == Qt::LeftButton && isDrawCut){
 
-        double x = xAxis->pixelToCoord(event->pos().x());
+        oldMouseX = xAxis->pixelToCoord(event->pos().x());
-        double y = yAxis->pixelToCoord(event->pos().y());
+        oldMouseY = yAxis->pixelToCoord(event->pos().y());
 
-        tempCut.push_back(QPointF(x,y));
+        tempCut.push_back(QPointF(oldMouseX,oldMouseY));
+
+        line->start->setCoords(oldMouseX, oldMouseY);
+        line->end->setCoords(oldMouseX, oldMouseY);
+        line->setVisible(true);
 
         DrawCut();
       }
 
+      //^================= right click
       if (event->button() == Qt::RightButton) {
         usingMenu = true;
         setSelectionRectMode(QCP::SelectionRectMode::srmNone);
@@ -122,8 +141,15 @@ public:
         QAction * a2 = menu->addAction("Clear hist.");
         QAction * a3 = menu->addAction("Toggle Stat.");
         QAction * a4 = menu->addAction("Create a Cut");
+        if( numCut > 0 ) {
+          menu->addSeparator();
+          menu->addAction("Add/Edit names to Cuts");
+          menu->addAction("Clear all Cuts");
+        }
         for( int i = 0; i < cutList.size(); i++){
-          menu->addAction("Delete Cut-" + QString::number(i));
+          if( cutList[i].isEmpty()) continue;
+          QString haha = "";
+          menu->addAction("Delete " + cutNameList[i] + " ["+ colorCycle[cutIDList[i]%colorCycle.count()].second+"]");
         }
 
         QAction *selectedAction = menu->exec(event->globalPosition().toPoint());
@@ -155,23 +181,133 @@ public:
 
         if( selectedAction == a4 ){
           tempCut.clear();
+          tempCutID ++;
           isDrawCut= true;
           usingMenu = false;
+          numCut ++;
+          qDebug() << "#### Create Cut Plottable count : " <<  plottableCount()  << ", numCut :" << numCut << ", " << lastPlottableID; 
         }
 
-        // if( selectedAction->text().contains("Delete Cut") ){
+        if( selectedAction && selectedAction->text().contains("Delete ") ){
-        //   qDebug() << "delete Cut";
+          QString haha = selectedAction->text();
-        // }
+          int index1 = haha.indexOf("-");
+          int index2 = haha.indexOf("[");
+          int cutID = haha.mid(index1+1, index2-index1-1).remove(' ').toInt();
           
-      }
+          removeItem(cutTextIDList[cutID]);
+          removePlottable(plottableIDList[cutID]);
+          replot();
+
+          numCut --;
+          cutList[cutID].clear();
+          cutIDList[cutID] = -1;
+          cutTextIDList[cutID] = -1;
+          plottableIDList[cutID] = -1;
+          cutNameList[cutID] = "";
+
+          for( int i = cutID + 1; i < cutTextIDList.count() ; i++){
+              cutTextIDList[i] --;
+              plottableIDList[i] --;
+          }
+
+          if( numCut == 0 ){
+            cutList.clear();
+            cutIDList.clear();
+            cutTextIDList.clear();
+            plottableIDList.clear();
+            cutNameList.clear();
+          }
+
+          qDebug() << "================= delete Cut-" << cutID;
+          qDebug() << "      cutIDList " << cutIDList ;
+          qDebug() << "plottableIDList " << plottableIDList << ", " << plottableCount();
+          qDebug() << "  cutTextIDList "  << cutTextIDList << ", " << itemCount();
+
+        }
+
+        if( selectedAction && selectedAction->text().contains("Clear all Cuts") ){
+          numCut = 0;
+          tempCutID = -1;
+          lastPlottableID = -1;
+          cutList.clear();
+          cutIDList.clear();
+          for( int i = cutTextIDList.count() - 1; i >= 0 ; i--){
+            if( cutTextIDList[i] < 0 ) continue;
+            removeItem(cutTextIDList[i]);
+            removePlottable(plottableIDList[i]);
+          }
+          replot();
+
+          cutTextIDList.clear();
+          plottableIDList.clear();
+          cutNameList.clear();
+
+        }
+
+        if( selectedAction && selectedAction->text().contains("Add/Edit names to Cuts") ){
+          
+          QDialog dialog(this);
+          dialog.setWindowTitle("Add/Edit name of cuts ");
+
+          QFormLayout layout(&dialog);
+
+          for(int i = 0; i < cutTextIDList.count(); i++){
+            if( cutTextIDList[i] < 0 ) continue;
+            QLineEdit * le = new QLineEdit(&dialog);
+            layout.addRow(colorCycle[i%colorCycle.count()].second, le);
+            le->setText( cutNameList[i] );
+            connect(le, &QLineEdit::textChanged, this, [=](){
+              le->setStyleSheet("color : blue;");
+            });
+            connect(le, &QLineEdit::returnPressed, this, [=](){
+              le->setStyleSheet("");
+              cutNameList[i] = le->text();
+              ((QCPItemText *) this->item(cutTextIDList[i]))->setText(le->text());
+              replot();
+            });
+          }
+
+          // QDialogButtonBox buttonBox(QDialogButtonBox::Cancel, Qt::Horizontal, &dialog);
+          // layout.addRow(&buttonBox);
+
+          // QObject::connect(&buttonBox, &QDialogButtonBox::rejected, [&]() { dialog.reject();});
+
+          dialog.exec();
+        }
+
+      }///================= end of right click
     });
 
     //connect( this, &QCustomPlot::mouseDoubleClick, this, [=](QMouseEvent *event){
     connect( this, &QCustomPlot::mouseDoubleClick, this, [=](){
-      isDrawCut = false;
+      if( isDrawCut) {
-      tempCut.push_back(tempCut[0]);
+        tempCut.push_back(tempCut[0]);
-      cutList.push_back(tempCut);
+        DrawCut();
-      DrawCut();
+        isDrawCut = false;
+        line->setVisible(false);
+
+        plottableIDList.push_back(plottableCount() -1 );
+
+        cutNameList.push_back("Cut-" + QString::number(cutList.count()));
+        //QCPItemText is not a plottable
+        QCPItemText * text = new QCPItemText(this);
+        text->setText(cutNameList.last());
+        text->position->setCoords(tempCut[0].rx(), tempCut[0].ry());
+        int colorID = tempCutID% colorCycle.count();
+        text->setColor(colorCycle[colorID].first);
+        cutTextIDList.push_back(itemCount() - 1);
+
+        replot();
+
+        cutList.push_back(tempCut);
+        cutIDList.push_back(tempCutID);
+
+        qDebug() << "----------- end of create cut";
+        qDebug() << "      cutIDList " << cutIDList ;
+        qDebug() << "plottableIDList " << plottableIDList << ", " << plottableCount();
+        qDebug() << "  cutTextIDList "  << cutTextIDList << ", " << itemCount();
+
+      }
     });
 
     connect(this, &QCustomPlot::mouseRelease, this, [=](){
@@ -219,16 +355,18 @@ public:
 
   void DrawCut(){
 
-    //TODO how to delete cut from plot and from QList
+    //The histogram is the 1st plottable.
+    // the lastPlottableID should be numCut+ 1
+    if( lastPlottableID != numCut ){
+      removePlottable(lastPlottableID);
+    }
     
-    plottableCount();
+    if(tempCut.size() > 0) {
-
-    //TODO how to not double plot?
-
-    if(tempCut.size() > 1) {
         QCPCurve *polygon = new QCPCurve(xAxis, yAxis);
+        lastPlottableID = plottableCount() - 1;
 
-        QPen pen(Qt::blue);
+        int colorID = tempCutID% colorCycle.count();
+        QPen pen(colorCycle[colorID].first);
         pen.setWidth(1);
         polygon->setPen(pen);
 
@@ -238,11 +376,12 @@ public:
         }
         polygon->data()->set(dataPoints, false);
     }
-
     replot();
+
+    qDebug() << "Plottable count : " <<  plottableCount() << ", cutList.count :" << cutList.count() << ", cutID :" << lastPlottableID;
   }
 
-  QList<QPolygonF> GetCutList() const{return cutList;}
+  QList<QPolygonF> GetCutList() const{return cutList;} // this list may contain empty element
 
 private:
   double xMin, xMax, yMin, yMax;
@@ -259,8 +398,18 @@ private:
   QCPItemText * txt[3][3];
   
   QPolygonF tempCut;
+  int tempCutID; // only incresing;
   QList<QPolygonF> cutList;
+  QList<int> cutIDList;
+  int numCut;
   bool isDrawCut;
+  int lastPlottableID;
+  QList<int> cutTextIDList;
+  QList<int> plottableIDList;
+  QList<QString> cutNameList;
+
+  QCPItemLine * line;
+  double oldMouseX = 0.0, oldMouseY = 0.0;
 
 };
 

SingleSpectra.cpp

@@ -121,6 +121,13 @@ SingleSpectra::SingleSpectra(Digitizer ** digi, unsigned int nDigi, QString rawD
       for( int j = 0; j < digi[i]->GetNChannels(); j++){
         if( i < nDigi ) {
           hist[i][j] = new Histogram1D("Digi-" + QString::number(digi[i]->GetSerialNumber()) +", Ch-" +  QString::number(j), "Raw Energy [ch]", nBin, xMin, xMax);
+          connect(hist[i][j], &Histogram1D::ReBinned , this, [=](){
+            enableSignalSlot = false;
+            sbNBin->setValue(hist[i][j]->GetNBin());
+            sbXMin->setValue(hist[i][j]->GetXMin());
+            sbXMax->setValue(hist[i][j]->GetXMax());
+            enableSignalSlot = true;
+          });
         }else{
           hist[i][j] = nullptr;
         }