Application: gvSIG desktop

import org.gvsig.fmap.dal.exception.DataEvaluatorRuntimeException;

import org.gvsig.fmap.dal.feature.FeatureQueryOrder.FeatureQueryOrderMember;

	public void testSort() {

		try {

			DataStoreParameters dbfParameters = this

					.getDefaultDataStoreParameters();

			FeatureStore store = (FeatureStore) dataManager

					.createStore(dbfParameters);

			FeatureSet set1;

			FeatureQuery query;

			Iterator iter1, attrIter;

			FeatureAttributeDescriptor attr;

			attrIter = store.getDefaultFeatureType().iterator();

			String[] names;

			while (attrIter.hasNext()) {

				attr = (FeatureAttributeDescriptor) attrIter.next();

				query = store.createFeatureQuery();

				String fieldName = attr.getName();

				names = new String[] { fieldName };

				query.setAttributeNames(names);

				query.getOrder().add(fieldName, getRandom().nextBoolean());

				set1 = store.getFeatureSet(query);

				if (getRandom().nextBoolean()) {

					iter1 = set1.fastIterator();

				} else {

					iter1 = set1.iterator();

				}

				assertTrue(checkSort(iter1, query));

				set1.dispose();

			}

			int ntimes = getRandom().nextInt(10) + 5;

			FeatureType type = store.getDefaultFeatureType();

			query = store.createFeatureQuery();

			for (int i = 0; i < ntimes; i++) {

				names = getRandomAttibuteList(type);

				int nShortFields = getRandom().nextInt(names.length) + 1;

				query.getOrder().clear();

				for (int j = 0; j < nShortFields; j++) {

					attr = store.getDefaultFeatureType().getAttributeDescriptor(names[getRandom().nextInt(names.length)]);

					if (attr.getDataType() == DataTypes.INT

							|| attr.getDataType() == DataTypes.LONG

							|| attr.getDataType() == DataTypes.DOUBLE

							|| attr.getDataType() == DataTypes.STRING

							|| attr.getDataType() == DataTypes.DATE

							|| attr.getDataType() == DataTypes.BOOLEAN

							|| attr.getDataType() == DataTypes.BYTE

							|| attr.getDataType() == DataTypes.FLOAT) {

						query.getOrder().add(attr.getName(),

								getRandom().nextBoolean());

					}

				}

				query.setAttributeNames(names);

				set1 = store.getFeatureSet(query);

				if (getRandom().nextBoolean()) {

					iter1 = set1.fastIterator();

				} else {

					iter1 = set1.iterator();

				}

				assertTrue(checkSort(iter1, query));

				set1.dispose();

			}

			store.dispose();

		} catch (DataException e3) {

			e3.printStackTrace();

			fail();

			return;

		}

	}

	public boolean checkSort(Iterator iter, FeatureQuery query) {

		FeatureQueryOrderMember order;

		Feature prevFeature = null;

		Feature currFeature = null;

		boolean isFirst = true;

		Comparable v1, v2;

		int v;

		FeatureQueryOrder queryOrder = query.getOrder();

		Iterator orderIter;

		//for debug only

		/*

		System.out.println("\nCheck order:");

		Iterator orderIter = queryOrder.iterator();

		while (orderIter.hasNext()) {

			order = (FeatureQueryOrderMember) orderIter.next();

			System.out.print(order.getAttributeName() + " ");

			if (order.getAscending()) {

				System.out.print("Asc, ");

			} else {

				System.out.print("Desc, ");

			}

		}

		System.out.println(";");

		*/

		while (iter.hasNext()) {

			currFeature = (Feature) iter.next();

			if (isFirst) {

				prevFeature = currFeature.getCopy();

				// printFeature(prevFeature, true, 15);

				isFirst = false;

				continue;

			}

			// printFeature(currFeature, false, 15);

			orderIter = queryOrder.iterator();

			while (orderIter.hasNext()) {

				order = (FeatureQueryOrderMember) orderIter.next();

				if (order.hasEvaluator()) {

					try {

						v1 = (Comparable) order.getEvaluator().evaluate(

								(EvaluatorData) prevFeature);

						v2 = (Comparable) order.getEvaluator().evaluate(

								(EvaluatorData) currFeature);

					} catch (EvaluatorException e) {

						throw new DataEvaluatorRuntimeException(e);

					}

				} else {

					v1 = (Comparable) prevFeature.get(order.getAttributeName());

					v2 = (Comparable) currFeature.get(order.getAttributeName());

				}

				v = v1.compareTo(v2);

				if (v != 0) {

					if (!order.getAscending()) {

						v = -v;

					}

				}

				if (v < 0) {

					break;

				} else if (v > 0) {

					return false;

				}

			}

			prevFeature = currFeature.getCopy();

		}

		return true;

	}