All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.tomitribe.util.Size Maven / Gradle / Ivy

There is a newer version: 2.0.0
Show 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 javax.xml.bind.annotation.adapters.XmlAdapter;
import javax.xml.bind.annotation.adapters.XmlJavaTypeAdapter;
import java.beans.PropertyEditorManager;
import java.util.ArrayList;
import java.util.List;

@XmlJavaTypeAdapter(Size.Adapter.class)
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)) {
                    // no-op
                } else {
                    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 long a;
        private long b;
        private 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);
        if (n.a > n.b) return 1;
        if (n.a == n.b) return 0;
        return -1;
    }

    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);
    }

    public static class Adapter extends XmlAdapter {

        @Override
        public Size unmarshal(String v) throws Exception {
            return new Size(v);
        }

        @Override
        public String marshal(Size v) throws Exception {
            return v.toString();
        }
    }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy