#include "FWTEveViewer.h"

Inheritance diagram for FWTEveViewer:

Public Member Functions
std::future< int >	CaptureAndSaveImage (const TString &file, int height=-1)

FWTGLViewer *	fwGlViewer ()

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

FWTGLViewer *	SpawnFWTGLViewer ()

	~FWTEveViewer () override

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 &)=delete

const FWTEveViewer &	operator= (const FWTEveViewer &)=delete

void	spawn_image_thread ()

Private Attributes
std::condition_variable	m_cnd

FWTGLViewer *	m_fwGlViewer

int	m_hh

std::vector< unsigned char >	m_imgBuffer

std::mutex	m_moo

TString	m_name

std::promise< int >	m_prom

std::thread *	m_thr = nullptr

bool	m_thr_exit = false

int	m_ww

Detailed Description

Description: [one line class summary]

Usage: <usage>

Definition at line 36 of file FWTEveViewer.h.

Constructor & Destructor Documentation

◆ FWTEveViewer() [1/2]

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

Definition at line 38 of file FWTEveViewer.cc.

38 : TEveViewer(n, t), m_fwGlViewer(nullptr) {}

◆ ~FWTEveViewer()

FWTEveViewer::~FWTEveViewer ( )

override

Definition at line 45 of file FWTEveViewer.cc.

                             {
   if (m_thr)
     m_thr->detach();
  
   {
     std::unique_lock<std::mutex> lk(m_moo);
  
     m_thr_exit = true;
     m_cnd.notify_one();
   }
  
   delete m_thr;
 }

References m_cnd, m_moo, m_thr, and m_thr_exit.

◆ FWTEveViewer() [2/2]

FWTEveViewer::FWTEveViewer ( const FWTEveViewer & )

privatedelete

Member Function Documentation

◆ CaptureAndSaveImage()

std::future< int > FWTEveViewer::CaptureAndSaveImage	(	const TString &	file,
		int	height = `-1`
	)

Definition at line 125 of file FWTEveViewer.cc.

                                                                                 {
   static const TString eh("FWTEveViewer::CaptureAndSaveImage");
  
   TGLFBO* fbo = nullptr;
   if (height == -1)
     fbo = m_fwGlViewer->MakeFbo();
   else
     fbo = m_fwGlViewer->MakeFboHeight(height);
  
   if (fbo == nullptr) {
     ::Error(eh, "Returned FBO is 0.");
     m_prom = std::promise<int>();
     m_prom.set_value(-1);
     return m_prom.get_future();
   }
  
   int ww, hh;
   if (fbo->GetIsRescaled()) {
     ww = TMath::Nint(fbo->GetW() * fbo->GetWScale());
     hh = TMath::Nint(fbo->GetH() * fbo->GetHScale());
   } else {
     ww = fbo->GetW();
     hh = fbo->GetH();
   }
  
   fbo->SetAsReadBuffer();
  
   size_t bufsize = 3 * ww * hh;
   if (bufsize != m_imgBuffer.size()) {
     m_imgBuffer.resize(bufsize);
   }
  
   glPixelStorei(GL_PACK_ALIGNMENT, 1);
   glReadPixels(0, 0, ww, hh, GL_RGB, GL_UNSIGNED_BYTE, &m_imgBuffer[0]);
  
   if (m_thr == nullptr)
     spawn_image_thread();
  
   {
     std::unique_lock<std::mutex> lk(m_moo);
  
     m_prom = std::promise<int>();
     m_name = file;
     m_ww = ww;
     m_hh = hh;
  
     m_cnd.notify_one();
   }
  
   return m_prom.get_future();
 }

References FrontierConditions_GlobalTag_cff::file, m_cnd, m_fwGlViewer, m_hh, m_imgBuffer, m_moo, m_name, m_prom, m_thr, m_ww, FWTGLViewer::MakeFbo(), FWTGLViewer::MakeFboHeight(), and spawn_image_thread().

◆ fwGlViewer()

FWTGLViewer* FWTEveViewer::fwGlViewer ( )

inline

Definition at line 50 of file FWTEveViewer.h.

50 { return m_fwGlViewer; }

References m_fwGlViewer.

Referenced by FWEveView::fwViewerGL().

◆ operator=()

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

privatedelete

◆ SaveJpg()

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

static

Definition at line 270 of file FWTEveViewer.cc.

                                                                            {
   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;
 }

References FrontierConditions_GlobalTag_cff::file, personalPlayback::fp, mps_fire::i, dqmiolumiharvest::j, postprocess-scan-build::rows, TRUE, and geometryCSVtoXML::xx.

Referenced by spawn_image_thread().

◆ SavePng()

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

static

Definition at line 185 of file FWTEveViewer.cc.

                                                                            {
   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, nullptr, nullptr, nullptr);
   if (png_ptr == nullptr) {
     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 == nullptr) {
     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;
 }

References FrontierConditions_GlobalTag_cff::file, personalPlayback::fp, mps_fire::i, dqmiolumiharvest::j, postprocess-scan-build::rows, and geometryCSVtoXML::xx.

Referenced by spawn_image_thread().

◆ spawn_image_thread()

void FWTEveViewer::spawn_image_thread ( )

private

Definition at line 77 of file FWTEveViewer.cc.

                                       {
   std::unique_lock<std::mutex> lko(m_moo);
  
   m_thr = new std::thread([=]() {
     {
       std::unique_lock<std::mutex> lk(m_moo);
       m_cnd.notify_one();
     }
     while (true) {
       {
         std::unique_lock<std::mutex> lk(m_moo);
         m_cnd.wait(lk);
  
         if (m_thr_exit) {
           return;
         }
       }
       if (m_name.EndsWith(".jpg")) {
         SaveJpg(m_name, &m_imgBuffer[0], m_ww, m_hh);
       } else {
         SavePng(m_name, &m_imgBuffer[0], m_ww, m_hh);
       }
  
       m_prom.set_value(0);
     }
   });
  
   m_cnd.wait(lko);
 }

References m_cnd, m_hh, m_imgBuffer, m_moo, m_name, m_prom, m_thr, m_thr_exit, m_ww, SaveJpg(), and SavePng().

Referenced by CaptureAndSaveImage().

◆ SpawnFWTGLViewer()

FWTGLViewer * FWTEveViewer::SpawnFWTGLViewer ( )

Definition at line 109 of file FWTEveViewer.cc.

                                             {
   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 == nullptr)
     PreUndock();
  
   return m_fwGlViewer;
 }