FWTEveViewer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     Subsystem/Package
// Class  :     FWTEveViewer
//
// Implementation:
//     [Notes on implementation]
//
// Original Author:
//         Created:  Tue, 03 Feb 2015 21:46:04 GMT
//

// system include files

#include "png.h"
#include "jpeglib.h"

// user include files

#include "TMath.h"
#include "TGLIncludes.h"
#include "TGLFBO.h"

#include "Fireworks/Core/interface/FWTEveViewer.h"
#include "Fireworks/Core/interface/FWTGLViewer.h"

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
FWTEveViewer::FWTEveViewer(const char* n, const char* t) : TEveViewer(n, t), m_fwGlViewer(nullptr) {}

// FWTEveViewer::FWTEveViewer(const FWTEveViewer& rhs)
// {
//    // do actual copying here;
// }

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

//
// assignment operators
//
// const FWTEveViewer& FWTEveViewer::operator=(const FWTEveViewer& rhs)
// {
//   //An exception safe implementation is
//   FWTEveViewer temp(rhs);
//   swap(rhs);
//
//   return *this;
// }

//==============================================================================

//
// member functions
//

void FWTEveViewer::spawn_image_thread() {
  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);
}

//------------------------------------------------------------------------------

FWTGLViewer* FWTEveViewer::SpawnFWTGLViewer() {
  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;
}

std::future<int> FWTEveViewer::CaptureAndSaveImage(const TString& file, int height) {
  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();
}

//
// const member functions
//

//
// static member functions
//

bool FWTEveViewer::SavePng(const TString& file, UChar_t* xx, int ww, int hh) {
  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;
}

bool FWTEveViewer::SaveJpg(const TString& file, UChar_t* xx, int ww, int hh) {
  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;
}