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 | |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 24 | #include "SkDocument.h" |
| 25 | #include "SkPicture.h" |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 26 | #include "SkPictureRecorder.h" |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 27 | #include "SkStream.h" |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 28 | #include "SkRect.h" |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 29 | |
sergeyv | dccca44 | 2016-03-21 15:38:21 -0700 | [diff] [blame] | 30 | #include <hwui/Canvas.h> |
| 31 | |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 32 | namespace android { |
| 33 | |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 34 | struct PageRecord { |
Svetoslav | 6811f4e | 2013-09-18 15:58:28 -0700 | [diff] [blame] | 35 | |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 36 | PageRecord(int width, int height, const SkRect& contentRect) |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 37 | : mPictureRecorder(new SkPictureRecorder()) |
| 38 | , mPicture(NULL) |
| 39 | , mWidth(width) |
| 40 | , mHeight(height) { |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 41 | mContentRect = contentRect; |
| 42 | } |
| 43 | |
| 44 | ~PageRecord() { |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 45 | delete mPictureRecorder; |
| 46 | if (NULL != mPicture) { |
| 47 | mPicture->unref(); |
| 48 | } |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 49 | } |
| 50 | |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 51 | SkPictureRecorder* mPictureRecorder; |
| 52 | SkPicture* mPicture; |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 53 | const int mWidth; |
| 54 | const int mHeight; |
| 55 | SkRect mContentRect; |
| 56 | }; |
| 57 | |
| 58 | class PdfDocument { |
| 59 | public: |
| 60 | PdfDocument() { |
| 61 | mCurrentPage = NULL; |
| 62 | } |
| 63 | |
| 64 | SkCanvas* startPage(int width, int height, |
| 65 | int contentLeft, int contentTop, int contentRight, int contentBottom) { |
| 66 | assert(mCurrentPage == NULL); |
| 67 | |
| 68 | SkRect contentRect = SkRect::MakeLTRB( |
| 69 | contentLeft, contentTop, contentRight, contentBottom); |
| 70 | PageRecord* page = new PageRecord(width, height, contentRect); |
| 71 | mPages.push_back(page); |
| 72 | mCurrentPage = page; |
| 73 | |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 74 | SkCanvas* canvas = page->mPictureRecorder->beginRecording( |
Mike Reed | 71487eb | 2014-11-19 16:13:20 -0500 | [diff] [blame] | 75 | SkRect::MakeWH(contentRect.width(), contentRect.height())); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 76 | |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 77 | return canvas; |
| 78 | } |
| 79 | |
| 80 | void finishPage() { |
| 81 | assert(mCurrentPage != NULL); |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 82 | assert(mCurrentPage->mPictureRecorder != NULL); |
| 83 | assert(mCurrentPage->mPicture == NULL); |
Mike Reed | 260ab72 | 2016-10-07 15:59:20 -0400 | [diff] [blame] | 84 | mCurrentPage->mPicture = mCurrentPage->mPictureRecorder->finishRecordingAsPicture().release(); |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 85 | delete mCurrentPage->mPictureRecorder; |
| 86 | mCurrentPage->mPictureRecorder = NULL; |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 87 | mCurrentPage = NULL; |
| 88 | } |
| 89 | |
| 90 | void write(SkWStream* stream) { |
Hal Canary | 4cb7bb5 | 2016-05-02 15:27:51 -0400 | [diff] [blame] | 91 | sk_sp<SkDocument> document = SkDocument::MakePDF(stream); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 92 | for (unsigned i = 0; i < mPages.size(); i++) { |
| 93 | PageRecord* page = mPages[i]; |
| 94 | |
| 95 | SkCanvas* canvas = document->beginPage(page->mWidth, page->mHeight, |
| 96 | &(page->mContentRect)); |
Derek Sollenberger | fa3e340 | 2017-08-04 08:35:10 -0400 | [diff] [blame] | 97 | canvas->drawPicture(page->mPicture); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 98 | |
| 99 | document->endPage(); |
| 100 | } |
| 101 | document->close(); |
| 102 | } |
| 103 | |
| 104 | void close() { |
Robert Phillips | b59508f | 2014-04-23 12:31:37 -0400 | [diff] [blame] | 105 | assert(NULL == mCurrentPage); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 106 | for (unsigned i = 0; i < mPages.size(); i++) { |
| 107 | delete mPages[i]; |
| 108 | } |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 109 | } |
| 110 | |
| 111 | private: |
| 112 | ~PdfDocument() { |
| 113 | close(); |
| 114 | } |
| 115 | |
| 116 | std::vector<PageRecord*> mPages; |
| 117 | PageRecord* mCurrentPage; |
| 118 | }; |
| 119 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 120 | static jlong nativeCreateDocument(JNIEnv* env, jobject thiz) { |
| 121 | return reinterpret_cast<jlong>(new PdfDocument()); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 122 | } |
| 123 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 124 | static jlong nativeStartPage(JNIEnv* env, jobject thiz, jlong documentPtr, |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 125 | jint pageWidth, jint pageHeight, |
Svetoslav | 6811f4e | 2013-09-18 15:58:28 -0700 | [diff] [blame] | 126 | jint contentLeft, jint contentTop, jint contentRight, jint contentBottom) { |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 127 | PdfDocument* document = reinterpret_cast<PdfDocument*>(documentPtr); |
Derek Sollenberger | 8872b38 | 2014-06-23 14:13:53 -0400 | [diff] [blame] | 128 | SkCanvas* canvas = document->startPage(pageWidth, pageHeight, |
| 129 | contentLeft, contentTop, contentRight, contentBottom); |
Derek Sollenberger | fa3e340 | 2017-08-04 08:35:10 -0400 | [diff] [blame] | 130 | return reinterpret_cast<jlong>(Canvas::create_canvas(canvas)); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 131 | } |
| 132 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 133 | static void nativeFinishPage(JNIEnv* env, jobject thiz, jlong documentPtr) { |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 134 | PdfDocument* document = reinterpret_cast<PdfDocument*>(documentPtr); |
| 135 | document->finishPage(); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 136 | } |
| 137 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 138 | static void nativeWriteTo(JNIEnv* env, jobject thiz, jlong documentPtr, jobject out, |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 139 | jbyteArray chunk) { |
| 140 | PdfDocument* document = reinterpret_cast<PdfDocument*>(documentPtr); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 141 | SkWStream* skWStream = CreateJavaOutputStreamAdaptor(env, out, chunk); |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 142 | document->write(skWStream); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 143 | delete skWStream; |
| 144 | } |
| 145 | |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 146 | static void nativeClose(JNIEnv* env, jobject thiz, jlong documentPtr) { |
Svetoslav | 35aacf2 | 2013-11-06 18:22:13 -0800 | [diff] [blame] | 147 | PdfDocument* document = reinterpret_cast<PdfDocument*>(documentPtr); |
| 148 | document->close(); |
| 149 | } |
| 150 | |
Daniel Micay | 76f6a86 | 2015-09-19 17:31:01 -0400 | [diff] [blame] | 151 | static const JNINativeMethod gPdfDocument_Methods[] = { |
Ashok Bhat | cdf3446 | 2014-01-23 15:29:55 +0000 | [diff] [blame] | 152 | {"nativeCreateDocument", "()J", (void*) nativeCreateDocument}, |
| 153 | {"nativeStartPage", "(JIIIIII)J", (void*) nativeStartPage}, |
| 154 | {"nativeFinishPage", "(J)V", (void*) nativeFinishPage}, |
| 155 | {"nativeWriteTo", "(JLjava/io/OutputStream;[B)V", (void*) nativeWriteTo}, |
| 156 | {"nativeClose", "(J)V", (void*) nativeClose} |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 157 | }; |
| 158 | |
Svetoslav | 6811f4e | 2013-09-18 15:58:28 -0700 | [diff] [blame] | 159 | int register_android_graphics_pdf_PdfDocument(JNIEnv* env) { |
Andreas Gampe | ed6b9df | 2014-11-20 22:02:20 -0800 | [diff] [blame] | 160 | return RegisterMethodsOrDie( |
Svetoslav | 6811f4e | 2013-09-18 15:58:28 -0700 | [diff] [blame] | 161 | env, "android/graphics/pdf/PdfDocument", gPdfDocument_Methods, |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 162 | NELEM(gPdfDocument_Methods)); |
Svetoslav Ganov | ff4adde5 | 2013-06-10 08:47:27 -0700 | [diff] [blame] | 163 | } |
| 164 | |
| 165 | }; |