/// <summary>
/// Write results to a rester.
/// </summary>
/// <param name="raster">DEM raster</param>
/// <param name="outputDataStore">Output datastore</param>
/// <param name="result">Elevation map with resuluts</param>
private void WriteToRaster(Raster raster, FileSystemDatastore outputDataStore, GeoMap result)
{
raster.SetNoDataValue(0);
raster.SetPixelType(RasterPixelType.DOUBLE);
RasterDataset resultRasterDataset = raster.SaveAs(tmpRasterName, outputDataStore, rasterFormat);
GarbageHelper.Instance.AddGarbage(Path.Combine(outputFolder, tmpRasterName));
Raster resultRaster = resultRasterDataset.CreateRaster(new int[1] {
0
});
resultRaster.Refresh();
if (!resultRaster.CanEdit())
{
MessageBox.Show("Cannot write to raster", "Error", System.Windows.MessageBoxButton.OK, System.Windows.MessageBoxImage.Error);
return;
}
PixelBlock pixelBlock = resultRaster.CreatePixelBlock(resultRaster.GetWidth(), resultRaster.GetHeight());
resultRaster.Read(0, 0, pixelBlock);
pixelBlock.Clear(0);
Array pixel = new double[resultRaster.GetWidth(), resultRaster.GetHeight()];
pixelBlock.SetPixelData(0, result.Transpose());
resultRaster.Write(0, 0, pixelBlock);
resultRaster.Refresh();
resultRaster.SaveAs(SettingsManager.Instance.CurrentSettings.OutputFilename, outputDataStore, rasterFormat);
}
/// <summary>
/// Open a Raster Dataset given a folder and a dataset name.
/// </summary>
/// <param name="folder">Full path to the folder containing the raster dataset.</param>
/// <param name="name">Name of the raster dataset to open.</param>
/// <returns></returns>
public static RasterDataset OpenRasterDataset(string folder, string name)
{
// Create a new raster dataset which is set to null
RasterDataset rasterDatasetToOpen = null;
try
{
// Create a new file system connection path to open raster datasets using the folder path.
FileSystemConnectionPath connectionPath = new FileSystemConnectionPath(new System.Uri(folder), FileSystemDatastoreType.Raster);
// Create a new file system data store for the connection path created above.
FileSystemDatastore dataStore = new FileSystemDatastore(connectionPath);
// Open the raster dataset.
rasterDatasetToOpen = dataStore.OpenDataset <RasterDataset>(name);
// Check if it is not null. If it is show a message box with the appropriate message.
if (rasterDatasetToOpen == null)
{
MessageBox.Show("Failed to open raster dataset: " + name);
}
}
catch (Exception exc)
{
// If an exception occurs, show a message box with the appropriate message.
MessageBox.Show("Exception caught in OpenRasterDataset for raster: " + name + exc.Message);
}
return(rasterDatasetToOpen);
}
/// <summary>
/// Create a new filesystem datastore.
/// </summary>
/// <returns></returns>
private FileSystemDatastore CreateNewDatastore()
{
FileSystemConnectionPath outputConnectionPath = new FileSystemConnectionPath(
new Uri(outputFolder), FileSystemDatastoreType.Raster);
FileSystemDatastore outputDataStore = new FileSystemDatastore(outputConnectionPath);
return(outputDataStore);
}
public static Task <Envelope> GetRasterExtent(string path)
{
Task <Envelope> task = QueuedTask.Run(() =>
{
string directory = System.IO.Path.GetDirectoryName(path);
string fileName = System.IO.Path.GetFileName(path);
FileSystemConnectionPath connectionPath = new FileSystemConnectionPath(new System.Uri(directory), FileSystemDatastoreType.Raster);
FileSystemDatastore dataStore = new FileSystemDatastore(connectionPath);
RasterDataset fileRasterDataset = dataStore.OpenDataset <RasterDataset>(fileName);
Raster raster = fileRasterDataset.CreateFullRaster();
Envelope env = raster.GetExtent();
return(env);
});
return(task);
}
private void ExportRaster(Raster raster, string path, string fileName)
{
// Open data store.
FileSystemConnectionPath connectionPath =
new FileSystemConnectionPath(new Uri(path), FileSystemDatastoreType.Raster);
using (FileSystemDatastore datastore = new FileSystemDatastore(connectionPath))
{
using (raster.SaveAs(fileName + ".tif", datastore, "TIFF"))
{
datastore.Dispose();
raster.Dispose();
}
datastore.Dispose();
}
}
private Raster OpenRasterFromDataset(string path, string rasterDatasetFileName)
{
Raster raster;
FileSystemConnectionPath connectionPath =
new FileSystemConnectionPath(new Uri(path), FileSystemDatastoreType.Raster);
using (FileSystemDatastore datastore = new FileSystemDatastore(connectionPath))
{
using (RasterDataset rasterDataset = datastore.OpenDataset <RasterDataset>(rasterDatasetFileName))
{
// Create a full raster from the raster dataset.
raster = rasterDataset.CreateFullRaster();
rasterDataset.Dispose();
}
datastore.Dispose();
}
return(raster);
}
/// <summary>
/// Asynchrnously start the visual magnitude analysis.
/// </summary>
public async void StartAnalysis()
{
if (!ValidateInputLayers())
{
return;
}
outputFolder = CreateOutputDirectory(outputFolderName);
if (File.Exists(outputFolder + "/" + SettingsManager.Instance.CurrentSettings.OutputFilename))
{
System.Windows.MessageBoxResult messageResult = MessageBox.Show("The output file already exists and will be overwritten. Continue?", "File exists!", System.Windows.MessageBoxButton.OKCancel, System.Windows.MessageBoxImage.Warning);
if (messageResult == System.Windows.MessageBoxResult.OK)
{
GarbageHelper.Instance.AddGarbage(outputFolder + "/" + SettingsManager.Instance.CurrentSettings.OutputFilename);
GarbageHelper.Instance.CleanUp();
}
else
{
return;
}
}
if (File.Exists(outputFolder + "/" + tmpRasterName))
{
GarbageHelper.Instance.AddGarbage(outputFolder + "/" + tmpRasterName);
GarbageHelper.Instance.CleanUp();
}
///most tasks have to run on MCT thread
await QueuedTask.Run(async() => {
outputFolder = CreateOutputDirectory(outputFolderName);
FileSystemDatastore outputDataStore = CreateNewDatastore();
//get viewpoints
Raster raster = SettingsManager.Instance.SelectedDemLayer.GetRaster();
Projection projection = new Projection(raster, outputFolder); //make the detection automatic
try {
if (await projection.CalculateViewpoints(SettingsManager.Instance.SelectedViewpointLayer) == false)
{
MessageBox.Show("Invalid viewpoint layer type.\nOnly points, lines and polylines are supported.", "Error", System.Windows.MessageBoxButton.OK, System.Windows.MessageBoxImage.Error);
return;
}
} catch (Exception) {
if (!SettingsManager.Instance.CurrentSettings.OffsetGlobal)
{
MessageBox.Show("Invalid viewpoint data. Do the viewpoints have OFFSET column specified?", "Error", System.Windows.MessageBoxButton.OK, System.Windows.MessageBoxImage.Error);
}
else if (SettingsManager.Instance.CurrentSettings.WeightedViewpoints)
{
MessageBox.Show("Invalid viewpoint data. Do the viewpoints have WEIGHT column specified?", "Error", System.Windows.MessageBoxButton.OK, System.Windows.MessageBoxImage.Error);
}
else
{
MessageBox.Show("Invalid viewpoint data.", "Error", System.Windows.MessageBoxButton.OK, System.Windows.MessageBoxImage.Error);
}
return;
}
GeoMap elevationMap = CreateElevationMap(raster);
//initialize work manager
WorkManager workManager = new WorkManager(SettingsManager.Instance.CurrentSettings.WorkerThreads);
int invalidViewpointsCount = 0;
foreach (SpatialUtils.ViewpointProps viewpoint in projection)
{
if (viewpoint.Y < 0 || viewpoint.X < 0)
{
invalidViewpointsCount++;
}
else
{
if (SettingsManager.Instance.CurrentSettings.OffsetGlobal)
{
viewpoint.ElevationOffset = SettingsManager.Instance.CurrentSettings.AltOffset;
}
workManager.AddWork(viewpoint);
}
}
if (invalidViewpointsCount > 0)
{
string message = invalidViewpointsCount.ToString()
+ (invalidViewpointsCount == 1
? " viewpoint was invalid or failed to process."
: " viewpoints were invalid or failed to process.");
MessageBox.Show(message, "Ignored viewpoints", System.Windows.MessageBoxButton.OK, System.Windows.MessageBoxImage.Warning);
}
//wait for the calculation of the visual magnitude for all viewpoints to finish
var watch = System.Diagnostics.Stopwatch.StartNew();
workManager.StartWorking(ref elevationMap);
WorkManager.AutoEvent.WaitOne();
GeoMap result = workManager.GetResult();
MessageBox.Show("Computation finished\n------------\nTime: " + watch.ElapsedMilliseconds / 1000 + " seconds\nViewpoints: " + projection.GetViewpointsCount(), "Finished", System.Windows.MessageBoxButton.OK, System.Windows.MessageBoxImage.Information);
//save and display the result
try {
WriteToRaster(raster, outputDataStore, result);
} catch (Exception) {
MessageBox.Show("Cannot write data to raster.", "Error", System.Windows.MessageBoxButton.OK, System.Windows.MessageBoxImage.Error);
}
LayerFactory.Instance.CreateLayer(new Uri(Path.Combine(outputFolder, SettingsManager.Instance.CurrentSettings.OutputFilename)),
MapView.Active.Map);
//clean up temporary files
GarbageHelper.Instance.CleanUp();
});
}
/// <summary>
/// Mask raster pixels based on the rectangle given and save the output in the
/// current project folder.
/// </summary>
/// <param name="geometry">Rectangle to use to mask raster pixels.</param>
public static async void MaskRaster(Geometry geometry)
{
try
{
// Check if there is an active map view.
if (MapView.Active != null)
{
// Get the active map view.
var mapView = MapView.Active;
// Get the list of selected layers.
IReadOnlyList <Layer> selectedLayerList = mapView.GetSelectedLayers();
if (selectedLayerList.Count == 0)
{
// If no layers are selected show a message box with the appropriate message.
MessageBox.Show("No Layers selected. Please select one Raster layer.");
}
else
{
// Get the most recently selected layer.
Layer firstSelectedLayer = mapView.GetSelectedLayers().First();
if (firstSelectedLayer is RasterLayer)
{
// Working with rasters requires the MCT.
await QueuedTask.Run(() =>
{
#region Get the raster dataset from the currently selected layer
// Get the raster layer from the selected layer.
RasterLayer currentRasterLayer = firstSelectedLayer as RasterLayer;
// Get the raster from the current selected raster layer.
Raster inputRaster = currentRasterLayer.GetRaster();
// Get the basic raster dataset from the raster.
BasicRasterDataset basicRasterDataset = inputRaster.GetRasterDataset();
if (!(basicRasterDataset is RasterDataset))
{
// If the dataset is not a raster dataset, show a message box with the appropriate message.
MessageBox.Show("No Raster Layers selected. Please select one Raster layer.");
return;
}
// Get the input raster dataset from the basic raster dataset.
RasterDataset rasterDataset = basicRasterDataset as RasterDataset;
#endregion
#region Save a copy of the raster dataset in the project folder and open it
// Create a full raster from the input raster dataset.
inputRaster = rasterDataset.CreateFullRaster();
// Setup the paths and name of the output file and folder inside the project folder.
string ouputFolderName = "MaskedOuput";
string outputFolder = Path.Combine(Project.Current.HomeFolderPath, ouputFolderName);;
string outputName = "MaskedRaster.tif";
// Delete the output directory if it exists and create it.
// Note: You will need write access to the project directory for this sample to work.
if (Directory.Exists(outputFolder))
{
Directory.Delete(outputFolder, true);
}
Directory.CreateDirectory(outputFolder);
// Create a new file system connection path to open raster datasets using the output folder path.
FileSystemConnectionPath outputConnectionPath = new FileSystemConnectionPath(
new System.Uri(outputFolder), FileSystemDatastoreType.Raster);
// Create a new file system data store for the connection path created above.
FileSystemDatastore outputFileSytemDataStore = new FileSystemDatastore(outputConnectionPath);
// Create a new raster storage definition.
RasterStorageDef rasterStorageDef = new RasterStorageDef();
// Set the pyramid level to 0 meaning no pyramids will be calculated. This is required
// because we are going to change the pixels after we save the raster dataset and if the
// pyramids are calculated prior to that, the pyramids will be incorrect and will have to
// be recalculated.
rasterStorageDef.SetPyramidLevel(0);
// Save a copy of the raster using the file system data store and the raster storage definition.
inputRaster.SaveAs(outputName, outputFileSytemDataStore, "TIFF", rasterStorageDef);
// Open the raster dataset you just saved.
rasterDataset = OpenRasterDataset(outputFolder, outputName);
// Create a full raster from it so we can modify pixels.
Raster outputRaster = rasterDataset.CreateFullRaster();
#endregion
#region Get the Min/Max Row/Column to mask
// If the map spatial reference is different from the spatial reference of the input raster,
// set the map spatial reference on the input raster. This will ensure the map points are
// correctly reprojected to image points.
if (mapView.Map.SpatialReference.Name != inputRaster.GetSpatialReference().Name)
{
inputRaster.SetSpatialReference(mapView.Map.SpatialReference);
}
// Use the MapToPixel method of the input raster to get the row and column values for the
// points of the rectangle.
Tuple <int, int> tlcTuple = inputRaster.MapToPixel(geometry.Extent.XMin, geometry.Extent.YMin);
Tuple <int, int> lrcTuple = inputRaster.MapToPixel(geometry.Extent.XMax, geometry.Extent.YMax);
int minCol = (int)tlcTuple.Item1;
int minRow = (int)tlcTuple.Item2;
int maxCol = (int)lrcTuple.Item1;
int maxRow = (int)lrcTuple.Item2;
// Ensure the min's are less than the max's.
if (maxCol < minCol)
{
int temp = maxCol;
maxCol = minCol;
minCol = temp;
}
if (maxRow < minRow)
{
int temp = maxRow;
maxRow = minRow;
minRow = temp;
}
// Ensure the mins and maxs are within the raster.
minCol = (minCol < 0) ? 0 : minCol;
minRow = (minRow < 0) ? 0 : minRow;
maxCol = (maxCol > outputRaster.GetWidth()) ? outputRaster.GetWidth() : maxCol;
maxRow = (maxRow > outputRaster.GetHeight()) ? outputRaster.GetHeight() : maxRow;
#endregion
#region Mask pixels based on the rectangle drawn by the user
// Calculate the width and height of the pixel block to create.
int pbWidth = maxCol - minCol;
int pbHeight = maxRow - minRow;
// Check to see if the output raster can be edited.
if (!outputRaster.CanEdit())
{
// If not, show a message box with the appropriate message.
MessageBox.Show("Cannot edit raster :(");
return;
}
// Create a new pixel block from the output raster of the height and width calculated above.
PixelBlock currentPixelBlock = outputRaster.CreatePixelBlock(pbWidth, pbHeight);
// Iterate over the bands of the output raster.
for (int plane = 0; plane < currentPixelBlock.GetPlaneCount(); plane++)
{
// For each band, clear the pixel block.
currentPixelBlock.Clear(plane);
//Array noDataMask = currentPixelBlock.GetNoDataMask(plane, true);
//for (int i = 0; i < noDataMask.GetLength(0); i++)
// noDataMask.SetValue(Convert.ToByte(0), i);
//currentPixelBlock.SetNoDataMask(plane, noDataMask);
}
// Write the cleared pixel block to the output raster dataset.
outputRaster.Write(minCol, minRow, currentPixelBlock);
// Refresh the properties of the output raster dataset.
outputRaster.Refresh();
#endregion
// Create a new layer from the masked raster dataset and add it to the map.
LayerFactory.Instance.CreateLayer(new Uri(Path.Combine(outputFolder, outputName)),
mapView.Map);
// Disable the layer representing the original raster dataset.
firstSelectedLayer.SetVisibility(false);
});
}
else
{
// If the selected layer is not a raster layer show a message box with the appropriate message.
MessageBox.Show("No Raster layers selected. Please select one Raster layer.");
}
}
}
}
catch (Exception exc)
{
MessageBox.Show("Exception caught in MaskRaster: " + exc.Message);
}
}
public async void RasterDatasetSnippets()
{
try
{
#region Open raster dataset in a folder.
// Create a FileSystemConnectionPath using the folder path.
FileSystemConnectionPath connectionPath = new FileSystemConnectionPath(new System.Uri(@"C:\Temp"), FileSystemDatastoreType.Raster);
// Create a new FileSystemDatastore using the FileSystemConnectionPath.
FileSystemDatastore dataStore = new FileSystemDatastore(connectionPath);
// Open the raster dataset.
RasterDataset fileRasterDataset = dataStore.OpenDataset <RasterDataset>("Sample.tif");
#endregion
#region Open raster dataset in a geodatabase.
// Create a FileGeodatabaseConnectionPath using the path to the gdb. Note: This can be a path to a .sde file.
FileGeodatabaseConnectionPath geodatabaseConnectionPath = new FileGeodatabaseConnectionPath(new Uri(@"C:\Temp\rasters.gdb"));
// Create a new Geodatabase object using the FileGeodatabaseConnectionPath.
Geodatabase geodatabase = new Geodatabase(geodatabaseConnectionPath);
// Open the raster dataset.
RasterDataset gdbRasterDataset = geodatabase.OpenDataset <RasterDataset>("sample");
#endregion
RasterDataset rasterDataset = fileRasterDataset;
#region Get the raster dataset definition from a raster dataset.
await QueuedTask.Run(() =>
{
RasterDatasetDefinition rasterDatasetDefinition = rasterDataset.GetDefinition();
rasterDatasetDefinition.GetBandCount();
});
#endregion
{
#region Access rows in a raster attribute table.
var raster = MapView.Active.Map.GetLayersAsFlattenedList().OfType <RasterLayer>().FirstOrDefault();
if (raster != null)
{
await QueuedTask.Run(() =>
{
var rasterTbl = raster.GetRaster().GetAttributeTable();
var cursor = rasterTbl.Search();
while (cursor.MoveNext())
{
var row = cursor.Current;
}
});
}
#endregion
}
{
#region Create a raster cursor to iterate through the raster data.
await QueuedTask.Run(() =>
{
// Create a full raster from the raster dataset.
ArcGIS.Core.Data.Raster.Raster raster = rasterDataset.CreateFullRaster();
// Calculate size of pixel blocks to process. Use 1000 or height/width of the raster, whichever is smaller.
int pixelBlockHeight = raster.GetHeight() > 1000 ? 1000 : raster.GetHeight();
int pixelBlockWidth = raster.GetWidth() > 1000 ? 1000 : raster.GetWidth();
// Create the raster cursor using the height and width calculated.
RasterCursor rasterCursor = raster.CreateCursor(pixelBlockWidth, pixelBlockHeight);
// Use a do-while loop to iterate through the pixel blocks of the raster using the raster cursor.
do
{
// Get the current pixel block from the cursor.
using (PixelBlock currentPixelBlock = rasterCursor.Current)
{
// Do something with the pixel block...
}
// Once you are done, move to the next pixel block.
} while (rasterCursor.MoveNext());
});
#endregion
}
{
#region Read and Write pixels from and to a raster dataset using pixel blocks.
await QueuedTask.Run(() =>
{
// Create a full raster from the raster dataset.
ArcGIS.Core.Data.Raster.Raster raster = rasterDataset.CreateFullRaster();
// Calculate size of pixel block to create. Use 128 or height/width of the raster, whichever is smaller.
int pixelBlockHeight = raster.GetHeight() > 128 ? 128 : raster.GetHeight();
int pixelBlockWidth = raster.GetWidth() > 128 ? 128 : raster.GetWidth();
// Create a new (blank) pixel block.
PixelBlock currentPixelBlock = raster.CreatePixelBlock(pixelBlockWidth, pixelBlockHeight);
// Read pixel values from the raster dataset into the pixel block starting from the given top left corner.
raster.Read(0, 0, currentPixelBlock);
// Do something with the pixel block...
// Write the pixel block to the raster dataset starting from the given top left corner.
raster.Write(0, 0, currentPixelBlock);
});
#endregion
}
{
// Create a full raster from the raster dataset.
ArcGIS.Core.Data.Raster.Raster raster = rasterDataset.CreateFullRaster();
// Calculate size of pixel block to create. Use 128 or height/width of the raster, whichever is smaller.
int pixelBlockHeight = raster.GetHeight() > 128 ? 128 : raster.GetHeight();
int pixelBlockWidth = raster.GetWidth() > 128 ? 128 : raster.GetWidth();
// Create a new (blank) pixel block.
PixelBlock currentPixelBlock = raster.CreatePixelBlock(pixelBlockWidth, pixelBlockHeight);
#region Process pixels using a pixel block
await QueuedTask.Run(() =>
{
// Read pixel values from the raster dataset into the pixel block starting from the given top left corner.
raster.Read(0, 0, currentPixelBlock);
// For each plane (band) in the pixel block
for (int plane = 0; plane < currentPixelBlock.GetPlaneCount(); plane++)
{
// Get a copy of the array of pixels from the pixel block corresponding to the current plane.
Array sourcePixels = currentPixelBlock.GetPixelData(plane, true);
// Get the height and width of the pixel block.
int pBHeight = currentPixelBlock.GetHeight();
int pBWidth = currentPixelBlock.GetWidth();
// Iterate through the pixels in the array.
for (int i = 0; i < pBHeight; i++)
{
for (int j = 0; j < pBWidth; j++)
{
// Get the NoData mask value to see if the pixel is a valid pixel.
if (Convert.ToByte(currentPixelBlock.GetNoDataMaskValue(plane, j, i)) == 1)
{
// Get the pixel value from the array and process it (add 5 to the value).
// Note: This is assuming the pixel type is Unisigned 8bit.
int pixelValue = Convert.ToInt16(sourcePixels.GetValue(j, i)) + 5;
// Make sure the pixel value does not go above the range of the pixel type.
pixelValue = pixelValue > 254 ? 254 : pixelValue;
// Set the new pixel value to the array.
// Note: This is assuming the pixel type is Unisigned 8bit.
sourcePixels.SetValue(Convert.ToByte(pixelValue), j, i);
}
}
}
// Set the modified array of pixels back to the pixel block.
currentPixelBlock.SetPixelData(plane, sourcePixels);
}
// Write the pixel block to the raster dataset starting from the given top left corner.
raster.Write(0, 0, currentPixelBlock);
});
#endregion
#region Calculate Raster statistics
//If a raster dataset has statistics, you can create a raster layer and get these statistics by accessing the colorizer.
await QueuedTask.Run(() =>
{
//Accessing the raster layer
var lyr = MapView.Active.Map.GetLayersAsFlattenedList().OfType <BasicRasterLayer>().FirstOrDefault();
//Getting the colorizer
var colorizer = lyr.GetColorizer() as CIMRasterStretchColorizer;
//Accessing the statistics
var stats = colorizer.StretchStats;
var max = stats.max;
var min = stats.min;
});
#endregion
}
}
catch (Exception)
{
}
}
