开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 数值算法/人工智能 > 生物技术 > 查看源码
cn3d_png.cpp
资源名称：ncbi_cxx_win--Jul_08_2004.zip　[点击查看]
上传用户：yhdzpy8989
上传日期：2007-06-13
资源大小：13604k
文件大小：35k
源码类别：

生物技术

开发平台：

C/C++

cn3d_png.cpp：源码内容
  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: cn3d_png.cpp,v $
  4.  * PRODUCTION Revision 1000.3  2004/06/01 18:28:18  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.21
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  10. /*  $Id: cn3d_png.cpp,v 1000.3 2004/06/01 18:28:18 gouriano Exp $
  11. * ===========================================================================
  12. *
  13. *                            PUBLIC DOMAIN NOTICE
  14. *               National Center for Biotechnology Information
  15. *
  16. *  This software/database is a "United States Government Work" under the
  17. *  terms of the United States Copyright Act.  It was written as part of
  18. *  the author's official duties as a United States Government employee and
  19. *  thus cannot be copyrighted.  This software/database is freely available
  20. *  to the public for use. The National Library of Medicine and the U.S.
  21. *  Government have not placed any restriction on its use or reproduction.
  22. *
  23. *  Although all reasonable efforts have been taken to ensure the accuracy
  24. *  and reliability of the software and data, the NLM and the U.S.
  25. *  Government do not and cannot warrant the performance or results that
  26. *  may be obtained by using this software or data. The NLM and the U.S.
  27. *  Government disclaim all warranties, express or implied, including
  28. *  warranties of performance, merchantability or fitness for any particular
  29. *  purpose.
  30. *
  31. *  Please cite the author in any work or product based on this material.
  32. *
  33. * ===========================================================================
  34. *
  35. * Authors:  Paul Thiessen
  36. *
  37. * File Description:
  38. *      save structure window to PNG file
  39. *
  40. * ===========================================================================
  41. */
  43. #ifdef _MSC_VER
  44. #pragma warning(disable:4018)   // disable signed/unsigned mismatch warning in MSVC
  45. #endif
  47. #include <ncbi_pch.hpp>
  48. #include <corelib/ncbistd.hpp>
  49. #include <corelib/ncbi_limits.h>
  51. // need GL headers for off-screen rendering
  52. #if defined(__WXMSW__)
  53. #include <windows.h>
  54. #include <GL/gl.h>
  55. #include <GL/glu.h>
  57. #elif defined(__WXGTK__)
  58. #include <GL/gl.h>
  59. #include <GL/glu.h>
  60. #include <GL/glx.h>
  61. #include <gdk/gdkx.h>
  63. #elif defined(__WXMAC__)
  64. #include <OpenGL/gl.h>
  65. #include <OpenGL/glu.h>
  66. #endif
  68. #include <wx/platform.h>
  69. #if defined(__WXGTK__) && defined(__LINUX__)
  70. // use system headers/libs for linux builds
  71. #include <png.h>
  72. #else
  73. // otherwise, use libs built into wxWindows
  74. #include "../cn3d/png.h"
  75. #endif
  77. #include "cn3d_png.hpp"
  78. #include "cn3d_glcanvas.hpp"
  79. #include "opengl_renderer.hpp"
  80. #include "progress_meter.hpp"
  81. #include "cn3d_tools.hpp"
  82. #include "messenger.hpp"
  85. ////////////////////////////////////////////////////////////////////////////////////////////////
  86. // The following is taken unmodified from wxDesigner's C++ code from png_dialog.wdr
  87. ////////////////////////////////////////////////////////////////////////////////////////////////
  89. #include <wx/image.h>
  90. #include <wx/statline.h>
  91. #include <wx/spinbutt.h>
  92. #include <wx/spinctrl.h>
  93. #include <wx/splitter.h>
  94. #include <wx/listctrl.h>
  95. #include <wx/treectrl.h>
  96. #include <wx/notebook.h>
  97. #include <wx/grid.h>
  99. // Declare window functions
  101. #define ID_TEXT 10000
  102. #define ID_B_BROWSE 10001
  103. #define ID_T_NAME 10002
  104. #define ID_T_WIDTH 10003
  105. #define ID_T_HEIGHT 10004
  106. #define ID_C_ASPECT 10005
  107. #define ID_C_INTERLACE 10006
  108. #define ID_B_OK 10007
  109. #define ID_B_CANCEL 10008
  110. wxSizer *SetupPNGOptionsDialog( wxPanel *parent, bool call_fit = TRUE, bool set_sizer = TRUE );
  112. ////////////////////////////////////////////////////////////////////////////////////////////////
  114. // fix jmpbuf symbol problem on aix
  115. #if defined(_AIX43) && defined(jmpbuf)
  116. #undef jmpbuf
  117. #endif
  119. USING_NCBI_SCOPE;
  122. BEGIN_SCOPE(Cn3D)
  124. static ProgressMeter *progressMeter = NULL;
  125. static const int PROGRESS_RESOLUTION = 100, MAX_BUFFER_PIXELS = 1000000;
  126. static int nRows;
  128. class PNGOptionsDialog : private wxDialog
  129. {
  130. public:
  131.     PNGOptionsDialog(wxWindow *parent);
  132.     bool Activate(int initialWidth, int initialHeight, bool initialInterlaced);
  133.     bool GetValues(wxString *outputFilename, int *width, int *height, bool *interlaced);
  135. private:
  136.     double initialAspect;
  137.     bool dontProcessChange;
  139.     // event callbacks
  140.     void OnButton(wxCommandEvent& event);
  141.     void OnChangeSize(wxCommandEvent& event);
  142.     void OnCheckbox(wxCommandEvent& event);
  143.     void OnCloseWindow(wxCloseEvent& event);
  145.     DECLARE_EVENT_TABLE()
  146. };
  148. #define DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(var, id, type) 
  149.     type *var; 
  150.     var = wxDynamicCast(FindWindow(id), type); 
  151.     if (!var) { 
  152.         ERRORMSG("Can't find window with id " << id); 
  153.         return; 
  154.     }
  156. #define DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(var, id, type, errResult) 
  157.     type *var; 
  158.     var = wxDynamicCast(FindWindow(id), type); 
  159.     if (!var) { 
  160.         ERRORMSG("Can't find window with id " << id); 
  161.         return errResult; 
  162.     }
  164. // to make sure values are legitimate integers; use wxString::ToDouble to avoid parsing
  165. // strings as octal or hexadecimal...
  166. #define GET_AND_IS_VALID_SIZE(textctrl, var) 
  167.     (textctrl->GetValue().ToDouble(&var) && var >= 1 && fmod(var, 1.0) == 0.0 && var <= kMax_Int)
  169. BEGIN_EVENT_TABLE(PNGOptionsDialog, wxDialog)
  170.     EVT_BUTTON      (-1,    PNGOptionsDialog::OnButton)
  171.     EVT_TEXT        (-1,    PNGOptionsDialog::OnChangeSize)
  172.     EVT_CHECKBOX    (-1,    PNGOptionsDialog::OnCheckbox)
  173.     EVT_CLOSE       (       PNGOptionsDialog::OnCloseWindow)
  174. END_EVENT_TABLE()
  176. PNGOptionsDialog::PNGOptionsDialog(wxWindow *parent) :
  177.     wxDialog(parent, -1, "Export Options", wxPoint(50, 50), wxDefaultSize,
  178.         wxCAPTION | wxSYSTEM_MENU | wxFRAME_NO_TASKBAR), // not resizable
  179.     dontProcessChange(false)
  180. {
  181.     // construct the panel
  182.     wxPanel *panel = new wxPanel(this, -1);
  183.     wxSizer *topSizer = SetupPNGOptionsDialog(panel, false);
  185.     // call sizer stuff
  186.     topSizer->Fit(panel);
  187.     SetClientSize(topSizer->GetMinSize());
  188. }
  190. void PNGOptionsDialog::OnButton(wxCommandEvent& event)
  191. {
  192.     switch (event.GetId()) {
  193.         case ID_B_BROWSE: {
  194.             DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(tName, ID_T_NAME, wxTextCtrl)
  195.             wxString filename = wxFileSelector(
  196.                 "Choose a filename for output", GetUserDir().c_str(), "",
  197.                 ".png", "All Files|*.*|PNG files (*.png)|*.png",
  198.                 wxSAVE | wxOVERWRITE_PROMPT);
  199.             if (filename.size() > 0) tName->SetValue(filename);
  200.             break;
  201.         }
  202.         case ID_B_OK: {
  203.             wxString str;
  204.             int w, h;
  205.             bool i;
  206.             if (GetValues(&str, &w, &h, &i))
  207.                 EndModal(wxOK);
  208.             else
  209.                 wxBell();
  210.             break;
  211.         }
  212.         case ID_B_CANCEL:
  213.             EndModal(wxCANCEL);
  214.             break;
  215.     }
  216. }
  218. void PNGOptionsDialog::OnChangeSize(wxCommandEvent& event)
  219. {
  220.     // avoid recursive calls to this, since SetValue() triggers this event; only process size fields
  221.     if (dontProcessChange || !(event.GetId() == ID_T_WIDTH || event.GetId() == ID_T_HEIGHT)) return;
  223.     DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(tWidth, ID_T_WIDTH, wxTextCtrl)
  224.     DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(tHeight, ID_T_HEIGHT, wxTextCtrl)
  225.     DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(cAspect, ID_C_ASPECT, wxCheckBox)
  226.     DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(cInterlace, ID_C_INTERLACE, wxCheckBox)
  228.     // post refreshes for both, in case background color changes
  229.     tWidth->Refresh(true);
  230.     tHeight->Refresh(true);
  232.     double w, h;
  233.     if (!GET_AND_IS_VALID_SIZE(tWidth, w)) {
  234.         tWidth->SetBackgroundColour(*wxRED);
  235.         return;
  236.     }
  237.     tWidth->SetBackgroundColour(*wxWHITE);
  238.     if (!GET_AND_IS_VALID_SIZE(tHeight, h)) {
  239.         tHeight->SetBackgroundColour(*wxRED);
  240.         return;
  241.     }
  242.     tHeight->SetBackgroundColour(*wxWHITE);
  244.     // adjust to aspect ratio if indicated
  245.     dontProcessChange = true;
  246.     if (cAspect->GetValue()) {
  247.         wxString num;
  248.         if (event.GetId() == ID_T_WIDTH) {
  249.             h = floor(w / initialAspect + 0.5);
  250.             num.Printf("%i", (int) h);
  251.             tHeight->SetValue(num);
  252.         } else {
  253.             w = floor(h * initialAspect + 0.5);
  254.             num.Printf("%i", (int) w);
  255.             tWidth->SetValue(num);
  256.         }
  257.     }
  258.     dontProcessChange = false;
  260.     // only allow interlacing if image is small enough
  261.     if (w * h > MAX_BUFFER_PIXELS) {
  262.         if (cInterlace->IsEnabled()) {
  263.             cInterlace->Enable(false);
  264.             cInterlace->SetValue(false);
  265.         }
  266.     } else {
  267.         if (!cInterlace->IsEnabled()) {
  268.             cInterlace->Enable(true);
  269.             cInterlace->SetValue(true);
  270.         }
  271.     }
  272. }
  274. void PNGOptionsDialog::OnCheckbox(wxCommandEvent& event)
  275. {
  276.     if (event.GetId() == ID_C_ASPECT) {
  277.         // adjust height when aspect checkbox is turned on
  278.         DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(cAspect, ID_C_ASPECT, wxCheckBox)
  279.         if (cAspect->GetValue()) {
  281.             DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(tWidth, ID_T_WIDTH, wxTextCtrl)
  282.             DECLARE_AND_FIND_WINDOW_RETURN_ON_ERR(tHeight, ID_T_HEIGHT, wxTextCtrl)
  283.             double w, h;
  285.             if (GET_AND_IS_VALID_SIZE(tWidth, w)) {
  286.                 wxString num;
  287.                 h = floor(w / initialAspect + 0.5);
  288.                 num.Printf("%i", (int) h);
  289.                 tHeight->SetValue(num);
  290.             }
  291.         }
  292.     }
  293. }
  295. void PNGOptionsDialog::OnCloseWindow(wxCloseEvent& event)
  296. {
  297.     EndModal(wxCANCEL);
  298. }
  300. bool PNGOptionsDialog::Activate(int initialWidth, int initialHeight, bool initialInterlaced)
  301. {
  302.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(tName, ID_T_NAME, wxTextCtrl, false)
  303.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(tWidth, ID_T_WIDTH, wxTextCtrl, false)
  304.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(tHeight, ID_T_HEIGHT, wxTextCtrl, false)
  305.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(cAspect, ID_C_ASPECT, wxCheckBox, false)
  306.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(cInterlace, ID_C_INTERLACE, wxCheckBox, false)
  308.     dontProcessChange = true;
  309.     wxString num;
  310.     num.Printf("%i", initialWidth);
  311.     tWidth->SetValue(num);
  312.     num.Printf("%i", initialHeight);
  313.     tHeight->SetValue(num);
  314.     initialAspect = ((double) initialWidth) / initialHeight;
  315.     cAspect->SetValue(true);
  316.     dontProcessChange = false;
  318.     if (initialWidth*initialHeight > MAX_BUFFER_PIXELS) {
  319.         cInterlace->Enable(false);
  320.         cInterlace->SetValue(false);
  321.     } else
  322.         cInterlace->SetValue(initialInterlaced);
  324.     // bring up file selector first
  325.     wxCommandEvent browse(wxEVT_COMMAND_BUTTON_CLICKED, ID_B_BROWSE);
  326.     OnButton(browse);
  327.     if (tName->GetValue().size() == 0) return false;    // cancelled
  329.     // return dialog result
  330.     return (ShowModal() == wxOK);
  331. }
  333. bool PNGOptionsDialog::GetValues(wxString *outputFilename, int *width, int *height, bool *interlaced)
  334. {
  335.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(tName, ID_T_NAME, wxTextCtrl, false)
  336.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(tWidth, ID_T_WIDTH, wxTextCtrl, false)
  337.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(tHeight, ID_T_HEIGHT, wxTextCtrl, false)
  338.     DECLARE_AND_FIND_WINDOW_RETURN_RESULT_ON_ERR(cInterlace, ID_C_INTERLACE, wxCheckBox, false)
  340.     *outputFilename = tName->GetValue();
  341.     double val;
  342.     if (!GET_AND_IS_VALID_SIZE(tWidth, val)) return false;
  343.     *width = (int) val;
  344.     if (!GET_AND_IS_VALID_SIZE(tHeight, val)) return false;
  345.     *height = (int) val;
  346.     *interlaced = (cInterlace->IsEnabled() && cInterlace->GetValue());
  347.     return true;
  348. }
  351. static bool GetOutputParameters(wxString *outputFilename, int *width, int *height, bool *interlaced)
  352. {
  353.     PNGOptionsDialog dialog(NULL);
  354.     bool ok = dialog.Activate(*width, *height, *interlaced);
  355.     if (ok) dialog.GetValues(outputFilename, width, height, interlaced);
  356.     return ok;
  357. }
  359. // callback used by PNG library to report errors
  360. static void writepng_error_handler(png_structp png_ptr, png_const_charp msg)
  361. {
  362.     ERRORMSG("PNG library error: " << msg);
  363. }
  365. #ifdef __WXGTK__
  366. static bool gotAnXError;
  367. int X_error_handler(Display *dpy, XErrorEvent *error)
  368. {
  369.     gotAnXError = true;
  370.     return 0;
  371. }
  372. #endif
  374. // called after each row is written to the file
  375. static void write_row_callback(png_structp png_ptr, png_uint_32 row, int pass)
  376. {
  377.     if (!progressMeter) return;
  378.     int progress = 0;
  380.     if (nRows < 0) { /* if negative, then we're doing interlacing */
  381.         double start, end;
  382.         switch (pass + 1) {
  383.             case 1: start = 0;  end = 1;  break;
  384.             case 2: start = 1;  end = 2;  break;
  385.             case 3: start = 2;  end = 3;  break;
  386.             case 4: start = 3;  end = 5;  break;
  387.             case 5: start = 5;  end = 7;  break;
  388.             case 6: start = 7;  end = 11; break;
  389.             case 7: start = 11; end = 15; break;
  390.             default: return;    // png lib gives final pass=7,row=0 to signal completion
  391.         }
  392.         progress = (int) (1.0 * PROGRESS_RESOLUTION *
  393.             ((start / 15) + (((double) row) / (-nRows)) * ((end - start) / 15)));
  394.     } else { /* not interlaced */
  395.         progress = (int) (1.0 * PROGRESS_RESOLUTION * row / nRows);
  396.     }
  398.     progressMeter->SetValue(progress);
  399. }
  401. bool ExportPNG(Cn3DGLCanvas *glCanvas)
  402. {
  403. #if !defined(__WXMSW__) && !defined(__WXGTK__) && !defined(__WXMAC__)
  404.     ERRORMSG("PNG export not (yet) implemented on this platform");
  405.     return false;
  406. #endif
  408.     if (!glCanvas || !glCanvas->renderer) {
  409.         ERRORMSG("ExportPNG() - bad glCanvas parameter");
  410.         return false;
  411.     }
  413.     bool success = false, shareDisplayLists = true, interlaced = true;
  414.     int outputWidth, outputHeight, bufferHeight, bytesPerPixel = 3, nChunks = 1;
  415.     wxString filename;
  416.     FILE *out = NULL;
  417.     unsigned char *rowStorage = NULL;
  418.     png_structp png_ptr = NULL;
  419.     png_infop info_ptr = NULL;
  421.     outputWidth = glCanvas->GetClientSize().GetWidth(); // initial size
  422.     outputHeight = glCanvas->GetClientSize().GetHeight();
  423.     if (!GetOutputParameters(&filename, &outputWidth, &outputHeight, &interlaced))
  424.         return true; // cancelled
  426.     try {
  427.         INFOMSG("saving PNG file '" << filename.c_str() << "'");
  429.         // need to avoid any GL calls in glCanvas while off-screen rendering is happening; so
  430.         // temporarily prevent glCanvas from responding to window resize/exposure, etc.
  431.         glCanvas->SuspendRendering(true);
  433.         int windowViewport[4];
  434.         glCanvas->renderer->GetViewport(windowViewport);
  435.         TRACEMSG("window viewport: x,y: " << windowViewport[0] << ',' << windowViewport[1]
  436.             << " size: " << windowViewport[2] << ',' << windowViewport[3]);
  437.         INFOMSG("output size: " << outputWidth << ',' << outputHeight);
  439.         // decide whether the in-memory image buffer can fit the whole drawing,
  440.         // or whether we need to split it up into separate horizontal chunks
  441.         bufferHeight = outputHeight;
  442.         if (outputWidth * outputHeight > MAX_BUFFER_PIXELS) {
  443.             bufferHeight = MAX_BUFFER_PIXELS / outputWidth;
  444.             nChunks = outputHeight / bufferHeight;              // whole chunks +
  445.             if (outputHeight % bufferHeight != 0) ++nChunks;    // partially occupied chunk
  446.             interlaced = false;
  447.         }
  449.         // create and show progress meter
  450.         wxString message;
  451.         message.Printf("Writing PNG file %s (%ix%i)",
  452.             (wxString(wxFileNameFromPath(filename.c_str()))).c_str(),
  453.             outputWidth, outputHeight);
  454.         progressMeter = new ProgressMeter(NULL, message, "Saving...", PROGRESS_RESOLUTION);
  456.         // open the output file for writing
  457.         out = fopen(filename.c_str(), "wb");
  458.         if (!out) throw "can't open file for writing";
  460. #if defined(__WXMSW__)
  461.         HDC hdc = NULL, current_hdc = NULL;
  462.         HGLRC hglrc = NULL, current_hglrc = NULL;
  463.         HGDIOBJ current_hgdiobj = NULL;
  464.         HBITMAP hbm = NULL;
  465.         PIXELFORMATDESCRIPTOR pfd;
  466.         int nPixelFormat;
  468.         current_hglrc = wglGetCurrentContext(); // save to restore later
  469.         current_hdc = wglGetCurrentDC();
  471.         // create bitmap of same color type as current rendering context
  472.         hbm = CreateCompatibleBitmap(current_hdc, outputWidth, bufferHeight);
  473.         if (!hbm) throw "failed to create rendering BITMAP";
  474.         hdc = CreateCompatibleDC(current_hdc);
  475.         if (!hdc) throw "CreateCompatibleDC failed";
  476.         current_hgdiobj = SelectObject(hdc, hbm);
  477.         if (!current_hgdiobj) throw "SelectObject failed";
  479.         memset(&pfd, 0, sizeof(PIXELFORMATDESCRIPTOR));
  480.         pfd.nSize = sizeof(PIXELFORMATDESCRIPTOR);
  481.         pfd.nVersion = 1;
  482.         pfd.dwFlags = PFD_DRAW_TO_BITMAP | PFD_SUPPORT_OPENGL;  // NOT doublebuffered, to save memory
  483.         pfd.iPixelType = PFD_TYPE_RGBA;
  484.         pfd.cColorBits = GetDeviceCaps(current_hdc, BITSPIXEL);
  485.         pfd.iLayerType = PFD_MAIN_PLANE;
  486.         nPixelFormat = ChoosePixelFormat(hdc, &pfd);
  487.         if (!nPixelFormat) {
  488.             ERRORMSG("ChoosePixelFormat failed");
  489.             throw GetLastError();
  490.         }
  491.         if (!SetPixelFormat(hdc, nPixelFormat, &pfd)) {
  492.             ERRORMSG("SetPixelFormat failed");
  493.             throw GetLastError();
  494.         }
  495.         hglrc = wglCreateContext(hdc);
  496.         if (!hglrc) {
  497.             ERRORMSG("wglCreateContext failed");
  498.             throw GetLastError();
  499.         }
  500.         // try to share display lists with regular window, to save memory and draw time
  501.         if (!wglShareLists(current_hglrc, hglrc)) {
  502.             WARNINGMSG("wglShareLists failed: " << GetLastError());
  503.             shareDisplayLists = false;
  504.         }
  505.         if (!wglMakeCurrent(hdc, hglrc)) {
  506.             ERRORMSG("wglMakeCurrent failed");
  507.             throw GetLastError();
  508.         }
  510. #elif defined(__WXGTK__)
  511.         GLint glSize;
  512.         int nAttribs, attribs[20];
  513.         XVisualInfo *visinfo = NULL;
  514.         bool localVI = false;
  515.         Pixmap xPixmap = 0;
  516.         GLXContext currentCtx = NULL, glCtx = NULL;
  517.         GLXPixmap glxPixmap = 0;
  518.         GLXDrawable currentXdrw = 0;
  519.         Display *display;
  520.         int (*currentXErrHandler)(Display *, XErrorEvent *) = NULL;
  522.         currentCtx = glXGetCurrentContext(); // save current context info
  523.         currentXdrw = glXGetCurrentDrawable();
  524.         display = GDK_DISPLAY();
  526.         currentXErrHandler = XSetErrorHandler(X_error_handler);
  527.         gotAnXError = false;
  529.         // first, try to get a non-doublebuffered visual, to economize on memory
  530.         nAttribs = 0;
  531.         attribs[nAttribs++] = GLX_USE_GL;
  532.         attribs[nAttribs++] = GLX_RGBA;
  533.         attribs[nAttribs++] = GLX_RED_SIZE;
  534.         glGetIntegerv(GL_RED_BITS, &glSize);
  535.         attribs[nAttribs++] = glSize;
  536.         attribs[nAttribs++] = GLX_GREEN_SIZE;
  537.         attribs[nAttribs++] = glSize;
  538.         attribs[nAttribs++] = GLX_BLUE_SIZE;
  539.         attribs[nAttribs++] = glSize;
  540.         attribs[nAttribs++] = GLX_DEPTH_SIZE;
  541.         glGetIntegerv(GL_DEPTH_BITS, &glSize);
  542.         attribs[nAttribs++] = glSize;
  543.         attribs[nAttribs++] = None;
  544.         visinfo = glXChooseVisual(display, DefaultScreen(display), attribs);
  546.         // if that fails, just revert to the one used for the regular window
  547.         if (visinfo)
  548.             localVI = true;
  549.         else
  550.             visinfo = (XVisualInfo *) (glCanvas->m_vi);
  552.         // create pixmap
  553.         xPixmap = XCreatePixmap(display,
  554.             RootWindow(display, DefaultScreen(display)),
  555.             outputWidth, bufferHeight, visinfo->depth);
  556.         if (!xPixmap) throw "failed to create Pixmap";
  557.         glxPixmap = glXCreateGLXPixmap(display, visinfo, xPixmap);
  558.         if (!glxPixmap) throw "failed to create GLXPixmap";
  559.         if (gotAnXError) throw "Got an X error creating GLXPixmap";
  561.         // try to share display lists with "regular" context
  562. // ... seems to be too flaky - fails on Linux/Mesa, Solaris
  563. //        glCtx = glXCreateContext(display, visinfo, currentCtx, False);
  564. //        if (!glCtx || !glXMakeCurrent(display, glxPixmap, glCtx)) {
  565. //            WARNINGMSG("failed to make GLXPixmap rendering context with shared display lists");
  566. //            if (glCtx) glXDestroyContext(display, glCtx);
  567.             shareDisplayLists = false;
  569.             // try to create context without shared lists
  570.             glCtx = glXCreateContext(display, visinfo, NULL, False);
  571.             if (!glCtx || !glXMakeCurrent(display, glxPixmap, glCtx))
  572.                 throw "failed to make GLXPixmap rendering context without shared display lists";
  573. //        }
  574.         if (gotAnXError) throw "Got an X error setting GLX context";
  576. #elif defined(__WXMAC__)
  577.         unsigned char *base = NULL;
  578.         GLint attrib[20], glSize;
  579.         int na = 0;
  580.         AGLPixelFormat fmt = NULL;
  581.         AGLContext ctx = NULL, currentCtx;
  583.      currentCtx = aglGetCurrentContext();
  585.      // Mac pixels seem to always be 32-bit
  586.      bytesPerPixel = 4;
  588.      base = new unsigned char[outputWidth * bufferHeight * bytesPerPixel];
  589.      if (!base) throw "failed to allocate image buffer";
  591.      // create an off-screen rendering context (NOT doublebuffered)
  592.      attrib[na++] = AGL_OFFSCREEN;
  593.      attrib[na++] = AGL_RGBA;
  594.      glGetIntegerv(GL_RED_BITS, &glSize);
  595.         attrib[na++] = AGL_RED_SIZE;
  596.         attrib[na++] = glSize;
  597.         attrib[na++] = AGL_GREEN_SIZE;
  598.         attrib[na++] = glSize;
  599.         attrib[na++] = AGL_BLUE_SIZE;
  600.         attrib[na++] = glSize;
  601.      glGetIntegerv(GL_DEPTH_BITS, &glSize);
  602.         attrib[na++] = AGL_DEPTH_SIZE;
  603.         attrib[na++] = glSize;
  604.         attrib[na++] = AGL_NONE;
  606.         if ((fmt=aglChoosePixelFormat(NULL, 0, attrib)) == NULL)
  607.          throw "aglChoosePixelFormat failed";
  608.         // try to share display lists with current "regular" context
  609.         if ((ctx=aglCreateContext(fmt, currentCtx)) == NULL) {
  610.             WARNINGMSG("aglCreateContext with shared lists failed");
  611.          shareDisplayLists = false;
  612.          if ((ctx=aglCreateContext(fmt, NULL)) == NULL)
  613.              throw "aglCreateContext without shared lists failed";
  614.         }
  616.         // attach off-screen buffer to this context
  617.         if (!aglSetOffScreen(ctx, outputWidth, bufferHeight, bytesPerPixel * outputWidth, base))
  618.          throw "aglSetOffScreen failed";
  619.         if (!aglSetCurrentContext(ctx))
  620.          throw "aglSetCurrentContext failed";
  622.         glCanvas->renderer->RecreateQuadric(); // Macs have context-sensitive quadrics...
  623. #endif
  625.         TRACEMSG("interlaced: " << interlaced << ", nChunks: " << nChunks
  626.             << ", buffer height: " << bufferHeight << ", shared: " << shareDisplayLists);
  628.         // allocate a row of pixel storage
  629.         rowStorage = new unsigned char[outputWidth * bytesPerPixel];
  630.         if (!rowStorage) throw "failed to allocate pixel row buffer";
  632.         /* set up the PNG writing (see libpng.txt) */
  633.         png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, writepng_error_handler, NULL);
  634.         if (!png_ptr) throw "can't create PNG write structure";
  635.         info_ptr = png_create_info_struct(png_ptr);
  636.         if (!info_ptr) throw "can't create PNG info structure";
  637.         if (setjmp(png_ptr->jmpbuf)) throw "setjmp failed";
  638.         png_init_io(png_ptr, out);
  640.         // sets callback that's called by PNG after each written row
  641.         png_set_write_status_fn(png_ptr, write_row_callback);
  643.         // set various PNG parameters
  644.         png_set_compression_level(png_ptr, Z_BEST_COMPRESSION);
  645.         png_set_IHDR(png_ptr, info_ptr, outputWidth, outputHeight,
  646.             8, PNG_COLOR_TYPE_RGB,
  647.             interlaced ? PNG_INTERLACE_ADAM7 : PNG_INTERLACE_NONE,
  648.             PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
  649.         png_write_info(png_ptr, info_ptr);
  651.         // set up camera so that it's the same view as it is in the window
  652.         glCanvas->renderer->Init();
  653.         glViewport(0, 0, outputWidth, outputHeight);
  654.         glCanvas->renderer->NewView();
  656.         // Redraw the model into the new off-screen context, then use glReadPixels
  657.         // to retrieve pixel data. It's much easier to use glReadPixels rather than
  658.         // trying to read directly from the off-screen buffer, since GL can do all
  659.         // the potentially tricky work of translating from whatever pixel format
  660.         // the buffer uses into "regular" RGB byte triples. If fonts need scaling,
  661.         // have to reconstruct lists regardless of whether display lists are shared.
  662.         if (!shareDisplayLists || outputHeight != glCanvas->GetClientSize().GetHeight()) {
  663.             glCanvas->SetGLFontFromRegistry(((double) outputHeight) / glCanvas->GetClientSize().GetHeight());
  664.             glCanvas->renderer->Construct();
  665.         }
  666.         glPixelStorei(GL_PACK_ALIGNMENT, 1);
  668.         // Write image row by row to avoid having to allocate another full image's
  669.         // worth of memory. Do multiple passes for interlacing, if necessary. Note
  670.         // that PNG's rows are stored top down, while GL's are bottom up.
  671.         if (interlaced) {
  673.             // need to draw only once
  674.             glCanvas->renderer->Display();
  676.             int pass, r;
  677.             nRows = -outputHeight; // signal to monitor that we're interlacing
  678.             if (png_set_interlace_handling(png_ptr) != 7) throw "confused by unkown PNG interlace scheme";
  679.             for (pass = 1; pass <= 7; ++pass) {
  680.                 for (int i = outputHeight - 1; i >= 0; --i) {
  681.                     r = outputHeight - i - 1;
  682.                     // when interlacing, only certain rows are actually read
  683.                     // during certain passes - avoid unncessary reads
  684.                     if (
  685.                         ((pass == 1 || pass == 2) && (r % 8 == 0)) ||
  686.                         ((pass == 3) && ((r - 4) % 8 == 0)) ||
  687.                         ((pass == 4) && (r % 4 == 0)) ||
  688.                         ((pass == 5) && ((r - 2) % 4 == 0)) ||
  689.                         ((pass == 6) && (r % 2 == 0)) ||
  690.                         ((pass == 7) && ((r - 1) % 2 == 0))
  691.                     )
  692.                         glReadPixels(0, i, outputWidth, 1, GL_RGB, GL_UNSIGNED_BYTE, rowStorage);
  693.                     // ... but still have to call this for each row in each pass */
  694.                     png_write_row(png_ptr, rowStorage);
  695.                 }
  696.             }
  697.         }
  699.         // not interlaced, but possibly chunked
  700.         else {
  701.             int bufferRow, bufferRowStart;
  702.             nRows = outputHeight;
  703.             for (int chunk = nChunks - 1; chunk >= 0; --chunk) {
  705.                 // set viewport for this chunk and redraw
  706.                 if (nChunks > 1) {
  707.                     TRACEMSG("drawing chunk #" << (chunk + 1));
  708.                     glViewport(0, -chunk*bufferHeight, outputWidth, outputHeight);
  709.                 }
  710.                 glCanvas->renderer->Display();
  712.                 // only draw "visible" part of top chunk
  713.                 if (chunk == nChunks - 1)
  714.                     bufferRowStart = outputHeight - 1 - bufferHeight * (nChunks - 1);
  715.                 else
  716.                     bufferRowStart = bufferHeight - 1;
  718.                 // dump chunk to PNG file
  719.                 for (bufferRow = bufferRowStart; bufferRow >= 0; --bufferRow) {
  720.                     glReadPixels(0, bufferRow, outputWidth, 1, GL_RGB, GL_UNSIGNED_BYTE, rowStorage);
  721.                     png_write_row(png_ptr, rowStorage);
  722.                 }
  723.             }
  724.         }
  726.         // finish up PNG write
  727.         png_write_end(png_ptr, info_ptr);
  728.         success = true;
  730.         // general cleanup
  731.         if (out) fclose(out);
  732.         if (rowStorage) delete rowStorage;
  733.         if (png_ptr) {
  734.             if (info_ptr)
  735.                 png_destroy_write_struct(&png_ptr, &info_ptr);
  736.             else
  737.                 png_destroy_write_struct(&png_ptr, NULL);
  738.         }
  739.         if (progressMeter) {
  740.             progressMeter->Close(true);
  741.             progressMeter->Destroy();
  742.             progressMeter = NULL;
  743.         }
  745.     // platform-specific cleanup, restore context
  746. #if defined(__WXMSW__)
  747.         if (current_hdc && current_hglrc) wglMakeCurrent(current_hdc, current_hglrc);
  748.         if (hglrc) wglDeleteContext(hglrc);
  749.         if (hbm) DeleteObject(hbm);
  750.         if (hdc) DeleteDC(hdc);
  752. #elif defined(__WXGTK__)
  753.         gotAnXError = false;
  754.         if (glCtx) {
  755.             glXMakeCurrent(display, currentXdrw, currentCtx);
  756.             glXDestroyContext(display, glCtx);
  757.         }
  758.         if (glxPixmap) glXDestroyGLXPixmap(display, glxPixmap);
  759.         if (xPixmap) XFreePixmap(display, xPixmap);
  760.         if (localVI && visinfo) XFree(visinfo);
  761.         if (gotAnXError) WARNINGMSG("Got an X error destroying GLXPixmap context");
  762.         XSetErrorHandler(currentXErrHandler);
  764. #elif defined(__WXMAC__)
  765.         aglSetCurrentContext(currentCtx);
  766.         if (ctx) aglDestroyContext(ctx);
  767.         if (fmt) aglDestroyPixelFormat(fmt);
  768.         if (base) delete[] base;
  769.         glCanvas->renderer->RecreateQuadric(); // Macs have context-sensitive quadrics...
  770. #endif
  772.     } catch (const char *err) {
  773.         ERRORMSG("Error creating PNG: " << err);
  774.     } catch (exception& e) {
  775.         ERRORMSG("Uncaught exception while creating PNG: " << e.what());
  776.     }
  778.     // reset font after "regular" context restore
  779.     glCanvas->SuspendRendering(false);
  780.     if (outputHeight != glCanvas->GetClientSize().GetHeight()) {
  781.         glCanvas->SetCurrent();
  782.         glCanvas->SetGLFontFromRegistry(1.0);
  783.         if (shareDisplayLists)
  784.             GlobalMessenger()->PostRedrawAllStructures();
  785.     }
  786.     glCanvas->Refresh(false);
  788.     return success;
  789. }
  791. END_SCOPE(Cn3D)
  794. ////////////////////////////////////////////////////////////////////////////////////////////////
  795. // The following is taken unmodified from wxDesigner's C++ code from png_dialog.wdr
  796. ////////////////////////////////////////////////////////////////////////////////////////////////
  798. wxSizer *SetupPNGOptionsDialog( wxPanel *parent, bool call_fit, bool set_sizer )
  799. {
  800.     wxBoxSizer *item0 = new wxBoxSizer( wxVERTICAL );
  802.     wxStaticBox *item2 = new wxStaticBox( parent, -1, "Output Settings" );
  803.     wxStaticBoxSizer *item1 = new wxStaticBoxSizer( item2, wxVERTICAL );
  805.     wxFlexGridSizer *item3 = new wxFlexGridSizer( 1, 0, 0, 0 );
  806.     item3->AddGrowableCol( 1 );
  808.     wxStaticText *item4 = new wxStaticText( parent, ID_TEXT, "File name:", wxDefaultPosition, wxDefaultSize, 0 );
  809.     item3->Add( item4, 0, wxALIGN_CENTRE|wxALL, 5 );
  811.     item3->Add( 20, 20, 0, wxALIGN_CENTRE|wxALL, 5 );
  813.     wxButton *item5 = new wxButton( parent, ID_B_BROWSE, "Browse", wxDefaultPosition, wxDefaultSize, 0 );
  814.     item3->Add( item5, 0, wxALIGN_CENTRE|wxLEFT|wxRIGHT|wxTOP, 5 );
  816.     item1->Add( item3, 0, wxGROW|wxALIGN_CENTER_VERTICAL, 5 );
  818.     wxTextCtrl *item6 = new wxTextCtrl( parent, ID_T_NAME, "", wxDefaultPosition, wxDefaultSize, 0 );
  819.     item1->Add( item6, 0, wxGROW|wxALIGN_CENTER_VERTICAL|wxLEFT|wxRIGHT|wxBOTTOM, 5 );
  821.     wxFlexGridSizer *item7 = new wxFlexGridSizer( 2, 0, 0, 0 );
  822.     item7->AddGrowableCol( 1 );
  823.     item7->AddGrowableRow( 1 );
  825.     wxStaticText *item8 = new wxStaticText( parent, ID_TEXT, "Width:", wxDefaultPosition, wxDefaultSize, 0 );
  826.     item7->Add( item8, 0, wxALIGN_CENTER_VERTICAL|wxLEFT|wxRIGHT|wxTOP, 5 );
  828.     item7->Add( 20, 20, 0, wxALIGN_CENTRE|wxALL, 5 );
  830.     wxStaticText *item9 = new wxStaticText( parent, ID_TEXT, "Height:", wxDefaultPosition, wxDefaultSize, 0 );
  831.     item7->Add( item9, 0, wxALIGN_CENTER_VERTICAL|wxLEFT|wxRIGHT|wxTOP, 5 );
  833.     wxTextCtrl *item10 = new wxTextCtrl( parent, ID_T_WIDTH, "", wxDefaultPosition, wxDefaultSize, 0 );
  834.     item7->Add( item10, 0, wxALIGN_CENTRE|wxLEFT|wxRIGHT|wxBOTTOM, 5 );
  836.     wxStaticText *item11 = new wxStaticText( parent, ID_TEXT, "x", wxDefaultPosition, wxDefaultSize, 0 );
  837.     item7->Add( item11, 0, wxALIGN_CENTRE|wxLEFT|wxRIGHT|wxBOTTOM, 5 );
  839.     wxTextCtrl *item12 = new wxTextCtrl( parent, ID_T_HEIGHT, "", wxDefaultPosition, wxDefaultSize, 0 );
  840.     item7->Add( item12, 0, wxALIGN_CENTRE|wxLEFT|wxRIGHT|wxBOTTOM, 5 );
  842.     item1->Add( item7, 0, wxGROW|wxALIGN_CENTER_VERTICAL|wxALL, 5 );
  844.     wxBoxSizer *item13 = new wxBoxSizer( wxHORIZONTAL );
  846.     wxCheckBox *item14 = new wxCheckBox( parent, ID_C_ASPECT, "Maintain original aspect ratio", wxDefaultPosition, wxDefaultSize, 0 );
  847.     item14->SetValue( TRUE );
  848.     item13->Add( item14, 0, wxALIGN_CENTRE|wxALL, 5 );
  850.     item13->Add( 20, 20, 0, wxALIGN_CENTRE|wxALL, 5 );
  852.     wxCheckBox *item15 = new wxCheckBox( parent, ID_C_INTERLACE, "Interlaced", wxDefaultPosition, wxDefaultSize, 0 );
  853.     item15->SetValue( TRUE );
  854.     item13->Add( item15, 0, wxALIGN_CENTRE|wxALL, 5 );
  856.     item1->Add( item13, 0, wxALIGN_CENTRE|wxLEFT|wxRIGHT|wxTOP, 5 );
  858.     item0->Add( item1, 0, wxALIGN_CENTRE|wxALL, 5 );
  860.     wxBoxSizer *item16 = new wxBoxSizer( wxHORIZONTAL );
  862.     wxButton *item17 = new wxButton( parent, ID_B_OK, "OK", wxDefaultPosition, wxDefaultSize, 0 );
  863.     item17->SetDefault();
  864.     item16->Add( item17, 0, wxALIGN_CENTRE|wxALL, 5 );
  866.     item16->Add( 20, 20, 0, wxALIGN_CENTRE|wxALL, 5 );
  868.     wxButton *item18 = new wxButton( parent, ID_B_CANCEL, "Cancel", wxDefaultPosition, wxDefaultSize, 0 );
  869.     item16->Add( item18, 0, wxALIGN_CENTRE|wxALL, 5 );
  871.     item0->Add( item16, 0, wxALIGN_CENTRE|wxALL, 5 );
  873.     if (set_sizer)
  874.     {
  875.         parent->SetAutoLayout( TRUE );
  876.         parent->SetSizer( item0 );
  877.         if (call_fit)
  878.         {
  879.             item0->Fit( parent );
  880.             item0->SetSizeHints( parent );
  881.         }
  882.     }
  884.     return item0;
  885. }
  887. /*
  888. * ---------------------------------------------------------------------------
  889. * $Log: cn3d_png.cpp,v $
  890. * Revision 1000.3  2004/06/01 18:28:18  gouriano
  891. * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.21
  892. *
  893. * Revision 1.21  2004/05/21 21:41:39  gorelenk
  894. * Added PCH ncbi_pch.hpp
  895. *
  896. * Revision 1.20  2004/03/15 17:30:06  thiessen
  897. * prefer prefix vs. postfix ++/-- operators
  898. *
  899. * Revision 1.19  2004/02/19 17:04:50  thiessen
  900. * remove cn3d/ from include paths; add pragma to disable annoying msvc warning
  901. *
  902. * Revision 1.18  2003/12/04 15:49:39  thiessen
  903. * fix stereo and PNG export problems on Mac
  904. *
  905. * Revision 1.17  2003/11/15 16:08:35  thiessen
  906. * add stereo
  907. *
  908. * Revision 1.16  2003/03/13 14:26:18  thiessen
  909. * add file_messaging module; split cn3d_main_wxwin into cn3d_app, cn3d_glcanvas, structure_window, cn3d_tools
  910. *
  911. * Revision 1.15  2003/02/03 19:20:03  thiessen
  912. * format changes: move CVS Log to bottom of file, remove std:: from .cpp files, and use new diagnostic macros
  913. *
  914. * Revision 1.14  2002/10/21 15:11:29  thiessen
  915. * don't share lists in wxGTK build
  916. *
  917. * Revision 1.13  2002/10/18 20:33:54  thiessen
  918. * workaround for linux/Mesa bug
  919. *
  920. * Revision 1.12  2002/10/18 15:42:59  thiessen
  921. * work around png header issue for linux
  922. *
  923. * Revision 1.11  2002/10/11 17:21:39  thiessen
  924. * initial Mac OSX build
  925. *
  926. * Revision 1.10  2002/08/15 22:13:14  thiessen
  927. * update for wx2.3.2+ only; add structure pick dialog; fix MultitextDialog bug
  928. *
  929. * Revision 1.9  2002/04/21 12:21:21  thiessen
  930. * minor fixes for AIX
  931. *
  932. * Revision 1.8  2001/10/25 17:16:43  thiessen
  933. * add PNG output to Mac version
  934. *
  935. * Revision 1.7  2001/10/25 14:19:44  thiessen
  936. * move png.h to top
  937. *
  938. * Revision 1.6  2001/10/25 00:06:29  thiessen
  939. * fix concurrent rendering problem in wxMSW PNG output
  940. *
  941. * Revision 1.5  2001/10/24 22:02:02  thiessen
  942. * fix wxGTK concurrent rendering problem
  943. *
  944. * Revision 1.4  2001/10/24 17:07:30  thiessen
  945. * add PNG output for wxGTK
  946. *
  947. * Revision 1.3  2001/10/24 11:25:20  thiessen
  948. * fix wxString problem
  949. *
  950. * Revision 1.2  2001/10/23 20:10:23  thiessen
  951. * fix scaling of fonts in high-res PNG output
  952. *
  953. * Revision 1.1  2001/10/23 13:53:38  thiessen
  954. * add PNG export
  955. *
  956. */