@kittl/pdfkit 0.17.4 → 0.17.5

package/js/pdfkit.js

@@ -941,14 +941,18 @@ class PDFGradient$1 {
     }
     color = this.doc._normalizeColor(color);
     if (this.stops.length === 0) {
-      if (color.length === 3) {
-        this._colorSpace = 'DeviceRGB';
-      } else if (color.length === 4) {
-        this._colorSpace = 'DeviceCMYK';
-      } else if (color.length === 1) {
-        this._colorSpace = 'DeviceGray';
+      if (this._activeColorProfile) {
+        this._colorSpace = this._activeColorProfile.ref;
       } else {
-        throw new Error('Unknown color space');
+        if (color.length === 3) {
+          this._colorSpace = 'DeviceRGB';
+        } else if (color.length === 4) {
+          this._colorSpace = 'DeviceCMYK';
+        } else if (color.length === 1) {
+          this._colorSpace = 'DeviceGray';
+        } else {
+          throw new Error('Unknown color space');
+        }
       }
     } else if (this._colorSpace === 'DeviceRGB' && color.length !== 3 || this._colorSpace === 'DeviceCMYK' && color.length !== 4 || this._colorSpace === 'DeviceGray' && color.length !== 1) {
       throw new Error('All gradient stops must use the same color space');
@@ -1239,7 +1243,7 @@ var ColorMixin = {
       if (color.length === 3) {
         color = color.map(part => part / 255);
       } else if (color.length === 4) {
-        color = color.map(part => part / 255);
+        color = color.map(part => part / 100);
       }
       return color;
     }
@@ -1315,7 +1319,7 @@ var ColorMixin = {
     this._doOpacity(null, opacity);
     return this;
   },
-  beingColorProfile(label) {
+  beginColorSpace(label) {
     const profile = this._colorProfiles[label];
     if (!profile) {
       throw Error('Invalid color space label, the profile should be set first');
@@ -1323,7 +1327,7 @@ var ColorMixin = {
     this._activeColorProfile = profile;
     return this;
   },
-  endColorProfile() {
+  endColorSpace() {
     this._activeColorProfile = null;
     return this;
   },
@@ -3947,10 +3951,102 @@ class PNGImage {
   }
 }
 
+class BitmapImage {
+  constructor(data, label, properties) {
+    this.label = label;
+    this.data = data;
+    this.width = properties.width;
+    this.height = properties.height;
+    this.channels = properties.channels;
+    this.colorSpace = properties.colorSpace;
+    this.hasAlphaChannel = properties.hasAlphaChannel;
+    this.obj = null;
+  }
+  embed(document) {
+    if (this.obj) {
+      return;
+    }
+    this.document = document;
+    if (!(this.data instanceof Uint8Array && this.data.length === this.channels * this.width * this.height)) {
+      throw Error("Invalid bitmap, data doesn't match the given properties.");
+    }
+    if (this.colorSpace) {
+      const profile = this.document._colorProfiles[this.colorSpace];
+      if (!profile) {
+        throw Error("PDFDocument doesn't support bitmap color space.");
+      }
+      const channels = this.hasAlphaChannel ? this.channels - 1 : this.channels;
+      if (profile.channels !== channels) {
+        throw Error("Color profile doesn't support image channels");
+      }
+      this.colorSpace = profile.ref;
+    } else {
+      if (!this.hasAlphaChannel) {
+        this.colorSpace = this.channels === 4 ? 'DeviceCMYK' : 'DeviceRGB';
+      } else {
+        this.colorSpace = this.channels === 5 ? 'DeviceCMYK' : 'DeviceRGB';
+      }
+    }
+    let pixelData = this.data;
+    let alphaData = undefined;
+    if (this.hasAlphaChannel) {
+      const pixelChannels = this.channels - 1;
+      pixelData = new Uint8Array(pixelChannels * this.width * this.height);
+      alphaData = new Uint8Array(this.width * this.height);
+      if (this.channels === 4) {
+        for (let src = 0, dst = 0, a = 0; a < this.data.length; src += 4, dst += 3, a++) {
+          pixelData[dst] = this.data[src];
+          pixelData[dst + 1] = this.data[src + 1];
+          pixelData[dst + 2] = this.data[src + 2];
+          alphaData[a] = this.data[src + 3];
+        }
+      } else if (this.channels === 5) {
+        for (let src = 0, dst = 0, a = 0; a < this.data.length; src += 5, dst += 4, a++) {
+          pixelData[dst] = this.data[src];
+          pixelData[dst + 1] = this.data[src + 1];
+          pixelData[dst + 2] = this.data[src + 2];
+          pixelData[dst + 3] = this.data[src + 3];
+          alphaData[a] = this.data[src + 4];
+        }
+      } else {
+        for (let src = 0, dst = 0, a = 0; a < this.data.length; src += this.channels, dst += pixelChannels, a++) {
+          for (let j = 0; j < pixelChannels; j++) {
+            pixelData[dst + j] = this.data[src + j];
+          }
+          alphaData[a] = this.data[src + pixelChannels];
+        }
+      }
+    }
+    this.obj = this.document.ref({
+      Type: 'XObject',
+      Subtype: 'Image',
+      BitsPerComponent: 8,
+      Width: this.width,
+      Height: this.height,
+      ColorSpace: this.colorSpace
+    });
+    if (alphaData) {
+      const sMask = this.document.ref({
+        Type: 'XObject',
+        Subtype: 'Image',
+        Height: this.height,
+        Width: this.width,
+        BitsPerComponent: 8,
+        ColorSpace: 'DeviceGray',
+        Decode: [0, 1]
+      });
+      sMask.end(alphaData);
+      this.obj.data['SMask'] = sMask;
+    }
+    this.obj.end(pixelData);
+    this.data = null;
+  }
+}
+
 class PDFImage {
-  static open(src, label) {
+  static open(src, label, properties) {
     let data;
-    if (Buffer.isBuffer(src)) {
+    if (src instanceof Uint8Array || Buffer.isBuffer(src)) {
       data = src;
     } else if (src instanceof ArrayBuffer) {
       data = Buffer.from(new Uint8Array(src));
@@ -3968,6 +4064,9 @@ class PDFImage {
         }
       }
     }
+    if (properties?.isBitmap) {
+      return new BitmapImage(data, label, properties);
+    }
     if (data[0] === 0xff && data[1] === 0xd8) {
       return new JPEG(data, label);
     } else if (data[0] === 0x89 && data.toString('ascii', 1, 4) === 'PNG') {
@@ -4151,13 +4250,13 @@ var ImagesMixin = {
     this.restore();
     return this;
   },
-  openImage(src) {
+  openImage(src, properties) {
     let image;
     if (typeof src === 'string') {
       image = this._imageRegistry[src];
     }
     if (!image) {
-      image = PDFImage.open(src, `I${++this._imageCount}`);
+      image = PDFImage.open(src, `I${++this._imageCount}`, properties);
       if (typeof src === 'string') {
         this._imageRegistry[src] = image;
       }
@@ -6032,8 +6131,7 @@ class ICCProfile {
       N: this.channels,
       Length: this.data.length
     });
-    this.streamRef.write(this.data);
-    this.streamRef.end();
+    this.streamRef.end(this.data);
     this.ref = this.document.ref([`ICCBased ${this.streamRef}`]);
     this.ref.end();
     this.data = null;