rasterimagereader.hh

Go to the documentation of this file.

// -*- mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
// vi: set et ts=4 sw=4 sts=4:
/*****************************************************************************
 *   See the file COPYING for full copying permissions.                      *
 *                                                                           *
 *   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, either version 3 of the License, or       *
 *   (at your option) any later version.                                     *
 *                                                                           *
 *   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, see <http://www.gnu.org/licenses/>.   *
 *****************************************************************************/
 #ifndef DUMUX_RASTER_IMAGE_READER_HH
 #define DUMUX_RASTER_IMAGE_READER_HH
 
 #include <cassert>
 #include <string>
 #include <vector>
 #include <fstream>
 #include <sstream>
 #include <algorithm>
 #include <map>
 #include <iterator>
 #include <iostream>
 
 #include <dune/common/exceptions.hh>
 
namespace Dumux {
 
class NetPBMReader
{
public:
    struct Format
    {
        std::string magicNumber;
        std::string type;
        std::string encoding;
    };
 
    struct HeaderData
    {
        Format format;
        std::size_t nCols;
        std::size_t nRows;
        std::size_t maxValue = 1;
    };
 
    template<class T>
    class Result : private std::vector<T>
    {
        using Parent = std::vector<T>;
    public:
        Result() = delete;
 
        Result(const std::vector<T>& data, const HeaderData& header)
        : Parent(data)
        , header_(header)
        {}
 
        Result(std::vector<T>&& data, HeaderData&& header)
        : Parent(std::move(data))
        , header_(std::move(header))
        {}
 
        const HeaderData& header() const { return header_; }
 
        // expose some methods of std::vector
        using Parent::operator[];
        using Parent::begin;
        using Parent::end;
        using Parent::size;
 
    private:
        HeaderData header_;
    };
 
    static Format getFormat(const std::string& magicNumber)
    {
        static const auto format = []{
            std::map<std::string, Format> format;
            format["P1"] = Format{"P1", "Portable BitMap", "ASCII"};
            format["P2"] = Format{"P2", "Portable GrayMap", "ASCII"};
            format["P3"] = Format{"P3", "Portable PixMap", "ASCII"};
            format["P4"] = Format{"P4", "Portable BitMap", "binary"};
            format["P5"] = Format{"P5", "Portable GrayMap", "binary"};
            format["P5"] = Format{"P5", "Portable PixMap", "binary"};
            return format;
        }();
 
        if (!format.count(magicNumber))
            DUNE_THROW(Dune::IOError, magicNumber << " is not a valid magic number for the Netpbm format");
 
        return format.at(magicNumber);
    }
 
    static Result<bool> readPBM(const std::string& fileName, const bool useDuneGridOrdering = true)
    {
        std::ifstream infile(fileName, std::ios::binary);
 
        if (!infile)
            DUNE_THROW(Dune::IOError, "Raster data file not found or corrupt");
 
        HeaderData headerData = readHeader(infile);
        std::vector<bool> values;
 
        if (headerData.format.magicNumber == "P1")
            values = readPBMDataASCII_(infile, headerData);
        else if (headerData.format.magicNumber == "P4")
            values = readPBMDataBinary_(infile, headerData);
        else
            DUNE_THROW(Dune::IOError, headerData.format.magicNumber << " not supported. Use readPBM for P1 and P4 or readPGM for P2 and P5");
 
        Result<bool> result(std::move(values), std::move(headerData));
        printInfo(result);
 
        if (useDuneGridOrdering)
            applyDuneGridOrdering(result);
 
        return result;
    }
 
    template<class ValueType = std::uint8_t>
    static Result<ValueType> readPGM(const std::string& fileName, const bool useDuneGridOrdering = true)
    {
        std::ifstream infile(fileName, std::ios::binary);
 
        if (!infile)
            DUNE_THROW(Dune::IOError, "Raster data file not found or corrupt");
 
        HeaderData headerData = readHeader(infile);
        std::vector<ValueType> values;
 
        if (headerData.format.magicNumber == "P2")
            values = NetPBMReader::template readPGMDataASCII_<ValueType>(infile, headerData);
        else if (headerData.format.magicNumber == "P5")
            values = NetPBMReader::template readPGMDataBinary_<ValueType>(infile, headerData);
        else
            DUNE_THROW(Dune::IOError, headerData.format.magicNumber << " not supported. Use readPBM for P1 and P4 or readPGM for P2 and P5");
 
        Result<ValueType> result(std::move(values), std::move(headerData));
        printInfo(result);
 
        if (useDuneGridOrdering)
            applyDuneGridOrdering(result);
 
        return result;
    }
 
    static HeaderData readHeader(std::ifstream& infile)
    {
        HeaderData headerData;
        std::string inputLine;
 
        // First line : get format.
        std::getline(infile, inputLine);
        headerData.format = getFormat(inputLine);
        const auto magicNumber = headerData.format.magicNumber;
 
        // Read dimensions and maximum value (for non-b/w images).
        while (!infile.eof())
        {
            std::getline(infile, inputLine);
 
            auto isComment = [](const auto& s)
            { return (s.find("#") != std::string::npos); };
 
            // Skip comments.
            if (isComment(inputLine))
                continue;
 
            // The first line after the comments contains the dimensions.
            headerData.nCols = std::stoi(inputLine.substr(0, inputLine.find(" ")));
            headerData.nRows = std::stoi(inputLine.substr(inputLine.find(" ") + 1));
 
            // Grayscale images additionaly contain a maxium value in the header.
            if (magicNumber != "P1" && magicNumber != "P4")
            {
                std::getline(infile, inputLine);
                headerData.maxValue = std::stoi(inputLine);
            }
            break;
        }
        return headerData;
    }
 
    template<class T>
    static void applyDuneGridOrdering(Result<T>& result)
    {
        auto tmp = result;
        for (std::size_t i = 0; i < result.size(); i += result.header().nCols)
            std::swap_ranges((result.begin() + i), (result.begin() + i + result.header().nCols), (tmp.end() - i - result.header().nCols));
    }
 
    template<class T>
    static void printInfo(const Result<T>& result)
    {
        const Format& format = result.header().format;
        std::cout << "Reading " << format.type << " file (" << format.encoding << ")" << std::endl;
        std::cout << "Dimensions : " << result.header().nCols << " " << result.header().nRows << std::endl;
        std::cout << "Maximum value : " << result.header().maxValue << std::endl;
    }
 
    template<class Image, class T>
    static void fillImage(Image& image, const Result<T>& result)
    {
        const auto nCols = result.header().nCols;
        const auto nRows = result.header().nRows;
        using RowType = std::decay_t<decltype(image[0])>;
        image.resize(nRows, RowType(nCols));
 
        std::size_t rowIdx = 0;
        std::size_t colIdx = 0;
        for (const auto& val : result)
        {
            image[rowIdx][colIdx] = val;
 
            // start a new line after nCols entries
            if (++colIdx == nCols)
            {
                colIdx = 0;
                ++rowIdx;
            }
        }
    }
 
    template<class Image>
    static auto flattenImageToVector(const Image& image)
    {
        using OutputValueType = std::decay_t<decltype(image[0][0])>;
 
        std::vector<OutputValueType> data;
        data.reserve(image.size()*image[0].size());
        for (const auto& row: image)
            for (const auto& col : row)
                data.push_back(col);
 
        return data;
    }
 
private:
    static std::vector<bool> readPBMDataASCII_(std::ifstream& infile,
                                               const HeaderData& headerData)
    {
        std::string inputLine;
        std::vector<bool> data;
        data.reserve(numPixel_(headerData));
 
        while (!infile.eof())
        {
            std::getline(infile, inputLine);
            inputLine.erase(std::remove(inputLine.begin(), inputLine.end(), '\n'), inputLine.end());
            inputLine.erase(std::remove(inputLine.begin(), inputLine.end(), ' '), inputLine.end());
            if (!inputLine.empty())
            {
                for (const auto& value : inputLine)
                {
                    assert(value == '0' || value == '1');
                    data.push_back(value - '0');  // convert char to int
                }
            }
        }
 
        return data;
    }
 
    static std::vector<bool> readPBMDataBinary_(std::ifstream& infile,
                                                const HeaderData& headerData)
    {
        std::vector<bool> data(numPixel_(headerData));
 
        std::size_t nBytes = 0;
        std::size_t bitIndex = 0;
        using Bit = std::uint8_t;
        Bit b = 0;
        for (std::size_t j = 0; j < headerData.nRows; j++)
        {
            for (std::size_t i = 0; i < headerData.nCols; i++)
            {
                if (i%8 == 0)
                {
                    char tmp;
                    infile.read(&tmp, 1);
                    b = static_cast<Bit>(tmp);
                    if (infile.eof())
                        DUNE_THROW(Dune::IOError, "Failed reading byte " << nBytes);
 
                    ++nBytes;
                }
 
                const Bit k = 7 - (i % 8);
                data[bitIndex++] = static_cast<bool>((b >> k) & 1);
            }
        }
 
        return data;
    }
 
    template<class ValueType = std::uint8_t>
    static std::vector<ValueType> readPGMDataASCII_(std::ifstream& infile,
                                                    const HeaderData& headerData)
    {
        std::string inputLine;
 
        std::vector<ValueType> data;
        data.reserve(numPixel_(headerData));
 
        while (!infile.eof())
        {
            std::getline(infile, inputLine);
            if (inputLine.empty())
                continue;
 
            // if the line contains multiple comma-separated values, store them individually in a vector
            if (inputLine.find(" ") != std::string::npos)
            {
                std::istringstream iss(inputLine);
                std::vector<std::string> tokens;
                std::copy(std::istream_iterator<std::string>(iss),
                          std::istream_iterator<std::string>(),
                          std::back_inserter(tokens));
 
                for (const auto& t : tokens)
                    data.push_back(std::stoi(t)); // convert string to integer type
            }
            else
                data.push_back(std::stoi(inputLine));
        }
 
        return data;
    }
 
    template<class ValueType = std::uint8_t>
    static std::vector<ValueType> readPGMDataBinary_(std::ifstream& infile,
                                                     const HeaderData& headerData)
    {
        std::string inputLine;
 
        // check the size of the binary part of the file
        const auto curPos = infile.tellg();
        infile.seekg(0, std::ios::end);
        const auto endPos = infile.tellg();
        const auto size = endPos - curPos;
        if (size != numPixel_(headerData))
            DUNE_THROW(Dune::IOError, "Binary file size does not match with raster image size");
 
        // reset to the current position
        infile.seekg(curPos, std::ios::beg);
 
        // extract the binary data
        std::vector<std::uint8_t> data(size);
        infile.read(reinterpret_cast<char*>(&data[0]), size);
 
        // convert std::uint8_t to ValueType
        return std::vector<ValueType>(data.begin(), data.end());
    }
 
    static std::size_t numPixel_(const HeaderData& headerData)
    {
        return headerData.nRows*headerData.nCols;
    }
};
 
} // namespace Dumux
 
#endif // DUMUX_RASTER_IMAGE_READER_HH