hd3gtv.mydmam.transcode.images.ImageMagickAnalyser.java Source code

Java tutorial

Introduction

Here is the source code for hd3gtv.mydmam.transcode.images.ImageMagickAnalyser.java

Source

/*
 * This file is part of MyDMAM.
 * 
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * 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 Lesser General Public License for more details.
 * 
 * Copyright (C) hdsdi3g for hd3g.tv 2015
 * 
*/
package hd3gtv.mydmam.transcode.images;

import hd3gtv.configuration.Configuration;
import hd3gtv.log2.Log2;
import hd3gtv.log2.Log2Dump;
import hd3gtv.mydmam.metadata.MetadataGeneratorAnalyser;
import hd3gtv.mydmam.metadata.container.Container;
import hd3gtv.mydmam.metadata.container.EntryAnalyser;
import hd3gtv.mydmam.metadata.container.ContainerOperations;
import hd3gtv.tools.ExecprocessBadExecutionException;
import hd3gtv.tools.ExecprocessGettext;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import com.google.gson.JsonObject;
import com.google.gson.JsonParser;

public class ImageMagickAnalyser implements MetadataGeneratorAnalyser {

    static final ArrayList<String> mimetype_list;
    static final ArrayList<String> convert_limits_params;

    static {
        mimetype_list = new ArrayList<String>();
        mimetype_list.add("image/jpeg");
        mimetype_list.add("image/png");
        mimetype_list.add("image/bmp");
        mimetype_list.add("image/gif");
        mimetype_list.add("image/vnd.adobe.photoshop");
        mimetype_list.add("image/tiff");
        mimetype_list.add("image/svg+xml");
        mimetype_list.add("application/postscript");
        mimetype_list.add("image/jp2");
        mimetype_list.add("application/dicom");
        mimetype_list.add("image/x-icon");
        mimetype_list.add("image/pict");
        mimetype_list.add("image/vndwapwbmp");
        mimetype_list.add("image/x-pcx");
        mimetype_list.add("image/x-portable-bitmap");
        mimetype_list.add("image/x-xbm");
        mimetype_list.add("image/xpm");
        mimetype_list.add("image/cineon");
        mimetype_list.add("image/dpx");
        mimetype_list.add("image/tga");
        mimetype_list.add("image/exr");
        mimetype_list.add("image/vnd.radiance");
        mimetype_list.add("image/webp");
        mimetype_list.add("image/sgi");
        mimetype_list.add("image/x-palm-pixmap");
        mimetype_list.add("image/x-g3-fax");
        mimetype_list.add("image/jpcd");
        mimetype_list.add("image/x-sct");
        mimetype_list.add("image/jbig");
        mimetype_list.add("image/x-miff");
        mimetype_list.add("image/x-sun");

        convert_limits_params = new ArrayList<String>(4);
        if (Configuration.global.isElementExists("imagemagick_limits")) {
            String memory = Configuration.global.getValue("imagemagick_limits", "memory", null);
            if (memory != null) {
                convert_limits_params.add("-limit");
                convert_limits_params.add("memory");
                convert_limits_params.add(memory);
            }

            String disk = Configuration.global.getValue("imagemagick_limits", "disk", null);
            if (disk != null) {
                convert_limits_params.add("-limit");
                convert_limits_params.add("disk");
                convert_limits_params.add(disk);
            }

            String file = Configuration.global.getValue("imagemagick_limits", "file", null);
            if (file != null) {
                convert_limits_params.add("-limit");
                convert_limits_params.add("file");
                convert_limits_params.add(file);
            }

            String time = Configuration.global.getValue("imagemagick_limits", "time", null);
            if (time != null) {
                convert_limits_params.add("-limit");
                convert_limits_params.add("time");
                convert_limits_params.add(time);
            }
        }
    }

    private String convert_bin;

    public ImageMagickAnalyser() {
        convert_bin = Configuration.global.getValue("transcoding", "convert_bin", "convert");
    }

    public EntryAnalyser process(Container container) throws Exception {
        ArrayList<String> param = new ArrayList<String>();
        ExecprocessGettext process = null;
        try {
            param.addAll(convert_limits_params);
            param.add(container.getPhysicalSource().getPath() + "[0]");
            param.add("json:-");

            process = new ExecprocessGettext(convert_bin, param);
            process.setEndlinewidthnewline(true);
            process.start();

            JsonParser p = new JsonParser();
            JsonObject result = p.parse(process.getResultstdout().toString()).getAsJsonObject();
            result = result.get("image").getAsJsonObject();

            if (result.has("profiles")) {
                JsonObject jo_profiles = result.get("profiles").getAsJsonObject();
                if (jo_profiles.has("iptc")) {
                    /**
                     * Import and inject IPTC
                     */
                    param.clear();
                    param.addAll(convert_limits_params);
                    param.add(container.getPhysicalSource().getPath() + "[0]");
                    param.add("iptctext:-");
                    process = new ExecprocessGettext(convert_bin, param);
                    process.setEndlinewidthnewline(true);
                    process.setExitcodemusttobe0(false);
                    process.start();

                    if (process.getRunprocess().getExitvalue() == 0) {
                        if (result.has("properties")) {
                            ImageAttributes.injectIPTCInProperties(process.getResultstdout().toString(),
                                    result.get("properties").getAsJsonObject());
                        } else {
                            JsonObject jo_properties = new JsonObject();
                            result.add("properties", jo_properties);
                            ImageAttributes.injectIPTCInProperties(process.getResultstdout().toString(),
                                    jo_properties);
                        }
                    }
                }
                result.remove("profiles");
            }

            result.remove("artifacts");
            result.remove("name");

            ImageAttributes ia = ContainerOperations.getGson().fromJson(result, ImageAttributes.class);
            container.getSummary().putSummaryContent(ia, ia.createSummary());

            return ia;
        } catch (IOException e) {
            if (e instanceof ExecprocessBadExecutionException) {
                Log2Dump dump = new Log2Dump();
                dump.add("param", param);
                if (process != null) {
                    dump.add("stdout", process.getResultstdout().toString().trim());
                    dump.add("stderr", process.getResultstderr().toString().trim());
                    dump.add("exitcode", process.getRunprocess().getExitvalue());
                }
                Log2.log.error("Problem with convert", null, dump);
            }
            throw e;
        }
    }

    public boolean canProcessThis(String mimetype) {
        return mimetype_list.contains(mimetype);
    }

    public boolean isEnabled() {
        return (new File(convert_bin)).exists();
    }

    public String getLongName() {
        return "ImageMagick Analyser";
    }

    public List<String> getMimeFileListCanUsedInMasterAsPreview() {
        return null;
    }

    public boolean isCanUsedInMasterAsPreview(Container container) {
        return false;
    }

    public Class<? extends EntryAnalyser> getRootEntryClass() {
        return ImageAttributes.class;
    }
}