diff options
Diffstat (limited to 'src/pbmimage.cpp')
-rw-r--r-- | src/pbmimage.cpp | 253 |
1 files changed, 253 insertions, 0 deletions
diff --git a/src/pbmimage.cpp b/src/pbmimage.cpp new file mode 100644 index 0000000..eff2be3 --- /dev/null +++ b/src/pbmimage.cpp @@ -0,0 +1,253 @@ +/* + * pbmimage.cpp (C) 2006, Aurélien Croc (AP²C) + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; version 2 of the License. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the + * Free Software Foundation, Inc., + * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. + * + * $Id: pbmimage.cpp 56 2006-12-13 15:20:59Z ap2c $ + * + */ +#include "pbmimage.h" +#include <string.h> +#include <unistd.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> +#include <unistd.h> +#include "error.h" + +/* + * Constructeur - Destructeur + * Init - Uninit + */ +PbmImage::PbmImage(const char *black, const char *cyan, const char *magenta, + const char *yellow) +{ + _color = !cyan && !magenta && !yellow ? false : true; + _blackFile = black; + _cyanFile = cyan; + _magentaFile = magenta; + _yellowFile = yellow; + _black = NULL; + _cyan = NULL; + _magenta = NULL; + _yellow = NULL; + + _width = 0; + _height = 0; + _lineSize = 0; + _line = 0; + _lineBuffer = NULL; + _currentColor = 1; + +} + +PbmImage::~PbmImage() +{ + unload(); +} + + + +/* + * Chargement de l'image + * Load the image + */ +void PbmImage::unload() +{ + if (_black) { + fclose(_black); + _blackFile = NULL; + _black = NULL; + } + if (_color) { + if (_cyan) { + fclose(_cyan); + _cyanFile = NULL; + _cyan = NULL; + } + if (_magenta) { + fclose(_magenta); + _magentaFile = NULL; + _magenta = NULL; + } + if (_yellow) { + fclose(_yellow); + _yellowFile = NULL; + _yellow = NULL; + } + } + if (_lineBuffer) { + delete[] _lineBuffer; + _lineBuffer= NULL; + } +} + +int PbmImage::load() +{ + uint32_t width, height; + char buffer[1024]; + + // Open the different files + if (!(_black = fopen(_blackFile, "r"))) { + fprintf(stderr, _("Cannot open black file %s\n"), _blackFile); + return -1; + } + if (_color) { + if (_cyanFile && !(_cyan = fopen(_cyanFile, "r"))) { + fprintf(stderr, _("Cannot open cyan file %s\n"), + _cyanFile); + return -1; + } + if (_magentaFile && !(_magenta = fopen(_magentaFile, "r"))) { + fprintf(stderr, _("Cannot open magenta file %s\n"), + _magentaFile); + return -1; + } + if (_yellowFile && !(_yellow = fopen(_yellowFile, "r"))) { + fprintf(stderr, _("Cannot open yellow file %s\n"), + _yellowFile); + return -1; + } + } + + // Read the PBM header + fgets((char *)&buffer, sizeof(buffer), _black); + if (strcmp((char *)&buffer, "P4\n")) { + fprintf(stderr, _("Invalid PBM file for file %s\n"), + _blackFile); + return -1; + } + fgets((char *)&buffer, sizeof(buffer), _black); + fscanf(_black, "%u %u\n", &width, &height); + _width = width; + _height = height; + _lineSize = (width + 7) >> 3; + + if (_color) { + unsigned int tmpW, tmpH; + + if (_cyan) { + fgets((char *)&buffer, sizeof(buffer), _cyan); + if (strcmp((char *)&buffer, "P4\n")) { + fprintf(stderr, _("Invalid PBM file for file " + "%s\n"), _cyanFile); + return -1; + } + fgets((char *)&buffer, sizeof(buffer), _cyan); + fscanf(_cyan, "%u %u\n", &tmpW, &tmpH); + if ((tmpW != width) || (tmpH != height)) { + fprintf(stderr, _("The different PBM layers " + "must have the same size\n")); + return -1; + } + } + if (_magenta) { + fgets((char *)&buffer, sizeof(buffer), _magenta); + if (strcmp((char *)&buffer, "P4\n")) { + fprintf(stderr, _("Invalid PBM file for file " + "%s\n"), _magentaFile); + return -1; + } + fgets((char *)&buffer, sizeof(buffer), _magenta); + fscanf(_magenta, "%u %u\n", &tmpW, &tmpH); + if ((tmpW != width) || (tmpH != height)) { + fprintf(stderr, _("The different PBM layers " + "must have the same size\n")); + return -1; + } + } + if (_yellow) { + fgets((char *)&buffer, sizeof(buffer), _yellow); + if (strcmp((char *)&buffer, "P4\n")) { + fprintf(stderr, _("Invalid PBM file for file " + "%s\n"), _yellowFile); + return -1; + } + fgets((char *)&buffer, sizeof(buffer), _yellow); + fscanf(_yellow, "%u %u\n", &tmpW, &tmpH); + if ((tmpW != width) || (tmpH != height)) { + fprintf(stderr, _("The different PBM layers " + "must have the same size\n")); + return -1; + } + } + } + + return 0; +} + +int PbmImage::loadPage(Printer *printer) +{ + printer->setCompVersion(0x11); + if (_line) + return 1; + + return 0; +} + + + +/* + * Lecture d'une ligne + * Read a line + */ +int PbmImage::readLine() +{ + if (!_lineBuffer) + _lineBuffer = new unsigned char[_lineSize]; + + if (_line >= _height) { + memset(_lineBuffer, 0x00, _lineSize); + return _lineSize; + } + + if (_color) { + switch(_currentColor) { + case 1: + if (_cyan) + fread(_lineBuffer, 1, _lineSize, _cyan); + else + memset(_lineBuffer, 0x00, _lineSize); + _currentColor++; + break; + case 2: + if (_magenta) + fread(_lineBuffer, 1, _lineSize, + _magenta); + else + memset(_lineBuffer, 0x00, _lineSize); + _currentColor++; + break; + case 3: + if (_yellow) + fread(_lineBuffer, 1, _lineSize, + _yellow); + else + memset(_lineBuffer, 0x00, _lineSize); + _currentColor++; + break; + case 4: + fread(_lineBuffer, 1, _lineSize, _black); + _currentColor = 1; + _line++; + break; + } + } else { + fread(_lineBuffer, 1, _lineSize, _black); + _line++; + } + return _lineSize; +} + |