bm12controller.cpp

Go to the documentation of this file.

#include "bm12controller.h"
#include <qwt_plot.h>
#include <qwt_plot_curve.h>
#include <qwt_symbol.h>
#include "mcacontroller.h"
#include <QFileDialog>
#include <QTextStream>

#include <stdio.h>

BM12Controller::BM12Controller(MCAController *ctrl, QwtPlot *graph, QObject *parent)
  : QObject(parent),
    m_NChannels(13),
    m_MCA("bm12det:aim_adc1"),
    m_ADC("bm12det:adc1"),
    m_HVPS("bm12det:hvps1"),
    m_Controller(ctrl),
    m_Graph(graph),
    m_CurrentChannel(0),
    m_CurrentWindow(0),
    m_SaveDataPath(QDir::currentPath()),
    m_SaveWindowsPath(QDir::currentPath())
{
//   printf("BM12Controller::BM12Controller\n");

  for (int i=0; i<m_NChannels; i++) {
    m_TCA.append(new EpicsPulseAnalyser(QString("bm12det:tca%1").arg(i+1)));
    m_Waveforms.append(new BM12Waveforms());
  }

  connect(&m_MCA, SIGNAL(valueChanged()), this, SLOT(mcaValueChanged()));

  m_ADC.setGain(2048);
  m_ADC.setRange(2048);
  m_ADC.setOffset(0);
  m_ADC.setLowerLevelDiscrim(5);
  m_ADC.setUpperLevelDiscrim(100);

  m_HVPS.setVoltage(500);
  m_HVPS.setLimit(500);
}

BM12Controller::~BM12Controller()
{
  while (!m_TCA.isEmpty()) {
    delete m_TCA.takeFirst();
  }

  while (!m_Waveforms.isEmpty()) {
    delete m_Waveforms.takeFirst();
  }
}

EpicsMCA* BM12Controller::mca()
{
  return &m_MCA;
}

EpicsA2DConverter* BM12Controller::a2d()
{
  return &m_ADC;
}

EpicsPulseAnalyser* BM12Controller::pulseAnalyser(int chan)
{
  return m_TCA.value(chan);
}

EpicsPowerSupply* BM12Controller::powerSupply()
{
  return &m_HVPS;
}

void BM12Controller::selectSCA(int ch, int win)
{
//   printf("BM12Controller::selectSCA(%d,%d)\n", ch,win);

//   pulseAnalyser(m_CurrentChannel)->setSelected(false);
//   pulseAnalyser(m_CurrentChannel)->setGated(m_CurrentWindow, false);

  m_CurrentChannel = ch;
  m_CurrentWindow  = win;

//   pulseAnalyser(m_CurrentChannel)->setSelected(true);
//   pulseAnalyser(m_CurrentChannel)->setGated(m_CurrentWindow, true);

  for (int i=0; i<m_NChannels; i++) {
    EpicsPulseAnalyser *p = pulseAnalyser(i);

    p->setSelected(i==ch);

    for (int j=0; j<p->channelCount(); j++) {
      p->setGated(j, j==(win-1));
    }
  }

  if (win == 0) {
    m_ADC.setCoincidence(1);
  } else {
    m_ADC.setCoincidence(0);
  }

  pulseAnalyser(0)->settle(0.25);

  updateGraph();
}

void BM12Controller::clear(int ch, int win)
{
  if (ch >= 0 && ch < 13) {
    BM12Waveforms *wf = m_Waveforms[ch];

    if (wf) {
      switch (win) {
      case 0:
        m_MCA.erase();
        wf -> YData().clear();
        break;

      case 1:
        wf -> YDataSaved1().clear();
        break;

      case 2:
        wf -> YDataSaved2().clear();
        break;

      case 3:
        wf -> YDataSaved3().clear();
        break;
      }
    }

    updateGraph();
  }
}

void BM12Controller::save(int ch, int win)
{
 //  printf("BM12Controller::save(%d,%d)\n", ch,win);

//   m_CurrentChannel = ch;
//   m_CurrentWindow  = win;

  if (ch >= 0 && ch < 13) {
    BM12Waveforms *wf = m_Waveforms[ch];

    if (wf) {
      switch (win) {
      case 1:
        wf -> YDataSaved1() = wf -> YData();
        break;

      case 2:
        wf -> YDataSaved2() = wf -> YData();
        break;

      case 3:
        wf -> YDataSaved3() = wf -> YData();
        break;
      }
    }

    updateGraph();
  }
}

void BM12Controller::count(int ch, int win)
{
//   printf("BM12Controller::count(%d,%d)\n", ch,win);

  m_CurrentChannel = ch;
  m_CurrentWindow  = win;

  m_MCA.eraseAndStart();

  mcaValueChanged();
}

void BM12Controller::mcaValueChanged()
{
//   printf("BM12Controller::mcaValueChanged() %d, %d\n", m_CurrentChannel, m_CurrentWindow);

  if (m_CurrentChannel >= 0 && m_CurrentChannel < 13) {
    BM12Waveforms *wf = m_Waveforms[m_CurrentChannel];

    if (wf) {
      switch (m_CurrentWindow) {
        
      case 0:
        wf -> YData() = m_MCA.value();
        break;

      case 1:
        wf -> YDataSCA1() = m_MCA.value();
        break;

      case 2:
        wf -> YDataSCA2() = m_MCA.value();
        break;

      case 3:
        wf -> YDataSCA3() = m_MCA.value();
        break;

      default:
        break;
      }
    }
  }

  updateGraph();
}

double BM12Controller::calibratedValue(int bin)
{
  return bin*0.0538287-0.060349;
}

void BM12Controller::updateGraph()
{
  if (m_CurrentChannel >= 0 && m_CurrentChannel < 13) {
    BM12Waveforms *wf = m_Waveforms[m_CurrentChannel];

    if (wf) {
      int xlen = wf->XData().size();
      int ylen = wf->YData().size();

//       if (ylen != xlen) {
      wf->XData().resize(ylen);
      for (int i=0; i<ylen; i++) {
        wf->XData()[i] = calibratedValue(i);
      }
//       }

      m_Graph -> detachItems(QwtPlotItem::Rtti_PlotCurve);
      m_Graph -> detachItems(QwtPlotItem::Rtti_PlotMarker);

      QwtPlotCurve *pc = new QwtPlotCurve("Total");
      pc -> setSamples(wf->XData().data(), wf->YData().data(), ylen);
      QPen pen(Qt::black);
      pc -> setPen(pen);
      pc -> setSymbol(new QwtSymbol(QwtSymbol::Ellipse,QBrush(),pen,QSize(3,3)));
      pc -> attach(m_Graph);
      
      if (ylen == wf->YDataSCA1().size()) {
        QwtPlotCurve *pc1 = new QwtPlotCurve("SCA1");
        QPen pen(Qt::red);
        pc1 -> setSamples(wf->XData().data(), wf->YDataSCA1().data(), ylen);
        pc1 -> setPen(pen);
        pc1 -> setSymbol(new QwtSymbol(QwtSymbol::Ellipse,QBrush(),pen,QSize(3,3)));
        pc1 -> setBrush(QBrush(QColor(255,0,0,20)));
        pc1 -> attach(m_Graph);
      }

      if (ylen == wf->YDataSCA2().size()) {
        QwtPlotCurve *pc1 = new QwtPlotCurve("SCA2");
        QPen pen(Qt::green);
        pc1 -> setSamples(wf->XData().data(), wf->YDataSCA2().data(), ylen);
        pc1 -> setPen(pen);
        pc1 -> setSymbol(new QwtSymbol(QwtSymbol::Ellipse,QBrush(),pen,QSize(3,3)));
        pc1 -> setBrush(QBrush(QColor(0,255,0,20)));
        pc1 -> attach(m_Graph);
      }

      if (ylen == wf->YDataSCA3().size()) {
        QwtPlotCurve *pc1 = new QwtPlotCurve("SCA3");
        QPen pen(Qt::blue);
        pc1 -> setSamples(wf->XData().data(), wf->YDataSCA3().data(), ylen);
        pc1 -> setPen(pen);
        pc1 -> setSymbol(new QwtSymbol(QwtSymbol::Ellipse,QBrush(),pen,QSize(3,3)));
        pc1 -> setBrush(QBrush(QColor(0,0,255,20)));
        pc1 -> attach(m_Graph);
      }

      if (ylen == wf->YDataSaved1().size()) {
        QwtPlotCurve *pc1 = new QwtPlotCurve("Saved A");
        QPen pen(Qt::darkRed);
        pc1 -> setSamples(wf->XData().data(), wf->YDataSaved1().data(), ylen);
        pc1 -> setPen(pen);
        pc1 -> attach(m_Graph);
      }

      if (ylen == wf->YDataSaved2().size()) {
        QwtPlotCurve *pc1 = new QwtPlotCurve("Saved B");
        QPen pen(Qt::darkGreen);
        pc1 -> setSamples(wf->XData().data(), wf->YDataSaved2().data(), ylen);
        pc1 -> setPen(pen);
        pc1 -> attach(m_Graph);
      }

      if (ylen == wf->YDataSaved3().size()) {
        QwtPlotCurve *pc1 = new QwtPlotCurve("Saved C");
        QPen pen(Qt::darkBlue);
        pc1 -> setSamples(wf->XData().data(), wf->YDataSaved3().data(), ylen);
        pc1 -> setPen(pen);
        pc1 -> attach(m_Graph);
      }

      m_Controller -> replot();
    }
  }
}

static void max(int& ncol, int &max, int sz)
{
  if (sz) {
    ncol++;

    if (sz > max) {
      max = sz;
    }
  }
}

void BM12Controller::saveData(int ch)
{
  int maxsize = 0;
  int ncolumns = 0;

  BM12Waveforms *wf;

  foreach(wf, m_Waveforms) {
    max(ncolumns, maxsize, wf->YData().size());
    max(ncolumns, maxsize, wf->YDataSCA1().size());
    max(ncolumns, maxsize, wf->YDataSCA2().size());
    max(ncolumns, maxsize, wf->YDataSCA3().size());
    max(ncolumns, maxsize, wf->YDataSaved1().size());
    max(ncolumns, maxsize, wf->YDataSaved2().size());
    max(ncolumns, maxsize, wf->YDataSaved3().size());
  }

  QString name = QFileDialog::getSaveFileName(NULL, "Save Data...", m_SaveDataPath);

  if (name.length()) {
    m_SaveDataPath = name;
    
//     printf("save mca data in file %s\n", qPrintable(name));
//     printf("dirName %s\n", qPrintable(m_SaveDataPath));

    FILE *ofile = fopen(qPrintable(name),"a+");

    fprintf(ofile, "#S 1\n");
    fprintf(ofile, "#N %d\n", ncolumns+2);
    fprintf(ofile, "#L  point  percent");

    for (int i=0; i<m_Waveforms.size(); i++) {
      wf = m_Waveforms[i];
      if (wf->YData().size())      fprintf(ofile,"  ch%d", i+1);
      if (wf->YDataSCA1().size())  fprintf(ofile,"  ch%dsca1", i+1);
      if (wf->YDataSCA2().size())  fprintf(ofile,"  ch%dsca2", i+1);
      if (wf->YDataSCA3().size())  fprintf(ofile,"  ch%dsca3", i+1);
      if (wf->YDataSaved1().size()) fprintf(ofile, "  ch%dsav1", i+1);
      if (wf->YDataSaved2().size()) fprintf(ofile, "  ch%dsav2", i+1);
      if (wf->YDataSaved3().size()) fprintf(ofile, "  ch%dsav3", i+1);
    }
    fprintf(ofile, "\n");

    for (int row=0; row<maxsize; row++) {
      fprintf(ofile,"%d\t%g", row, calibratedValue(row));

      for (int i=0; i<m_Waveforms.size(); i++) {
        wf = m_Waveforms[i];
        if (wf->YData().size()>row)      fprintf(ofile,"\t%g", wf->YData()[row]);
        if (wf->YDataSCA1().size()>row)  fprintf(ofile,"\t%g", wf->YDataSCA1()[row]);
        if (wf->YDataSCA2().size()>row)  fprintf(ofile,"\t%g", wf->YDataSCA2()[row]);
        if (wf->YDataSCA3().size()>row)  fprintf(ofile,"\t%g", wf->YDataSCA3()[row]);
        if (wf->YDataSaved1().size()>row) fprintf(ofile, "\t%g", wf->YDataSaved1()[row]);
        if (wf->YDataSaved2().size()>row) fprintf(ofile, "\t%g", wf->YDataSaved2()[row]);
        if (wf->YDataSaved3().size()>row) fprintf(ofile, "\t%g", wf->YDataSaved3()[row]);
      }
      fprintf(ofile, "\n");
    }

    fclose(ofile);
  }
}

void BM12Controller::loadEnergyWindows(int ch)
{
  QString name = QFileDialog::getOpenFileName(NULL, "Load Energy Windows...", m_SaveWindowsPath);

  if (name.length()) {
    m_SaveWindowsPath = name;
    
//     printf("load windows from file %s\n", qPrintable(name));

    FILE *ifile = fopen(qPrintable(name),"r");
    QTextStream istr(ifile);
    QString line;
    QRegExp match("epics_put\\(\\s*\"bm12det:tca(\\d+)SCA(\\d+)(\\S+)\",\\s*(\\S+)\\s*\\)\\;");

    do {
      line = istr.readLine();

      if (match.exactMatch(line)) {
        int chan = match.cap(1).toInt()-1;
        int win  = match.cap(2).toInt()-1;
        int islow = match.cap(3)=="_LOW";
        int ishigh = match.cap(3)=="_HI";

        double val = match.cap(4).toDouble();
                        
        if (islow) {
//        printf("set chan %d, win %d _LOW = %g\n", chan, win, val);
          pulseAnalyser(chan)->setLowerLevel(win,val);
        } else if (ishigh) {
//        printf("set chan %d, win %d _HI = %g\n", chan, win, val);
          pulseAnalyser(chan)->setUpperLevel(win,val);
        }
      }

    } while (!line.isNull());
  }
}

void BM12Controller::saveEnergyWindows(int ch)
{
  QString cmd;
  EpicsPulseAnalyser *tca;

  foreach(tca, m_TCA) {
    for (int i=0; i<tca->channelCount(); i++) {
      EpicsDouble *lll = tca->lowerLevel(i);
      EpicsDouble *hll = tca->upperLevel(i);

      cmd += QString("epics_put( \"%1\", %2 );\n").arg(lll->pvName()).arg(lll->value());
      cmd += QString("epics_put( \"%1\", %2 );\n").arg(hll->pvName()).arg(hll->value());
    }
  }
  
//   printf("Energy Windows\n%s\n", qPrintable(cmd));

  QString name = QFileDialog::getSaveFileName(NULL, "Save Energy Windows...", m_SaveWindowsPath);

  if (name.length()) {
    m_SaveWindowsPath = name;

//     printf("save windows in file %s\n", qPrintable(name));

    FILE *ofile = fopen(qPrintable(name), "w");
    fputs(qPrintable(cmd), ofile);
    fclose(ofile);
  }
}