package net.osmand.osm.edit;

import gnu.trove.list.array.TLongArrayList;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import net.osmand.data.LatLon;
import net.osmand.data.QuadRect;

public class Way extends Entity {

	// lazy loading
	private TLongArrayList nodeIds = null;
	private List<Node> nodes = null;

	public Way(long id) {
		super(id);
	}

	public Way(Way w) {
		super(w.getId());
		if (w.nodeIds != null) {
			nodeIds = new TLongArrayList(w.nodeIds);
		}
		if (w.nodes != null) {
			nodes = new ArrayList<Node>(w.nodes);
		}
	}

	public Way(long id, List<Node> nodes) {
		super(id);
		this.nodes = new ArrayList<Node>(nodes);
		nodeIds = new TLongArrayList(nodes.size());
		for (Node n : nodes) {
			nodeIds.add(n.getId());
		}
	}

	public void addNode(long id) {
		if (nodeIds == null) {
			nodeIds = new TLongArrayList();
		}
		nodeIds.add(id);
	}

	public long getFirstNodeId() {
		if (nodeIds == null) {
			return -1;
		}
		return nodeIds.get(0);
	}

	public long getLastNodeId() {
		if (nodeIds == null) {
			return -1;
		}
		return nodeIds.get(nodeIds.size() - 1);
	}

	public Node getFirstNode() {
		if (nodes == null || nodes.size() == 0) {
			return null;
		}
		return nodes.get(0);
	}

	public Node getLastNode() {
		if (nodes == null || nodes.size() == 0) {
			return null;
		}
		return nodes.get(nodes.size() - 1);
	}

	public void addNode(Node n) {
		if (nodeIds == null) {
			nodeIds = new TLongArrayList();
		}
		if (nodes == null) {
			nodes = new ArrayList<Node>();
		}
		nodeIds.add(n.getId());
		nodes.add(n);
	}

	public void addNode(Node n, int index) {
		if (nodeIds == null) {
			nodeIds = new TLongArrayList();
		}
		if (nodes == null) {
			nodes = new ArrayList<Node>();
		}
		nodeIds.insert(index, n.getId());
		nodes.add(index, n);
	}

	public long removeNodeByIndex(int i) {
		if (nodeIds == null) {
			return -1;
		}
		long toReturn = nodeIds.removeAt(i);
		if (nodes != null && nodes.size() > i) {
			nodes.remove(i);
		}
		return toReturn;
	}

	public TLongArrayList getNodeIds() {
		if (nodeIds == null) {
			return new TLongArrayList(0);
		}
		return nodeIds;
	}

	public List<EntityId> getEntityIds() {
		if (nodeIds == null) {
			return Collections.emptyList();
		}
		List<EntityId> ls = new ArrayList<EntityId>();
		for (int i = 0; i < nodeIds.size(); i++) {
			ls.add(new EntityId(EntityType.NODE, nodeIds.get(i)));
		}
		return ls;
	}

	public List<Node> getNodes() {
		if (nodes == null) {
			return Collections.emptyList();
		}
		return nodes;
	}

	@Override
	public void initializeLinks(Map<EntityId, Entity> entities) {
		if (nodeIds != null) {
			if (nodes == null) {
				nodes = new ArrayList<Node>();
			} else {
				nodes.clear();
			}
			int nIsize = nodeIds.size();
			for (int i = 0; i < nIsize; i++) {
				nodes.add((Node) entities.get(new EntityId(EntityType.NODE, nodeIds.get(i))));
			}
		}
	}

	public QuadRect getLatLonBBox() {
		QuadRect qr = null;
		if (nodes != null) {
			for (Node n : nodes) {
				if (qr == null) {
					qr = new QuadRect();
					qr.left = (float) n.getLongitude();
					qr.right = (float) n.getLongitude();
					qr.top = (float) n.getLatitude();
					qr.bottom = (float) n.getLatitude();
				}
				if (n.getLongitude() < qr.left) {
					qr.left = (float) n.getLongitude();
				} else if (n.getLongitude() > qr.right) {
					qr.right = (float) n.getLongitude();
				}
				if (n.getLatitude() > qr.top) {
					qr.top = (float) n.getLatitude();
				} else if (n.getLatitude() < qr.bottom) {
					qr.bottom = (float) n.getLatitude();
				}
			}
		}
		return qr;
	}

	@Override
	public LatLon getLatLon() {
		if (nodes == null) {
			return null;
		}
		return OsmMapUtils.getWeightCenterForWay(this);
	}


	public void reverseNodes() {
		if (nodes != null) {
			Collections.reverse(nodes);
		}
		if (nodeIds != null) {
			nodeIds.reverse();
			;
		}
	}
}