Changeset 588 for trunk/geant4/visualization/OpenGL/src

Timestamp:

Sep 28, 2007, 5:16:30 PM (17 years ago)

Author:

garnier

Message:

r617@mac-90108: laurentgarnier | 2007-09-26 18:41:06 +0200
eps ok

File:

• trunk/geant4/visualization/OpenGL/src/G4OpenGLQtViewer.cc (modified) (7 diffs)

trunk/geant4/visualization/OpenGL/src/G4OpenGLQtViewer.cc

@@ -56 +56 @@
 #include <QMessageBox>
 #include <QFileDialog.h>
+#include <QPrinter.h>
+#include <QPainter.h>
 
 //////////////////////////////////////////////////////////////////////////////
@@ -664 +666 @@
     //    }
   }
-  filters += "eps";
+  filters += "eps;;";
+  filters += "ps;;";
+  filters += "pdf";
   QString* selectedFilter = new QString();
   QString nomFich =  QFileDialog::getSaveFileName ( GLWindow,
@@ -701 +705 @@
     } else if (nomFich.endsWith(".eps")) {
       generateEPS(nomFich,exportDialog->getNbColor(),image);
+    } else if (nomFich.endsWith(".ps") ||nomFich.endsWith(".pdf")) {
+      generatePS_PDF(nomFich,exportDialog->getNbColor(),image);
     } else if (nomFich.endsWith(".tif") ||
                nomFich.endsWith(".tiff") ||
@@ -722 +728 @@
 /*
 // http://www.google.com/codesearch?hl=en&q=+jpg+Qt+quality+QDialog+show:FZkUoth8oiw:TONpW2mR-_c:tyTfrKMO-xI&sa=N&cd=2&ct=rc&cs_p=http://soft.proindependent.com/src/qtiplot-0.8.9.zip&cs_f=qtiplot-0.8.9/qtiplot/src/application.cpp#a0
-void ApplicationWindow::exportLayer()
-{
-  QWidget *w=ws->activeWindow();
-  if (!w || !w->isA("MultiLayer"))
-    return;
-  
-  Graph* g = ((MultiLayer*)w)->activeGraph();
-  if (!g)
-    return;
-  
-  ImageExportDialog *ied = new ImageExportDialog(this, 0);
-  ied->setDir(workingDir);
-  if ( ied->exec() == QDialog::Accepted )
-    {
-      workingDir = ied->dirPath();
-      QString fname = ied->selectedFile();
-      QString selectedFilter = ied->selectedFilter();
-  
-      QFileInfo fi(fname);
-      QString baseName = fi.fileName();
-  
-      if (baseName.contains(".")==0)
-        fname.append(selectedFilter.remove("*"));
-  
-      if ( QFile::exists(fname) &&
-           QMessageBox::question(0, tr("QtiPlot - Overwrite File?"),
-                                 tr("A file called: <p><b>%1</b><p>already exists. "
-                                    "Do you want to overwrite it?")
-                                 .arg(fname),
-                                 tr("&Yes"), tr("&No"),
-                                 QString::null, 0, 1 ) )
-        return ;
-      else
-        {
-          QFile f(fname);
-          if ( !f.open( IO_WriteOnly ) )
-            {
-              QMessageBox::critical(0, tr("QtiPlot - Export Error"),
-                                    tr("Could not write to file: <br><h4> %1 </h4><p>Please verify that you have the right to write to this location!").arg(fname));
-              return;
-            }
-  
-          if (selectedFilter.contains(".eps"))
-            {
-              if (ied->showExportOptions())
-                {
-                  epsExportDialog *ed= new epsExportDialog (fname, this, "exportDialog", TRUE, WStyle_Tool|WDestructiveClose);
-                  connect (ed, SIGNAL(exportToEPS(const QString&, int, QPrinter::Orientation, QPrinter::PageSize, QPrinter::ColorMode)),
-                           g, SLOT(exportToEPS(const QString&, int, QPrinter::Orientation, QPrinter::PageSize, QPrinter::ColorMode)));
-  
-                  ed->showNormal();
-                  ed->setActiveWindow();
-                }
-              else
-                g->exportToEPS(fname);
-  
-              if (((MultiLayer*)w)->hasOverlapingLayers())
-                ((MultiLayer*)w)->updateTransparency();
-              return;
-            }
-          else if (selectedFilter.contains(".svg"))
-            {
-              g->exportToSVG(fname);
-              return;
-            }
-          //  else if (selectedFilter.contains(".wmf"))
-          //  {
-          //  g->exportToWmf(fname);
-          //  return;
-        }
-QStringList list=QImage::outputFormatList ();
-      for (int i=0; i<(int)list.count(); i++)
-        {
-          if (selectedFilter.contains("."+(list[i]).lower()))
-            {
-            if (ied->showExportOptions())
-              {
-              imageExportDialog* ed= new imageExportDialog(false, this,"exportDialog",TRUE,WStyle_Tool|WDestructiveClose);
-                  connect (ed, SIGNAL(options(const QString&, const QString&, int, bool)),
-                           g, SLOT(exportImage(const QString&, const QString&, int, bool)));
-  
-                  ed->setExportPath(fname, list[i]);
-                  ed->enableTransparency();
-                  ed->showNormal();
-                  ed->setActiveWindow();
-                }
-              else
-                g->exportImage(fname, list[i], 100, true);
-  
-              if (((MultiLayer*)w)->hasOverlapingLayers())
-                ((MultiLayer*)w)->updateTransparency();
-              return;
-            }
-        }
-    }
-}
-}
 
 void Graph::exportToSVG(const QString& fname)
@@ -952 +861 @@
    @param aImage : Image to print
 */
-int G4OpenGLQtViewer::generateEPS (
+bool G4OpenGLQtViewer::generateEPS (
  QString aFilename
 ,int aInColor
@@ -965 +874 @@
 
   const uchar * pixels = aImage.bits ();
-    //  pixels = grabPixels (inColour, width, height);
     
   if (pixels == NULL)
-    return 1;
+    return false;
   
   fp = fopen (aFilename.toStdString().c_str(), "w");
   if (fp == NULL) {
-    return 2;
+    return false;
   }
   
@@ -1006 +914 @@
   fprintf (fp, "{currentfile picstr readhexstring pop}\n");
   fprintf (fp, "false %d\n", aInColor);
-  //  fprintf (fp, "%%BeginData:       %d ASCII Bytes\n",size);
   fprintf (fp, "colorimage\n");
   
-  printf("boucle jusque :%d/%d par ligne:%d \n",aImage.width()*aImage.height()*aInColor,aImage.numBytes(),aImage.bytesPerLine ());
-
-  QRgb color;
-  if (aInColor == 3) {  // for RGB images
-    for (int j=aImage.height()-1;j>=0;j--){
-      for (int i=0;i<aImage.width();i++){
-        color = aImage.pixel (i,j);
-        fprintf (fp, " %02hx %02hx %02hx",qBlue(color),qGreen(color),qRed(color));
+
+  int width = aImage.width();
+  int height = aImage.height();
+  int depth = aImage.depth();
+  int size = width*height;
+  
+  if (depth == 1)
+    size = (width+7)/8*height;
+  else if (aInColor == 1)
+    size = size*3;
+  
+  int i = 0;
+  if (depth == 1) {
+    //  To be implemented
+    //    QImage::Endian bitOrder = aImage.bitOrder();
+    /*    for(int y=0; y < height; y++) {
+      const uchar * s = aImage.scanLine(y);
+      for(int x=0; x < width; x++) {
+        // need to copy bit for bit...
+        bool b = (bitOrder == QImage::LittleEndian) ?
+          (*(s + (x >> 3)) >> (x & 7)) & 1 :
+          (*(s + (x >> 3)) << (x & 7)) & 0x80 ;
+        if (b)
+          pixel[i >> 3] ^= (0x80 >> (i & 7));
+        i++;
       }
-      fprintf (fp, "\n");
-    }  
-  } else if (aInColor == 4) {
-    // for RGB images
-    for (int j=aImage.height()-1;j>=0;j--){
-      for (int i=0;i<aImage.width();i++){
-        color = aImage.pixel (i,j);
-        fprintf (fp, " %02hx %02hx %02hx %02hx",qAlpha(color), qRed(color),qGreen(color),qBlue(color));
+      // we need to align to 8 bit here
+      i = (i+7) & 0xffffff8;
+    }
+    */
+  } else if (depth == 8) {
+    printf("has 8 bit\n");
+    for(int y=height-1; y >=0 ; y--) {
+      const uchar * s = aImage.scanLine(y);
+      for(int x=0; x <width; x++) {
+        QRgb rgb = aImage.color(s[x]);
+        if (aInColor == 1) {
+          fprintf (fp, " %02hx ",(unsigned char)qGray(rgb));
+          i++;
+        } else {
+          fprintf (fp, " %02hx %02hx %02hx",
+                   (unsigned char) qRed(rgb),
+                   (unsigned char) qGreen(rgb),
+                   (unsigned char) qBlue(rgb));
+          i += 3;
+        }
       }
       fprintf (fp, "\n");
     }
-  }
-  else if (aInColor == 1) { // for grayscale
-  }
-  
+  } else {
+    bool alpha = aImage.hasAlphaChannel();
+    printf("has else %d alpha %d\n",depth,alpha);
+    for(int y=height-1; y >=0 ; y--) {
+      QRgb * s = (QRgb*)(aImage.scanLine(y));
+      for(int x=0; x <width; x++) {
+        QRgb rgb = (*s++);
+        if (alpha && qAlpha(rgb) < 0x40) // 25% alpha, convert to white -
+          rgb = qRgb(0xff, 0xff, 0xff);
+        if (aInColor == 1) {
+          fprintf (fp, " %02hx ",(unsigned char)qGray(rgb));
+          i++;
+        } else {
+          fprintf (fp, " %02hx %02hx %02hx",
+                   (unsigned char) qRed(rgb),
+                   (unsigned char) qGreen(rgb),
+                   (unsigned char) qBlue(rgb));
+          i += 3;
+        }
+      }
+      fprintf (fp, "\n");
+    } 
+  }
+
   fprintf (fp, "grestore\n");
   fprintf (fp, "showpage\n");
-  //  delete pixels;
   fclose (fp);
-  return 0;
-}
+
+  return true;
+}
+/**
+   Generate Postscript or PDF form image
+   @param aFilename : name of file
+   @param aInColor : numbers of colors : 1->BW 2->RGB
+   @param aImage : Image to print
+*/
+bool G4OpenGLQtViewer::generatePS_PDF (
+ QString aFilename
+,int aInColor
+,QImage aImage
+)
+{
+  QPrinter printer;
+  //  printer.setPageSize(pageSize);
+  if (aInColor == 1) {
+    printer.setColorMode(QPrinter::GrayScale);
+  } else {
+    printer.setColorMode(QPrinter::Color);
+  }
+
+  if (aFilename.endsWith(".ps")) {
+    printer.setOutputFormat(QPrinter::PostScriptFormat);
+  } else {
+    printer.setOutputFormat(QPrinter::PdfFormat);
+  }
+  printer.setOutputFileName(aFilename);
+  //  printer.setFullPage ( true);
+  QPainter paint(&printer);
+  paint.drawImage (0,0,aImage );
+  paint.end();
+  return true;
+}
+
 #endif
+
+/*
+
+void MultiLayer::exportToSVG(const QString& fname)
+{
+  QPicture picture;
+  QPainter p(&picture);
+  for (int i=0;i<(int)graphsList->count();i++)
+    {
+      Graph *gr=(Graph *)graphsList->at(i);
+      Plot *myPlot= (Plot *)gr->plotWidget();
+      
+      QPoint pos=gr->pos();
+      
+      int width=int(myPlot->frameGeometry().width());
+      int height=int(myPlot->frameGeometry().height());
+      
+      myPlot->print(&p, QRect(pos,QSize(width,height)));
+    }
+  
+  p.end();
+  picture.save(fname, "svg");
+}
+*/