Private/Start-GUIChiaPlotting.ps1
|
function Start-GUIChiaPlotting { [CmdletBinding()] param( $ChiaRun, $ChiaQueue, $ChiaJob ) try{ #not really needed, but just wanted to make each parameter its own variable $PlottingParameters = $ChiaRun.PlottingParameters $KSize = $PlottingParameters.KSize.KSizeValue $Buffer = $PlottingParameters.RAM $Threads = $PlottingParameters.Threads $DisableBitfield = $PlottingParameters.DisableBitField $ExcludeFinalDirectory = $PlottingParameters.ExcludeFinalDirectory $TempDirectoryPath = $PlottingParameters.TempVolume.DirectoryPath $FinalDirectoryPath = $PlottingParameters.FinalVolume.DirectoryPath $LogDirectoryPath = $PlottingParameters.LogDirectory $SecondTempDirectoryPath = $PlottingParameters.SecondTempVolume.DirectoryPath $PoolPublicKey = $PlottingParameters.PoolPublicKey $FarmerPublicKey = $PlottingParameters.FarmerPublicKey $Buckets = $PlottingParameters.Buckets $PoolContractEnabled = $PlottingParameters.PoolContractEnabled $PoolContractAddress = $PlottingParameters.PoolContractAddress $ReplotEnabled = $PlottingParameters.ReplotEnabled $AlternativePlotterEnabled = $PlottingParameters.AlternativePlotterEnabled $AlternativePlotterPath = $PlottingParameters.AlternativePlotterPath $Phase3and4Buckets = $PlottingParameters.PhaseThreeFourBuckets $Phase3and4BucketsEnabled = $PlottingParameters.Phase3and4BucketsEnabled $PlotWhileCopy = $PlottingParameters.PlotWhileCopy #used for replotting $OldPlotDeleted = $false $E = if ($DisableBitfield){"-e"} $X = if ($ExcludeFinalDirectory){"-x"} if ($PlotWhileCopy -eq $true){ if ($AlternativePlotterEnabled -eq $true){ $FinalDirectoryPath = $TempDirectoryPath } else{ if (-not[string]::IsNullOrWhiteSpace($SecondTempDirectoryPath)){ $FinalDirectoryPath = $SecondTempDirectoryPath } else{ $FinalDirectoryPath = $TempDirectoryPath } } } #remove any trailing '\' since chia.exe hates them $TempDirectoryPath = $TempDirectoryPath.TrimEnd(@('\','/')) $FinalDirectoryPath = $FinalDirectoryPath.TrimEnd(@('\','/')) #path to chia.exe or alternative plotter $ChiaPath = (Get-Item -Path "$ENV:LOCALAPPDATA\Programs\Chia\resources\app.asar.unpacked\daemon\chia.exe").FullName if ($AlternativePlotterEnabled -eq $true){ $TempDirectoryPath = $TempDirectoryPath + '\' $FinalDirectoryPath = $FinalDirectoryPath + '\' $PlotterPath = $AlternativePlotterPath $ChiaArguments = "-u $Buckets -r $Threads -t $TempDirectoryPath -d $FinalDirectoryPath" if ($Phase3and4BucketsEnabled -eq $true){ $ChiaArguments += " -v $Phase3and4Buckets" } } else{ $PlotterPath = $ChiaPath $ChiaArguments = "plots create -k $KSize -b $Buffer -u $Buckets -r $Threads -t `"$TempDirectoryPath`" -d `"$FinalDirectoryPath`" $E $X" } if ($PoolContractEnabled){ if (-not[string]::IsNullOrEmpty($PoolContractAddress)){ $ChiaArguments += " -c $PoolContractAddress" } } elseif (-not[string]::IsNullOrWhiteSpace($PoolPublicKey)){ $ChiaArguments += " -p $PoolPublicKey" } if (-not[string]::IsNullOrWhiteSpace($FarmerPublicKey)){ $ChiaArguments += " -f $FarmerPublicKey" } if ($ChiaJob.BasicPlotting){ if ($ChiaRun.PlottingParameters.EnableBasicSecondTempDirectory){ if (-not[string]::IsNullOrWhiteSpace($SecondTempDirectoryPath)){ $SecondTempDirectoryPath = $SecondTempDirectoryPath.TrimEnd(@('\','/')) if ($AlternativePlotterEnabled -eq $true){ $SecondTempDirectoryPath = $SecondTempDirectoryPath + '\' $ChiaArguments += " -2 $SecondTempDirectoryPath" } else{ $ChiaArguments += " -2 `"$SecondTempDirectoryPath`"" } } } } else{ if (-not[string]::IsNullOrWhiteSpace($SecondTempDirectoryPath)){ $SecondTempDirectoryPath = $SecondTempDirectoryPath.TrimEnd(@('\','/')) if ($AlternativePlotterEnabled -eq $true){ $SecondTempDirectoryPath = $SecondTempDirectoryPath + '\' $ChiaArguments += " -2 $SecondTempDirectoryPath" } else{ $ChiaArguments += " -2 `"$SecondTempDirectoryPath`"" } } } if ($KSize -eq 25){ $ChiaArguments += " --override-k" } if ($PlotterPath){ Write-Information "Chia path exists, starting the plotting process" try{ $LogPath = Join-Path $LogDirectoryPath ((Get-Date -Format yyyy_MM_dd_hh-mm-ss-tt_) + "plotlog-" + $ChiaQueue.QueueNumber + "-" + $ChiaRun.RunNumber + ".log") $ChiaRun.LogPath = $LogPath $PlottingParam = @{ FilePath = $PlotterPath ArgumentList = $ChiaArguments RedirectStandardOutput = $LogPath } $chiaProcess = Start-Process @PlottingParam -PassThru -WindowStyle Hidden #this is 100% require for th exit code to be seen by powershell when redirectingstandardoutput $handle = $chiaProcess.Handle $ChiaRun.ChiaProcess = $ChiaProcess $ChiaRun.ProcessId = $ChiaProcess.Id $ChiaJob.RunsInProgress.Add($ChiaRun) if (-not$ChiaJob.BasicPlotting){ $TempMasterVolume = $DataHash.MainViewModel.AllVolumes | where UniqueId -eq $ChiaRun.PlottingParameters.TempVolume.UniqueId $TempMasterVolume.CurrentChiaRuns.Add($ChiaRun) $FinalMasterVolume = $DataHash.MainViewModel.AllVolumes | where UniqueId -eq $ChiaRun.PlottingParameters.FinalVolume.UniqueId $FinalMasterVolume.PendingFinalRuns.Add($ChiaRun) } $ChiaQueue.CurrentRun = $ChiaRun $DataHash.MainViewModel.CurrentRuns.Add($ChiaRun) $DataHash.MainViewModel.AllRuns.Add($ChiaRun) #Have noticed that giving the process a second to start before checking the logs works better Start-Sleep 1 while (!$chiaProcess.HasExited){ try{ if ($AlternativePlotterEnabled -eq $true){ $progress = Get-ChiaPlotProgress -LogPath $LogPath -MadMax -ErrorAction Stop } else{ $progress = Get-ChiaPlotProgress -LogPath $LogPath -ErrorAction Stop } $plotid = $progress.PlotId $ChiaRun.Progress = $progress.progress $ChiaRun.PlotId = $plotid $ChiaQueue.CurrentTime = [DateTime]::Now $ChiaRun.CurrentTime = [DateTime]::Now $ChiaRun.Phase = $progress.Phase if ($progress.EST_TimeRemaining.TotalSeconds -le 0){ $ChiaRun.EstTimeRemaining = New-TimeSpan -Seconds 0 } else{ $ChiaRun.EstTimeRemaining = $progress.EST_TimeRemaining } switch ($progress.Phase) { "Phase 1" {$ChiaRun.CurrentPhaseProgress = $progress.Phase1Progess} "Phase 2" {$ChiaRun.CurrentPhaseProgress = $progress.Phase2Progess} "Phase 3" {$ChiaRun.CurrentPhaseProgress = $progress.Phase3Progess} "Phase 4" {$ChiaRun.CurrentPhaseProgress = $progress.Phase4Progess} "Copying" {$ChiaRun.CurrentPhaseProgress = $progress.CopyProgess} } if (($progress.Phase -eq "Phase 4" -or $progress.Phase -eq "Copying" -or $progress.Phase3Progess -gt 75) -and $ReplotEnabled -and -not$OldPlotDeleted){ $OldPlotDeleted = $true $oldDirectories = $ChiaRun.PlottingParameters.FinalVolume.OldPlotDirectories.Path $OldPlot = Get-ChildItem -Path $oldDirectories -Filter "plot-k$($KSize)*.plot" | Select-Object -First 1 if ($Null -ne $OldPlot){ $OldPlot | Remove-Item -Force } } $ChiaRun.TempSize = Get-ChiaTempSize -DirectoryPath $TempDirectoryPath -PlotId $plotid Start-Sleep (5 + $ChiaQueue.QueueNumber) } catch{ Write-PSChiaPlotterLog -LogType ERROR -ErrorObject $_ Start-Sleep 30 } } #while if ($PlotWhileCopy -eq $true -and ($ChiaRun.ChiaProcess.ExitCode -eq 0)){ if (-not$ChiaJob.BasicPlotting){ Start-PlotCopyPhase -RunToCopy $ChiaRun -FinalVol $FinalMasterVolume -TempVol $TempMasterVolume } else{ Start-PlotCopyPhase -RunToCopy $ChiaRun } return #need to return so that the next plot can start } $ChiaJob.RunsInProgress.Remove($ChiaRun) $ChiaJob.CompletedRunCount++ if (-not$ChiaJob.BasicPlotting){ $FinalMasterVolume.PendingFinalRuns.Remove($ChiaRun) $TempMasterVolume.CurrentChiaRuns.Remove($ChiaRun) } $ChiaRun.ExitCode = $ChiaRun.ChiaPRocess.ExitCode #if this is null then an error will occur if we try to set this property if ($ChiaRun.ExitTime){ $ChiaRun.ExitTime = $ChiaProcess.ExitTime } if ($ChiaRun.ChiaProcess.ExitCode -ne 0){ $ChiaRun.Status = "Failed" $ChiaQueue.FailedPlotCount++ $ChiaJob.FailedPlotCount++ $DataHash.MainViewModel.FailedRuns.Add($ChiaRun) Start-Sleep -Seconds 1 Get-ChildItem -Path $TempDirectoryPath -Filter "*$plotid*.tmp" | foreach { try{ Remove-Item -Path $_.FullName -Force -ErrorAction Stop } catch{ Show-Messagebox -Text $_.Exception.Message | Out-Null } } if (-not[string]::IsNullOrWhiteSpace($SecondTempDirectoryPath)){ Get-ChildItem -Path $SecondTempDirectoryPath -Filter "*$plotid*.tmp" | foreach { try{ Remove-Item -Path $_.FullName -Force -ErrorAction Stop } catch{ Show-Messagebox -Text $_.Exception.Message | Out-Null } } #foreach file } } else{ $ChiaRun.Status = "Completed" $ChiaJob.CompletedPlotCount++ $ChiaQueue.CompletedPlotCount++ $DataHash.MainViewModel.CompletedRuns.Add($ChiaRun) $ChiaRun.CheckPlotPowershellCommand = "&'$ChiaPath' plots check -g $plotid" Update-ChiaGUISummary -Success if ($ChiaRun.PlottingParameters.AutoPlotCheckEnabled){ $PlotCheckResults = Test-ChiaPlot -Path $plotid -ErrorAction Continue if ($Null -ne $PlotCheckResults){ $ChiaRun.PlotCheckRatio = [math]::Round($PlotCheckResults.Ratio,2) } } #if autoplotcheck } $ChiaQueue.CurrentRun = $null $DataHash.MainViewModel.CurrentRuns.Remove($ChiaRun) } catch{ if (-not$DataHash.MainViewModel.FailedRuns.Contains($ChiaRun)){ $DataHash.MainViewModel.FailedRuns.Add($ChiaRun) } if ($DataHash.MainViewModel.CurrentRuns.Contains($ChiaRun)){ $DataHash.MainViewModel.CurrentRuns.Remove($ChiaRun) } if ($ChiaJob.RunsInProgress.Contains($ChiaRun)){ $ChiaJob.RunsInProgress.Remove($ChiaRun) } if (-not$ChiaJob.BasicPlotting){ if ($FinalMasterVolume){ if ($FinalMasterVolume.PendingFinalRuns.Contains($ChiaRun)){ $FinalMasterVolume.PendingFinalRuns.Remove($ChiaRun) } } } $PSCmdlet.WriteError($_) } } #if chia path exits else{ $Message = "chia.exe was not found" $ErrorRecord = [System.Management.Automation.ErrorRecord]::new( [System.IO.FileNotFoundException]::new($Message,"$ENV:LOCALAPPDATA\Programs\Chia\resources\app.asar.unpacked\daemon\chia.exe"), 'ChiaPathInvalid', [System.Management.Automation.ErrorCategory]::ObjectNotFound, "$ENV:LOCALAPPDATA\Programs\Chia\resources\app.asar.unpacked\daemon\chia.exe" ) $PSCmdlet.ThrowTerminatingError($ErrorRecord) $PSCmdlet.ThrowTerminatingError("Invalid Log Path Directory: $LogDirectoryPath") } } catch{ Write-PSChiaPlotterLog -LogType ERROR -ErrorObject $_ $PSCmdlet.WriteError($_) } } |