exiv2/samples/crwedit.cpp

// ***************************************************************** -*- C++ -*-
// crwedit.cpp
// Print the CIFF structure of a CRW file

#include "crwimage.hpp"
#include "crwimage_int.hpp"
#include "futils.hpp"

#include <cstring>
#include <iostream>
#include <string>

void remove(Exiv2::Internal::CiffHeader* pHead);
void add(Exiv2::Internal::CiffHeader* pHead);
void help();
void write(const std::string& filename, const Exiv2::Internal::CiffHeader* pHead);

int main(int argc, char* const argv[]) try {
    if (argc != 2) {
        std::cout << "Usage: " << argv[0] << " file\n";
        std::cout << "Edit the CIFF structure of a CRW file\n";
        return 1;
    }

    std::string filename(argv[1]);
    Exiv2::FileIo io(filename);
    if (io.open() != 0) {
        throw Exiv2::Error(Exiv2::kerDataSourceOpenFailed, io.path(), Exiv2::strError());
    }
    Exiv2::IoCloser closer(io);

    // Ensure that this is a CRW image
    if (!Exiv2::isCrwType(io, false)) {
        if (io.error() || io.eof())
            throw Exiv2::Error(Exiv2::kerFailedToReadImageData);
        throw Exiv2::Error(Exiv2::kerNotACrwImage);
    }

    // Read the image into a memory buffer
    long len = (long)io.size();
    Exiv2::DataBuf buf(len);
    io.read(buf.pData_, len);
    if (io.error() || io.eof())
        throw Exiv2::Error(Exiv2::kerFailedToReadImageData);

    // Parse the image, starting with a CIFF header component
    Exiv2::Internal::CiffHeader::AutoPtr parseTree(new Exiv2::Internal::CiffHeader);
    parseTree->read(buf.pData_, buf.size_);

    // Allow user to make changes
    bool go = true;
    while (go) {
        char cmd;
        std::cout << "command> ";
        std::cin >> cmd;
        switch (cmd) {
            case 'q':
                go = false;
                break;
            case 'p':
                parseTree->print(std::cout);
                break;
            case 'a':
                add(parseTree.get());
                break;
            case 'd':
                remove(parseTree.get());
                break;
            case 'w':
                write(filename, parseTree.get());
                break;
            case 'h':
                help();
                break;
        }
    }

    return 0;
} catch (Exiv2::AnyError& e) {
    std::cerr << e << "\n";
    return -1;
}

void write(const std::string& filename, const Exiv2::Internal::CiffHeader* pHead)
{
    Exiv2::Blob blob;
    pHead->write(blob);

    Exiv2::FileIo io(filename);
    if (io.open("wb") != 0) {
        throw Exiv2::Error(Exiv2::kerDataSourceOpenFailed, io.path(), Exiv2::strError());
    }
    Exiv2::IoCloser closer(io);
    long ret = io.write(&blob[0], (long)blob.size());
    if (static_cast<size_t>(ret) != blob.size())
        throw Exiv2::Error(Exiv2::kerImageWriteFailed);
    io.close();
}

void remove(Exiv2::Internal::CiffHeader* pHead)
{
    uint16_t crwTag, crwDir;
    std::cout << "crwTag> 0x";
    std::cin >> std::hex >> crwTag;
    std::cout << "crwDir> 0x";
    std::cin >> std::hex >> crwDir;
    std::cout << "Deleting tag 0x" << std::hex << crwTag << " in dir 0x" << crwDir << ", ok? ";
    char cmd;
    std::cin >> cmd;
    if (cmd != 'n' && cmd != 'N') {
        pHead->remove(crwTag, crwDir);
    } else {
        std::cout << "Canceled.\n";
    }
}

void add(Exiv2::Internal::CiffHeader* pHead)
{
    uint16_t crwTag, crwDir;
    uint32_t size;
    std::cout << "crwTag> 0x";
    std::cin >> std::hex >> crwTag;
    std::cout << "crwDir> 0x";
    std::cin >> std::hex >> crwDir;
    std::cout << "size> ";
    std::cin >> std::dec >> size;
    std::cout << "Adding tag 0x" << std::hex << crwTag << " in dir 0x" << crwDir << ", " << size << " bytes, ok? ";
    char cmd;
    std::cin >> cmd;
    if (cmd != 'n' && cmd != 'N') {
        Exiv2::DataBuf buf(size);
        std::memset(buf.pData_, 0xaa, size);
        pHead->add(crwTag, crwDir, buf);
    } else {
        std::cout << "Canceled.\n";
    }
}

void help()
{
    std::cout << "a: add tag, d: delete tag, p: print tags, w: write file, q: quit\n";
}