/*
* Copyright 1998-2009 University Corporation for Atmospheric Research/Unidata
*
* Portions of this software were developed by the Unidata Program at the
* University Corporation for Atmospheric Research.
*
* Access and use of this software shall impose the following obligations
* and understandings on the user. The user is granted the right, without
* any fee or cost, to use, copy, modify, alter, enhance and distribute
* this software, and any derivative works thereof, and its supporting
* documentation for any purpose whatsoever, provided that this entire
* notice appears in all copies of the software, derivative works and
* supporting documentation. Further, UCAR requests that the user credit
* UCAR/Unidata in any publications that result from the use of this
* software or in any product that includes this software. The names UCAR
* and/or Unidata, however, may not be used in any advertising or publicity
* to endorse or promote any products or commercial entity unless specific
* written permission is obtained from UCAR/Unidata. The user also
* understands that UCAR/Unidata is not obligated to provide the user with
* any support, consulting, training or assistance of any kind with regard
* to the use, operation and performance of this software nor to provide
* the user with any updates, revisions, new versions or "bug fixes."
*
* THIS SOFTWARE IS PROVIDED BY UCAR/UNIDATA "AS IS" AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL UCAR/UNIDATA BE LIABLE FOR ANY SPECIAL,
* INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING
* FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
* NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION
* WITH THE ACCESS, USE OR PERFORMANCE OF THIS SOFTWARE.
*/
package ucar.nc2;
import junit.framework.*;
import org.junit.experimental.categories.Category;
import ucar.ma2.*;
import ucar.unidata.util.test.category.NeedsCdmUnitTest;
import ucar.unidata.util.test.UtilsMa2Test;
import ucar.unidata.util.test.TestDir;
import java.io.*;
import java.util.List;
import java.util.ArrayList;
/** Test reading variable data */
public class TestReadStrides extends TestCase {
public TestReadStrides( String name) {
super(name);
}
public void testReadStridesCached() throws IOException, InvalidRangeException {
NetcdfFile ncfile = TestDir.openFileLocal("ncml/nc/time0.nc");
Variable temp = null;
assert(null != (temp = ncfile.findVariable("T")));
// read entire array
Array A = temp.read("0:2,0:3");
assert (A.getRank() == 2);
Index ima = A.getIndex();
int[] shape = A.getShape();
assert shape[0] == 3;
assert shape[1] == 4;
for (int i=0; i<shape[0]; i++) {
for (int j=0; j<shape[1]; j++) {
double dval = A.getDouble(ima.set(i,j));
assert( dval == (double) (i*10+j)) : dval;
}
}
A = temp.read("0:2:1,0:3:1");
assert (A.getRank() == 2);
ima = A.getIndex();
shape = A.getShape();
assert shape[0] == 3;
assert shape[1] == 4;
for (int i=0; i<shape[0]; i++) {
for (int j = 0; j < shape[1]; j++) {
double dval = A.getDouble(ima.set(i, j));
assert(dval == (double) (i * 10 + j)):dval;
}
}
A = temp.read("0:2:2,0:3:2");
assert (A.getRank() == 2);
ima = A.getIndex();
shape = A.getShape();
assert shape[0] == 2;
assert shape[1] == 2;
for (int i=0; i<shape[0]; i++) {
for (int j = 0; j < shape[1]; j++) {
double dval = A.getDouble(ima.set(i, j));
assert(dval == (double) (i * 20 + j*2)):dval;
}
}
A = temp.read(":,0:3:2");
assert (A.getRank() == 2);
ima = A.getIndex();
shape = A.getShape();
assert shape[0] == 3;
assert shape[1] == 2;
for (int i=0; i<shape[0]; i++) {
for (int j = 0; j < shape[1]; j++) {
double dval = A.getDouble(ima.set(i, j));
assert(dval == (double) (i * 10 + j*2)):dval;
}
}
A = temp.read("0:2:2,:");
assert (A.getRank() == 2);
ima = A.getIndex();
shape = A.getShape();
assert shape[0] == 2;
assert shape[1] == 4;
for (int i=0; i<shape[0]; i++) {
for (int j = 0; j < shape[1]; j++) {
double dval = A.getDouble(ima.set(i, j));
assert(dval == (double) (i * 20 + j)):dval;
}
}
ncfile.close();
}
public void testReadStridesNoCache() throws IOException, InvalidRangeException {
NetcdfFile ncfile = TestDir.openFileLocal("ncml/nc/time0.nc");
Variable temp = null;
assert(null != (temp = ncfile.findVariable("T")));
temp.setCaching(false);
Array A;
Index ima;
int shape[];
A = temp.read("0:2:1,0:3:1");
assert (A.getRank() == 2);
ima = A.getIndex();
shape = A.getShape();
assert shape[0] == 3;
assert shape[1] == 4;
for (int i=0; i<shape[0]; i++) {
for (int j = 0; j < shape[1]; j++) {
double dval = A.getDouble(ima.set(i, j));
assert(dval == (double) (i * 10 + j)):dval;
}
}
A = temp.read("0:2:2,0:3:2");
assert (A.getRank() == 2);
ima = A.getIndex();
shape = A.getShape();
assert shape[0] == 2;
assert shape[1] == 2;
for (int i=0; i<shape[0]; i++) {
for (int j = 0; j < shape[1]; j++) {
double dval = A.getDouble(ima.set(i, j));
assert(dval == (double) (i * 20 + j*2)):dval;
}
}
A = temp.read(":,0:3:2");
assert (A.getRank() == 2);
ima = A.getIndex();
shape = A.getShape();
assert shape[0] == 3;
assert shape[1] == 2;
for (int i=0; i<shape[0]; i++) {
for (int j = 0; j < shape[1]; j++) {
double dval = A.getDouble(ima.set(i, j));
assert(dval == (double) (i * 10 + j*2)):dval;
}
}
A = temp.read("0:2:2,:");
assert (A.getRank() == 2);
ima = A.getIndex();
shape = A.getShape();
assert shape[0] == 2;
assert shape[1] == 4;
for (int i=0; i<shape[0]; i++) {
for (int j = 0; j < shape[1]; j++) {
double dval = A.getDouble(ima.set(i, j));
assert(dval == (double) (i * 20 + j)):dval;
}
}
ncfile.close();
}
@Category(NeedsCdmUnitTest.class)
public void testReadStridesAll() throws IOException, InvalidRangeException {
testReadStrides(TestDir.cdmLocalTestDataDir+"ncml/nc/time0.nc");
testReadStrides(TestDir.cdmUnitTestDir+"formats/gini/HI-NATIONAL_14km_IR_20050918_2000.gini");
}
private void testReadStrides(String filename) throws IOException, InvalidRangeException {
NetcdfFile ncfile = TestDir.open(filename);
List vars = ncfile.getVariables();
for (int i = 0; i < vars.size(); i++) {
Variable v = (Variable) vars.get(i);
if (v.getRank() == 0 ) continue;
if (!v.hasCachedData())
v.setCaching(false);
//if (v.getRank() == 1 )
testVariableReadStrides(v);
}
ncfile.close();
}
private void testVariableReadStrides(Variable v) throws IOException, InvalidRangeException {
Array allData = v.read();
int[] shape = v.getShape();
if(shape.length < 5)
return;
for (int first=0;first<3;first++) {
for (int stride=2;stride<5;stride++) {
ArrayList ranges = new ArrayList();
for (int i = 0; i < shape.length; i++) {
int last = shape[i] - 1;
Range r = new Range(first, last, stride);
ranges.add(r);
}
System.out.println( v.getFullName()+" test range= "+Range.makeSectionSpec(ranges));
Array sectionRead = v.read(ranges);
Array sectionMake = allData.sectionNoReduce( ranges);
UtilsMa2Test.testEquals(sectionRead, sectionMake);
}
}
}
}