2016-08-02 22:55:18 +00:00
|
|
|
/*
|
|
|
|
* Minecraft Forge
|
2018-07-01 21:17:28 +00:00
|
|
|
* Copyright (c) 2016-2018.
|
2016-08-02 22:55:18 +00:00
|
|
|
*
|
|
|
|
* 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 version 2.1
|
|
|
|
* of the License.
|
|
|
|
*
|
|
|
|
* 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 library; if not, write to the Free Software
|
|
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
|
|
*/
|
|
|
|
|
2017-06-19 22:02:18 +00:00
|
|
|
package net.minecraftforge.registries;
|
2016-08-02 22:55:18 +00:00
|
|
|
|
|
|
|
import java.util.List;
|
|
|
|
|
|
|
|
import com.google.common.collect.Lists;
|
|
|
|
|
|
|
|
import net.minecraft.util.ResourceLocation;
|
2017-06-19 22:02:18 +00:00
|
|
|
import net.minecraftforge.registries.IForgeRegistry.*;
|
2016-08-02 22:55:18 +00:00
|
|
|
|
2017-01-11 23:17:56 +00:00
|
|
|
import javax.annotation.Nullable;
|
|
|
|
|
2018-09-06 03:06:18 +00:00
|
|
|
public class RegistryBuilder<T extends IForgeRegistryEntry<T>>
|
2016-08-02 22:55:18 +00:00
|
|
|
{
|
|
|
|
private ResourceLocation registryName;
|
|
|
|
private Class<T> registryType;
|
|
|
|
private ResourceLocation optionalDefaultKey;
|
2017-06-19 22:02:18 +00:00
|
|
|
private int minId = 0;
|
2017-06-27 22:18:03 +00:00
|
|
|
private int maxId = Integer.MAX_VALUE - 1;
|
2016-08-02 22:55:18 +00:00
|
|
|
private List<AddCallback<T>> addCallback = Lists.newArrayList();
|
|
|
|
private List<ClearCallback<T>> clearCallback = Lists.newArrayList();
|
|
|
|
private List<CreateCallback<T>> createCallback = Lists.newArrayList();
|
2018-06-27 05:00:53 +00:00
|
|
|
private List<ValidateCallback<T>> validateCallback = Lists.newArrayList();
|
2017-06-15 20:00:46 +00:00
|
|
|
private boolean saveToDisc = true;
|
2017-06-19 22:02:18 +00:00
|
|
|
private boolean allowOverrides = true;
|
|
|
|
private boolean allowModifications = false;
|
|
|
|
private DummyFactory<T> dummyFactory;
|
2017-08-15 12:10:07 +00:00
|
|
|
private MissingFactory<T> missingFactory;
|
2016-08-02 22:55:18 +00:00
|
|
|
|
|
|
|
public RegistryBuilder<T> setName(ResourceLocation name)
|
|
|
|
{
|
|
|
|
this.registryName = name;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public RegistryBuilder<T> setType(Class<T> type)
|
|
|
|
{
|
|
|
|
this.registryType = type;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public RegistryBuilder<T> setIDRange(int min, int max)
|
|
|
|
{
|
|
|
|
this.minId = min;
|
|
|
|
this.maxId = max;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
2017-06-19 22:02:18 +00:00
|
|
|
public RegistryBuilder<T> setMaxID(int max)
|
|
|
|
{
|
|
|
|
return this.setIDRange(0, max);
|
|
|
|
}
|
|
|
|
|
2017-03-23 18:23:54 +00:00
|
|
|
public RegistryBuilder<T> setDefaultKey(ResourceLocation key)
|
|
|
|
{
|
|
|
|
this.optionalDefaultKey = key;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
2016-08-02 22:55:18 +00:00
|
|
|
@SuppressWarnings("unchecked")
|
|
|
|
public RegistryBuilder<T> addCallback(Object inst)
|
|
|
|
{
|
|
|
|
if (inst instanceof AddCallback)
|
|
|
|
this.add((AddCallback<T>)inst);
|
|
|
|
if (inst instanceof ClearCallback)
|
|
|
|
this.add((ClearCallback<T>)inst);
|
|
|
|
if (inst instanceof CreateCallback)
|
|
|
|
this.add((CreateCallback<T>)inst);
|
2018-06-27 05:00:53 +00:00
|
|
|
if (inst instanceof ValidateCallback)
|
|
|
|
this.add((ValidateCallback<T>)inst);
|
2017-06-19 22:02:18 +00:00
|
|
|
if (inst instanceof DummyFactory)
|
|
|
|
this.set((DummyFactory<T>)inst);
|
2017-08-15 12:10:07 +00:00
|
|
|
if (inst instanceof MissingFactory)
|
|
|
|
this.set((MissingFactory<T>)inst);
|
2016-08-02 22:55:18 +00:00
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public RegistryBuilder<T> add(AddCallback<T> add)
|
|
|
|
{
|
|
|
|
this.addCallback.add(add);
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public RegistryBuilder<T> add(ClearCallback<T> clear)
|
|
|
|
{
|
|
|
|
this.clearCallback.add(clear);
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public RegistryBuilder<T> add(CreateCallback<T> create)
|
|
|
|
{
|
|
|
|
this.createCallback.add(create);
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
2018-06-27 05:00:53 +00:00
|
|
|
public RegistryBuilder<T> add(ValidateCallback<T> validate)
|
|
|
|
{
|
|
|
|
this.validateCallback.add(validate);
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
2017-06-19 22:02:18 +00:00
|
|
|
public RegistryBuilder<T> set(DummyFactory<T> factory)
|
2016-08-02 22:55:18 +00:00
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
this.dummyFactory = factory;
|
2016-08-02 22:55:18 +00:00
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
2017-08-15 12:10:07 +00:00
|
|
|
public RegistryBuilder<T> set(MissingFactory<T> missing)
|
|
|
|
{
|
|
|
|
this.missingFactory = missing;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
2017-06-19 22:02:18 +00:00
|
|
|
public RegistryBuilder<T> disableSaving()
|
2017-06-14 17:14:56 +00:00
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
this.saveToDisc = false;
|
2017-06-14 17:14:56 +00:00
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
2017-06-19 22:02:18 +00:00
|
|
|
public RegistryBuilder<T> disableOverrides()
|
2017-06-14 17:14:56 +00:00
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
this.allowOverrides = false;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
|
|
|
public RegistryBuilder<T> allowModification()
|
|
|
|
{
|
|
|
|
this.allowModifications = true;
|
2017-06-14 17:14:56 +00:00
|
|
|
return this;
|
|
|
|
}
|
|
|
|
|
2016-08-02 22:55:18 +00:00
|
|
|
public IForgeRegistry<T> create()
|
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
return RegistryManager.ACTIVE.createRegistry(registryName, registryType, optionalDefaultKey, minId, maxId,
|
2018-06-27 05:00:53 +00:00
|
|
|
getAdd(), getClear(), getCreate(), getValidate(), saveToDisc, allowOverrides, allowModifications, dummyFactory, missingFactory);
|
2016-08-02 22:55:18 +00:00
|
|
|
}
|
|
|
|
|
2017-01-11 23:17:56 +00:00
|
|
|
@Nullable
|
2016-08-02 22:55:18 +00:00
|
|
|
private AddCallback<T> getAdd()
|
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
if (addCallback.isEmpty())
|
2016-08-02 22:55:18 +00:00
|
|
|
return null;
|
2017-06-19 22:02:18 +00:00
|
|
|
if (addCallback.size() == 1)
|
|
|
|
return addCallback.get(0);
|
2016-08-02 22:55:18 +00:00
|
|
|
|
2017-06-26 03:45:48 +00:00
|
|
|
return (owner, stage, id, obj, old) ->
|
2016-08-02 22:55:18 +00:00
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
for (AddCallback<T> cb : this.addCallback)
|
2017-06-26 03:45:48 +00:00
|
|
|
cb.onAdd(owner, stage, id, obj, old);
|
2016-08-02 22:55:18 +00:00
|
|
|
};
|
|
|
|
}
|
2017-01-11 23:17:56 +00:00
|
|
|
|
|
|
|
@Nullable
|
2016-08-02 22:55:18 +00:00
|
|
|
private ClearCallback<T> getClear()
|
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
if (clearCallback.isEmpty())
|
2016-08-02 22:55:18 +00:00
|
|
|
return null;
|
2017-06-19 22:02:18 +00:00
|
|
|
if (clearCallback.size() == 1)
|
|
|
|
return clearCallback.get(0);
|
2016-08-02 22:55:18 +00:00
|
|
|
|
2017-06-19 22:02:18 +00:00
|
|
|
return (owner, stage) ->
|
2016-08-02 22:55:18 +00:00
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
for (ClearCallback<T> cb : this.clearCallback)
|
|
|
|
cb.onClear(owner, stage);
|
2016-08-02 22:55:18 +00:00
|
|
|
};
|
|
|
|
}
|
2017-01-11 23:17:56 +00:00
|
|
|
|
|
|
|
@Nullable
|
2016-08-02 22:55:18 +00:00
|
|
|
private CreateCallback<T> getCreate()
|
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
if (createCallback.isEmpty())
|
2016-08-02 22:55:18 +00:00
|
|
|
return null;
|
2017-06-19 22:02:18 +00:00
|
|
|
if (createCallback.size() == 1)
|
|
|
|
return createCallback.get(0);
|
2016-08-02 22:55:18 +00:00
|
|
|
|
2017-06-19 22:02:18 +00:00
|
|
|
return (owner, stage) ->
|
2016-08-02 22:55:18 +00:00
|
|
|
{
|
2017-06-19 22:02:18 +00:00
|
|
|
for (CreateCallback<T> cb : this.createCallback)
|
|
|
|
cb.onCreate(owner, stage);
|
2016-08-02 22:55:18 +00:00
|
|
|
};
|
|
|
|
}
|
2018-06-27 05:00:53 +00:00
|
|
|
|
|
|
|
@Nullable
|
|
|
|
private ValidateCallback<T> getValidate()
|
|
|
|
{
|
|
|
|
if (validateCallback.isEmpty())
|
|
|
|
return null;
|
|
|
|
if (validateCallback.size() == 1)
|
|
|
|
return validateCallback.get(0);
|
|
|
|
|
|
|
|
return (owner, stage, id, key, obj) ->
|
|
|
|
{
|
|
|
|
for (ValidateCallback<T> cb : this.validateCallback)
|
|
|
|
cb.onValidate(owner, stage, id, key, obj);
|
|
|
|
};
|
|
|
|
}
|
2016-08-02 22:55:18 +00:00
|
|
|
}
|