ViacheslavKlimov
3 years ago
29 changed files with 444 additions and 289 deletions
@ -0,0 +1,29 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data; |
||||
|
|
||||
|
import com.fasterxml.jackson.annotation.JsonInclude; |
||||
|
import lombok.Data; |
||||
|
|
||||
|
@Data |
||||
|
@JsonInclude(JsonInclude.Include.NON_NULL) |
||||
|
public class ImageDescriptor { |
||||
|
private String mediaType; |
||||
|
private int width; |
||||
|
private int height; |
||||
|
private long size; |
||||
|
private ImageDescriptor previewDescriptor; |
||||
|
} |
||||
@ -1,96 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2023 The Thingsboard Authors |
|
||||
* |
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
||||
* you may not use this file except in compliance with the License. |
|
||||
* You may obtain a copy of the License at |
|
||||
* |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* |
|
||||
* Unless required by applicable law or agreed to in writing, software |
|
||||
* distributed under the License is distributed on an "AS IS" BASIS, |
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
||||
* See the License for the specific language governing permissions and |
|
||||
* limitations under the License. |
|
||||
*/ |
|
||||
package org.thingsboard.server.common.data.util; |
|
||||
|
|
||||
import lombok.AccessLevel; |
|
||||
import lombok.Data; |
|
||||
import lombok.NoArgsConstructor; |
|
||||
import org.springframework.util.MimeType; |
|
||||
import org.springframework.util.MimeTypeUtils; |
|
||||
|
|
||||
import javax.imageio.ImageIO; |
|
||||
import java.awt.image.BufferedImage; |
|
||||
import java.io.ByteArrayInputStream; |
|
||||
import java.io.ByteArrayOutputStream; |
|
||||
import java.io.IOException; |
|
||||
import java.util.Map; |
|
||||
|
|
||||
@NoArgsConstructor(access = AccessLevel.PRIVATE) |
|
||||
public class ImageUtils { |
|
||||
|
|
||||
private static final Map<String, String> mediaTypeMappings = Map.of( |
|
||||
"jpeg", "jpg", |
|
||||
"svg+xml", "svg" |
|
||||
); |
|
||||
|
|
||||
public static String mediaTypeToFileExtension(String mimeType) { |
|
||||
String subtype = MimeTypeUtils.parseMimeType(mimeType).getSubtype(); |
|
||||
return mediaTypeMappings.getOrDefault(subtype, subtype); |
|
||||
} |
|
||||
|
|
||||
public static String fileExtensionToMediaType(String type, String extension) { |
|
||||
String subtype = mediaTypeMappings.entrySet().stream() |
|
||||
.filter(mapping -> mapping.getValue().equals(extension)) |
|
||||
.map(Map.Entry::getKey).findFirst().orElse(extension); |
|
||||
return new MimeType(type, subtype).toString(); |
|
||||
} |
|
||||
|
|
||||
public static ImageInfo processImage(byte[] imageData) throws IOException { |
|
||||
BufferedImage image = ImageIO.read(new ByteArrayInputStream(imageData)); |
|
||||
ImageThumbnail thumbnail = getImageThumbnail(image, 250); |
|
||||
return new ImageInfo(image.getWidth(), image.getHeight(), thumbnail); |
|
||||
} |
|
||||
|
|
||||
private static ImageThumbnail getImageThumbnail(BufferedImage originalImage, int maxDimension) throws IOException { |
|
||||
int originalWidth = originalImage.getWidth(); |
|
||||
int originalHeight = originalImage.getHeight(); |
|
||||
int thumbnailWidth; |
|
||||
int thumbnailHeight; |
|
||||
if (originalWidth <= maxDimension && originalHeight <= maxDimension) { |
|
||||
thumbnailWidth = originalWidth; |
|
||||
thumbnailHeight = originalHeight; |
|
||||
} else { |
|
||||
double aspectRatio = (double) originalWidth / originalHeight; |
|
||||
if (originalWidth > originalHeight) { |
|
||||
thumbnailWidth = maxDimension; |
|
||||
thumbnailHeight = (int) (maxDimension / aspectRatio); |
|
||||
} else { |
|
||||
thumbnailWidth = (int) (maxDimension * aspectRatio); |
|
||||
thumbnailHeight = maxDimension; |
|
||||
} |
|
||||
} |
|
||||
BufferedImage thumbnail = new BufferedImage(thumbnailWidth, thumbnailHeight, BufferedImage.TYPE_INT_RGB); |
|
||||
thumbnail.getGraphics().drawImage(originalImage, 0, 0, thumbnailWidth, thumbnailHeight, null); |
|
||||
ByteArrayOutputStream os = new ByteArrayOutputStream(); |
|
||||
ImageIO.write(thumbnail, "ignored", os); |
|
||||
return new ImageThumbnail(thumbnail.getWidth(), thumbnail.getHeight(), os.toByteArray()); |
|
||||
} |
|
||||
|
|
||||
@Data |
|
||||
public static class ImageInfo { |
|
||||
private final int width; |
|
||||
private final int height; |
|
||||
private final ImageThumbnail thumbnail; |
|
||||
} |
|
||||
|
|
||||
@Data |
|
||||
public static class ImageThumbnail { |
|
||||
private final int width; |
|
||||
private final int height; |
|
||||
private final byte[] data; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
@ -0,0 +1,144 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2023 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.dao.util; |
||||
|
|
||||
|
import lombok.AccessLevel; |
||||
|
import lombok.Data; |
||||
|
import lombok.NoArgsConstructor; |
||||
|
import org.apache.batik.transcoder.TranscoderInput; |
||||
|
import org.apache.batik.transcoder.TranscoderOutput; |
||||
|
import org.apache.batik.transcoder.image.PNGTranscoder; |
||||
|
import org.springframework.util.MimeType; |
||||
|
import org.springframework.util.MimeTypeUtils; |
||||
|
|
||||
|
import javax.imageio.ImageIO; |
||||
|
import java.awt.image.BufferedImage; |
||||
|
import java.io.ByteArrayInputStream; |
||||
|
import java.io.ByteArrayOutputStream; |
||||
|
import java.util.Map; |
||||
|
|
||||
|
@NoArgsConstructor(access = AccessLevel.PRIVATE) |
||||
|
public class ImageUtils { |
||||
|
|
||||
|
private static final Map<String, String> mediaTypeMappings = Map.of( |
||||
|
"jpeg", "jpg", |
||||
|
"svg+xml", "svg" |
||||
|
); |
||||
|
|
||||
|
public static String mediaTypeToFileExtension(String mimeType) { |
||||
|
String subtype = MimeTypeUtils.parseMimeType(mimeType).getSubtype(); |
||||
|
return mediaTypeMappings.getOrDefault(subtype, subtype); |
||||
|
} |
||||
|
|
||||
|
public static String fileExtensionToMediaType(String type, String extension) { |
||||
|
String subtype = mediaTypeMappings.entrySet().stream() |
||||
|
.filter(mapping -> mapping.getValue().equals(extension)) |
||||
|
.map(Map.Entry::getKey).findFirst().orElse(extension); |
||||
|
return new MimeType(type, subtype).toString(); |
||||
|
} |
||||
|
|
||||
|
public static ProcessedImage processImage(byte[] data, String mediaType, int thumbnailMaxDimension) throws Exception { |
||||
|
if (mediaTypeToFileExtension(mediaType).equals("svg")) { |
||||
|
return processSvgImage(data, thumbnailMaxDimension); |
||||
|
} |
||||
|
|
||||
|
BufferedImage bufferedImage = ImageIO.read(new ByteArrayInputStream(data)); |
||||
|
ProcessedImage image = new ProcessedImage(); |
||||
|
image.setMediaType(mediaType); |
||||
|
image.setWidth(bufferedImage.getWidth()); |
||||
|
image.setHeight(bufferedImage.getHeight()); |
||||
|
image.setData(data); |
||||
|
image.setSize(data.length); |
||||
|
|
||||
|
ProcessedImage preview = new ProcessedImage(); |
||||
|
int[] thumbnailDimensions = getThumbnailDimensions(image.getWidth(), image.getHeight(), thumbnailMaxDimension); |
||||
|
preview.setWidth(thumbnailDimensions[0]); |
||||
|
preview.setHeight(thumbnailDimensions[1]); |
||||
|
|
||||
|
if (preview.getWidth() == image.getWidth() && preview.getHeight() == image.getHeight()) { |
||||
|
if (mediaType.equals("image/png")) { |
||||
|
preview.setMediaType(mediaType); |
||||
|
preview.setData(null); |
||||
|
preview.setSize(data.length); |
||||
|
image.setPreview(preview); |
||||
|
return image; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
BufferedImage thumbnail = new BufferedImage(preview.getWidth(), preview.getHeight(), BufferedImage.TYPE_INT_RGB); |
||||
|
thumbnail.getGraphics().drawImage(bufferedImage, 0, 0, preview.getWidth(), preview.getHeight(), null); |
||||
|
ByteArrayOutputStream out = new ByteArrayOutputStream(); |
||||
|
ImageIO.write(thumbnail, "png", out); |
||||
|
|
||||
|
preview.setMediaType("image/png"); |
||||
|
preview.setData(out.toByteArray()); |
||||
|
preview.setSize(preview.getData().length); |
||||
|
image.setPreview(preview); |
||||
|
return image; |
||||
|
} |
||||
|
|
||||
|
public static ProcessedImage processSvgImage(byte[] data, int thumbnailMaxDimension) throws Exception { |
||||
|
ProcessedImage image = new ProcessedImage(); |
||||
|
image.setWidth(0); |
||||
|
image.setHeight(0); |
||||
|
image.setData(data); |
||||
|
image.setSize(data.length); |
||||
|
|
||||
|
PNGTranscoder transcoder = new PNGTranscoder(); |
||||
|
transcoder.addTranscodingHint(PNGTranscoder.KEY_MAX_WIDTH, (float) thumbnailMaxDimension); |
||||
|
transcoder.addTranscodingHint(PNGTranscoder.KEY_MAX_HEIGHT, (float) thumbnailMaxDimension); |
||||
|
ByteArrayOutputStream out = new ByteArrayOutputStream(); |
||||
|
transcoder.transcode(new TranscoderInput(new ByteArrayInputStream(data)), new TranscoderOutput(out)); |
||||
|
byte[] pngThumbnail = out.toByteArray(); |
||||
|
|
||||
|
ProcessedImage preview = new ProcessedImage(); |
||||
|
preview.setWidth(thumbnailMaxDimension); |
||||
|
preview.setHeight(thumbnailMaxDimension); |
||||
|
preview.setMediaType("image/png"); |
||||
|
preview.setData(pngThumbnail); |
||||
|
preview.setSize(pngThumbnail.length); |
||||
|
image.setPreview(preview); |
||||
|
return image; |
||||
|
} |
||||
|
|
||||
|
private static int[] getThumbnailDimensions(int originalWidth, int originalHeight, int maxDimension) { |
||||
|
if (originalWidth <= maxDimension && originalHeight <= maxDimension) { |
||||
|
return new int[]{originalWidth, originalHeight}; |
||||
|
} |
||||
|
int thumbnailWidth; |
||||
|
int thumbnailHeight; |
||||
|
double aspectRatio = (double) originalWidth / originalHeight; |
||||
|
if (originalWidth > originalHeight) { |
||||
|
thumbnailWidth = maxDimension; |
||||
|
thumbnailHeight = (int) (maxDimension / aspectRatio); |
||||
|
} else { |
||||
|
thumbnailWidth = (int) (maxDimension * aspectRatio); |
||||
|
thumbnailHeight = maxDimension; |
||||
|
} |
||||
|
return new int[]{thumbnailWidth, thumbnailHeight}; |
||||
|
} |
||||
|
|
||||
|
@Data |
||||
|
public static class ProcessedImage { |
||||
|
private String mediaType; |
||||
|
private int width; |
||||
|
private int height; |
||||
|
private byte[] data; |
||||
|
private long size; |
||||
|
private ProcessedImage preview; |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue