main.cpp

Go to the documentation of this file.

 
#include "kcenon/pacs/core/dicom_dictionary.h"
#include "kcenon/pacs/core/dicom_file.h"
#include "kcenon/pacs/core/dicom_tag_constants.h"
#include "kcenon/pacs/encoding/transfer_syntax.h"
#include "kcenon/pacs/encoding/vr_type.h"
 
#include <algorithm>
#include <cstdint>
#include <filesystem>
#include <fstream>
#include <iomanip>
#include <iostream>
#include <sstream>
#include <string>
#include <vector>
 
namespace {
 
enum class bulk_data_mode {
    inline_base64,  
    uri,            
    exclude         
};
 
struct options {
    std::filesystem::path input_path;
    std::filesystem::path output_path;
    bool pretty_print{true};
    bool compact{false};
    bulk_data_mode bulk_mode{bulk_data_mode::exclude};
    std::string bulk_data_uri_prefix{"file://"};
    std::filesystem::path bulk_data_dir;
    std::vector<std::string> filter_tags;
    bool no_pixel{false};
    bool recursive{false};
    bool include_meta{true};
    bool verbose{false};
    bool quiet{false};
};
 
constexpr char base64_chars[] =
    "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
    "abcdefghijklmnopqrstuvwxyz"
    "0123456789+/";
 
[[nodiscard]] std::string to_base64(std::span<const uint8_t> data) {
    std::string result;
    result.reserve(((data.size() + 2) / 3) * 4);
 
    size_t i = 0;
    while (i < data.size()) {
        uint32_t octet_a = i < data.size() ? data[i++] : 0;
        uint32_t octet_b = i < data.size() ? data[i++] : 0;
        uint32_t octet_c = i < data.size() ? data[i++] : 0;
 
        uint32_t triple = (octet_a << 16) | (octet_b << 8) | octet_c;
 
        result += base64_chars[(triple >> 18) & 0x3F];
        result += base64_chars[(triple >> 12) & 0x3F];
        result += (i > data.size() + 1) ? '=' : base64_chars[(triple >> 6) & 0x3F];
        result += (i > data.size()) ? '=' : base64_chars[triple & 0x3F];
    }
 
    return result;
}
 
[[nodiscard]] std::string json_escape(const std::string& str) {
    std::ostringstream oss;
    for (unsigned char c : str) {
        switch (c) {
            case '"':
                oss << "\\\"";
                break;
            case '\\':
                oss << "\\\\";
                break;
            case '\b':
                oss << "\\b";
                break;
            case '\f':
                oss << "\\f";
                break;
            case '\n':
                oss << "\\n";
                break;
            case '\r':
                oss << "\\r";
                break;
            case '\t':
                oss << "\\t";
                break;
            default:
                if (c < 0x20) {
                    oss << "\\u" << std::hex << std::setfill('0') << std::setw(4)
                        << static_cast<int>(c);
                } else {
                    oss << c;
                }
        }
    }
    return oss.str();
}
 
[[nodiscard]] std::string format_tag_key(const kcenon::pacs::core::dicom_tag& tag) {
    std::ostringstream oss;
    oss << std::hex << std::uppercase << std::setfill('0')
        << std::setw(4) << tag.group()
        << std::setw(4) << tag.element();
    return oss.str();
}
 
[[nodiscard]] bool is_bulk_data_vr(kcenon::pacs::encoding::vr_type vr) {
    using namespace kcenon::pacs::encoding;
    return vr == vr_type::OB || vr == vr_type::OD || vr == vr_type::OF ||
           vr == vr_type::OL || vr == vr_type::OV || vr == vr_type::OW ||
           vr == vr_type::UN;
}
 
[[nodiscard]] bool is_pixel_data_tag(const kcenon::pacs::core::dicom_tag& tag) {
    return tag.group() == 0x7FE0 && tag.element() == 0x0010;
}
 
void print_usage(const char* program_name) {
    std::cout << R"(
DICOM to JSON Converter (DICOM PS3.18)
 
Usage: )" << program_name
              << R"( <dicom-file> [output-file] [options]
 
Arguments:
  dicom-file        Input DICOM file or directory
  output-file       Output JSON file (optional, stdout if omitted)
 
Options:
  -h, --help              Show this help message
  -p, --pretty            Pretty-print formatting (default)
  -c, --compact           Compact output (no formatting)
  --bulk-data <mode>      Binary data handling:
                            inline  - Include as Base64 (InlineBinary)
                            uri     - Save to file, reference by URI
                            exclude - Completely exclude (default)
  --bulk-data-uri <pfx>   BulkDataURI prefix (default: file://)
  --bulk-data-dir <dir>   Directory for bulk data files
  -t, --tag <tag>         Output specific tag only (e.g., 0010,0010)
  --no-pixel              Exclude pixel data
  --no-meta               Exclude File Meta Information
  -r, --recursive         Process directories recursively
  -v, --verbose           Verbose output
  -q, --quiet             Quiet mode (errors only)
 
Examples:
  )" << program_name
              << R"( image.dcm
  )" << program_name
              << R"( image.dcm output.json --pretty
  )" << program_name
              << R"( image.dcm --bulk-data inline
  )" << program_name
              << R"( image.dcm --bulk-data uri --bulk-data-dir ./bulk/
  )" << program_name
              << R"( image.dcm -t 0010,0010 -t 0010,0020
  )" << program_name
              << R"( ./dicom_folder/ --recursive --no-pixel
 
Output Format (DICOM PS3.18 JSON):
  {
    "00100010": {
      "vr": "PN",
      "Value": [{"Alphabetic": "DOE^JOHN"}]
    },
    "00100020": {
      "vr": "LO",
      "Value": ["12345678"]
    }
  }
 
Exit Codes:
  0  Success
  1  Invalid arguments
  2  File error or invalid DICOM
)";
}
 
[[nodiscard]] std::optional<kcenon::pacs::core::dicom_tag> parse_tag_string(
    const std::string& tag_str) {
    std::string s = tag_str;
    // Remove parentheses if present
    if (!s.empty() && s.front() == '(') s.erase(0, 1);
    if (!s.empty() && s.back() == ')') s.pop_back();
    // Remove comma
    s.erase(std::remove(s.begin(), s.end(), ','), s.end());
 
    if (s.length() != 8) {
        return std::nullopt;
    }
 
    try {
        uint16_t group = static_cast<uint16_t>(std::stoul(s.substr(0, 4), nullptr, 16));
        uint16_t elem = static_cast<uint16_t>(std::stoul(s.substr(4, 4), nullptr, 16));
        return kcenon::pacs::core::dicom_tag{group, elem};
    } catch (...) {
        return std::nullopt;
    }
}
 
bool parse_arguments(int argc, char* argv[], options& opts) {
    if (argc < 2) {
        return false;
    }
 
    for (int i = 1; i < argc; ++i) {
        std::string arg = argv[i];
 
        if (arg == "--help" || arg == "-h") {
            return false;
        } else if (arg == "--pretty" || arg == "-p") {
            opts.pretty_print = true;
            opts.compact = false;
        } else if (arg == "--compact" || arg == "-c") {
            opts.compact = true;
            opts.pretty_print = false;
        } else if (arg == "--bulk-data" && i + 1 < argc) {
            std::string mode = argv[++i];
            if (mode == "inline") {
                opts.bulk_mode = bulk_data_mode::inline_base64;
            } else if (mode == "uri") {
                opts.bulk_mode = bulk_data_mode::uri;
            } else if (mode == "exclude") {
                opts.bulk_mode = bulk_data_mode::exclude;
            } else {
                std::cerr << "Error: Unknown bulk-data mode '" << mode
                          << "'. Use: inline, uri, exclude\n";
                return false;
            }
        } else if (arg == "--bulk-data-uri" && i + 1 < argc) {
            opts.bulk_data_uri_prefix = argv[++i];
        } else if (arg == "--bulk-data-dir" && i + 1 < argc) {
            opts.bulk_data_dir = argv[++i];
        } else if ((arg == "--tag" || arg == "-t") && i + 1 < argc) {
            opts.filter_tags.push_back(argv[++i]);
        } else if (arg == "--no-pixel") {
            opts.no_pixel = true;
        } else if (arg == "--no-meta") {
            opts.include_meta = false;
        } else if (arg == "--recursive" || arg == "-r") {
            opts.recursive = true;
        } else if (arg == "--verbose" || arg == "-v") {
            opts.verbose = true;
        } else if (arg == "--quiet" || arg == "-q") {
            opts.quiet = true;
        } else if (arg[0] == '-') {
            std::cerr << "Error: Unknown option '" << arg << "'\n";
            return false;
        } else if (opts.input_path.empty()) {
            opts.input_path = arg;
        } else if (opts.output_path.empty()) {
            opts.output_path = arg;
        } else {
            std::cerr << "Error: Too many arguments\n";
            return false;
        }
    }
 
    if (opts.input_path.empty()) {
        std::cerr << "Error: No input file specified\n";
        return false;
    }
 
    if (opts.quiet) {
        opts.verbose = false;
    }
 
    return true;
}
 
[[nodiscard]] bool should_include_tag(const kcenon::pacs::core::dicom_tag& tag,
                                       const options& opts) {
    // Exclude pixel data if requested
    if (opts.no_pixel && is_pixel_data_tag(tag)) {
        return false;
    }
 
    // If filter tags specified, only include those
    if (!opts.filter_tags.empty()) {
        for (const auto& filter : opts.filter_tags) {
            auto parsed = parse_tag_string(filter);
            if (parsed && *parsed == tag) {
                return true;
            }
        }
        return false;
    }
 
    return true;
}
 
// Forward declaration
void write_dataset_json(std::ostream& out,
                        const kcenon::pacs::core::dicom_dataset& dataset,
                        const options& opts,
                        const std::filesystem::path& base_path,
                        int indent_level,
                        size_t& bulk_data_counter);
 
void write_element_value_json(std::ostream& out,
                               const kcenon::pacs::core::dicom_element& element,
                               const options& opts,
                               const std::filesystem::path& base_path,
                               int indent_level,
                               size_t& bulk_data_counter) {
    using namespace kcenon::pacs::encoding;
 
    const std::string indent = opts.compact ? "" : std::string(indent_level * 2, ' ');
    const std::string newline = opts.compact ? "" : "\n";
    const std::string space = opts.compact ? "" : " ";
 
    auto vr = element.vr();
    auto vr_str = to_string(vr);
 
    // Handle sequences
    if (element.is_sequence()) {
        out << "{" << newline;
        out << indent << "  \"vr\":" << space << "\"SQ\"";
 
        const auto& items = element.sequence_items();
        if (!items.empty()) {
            out << "," << newline << indent << "  \"Value\":" << space << "[" << newline;
            for (size_t i = 0; i < items.size(); ++i) {
                out << indent << "    ";
                write_dataset_json(out, items[i], opts, base_path,
                                   indent_level + 2, bulk_data_counter);
                if (i < items.size() - 1) {
                    out << ",";
                }
                out << newline;
            }
            out << indent << "  ]" << newline;
        } else {
            out << newline;
        }
        out << indent << "}";
        return;
    }
 
    // Handle empty elements
    if (element.is_empty()) {
        out << "{" << newline;
        out << indent << "  \"vr\":" << space << "\"" << vr_str << "\"" << newline;
        out << indent << "}";
        return;
    }
 
    // Handle bulk data (binary VRs)
    if (is_bulk_data_vr(vr)) {
        out << "{" << newline;
        out << indent << "  \"vr\":" << space << "\"" << vr_str << "\"";
 
        switch (opts.bulk_mode) {
            case bulk_data_mode::inline_base64: {
                auto data = element.raw_data();
                out << "," << newline << indent << "  \"InlineBinary\":" << space
                    << "\"" << to_base64(data) << "\"" << newline;
                break;
            }
            case bulk_data_mode::uri: {
                // Generate bulk data file
                std::string filename = "bulk_" + std::to_string(bulk_data_counter++) + ".raw";
                std::filesystem::path bulk_path = opts.bulk_data_dir.empty()
                    ? base_path / filename
                    : opts.bulk_data_dir / filename;
 
                // Write bulk data to file
                std::ofstream bulk_file(bulk_path, std::ios::binary);
                if (bulk_file) {
                    auto data = element.raw_data();
                    bulk_file.write(reinterpret_cast<const char*>(data.data()),
                                    static_cast<std::streamsize>(data.size()));
                }
 
                std::string uri = opts.bulk_data_uri_prefix + bulk_path.string();
                out << "," << newline << indent << "  \"BulkDataURI\":" << space
                    << "\"" << json_escape(uri) << "\"" << newline;
                break;
            }
            case bulk_data_mode::exclude:
            default:
                out << newline;
                break;
        }
        out << indent << "}";
        return;
    }
 
    // Handle PersonName (PN) VR specially per PS3.18
    if (vr == vr_type::PN) {
        out << "{" << newline;
        out << indent << "  \"vr\":" << space << "\"PN\"," << newline;
        out << indent << "  \"Value\":" << space << "[" << newline;
 
        auto result = element.as_string();
        if (result.is_ok()) {
            std::string value = result.value();
            // Split by backslash for VM > 1
            std::vector<std::string> names;
            std::stringstream ss(value);
            std::string name;
            while (std::getline(ss, name, '\\')) {
                names.push_back(name);
            }
 
            for (size_t i = 0; i < names.size(); ++i) {
                out << indent << "    {" << newline;
                out << indent << "      \"Alphabetic\":" << space
                    << "\"" << json_escape(names[i]) << "\"" << newline;
                out << indent << "    }";
                if (i < names.size() - 1) {
                    out << ",";
                }
                out << newline;
            }
        }
 
        out << indent << "  ]" << newline;
        out << indent << "}";
        return;
    }
 
    // Handle string VRs
    if (is_string_vr(vr)) {
        out << "{" << newline;
        out << indent << "  \"vr\":" << space << "\"" << vr_str << "\"," << newline;
        out << indent << "  \"Value\":" << space << "[";
 
        auto result = element.as_string();
        if (result.is_ok()) {
            std::string value = result.value();
            // Split by backslash for VM > 1
            std::vector<std::string> values;
            std::stringstream ss(value);
            std::string v;
            while (std::getline(ss, v, '\\')) {
                values.push_back(v);
            }
 
            for (size_t i = 0; i < values.size(); ++i) {
                if (i > 0) out << ",";
                out << "\"" << json_escape(values[i]) << "\"";
            }
        }
 
        out << "]" << newline;
        out << indent << "}";
        return;
    }
 
    // Handle numeric VRs
    if (is_numeric_vr(vr)) {
        out << "{" << newline;
        out << indent << "  \"vr\":" << space << "\"" << vr_str << "\"," << newline;
        out << indent << "  \"Value\":" << space << "[";
 
        auto write_numeric_values = [&]<typename T>() {
            auto result = element.as_numeric_list<T>();
            if (result.is_ok()) {
                const auto& values = result.value();
                for (size_t i = 0; i < values.size(); ++i) {
                    if (i > 0) out << ",";
                    if constexpr (std::is_floating_point_v<T>) {
                        out << std::setprecision(17) << values[i];
                    } else {
                        out << static_cast<int64_t>(values[i]);
                    }
                }
            }
        };
 
        switch (vr) {
            case vr_type::US:
                write_numeric_values.template operator()<uint16_t>();
                break;
            case vr_type::SS:
                write_numeric_values.template operator()<int16_t>();
                break;
            case vr_type::UL:
                write_numeric_values.template operator()<uint32_t>();
                break;
            case vr_type::SL:
                write_numeric_values.template operator()<int32_t>();
                break;
            case vr_type::FL:
                write_numeric_values.template operator()<float>();
                break;
            case vr_type::FD:
                write_numeric_values.template operator()<double>();
                break;
            case vr_type::UV:
                write_numeric_values.template operator()<uint64_t>();
                break;
            case vr_type::SV:
                write_numeric_values.template operator()<int64_t>();
                break;
            default:
                // Fallback: try as string
                if (auto result = element.as_string(); result.is_ok()) {
                    out << "\"" << json_escape(result.value()) << "\"";
                }
                break;
        }
 
        out << "]" << newline;
        out << indent << "}";
        return;
    }
 
    // Handle AT (Attribute Tag) VR
    if (vr == vr_type::AT) {
        out << "{" << newline;
        out << indent << "  \"vr\":" << space << "\"AT\"," << newline;
        out << indent << "  \"Value\":" << space << "[";
 
        auto data = element.raw_data();
        for (size_t i = 0; i + 4 <= data.size(); i += 4) {
            if (i > 0) out << ",";
            uint16_t group = static_cast<uint16_t>(data[i] | (data[i + 1] << 8));
            uint16_t elem = static_cast<uint16_t>(data[i + 2] | (data[i + 3] << 8));
            out << "\"" << std::hex << std::uppercase << std::setfill('0')
                << std::setw(4) << group << std::setw(4) << elem << "\"";
        }
 
        out << "]" << newline << std::dec;
        out << indent << "}";
        return;
    }
 
    // Fallback: treat as string
    out << "{" << newline;
    out << indent << "  \"vr\":" << space << "\"" << vr_str << "\"," << newline;
    out << indent << "  \"Value\":" << space << "[";
 
    if (auto result = element.as_string(); result.is_ok()) {
        out << "\"" << json_escape(result.value()) << "\"";
    }
 
    out << "]" << newline;
    out << indent << "}";
}
 
void write_dataset_json(std::ostream& out,
                        const kcenon::pacs::core::dicom_dataset& dataset,
                        const options& opts,
                        const std::filesystem::path& base_path,
                        int indent_level,
                        size_t& bulk_data_counter) {
    const std::string indent = opts.compact ? "" : std::string(indent_level * 2, ' ');
    const std::string newline = opts.compact ? "" : "\n";
    const std::string space = opts.compact ? "" : " ";
 
    out << "{" << newline;
 
    bool first = true;
    for (const auto& [tag, element] : dataset) {
        if (!should_include_tag(tag, opts)) {
            continue;
        }
 
        if (!first) {
            out << "," << newline;
        }
        first = false;
 
        out << indent << "  \"" << format_tag_key(tag) << "\":" << space;
        write_element_value_json(out, element, opts, base_path,
                                  indent_level + 1, bulk_data_counter);
    }
 
    out << newline << indent << "}";
}
 
int convert_file(const std::filesystem::path& input_path,
                 std::ostream& output,
                 const options& opts) {
    using namespace kcenon::pacs::core;
 
    auto result = dicom_file::open(input_path);
    if (result.is_err()) {
        std::cerr << "Error: Failed to open '" << input_path.string()
                  << "': " << result.error().message << "\n";
        return 2;
    }
 
    auto& file = result.value();
    size_t bulk_data_counter = 0;
    std::filesystem::path base_path = input_path.parent_path();
 
    const std::string newline = opts.compact ? "" : "\n";
    const std::string space = opts.compact ? "" : " ";
 
    output << "{" << newline;
 
    bool first = true;
 
    // Include File Meta Information if requested
    if (opts.include_meta && !file.meta_information().empty()) {
        first = false;
        output << "  \"00020000\":" << space;
        write_dataset_json(output, file.meta_information(), opts, base_path, 1, bulk_data_counter);
    }
 
    // Write main dataset
    for (const auto& [tag, element] : file.dataset()) {
        if (!should_include_tag(tag, opts)) {
            continue;
        }
 
        if (!first) {
            output << "," << newline;
        }
        first = false;
 
        const std::string indent = opts.compact ? "" : "  ";
        output << indent << "\"" << format_tag_key(tag) << "\":" << space;
        write_element_value_json(output, element, opts, base_path, 1, bulk_data_counter);
    }
 
    output << newline << "}" << newline;
 
    return 0;
}
 
int process_directory(const std::filesystem::path& dir_path, const options& opts) {
    int exit_code = 0;
 
    auto process = [&](const std::filesystem::path& file_path) {
        auto ext = file_path.extension().string();
        std::transform(ext.begin(), ext.end(), ext.begin(), ::tolower);
        if (ext != ".dcm" && ext != ".dicom" && !ext.empty()) {
            return;
        }
 
        // Generate output filename
        std::filesystem::path output_path = file_path;
        output_path.replace_extension(".json");
 
        if (!opts.quiet) {
            std::cout << "Converting: " << file_path.string() << " -> "
                      << output_path.string() << "\n";
        }
 
        std::ofstream output(output_path);
        if (!output) {
            std::cerr << "Error: Cannot create output file: " << output_path.string() << "\n";
            exit_code = 2;
            return;
        }
 
        if (convert_file(file_path, output, opts) != 0) {
            exit_code = 2;
        }
    };
 
    if (opts.recursive) {
        for (const auto& entry :
             std::filesystem::recursive_directory_iterator(dir_path)) {
            if (entry.is_regular_file()) {
                process(entry.path());
            }
        }
    } else {
        for (const auto& entry : std::filesystem::directory_iterator(dir_path)) {
            if (entry.is_regular_file()) {
                process(entry.path());
            }
        }
    }
 
    return exit_code;
}
 
}  // namespace
 
int main(int argc, char* argv[]) {
    options opts;
 
    if (!parse_arguments(argc, argv, opts)) {
        std::cout << R"(
  ____   ____ __  __   _____  ___        _ ____   ___  _   _
 |  _ \ / ___|  \/  | |_   _|/ _ \      | / ___| / _ \| \ | |
 | | | | |   | |\/| |   | | | | | |  _  | \___ \| | | |  \| |
 | |_| | |___| |  | |   | | | |_| | | |_| |___) | |_| | |\  |
 |____/ \____|_|  |_|   |_|  \___/   \___/|____/ \___/|_| \_|
 
        DICOM to JSON Converter (PS3.18)
)" << "\n";
        print_usage(argv[0]);
        return 1;
    }
 
    // Check if input path exists
    if (!std::filesystem::exists(opts.input_path)) {
        std::cerr << "Error: Path does not exist: " << opts.input_path.string() << "\n";
        return 2;
    }
 
    // Create bulk data directory if needed
    if (opts.bulk_mode == bulk_data_mode::uri && !opts.bulk_data_dir.empty()) {
        std::filesystem::create_directories(opts.bulk_data_dir);
    }
 
    // Show banner in non-quiet mode
    if (!opts.quiet) {
        std::cout << R"(
  ____   ____ __  __   _____  ___        _ ____   ___  _   _
 |  _ \ / ___|  \/  | |_   _|/ _ \      | / ___| / _ \| \ | |
 | | | | |   | |\/| |   | | | | | |  _  | \___ \| | | |  \| |
 | |_| | |___| |  | |   | | | |_| | | |_| |___) | |_| | |\  |
 |____/ \____|_|  |_|   |_|  \___/   \___/|____/ \___/|_| \_|
 
        DICOM to JSON Converter (PS3.18)
)" << "\n";
    }
 
    // Handle directory vs file
    if (std::filesystem::is_directory(opts.input_path)) {
        return process_directory(opts.input_path, opts);
    } else {
        // Single file
        if (opts.output_path.empty()) {
            // Output to stdout
            return convert_file(opts.input_path, std::cout, opts);
        } else {
            // Output to file
            std::ofstream output(opts.output_path);
            if (!output) {
                std::cerr << "Error: Cannot create output file: "
                          << opts.output_path.string() << "\n";
                return 2;
            }
            return convert_file(opts.input_path, output, opts);
        }
    }
}