tomwalters@116: // Copyright 2006, Willem van Engen
tomwalters@116: //
tomwalters@116: // AIM-C: A C++ implementation of the Auditory Image Model
tomwalters@116: // http://www.acousticscale.org/AIMC
tomwalters@116: //
tomwalters@116: // Licensed under the Apache License, Version 2.0 (the "License");
tomwalters@116: // you may not use this file except in compliance with the License.
tomwalters@116: // You may obtain a copy of the License at
tomwalters@116: //
tomwalters@116: //     http://www.apache.org/licenses/LICENSE-2.0
tomwalters@116: //
tomwalters@116: // Unless required by applicable law or agreed to in writing, software
tomwalters@116: // distributed under the License is distributed on an "AS IS" BASIS,
tomwalters@116: // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
tomwalters@116: // See the License for the specific language governing permissions and
tomwalters@116: // limitations under the License.
tomwalters@116: 
tomwalters@116: /*!
tomwalters@116:  * \file
tomwalters@116:  * \brief Output device for output to a graphics file using plotutils
tomwalters@116:  *
tomwalters@116:  * \author Willem van Engen <cnbh@willem.engen.nl>
tomwalters@116:  * \date created 2006/10/13
tomwalters@116:  * \version \$Id: GraphicsOutputDevicePlotutils.cpp 493 2007-11-27 10:59:20Z tom $
tomwalters@116:  */
tomwalters@116: #include "Support/Common.h"
tomwalters@116: 
tomwalters@116: #include <sys/types.h>
tomwalters@116: #include <sys/stat.h>
tomwalters@116: #include <string.h>
tomwalters@116: #include <stdio.h>
tomwalters@116: #include <math.h>
tomwalters@116: 
tomwalters@116: #include "Support/util.h"
tomwalters@116: #include "Modules/Output/Graphics/Devices/GraphicsOutputDevicePlotutils.h"
tomwalters@116: 
tomwalters@116: GraphicsOutputDevicePlotutils::GraphicsOutputDevicePlotutils(Parameters *pParam)
tomwalters@116:     : GraphicsOutputDevice(pParam) {
tomwalters@228:   m_iPlotHandle = 0;
tomwalters@228:   m_pOutputFile = NULL;
tomwalters@228:   m_iFileNumber = 0;
tomwalters@228:   m_iVertexType = VertexTypeNone;
tomwalters@228:   m_bUseMemoryBuffer = false;
tomwalters@228:   m_pMemoryBuffer = NULL;
tomwalters@228:   m_uWidth=0;
tomwalters@228:   m_uHeight=0;
tomwalters@116: }
tomwalters@116: 
tomwalters@116: bool GraphicsOutputDevicePlotutils::Initialize(const char *sDir) {
tomwalters@116:   Init();
tomwalters@116: 
tomwalters@228:   //! \todo Output to file if sDir is a file, to directory with multiple
tomwalters@116:   //! images if it's a directory.
tomwalters@228:   strncpy(m_sDir, sDir, sizeof(m_sDir)/sizeof(m_sDir[0]));
tomwalters@116: 
tomwalters@228:   /* Try to open an image to see if everything is allright. We want to avoid
tomwalters@228:    * errors in the main Process()ing loop. */
tomwalters@228:   if (!OpenFile(0)) {
tomwalters@228:     //! \todo Better error message that is more specific about the cause.
tomwalters@228:     AIM_ERROR(_T("Could not open output directory '%s' using graphics format '%s'."),
tomwalters@228:              m_sDir, m_pParam->GetString("output.img.format") );
tomwalters@228:     return false;
tomwalters@228:   }
tomwalters@228:   CloseFile();
tomwalters@228:   return true;
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::Init() {
tomwalters@116:   AIM_ASSERT(m_pParam);
tomwalters@228:   /*
tomwalters@228:    * Set parameters
tomwalters@228:    */
tomwalters@228:   pl_parampl("BG_COLOR", (void*)m_pParam->GetString("output.img.color.background"));
tomwalters@228:   // Handle GIFs as other output formats, don't merge frames into single GIF.
tomwalters@228:   pl_parampl("GIF_ANIMATION", (void*)"no");
tomwalters@228:   pl_parampl("INTERLACE", (void*)"no");
tomwalters@116: 
tomwalters@228:   m_bInvertColors = m_pParam->GetBool("output.img.color.invert");
tomwalters@116: 
tomwalters@228:   // Output size!
tomwalters@228:   m_uWidth = m_pParam->GetUInt("output.img.width");
tomwalters@228:   m_uHeight = m_pParam->GetUInt("output.img.height");
tomwalters@228:   char strSize[100];
tomwalters@228:   snprintf(strSize, sizeof(strSize)/sizeof(strSize[0]), "%ux%u", m_uWidth, m_uHeight);
tomwalters@228:   pl_parampl("BITMAPSIZE", (void*)strSize);
tomwalters@116: }
tomwalters@116: 
tomwalters@116: bool GraphicsOutputDevicePlotutils::Initialize() {
tomwalters@116:   Init();
tomwalters@116:   m_bUseMemoryBuffer = true;
tomwalters@116:   return(true);
tomwalters@116: }
tomwalters@116: 
tomwalters@116: bool GraphicsOutputDevicePlotutils::OpenFile(unsigned int index) {
tomwalters@116:   const char *strPlottype = m_pParam->GetString("output.img.format");
tomwalters@116:   if (!m_bUseMemoryBuffer) {
tomwalters@116:     char sFilename[PATH_MAX];
tomwalters@116:     struct stat fileinfo;
tomwalters@116: 
tomwalters@116:     // Get filename without trailing slash
tomwalters@116:     strncpy(sFilename, m_sDir, sizeof(sFilename)/sizeof(sFilename[0]));
tomwalters@116: #ifdef _WINDOWS
tomwalters@116:     if (sFilename[strlen(sFilename)-1]=='\\') {
tomwalters@116:       sFilename[strlen(sFilename)-1]='\0';
tomwalters@116:     }
tomwalters@116: #else
tomwalters@116:     if (sFilename[strlen(sFilename)-1]=='/') {
tomwalters@116:       sFilename[strlen(sFilename)-1]='\0';
tomwalters@116:     }
tomwalters@116: #endif
tomwalters@116:     // Enumerate files it m_sDir is a directory.
tomwalters@116:     if ( stat(sFilename, &fileinfo)==0 && (fileinfo.st_mode & S_IFDIR) ) {
tomwalters@116:       // We have a directory: enumerate with index
tomwalters@116:       snprintf(sFilename, sizeof(sFilename)/sizeof(sFilename[0]),"%s%06d.%s", m_sDir, index, strPlottype);
tomwalters@116:       // If type is 'auto', fallback to 'png'
tomwalters@116:       if (strcmp(strPlottype, "auto")==0)
tomwalters@116:           strPlottype = "png";
tomwalters@116:       } else {
tomwalters@116:         // We have a (probabely non-existant) file. Auto-detect type by extension if requested
tomwalters@116:         strncpy(sFilename, m_sDir, sizeof(sFilename)/sizeof(sFilename[0]));
tomwalters@116:         char *pDot = strrchr(sFilename, '.');
tomwalters@116:         if (!pDot) {
tomwalters@116:           AIM_ERROR(_T("Please supply extension on filename when using 'auto' format: '%s'"), sFilename);
tomwalters@116:           return false;
tomwalters@116:         }
tomwalters@116:         strPlottype = &pDot[1];
tomwalters@116:       }
tomwalters@116:       if ((m_pOutputFile = fopen(sFilename, "wb")) == NULL) {
tomwalters@116:         return false;
tomwalters@116:       }
tomwalters@116:     } else {
tomwalters@116: #if !defined(_WINDOWS) && !defined(_MACOSX)
tomwalters@116:     if ((m_pOutputFile=open_memstream(&m_pMemoryBuffer, &m_sMemoryBufferSize))==NULL)
tomwalters@116: #endif
tomwalters@116:       return false;
tomwalters@116:       strPlottype="pnm"; // Force pnm format or this doesn't work
tomwalters@116:     }
tomwalters@116: 
tomwalters@228:   /*
tomwalters@228:    * Create a plotter
tomwalters@228:    *
tomwalters@228:    * Plotutils knows the following types for file output:
tomwalters@228:    *   pnm gif ai ps gif pcl hpgl tek meta
tomwalters@228:    */
tomwalters@228:   if ((m_iPlotHandle = pl_newpl(strPlottype,
tomwalters@116:                                 NULL,
tomwalters@116:                                 m_pOutputFile,
tomwalters@116:                                 stderr)) < 0 ) {
tomwalters@228:     return false;
tomwalters@228:   }
tomwalters@228:   pl_selectpl(m_iPlotHandle);
tomwalters@116: 
tomwalters@228:   if ( pl_openpl() < 0) {
tomwalters@228:     return false;
tomwalters@228:   }
tomwalters@116: 
tomwalters@228:   // Now setup things for this plotter
tomwalters@228:   pl_fontname(m_pParam->GetString("output.img.fontname"));
tomwalters@228:   //! \todo Make fontsize work in Plotutils, currently disabled
tomwalters@228:   //pl_ffontsize(m_pParam->GetUInt("output.fontsize"));
tomwalters@116: 
tomwalters@228:   return true;
tomwalters@116: }
tomwalters@116: 
tomwalters@116: char* GraphicsOutputDevicePlotutils::GetBuffer() {
tomwalters@116:   if(m_bUseMemoryBuffer && (m_pMemoryBuffer!=NULL))
tomwalters@116:     return (&m_pMemoryBuffer[m_sMemoryBufferSize-(m_uWidth*m_uHeight*3)]);
tomwalters@116:   else
tomwalters@116:     return NULL;
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::CloseFile() {
tomwalters@228:   // Plotting library
tomwalters@228:   if (m_iPlotHandle>0) {
tomwalters@228:     pl_closepl();
tomwalters@116: 
tomwalters@228:     pl_selectpl(0);
tomwalters@228:     pl_deletepl(m_iPlotHandle);
tomwalters@228:     m_iPlotHandle = 0;
tomwalters@228:   }
tomwalters@116: 
tomwalters@228:   // And the output file
tomwalters@228:   if (m_pOutputFile) {
tomwalters@228:     fclose(m_pOutputFile);
tomwalters@228:     m_pOutputFile = NULL;
tomwalters@228:   }
tomwalters@116: }
tomwalters@116: 
tomwalters@116: GraphicsOutputDevicePlotutils::~GraphicsOutputDevicePlotutils() {
tomwalters@228:   // Output file should be closed by gRelease()
tomwalters@228:   AIM_ASSERT(!m_pOutputFile);
tomwalters@228:   AIM_ASSERT(!m_iPlotHandle);
tomwalters@228:   CloseFile();
tomwalters@116: }
tomwalters@116: 
tomwalters@116: PixelFormat GraphicsOutputDevicePlotutils::GetPixelFormat() {
tomwalters@116:     return AIM_PIX_FMT_RGB24_24;
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::gGrab() {
tomwalters@228:   // Open file
tomwalters@228:   if (!OpenFile(m_iFileNumber)) {
tomwalters@228:     return;
tomwalters@116:   }
tomwalters@228:   // Setup plotting area
tomwalters@228:   pl_fspace(0.0, 0.0, 1.0, 1.0);
tomwalters@228:   pl_flinewidth(0.0001);
tomwalters@228:   pl_pencolorname("darkblue");
tomwalters@228:   pl_erase();
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::gBeginLineStrip() {
tomwalters@228:   m_bIsFirstVertex = true;
tomwalters@228:   m_iVertexType = VertexTypeLine;
tomwalters@228:   pl_filltype(0);
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::gBeginQuadStrip() {
tomwalters@228:   m_bIsFirstVertex = true;
tomwalters@228:   m_iVertexType = VertexTypeQuad;
tomwalters@228:   m_iPrevVertexCount = 0;
tomwalters@228:   pl_filltype(1);
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::gColor3f(float r, float g, float b) {
tomwalters@228:   if (m_bInvertColors) {
tomwalters@228:     r = 1-r;
tomwalters@228:     g = 1-g;
tomwalters@228:     b = 1-b;
tomwalters@228:   }
tomwalters@228:   int ir = (int)(r*0xffff);
tomwalters@228:   int ig = (int)(g*0xffff);
tomwalters@228:   int ib = (int)(b*0xffff);
tomwalters@228:   ir = MIN(0xffff, MAX(0,ir));
tomwalters@228:   ig = MIN(0xffff, MAX(0,ig));
tomwalters@228:   ib = MIN(0xffff, MAX(0,ib));
tomwalters@228:   pl_color(ir, ig, ib);
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::gVertex3f(float x, float y, float z) {
tomwalters@228:   switch(m_iVertexType) {
tomwalters@228:   case VertexTypeLine:
tomwalters@228:     if (m_bIsFirstVertex) {
tomwalters@228:       m_bIsFirstVertex = false;
tomwalters@228:       pl_fmove(x, y);
tomwalters@228:     } else {
tomwalters@228:       pl_fcont(x, y);
tomwalters@228:     }
tomwalters@228:     break;
tomwalters@228:   case VertexTypeQuad:
tomwalters@228:     /* Store vertices until we got four in a row.
tomwalters@228:      * The order of vertices when processing quads is:
tomwalters@228:      *    1-----3-----5
tomwalters@228:      *    |     |     |
tomwalters@228:      *    0-----2-----4
tomwalters@228:      */
tomwalters@228:     if (m_iPrevVertexCount>=3) {
tomwalters@228:       // Plot this quad
tomwalters@228:       pl_fmove(m_aPrevX[0],m_aPrevY[0]);
tomwalters@228:       pl_fcont(m_aPrevX[1],m_aPrevY[1]);
tomwalters@228:       pl_fcont(x,y);
tomwalters@228:       pl_fcont(m_aPrevX[2],m_aPrevY[2]);
tomwalters@228:       pl_endpath();
tomwalters@228:       // Last vertices of this quad are the first of the next
tomwalters@228:       m_aPrevX[0] = m_aPrevX[2];
tomwalters@228:       m_aPrevY[0] = m_aPrevY[2];
tomwalters@228:       m_aPrevX[1] = x;
tomwalters@228:       m_aPrevY[1] = y;
tomwalters@228:       m_iPrevVertexCount = 2;
tomwalters@228:     } else {
tomwalters@228:       // Not at the fourth, keep storing
tomwalters@228:       m_aPrevX[m_iPrevVertexCount] = x;
tomwalters@228:       m_aPrevY[m_iPrevVertexCount] = y;
tomwalters@228:       m_iPrevVertexCount++;
tomwalters@228:     }
tomwalters@228:     break;
tomwalters@228:   default:
tomwalters@228:     // Should not happen
tomwalters@228:     AIM_ASSERT(0);
tomwalters@228:   }
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::gEnd() {
tomwalters@228:   pl_endpath();
tomwalters@228:   m_iVertexType = VertexTypeNone;
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::gText3f(float x,
tomwalters@116:                                             float y,
tomwalters@116:                                             float z,
tomwalters@116:                                             const char *sStr,
tomwalters@116:                                             bool bRotated) {
tomwalters@228:   if (bRotated) {
tomwalters@228:     pl_textangle(90);
tomwalters@228:     pl_fmove(x, y);
tomwalters@228:     pl_alabel('l', 't', sStr);
tomwalters@228:   } else {
tomwalters@228:     pl_textangle(0);
tomwalters@228:     pl_fmove(x, y);
tomwalters@228:     pl_alabel('l', 'b', sStr);
tomwalters@228:   }
tomwalters@116: }
tomwalters@116: 
tomwalters@116: void GraphicsOutputDevicePlotutils::gRelease() {
tomwalters@228:   AIM_ASSERT(m_pOutputFile);
tomwalters@228:   AIM_ASSERT(m_iPlotHandle>0);
tomwalters@228:   CloseFile();
tomwalters@228:   m_iFileNumber++;
tomwalters@116: }