least_significant_bit.hpp

/* This file is a part of "Steganography" a C++ steganography tool.

Copyright (C) 2019 James Lee <jamesl33info@gmail.com>.

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/>. */

#include <iostream>
#include <string>
#include <thread>
#include <vector>
#include <boost/filesystem.hpp>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include "steganography.hpp"
#include "exceptions.hpp"

#ifndef LEAST_SIGNIFICANT_BIT_HPP
#define LEAST_SIGNIFICANT_BIT_HPP

class LeastSignificantBit : public Steganography
{
    public:
        LeastSignificantBit(const boost::filesystem::path &image_path) : Steganography(image_path)
        {
            this->image_capacity = (this->image.rows * this->image.cols * this->image.channels()) - 64;
        }

        void Encode(const boost::filesystem::path &);

        void Decode();

    private:
        int image_capacity;

        void EncodeChunk(const int &start, std::vector<unsigned char>::iterator it, std::vector<unsigned char>::iterator en);

        void EncodeChunkLength(const int &start, const unsigned int &chunk_length);

        void DecodeChunk(const int start, std::vector<unsigned char>::iterator it, std::vector<unsigned char>::iterator en);

        unsigned int DecodeChunkLength(const int &start);
};

#endif // LEAST_SIGNIFICANT_BIT_HPP