internal Spreadsheet(string name, XlsxWriter workbook)
{
this._name = name;
this._sharedStrings = workbook._sharedStrings;
string tmpFile = Path.GetTempFileName();
// Give OS a hint to cache file in memory, if possible
File.SetAttributes(tmpFile, FileAttributes.Temporary);
//Delete on close
this._stream = new FileStream(
tmpFile,
FileMode.Open,
FileAccess.ReadWrite,
FileShare.Read,
BUFFER_SIZE,
FileOptions.DeleteOnClose);
this._writer = XmlWriter.Create(this._stream);
this.InitXmlDoc();
}
public static void Run(string inFile)
{
string logFile = Path.ChangeExtension(inFile, "log");
// string outFile = Path.ChangeExtension(inFile, "bin");
// parameter 3 should NOT be null! epanet2.dll does not check it for NULL.
EpanetException.Check(UnsafeNativeMethods.ENopen(inFile, logFile, string.Empty));
try {
using (var workbook = new XlsxWriter()) {
int nodeCount, linkCount;
EpanetException.Check(UnsafeNativeMethods.ENgetcount(CountType.Node, out nodeCount));
EpanetException.Check(UnsafeNativeMethods.ENgetcount(CountType.Link, out linkCount));
// set string builder buffer size to be at least MAXID chars
var sb = new StringBuilder(UnsafeNativeMethods.MAXID);
var nodesHead = new object[nodeCount + 1];
nodesHead[0] = "Time/Node";
// Fill list of nodes IDs
for (int i = 1; i <= nodeCount; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetnodeid(i, sb));
nodesHead[i] = sb.ToString();
}
var linksHead = new object[linkCount + 1];
linksHead[0] = "Time/Link";
// Fill list of links IDs
for (int i = 1; i <= linkCount; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetlinkid(i, sb));
linksHead[i] = sb.ToString();
}
// Sheet definitions
var sheetsList = new[] {
new { text = "Node head", isNode = true },
new { text = "Node actual demand", isNode = true },
new { text = "Node pressure", isNode = true },
new { text = "Link flows", isNode = false },
new { text = "Link velocity", isNode = false },
new { text = "Link unit headloss", isNode = false },
new { text = "Link setting", isNode = false }
};
// Fill workbook with sheets
for (int i = 0; i < sheetsList.Length; i++)
{
workbook.AddSheet(sheetsList[i].text);
workbook[i].AddRow(sheetsList[i].isNode ? nodesHead : linksHead);
}
var nodeRow = new object[nodeCount + 1];
var linkRow = new object[linkCount + 1];
EpanetException.Check(UnsafeNativeMethods.ENopenH());
EpanetException.Check(UnsafeNativeMethods.ENinitH(SaveOptions.None));
int tstep;
do
{
int t;
EpanetException.Check(UnsafeNativeMethods.ENrunH(out t));
TimeSpan span = TimeSpan.FromSeconds(t);
// Set first cell to current run time
nodeRow[0] =
linkRow[0] =
string.Format("{0:d2}:{1:d2}:{2:d2}", (int)span.TotalHours, span.Minutes, span.Seconds);
WriteStep(workbook, nodeRow, linkRow);
EpanetException.Check(UnsafeNativeMethods.ENnextH(out tstep));
}while (tstep > 0);
workbook.Save(Path.ChangeExtension(inFile, "xlsx"));
}
}
finally {
EpanetException.Check(UnsafeNativeMethods.ENclose());
}
}
/// <summary>Retrivies single time step data from EPANET and writes it to workbook.</summary>
private static void WriteStep(XlsxWriter workbook, object[] nodeRow, object[] linkRow)
{
float value;
// NODES HEADS
for (int i = 1; i < nodeRow.Length; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetnodevalue(i, NodeValue.Head, out value));
nodeRow[i] = value;
}
workbook[0].AddRow(nodeRow);
// NODES DEMANDS
for (int i = 1; i < nodeRow.Length; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetnodevalue(i, NodeValue.Demand, out value));
nodeRow[i] = value;
}
workbook[1].AddRow(nodeRow);
// NODES PRESSURE
for (int i = 1; i < nodeRow.Length; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetnodevalue(i, NodeValue.Pressure, out value));
nodeRow[i] = value;
}
workbook[2].AddRow(nodeRow);
// LINK FLOW
for (int i = 1; i < linkRow.Length; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetlinkvalue(i, LinkValue.Flow, out value));
linkRow[i] = value;
}
workbook[3].AddRow(linkRow);
// LINK VELOCITY
for (int i = 1; i < linkRow.Length; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetlinkvalue(i, LinkValue.Velocity, out value));
linkRow[i] = value;
}
workbook[4].AddRow(linkRow);
// LINK HEADLOSS
for (int i = 1; i < linkRow.Length; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetlinkvalue(i, LinkValue.HeadLoss, out value));
linkRow[i] = value;
}
workbook[5].AddRow(linkRow);
// LINK FRICTION
for (int i = 1; i < linkRow.Length; i++)
{
EpanetException.Check(UnsafeNativeMethods.ENgetlinkvalue(i, LinkValue.Setting, out value));
linkRow[i] = value;
}
workbook[6].AddRow(linkRow);
}
