org.tomitribe.util.Size Maven / Gradle / Ivy
The newest version!
/*
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you 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.tomitribe.util;
import org.tomitribe.util.editor.Editors;
import java.beans.PropertyEditorManager;
import java.util.ArrayList;
import java.util.List;
public class Size implements Comparable {
private long size;
private SizeUnit unit;
public Size() {
}
public Size(final long size, final SizeUnit unit) {
this.size = size;
this.unit = unit;
}
public Size(final String string) {
this(string, null);
}
public Size(final String string, final SizeUnit defaultUnit) {
final String[] strings = string.split(",| and ");
Size total = new Size();
for (String s : strings) {
final Size part = new Size();
s = s.trim();
final StringBuilder t = new StringBuilder();
final StringBuilder u = new StringBuilder();
int i = 0;
// get the number
for (; i < s.length(); i++) {
final char c = s.charAt(i);
if (Character.isDigit(c) || i == 0 && c == '-' || i > 0 && c == '.') {
t.append(c);
} else {
break;
}
}
if (t.length() == 0) {
invalidFormat(s);
}
// skip whitespace
for (; i < s.length(); i++) {
final char c = s.charAt(i);
if (!Character.isWhitespace(c)) {
break;
}
}
// get time unit text part
for (; i < s.length(); i++) {
final char c = s.charAt(i);
if (Character.isLetter(c)) {
u.append(c);
} else {
invalidFormat(s);
}
}
part.unit = parseUnit(u.toString());
if (part.unit == null) {
part.unit = defaultUnit;
}
final String size = t.toString();
if (size.contains(".")) {
if (part.unit == null) {
throw new IllegalArgumentException("unit must be specified with floating point numbers");
}
final double d = Double.parseDouble(size);
final long bytes = part.unit.toBytes(1);
part.size = (long) (bytes * d);
part.unit = SizeUnit.BYTES;
} else {
part.size = Integer.parseInt(size);
}
total = total.add(part);
}
this.size = total.size;
this.unit = total.unit;
}
public long getSize() {
return size;
}
public long getSize(final SizeUnit unit) {
return unit.convert(this.size, this.unit);
}
public void setSize(final long size) {
this.size = size;
}
public SizeUnit getUnit() {
return unit;
}
public void setUnit(final SizeUnit unit) {
this.unit = unit;
}
private static class Normalize {
private final long a;
private final long b;
private final SizeUnit base;
private Normalize(final Size a, final Size b) {
this.base = lowest(a, b);
this.a = a.unit == null ? a.size : base.convert(a.size, a.unit);
this.b = b.unit == null ? b.size : base.convert(b.size, b.unit);
}
private static SizeUnit lowest(final Size a, final Size b) {
if (a.unit == null) return b.unit;
if (b.unit == null) return a.unit;
if (a.size == 0) return b.unit;
if (b.size == 0) return a.unit;
return SizeUnit.values()[Math.min(a.unit.ordinal(), b.unit.ordinal())];
}
}
public boolean equals(final Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
final Size that = (Size) o;
final Normalize n = new Normalize(this, that);
return n.a == n.b;
}
@Override
public int hashCode() {
int result = (int) (size ^ (size >>> 32));
result = 31 * result + (unit != null ? unit.hashCode() : 0);
return result;
}
public Size add(final Size that) {
final Normalize n = new Normalize(this, that);
return new Size(n.a + n.b, n.base);
}
public Size subtract(final Size that) {
final Normalize n = new Normalize(this, that);
return new Size(n.a - n.b, n.base);
}
public Size to(SizeUnit unit) {
return new Size(unit.convert(this.size, this.unit), unit);
}
public static Size parse(final String text) {
return new Size(text);
}
private static void invalidFormat(final String text) {
throw new IllegalArgumentException("Illegal size format: '" + text + "'. Valid examples are '10kb' or '10 kilobytes'.");
}
@Override
public String toString() {
final StringBuilder sb = new StringBuilder();
sb.append(size);
if (unit != null) {
sb.append(abbreviate(unit));
}
return sb.toString();
}
private static SizeUnit parseUnit(final String u) {
if (u.length() == 0) return null;
if ("BYTES".equalsIgnoreCase(u)) return SizeUnit.BYTES;
if ("BYTE".equalsIgnoreCase(u)) return SizeUnit.BYTES;
if ("B".equalsIgnoreCase(u)) return SizeUnit.BYTES;
if ("KILOBYTES".equalsIgnoreCase(u)) return SizeUnit.KILOBYTES;
if ("KILOBYTE".equalsIgnoreCase(u)) return SizeUnit.KILOBYTES;
if ("KB".equalsIgnoreCase(u)) return SizeUnit.KILOBYTES;
if ("K".equalsIgnoreCase(u)) return SizeUnit.KILOBYTES;
if ("MEGABYTES".equalsIgnoreCase(u)) return SizeUnit.MEGABYTES;
if ("MEGABYTE".equalsIgnoreCase(u)) return SizeUnit.MEGABYTES;
if ("MB".equalsIgnoreCase(u)) return SizeUnit.MEGABYTES;
if ("M".equalsIgnoreCase(u)) return SizeUnit.MEGABYTES;
if ("GIGABYTES".equalsIgnoreCase(u)) return SizeUnit.GIGABYTES;
if ("GIGABYTE".equalsIgnoreCase(u)) return SizeUnit.GIGABYTES;
if ("GB".equalsIgnoreCase(u)) return SizeUnit.GIGABYTES;
if ("G".equalsIgnoreCase(u)) return SizeUnit.GIGABYTES;
if ("TERABYTES".equalsIgnoreCase(u)) return SizeUnit.TERABYTES;
if ("TERABYTE".equalsIgnoreCase(u)) return SizeUnit.TERABYTES;
if ("TB".equalsIgnoreCase(u)) return SizeUnit.TERABYTES;
if ("T".equalsIgnoreCase(u)) return SizeUnit.TERABYTES;
throw new IllegalArgumentException("Unknown size unit '" + u + "'. Supported units " + Join.join(", ", lowercase(SizeUnit.values())));
}
private static String abbreviate(final SizeUnit u) {
switch (u) {
case BYTES:
return "bytes";
case KILOBYTES:
return "kb";
case MEGABYTES:
return "mb";
case GIGABYTES:
return "gb";
case TERABYTES:
return "tb";
default:
throw new IllegalArgumentException("Unknown size unit '" + u + "'. Supported units " + Join.join(", ", lowercase(SizeUnit.values())));
}
}
@Override
public int compareTo(final Size that) {
final Normalize n = new Normalize(this, that);
return Long.compare(n.a, n.b);
}
private SizeUnit lowest(SizeUnit a, SizeUnit b) {
final int min = Math.min(a.ordinal(), b.ordinal());
return SizeUnit.values()[min];
}
private static List lowercase(final Enum... units) {
final List list = new ArrayList();
for (final Enum unit : units) {
list.add(unit.name().toLowerCase());
}
return list;
}
public static class SizeEditor extends java.beans.PropertyEditorSupport {
public void setAsText(final String text) {
final Size d = Size.parse(text);
setValue(d);
}
}
static {
PropertyEditorManager.registerEditor(Size.class, SizeEditor.class);
Editors.get(Size.class);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy