void SET2(int I, int J, float *DXX, float DD, linearpart<float>& elevDEM, linearpart<float>& flowDir, linearpart<float>& slope) {
    double dxA = elevDEM.getdxA();
    double dyA = elevDEM.getdyA();
    float SK[9];
    float ANGLE[9];
    float SMAX;
    float tempFloat;
    int K;
    int KD;

    int ID1[] = {0, 1, 2, 2, 1, 1, 2, 2, 1};
    int ID2[] = {0, 2, 1, 1, 2, 2, 1, 1, 2};
    int I1[] = {0, 0, -1, -1, 0, 0, 1, 1, 0};
    int I2[] = {0, -1, -1, -1, -1, 1, 1, 1, 1};
    int J1[] = {0, 1, 0, 0, -1, -1, 0, 0, 1};
    int J2[] = {0, 1, 1, -1, -1, -1, -1, 1, 1};
    float ANGC[] = {0, 0., 1., 1., 2., 2., 3., 3., 4.};
    float ANGF[] = {0, 1., -1., 1., -1., 1., -1., 1., -1.};


    for (K = 1; K <= 8; K++) {
        VSLOPE(
                elevDEM.getData(J, I, tempFloat), //felevg.d[J][I],
                elevDEM.getData(J + J1[K], I + I1[K], tempFloat), //[felevg.d[J+J1[K]][I+I1[K]],
                elevDEM.getData(J + J2[K], I + I2[K], tempFloat), //felevg.d[J+J2[K]][I+I2[K]],
                DXX[ID1[K]],
                DXX[ID2[K]],
                DD,
                &SK[K],
                &ANGLE[K]
                );
    }
    tempFloat = -1;
    SMAX = 0.;
    KD = 0;
    flowDir.setData(J, I, tempFloat); //USE -1 TO INDICATE DIRECTION NOT YET SET 
    for (K = 1; K <= 8; K++) {
        if (SK[K] > SMAX) {
            SMAX = SK[K];
            KD = K;
        }
    }

    if (KD > 0) {
        tempFloat = (float) (ANGC[KD]*(PI / 2) + ANGF[KD] * ANGLE[KD]);
        flowDir.setData(J, I, tempFloat); //set to angle
    }
    slope.setData(J, I, SMAX);
}
示例#2
0
文件: d8.cpp 项目: kornholi/TauDEM
// Sets only flowDir only where there is a positive slope
// Returns number of cells which are flat
int setPosDir(linearpart<float>& elevDEM, linearpart<short>& flowDir)
{
    double dxA = elevDEM.getdxA();
    double dyA = elevDEM.getdyA();
    int nx = elevDEM.getnx();
    int ny = elevDEM.getny();
    int numFlat = 0;

    double tempdxc, tempdyc;

    fact = new double*[ny]; // initialize 2d array by Nazmus 2/16

    for(int m = 0; m<ny; m++)
        fact[m] = new double[9];

    for (int m = 0; m<ny; m++) {
        for (int k = 1; k <= 8; k++) {
            elevDEM.getdxdyc(m, tempdxc, tempdyc);
            fact[m][k] = (double) (1./sqrt(d1[k]*d1[k]*tempdxc*tempdxc + d2[k]*d2[k]*tempdyc*tempdyc));
        }
    }

    for (int j = 0; j < ny; j++) {
        for (int i=0; i < nx; i++) {
            //FlowDir is nodata if it is on the border OR elevDEM has no data
            if (elevDEM.isNodata(i,j) || !elevDEM.hasAccess(i-1,j) || !elevDEM.hasAccess(i+1,j) ||
                    !elevDEM.hasAccess(i,j-1) || !elevDEM.hasAccess(i,j+1)) {
                //do nothing
                continue;
            }

            //Check if cell is "contaminated" (neighbors have no data)
            //  set flowDir to noData if contaminated
            bool contaminated = false;
            for (int k=1; k<=8; k++) {
                int in=i+d1[k];
                int jn=j+d2[k];

                if (elevDEM.isNodata(in,jn)) {
                    contaminated = true;
                    break;
                }
            }

            if (contaminated) {
                flowDir.setToNodata(i,j);
            } else {
                // If cell is not contaminated,
                flowDir.setData(i, j, 0);
                setFlow(i,j, flowDir, elevDEM);

                if (flowDir.getData(i,j) == 0) {
                    numFlat++;
                }
            }
        }
    }

    return numFlat;
}
示例#3
0
文件: d8.cpp 项目: kornholi/TauDEM
//Set positive flowdirections of elevDEM
void setFlow(int i, int j, linearpart<short>& flowDir, linearpart<float>& elevDEM)
{
    int in,jn;
    int amax=0;
    float smax=0;

    float elev = elevDEM.getData(i, j);
    
    // fixme: return instead of checking isNodata

    for (short k=1; k<=8 && !flowDir.isNodata(i,j); k+=2) {
        in=i+d1[k];
        jn=j+d2[k];
        
        float slope = fact[j][k] * (elev - elevDEM.getData(in,jn));

        if (slope > smax) {
            smax=slope;
            short dirnb=flowDir.getData(in,jn);

            if (dirnb > 0 && abs(dirnb-k) == 4) {
                flowDir.setToNodata(i,j);
            } else {
                flowDir.setData(i,j,k);
            }
        }
    }

    for (short k=2; k<=8 && !flowDir.isNodata(i,j); k+=2) {
        in=i+d1[k];
        jn=j+d2[k];

        float slope = fact[j][k] * (elev - elevDEM.getData(in,jn));

        if (slope > smax && !cellsCross(k,i,j,flowDir)) {
            smax = slope;
            short dirnb = flowDir.getData(in,jn);

            if (dirnb > 0 && abs(dirnb-k) == 4) { 
                flowDir.setToNodata(i,j);
            } else {
                flowDir.setData(i,j,k);
            }
        }
    }
}
示例#4
0
文件: d8.cpp 项目: kornholi/TauDEM
void setFlow2(int i, int j, linearpart<short>& flowDir, T& elev, SparsePartition<int>& inc)
{
    /*  This function sets directions based upon secondary elevations for
      assignment of flow directions across flats according to Garbrecht and Martz
      scheme.  There are two possibilities:
    	A.  The neighbor is outside the flat set
    	B.  The neighbor is in the flat set.
    	In the case of A the input elevations are used and if a draining neighbor is found it is selected.
    	Case B requires slope to be positive.  Remaining flats are removed by iterating this process
    */
    int nx = flowDir.getnx();
    int ny = flowDir.getny();

    const short order[8]= {1,3,5,7,2,4,6,8};

    float slopeMax = 0;

    for (short k : order) {
        int in = i+d1[k];
        int jn = j+d2[k];

        if (!flowDir.hasAccess(in, jn))
            continue;

        if (inc.getData(in, jn) > 0) {
            // Neighbor is in flat
            float slope = fact[j][k]*(inc.getData(i, j) - inc.getData(in, jn));

            if (slope > slopeMax) {
                flowDir.setData(i, j, k);
                slopeMax = slope;
            }
        } else {
            // Neighbor is not in flat
            auto ed = elev.getData(i, j) - elev.getData(in, jn);

            if (ed >= 0) {
                // Found a way out - this is outlet
                flowDir.setData(i, j, k);
                break;  
            }
        }
    }
}
long setPosDirDinf(linearpart<float>& elevDEM, linearpart<float>& flowDir, linearpart<float>& slope, int useflowfile) {
    double dxA = elevDEM.getdxA();
    double dyA = elevDEM.getdyA();
    long nx = elevDEM.getnx();
    long ny = elevDEM.getny();
    float tempFloat;
    double tempdxc, tempdyc;
    int i, j, k, in, jn, con;
    long numFlat = 0;

    tempFloat = 0;
    for (j = 0; j < ny; j++) {
        for (i = 0; i < nx; i++) {


            //FlowDir is nodata if it is on the border OR elevDEM has no data
            if (elevDEM.isNodata(i, j) || !elevDEM.hasAccess(i - 1, j) || !elevDEM.hasAccess(i + 1, j) ||
                    !elevDEM.hasAccess(i, j - 1) || !elevDEM.hasAccess(i, j + 1)) {
                //do nothing			
            } else {
                //Check if cell is "contaminated" (neighbors have no data)
                //  set flowDir to noData if contaminated
                con = 0;
                for (k = 1; k <= 8 && con != -1; k++) {
                    in = i + d1[k];
                    jn = j + d2[k];
                    if (elevDEM.isNodata(in, jn)) con = -1;
                }
                if (con == -1) 
                    flowDir.setToNodata(i, j);
                    //If cell is not contaminated,
                else {
                    tempFloat = -1.;
                    flowDir.setData(i, j, tempFloat); //set to -1
                    elevDEM.getdxdyc(j, tempdxc, tempdyc);


                    float DXX[3] = {0, tempdxc, tempdyc}; //tardemlib.cpp ln 1291
                    float DD = sqrt(tempdxc * tempdxc + tempdyc * tempdyc); //tardemlib.cpp ln 1293
                    SET2(j, i, DXX, DD, elevDEM, flowDir, slope); //i=y in function form old code j is x switched on purpose
                    //  Use SET2 from serial code here modified to get what it has as felevg.d from elevDEM partition
                    //  Modify to return 0 if there is a 0 slope.  Modify SET2 to output flowDIR as no data (do nothing 
                    //  if verified initialization to nodata) and 
                    //  slope as 0 if a positive slope is not found

                    //setFlow( i,j, flowDir, elevDEM, area, useflowfile);
                    if (flowDir.getData(i, j, tempFloat) == -1)
                        numFlat++;
                }
            }
        }
    }
    return numFlat;
}
示例#6
0
文件: d8.cpp 项目: kornholi/TauDEM
//Calculate the slope information of flowDir to slope
void calcSlope(linearpart<short>& flowDir, linearpart<float>& elevDEM, linearpart<float>& slope)
{
    int nx = elevDEM.getnx();
    int ny = elevDEM.getny();

    for (int j=0; j < ny; j++) {
        for (int i=0; i < nx; i++) {
            // If i,j is on the border or flowDir has no data, set slope(i,j) to slopeNoData
            if (flowDir.isNodata(i,j) || !flowDir.hasAccess(i-1,j) || !flowDir.hasAccess(i+1,j) ||
                    !flowDir.hasAccess(i,j-1) || !flowDir.hasAccess(i,j+1)) {
                slope.setToNodata(i, j);
            } else {
                short flowDirection = flowDir.getData(i,j);
  
                int in = i + d1[flowDirection];
                int jn = j + d2[flowDirection];

                float elevDiff = elevDEM.getData(i,j) - elevDEM.getData(in,jn);
                slope.setData(i,j, elevDiff*fact[j][flowDirection]);
            }
        }
    }
}
void SET2(int I, int J, float *DXX, float DD, T& elevDEM, SparsePartition<short>& elev2, linearpart<float>& flowDir) {
    float SK[9];
    float ANGLE[9];
    float SMAX = 0.0;
    float tempFloat;
    short tempShort, tempShort1, tempShort2;
    int K;
    int KD = 0;

    int ID1[] = {0, 1, 2, 2, 1, 1, 2, 2, 1};
    int ID2[] = {0, 2, 1, 1, 2, 2, 1, 1, 2};
    int I1[] = {0, 0, -1, -1, 0, 0, 1, 1, 0};
    int I2[] = {0, -1, -1, -1, -1, 1, 1, 1, 1};
    int J1[] = {0, 1, 0, 0, -1, -1, 0, 0, 1};
    int J2[] = {0, 1, 1, -1, -1, -1, -1, 1, 1};
    float ANGC[] = {0, 0., 1., 1., 2., 2., 3., 3., 4.};
    float ANGF[] = {0, 1., -1., 1., -1., 1., -1., 1., -1.};
    bool diagOutFound = false;  

    for (K = 1; K <= 8; K++) {
        tempShort1 = elev2.getData(J + J1[K], I + I1[K]);
        tempShort2 = elev2.getData(J + J2[K], I + I2[K]);
        
        if (tempShort1 <= 0 && tempShort2 <= 0) { //Both E1 and E2 are outside the flat get slope and angle
            float a = elevDEM.getData(J, I);
            float b = elevDEM.getData(J + J1[K], I + I1[K]);
            float c = elevDEM.getData(J + J2[K], I + I2[K]);
            VSLOPE(
                    a, //E0
                    b, //E1
                    c, //E2
                    DXX[ID1[K]], //dx or dy depending on ID1
                    DXX[ID2[K]], //dx or dy depending on ID2
                    DD, //Hypotenuse
                    &SK[K], //Slope Returned
                    &ANGLE[K]//Angle Returned
                    );
            if (SK[K] >= 0.0) //  Found an outlet
            {
                if (b > a) // Outlet found had better be a diagonal, because it is not an edge
                {
                    if (!diagOutFound) {
                        diagOutFound = true;
                        KD = K;
                    }
                } else { //  Here it is an adjacent outlet
                    KD = K;
                    break;
                }
            }

        } else if (tempShort1 <= 0 && tempShort2 > 0) {//E1 is outside of the flat and E2 is inside the flat. Use DEM elevations. tempShort2/E2 is in the artificial grid
            float a = elevDEM.getData(J, I);
            float b = elevDEM.getData(J + J1[K], I + I1[K]);

            if (a >= b) {
                ANGLE[K] = 0.0;
                SK[K] = 0.0;
                KD = K;
                break;
            }
            short a1 = elev2.getData(J, I);
            short c1 = elev2.getData(J + J2[K], I + I2[K]);
            short b1 = max(a1, c1);
            VSLOPE(
                    (float) a1, //felevg.d[J][I],
                    (float) b1, //[felevg.d[J+J1[K]][I+I1[K]],
                    (float) c1, //felevg.d[J+J2[K]][I+I2[K]],
                    DXX[ID1[K]], //dx or dy
                    DXX[ID2[K]], //dx or dy
                    DD, //Hypotenuse
                    &SK[K], //Slope Returned
                    &ANGLE[K]//Angle Reutnred
                    );
            if (SK[K] > SMAX) {
                SMAX = SK[K];
                KD = K;
            }
        } else if (tempShort1 > 0 && tempShort2 <= 0) {//E2 is out side of the flat and E1 is inside the flat, use DEM elevations
            float a = elevDEM.getData(J, I);
            //float b=elevDEM->getData(J+J1[K],I+I1[K],tempFloat);
            float c = elevDEM.getData(J + J2[K], I + I2[K]);
            if (a >= c) {
                if (!diagOutFound) {
                    ANGLE[K] = (float) atan2(DXX[ID2[K]], DXX[ID1[K]]);
                    SK[K] = 0.0;
                    KD = K;
                    diagOutFound = true;
                }
            } else {
                short a1 = elev2.getData(J, I);
                short b1 = elev2.getData(J + J1[K], I + I1[K]);
                short c1 = max(a1, b1);
                VSLOPE(
                        (float) a1, //felevg.d[J][I],
                        (float) b1, //[felevg.d[J+J1[K]][I+I1[K]],
                        (float) c1, //felevg.d[J+J2[K]][I+I2[K]],
                        DXX[ID1[K]], //dx or dy
                        DXX[ID2[K]], //dx or dy
                        DD, //Hypotenuse
                        &SK[K], //Slope Returned
                        &ANGLE[K]//Angle Reutnred
                        );
                if (SK[K] > SMAX) {
                    SMAX = SK[K];
                    KD = K;
                }

            }
        } else {//Both E1 and E2 are in the flat. Use artificial elevation to get slope and angle
            short a, b, c;
            a = elev2.getData(J, I);
            b = elev2.getData(J + J1[K], I + I1[K]);
            c = elev2.getData(J + J2[K], I + I2[K]);
            VSLOPE(
                    (float) a, //felevg.d[J][I],
                    (float) b, //[felevg.d[J+J1[K]][I+I1[K]],
                    (float) c, //felevg.d[J+J2[K]][I+I2[K]],
                    DXX[ID1[K]], //dx or dy
                    DXX[ID2[K]], //dx or dy
                    DD, //Hypotenuse
                    &SK[K], //Slope Returned
                    &ANGLE[K]//Angle Reutnred
                    );
            if (SK[K] > SMAX) {
                SMAX = SK[K];
                KD = K;
            }
        }
    }
    //USE -1 TO INDICATE DIRECTION NOT YET SET, 
    // but only for non pit grid cells.  Pits will have flowDir as no data
    if (!flowDir.isNodata(J, I)) {
        tempFloat = -1;
        flowDir.setData(J, I, tempFloat);
    }

    if (KD > 0)//We have a flow direction.  Calculate the Angle and save/write it.
    {
        tempFloat = (float) (ANGC[KD]*(PI / 2) + ANGF[KD] * ANGLE[KD]); //Calculate the Angle
        if (tempFloat >= 0.0)//Make sure the angle is positive
            flowDir.setData(J, I, tempFloat); //set the angle in the flowPartition
    }
}