DicomAnonymizer.hpp

/* ***** BEGIN LICENSE BLOCK *****
 * FW4SPL - Copyright (C) IRCAD, 2009-2017.
 * Distributed under the terms of the GNU Lesser General Public License (LGPL) as
 * published by the Free Software Foundation.
 * ****** END LICENSE BLOCK ****** */

#ifndef __FWGDCMIO_HELPER_DICOMANONYMIZER_HPP__
#define __FWGDCMIO_HELPER_DICOMANONYMIZER_HPP__

#include "fwGdcmIO/config.hpp"
#include "fwGdcmIO/helper/tags.hpp"

#include <fwZip/WriteZipArchive.hpp>

#include <boost/date_time/gregorian/gregorian_types.hpp>
#include <boost/filesystem/path.hpp>

#include <gdcmAnonymizer.h>
#include <gdcmDicts.h>
#include <gdcmStringFilter.h>

#include <iostream>
#include <map>
#include <set>
#include <string>

namespace fwJobs
{
class IJob;
class Observer;
}

namespace fwGdcmIO
{
namespace helper
{

class FWGDCMIO_CLASS_API DicomAnonymizer
{
public:

    fwCoreClassFactoryMacro((DicomAnonymizer), (()), new DicomAnonymizer );

    FWGDCMIO_API DicomAnonymizer();

    FWGDCMIO_API virtual ~DicomAnonymizer();

    typedef std::map< std::string, std::string > UIDMap;

    typedef std::map< ::gdcm::Tag, std::string > ExceptionTagMapType;

    FWGDCMIO_API void anonymize(const ::boost::filesystem::path& dirPath);

    FWGDCMIO_API void anonymize(std::istream& inputStream, std::ostream& outputStream);

    FWGDCMIO_API void addExceptionTag(uint16_t group, uint16_t element, const std::string& value = "");

    FWGDCMIO_API static void copyDirectory(const ::boost::filesystem::path& input,
                                           const ::boost::filesystem::path& output);

    FWGDCMIO_API SPTR(::fwJobs::IJob) getJob() const;

    FWGDCMIO_API unsigned int getNextIndex();

    FWGDCMIO_API void resetIndex();

    FWGDCMIO_API void removeAnonymizeTag(const ::gdcm::Tag& tag);

    FWGDCMIO_API void setReferenceDate(const ::boost::gregorian::date& referenceDate);

    FWGDCMIO_API void addShiftDateTag(const ::gdcm::Tag& tag);

    FWGDCMIO_API void preservePrivateTag(const ::gdcm::Tag& tag);

    typedef std::set< ::gdcm::Tag > TagContainerType;
    FWGDCMIO_API const TagContainerType& getActionCodeDTags();
    FWGDCMIO_API const TagContainerType& getActionCodeZTags();
    FWGDCMIO_API const TagContainerType& getActionCodeXTags();
    FWGDCMIO_API const TagContainerType& getActionCodeKTags();
    FWGDCMIO_API const TagContainerType& getActionCodeCTags();
    FWGDCMIO_API const TagContainerType& getActionCodeUTags();
private:

    void anonymizationProcess(const ::boost::filesystem::path& dirPath);

    void applyActionCodeD(const ::gdcm::Tag& tag);

    void applyActionCodeZ(const ::gdcm::Tag& tag);

    void applyActionCodeX(const ::gdcm::Tag& tag);

    void applyActionCodeK(const ::gdcm::Tag& tag);

    void applyActionCodeC(const ::gdcm::Tag& tag);

    void applyActionCodeU(const ::gdcm::Tag& tag);

    void applyActionShiftDate(const ::gdcm::Tag& tag);

    void generateDummyValue(const ::gdcm::Tag& tag);

    ::gdcm::Anonymizer m_anonymizer;

    ::gdcm::StringFilter m_stringFilter;

    const ::gdcm::Dict& m_publicDictionary;

    UIDMap m_uidMap;

    ExceptionTagMapType m_exceptionTagMap;

    static const unsigned int s_NUMBER_OF_TAGS;

    SPTR(::fwJobs::Observer) m_observer;

    bool m_archiving;

    unsigned int m_fileIndex;

    ::boost::gregorian::date m_referenceDate;

    TagContainerType m_actionCodeDTags;
    TagContainerType m_actionCodeZTags;
    TagContainerType m_actionCodeXTags;
    TagContainerType m_actionCodeKTags;
    TagContainerType m_actionCodeCTags;
    TagContainerType m_actionCodeUTags;
    TagContainerType m_actionShiftDateTags;

    ::fwGdcmIO::helper::PrivateTagVecType m_privateTags;
};

} // namespace helper
} // namespace fwGdcmIO

#endif /*__FWGDCMIO_HELPER_DICOMANONYMIZER_HPP__*/