FWTEveViewer Class Reference

#include "FWTEveViewer.h"

Inheritance diagram for FWTEveViewer:

Public Member Functions
std::thread	CaptureAndSaveImage (const TString &file, int height=-1)
FWTGLViewer *	fwGlViewer ()
	FWTEveViewer (const char *n="FWTEveViewer", const char *t="")
FWTGLViewer *	SpawnFWTGLViewer ()
virtual	~FWTEveViewer ()

Static Public Member Functions
static bool	SaveJpg (const TString &file, UChar_t *xx, int ww, int hh)
static bool	SavePng (const TString &file, UChar_t *xx, int ww, int hh)

Private Member Functions
	FWTEveViewer (const FWTEveViewer &)
const FWTEveViewer &	operator= (const FWTEveViewer &)

Private Attributes
FWTGLViewer *	m_fwGlViewer

Detailed Description

Description: [one line class summary]

Usage: <usage>

Definition at line 35 of file FWTEveViewer.h.

Constructor & Destructor Documentation

FWTEveViewer::FWTEveViewer	(	const char *	n = "FWTEveViewer",
		const char *	t = ""
	)

Definition at line 41 of file FWTEveViewer.cc.

                                                       :
   TEveViewer(n, t),
   m_fwGlViewer(0)
{}

FWTEveViewer::~FWTEveViewer ( )

virtual

Definition at line 51 of file FWTEveViewer.cc.

{}

FWTEveViewer::FWTEveViewer ( const FWTEveViewer &  )

private

Member Function Documentation

std::thread FWTEveViewer::CaptureAndSaveImage	(	const TString &	file,
		int	height = -1
	)

Definition at line 87 of file FWTEveViewer.cc.

References m_fwGlViewer, FWTGLViewer::MakeFbo(), FWTGLViewer::MakeFboHeight(), SaveJpg(), and SavePng().

{
   static const TString eh("FWTEveViewer::CaptureAndSaveImage");

   TGLFBO *fbo = 0;
   if (height == -1)
      fbo = m_fwGlViewer->MakeFbo();
   else
      fbo = m_fwGlViewer->MakeFboHeight(height);

   if (fbo == 0)
   {
      ::Error(eh, "Returned FBO is 0.");
      return std::thread();
   }

   int ww, hh;
   if (fbo->fIsRescaled)
   {
      ww = TMath::Nint(fbo->fW * fbo->fWScale);
      hh = TMath::Nint(fbo->fH * fbo->fHScale);
   }
   else
   {
      ww = fbo->fW;
      hh = fbo->fH;
   }

   fbo->SetAsReadBuffer();

   UChar_t* xx = new UChar_t[3 * ww * hh];
   glPixelStorei(GL_PACK_ALIGNMENT, 1);
   glReadPixels(0, 0, ww, hh, GL_RGB, GL_UNSIGNED_BYTE, xx);

   delete fbo;

   std::thread thr([=]()
   {
      if (file.EndsWith(".jpg"))
      {
         SaveJpg(file, xx, ww, hh);
      }
      else
      {
         SavePng(file, xx, ww, hh);
      }

      delete [] xx;
   });

   return thr;
}

FWTGLViewer* FWTEveViewer::fwGlViewer ( )

inline

Definition at line 51 of file FWTEveViewer.h.

References m_fwGlViewer.

Referenced by FWEveView::fwViewerGL().

{ return m_fwGlViewer; }

const FWTEveViewer& FWTEveViewer::operator= ( const FWTEveViewer &  )

private

bool FWTEveViewer::SaveJpg ( const TString &  file, UChar_t *  xx, int  ww, int  hh  )

static

Definition at line 228 of file FWTEveViewer.cc.

References i, j, tablePrinter::rows, and TRUE.

Referenced by CaptureAndSaveImage().

{
   struct   jpeg_compress_struct JpegInfo;
   struct   jpeg_error_mgr       Error;

   JpegInfo.err = jpeg_std_error(&Error);

   // Now we can initialize the JPEG compression object.
   jpeg_create_compress(&JpegInfo);

   FILE *fp = fopen(file, "w");
   jpeg_stdio_dest(&JpegInfo, fp);

   JpegInfo.image_width      = ww;
   JpegInfo.image_height     = hh;
   JpegInfo.input_components = 3;
   JpegInfo.in_color_space   = JCS_RGB;

   jpeg_set_defaults(&JpegInfo);

   JpegInfo.write_JFIF_header = TRUE;

   // Set the quality output
   // const int quality = 98;
   // jpeg_set_quality(&JpegInfo, quality, true); // bool force_baseline ????

   jpeg_start_compress(&JpegInfo, TRUE);

   std::vector<UChar_t*> rows(hh);
   {
      int j = hh - 1;
      for (int i = 0; i < hh; i++, j--) {
         rows[i] = xx + j * ww * 3;
      }
   }

   jpeg_write_scanlines(&JpegInfo, &rows[0], hh);

   // Step 6: Finish compression
   jpeg_finish_compress(&JpegInfo);

   // Step 7: release JPEG compression object

   // This is an important step since it will release a good deal of memory.
   jpeg_destroy_compress(&JpegInfo);

   return true;
}

bool FWTEveViewer::SavePng ( const TString &  file, UChar_t *  xx, int  ww, int  hh  )

static

Definition at line 148 of file FWTEveViewer.cc.

References i, j, NULL, and tablePrinter::rows.

Referenced by CaptureAndSaveImage().

{
   png_structp     png_ptr;
   png_infop       info_ptr;

   /* Create and initialize the png_struct with the desired error handler
    * functions.  If you want to use the default stderr and longjump method,
    * you can supply NULL for the last three parameters.  We also check that
    * the library version is compatible with the one used at compile time,
    * in case we are using dynamically linked libraries.  REQUIRED.
    */
   png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, 0, 0);
   if (png_ptr == NULL) {
      printf("Error creating png write struct\n");
      return false;
   }

   // Allocate/initialize the image information data.      REQUIRED
   info_ptr = png_create_info_struct(png_ptr);
   if (info_ptr == NULL) {
      printf("Error creating png info struct\n");
      png_destroy_write_struct(&png_ptr, &info_ptr);
      return false;
   }

   /*// Set error handling.  REQUIRED if you aren't supplying your own
   //      error handling functions in the png_create_write_struct() call.
   if (setjmp(png_jmpbuf(png_ptr))) {
   // If we get here, we had a problem reading the file
   png_destroy_write_struct(&png_ptr, &info_ptr);
   ilSetError(IL_LIB_PNG_ERROR);
   return IL_FALSE;
   }*/

   FILE *fp = fopen(file, "w");

   png_init_io(png_ptr, fp);


   // Use PNG_INTERLACE_ADAM7 for interlacing
   png_set_IHDR(png_ptr, info_ptr, ww, hh, 8, PNG_COLOR_TYPE_RGB,
                PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);

   /* Optional gamma chunk is strongly suggested if you have any guess
    * as to the correct gamma of the image.
    */
   // png_set_gAMA(png_ptr, info_ptr, gamma);

   // Optionally write comments into the image.
   // png_text text;
   // text.key  = "Generated by";
   // text.text = "Generated by cmsShow";
   // text.compression = PNG_TEXT_COMPRESSION_NONE;
   // png_set_text(png_ptr, info_ptr, &text, 1);

   // Write the file header information.  REQUIRED.
   png_write_info(png_ptr, info_ptr);

   std::vector<UChar_t*> rows(hh);
   {
      int j = hh - 1;
      for (int i = 0; i < hh; i++, j--) {
         rows[i] = xx + j * ww * 3;
      }
   }

   // Writes the image.
   png_write_image(png_ptr, &rows[0]);

   // It is REQUIRED to call this to finish writing the rest of the file
   png_write_end(png_ptr, info_ptr);

   // clean up after the write, and ifree any memory allocated
   png_destroy_write_struct(&png_ptr, &info_ptr);

   fclose(fp);

   return true;
}

FWTGLViewer * FWTEveViewer::SpawnFWTGLViewer

(

)

Definition at line 70 of file FWTEveViewer.cc.

References m_fwGlViewer.

Referenced by FWEveView::FWEveView().

{
   TGCompositeFrame* cf = GetGUICompositeFrame();

   m_fwGlViewer = new FWTGLViewer(cf);
   SetGLViewer(m_fwGlViewer, m_fwGlViewer->GetFrame());

   cf->AddFrame(fGLViewerFrame, new TGLayoutHints(kLHintsNormal | kLHintsExpandX | kLHintsExpandY));

   fGLViewerFrame->MapWindow();

   if (fEveFrame == 0)
      PreUndock();

   return m_fwGlViewer;
}

Member Data Documentation

FWTGLViewer* FWTEveViewer::m_fwGlViewer

private

Definition at line 64 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), fwGlViewer(), and SpawnFWTGLViewer().