/* ============================================================ 

DRS Data Display

Qt-based graphical user interface for displaying data taken with
the drsdaq program.
There are two operation modes: inspection of a raw data file and
data taking via the socket interface. Connecting to a socket server
automatically disables raw data file operation and closes any open
file.

Oliver Grimm

============================================================ */

#include "GUI.h"

ddd::ddd() {
  
  WaitForData = false;
  
  // Instantiate without console output
  RD = new RawDataCTX(true);
  PixMap = new PixelMap("../../config/PixelMap.txt", false);
   
  //---------------------------------------------------------------------
  //**************************** Main window ****************************
  //---------------------------------------------------------------------

  Central = new QWidget(this);
  setCentralWidget(Central);

  Socket = new QTcpSocket(Central);
  connect(Socket, SIGNAL(readyRead()), this, SLOT(ReadFromSocket()));   
  connect(Socket, SIGNAL(disconnected()), this, SLOT(GotDisconnected()));   
 
  // SpinBox for event number
  EventNo = new QSpinBox(Central);
  EventNo->setEnabled(false);
  connect(EventNo, SIGNAL(valueChanged(int)), this, SLOT(DisplayEvent(int)));
  EventNo->setToolTip("Event number in file");
  SpinLayout = new QFormLayout();
  SpinLayout->setRowWrapPolicy(QFormLayout::WrapAllRows);
  SpinLayout->addRow("E&vent #", EventNo);

  // SpinBox for channel number 
  ChannelNo = new QSpinBox(Central);
  ChannelNo->setEnabled(false);
  connect(ChannelNo, SIGNAL(valueChanged(int)), this, SLOT(DisplayEvent(int)));
  ChannelNo->setToolTip("Channels 0-19 for 2 DRS chips per mezzanine board");
  
  // SpinBox for board number 
  BoardNo = new QSpinBox(Central);
  BoardNo->setEnabled(false);
  connect(BoardNo, SIGNAL(valueChanged(int)), this, SLOT(DisplayEvent(int)));
  BoardNo->setToolTip("Mezzanine board number");
 
   // TextBox for pixel ID
  PixelID = new QLineEdit(Central);
  PixelID->setEnabled(false);
  connect(PixelID, SIGNAL(returnPressed()), this, SLOT(TranslatePixelID()));
  PixelID->setToolTip("Pixel identification");
 
  // Layout of pixel addressing widgets
  FormLayout = new QFormLayout;
  FormLayout->setRowWrapPolicy(QFormLayout::WrapAllRows);
  FormLayout->addRow("&Channel #", ChannelNo);
  FormLayout->addRow("&Board #", BoardNo);
  FormLayout->addRow("Pixel ID", PixelID);

  // Socket button
  SocketButton = new QPushButton("Socke&t\nInterface", Central);
  SocketButton->setFont(QFont("Times", 10, QFont::Bold));
  connect(SocketButton, SIGNAL(clicked()), this, SLOT(OpenSocketWindow()));
  SocketButton->setToolTip("Open window for socket communication");

  // Acquire button and Continuous check box
  GetButton = new QPushButton("&Acquire", Central);
  GetButton->setFont(QFont("Times", 10, QFont::Bold));
  GetButton->setEnabled(false);
  connect(GetButton, SIGNAL(clicked()), this, SLOT(GetSignalFromSocket()));
  GetButton->setToolTip("Acquire event over socket interface");
  ContinuousBox = new QCheckBox("Continuous", Central);
  ContinuousBox->setFont(QFont("Times", 10, QFont::Bold));
  ContinuousBox->setToolTip("Acquire continuously");

  // Plot area
  Graph = new QwtPlot(Central);
  Graph->setAxisTitle(QwtPlot::xBottom, "Time (ns)");
  Graph->setAxisTitle(QwtPlot::yLeft, "Signal (mV)");
  Graph->setAutoReplot(true);
  Graph->setCanvasBackground(QColor(Qt::yellow));
  Graph->setToolTip("Left mouse button to zoom (+Shift to pan)");
  Zoomer = new QwtPlotZoomer(QwtPlot::xBottom,QwtPlot::yLeft,Graph->canvas());
  Panner = new QwtPlotPanner(Graph->canvas());
  Panner->setMouseButton(Qt::LeftButton, Qt::ShiftModifier);
  Grid = new QwtPlotGrid;
  Grid->setMajPen(QPen(Qt::gray, 0, Qt::DotLine));
  Grid->attach(Graph);
  Signal = new QwtPlotCurve;
  Signal->attach(Graph);
  Signal->setStyle(QwtPlotCurve::Steps);
      
  // Text boxes for run and event header
  RunHeaderDisplay = new QPlainTextEdit(Central);
  EventHeaderDisplay = new QPlainTextEdit(Central);
  RunHeaderDisplay->setReadOnly(true);
  EventHeaderDisplay->setReadOnly(true);

  // Browse botton
  LoadButton = new QToolButton(Central);
  LoadButton->setToolButtonStyle (Qt::ToolButtonTextOnly);
  LoadButton->setText("...");
  connect(LoadButton, SIGNAL(clicked()), this, SLOT(FileDialog()));
  LoadButton->setToolTip("Open file dialog to select raw data file");

  // Filename box
  FilenameBox = new QLineEdit(Central);
  connect(FilenameBox, SIGNAL(returnPressed()), this, SLOT(OpenDatafile()));
  FilenameBox->setToolTip("Raw data file name");

  // Tab widget
  TabWidget = new QTabWidget(Central);
  TabWidget->addTab(Graph, "&Signal");
  TabWidget->addTab(RunHeaderDisplay, "&Run Header");
  TabWidget->addTab(EventHeaderDisplay, "&Event Header");
 
  // Layout of all widgets
  MainLayout = new QGridLayout(Central);
  MainLayout->addWidget(FilenameBox, 0, 0, 1, 3);
  MainLayout->addWidget(LoadButton, 0, 3);
  MainLayout->addLayout(SpinLayout, 1, 0);
  MainLayout->addLayout(FormLayout, 2, 0);
  MainLayout->addWidget(SocketButton, 6,0);
  MainLayout->addWidget(GetButton, 4,0);
  MainLayout->addWidget(ContinuousBox, 5,0);
  MainLayout->addWidget(TabWidget, 1, 1, 6, 5);
  MainLayout->setColumnStretch(1, 10);
  
  // Menu bar
  QMenu* Menu = menuBar()->addMenu("&Menu");
  OpenAction = Menu->addAction("Open data file", this, SLOT(FileDialog()));
  OpenAction->setShortcut(Qt::CTRL + Qt::Key_O);
  QAction* SaveAction = Menu->addAction("Save plot", this, SLOT(MenuSave()));
  SaveAction->setShortcut(Qt::CTRL + Qt::Key_S);
  Menu->addAction("Print plot", this, SLOT(MenuPrint()));
  Menu->addSeparator();
  ConnectAction = Menu->addAction("Connect", this, SLOT(MakeConnection()));
  ConnectAction->setShortcut(Qt::CTRL + Qt::Key_C);
  Menu->addSeparator();
  Menu->addAction("Help", this, SLOT(MenuHelp()));
  Menu->addAction("About", this, SLOT(MenuAbout()));
  Menu->addSeparator();
  QAction* QuitAction = Menu->addAction("Quit", qApp, SLOT(quit()));
  QuitAction->setShortcut(Qt::CTRL + Qt::Key_Q);
    
  //-----------------------------------------------------------------------
  //**************************** Socket window ****************************
  //-----------------------------------------------------------------------
  
  // Create widget (initially hidden)
  SocketWindow = new QWidget();
  SocketWindow->setWindowTitle("ddd - Socket Interface");
  
  // Edit box for IP Address  
  IPAddress = new QLineEdit(SocketWindow);
  IPAddress->setText("eth-vme02");
  IPAddress->setToolTip("Address of socket server");
  AddressLayout = new QFormLayout;
  AddressLayout->addRow("&Remote Address", IPAddress);
  
  // SpinBox for port selection
  Port = new QSpinBox(SocketWindow);
  Port->setRange(0,65535);
  Port->setValue(2000);
  Port->setToolTip("Port number of socket server");
  PortLayout = new QFormLayout;  
  PortLayout->addRow("&Port", Port);

  // Button to make connection
  Connect = new QPushButton("Connect", SocketWindow);
  Connect->setFont(QFont("Times", 10, QFont::Bold));
  connect(Connect, SIGNAL(clicked()), this, SLOT(MakeConnection()));
  Connect->setToolTip("Connect to server");
    
  // Edit box for command
  Command = new QLineEdit(SocketWindow);
  CommandLayout = new QFormLayout;  
  CommandLayout->addRow("&Command", Command);
  Command->setEnabled(false);
  connect(Command, SIGNAL(returnPressed()), this, SLOT(SendToSocket()));
  Command->setToolTip("Command to send to socket server");

  // Text box for socket output
  SocketOutput= new QPlainTextEdit(SocketWindow);
  SocketOutput->setReadOnly(true);
  SocketOutput->setMaximumBlockCount(MAX_OUTPUT_LINES);
  SocketOutput->setToolTip("Output of socket server");
  
  // Layout of all widgets
  SocketLayout = new QGridLayout(SocketWindow);
  SocketLayout->addLayout(AddressLayout, 0, 0, 0, 1);
  SocketLayout->addLayout(PortLayout, 0, 2);
  SocketLayout->addWidget(Connect, 0, 3);
  SocketLayout->addLayout(CommandLayout, 1, 0, 1, 4);
  SocketLayout->addWidget(SocketOutput, 2, 0, 4, 4);  
}

ddd::~ddd() {
  // Qwt items
  delete Grid;			delete Signal;
  // Layout items
  delete PortLayout;		delete CommandLayout;
  delete AddressLayout;		delete FormLayout;
  delete SpinLayout;
  // Other items 		
  delete SocketWindow;
  delete PixMap;		delete RD;
}


//---------------------------------------------------------------------
//**************************** Main program ***************************
//---------------------------------------------------------------------

int main(int argc, char *argv[]) {
  QApplication app(argc, argv);
  
  ddd MainWindow;
  MainWindow.setGeometry(100, 100, 800, 500);
  MainWindow.setWindowTitle("ddd - DRS Data Display");
  MainWindow.show();
  
  return app.exec();
}