Files
ImportExcel/plot.ps1

152 lines
4.2 KiB
PowerShell

class PSPlot {
hidden $path
hidden $pkg
hidden $ws
hidden $chart
PSPlot() {
$this.path=[System.IO.Path]::GetTempFileName() -replace "\.tmp", ".xlsx"
$this.pkg = New-Object OfficeOpenXml.ExcelPackage $this.path
$this.ws=$this.pkg.Workbook.Worksheets.Add("plot")
}
Plot($yValues) {
$this.NewChart()
$xValues = 0..$yValues.Count
$xCol = 'A'
$yCol = 'B'
$this.AddDataToSheet($xCol,$yCol,'x','y',$xValues,$yValues)
$this.AddSeries($xCol,$yCol,$yValues)
$this.SetChartPosition($yCol)
}
Plot($xValues,$yValues) {
$this.NewChart()
$xCol = 'A'
$yCol = 'B'
$this.AddDataToSheet($xCol,$yCol,'x','y',$xValues,$yValues)
$this.AddSeries($xCol,$yCol,$yValues)
$this.SetChartPosition($yCol)
}
Plot($xValues,$yValues,$x1Values,$y1Values) {
$this.NewChart()
$xCol = 'A'
$yCol = 'B'
$this.AddDataToSheet($xCol,$yCol,'x','y',$xValues,$yValues)
$this.AddSeries($xCol,$yCol,$yValues)
$xCol=$this.GetNextColumnName($yCol)
$yCol=$this.GetNextColumnName($xCol)
$this.AddDataToSheet($xCol,$yCol,'x1','y1',$x1Values,$y1Values)
$this.AddSeries($xCol,$yCol,$y1Values)
$this.SetChartPosition($yCol)
}
Plot($xValues,$yValues,$x1Values,$y1Values,$x2Values,$y2Values) {
$this.NewChart()
$xCol = 'A'
$yCol = 'B'
$this.AddDataToSheet($xCol,$yCol,'x','y',$xValues,$yValues)
$this.AddSeries($xCol,$yCol,$yValues)
$xCol=$this.GetNextColumnName($yCol)
$yCol=$this.GetNextColumnName($xCol)
$this.AddDataToSheet($xCol,$yCol,'x1','y1',$x1Values,$y1Values)
$this.AddSeries($xCol,$yCol,$y1Values)
$xCol=$this.GetNextColumnName($yCol)
$yCol=$this.GetNextColumnName($xCol)
$this.AddDataToSheet($xCol,$yCol,'x2','y2',$x2Values,$y2Values)
$this.AddSeries($xCol,$yCol,$y2Values)
$this.SetChartPosition($yCol)
}
SetChartPosition($yCol) {
$columnNumber = $this.GetColumnNumber($yCol)+1
$this.chart.SetPosition(1,0,$columnNumber,0)
}
AddSeries($xCol,$yCol,$yValues) {
$yRange = "{0}2:{0}{1}" -f $yCol,($yValues.Count+1)
$xRange = "{0}2:{0}{1}" -f $xCol,($yValues.Count+1)
$Series=$this.chart.Series.Add($yRange,$xRange)
}
hidden [string]GetNextColumnName($columnName) {
return $this.GetColumnName($this.GetColumnNumber($columnName)+1)
}
hidden [int]GetColumnNumber($columnName) {
$sum=0
$columnName.ToCharArray() |
ForEach {
$sum*=26
$sum+=[char]$_.tostring().toupper()-[char]'A'+1
}
return $sum
}
hidden [string]GetColumnName($columnNumber) {
$dividend = $columnNumber
$columnName = @()
while($dividend -gt 0) {
$modulo = ($dividend - 1) % 26
$columnName += [char](65 + $modulo)
$dividend = [int](($dividend -$modulo)/26)
}
return ($columnName -join '')
}
hidden AddDataToSheet($xColumn,$yColumn,$xHeader,$yHeader,$xValues,$yValues) {
$count=$yValues.Count
$this.ws.Cells["$($xColumn)1"].Value=$xHeader
$this.ws.Cells["$($yColumn)1"].Value=$yHeader
for ($idx= 0; $idx-lt $count; $idx++) {
$row=$idx+2
$this.ws.Cells["$($xColumn)$($row)"].Value=$xValues[$idx]
$this.ws.Cells["$($yColumn)$($row)"].Value=$yValues[$idx]
}
}
hidden NewChart() {
$chartType="XYScatter"
#$chartType="line"
$this.chart=$this.ws.Drawings.AddChart("plot", $chartType)
$this.chart.Title.Text = 'Plot'
$this.chart.Legend.Remove()
$this.SetChartSize(300,300)
}
SetChartSize([int]$width,[int]$height){
$this.chart.SetSize($width, $height)
}
Show() {
$this.pkg.Save()
$this.pkg.Dispose()
Invoke-Item $this.path
}
}