src.it.unimi.dsi.big.util.StringMaps Maven / Gradle / Ivy
package it.unimi.dsi.big.util;
/*
* DSI utilities
*
* Copyright (C) 2008-2017 Sebastiano Vigna
*
* This library 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 (at your option)
* any later version.
*
* This library 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.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, see .
*
*/
import it.unimi.dsi.fastutil.objects.AbstractObject2LongFunction;
import it.unimi.dsi.fastutil.objects.AbstractObject2ObjectFunction;
import it.unimi.dsi.fastutil.objects.Object2ObjectFunction;
import it.unimi.dsi.fastutil.objects.Object2ObjectFunctions;
import it.unimi.dsi.fastutil.objects.ObjectBigList;
import it.unimi.dsi.fastutil.objects.ObjectBigLists;
import it.unimi.dsi.util.Interval;
import it.unimi.dsi.util.LongInterval;
import java.io.Serializable;
/** A class providing static methods and objects that do useful things with {@linkplain StringMap string maps}
* and {@linkplain PrefixMap prefix maps}.
*
* @see StringMap
* @see PrefixMap
* @author Sebastiano Vigna
* @since 2.0
*/
public class StringMaps {
private StringMaps() {}
protected static class SynchronizedStringMap implements StringMap, Serializable {
private static final long serialVersionUID = 1L;
protected final StringMap stringMap;
protected ObjectBigList extends S> list;
public SynchronizedStringMap(final StringMap stringMap) {
this.stringMap = stringMap;
}
@Override
public synchronized long size64() {
return stringMap.size64();
}
@Override
public synchronized ObjectBigList extends S> list() {
if (list == null) {
list = stringMap.list();
if(list != null) list = ObjectBigLists.synchronize(list, this);
}
return list;
}
@Override
public synchronized long getLong(Object s) {
return stringMap.getLong(s);
}
@Override
@SuppressWarnings("deprecation")
public synchronized Long get(Object key) {
return stringMap.get(key);
}
@Override
public synchronized long put(CharSequence key, long value) {
return stringMap.put(key, value);
}
@Override
@SuppressWarnings("deprecation")
public synchronized Long put(CharSequence key, Long value) {
return stringMap.put(key, value);
}
@Override
@SuppressWarnings("deprecation")
public synchronized Long remove(Object key) {
return stringMap.remove(key);
}
@Override
public synchronized long removeLong(Object key) {
return stringMap.removeLong(key);
}
@Override
public synchronized void clear() {
stringMap.clear();
}
@Override
public synchronized boolean containsKey(Object key) {
return stringMap.containsKey(key);
}
@Override
public synchronized long defaultReturnValue() {
return stringMap.defaultReturnValue();
}
@Override
public synchronized void defaultReturnValue(long rv) {
stringMap.defaultReturnValue(rv);
}
}
protected static class SynchronizedPrefixMap extends SynchronizedStringMap implements PrefixMap, Serializable {
private static final long serialVersionUID = 1L;
protected final PrefixMap map;
protected Object2ObjectFunction prefixMap;
protected Object2ObjectFunction rangeMap;
public SynchronizedPrefixMap(final PrefixMap map) {
super(map);
this.map = map;
}
@Override
public synchronized Object2ObjectFunction prefixMap() {
if (prefixMap == null) {
prefixMap = map.prefixMap();
if (prefixMap != null) prefixMap = Object2ObjectFunctions.synchronize(prefixMap, this);
}
return prefixMap;
}
@Override
public synchronized Object2ObjectFunction rangeMap() {
if (rangeMap == null) {
rangeMap = map.rangeMap();
if (rangeMap != null) rangeMap = Object2ObjectFunctions.synchronize(rangeMap, this);
}
return rangeMap;
}
}
/** Returns a synchronized string map backed by the given string map.
*
* @param stringMap the string map to be wrapped in a synchronized map.
* @return a synchronized view of the specified string map.
*/
public static StringMap synchronize(final StringMap stringMap) {
return stringMap instanceof PrefixMap ? new SynchronizedPrefixMap<>((PrefixMap)stringMap) : new SynchronizedStringMap<>(stringMap);
}
/** Returns a synchronized prefix map backed by the given prefix map.
*
* @param prefixMap the prefix map to be wrapped in a synchronized map.
* @return a synchronized view of the specified prefix map.
*/
public static PrefixMap synchronize(final PrefixMap prefixMap) {
return new SynchronizedPrefixMap<>(prefixMap);
}
protected static class StringMapWrapper extends AbstractObject2LongFunction implements StringMap {
private static final long serialVersionUID = 1L;
private final it.unimi.dsi.util.StringMap stringMap;
public StringMapWrapper(final it.unimi.dsi.util.StringMap stringMap) {
this.stringMap = stringMap;
}
@Override
public long getLong(final Object key) {
return stringMap.getLong(key);
}
@Override
public boolean containsKey(final Object key) {
return stringMap.containsKey(key);
}
@Override
public long size64() {
return stringMap.size();
}
@Override
public ObjectBigList extends T> list() {
return ObjectBigLists.asBigList(stringMap.list());
}
}
/** Returns an immutable (big) {@link StringMap} view of a standard {@link it.unimi.dsi.util.StringMap}.
*
* @param stringMap a string map.
* @return a {@link StringMap} view of {@code stringMap}.
*/
public static StringMap wrap(final it.unimi.dsi.util.StringMap stringMap) {
return new StringMapWrapper<>(stringMap);
}
protected static class PrefixMapWrapper extends StringMapWrapper implements PrefixMap {
private static final long serialVersionUID = 1L;
private final Object2ObjectFunction rangeMap;
public PrefixMapWrapper(final it.unimi.dsi.util.PrefixMap prefixMap) {
super(prefixMap);
rangeMap = new AbstractObject2ObjectFunction() {
private static final long serialVersionUID = 1L;
private final Object2ObjectFunction prefixMapRangeMap = prefixMap.rangeMap();
@Override
public LongInterval get(final Object key) {
final Interval interval = prefixMapRangeMap.get(key);
return LongInterval.valueOf(interval.left, interval.right);
}
@Override
public boolean containsKey(final Object key) {
return prefixMapRangeMap.containsKey(key);
}
@Override
public int size() {
return prefixMapRangeMap.size();
}
};
}
@Override
public Object2ObjectFunction rangeMap() {
return rangeMap;
}
@Override
public Object2ObjectFunction prefixMap() {
return null;
}
}
/** Returns an immutable (big) {@link PrefixMap} view of a standard {@link it.unimi.dsi.util.PrefixMap}. Note that
* the returned prefix map does not implement {@link PrefixMap#prefixMap()}.
*
* @param prefixMap a prefix map.
* @return a {@link PrefixMap} view of {@code prefixMap}.
*/
public static PrefixMap wrap(final it.unimi.dsi.util.PrefixMap prefixMap) {
return new PrefixMapWrapper<>(prefixMap);
}
}