forked from ddrilling/AsbCloudServer
285 lines
10 KiB
C#
285 lines
10 KiB
C#
using AsbCloudDb.Model;
|
||
using ClosedXML.Excel;
|
||
using Microsoft.EntityFrameworkCore;
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.IO;
|
||
using System.Linq;
|
||
using System.Reflection;
|
||
using System.Threading;
|
||
using System.Threading.Tasks;
|
||
using AsbCloudApp.Data.DetectedOperation;
|
||
using AsbCloudInfrastructure.Services.DetectOperations.Detectors;
|
||
using AsbCloudApp.Repositories;
|
||
using Microsoft.AspNetCore.Http.Extensions;
|
||
using AsbCloudApp.Exceptions;
|
||
|
||
namespace AsbCloudInfrastructure.Services.DetectOperations;
|
||
|
||
public class DetectedOperationExportService
|
||
{
|
||
private readonly DetectorAbstract[] detectors =
|
||
{
|
||
new DetectorDrilling(),
|
||
new DetectorSlipsTime()
|
||
};
|
||
|
||
private const int headerRowsCount = 1;
|
||
|
||
private const string cellDepositName = "B1";
|
||
private const string cellClusterName = "B2";
|
||
private const string cellWellName = "B3";
|
||
private const string cellDeltaDate = "H2";
|
||
|
||
private const int columnOperationName = 1;
|
||
private const int columnDateStart = 2;
|
||
private const int columnDateEnd = 3;
|
||
private const int columnDuration = 4;
|
||
private const int columnDepthStart = 5;
|
||
private const int columnDepthEnd = 6;
|
||
private const int columnDeltaDepth = 7;
|
||
private const int columnDepth = 8;
|
||
private const int columnIdReasonOfEnd = 9;
|
||
private const int columnComment = 10;
|
||
|
||
private readonly IAsbCloudDbContext dbContext;
|
||
private readonly IWellOperationRepository wellOperationRepository;
|
||
|
||
public DetectedOperationExportService(IAsbCloudDbContext dbContext, IWellOperationRepository wellOperationRepository)
|
||
{
|
||
this.dbContext = dbContext;
|
||
this.wellOperationRepository = wellOperationRepository;
|
||
}
|
||
|
||
/// <summary>
|
||
/// Экспорт excel файла с операциями по скважине
|
||
/// </summary>
|
||
/// <param name="idWell">ключ скважины</param>
|
||
/// <param name="host">хост</param>
|
||
/// <param name="cancellationToken"></param>
|
||
/// <returns></returns>
|
||
/// <exception cref="ArgumentInvalidException"></exception>
|
||
public async Task<Stream> ExportAsync(int idWell, string host, CancellationToken cancellationToken)
|
||
{
|
||
var well = await dbContext.Set<Well>()
|
||
.Include(w => w.Cluster)
|
||
.ThenInclude(c => c.Deposit)
|
||
.FirstOrDefaultAsync(w => w.Id == idWell, cancellationToken);
|
||
|
||
if (well is null)
|
||
throw new ArgumentInvalidException(nameof(idWell), $"Well {idWell} does not exist");
|
||
|
||
if (!well.IdTelemetry.HasValue)
|
||
throw new ArgumentInvalidException(nameof(idWell), $"Well {idWell} has no telemetry");
|
||
|
||
var operations = await DetectOperationsAsync(well.IdTelemetry.Value, DateTime.UnixEpoch, cancellationToken);
|
||
|
||
return await GenerateExcelFileStreamAsync(well, host, operations, cancellationToken);
|
||
}
|
||
|
||
private async Task<Stream> GenerateExcelFileStreamAsync(Well well, string host, IEnumerable<OperationDetectorResult> operationDetectorResults,
|
||
CancellationToken cancellationToken)
|
||
{
|
||
using var excelTemplateStream = await GetExcelTemplateStreamAsync(cancellationToken);
|
||
|
||
using var workbook = new XLWorkbook(excelTemplateStream, XLEventTracking.Disabled);
|
||
|
||
await AddToWorkbookAsync(workbook, well, host, operationDetectorResults, cancellationToken);
|
||
|
||
MemoryStream memoryStream = new MemoryStream();
|
||
workbook.SaveAs(memoryStream, new SaveOptions { });
|
||
memoryStream.Seek(0, SeekOrigin.Begin);
|
||
return memoryStream;
|
||
}
|
||
|
||
private async Task AddToWorkbookAsync(XLWorkbook workbook, Well well, string host, IEnumerable<OperationDetectorResult> operationDetectorResults,
|
||
CancellationToken cancellationToken)
|
||
{
|
||
const string sheetName = "Операции";
|
||
|
||
if (!operationDetectorResults.Any())
|
||
return;
|
||
|
||
var sheet = workbook.Worksheets.FirstOrDefault(ws => ws.Name == sheetName)
|
||
?? throw new FileFormatException($"Книга excel не содержит листа {sheetName}.");
|
||
|
||
await AddToSheetAsync(sheet, well, host, operationDetectorResults
|
||
.OrderBy(x => x.Operation.DateStart).ThenBy(x => x.Operation.DepthStart).ToArray(),
|
||
cancellationToken);
|
||
}
|
||
|
||
private async Task AddToSheetAsync(IXLWorksheet sheet, Well well, string host, IList<OperationDetectorResult> operationDetectorResults,
|
||
CancellationToken cancellationToken)
|
||
{
|
||
var wellOperationCategories = await dbContext.WellOperationCategories.ToListAsync(cancellationToken);
|
||
|
||
sheet.Cell(cellDepositName).Value = well.Cluster.Deposit.Caption;
|
||
sheet.Cell(cellClusterName).Value = well.Cluster.Caption;
|
||
sheet.Cell(cellWellName).Value = well.Caption;
|
||
sheet.Cell(cellDeltaDate).Value = operationDetectorResults.Max(o => o.Operation.DateEnd) - operationDetectorResults.Min(o => o.Operation.DateStart);
|
||
|
||
var detectedOperations = operationDetectorResults.Select(o => o.Operation).ToArray();
|
||
|
||
for (int i = 0; i < operationDetectorResults.Count; i++)
|
||
{
|
||
var current = detectedOperations[i];
|
||
var dateStart = current.DateStart.ToRemoteDateTime(well.Timezone.Hours);
|
||
var dateEnd = current.DateEnd.ToRemoteDateTime(well.Timezone.Hours);
|
||
|
||
var row = sheet.Row(5 + i + headerRowsCount);
|
||
|
||
row.Cell(columnOperationName).Value = GetCategoryName(wellOperationCategories, current);
|
||
row.Cell(columnDateEnd).Value = dateEnd;
|
||
row.Cell(columnDuration).Value = (dateEnd - dateStart).TotalMinutes;
|
||
row.Cell(columnDepthStart).Value = current.DepthStart;
|
||
row.Cell(columnDepthEnd).Value = current.DepthEnd;
|
||
row.Cell(columnDepth).Value = current.DepthEnd - current.DepthStart;
|
||
|
||
if (current.ExtraData.TryGetValue("IdReasonOfEnd", out object? idReasonOfEndObject)
|
||
&& idReasonOfEndObject is int idReasonOfEnd)
|
||
row.Cell(columnIdReasonOfEnd).Value = GetIdReasonOfEnd(idReasonOfEnd);
|
||
|
||
var query = new QueryBuilder();
|
||
query.Add("end", dateStart.AddSeconds(1800 * 0.9).ToString("yyyy-MM-ddTHH:mm:ss.fff"));
|
||
query.Add("range", "1800");
|
||
|
||
var link = $"{host}/well/{well.Id}/telemetry/monitoring{query}";
|
||
|
||
row.Cell(columnDateStart).Value = dateStart;
|
||
row.Cell(columnDateStart).SetHyperlink(new XLHyperlink(link));
|
||
|
||
row.Cell(columnDeltaDepth).Value = i > 0 && i + 1 < detectedOperations.Length
|
||
? current.DepthStart - detectedOperations[i - 1].DepthEnd
|
||
: 0;
|
||
|
||
row.Cell(columnComment).Value = CreateComment(operationDetectorResults[i]);
|
||
}
|
||
}
|
||
|
||
private static string GetCategoryName(IEnumerable<WellOperationCategory> wellOperationCategories, DetectedOperation current)
|
||
{
|
||
var idCategory = current.IdCategory;
|
||
if (idCategory == WellOperationCategory.IdSlide &&
|
||
EnabledSubsystemsFlags.AutoOscillation.HasEnabledSubsystems(current.EnabledSubsystems))
|
||
return "Бурение в слайде с осцилляцией";
|
||
|
||
var category = wellOperationCategories.FirstOrDefault(o => o.Id == current.IdCategory);
|
||
|
||
if(category is not null)
|
||
return category.Name;
|
||
|
||
return $"Операция №{idCategory}";
|
||
}
|
||
|
||
private static string GetIdReasonOfEnd(int idReasonOfEnd)
|
||
=> idReasonOfEnd switch {
|
||
0 => "Не определена",
|
||
1 => "Не определено начало операции",
|
||
101 => "Разница глубин забоя и положением долота",
|
||
300 => "Низкое давление",
|
||
_ => idReasonOfEnd.ToString($"Причина № {idReasonOfEnd}"),
|
||
|
||
};
|
||
|
||
private async Task<Stream> GetExcelTemplateStreamAsync(CancellationToken cancellationToken)
|
||
{
|
||
string resourceName = Assembly.GetExecutingAssembly()
|
||
.GetManifestResourceNames()
|
||
.FirstOrDefault(n => n.EndsWith("DetectOperations.xlsx"))!;
|
||
|
||
using var stream = Assembly.GetExecutingAssembly()
|
||
.GetManifestResourceStream(resourceName)!;
|
||
|
||
var memoryStream = new MemoryStream();
|
||
await stream.CopyToAsync(memoryStream, cancellationToken);
|
||
memoryStream.Position = 0;
|
||
|
||
return memoryStream;
|
||
}
|
||
|
||
private static string CreateComment(OperationDetectorResult operationDetectorResult)
|
||
{
|
||
var operation = operationDetectorResult.Operation;
|
||
switch (operation.IdCategory)
|
||
{
|
||
case WellOperationCategory.IdRotor:
|
||
case WellOperationCategory.IdSlide:
|
||
var comment = "";
|
||
if (operation.ExtraData.TryGetValue(DetectorDrilling.ExtraDataKeyAvgRotorSpeed, out object? oAvgRotorSpeed))
|
||
comment += $"Средняя скорость оборотов ротора: {oAvgRotorSpeed}\r\n";
|
||
|
||
if (operation.ExtraData.TryGetValue(DetectorDrilling.ExtraDataKeyDispersionOfNormalizedRotorSpeed, out object? oDispersionOfNormalizedRotorSpeed))
|
||
comment += $"Дисперсия нормированных оборотов ротора: {oDispersionOfNormalizedRotorSpeed}";
|
||
|
||
return comment;
|
||
|
||
default:
|
||
return string.Empty;
|
||
}
|
||
}
|
||
|
||
private async Task<IEnumerable<OperationDetectorResult>> DetectOperationsAsync(int idTelemetry, DateTimeOffset begin,
|
||
CancellationToken token)
|
||
{
|
||
var query = dbContext.TelemetryDataSaub
|
||
.AsNoTracking()
|
||
.Where(d => d.IdTelemetry == idTelemetry)
|
||
.Where(d => d.BlockPosition >= 0)
|
||
.Select(d => new DetectableTelemetry
|
||
{
|
||
DateTime = d.DateTime,
|
||
IdUser = d.IdUser,
|
||
WellDepth = d.WellDepth,
|
||
Pressure = d.Pressure,
|
||
HookWeight = d.HookWeight,
|
||
BlockPosition = d.BlockPosition,
|
||
BitDepth = d.BitDepth,
|
||
RotorSpeed = d.RotorSpeed,
|
||
})
|
||
.OrderBy(d => d.DateTime);
|
||
|
||
var startDate = begin;
|
||
var detectedOperationResults = new List<OperationDetectorResult>(8);
|
||
DetectedOperation? lastDetectedOperation = null;
|
||
const int minOperationLength = 5;
|
||
const int maxDetectorsInterpolationFrameLength = 30;
|
||
const int gap = maxDetectorsInterpolationFrameLength + minOperationLength;
|
||
|
||
while (true)
|
||
{
|
||
var data = await query
|
||
.Where(d => d.DateTime > startDate)
|
||
.ToArrayAsync(token);
|
||
|
||
if (data.Length < gap)
|
||
break;
|
||
|
||
var isDetected = false;
|
||
var positionBegin = 0;
|
||
var positionEnd = data.Length - gap;
|
||
while (positionEnd > positionBegin)
|
||
{
|
||
foreach (var detector in detectors)
|
||
{
|
||
if (!detector.TryDetect(idTelemetry, data, positionBegin, positionEnd, lastDetectedOperation, out var result))
|
||
continue;
|
||
|
||
detectedOperationResults.Add(result!);
|
||
lastDetectedOperation = result!.Operation;
|
||
isDetected = true;
|
||
positionBegin = result.TelemetryEnd;
|
||
break;
|
||
}
|
||
|
||
positionBegin += 1;
|
||
}
|
||
|
||
if (isDetected)
|
||
startDate = lastDetectedOperation!.DateEnd;
|
||
else
|
||
startDate = data[positionEnd].DateTime;
|
||
}
|
||
|
||
return detectedOperationResults;
|
||
}
|
||
} |