FWTEveViewer Class Reference

#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="")
	FWTEveViewer (const FWTEveViewer &)=delete
const FWTEveViewer &	operator= (const FWTEveViewer &)=delete
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
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.

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

~FWTEveViewer()

FWTEveViewer::~FWTEveViewer ( )

override

Definition at line 45 of file FWTEveViewer.cc.

References m_cnd, m_moo, m_thr, and m_thr_exit.

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

FWTEveViewer() [2/2]

FWTEveViewer::FWTEveViewer ( const FWTEveViewer &  )

delete

Member Function Documentation

CaptureAndSaveImage()

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

Definition at line 125 of file FWTEveViewer.cc.

References geometryDiff::file, ALPAKA_ACCELERATOR_NAMESPACE::caPixelDoublets::hh, m_cnd, m_fwGlViewer, m_hh, m_imgBuffer, m_moo, m_name, m_prom, m_thr, m_ww, FWTGLViewer::MakeFbo(), FWTGLViewer::MakeFboHeight(), spawn_image_thread(), and ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::ww.

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

fwGlViewer()

FWTGLViewer* FWTEveViewer::fwGlViewer ( )

inline

Definition at line 50 of file FWTEveViewer.h.

References m_fwGlViewer.

Referenced by FWEveView::fwViewerGL().

{ return m_fwGlViewer; }

operator=()

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

delete

SaveJpg()

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

static

Definition at line 270 of file FWTEveViewer.cc.

References geometryDiff::file, personalPlayback::fp, ALPAKA_ACCELERATOR_NAMESPACE::caPixelDoublets::hh, mps_fire::i, dqmiolumiharvest::j, mysort::rows, TRUE, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::ww, and geometryCSVtoXML::xx.

Referenced by spawn_image_thread().

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

SavePng()

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

static

Definition at line 185 of file FWTEveViewer.cc.

References geometryDiff::file, personalPlayback::fp, ALPAKA_ACCELERATOR_NAMESPACE::caPixelDoublets::hh, mps_fire::i, dqmiolumiharvest::j, mysort::rows, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::ww, and geometryCSVtoXML::xx.

Referenced by spawn_image_thread().

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

spawn_image_thread()

void FWTEveViewer::spawn_image_thread ( )

private

Definition at line 77 of file FWTEveViewer.cc.

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().

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

SpawnFWTGLViewer()

FWTGLViewer * FWTEveViewer::SpawnFWTGLViewer

(

)

Definition at line 109 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 == nullptr)
    PreUndock();

  return m_fwGlViewer;
}

Member Data Documentation

m_cnd

std::condition_variable FWTEveViewer::m_cnd

private

Definition at line 75 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), spawn_image_thread(), and ~FWTEveViewer().

m_fwGlViewer

FWTGLViewer* FWTEveViewer::m_fwGlViewer

private

Definition at line 65 of file FWTEveViewer.h.

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

m_hh

int FWTEveViewer::m_hh

private

Definition at line 70 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), and spawn_image_thread().

m_imgBuffer

std::vector<unsigned char> FWTEveViewer::m_imgBuffer

private

Definition at line 67 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), and spawn_image_thread().

m_moo

std::mutex FWTEveViewer::m_moo

private

Definition at line 74 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), spawn_image_thread(), and ~FWTEveViewer().

m_name

TString FWTEveViewer::m_name

private

Definition at line 69 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), and spawn_image_thread().

m_prom

std::promise<int> FWTEveViewer::m_prom

private

Definition at line 73 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), and spawn_image_thread().

m_thr

std::thread* FWTEveViewer::m_thr = nullptr

private

Definition at line 72 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), spawn_image_thread(), and ~FWTEveViewer().

m_thr_exit

bool FWTEveViewer::m_thr_exit = false

private

Definition at line 71 of file FWTEveViewer.h.

Referenced by spawn_image_thread(), and ~FWTEveViewer().

m_ww

int FWTEveViewer::m_ww

private

Definition at line 70 of file FWTEveViewer.h.

Referenced by CaptureAndSaveImage(), and spawn_image_thread().