PDF3D/PDFGenerator/Program.cs
2023-05-12 18:26:26 +08:00

122 lines
4.8 KiB
C#

using iText.IO.Image;
using iText.Kernel.Geom;
using iText.Kernel.Pdf;
using iText.Kernel.Pdf.Annot;
using iText.Kernel.Pdf.Canvas;
using iText.Layout;
using iText.Layout.Element;
using Newtonsoft.Json;
using Org.BouncyCastle.Utilities.IO;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Linq;
using System.Reflection;
public class Program
{
public static string IDTFTool = Directory.GetParent(Assembly.GetExecutingAssembly().Location) + @"\\IDTFTool\\IDTFConverter.exe";
static void Main(string[] args)
{
GeneratePDF(args[0], args[1], args[2]);
}
public static void GenerateU3D(string idtf, string u3d)
{
System.Diagnostics.Process exep = new System.Diagnostics.Process();
System.Diagnostics.ProcessStartInfo startInfo = new System.Diagnostics.ProcessStartInfo();
startInfo.FileName = IDTFTool;
startInfo.Arguments = $"-input \"{idtf}\" -output \"{u3d}\"";
startInfo.CreateNoWindow = true;
startInfo.UseShellExecute = false;
exep.StartInfo = startInfo;
exep.Start();
exep.WaitForExit();
}
public static void GeneratePDF(string idtf, string view, string pdf)
{
var u3d = System.IO.Path.GetTempPath() + System.IO.Path.GetRandomFileName();
GenerateU3D(idtf, u3d);
var writer = new PdfWriter(pdf);
PdfDocument pdfDoc = new PdfDocument(writer);
var size = PageSize.A4;
Document doc = new Document(pdfDoc);
var views = Generate3DView(view);
PdfStream stream3D = new PdfStream(pdfDoc, new FileStream(u3d, FileMode.Open));
stream3D.Put(PdfName.Type, new PdfName("3D"));
stream3D.Put(PdfName.Subtype, new PdfName("U3D"));
stream3D.Put(new PdfName("VA"), new PdfArray(views.ToArray()));
stream3D.SetCompressionLevel(CompressionConstants.DEFAULT_COMPRESSION);
stream3D.Flush();
Rectangle rect = new Rectangle(50, 50, size.GetWidth() - 100, size.GetHeight() - 100);
Pdf3DAnnotation annot = new Pdf3DAnnotation(rect, stream3D);
annot.SetContents(new PdfString("3D Model"));
annot.Put(PdfName._3DV, PdfName.L);//default view
var page = pdfDoc.AddNewPage(size).SetRotation(90);
page.AddAnnotation(annot);
//logo
ImageData imageData = ImageDataFactory.Create(System.IO.Directory.GetParent(Assembly.GetExecutingAssembly().Location) + "\\logo.png");
Image image = new Image(imageData);
image.SetHeight(25);
image.SetMarginLeft(-20);
image.SetMarginTop(-20);
image.SetRotationAngle(Math.PI / 2);
doc.Add(image);
// 关闭文档
pdfDoc.Close();
doc.Close();
}
private static List<PdfDictionary> Generate3DView(string view)
{
List<PdfDictionary> list = new List<PdfDictionary>();
string viewStr = File.ReadAllText(view);
var views = JsonConvert.DeserializeObject<List<IDTF.Net.View>>(viewStr);
foreach (var item in views)
{
if (item.Parents.Any())
{
PdfDictionary dict3D = new PdfDictionary();
dict3D.Put(PdfName.Type, new PdfName("3DView"));
dict3D.Put(new PdfName("XN"), new PdfString(item.Name));
dict3D.Put(new PdfName("IN"), new PdfString(item.Name));
dict3D.Put(new PdfName("RM"), RenderMode());
dict3D.Put(new PdfName("LS"), RenderLight());
dict3D.Put(new PdfName("MS"), PdfName.M);
dict3D.Put(new PdfName("C2W"),
new PdfArray(new double[] { item.Parents[0].Transform.c0r0, item.Parents[0].Transform.c0r1, item.Parents[0].Transform.c0r2,
item.Parents[0].Transform.c1r0, item.Parents[0].Transform.c1r1, item.Parents[0].Transform.c1r2,
item.Parents[0].Transform.c2r0, item.Parents[0].Transform.c2r1, item.Parents[0].Transform.c2r2,
item.Parents[0].Transform.c3r0, item.Parents[0].Transform.c3r1, item.Parents[0].Transform.c3r2 }));
dict3D.Flush();
list.Add(dict3D);
}
}
return list;
}
private static PdfDictionary RenderMode()
{
PdfDictionary dict3D = new PdfDictionary();
dict3D.Put(PdfName.Type, new PdfName("3DRenderMode"));
dict3D.Put(PdfName.Subtype, new PdfName("SolidOutline"));
return dict3D;
}
private static PdfDictionary RenderLight()
{
PdfDictionary dict3D = new PdfDictionary();
dict3D.Put(PdfName.Type, new PdfName("3DLightingScheme"));
dict3D.Put(PdfName.Subtype, new PdfName("Day"));
return dict3D;
}
}