Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2013 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #include "jni.h" |
| 18 | #include "GraphicsJNI.h" |
Andreas Gampe | ed6b9df | 2014-11-20 22:02:20 -0800 | [diff] [blame] | 19 | #include "core_jni_helpers.h" |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 20 | #include <vector> |
| 21 | |
| 22 | #include "CreateJavaOutputStreamAdaptor.h" |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 23 | |
Matt Sarett | ea70d22 | 2017-03-29 16:25:10 -0400 | [diff] [blame] | 24 | #include "SkColorSpaceXformCanvas.h" |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 25 | #include "SkDocument.h" |
| 26 | #include "SkPicture.h" |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 27 | #include "SkPictureRecorder.h" |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 28 | #include "SkStream.h" |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 29 | #include "SkRect.h" |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 30 | |
sergeyv | dccca44 | 2016-03-21 15:38:21 -0700 | [diff] [blame] | 31 | #include <hwui/Canvas.h> |
| 32 | |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 33 | namespace android { |
| 34 | |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 35 | struct PageRecord { |
Svetoslav | 6811f4e | 2013-09-18 15:58:28 -0700 | [diff] [blame] | 36 | |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 37 | PageRecord(int width, int height, const SkRect& contentRect) |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 38 | : mPictureRecorder(new SkPictureRecorder()) |
| 39 | , mPicture(NULL) |
| 40 | , mWidth(width) |
| 41 | , mHeight(height) { |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 42 | mContentRect = contentRect; |
| 43 | } |
| 44 | |
| 45 | ~PageRecord() { |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 46 | delete mPictureRecorder; |
| 47 | if (NULL != mPicture) { |
| 48 | mPicture->unref(); |
| 49 | } |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 50 | } |
| 51 | |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 52 | SkPictureRecorder* mPictureRecorder; |
| 53 | SkPicture* mPicture; |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 54 | const int mWidth; |
| 55 | const int mHeight; |
| 56 | SkRect mContentRect; |
| 57 | }; |
| 58 | |
| 59 | class PdfDocument { |
| 60 | public: |
| 61 | PdfDocument() { |
| 62 | mCurrentPage = NULL; |
| 63 | } |
| 64 | |
| 65 | SkCanvas* startPage(int width, int height, |
| 66 | int contentLeft, int contentTop, int contentRight, int contentBottom) { |
| 67 | assert(mCurrentPage == NULL); |
| 68 | |
| 69 | SkRect contentRect = SkRect::MakeLTRB( |
| 70 | contentLeft, contentTop, contentRight, contentBottom); |
| 71 | PageRecord* page = new PageRecord(width, height, contentRect); |
| 72 | mPages.push_back(page); |
| 73 | mCurrentPage = page; |
| 74 | |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 75 | SkCanvas* canvas = page->mPictureRecorder->beginRecording( |
Mike Reed | 71487eb | 2014-11-19 16:13:20 -0500 | [diff] [blame] | 76 | SkRect::MakeWH(contentRect.width(), contentRect.height())); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 77 | |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 78 | return canvas; |
| 79 | } |
| 80 | |
| 81 | void finishPage() { |
| 82 | assert(mCurrentPage != NULL); |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 83 | assert(mCurrentPage->mPictureRecorder != NULL); |
| 84 | assert(mCurrentPage->mPicture == NULL); |
Mike Reed | 260ab72 | 2016-10-07 15:59:20 -0400 | [diff] [blame] | 85 | mCurrentPage->mPicture = mCurrentPage->mPictureRecorder->finishRecordingAsPicture().release(); |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 86 | delete mCurrentPage->mPictureRecorder; |
| 87 | mCurrentPage->mPictureRecorder = NULL; |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 88 | mCurrentPage = NULL; |
| 89 | } |
| 90 | |
| 91 | void write(SkWStream* stream) { |
Hal Canary | 4cb7bb5 | 2016-05-02 15:27:51 -0400 | [diff] [blame] | 92 | sk_sp<SkDocument> document = SkDocument::MakePDF(stream); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 93 | for (unsigned i = 0; i < mPages.size(); i++) { |
| 94 | PageRecord* page = mPages[i]; |
| 95 | |
| 96 | SkCanvas* canvas = document->beginPage(page->mWidth, page->mHeight, |
| 97 | &(page->mContentRect)); |
Matt Sarett | ea70d22 | 2017-03-29 16:25:10 -0400 | [diff] [blame] | 98 | std::unique_ptr<SkCanvas> toSRGBCanvas = |
| 99 | SkCreateColorSpaceXformCanvas(canvas, SkColorSpace::MakeSRGB()); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 100 | |
Matt Sarett | ea70d22 | 2017-03-29 16:25:10 -0400 | [diff] [blame] | 101 | toSRGBCanvas->drawPicture(page->mPicture); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 102 | |
| 103 | document->endPage(); |
| 104 | } |
| 105 | document->close(); |
| 106 | } |
| 107 | |
| 108 | void close() { |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 109 | assert(NULL == mCurrentPage); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 110 | for (unsigned i = 0; i < mPages.size(); i++) { |
| 111 | delete mPages[i]; |
| 112 | } |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 113 | } |
| 114 | |
| 115 | private: |
| 116 | ~PdfDocument() { |
| 117 | close(); |
| 118 | } |
| 119 | |
| 120 | std::vector<PageRecord*> mPages; |
| 121 | PageRecord* mCurrentPage; |
| 122 | }; |
| 123 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 124 | static jlong nativeCreateDocument(JNIEnv* env, jobject thiz) { |
| 125 | return reinterpret_cast<jlong>(new PdfDocument()); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 126 | } |
| 127 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 128 | static jlong nativeStartPage(JNIEnv* env, jobject thiz, jlong documentPtr, |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 129 | jint pageWidth, jint pageHeight, |
Svetoslav | 6811f4e | 2013-09-18 15:58:28 -0700 | [diff] [blame] | 130 | jint contentLeft, jint contentTop, jint contentRight, jint contentBottom) { |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 131 | PdfDocument* document = reinterpret_cast<PdfDocument*>(documentPtr); |
Derek Sollenberger | 8872b38 | 2014-06-23 14:13:53 -0400 | [diff] [blame] | 132 | SkCanvas* canvas = document->startPage(pageWidth, pageHeight, |
| 133 | contentLeft, contentTop, contentRight, contentBottom); |
Matt Sarett | ea70d22 | 2017-03-29 16:25:10 -0400 | [diff] [blame] | 134 | return reinterpret_cast<jlong>(Canvas::create_canvas(canvas, Canvas::XformToSRGB::kDefer)); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 135 | } |
| 136 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 137 | static void nativeFinishPage(JNIEnv* env, jobject thiz, jlong documentPtr) { |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 138 | PdfDocument* document = reinterpret_cast<PdfDocument*>(documentPtr); |
| 139 | document->finishPage(); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 140 | } |
| 141 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 142 | static void nativeWriteTo(JNIEnv* env, jobject thiz, jlong documentPtr, jobject out, |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 143 | jbyteArray chunk) { |
| 144 | PdfDocument* document = reinterpret_cast<PdfDocument*>(documentPtr); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 145 | SkWStream* skWStream = CreateJavaOutputStreamAdaptor(env, out, chunk); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 146 | document->write(skWStream); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 147 | delete skWStream; |
| 148 | } |
| 149 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 150 | static void nativeClose(JNIEnv* env, jobject thiz, jlong documentPtr) { |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 151 | PdfDocument* document = reinterpret_cast<PdfDocument*>(documentPtr); |
| 152 | document->close(); |
| 153 | } |
| 154 | |
Daniel Micay | 76f6a86 | 2015-09-19 17:31:01 -0400 | [diff] [blame] | 155 | static const JNINativeMethod gPdfDocument_Methods[] = { |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 156 | {"nativeCreateDocument", "()J", (void*) nativeCreateDocument}, |
| 157 | {"nativeStartPage", "(JIIIIII)J", (void*) nativeStartPage}, |
| 158 | {"nativeFinishPage", "(J)V", (void*) nativeFinishPage}, |
| 159 | {"nativeWriteTo", "(JLjava/io/OutputStream;[B)V", (void*) nativeWriteTo}, |
| 160 | {"nativeClose", "(J)V", (void*) nativeClose} |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 161 | }; |
| 162 | |
Svetoslav | 6811f4e | 2013-09-18 15:58:28 -0700 | [diff] [blame] | 163 | int register_android_graphics_pdf_PdfDocument(JNIEnv* env) { |
Andreas Gampe | ed6b9df | 2014-11-20 22:02:20 -0800 | [diff] [blame] | 164 | return RegisterMethodsOrDie( |
Svetoslav | 6811f4e | 2013-09-18 15:58:28 -0700 | [diff] [blame] | 165 | env, "android/graphics/pdf/PdfDocument", gPdfDocument_Methods, |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 166 | NELEM(gPdfDocument_Methods)); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 167 | } |
| 168 | |
| 169 | }; |