runelite/src/main/java/info/sigterm/deob/Methods.java

package info.sigterm.deob;

import info.sigterm.deob.pool.NameAndType;
import info.sigterm.deob.signature.Signature;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class Methods
{
	ClassFile classFile;

	private List<Method> methods = new ArrayList<>();

	Methods(ClassFile cf) throws IOException
	{
		classFile = cf;

		DataInputStream is = cf.getStream();

		int count = is.readUnsignedShort();

		for (int i = 0; i < count; ++i)
			methods.add(new Method(this));
	}
	
	public void write(DataOutputStream out) throws IOException
	{
		out.writeShort(methods.size());
		for (Method m : methods)
			m.write(out);
	}
	
	public void removeMethod(Method m)
	{
		m.remove();
		methods.remove(m);
	}

	public ClassFile getClassFile()
	{
		return classFile;
	}
	
	public List<Method> getMethods()
	{
		return methods;
	}
	
	public Method findMethod(NameAndType nat)
	{
		for (Method m : methods)
			if (m.getName().equals(nat.getName()) && m.getDescriptor().equals(nat.getDescriptor()))
				return m;
		return null;
	}
	
	public Method findMethod(String name, Signature type)
	{
		for (Method m : methods)
			if (m.getName().equals(name) && m.getDescriptor().equals(type))
				return m;
		return null;
	}
	
	public Method findMethod(String name)
	{
		for (Method m : methods)
			if (m.getName().equals(name))
				return m;
		return null;
	}

	public void buildInstructionGraph()
	{
		for (Method m : methods)
			m.buildInstructionGraph();
	}
}