int main (int nNumberofArgs,char *argv[])
{
//Test for correct input arguments
if (nNumberofArgs!=3)
{
cout << "FATAL ERROR: wrong number inputs. The program needs the path name and the file name" << endl;
exit(EXIT_SUCCESS);
}
string path_name = argv[1];
// make sure there is a slash on the end of the file
string lchar = path_name.substr(path_name.length()-2,1);
string slash = "/";
if (lchar != slash)
{
cout << "You forgot the frontslash at the end of the path. Appending." << endl;
path_name = path_name+slash;
}
string f_name = argv[2];
cout << "\nYou are running the write junctions driver." << endl
<<"IMPORTANT: this has been updated to load an ENVI DEM, whith extension .bil" << endl
<<"You can convert your DEM to this file format using gdal_translate, with -of ENVI" << endl
<<"See documentation at: http://www.geos.ed.ac.uk/~smudd/LSDTT_docs/html/gdal_notes.html" << endl << endl;
cout << "The path is: " << path_name << " and the filename is: " << f_name << endl;
string full_name = path_name+f_name;
ifstream file_info_in;
file_info_in.open(full_name.c_str());
if( file_info_in.fail() )
{
cout << "\nFATAL ERROR: the header file \"" << full_name
<< "\" doesn't exist" << endl;
exit(EXIT_FAILURE);
}
string DEM_name;
string fill_ext = "_fill";
file_info_in >> DEM_name;
int junction_number;
float pruning_threshold;
int threshold;
float A_0;
int minimum_segment_length;
float sigma;
float start_movern;
float d_movern;
float Minimum_Slope;
int n_movern;
int target_nodes;
int n_iterations;
float fraction_dchi_for_variation;
float vertical_interval;
float horizontal_interval;
float area_thin_frac;
int target_skip;
file_info_in >> Minimum_Slope >> threshold >> junction_number
>> pruning_threshold >> A_0 >> minimum_segment_length >> sigma >> start_movern
>> d_movern >> n_movern >> target_nodes >> n_iterations >> fraction_dchi_for_variation
>> vertical_interval >> horizontal_interval >> area_thin_frac >> target_skip;
cout << "Paramters of this run: " << endl
<< "junction number: " << junction_number << endl
<< "pruning_threshold: " << pruning_threshold << endl
<< "threshold: " << threshold << endl
<< "A_0: " << A_0 << endl
<< "minimum_segment_length: " << minimum_segment_length << endl
<< "sigma: " << sigma << endl
<< "start_movern " << start_movern << endl
<< "d_movern: " << d_movern << endl
<< "n_movern: " << n_movern << endl
<< "target_nodes: " << target_nodes << endl
<< "n_iterarions: " << n_iterations << endl
<< "fraction_dchi_for_variation: " << fraction_dchi_for_variation << endl
<< "vertical interval: " << vertical_interval << endl
<< "horizontal interval: " << horizontal_interval << endl
<< "area thinning fraction for SA analysis: " << area_thin_frac << endl
<< "target_skip is: " << target_skip << endl;
string jn_name = itoa(junction_number);
string uscore = "_";
jn_name = uscore+jn_name;
file_info_in.close();
string DEM_f_name = path_name+DEM_name+fill_ext;
string DEM_bil_extension = "bil";
// set no flux boundary conditions
vector<string> boundary_conditions(4);
boundary_conditions[0] = "No";
boundary_conditions[1] = "no flux";
boundary_conditions[2] = "no flux";
boundary_conditions[3] = "No flux";
// load the filled DEM
LSDRaster filled_topo_test((path_name+DEM_name+fill_ext), DEM_bil_extension);
// get a flow info object
LSDFlowInfo FlowInfo(boundary_conditions,filled_topo_test);
// calcualte the distance from outlet
LSDRaster DistanceFromOutlet = FlowInfo.distance_from_outlet();
LSDIndexRaster ContributingPixels = FlowInfo.write_NContributingNodes_to_LSDIndexRaster();
// get the sources
vector<int> sources;
sources = FlowInfo.get_sources_index_threshold(ContributingPixels, threshold);
// now get the junction network
LSDJunctionNetwork ChanNetwork(sources, FlowInfo);
// now get a junction and look for the longest channel upstream
cout << "creating main stem" << endl;
LSDIndexChannel main_stem = ChanNetwork.generate_longest_index_channel_in_basin(junction_number, FlowInfo, DistanceFromOutlet);
cout << "got main stem channel, with n_nodes " << main_stem.get_n_nodes_in_channel() << endl;
string Basin_name = "_basin";
LSDIndexRaster BasinArray = ChanNetwork.extract_basin_from_junction(junction_number,junction_number,FlowInfo);
BasinArray.write_raster((path_name+DEM_name+Basin_name+jn_name),DEM_bil_extension);
// now get the best fit m over n for all the tributaries
int organization_switch = 1;
int pruning_switch = 1;
LSDIndexChannelTree ChannelTree(FlowInfo, ChanNetwork, junction_number, organization_switch,
DistanceFromOutlet, pruning_switch, pruning_threshold);
// print a file that can be ingested bt the chi fitting algorithm
string Chan_fname = "_ChanNet";
string Chan_ext = ".chan";
string Chan_for_chi_ingestion_fname = path_name+DEM_name+Chan_fname+jn_name+Chan_ext;
ChannelTree.print_LSDChannels_for_chi_network_ingestion(FlowInfo,
filled_topo_test, DistanceFromOutlet, Chan_for_chi_ingestion_fname);
ChannelTree.convert_chan_file_for_ArcMap_ingestion(Chan_for_chi_ingestion_fname);
}
int main (int nNumberofArgs,char *argv[])
{
//Test for correct input arguments
if (nNumberofArgs!=3)
{
cout << "FATAL ERROR: wrong number inputs. The program needs the path name and the file name" << endl;
exit(EXIT_SUCCESS);
}
string path_name = argv[1];
string f_name = argv[2];
cout << "The path is: " << path_name << " and the filename is: " << f_name << endl;
string full_name = path_name+f_name;
ifstream file_info_in;
file_info_in.open(full_name.c_str());
if( file_info_in.fail() )
{
cout << "\nFATAL ERROR: the header file \"" << full_name
<< "\" doesn't exist" << endl;
exit(EXIT_FAILURE);
}
string DEM_name;
string fill_ext = "_fill";
file_info_in >> DEM_name;
int threshold;
float Minimum_Slope;
float curv_threshold;
float minimum_catchment_area;
file_info_in >> Minimum_Slope >> threshold >> curv_threshold >> minimum_catchment_area;
// get some file names
string DEM_f_name = path_name+DEM_name+fill_ext;
string DEM_flt_extension = "bil";
// load the DEM
LSDRaster topo_test((path_name+DEM_name), DEM_flt_extension);
LSDRasterSpectral SpectralRaster(topo_test);
// int FilterType = 2;
// float FLow = 0.01;
// float FHigh = 0.1;
// LSDRaster topo_test_filtered = SpectralRaster.fftw2D_filter(FilterType, FLow, FHigh);
LSDRaster topo_test_wiener = SpectralRaster.fftw2D_wiener();
int border_width = 100;
// topo_test_filtered = topo_test_filtered.border_with_nodata(border_width);
topo_test_wiener = topo_test_wiener.border_with_nodata(border_width);
// Set the no flux boundary conditions
vector<string> boundary_conditions(4);
boundary_conditions[0] = "No";
boundary_conditions[1] = "no flux";
boundary_conditions[2] = "no flux";
boundary_conditions[3] = "No flux";
// get the filled file
cout << "Filling the DEM" << endl;
// LSDRaster filled_topo_test = topo_test_filtered.fill(Minimum_Slope);
LSDRaster filled_topo_test = topo_test_wiener.fill(Minimum_Slope);
filled_topo_test.write_raster((DEM_f_name),DEM_flt_extension);
//get a FlowInfo object
LSDFlowInfo FlowInfo(boundary_conditions,filled_topo_test);
LSDIndexRaster ContributingPixels = FlowInfo.write_NContributingNodes_to_LSDIndexRaster();
vector<int> sources;
sources = FlowInfo.get_sources_index_threshold(ContributingPixels, threshold);
// now get the junction network
LSDJunctionNetwork ChanNetwork(sources, FlowInfo);
// Get the valleys using the contour curvature
int surface_fitting_window_radius = 7;
int surface_fitting_window_radius_LW = 25;
vector<LSDRaster> surface_fitting, surface_fitting_LW;
LSDRaster tan_curvature;
LSDRaster tan_curvature_LW;
string curv_name = "_tan_curv";
vector<int> raster_selection(8, 0);
raster_selection[6] = 1;
surface_fitting = topo_test_wiener.calculate_polyfit_surface_metrics(surface_fitting_window_radius, raster_selection);
surface_fitting_LW = topo_test_wiener.calculate_polyfit_surface_metrics(surface_fitting_window_radius_LW, raster_selection);
for(int i = 0; i<int(raster_selection.size()); ++i)
{
if(raster_selection[i]==1)
{
tan_curvature = surface_fitting[i];
tan_curvature.write_raster((path_name+DEM_name+curv_name), DEM_flt_extension);
tan_curvature_LW = surface_fitting[i];
tan_curvature_LW.write_raster((path_name+DEM_name+curv_name+"_LW"), DEM_flt_extension);
}
}
string CH_name = "_CH_Pelletier";
Array2D<float> topography = filled_topo_test.get_RasterData();
Array2D<float> curvature = tan_curvature.get_RasterData();
Array2D<float> curvature_LW = tan_curvature_LW.get_RasterData();
cout << "\tLocating channel heads..." << endl;
vector<int> ChannelHeadNodes = ChanNetwork.calculate_pelletier_channel_heads_DTM(FlowInfo, topography, curv_threshold, curvature,curvature_LW);
// Now filter out false positives along channel according to a threshold
// catchment area
cout << "\tFiltering out false positives..." << endl;
LSDJunctionNetwork ChanNetworkNew(ChannelHeadNodes, FlowInfo);
vector<int> ChannelHeadNodesFilt;
int count = 0;
for(int i = 0; i<int(ChannelHeadNodes.size()); ++i)
{
int upstream_junc = ChanNetworkNew.get_Junction_of_Node(ChannelHeadNodes[i], FlowInfo);
int test_node = ChanNetworkNew.get_penultimate_node_from_stream_link(upstream_junc, FlowInfo);
float catchment_area = float(FlowInfo.retrieve_contributing_pixels_of_node(test_node)) * FlowInfo.get_DataResolution();
if (catchment_area >= minimum_catchment_area) ChannelHeadNodesFilt.push_back(ChannelHeadNodes[i]);
else ++count;
}
cout << "\t...removed " << count << " nodes out of " << ChannelHeadNodes.size() << endl;
FlowInfo.print_vector_of_nodeindices_to_csv_file(ChannelHeadNodesFilt,(path_name+DEM_name+CH_name));
//LSDIndexRaster Channel_heads_raster = FlowInfo.write_NodeIndexVector_to_LSDIndexRaster(ChannelHeadNodesFilt);
//Channel_heads_raster.write_raster((path_name+DEM_name+CH_name),DEM_flt_extension);
//create a channel network based on these channel heads
LSDJunctionNetwork NewChanNetwork(ChannelHeadNodesFilt, FlowInfo);
LSDIndexRaster SOArrayNew = NewChanNetwork.StreamOrderArray_to_LSDIndexRaster();
string SO_name_new = "_SO_Pelletier";
SOArrayNew.write_raster((path_name+DEM_name+SO_name_new),DEM_flt_extension);
}
