快捷導(dǎo)航

c# NPOI 如何在指定單元格導(dǎo)入導(dǎo)出圖片

更新時(shí)間：2021年03月03日 08:27:06 作者：WeihanLi

這篇文章主要介紹了c# NPOI 如何在指定單元格導(dǎo)入導(dǎo)出圖片，幫助大家更好的理解和學(xué)習(xí)使用c#，感興趣的朋友可以了解下

Intro

我維護(hù)了一個(gè) NPOI 的擴(kuò)展(WeihanLi.Npoi)，主要用來導(dǎo)入導(dǎo)出 Excel 數(shù)據(jù)，最近有網(wǎng)友提出了導(dǎo)入 Excel 的時(shí)候解析圖片的需求，于是就有了本文的探索

導(dǎo)入Excel 時(shí)解析圖片

xls 和 xlsx 的 API 稍有不同，詳細(xì)可以直接參考以下代碼，實(shí)現(xiàn)代碼如下：

public static Dictionary<CellPosition, IPictureData> GetPicturesAndPosition(this ISheet sheet)
{
  var dictionary = new Dictionary<CellPosition, IPictureData>();
  if (sheet.Workbook is HSSFWorkbook)
  {
    foreach (var shape in ((HSSFPatriarch)sheet.DrawingPatriarch).Children)
    {
      if (shape is HSSFPicture picture)
      {
        var position = new CellPosition(picture.ClientAnchor.Row1, picture.ClientAnchor.Col1);
        dictionary[position] = picture.PictureData;
      }
    }
  }
  else if (sheet.Workbook is XSSFWorkbook)
  {
    foreach (var shape in ((XSSFDrawing)sheet.DrawingPatriarch).GetShapes())
    {
      if (shape is XSSFPicture picture)
      {
        var position = new CellPosition(picture.ClientAnchor.Row1, picture.ClientAnchor.Col1);
        dictionary[position] = picture.PictureData;
      }
    }
  }
  return dictionary;
}

CellPosition 是一個(gè)自定義的結(jié)構(gòu)體，表示當(dāng)前單元格的位置，源碼如下：

public readonly struct CellPosition : IEquatable<CellPosition>
{
  public CellPosition(int row, int col)
  {
    Row = row;
    Column = col;
  }

  public int Row { get; }
  public int Column { get; }

  public bool Equals(CellPosition other)
  {
    return Row == other.Row && Column == other.Column;
  }

  public override bool Equals(object? obj) => obj is CellPosition other && Equals(other);

  public override int GetHashCode() => $"{Row}_{Column}".GetHashCode();
}

根據(jù)上面的代碼，我們就可以獲取到獲取到所有的圖片以及圖片的所在位置，這樣根據(jù)單元格位置去找圖片信息的時(shí)候就會(huì)很方便了

導(dǎo)出 Excel 時(shí)設(shè)置圖片

實(shí)現(xiàn)代碼如下：

public static bool TryAddPicture(this ISheet sheet, int row, int col, byte[] pictureBytes, PictureType pictureType = PictureType.PNG)
{
  if (sheet is null)
  {
    throw new ArgumentNullException(nameof(sheet));
  }

  try
  {
    var pictureIndex = sheet.Workbook.AddPicture(pictureBytes, pictureType);

    var clientAnchor = sheet.Workbook.GetCreationHelper().CreateClientAnchor();
    clientAnchor.Row1 = row;
    clientAnchor.Col1 = col;

    var picture = (sheet.DrawingPatriarch ?? sheet.CreateDrawingPatriarch())
      .CreatePicture(clientAnchor, pictureIndex);
    picture.Resize();
    return true;
  }
  catch (Exception e)
  {
    Debug.WriteLine(e);
  }

  return false;
}

通過上面的代碼我們就可以在指定的單元格設(shè)置圖片，目前沒有支持單元格合并操作，有需要自己進(jìn)行修改

WeihanLi.Npoi

WeihanLi.Npoi 在 1.15.0 版本中增加了圖片導(dǎo)入導(dǎo)出的支持，使用示例可以參考下面的單元測試：

[Theory]
[ExcelFormatData]
public async Task ImageImportExportTest(ExcelFormat excelFormat)
{
  using var httpClient = new HttpClient();
  var imageBytes = await httpClient.GetByteArrayAsync("https://weihanli.xyz/assets/avator.jpg");
  var list = Enumerable.Range(1, 5)
    .Select(x => new ImageTest() { Id = x, Image = imageBytes })
    .ToList();
  var excelBytes = list.ToExcelBytes(excelFormat);
  var importResult = ExcelHelper.ToEntityList<ImageTest>(excelBytes, excelFormat);
  Assert.NotNull(importResult);
  Assert.Equal(list.Count, importResult.Count);
  for (var i = 0; i < list.Count; i++)
  {
    Assert.NotNull(importResult[i]);
    var result = importResult[i]!;
    Assert.Equal(list[i].Id, result.Id);
    Assert.NotNull(result.Image);
    Assert.True(list[i].Image.SequenceEqual(result.Image));
  }
}

private class ImageTest
{
  public int Id { get; set; }

  public byte[] Image { get; set; } = null!;
}

導(dǎo)入時(shí)會(huì)自動(dòng)將 byte[] 類型的屬性嘗試獲取對(duì)應(yīng)的單元格位置的圖片，如果在對(duì)應(yīng)的位置找到了圖片就能夠讀取到圖片的字節(jié)數(shù)組信息映射到 model 里的字節(jié)數(shù)組屬性

除此之外，還支持直接導(dǎo)入的時(shí)候?qū)D片信息 Map 到 IPictureData 屬性上，但是覺得還是字節(jié)數(shù)組更通用一些，如果對(duì)此感興趣可以參考項(xiàng)目源碼以及單元測試