2ef12d4bb5
Code by Arthur Hsu, original code review: http://codereview.appspot.com/4650060/ Review URL: http://codereview.appspot.com/4639102 git-svn-id: http://skia.googlecode.com/svn/trunk@1812 2bbb7eff-a529-9590-31e7-b0007b416f81
270 lines
11 KiB
C++
270 lines
11 KiB
C++
/*
|
|
* Copyright (C) 2010 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <string>
|
|
|
|
#include "Test.h"
|
|
#include "SkData.h"
|
|
#include "SkPDFCatalog.h"
|
|
#include "SkPDFStream.h"
|
|
#include "SkPDFTypes.h"
|
|
#include "SkScalar.h"
|
|
#include "SkStream.h"
|
|
|
|
class SkPDFTestDict : public SkPDFDict {
|
|
public:
|
|
void getResources(SkTDArray<SkPDFObject*>* resourceList) {
|
|
resourceList->setReserve(resourceList->count() + fResources.count());
|
|
for (int i = 0; i < fResources.count(); i++) {
|
|
resourceList->push(fResources[i]);
|
|
fResources[i]->ref();
|
|
}
|
|
}
|
|
|
|
void addResource(SkPDFObject* object) {
|
|
fResources.append(1, &object);
|
|
}
|
|
|
|
private:
|
|
SkTDArray<SkPDFObject*> fResources;
|
|
};
|
|
|
|
static bool stream_equals(const SkDynamicMemoryWStream& stream, size_t offset,
|
|
const void* buffer, size_t len) {
|
|
SkAutoDataUnref data(stream.copyToData());
|
|
if (offset + len > data.size()) {
|
|
return false;
|
|
}
|
|
return memcmp(data.bytes() + offset, buffer, len) == 0;
|
|
}
|
|
|
|
static void CheckObjectOutput(skiatest::Reporter* reporter, SkPDFObject* obj,
|
|
const std::string& representation,
|
|
bool indirect) {
|
|
SkPDFCatalog catalog;
|
|
size_t directSize = obj->getOutputSize(&catalog, false);
|
|
REPORTER_ASSERT(reporter, directSize == representation.size());
|
|
|
|
SkDynamicMemoryWStream buffer;
|
|
obj->emit(&buffer, &catalog, false);
|
|
REPORTER_ASSERT(reporter, directSize == buffer.getOffset());
|
|
REPORTER_ASSERT(reporter, stream_equals(buffer, 0, representation.c_str(),
|
|
directSize));
|
|
|
|
if (indirect) {
|
|
// Indirect output.
|
|
static char header[] = "1 0 obj\n";
|
|
static size_t headerLen = strlen(header);
|
|
static char footer[] = "\nendobj\n";
|
|
static size_t footerLen = strlen(footer);
|
|
|
|
catalog.addObject(obj, false);
|
|
|
|
size_t indirectSize = obj->getOutputSize(&catalog, true);
|
|
REPORTER_ASSERT(reporter,
|
|
indirectSize == directSize + headerLen + footerLen);
|
|
|
|
buffer.reset();
|
|
obj->emit(&buffer, &catalog, true);
|
|
REPORTER_ASSERT(reporter, indirectSize == buffer.getOffset());
|
|
REPORTER_ASSERT(reporter, stream_equals(buffer, 0, header, headerLen));
|
|
REPORTER_ASSERT(reporter, stream_equals(buffer, headerLen,
|
|
representation.c_str(),
|
|
directSize));
|
|
REPORTER_ASSERT(reporter, stream_equals(buffer, headerLen + directSize,
|
|
footer, footerLen));
|
|
}
|
|
}
|
|
|
|
static void TestCatalog(skiatest::Reporter* reporter) {
|
|
SkPDFCatalog catalog;
|
|
SkRefPtr<SkPDFInt> int1 = new SkPDFInt(1);
|
|
int1->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFInt> int2 = new SkPDFInt(2);
|
|
int2->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFInt> int3 = new SkPDFInt(3);
|
|
int3->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFInt> int1Again(int1.get());
|
|
|
|
catalog.addObject(int1.get(), false);
|
|
catalog.addObject(int2.get(), false);
|
|
catalog.addObject(int3.get(), false);
|
|
|
|
REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int1.get()) == 3);
|
|
REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int2.get()) == 3);
|
|
REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int3.get()) == 3);
|
|
|
|
SkDynamicMemoryWStream buffer;
|
|
catalog.emitObjectNumber(&buffer, int1.get());
|
|
catalog.emitObjectNumber(&buffer, int2.get());
|
|
catalog.emitObjectNumber(&buffer, int3.get());
|
|
catalog.emitObjectNumber(&buffer, int1Again.get());
|
|
char expectedResult[] = "1 02 03 01 0";
|
|
REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult,
|
|
strlen(expectedResult)));
|
|
}
|
|
|
|
static void TestObjectRef(skiatest::Reporter* reporter) {
|
|
SkRefPtr<SkPDFInt> int1 = new SkPDFInt(1);
|
|
int1->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFInt> int2 = new SkPDFInt(2);
|
|
int2->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFObjRef> int2ref = new SkPDFObjRef(int2.get());
|
|
int2ref->unref(); // SkRefPtr and new both took a reference.
|
|
|
|
SkPDFCatalog catalog;
|
|
catalog.addObject(int1.get(), false);
|
|
catalog.addObject(int2.get(), false);
|
|
REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int1.get()) == 3);
|
|
REPORTER_ASSERT(reporter, catalog.getObjectNumberSize(int2.get()) == 3);
|
|
|
|
char expectedResult[] = "2 0 R";
|
|
SkDynamicMemoryWStream buffer;
|
|
int2ref->emitObject(&buffer, &catalog, false);
|
|
REPORTER_ASSERT(reporter, buffer.getOffset() == strlen(expectedResult));
|
|
REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult,
|
|
buffer.getOffset()));
|
|
}
|
|
|
|
static void TestSubstitute(skiatest::Reporter* reporter) {
|
|
SkRefPtr<SkPDFTestDict> proxy = new SkPDFTestDict();
|
|
proxy->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFTestDict> stub = new SkPDFTestDict();
|
|
stub->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFInt> int33 = new SkPDFInt(33);
|
|
int33->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFDict> stubResource = new SkPDFDict();
|
|
stubResource->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFInt> int44 = new SkPDFInt(44);
|
|
int44->unref(); // SkRefPtr and new both took a reference.
|
|
|
|
stub->insert("Value", int33.get());
|
|
stubResource->insert("InnerValue", int44.get());
|
|
stub->addResource(stubResource.get());
|
|
|
|
SkPDFCatalog catalog;
|
|
catalog.addObject(proxy.get(), false);
|
|
catalog.setSubstitute(proxy.get(), stub.get());
|
|
|
|
SkDynamicMemoryWStream buffer;
|
|
proxy->emit(&buffer, &catalog, false);
|
|
catalog.emitSubstituteResources(&buffer, false);
|
|
|
|
char expectedResult[] =
|
|
"<</Value 33\n>>1 0 obj\n<</InnerValue 44\n>>\nendobj\n";
|
|
REPORTER_ASSERT(reporter, buffer.getOffset() == strlen(expectedResult));
|
|
REPORTER_ASSERT(reporter, stream_equals(buffer, 0, expectedResult,
|
|
buffer.getOffset()));
|
|
}
|
|
|
|
static void TestPDFPrimitives(skiatest::Reporter* reporter) {
|
|
SkRefPtr<SkPDFInt> int42 = new SkPDFInt(42);
|
|
int42->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, int42.get(), "42", true);
|
|
|
|
SkRefPtr<SkPDFScalar> realHalf = new SkPDFScalar(SK_ScalarHalf);
|
|
realHalf->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, realHalf.get(), "0.5", true);
|
|
|
|
#if defined(SK_SCALAR_IS_FLOAT)
|
|
SkRefPtr<SkPDFScalar> bigScalar = new SkPDFScalar(110999.75);
|
|
bigScalar->unref(); // SkRefPtr and new both took a reference.
|
|
#if !defined(SK_ALLOW_LARGE_PDF_SCALARS)
|
|
CheckObjectOutput(reporter, bigScalar.get(), "111000", true);
|
|
#else
|
|
CheckObjectOutput(reporter, bigScalar.get(), "110999.75", true);
|
|
|
|
SkRefPtr<SkPDFScalar> biggerScalar = new SkPDFScalar(50000000.1);
|
|
biggerScalar->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, biggerScalar.get(), "50000000", true);
|
|
|
|
SkRefPtr<SkPDFScalar> smallestScalar = new SkPDFScalar(1.0/65536);
|
|
smallestScalar->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, smallestScalar.get(), "0.00001526", true);
|
|
#endif
|
|
#endif
|
|
|
|
SkRefPtr<SkPDFString> stringSimple = new SkPDFString("test ) string ( foo");
|
|
stringSimple->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, stringSimple.get(), "(test \\) string \\( foo)",
|
|
true);
|
|
SkRefPtr<SkPDFString> stringComplex =
|
|
new SkPDFString("\ttest ) string ( foo");
|
|
stringComplex->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, stringComplex.get(),
|
|
"<0974657374202920737472696E67202820666F6F>", true);
|
|
|
|
SkRefPtr<SkPDFName> name = new SkPDFName("Test name\twith#tab");
|
|
name->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, name.get(), "/Test#20name#09with#23tab", false);
|
|
|
|
SkRefPtr<SkPDFArray> array = new SkPDFArray;
|
|
array->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, array.get(), "[]", true);
|
|
array->append(int42.get());
|
|
CheckObjectOutput(reporter, array.get(), "[42]", true);
|
|
array->append(realHalf.get());
|
|
CheckObjectOutput(reporter, array.get(), "[42 0.5]", true);
|
|
SkRefPtr<SkPDFInt> int0 = new SkPDFInt(0);
|
|
int0->unref(); // SkRefPtr and new both took a reference.
|
|
array->append(int0.get());
|
|
CheckObjectOutput(reporter, array.get(), "[42 0.5 0]", true);
|
|
SkRefPtr<SkPDFInt> int1 = new SkPDFInt(1);
|
|
int1->unref(); // SkRefPtr and new both took a reference.
|
|
array->setAt(0, int1.get());
|
|
CheckObjectOutput(reporter, array.get(), "[1 0.5 0]", true);
|
|
|
|
SkRefPtr<SkPDFDict> dict = new SkPDFDict;
|
|
dict->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, dict.get(), "<<>>", true);
|
|
SkRefPtr<SkPDFName> n1 = new SkPDFName("n1");
|
|
n1->unref(); // SkRefPtr and new both took a reference.
|
|
dict->insert(n1.get(), int42.get());
|
|
CheckObjectOutput(reporter, dict.get(), "<</n1 42\n>>", true);
|
|
SkRefPtr<SkPDFName> n2 = new SkPDFName("n2");
|
|
n2->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFName> n3 = new SkPDFName("n3");
|
|
n3->unref(); // SkRefPtr and new both took a reference.
|
|
dict->insert(n2.get(), realHalf.get());
|
|
dict->insert(n3.get(), array.get());
|
|
CheckObjectOutput(reporter, dict.get(),
|
|
"<</n1 42\n/n2 0.5\n/n3 [1 0.5 0]\n>>", true);
|
|
|
|
char streamBytes[] = "Test\nFoo\tBar";
|
|
SkRefPtr<SkMemoryStream> streamData = new SkMemoryStream(
|
|
streamBytes, strlen(streamBytes), true);
|
|
streamData->unref(); // SkRefPtr and new both took a reference.
|
|
SkRefPtr<SkPDFStream> stream = new SkPDFStream(streamData.get());
|
|
stream->unref(); // SkRefPtr and new both took a reference.
|
|
CheckObjectOutput(reporter, stream.get(),
|
|
"<</Length 12\n>> stream\nTest\nFoo\tBar\nendstream",
|
|
true);
|
|
stream->insert(n1.get(), int42.get());
|
|
CheckObjectOutput(reporter, stream.get(),
|
|
"<</Length 12\n/n1 42\n>> stream\nTest\nFoo\tBar"
|
|
"\nendstream",
|
|
true);
|
|
|
|
TestCatalog(reporter);
|
|
|
|
TestObjectRef(reporter);
|
|
|
|
TestSubstitute(reporter);
|
|
}
|
|
|
|
#include "TestClassDef.h"
|
|
DEFINE_TESTCLASS("PDFPrimitives", PDFPrimitivesTestClass, TestPDFPrimitives)
|