mirror of https://github.com/mindoc-org/mindoc.git
parent
5b5752cf1b
commit
8441b7e338
|
@ -10,6 +10,10 @@ MinDoc 的前身是 SmartWiki 文档系统。SmartWiki 是基于 PHP 框架 lara
|
||||||
|
|
||||||
# 安装与使用
|
# 安装与使用
|
||||||
|
|
||||||
|
**Windows 教程: [https://github.com/lifei6671/godoc/blob/master/README_WIN.md](https://github.com/lifei6671/godoc/blob/master/README_WIN.md) **
|
||||||
|
|
||||||
|
**Linux 教程**
|
||||||
|
|
||||||
对于没有Golang使用经验的用户,可以从 [https://github.com/lifei6671/godoc/releases](https://github.com/lifei6671/godoc/releases) 这里下载编译完的程序。
|
对于没有Golang使用经验的用户,可以从 [https://github.com/lifei6671/godoc/releases](https://github.com/lifei6671/godoc/releases) 这里下载编译完的程序。
|
||||||
|
|
||||||
如果有Golang开发经验,建议通过编译安装。
|
如果有Golang开发经验,建议通过编译安装。
|
||||||
|
|
|
@ -0,0 +1,106 @@
|
||||||
|
# Windows 下安装和配置 MinDoc
|
||||||
|
|
||||||
|
**第一步 下载可执行文件**
|
||||||
|
|
||||||
|
请从 [https://github.com/lifei6671/godoc/releases](https://github.com/lifei6671/godoc/releases) 下载最新版的可执行文件,一般文件名为 godoc_linux_amd.tar.gz .
|
||||||
|
|
||||||
|
**第二步 解压压缩包**
|
||||||
|
|
||||||
|
请将刚才下载的文件解压,请执行如下命令解压:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
tar -xzvf godoc_linux_amd64.tar.gz
|
||||||
|
```
|
||||||
|
|
||||||
|
**第三步 创建数据库**
|
||||||
|
|
||||||
|
请创建一个编码为utf8mb4格式的数据库,如果没有GUI管理工具,推荐用下面的脚本创建:
|
||||||
|
|
||||||
|
```sql
|
||||||
|
CREATE DATABASE mindoc_db DEFAULT CHARSET utf8mb4 COLLATE utf8mb4_general_ci;
|
||||||
|
```
|
||||||
|
|
||||||
|
**第四步 配置数据库**
|
||||||
|
|
||||||
|
请将刚才解压目录下 conf/app.conf.example 重名为 app.conf:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cp conf/app.conf.example conf/app.conf
|
||||||
|
```
|
||||||
|
|
||||||
|
同时配置如下节点:
|
||||||
|
|
||||||
|
```ini
|
||||||
|
#数据库配置
|
||||||
|
|
||||||
|
#mysql数据库的IP
|
||||||
|
db_host=127.0.0.1
|
||||||
|
|
||||||
|
#mysql数据库的端口号一般为3306
|
||||||
|
db_port=3306
|
||||||
|
|
||||||
|
#刚才创建的数据库的名称
|
||||||
|
db_database=mindoc_db
|
||||||
|
|
||||||
|
#访问数据库的账号和密码
|
||||||
|
db_username=root
|
||||||
|
db_password=123456
|
||||||
|
|
||||||
|
```
|
||||||
|
|
||||||
|
**第五步 启动程序**
|
||||||
|
|
||||||
|
执行如下命令启动程序:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
./godoc_linux_amd64
|
||||||
|
```
|
||||||
|
|
||||||
|
稍等一分钟,程序会自动初始化数据库,并创建一个超级管理员账号:admin 密码:123456
|
||||||
|
|
||||||
|
此时访问 http://localhost:8181 就能访问 MinDoc 了。
|
||||||
|
|
||||||
|
**第六步 配置代理**
|
||||||
|
|
||||||
|
这一步可选,如果你不想用端口号访问 MinDoc 就需要配置一个代理了。
|
||||||
|
|
||||||
|
Nginx 代理的配置文件如下:
|
||||||
|
|
||||||
|
```ini
|
||||||
|
server {
|
||||||
|
listen 80;
|
||||||
|
|
||||||
|
#此处应该配置你的域名:
|
||||||
|
server_name webhook.iminho.me;
|
||||||
|
|
||||||
|
charset utf-8;
|
||||||
|
|
||||||
|
#此处配置你的访问日志,请手动创建该目录:
|
||||||
|
access_log /var/log/nginx/webhook.iminho.me/access.log;
|
||||||
|
|
||||||
|
root "/var/go/src/go-git-webhook";
|
||||||
|
|
||||||
|
location ~ .*\.(ttf|woff2|eot|otf|map|swf|svg|gif|jpg|jpeg|bmp|png|ico|txt|js|css)$ {
|
||||||
|
|
||||||
|
#此处将路径执行 MinDoc 的跟目录
|
||||||
|
root "/var/go/godoc";
|
||||||
|
expires 30m;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
location / {
|
||||||
|
try_files /_not_exists_ @backend;
|
||||||
|
}
|
||||||
|
|
||||||
|
# 这里为具体的服务代理配置
|
||||||
|
location @backend {
|
||||||
|
proxy_set_header X-Forwarded-For $remote_addr;
|
||||||
|
proxy_set_header Host $http_host;
|
||||||
|
proxy_set_header X-Forwarded-Proto $scheme;
|
||||||
|
|
||||||
|
#此处配置 MinDoc 程序的地址和端口号
|
||||||
|
proxy_pass http://127.0.0.1:8181;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
```
|
|
@ -0,0 +1,96 @@
|
||||||
|
# Windows 下安装和配置 MinDoc
|
||||||
|
|
||||||
|
**第一步 下载可执行文件**
|
||||||
|
|
||||||
|
请从 [https://github.com/lifei6671/godoc/releases](https://github.com/lifei6671/godoc/releases) 下载最新版的可执行文件,一般文件名为 godoc_windows_amd.zip .
|
||||||
|
|
||||||
|
**第二步 解压压缩包**
|
||||||
|
|
||||||
|
请将刚才下载的文件解压,推荐使用好压解压到任意目录。建议不用用中文明明目录名称。
|
||||||
|
|
||||||
|
**第三步 创建数据库**
|
||||||
|
|
||||||
|
请创建一个编码为utf8mb4格式的数据库,如果没有GUI管理工具,推荐用下面的脚本创建:
|
||||||
|
|
||||||
|
```sql
|
||||||
|
CREATE DATABASE mindoc_db DEFAULT CHARSET utf8mb4 COLLATE utf8mb4_general_ci;
|
||||||
|
```
|
||||||
|
|
||||||
|
**第四步 配置数据库**
|
||||||
|
|
||||||
|
请将刚才解压目录下 conf/app.conf.example 重名为 app.conf。同时配置如下节点:
|
||||||
|
|
||||||
|
```ini
|
||||||
|
#数据库配置
|
||||||
|
|
||||||
|
#mysql数据库的IP
|
||||||
|
db_host=127.0.0.1
|
||||||
|
|
||||||
|
#mysql数据库的端口号一般为3306
|
||||||
|
db_port=3306
|
||||||
|
|
||||||
|
#刚才创建的数据库的名称
|
||||||
|
db_database=mindoc_db
|
||||||
|
|
||||||
|
#访问数据库的账号和密码
|
||||||
|
db_username=root
|
||||||
|
db_password=123456
|
||||||
|
|
||||||
|
```
|
||||||
|
|
||||||
|
**第五步 启动程序**
|
||||||
|
|
||||||
|
此时,双击 godoc_windows_amd64.exe 文件,该程序会自动在后台执行,打开任务管理器会看到运行中的程序。
|
||||||
|
|
||||||
|
稍等一分钟,程序会自动初始化数据库,并创建一个超级管理员账号:admin 密码:123456
|
||||||
|
|
||||||
|
此时访问 http://localhost:8181 就能访问 MinDoc 了。
|
||||||
|
|
||||||
|
**第六步 配置代理**
|
||||||
|
|
||||||
|
这一步可选,如果你不想用端口号访问 MinDoc 就需要配置一个代理了。
|
||||||
|
|
||||||
|
推荐使用nginx做前端代理,当然,也可以用IIS做代理。
|
||||||
|
|
||||||
|
IIS的代理教程请参见 : [http://blog.csdn.net/yuanguozhengjust/article/details/23576033?utm_source=tuicool&utm_medium=referral](http://blog.csdn.net/yuanguozhengjust/article/details/23576033?utm_source=tuicool&utm_medium=referral)
|
||||||
|
|
||||||
|
Nginx 代理的配置文件如下:
|
||||||
|
|
||||||
|
```ini
|
||||||
|
server {
|
||||||
|
listen 80;
|
||||||
|
|
||||||
|
#此处应该配置你的域名:
|
||||||
|
server_name webhook.iminho.me;
|
||||||
|
|
||||||
|
charset utf-8;
|
||||||
|
|
||||||
|
#此处配置你的访问日志,请手动创建该目录:
|
||||||
|
access_log /var/log/nginx/webhook.iminho.me/access.log;
|
||||||
|
|
||||||
|
root "/var/go/src/go-git-webhook";
|
||||||
|
|
||||||
|
location ~ .*\.(ttf|woff2|eot|otf|map|swf|svg|gif|jpg|jpeg|bmp|png|ico|txt|js|css)$ {
|
||||||
|
|
||||||
|
#此处将路径执行 MinDoc 的跟目录
|
||||||
|
root "/var/go/godoc";
|
||||||
|
expires 30m;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
location / {
|
||||||
|
try_files /_not_exists_ @backend;
|
||||||
|
}
|
||||||
|
|
||||||
|
# 这里为具体的服务代理配置
|
||||||
|
location @backend {
|
||||||
|
proxy_set_header X-Forwarded-For $remote_addr;
|
||||||
|
proxy_set_header Host $http_host;
|
||||||
|
proxy_set_header X-Forwarded-Proto $scheme;
|
||||||
|
|
||||||
|
#此处配置 MinDoc 程序的地址和端口号
|
||||||
|
proxy_pass http://127.0.0.1:8181;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
```
|
|
@ -29,17 +29,18 @@ func (m *DocumentSearchResult) FindToPager(keyword string,page_index,page_size,m
|
||||||
o := orm.NewOrm()
|
o := orm.NewOrm()
|
||||||
|
|
||||||
offset := (page_index - 1) * page_size
|
offset := (page_index - 1) * page_size
|
||||||
|
keyword = "%"+keyword+"%"
|
||||||
|
|
||||||
if member_id <= 0 {
|
if member_id <= 0 {
|
||||||
sql1 := `SELECT count(doc.document_id) as total_count FROM md_documents AS doc
|
sql1 := `SELECT count(doc.document_id) as total_count FROM md_documents AS doc
|
||||||
LEFT JOIN md_books as book ON doc.book_id = book.book_id
|
LEFT JOIN md_books as book ON doc.book_id = book.book_id
|
||||||
WHERE book.privately_owned = 0 AND (doc.document_name LIKE '%?%' OR doc.release LIKE '%?%') `
|
WHERE book.privately_owned = 0 AND (doc.document_name LIKE ? OR doc.release LIKE ?) `
|
||||||
|
|
||||||
sql2 := `SELECT doc.document_id,doc.modify_time,doc.create_time,doc.document_name,doc.identify,doc.release as description,doc.modify_time,book.identify as book_identify,book.book_name,rel.member_id,member.account AS author FROM md_documents AS doc
|
sql2 := `SELECT doc.document_id,doc.modify_time,doc.create_time,doc.document_name,doc.identify,doc.release as description,doc.modify_time,book.identify as book_identify,book.book_name,rel.member_id,member.account AS author FROM md_documents AS doc
|
||||||
LEFT JOIN md_books as book ON doc.book_id = book.book_id
|
LEFT JOIN md_books as book ON doc.book_id = book.book_id
|
||||||
LEFT JOIN md_relationship AS rel ON book.book_id = rel.book_id AND role_id = 0
|
LEFT JOIN md_relationship AS rel ON book.book_id = rel.book_id AND role_id = 0
|
||||||
LEFT JOIN md_members as member ON rel.member_id = member.member_id
|
LEFT JOIN md_members as member ON rel.member_id = member.member_id
|
||||||
WHERE book.privately_owned = 0 AND (doc.document_name LIKE '%?%' OR doc.release LIKE '%?%')
|
WHERE book.privately_owned = 0 AND (doc.document_name LIKE ? OR doc.release LIKE ?)
|
||||||
ORDER BY doc.document_id DESC LIMIT ?,? `
|
ORDER BY doc.document_id DESC LIMIT ?,? `
|
||||||
|
|
||||||
err = o.Raw(sql1,keyword,keyword).QueryRow(&total_count)
|
err = o.Raw(sql1,keyword,keyword).QueryRow(&total_count)
|
||||||
|
@ -63,7 +64,6 @@ WHERE (book.privately_owned = 0 OR rel.relationship_id > 0) AND (doc.document_n
|
||||||
WHERE (book.privately_owned = 0 OR rel.relationship_id > 0) AND (doc.document_name LIKE ? OR doc.release LIKE ?)
|
WHERE (book.privately_owned = 0 OR rel.relationship_id > 0) AND (doc.document_name LIKE ? OR doc.release LIKE ?)
|
||||||
ORDER BY doc.document_id DESC LIMIT ?,? `
|
ORDER BY doc.document_id DESC LIMIT ?,? `
|
||||||
|
|
||||||
keyword = "%"+keyword+"%"
|
|
||||||
|
|
||||||
err = o.Raw(sql1,keyword,keyword).QueryRow(&total_count)
|
err = o.Raw(sql1,keyword,keyword).QueryRow(&total_count)
|
||||||
if err != nil{
|
if err != nil{
|
||||||
|
|
|
@ -0,0 +1,202 @@
|
||||||
|
|
||||||
|
Apache License
|
||||||
|
Version 2.0, January 2004
|
||||||
|
http://www.apache.org/licenses/
|
||||||
|
|
||||||
|
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||||
|
|
||||||
|
1. Definitions.
|
||||||
|
|
||||||
|
"License" shall mean the terms and conditions for use, reproduction,
|
||||||
|
and distribution as defined by Sections 1 through 9 of this document.
|
||||||
|
|
||||||
|
"Licensor" shall mean the copyright owner or entity authorized by
|
||||||
|
the copyright owner that is granting the License.
|
||||||
|
|
||||||
|
"Legal Entity" shall mean the union of the acting entity and all
|
||||||
|
other entities that control, are controlled by, or are under common
|
||||||
|
control with that entity. For the purposes of this definition,
|
||||||
|
"control" means (i) the power, direct or indirect, to cause the
|
||||||
|
direction or management of such entity, whether by contract or
|
||||||
|
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||||
|
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||||
|
|
||||||
|
"You" (or "Your") shall mean an individual or Legal Entity
|
||||||
|
exercising permissions granted by this License.
|
||||||
|
|
||||||
|
"Source" form shall mean the preferred form for making modifications,
|
||||||
|
including but not limited to software source code, documentation
|
||||||
|
source, and configuration files.
|
||||||
|
|
||||||
|
"Object" form shall mean any form resulting from mechanical
|
||||||
|
transformation or translation of a Source form, including but
|
||||||
|
not limited to compiled object code, generated documentation,
|
||||||
|
and conversions to other media types.
|
||||||
|
|
||||||
|
"Work" shall mean the work of authorship, whether in Source or
|
||||||
|
Object form, made available under the License, as indicated by a
|
||||||
|
copyright notice that is included in or attached to the work
|
||||||
|
(an example is provided in the Appendix below).
|
||||||
|
|
||||||
|
"Derivative Works" shall mean any work, whether in Source or Object
|
||||||
|
form, that is based on (or derived from) the Work and for which the
|
||||||
|
editorial revisions, annotations, elaborations, or other modifications
|
||||||
|
represent, as a whole, an original work of authorship. For the purposes
|
||||||
|
of this License, Derivative Works shall not include works that remain
|
||||||
|
separable from, or merely link (or bind by name) to the interfaces of,
|
||||||
|
the Work and Derivative Works thereof.
|
||||||
|
|
||||||
|
"Contribution" shall mean any work of authorship, including
|
||||||
|
the original version of the Work and any modifications or additions
|
||||||
|
to that Work or Derivative Works thereof, that is intentionally
|
||||||
|
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||||
|
or by an individual or Legal Entity authorized to submit on behalf of
|
||||||
|
the copyright owner. For the purposes of this definition, "submitted"
|
||||||
|
means any form of electronic, verbal, or written communication sent
|
||||||
|
to the Licensor or its representatives, including but not limited to
|
||||||
|
communication on electronic mailing lists, source code control systems,
|
||||||
|
and issue tracking systems that are managed by, or on behalf of, the
|
||||||
|
Licensor for the purpose of discussing and improving the Work, but
|
||||||
|
excluding communication that is conspicuously marked or otherwise
|
||||||
|
designated in writing by the copyright owner as "Not a Contribution."
|
||||||
|
|
||||||
|
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||||
|
on behalf of whom a Contribution has been received by Licensor and
|
||||||
|
subsequently incorporated within the Work.
|
||||||
|
|
||||||
|
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
copyright license to reproduce, prepare Derivative Works of,
|
||||||
|
publicly display, publicly perform, sublicense, and distribute the
|
||||||
|
Work and such Derivative Works in Source or Object form.
|
||||||
|
|
||||||
|
3. Grant of Patent License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
(except as stated in this section) patent license to make, have made,
|
||||||
|
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||||
|
where such license applies only to those patent claims licensable
|
||||||
|
by such Contributor that are necessarily infringed by their
|
||||||
|
Contribution(s) alone or by combination of their Contribution(s)
|
||||||
|
with the Work to which such Contribution(s) was submitted. If You
|
||||||
|
institute patent litigation against any entity (including a
|
||||||
|
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||||
|
or a Contribution incorporated within the Work constitutes direct
|
||||||
|
or contributory patent infringement, then any patent licenses
|
||||||
|
granted to You under this License for that Work shall terminate
|
||||||
|
as of the date such litigation is filed.
|
||||||
|
|
||||||
|
4. Redistribution. You may reproduce and distribute copies of the
|
||||||
|
Work or Derivative Works thereof in any medium, with or without
|
||||||
|
modifications, and in Source or Object form, provided that You
|
||||||
|
meet the following conditions:
|
||||||
|
|
||||||
|
(a) You must give any other recipients of the Work or
|
||||||
|
Derivative Works a copy of this License; and
|
||||||
|
|
||||||
|
(b) You must cause any modified files to carry prominent notices
|
||||||
|
stating that You changed the files; and
|
||||||
|
|
||||||
|
(c) You must retain, in the Source form of any Derivative Works
|
||||||
|
that You distribute, all copyright, patent, trademark, and
|
||||||
|
attribution notices from the Source form of the Work,
|
||||||
|
excluding those notices that do not pertain to any part of
|
||||||
|
the Derivative Works; and
|
||||||
|
|
||||||
|
(d) If the Work includes a "NOTICE" text file as part of its
|
||||||
|
distribution, then any Derivative Works that You distribute must
|
||||||
|
include a readable copy of the attribution notices contained
|
||||||
|
within such NOTICE file, excluding those notices that do not
|
||||||
|
pertain to any part of the Derivative Works, in at least one
|
||||||
|
of the following places: within a NOTICE text file distributed
|
||||||
|
as part of the Derivative Works; within the Source form or
|
||||||
|
documentation, if provided along with the Derivative Works; or,
|
||||||
|
within a display generated by the Derivative Works, if and
|
||||||
|
wherever such third-party notices normally appear. The contents
|
||||||
|
of the NOTICE file are for informational purposes only and
|
||||||
|
do not modify the License. You may add Your own attribution
|
||||||
|
notices within Derivative Works that You distribute, alongside
|
||||||
|
or as an addendum to the NOTICE text from the Work, provided
|
||||||
|
that such additional attribution notices cannot be construed
|
||||||
|
as modifying the License.
|
||||||
|
|
||||||
|
You may add Your own copyright statement to Your modifications and
|
||||||
|
may provide additional or different license terms and conditions
|
||||||
|
for use, reproduction, or distribution of Your modifications, or
|
||||||
|
for any such Derivative Works as a whole, provided Your use,
|
||||||
|
reproduction, and distribution of the Work otherwise complies with
|
||||||
|
the conditions stated in this License.
|
||||||
|
|
||||||
|
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||||
|
any Contribution intentionally submitted for inclusion in the Work
|
||||||
|
by You to the Licensor shall be under the terms and conditions of
|
||||||
|
this License, without any additional terms or conditions.
|
||||||
|
Notwithstanding the above, nothing herein shall supersede or modify
|
||||||
|
the terms of any separate license agreement you may have executed
|
||||||
|
with Licensor regarding such Contributions.
|
||||||
|
|
||||||
|
6. Trademarks. This License does not grant permission to use the trade
|
||||||
|
names, trademarks, service marks, or product names of the Licensor,
|
||||||
|
except as required for reasonable and customary use in describing the
|
||||||
|
origin of the Work and reproducing the content of the NOTICE file.
|
||||||
|
|
||||||
|
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||||
|
agreed to in writing, Licensor provides the Work (and each
|
||||||
|
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||||
|
implied, including, without limitation, any warranties or conditions
|
||||||
|
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||||
|
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||||
|
appropriateness of using or redistributing the Work and assume any
|
||||||
|
risks associated with Your exercise of permissions under this License.
|
||||||
|
|
||||||
|
8. Limitation of Liability. In no event and under no legal theory,
|
||||||
|
whether in tort (including negligence), contract, or otherwise,
|
||||||
|
unless required by applicable law (such as deliberate and grossly
|
||||||
|
negligent acts) or agreed to in writing, shall any Contributor be
|
||||||
|
liable to You for damages, including any direct, indirect, special,
|
||||||
|
incidental, or consequential damages of any character arising as a
|
||||||
|
result of this License or out of the use or inability to use the
|
||||||
|
Work (including but not limited to damages for loss of goodwill,
|
||||||
|
work stoppage, computer failure or malfunction, or any and all
|
||||||
|
other commercial damages or losses), even if such Contributor
|
||||||
|
has been advised of the possibility of such damages.
|
||||||
|
|
||||||
|
9. Accepting Warranty or Additional Liability. While redistributing
|
||||||
|
the Work or Derivative Works thereof, You may choose to offer,
|
||||||
|
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||||
|
or other liability obligations and/or rights consistent with this
|
||||||
|
License. However, in accepting such obligations, You may act only
|
||||||
|
on Your own behalf and on Your sole responsibility, not on behalf
|
||||||
|
of any other Contributor, and only if You agree to indemnify,
|
||||||
|
defend, and hold each Contributor harmless for any liability
|
||||||
|
incurred by, or claims asserted against, such Contributor by reason
|
||||||
|
of your accepting any such warranty or additional liability.
|
||||||
|
|
||||||
|
END OF TERMS AND CONDITIONS
|
||||||
|
|
||||||
|
APPENDIX: How to apply the Apache License to your work.
|
||||||
|
|
||||||
|
To apply the Apache License to your work, attach the following
|
||||||
|
boilerplate notice, with the fields enclosed by brackets "[]"
|
||||||
|
replaced with your own identifying information. (Don't include
|
||||||
|
the brackets!) The text should be enclosed in the appropriate
|
||||||
|
comment syntax for the file format. We also recommend that a
|
||||||
|
file or class name and description of purpose be included on the
|
||||||
|
same "printed page" as the copyright notice for easier
|
||||||
|
identification within third-party archives.
|
||||||
|
|
||||||
|
Copyright [yyyy] [name of copyright owner]
|
||||||
|
|
||||||
|
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.
|
|
@ -0,0 +1,684 @@
|
||||||
|
/*
|
||||||
|
Copyright 2011 Google Inc.
|
||||||
|
|
||||||
|
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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
// Package memcache provides a client for the memcached cache server.
|
||||||
|
package memcache
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"net"
|
||||||
|
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Similar to:
|
||||||
|
// http://code.google.com/appengine/docs/go/memcache/reference.html
|
||||||
|
|
||||||
|
var (
|
||||||
|
// ErrCacheMiss means that a Get failed because the item wasn't present.
|
||||||
|
ErrCacheMiss = errors.New("memcache: cache miss")
|
||||||
|
|
||||||
|
// ErrCASConflict means that a CompareAndSwap call failed due to the
|
||||||
|
// cached value being modified between the Get and the CompareAndSwap.
|
||||||
|
// If the cached value was simply evicted rather than replaced,
|
||||||
|
// ErrNotStored will be returned instead.
|
||||||
|
ErrCASConflict = errors.New("memcache: compare-and-swap conflict")
|
||||||
|
|
||||||
|
// ErrNotStored means that a conditional write operation (i.e. Add or
|
||||||
|
// CompareAndSwap) failed because the condition was not satisfied.
|
||||||
|
ErrNotStored = errors.New("memcache: item not stored")
|
||||||
|
|
||||||
|
// ErrServer means that a server error occurred.
|
||||||
|
ErrServerError = errors.New("memcache: server error")
|
||||||
|
|
||||||
|
// ErrNoStats means that no statistics were available.
|
||||||
|
ErrNoStats = errors.New("memcache: no statistics available")
|
||||||
|
|
||||||
|
// ErrMalformedKey is returned when an invalid key is used.
|
||||||
|
// Keys must be at maximum 250 bytes long and not
|
||||||
|
// contain whitespace or control characters.
|
||||||
|
ErrMalformedKey = errors.New("malformed: key is too long or contains invalid characters")
|
||||||
|
|
||||||
|
// ErrNoServers is returned when no servers are configured or available.
|
||||||
|
ErrNoServers = errors.New("memcache: no servers configured or available")
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// DefaultTimeout is the default socket read/write timeout.
|
||||||
|
DefaultTimeout = 100 * time.Millisecond
|
||||||
|
|
||||||
|
// DefaultMaxIdleConns is the default maximum number of idle connections
|
||||||
|
// kept for any single address.
|
||||||
|
DefaultMaxIdleConns = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
const buffered = 8 // arbitrary buffered channel size, for readability
|
||||||
|
|
||||||
|
// resumableError returns true if err is only a protocol-level cache error.
|
||||||
|
// This is used to determine whether or not a server connection should
|
||||||
|
// be re-used or not. If an error occurs, by default we don't reuse the
|
||||||
|
// connection, unless it was just a cache error.
|
||||||
|
func resumableError(err error) bool {
|
||||||
|
switch err {
|
||||||
|
case ErrCacheMiss, ErrCASConflict, ErrNotStored, ErrMalformedKey:
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func legalKey(key string) bool {
|
||||||
|
if len(key) > 250 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
for i := 0; i < len(key); i++ {
|
||||||
|
if key[i] <= ' ' || key[i] == 0x7f {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
crlf = []byte("\r\n")
|
||||||
|
space = []byte(" ")
|
||||||
|
resultOK = []byte("OK\r\n")
|
||||||
|
resultStored = []byte("STORED\r\n")
|
||||||
|
resultNotStored = []byte("NOT_STORED\r\n")
|
||||||
|
resultExists = []byte("EXISTS\r\n")
|
||||||
|
resultNotFound = []byte("NOT_FOUND\r\n")
|
||||||
|
resultDeleted = []byte("DELETED\r\n")
|
||||||
|
resultEnd = []byte("END\r\n")
|
||||||
|
resultOk = []byte("OK\r\n")
|
||||||
|
resultTouched = []byte("TOUCHED\r\n")
|
||||||
|
|
||||||
|
resultClientErrorPrefix = []byte("CLIENT_ERROR ")
|
||||||
|
)
|
||||||
|
|
||||||
|
// New returns a memcache client using the provided server(s)
|
||||||
|
// with equal weight. If a server is listed multiple times,
|
||||||
|
// it gets a proportional amount of weight.
|
||||||
|
func New(server ...string) *Client {
|
||||||
|
ss := new(ServerList)
|
||||||
|
ss.SetServers(server...)
|
||||||
|
return NewFromSelector(ss)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewFromSelector returns a new Client using the provided ServerSelector.
|
||||||
|
func NewFromSelector(ss ServerSelector) *Client {
|
||||||
|
return &Client{selector: ss}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Client is a memcache client.
|
||||||
|
// It is safe for unlocked use by multiple concurrent goroutines.
|
||||||
|
type Client struct {
|
||||||
|
// Timeout specifies the socket read/write timeout.
|
||||||
|
// If zero, DefaultTimeout is used.
|
||||||
|
Timeout time.Duration
|
||||||
|
|
||||||
|
// MaxIdleConns specifies the maximum number of idle connections that will
|
||||||
|
// be maintained per address. If less than one, DefaultMaxIdleConns will be
|
||||||
|
// used.
|
||||||
|
//
|
||||||
|
// Consider your expected traffic rates and latency carefully. This should
|
||||||
|
// be set to a number higher than your peak parallel requests.
|
||||||
|
MaxIdleConns int
|
||||||
|
|
||||||
|
selector ServerSelector
|
||||||
|
|
||||||
|
lk sync.Mutex
|
||||||
|
freeconn map[string][]*conn
|
||||||
|
}
|
||||||
|
|
||||||
|
// Item is an item to be got or stored in a memcached server.
|
||||||
|
type Item struct {
|
||||||
|
// Key is the Item's key (250 bytes maximum).
|
||||||
|
Key string
|
||||||
|
|
||||||
|
// Value is the Item's value.
|
||||||
|
Value []byte
|
||||||
|
|
||||||
|
// Flags are server-opaque flags whose semantics are entirely
|
||||||
|
// up to the app.
|
||||||
|
Flags uint32
|
||||||
|
|
||||||
|
// Expiration is the cache expiration time, in seconds: either a relative
|
||||||
|
// time from now (up to 1 month), or an absolute Unix epoch time.
|
||||||
|
// Zero means the Item has no expiration time.
|
||||||
|
Expiration int32
|
||||||
|
|
||||||
|
// Compare and swap ID.
|
||||||
|
casid uint64
|
||||||
|
}
|
||||||
|
|
||||||
|
// conn is a connection to a server.
|
||||||
|
type conn struct {
|
||||||
|
nc net.Conn
|
||||||
|
rw *bufio.ReadWriter
|
||||||
|
addr net.Addr
|
||||||
|
c *Client
|
||||||
|
}
|
||||||
|
|
||||||
|
// release returns this connection back to the client's free pool
|
||||||
|
func (cn *conn) release() {
|
||||||
|
cn.c.putFreeConn(cn.addr, cn)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cn *conn) extendDeadline() {
|
||||||
|
cn.nc.SetDeadline(time.Now().Add(cn.c.netTimeout()))
|
||||||
|
}
|
||||||
|
|
||||||
|
// condRelease releases this connection if the error pointed to by err
|
||||||
|
// is nil (not an error) or is only a protocol level error (e.g. a
|
||||||
|
// cache miss). The purpose is to not recycle TCP connections that
|
||||||
|
// are bad.
|
||||||
|
func (cn *conn) condRelease(err *error) {
|
||||||
|
if *err == nil || resumableError(*err) {
|
||||||
|
cn.release()
|
||||||
|
} else {
|
||||||
|
cn.nc.Close()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) putFreeConn(addr net.Addr, cn *conn) {
|
||||||
|
c.lk.Lock()
|
||||||
|
defer c.lk.Unlock()
|
||||||
|
if c.freeconn == nil {
|
||||||
|
c.freeconn = make(map[string][]*conn)
|
||||||
|
}
|
||||||
|
freelist := c.freeconn[addr.String()]
|
||||||
|
if len(freelist) >= c.maxIdleConns() {
|
||||||
|
cn.nc.Close()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.freeconn[addr.String()] = append(freelist, cn)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) getFreeConn(addr net.Addr) (cn *conn, ok bool) {
|
||||||
|
c.lk.Lock()
|
||||||
|
defer c.lk.Unlock()
|
||||||
|
if c.freeconn == nil {
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
freelist, ok := c.freeconn[addr.String()]
|
||||||
|
if !ok || len(freelist) == 0 {
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
cn = freelist[len(freelist)-1]
|
||||||
|
c.freeconn[addr.String()] = freelist[:len(freelist)-1]
|
||||||
|
return cn, true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) netTimeout() time.Duration {
|
||||||
|
if c.Timeout != 0 {
|
||||||
|
return c.Timeout
|
||||||
|
}
|
||||||
|
return DefaultTimeout
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) maxIdleConns() int {
|
||||||
|
if c.MaxIdleConns > 0 {
|
||||||
|
return c.MaxIdleConns
|
||||||
|
}
|
||||||
|
return DefaultMaxIdleConns
|
||||||
|
}
|
||||||
|
|
||||||
|
// ConnectTimeoutError is the error type used when it takes
|
||||||
|
// too long to connect to the desired host. This level of
|
||||||
|
// detail can generally be ignored.
|
||||||
|
type ConnectTimeoutError struct {
|
||||||
|
Addr net.Addr
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cte *ConnectTimeoutError) Error() string {
|
||||||
|
return "memcache: connect timeout to " + cte.Addr.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) dial(addr net.Addr) (net.Conn, error) {
|
||||||
|
type connError struct {
|
||||||
|
cn net.Conn
|
||||||
|
err error
|
||||||
|
}
|
||||||
|
|
||||||
|
nc, err := net.DialTimeout(addr.Network(), addr.String(), c.netTimeout())
|
||||||
|
if err == nil {
|
||||||
|
return nc, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if ne, ok := err.(net.Error); ok && ne.Timeout() {
|
||||||
|
return nil, &ConnectTimeoutError{addr}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) getConn(addr net.Addr) (*conn, error) {
|
||||||
|
cn, ok := c.getFreeConn(addr)
|
||||||
|
if ok {
|
||||||
|
cn.extendDeadline()
|
||||||
|
return cn, nil
|
||||||
|
}
|
||||||
|
nc, err := c.dial(addr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
cn = &conn{
|
||||||
|
nc: nc,
|
||||||
|
addr: addr,
|
||||||
|
rw: bufio.NewReadWriter(bufio.NewReader(nc), bufio.NewWriter(nc)),
|
||||||
|
c: c,
|
||||||
|
}
|
||||||
|
cn.extendDeadline()
|
||||||
|
return cn, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) onItem(item *Item, fn func(*Client, *bufio.ReadWriter, *Item) error) error {
|
||||||
|
addr, err := c.selector.PickServer(item.Key)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
cn, err := c.getConn(addr)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer cn.condRelease(&err)
|
||||||
|
if err = fn(c, cn.rw, item); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) FlushAll() error {
|
||||||
|
return c.selector.Each(c.flushAllFromAddr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get gets the item for the given key. ErrCacheMiss is returned for a
|
||||||
|
// memcache cache miss. The key must be at most 250 bytes in length.
|
||||||
|
func (c *Client) Get(key string) (item *Item, err error) {
|
||||||
|
err = c.withKeyAddr(key, func(addr net.Addr) error {
|
||||||
|
return c.getFromAddr(addr, []string{key}, func(it *Item) { item = it })
|
||||||
|
})
|
||||||
|
if err == nil && item == nil {
|
||||||
|
err = ErrCacheMiss
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Touch updates the expiry for the given key. The seconds parameter is either
|
||||||
|
// a Unix timestamp or, if seconds is less than 1 month, the number of seconds
|
||||||
|
// into the future at which time the item will expire. ErrCacheMiss is returned if the
|
||||||
|
// key is not in the cache. The key must be at most 250 bytes in length.
|
||||||
|
func (c *Client) Touch(key string, seconds int32) (err error) {
|
||||||
|
return c.withKeyAddr(key, func(addr net.Addr) error {
|
||||||
|
return c.touchFromAddr(addr, []string{key}, seconds)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) withKeyAddr(key string, fn func(net.Addr) error) (err error) {
|
||||||
|
if !legalKey(key) {
|
||||||
|
return ErrMalformedKey
|
||||||
|
}
|
||||||
|
addr, err := c.selector.PickServer(key)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return fn(addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) withAddrRw(addr net.Addr, fn func(*bufio.ReadWriter) error) (err error) {
|
||||||
|
cn, err := c.getConn(addr)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer cn.condRelease(&err)
|
||||||
|
return fn(cn.rw)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) withKeyRw(key string, fn func(*bufio.ReadWriter) error) error {
|
||||||
|
return c.withKeyAddr(key, func(addr net.Addr) error {
|
||||||
|
return c.withAddrRw(addr, fn)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) getFromAddr(addr net.Addr, keys []string, cb func(*Item)) error {
|
||||||
|
return c.withAddrRw(addr, func(rw *bufio.ReadWriter) error {
|
||||||
|
if _, err := fmt.Fprintf(rw, "gets %s\r\n", strings.Join(keys, " ")); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := rw.Flush(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := parseGetResponse(rw.Reader, cb); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// flushAllFromAddr send the flush_all command to the given addr
|
||||||
|
func (c *Client) flushAllFromAddr(addr net.Addr) error {
|
||||||
|
return c.withAddrRw(addr, func(rw *bufio.ReadWriter) error {
|
||||||
|
if _, err := fmt.Fprintf(rw, "flush_all\r\n"); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := rw.Flush(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
line, err := rw.ReadSlice('\n')
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case bytes.Equal(line, resultOk):
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("memcache: unexpected response line from flush_all: %q", string(line))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) touchFromAddr(addr net.Addr, keys []string, expiration int32) error {
|
||||||
|
return c.withAddrRw(addr, func(rw *bufio.ReadWriter) error {
|
||||||
|
for _, key := range keys {
|
||||||
|
if _, err := fmt.Fprintf(rw, "touch %s %d\r\n", key, expiration); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := rw.Flush(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
line, err := rw.ReadSlice('\n')
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case bytes.Equal(line, resultTouched):
|
||||||
|
break
|
||||||
|
case bytes.Equal(line, resultNotFound):
|
||||||
|
return ErrCacheMiss
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("memcache: unexpected response line from touch: %q", string(line))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetMulti is a batch version of Get. The returned map from keys to
|
||||||
|
// items may have fewer elements than the input slice, due to memcache
|
||||||
|
// cache misses. Each key must be at most 250 bytes in length.
|
||||||
|
// If no error is returned, the returned map will also be non-nil.
|
||||||
|
func (c *Client) GetMulti(keys []string) (map[string]*Item, error) {
|
||||||
|
var lk sync.Mutex
|
||||||
|
m := make(map[string]*Item)
|
||||||
|
addItemToMap := func(it *Item) {
|
||||||
|
lk.Lock()
|
||||||
|
defer lk.Unlock()
|
||||||
|
m[it.Key] = it
|
||||||
|
}
|
||||||
|
|
||||||
|
keyMap := make(map[net.Addr][]string)
|
||||||
|
for _, key := range keys {
|
||||||
|
if !legalKey(key) {
|
||||||
|
return nil, ErrMalformedKey
|
||||||
|
}
|
||||||
|
addr, err := c.selector.PickServer(key)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
keyMap[addr] = append(keyMap[addr], key)
|
||||||
|
}
|
||||||
|
|
||||||
|
ch := make(chan error, buffered)
|
||||||
|
for addr, keys := range keyMap {
|
||||||
|
go func(addr net.Addr, keys []string) {
|
||||||
|
ch <- c.getFromAddr(addr, keys, addItemToMap)
|
||||||
|
}(addr, keys)
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
for _ = range keyMap {
|
||||||
|
if ge := <-ch; ge != nil {
|
||||||
|
err = ge
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return m, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// parseGetResponse reads a GET response from r and calls cb for each
|
||||||
|
// read and allocated Item
|
||||||
|
func parseGetResponse(r *bufio.Reader, cb func(*Item)) error {
|
||||||
|
for {
|
||||||
|
line, err := r.ReadSlice('\n')
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if bytes.Equal(line, resultEnd) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
it := new(Item)
|
||||||
|
size, err := scanGetResponseLine(line, it)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
it.Value, err = ioutil.ReadAll(io.LimitReader(r, int64(size)+2))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if !bytes.HasSuffix(it.Value, crlf) {
|
||||||
|
return fmt.Errorf("memcache: corrupt get result read")
|
||||||
|
}
|
||||||
|
it.Value = it.Value[:size]
|
||||||
|
cb(it)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// scanGetResponseLine populates it and returns the declared size of the item.
|
||||||
|
// It does not read the bytes of the item.
|
||||||
|
func scanGetResponseLine(line []byte, it *Item) (size int, err error) {
|
||||||
|
pattern := "VALUE %s %d %d %d\r\n"
|
||||||
|
dest := []interface{}{&it.Key, &it.Flags, &size, &it.casid}
|
||||||
|
if bytes.Count(line, space) == 3 {
|
||||||
|
pattern = "VALUE %s %d %d\r\n"
|
||||||
|
dest = dest[:3]
|
||||||
|
}
|
||||||
|
n, err := fmt.Sscanf(string(line), pattern, dest...)
|
||||||
|
if err != nil || n != len(dest) {
|
||||||
|
return -1, fmt.Errorf("memcache: unexpected line in get response: %q", line)
|
||||||
|
}
|
||||||
|
return size, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set writes the given item, unconditionally.
|
||||||
|
func (c *Client) Set(item *Item) error {
|
||||||
|
return c.onItem(item, (*Client).set)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) set(rw *bufio.ReadWriter, item *Item) error {
|
||||||
|
return c.populateOne(rw, "set", item)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add writes the given item, if no value already exists for its
|
||||||
|
// key. ErrNotStored is returned if that condition is not met.
|
||||||
|
func (c *Client) Add(item *Item) error {
|
||||||
|
return c.onItem(item, (*Client).add)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) add(rw *bufio.ReadWriter, item *Item) error {
|
||||||
|
return c.populateOne(rw, "add", item)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Replace writes the given item, but only if the server *does*
|
||||||
|
// already hold data for this key
|
||||||
|
func (c *Client) Replace(item *Item) error {
|
||||||
|
return c.onItem(item, (*Client).replace)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) replace(rw *bufio.ReadWriter, item *Item) error {
|
||||||
|
return c.populateOne(rw, "replace", item)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CompareAndSwap writes the given item that was previously returned
|
||||||
|
// by Get, if the value was neither modified or evicted between the
|
||||||
|
// Get and the CompareAndSwap calls. The item's Key should not change
|
||||||
|
// between calls but all other item fields may differ. ErrCASConflict
|
||||||
|
// is returned if the value was modified in between the
|
||||||
|
// calls. ErrNotStored is returned if the value was evicted in between
|
||||||
|
// the calls.
|
||||||
|
func (c *Client) CompareAndSwap(item *Item) error {
|
||||||
|
return c.onItem(item, (*Client).cas)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) cas(rw *bufio.ReadWriter, item *Item) error {
|
||||||
|
return c.populateOne(rw, "cas", item)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) populateOne(rw *bufio.ReadWriter, verb string, item *Item) error {
|
||||||
|
if !legalKey(item.Key) {
|
||||||
|
return ErrMalformedKey
|
||||||
|
}
|
||||||
|
var err error
|
||||||
|
if verb == "cas" {
|
||||||
|
_, err = fmt.Fprintf(rw, "%s %s %d %d %d %d\r\n",
|
||||||
|
verb, item.Key, item.Flags, item.Expiration, len(item.Value), item.casid)
|
||||||
|
} else {
|
||||||
|
_, err = fmt.Fprintf(rw, "%s %s %d %d %d\r\n",
|
||||||
|
verb, item.Key, item.Flags, item.Expiration, len(item.Value))
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if _, err = rw.Write(item.Value); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if _, err := rw.Write(crlf); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := rw.Flush(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
line, err := rw.ReadSlice('\n')
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case bytes.Equal(line, resultStored):
|
||||||
|
return nil
|
||||||
|
case bytes.Equal(line, resultNotStored):
|
||||||
|
return ErrNotStored
|
||||||
|
case bytes.Equal(line, resultExists):
|
||||||
|
return ErrCASConflict
|
||||||
|
case bytes.Equal(line, resultNotFound):
|
||||||
|
return ErrCacheMiss
|
||||||
|
}
|
||||||
|
return fmt.Errorf("memcache: unexpected response line from %q: %q", verb, string(line))
|
||||||
|
}
|
||||||
|
|
||||||
|
func writeReadLine(rw *bufio.ReadWriter, format string, args ...interface{}) ([]byte, error) {
|
||||||
|
_, err := fmt.Fprintf(rw, format, args...)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := rw.Flush(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
line, err := rw.ReadSlice('\n')
|
||||||
|
return line, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func writeExpectf(rw *bufio.ReadWriter, expect []byte, format string, args ...interface{}) error {
|
||||||
|
line, err := writeReadLine(rw, format, args...)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case bytes.Equal(line, resultOK):
|
||||||
|
return nil
|
||||||
|
case bytes.Equal(line, expect):
|
||||||
|
return nil
|
||||||
|
case bytes.Equal(line, resultNotStored):
|
||||||
|
return ErrNotStored
|
||||||
|
case bytes.Equal(line, resultExists):
|
||||||
|
return ErrCASConflict
|
||||||
|
case bytes.Equal(line, resultNotFound):
|
||||||
|
return ErrCacheMiss
|
||||||
|
}
|
||||||
|
return fmt.Errorf("memcache: unexpected response line: %q", string(line))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Delete deletes the item with the provided key. The error ErrCacheMiss is
|
||||||
|
// returned if the item didn't already exist in the cache.
|
||||||
|
func (c *Client) Delete(key string) error {
|
||||||
|
return c.withKeyRw(key, func(rw *bufio.ReadWriter) error {
|
||||||
|
return writeExpectf(rw, resultDeleted, "delete %s\r\n", key)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteAll deletes all items in the cache.
|
||||||
|
func (c *Client) DeleteAll() error {
|
||||||
|
return c.withKeyRw("", func(rw *bufio.ReadWriter) error {
|
||||||
|
return writeExpectf(rw, resultDeleted, "flush_all\r\n")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Increment atomically increments key by delta. The return value is
|
||||||
|
// the new value after being incremented or an error. If the value
|
||||||
|
// didn't exist in memcached the error is ErrCacheMiss. The value in
|
||||||
|
// memcached must be an decimal number, or an error will be returned.
|
||||||
|
// On 64-bit overflow, the new value wraps around.
|
||||||
|
func (c *Client) Increment(key string, delta uint64) (newValue uint64, err error) {
|
||||||
|
return c.incrDecr("incr", key, delta)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decrement atomically decrements key by delta. The return value is
|
||||||
|
// the new value after being decremented or an error. If the value
|
||||||
|
// didn't exist in memcached the error is ErrCacheMiss. The value in
|
||||||
|
// memcached must be an decimal number, or an error will be returned.
|
||||||
|
// On underflow, the new value is capped at zero and does not wrap
|
||||||
|
// around.
|
||||||
|
func (c *Client) Decrement(key string, delta uint64) (newValue uint64, err error) {
|
||||||
|
return c.incrDecr("decr", key, delta)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Client) incrDecr(verb, key string, delta uint64) (uint64, error) {
|
||||||
|
var val uint64
|
||||||
|
err := c.withKeyRw(key, func(rw *bufio.ReadWriter) error {
|
||||||
|
line, err := writeReadLine(rw, "%s %s %d\r\n", verb, key, delta)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case bytes.Equal(line, resultNotFound):
|
||||||
|
return ErrCacheMiss
|
||||||
|
case bytes.HasPrefix(line, resultClientErrorPrefix):
|
||||||
|
errMsg := line[len(resultClientErrorPrefix) : len(line)-2]
|
||||||
|
return errors.New("memcache: client error: " + string(errMsg))
|
||||||
|
}
|
||||||
|
val, err = strconv.ParseUint(string(line[:len(line)-2]), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
return val, err
|
||||||
|
}
|
|
@ -0,0 +1,129 @@
|
||||||
|
/*
|
||||||
|
Copyright 2011 Google Inc.
|
||||||
|
|
||||||
|
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.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package memcache
|
||||||
|
|
||||||
|
import (
|
||||||
|
"hash/crc32"
|
||||||
|
"net"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ServerSelector is the interface that selects a memcache server
|
||||||
|
// as a function of the item's key.
|
||||||
|
//
|
||||||
|
// All ServerSelector implementations must be safe for concurrent use
|
||||||
|
// by multiple goroutines.
|
||||||
|
type ServerSelector interface {
|
||||||
|
// PickServer returns the server address that a given item
|
||||||
|
// should be shared onto.
|
||||||
|
PickServer(key string) (net.Addr, error)
|
||||||
|
Each(func(net.Addr) error) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// ServerList is a simple ServerSelector. Its zero value is usable.
|
||||||
|
type ServerList struct {
|
||||||
|
mu sync.RWMutex
|
||||||
|
addrs []net.Addr
|
||||||
|
}
|
||||||
|
|
||||||
|
// staticAddr caches the Network() and String() values from any net.Addr.
|
||||||
|
type staticAddr struct {
|
||||||
|
ntw, str string
|
||||||
|
}
|
||||||
|
|
||||||
|
func newStaticAddr(a net.Addr) net.Addr {
|
||||||
|
return &staticAddr{
|
||||||
|
ntw: a.Network(),
|
||||||
|
str: a.String(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *staticAddr) Network() string { return s.ntw }
|
||||||
|
func (s *staticAddr) String() string { return s.str }
|
||||||
|
|
||||||
|
// SetServers changes a ServerList's set of servers at runtime and is
|
||||||
|
// safe for concurrent use by multiple goroutines.
|
||||||
|
//
|
||||||
|
// Each server is given equal weight. A server is given more weight
|
||||||
|
// if it's listed multiple times.
|
||||||
|
//
|
||||||
|
// SetServers returns an error if any of the server names fail to
|
||||||
|
// resolve. No attempt is made to connect to the server. If any error
|
||||||
|
// is returned, no changes are made to the ServerList.
|
||||||
|
func (ss *ServerList) SetServers(servers ...string) error {
|
||||||
|
naddr := make([]net.Addr, len(servers))
|
||||||
|
for i, server := range servers {
|
||||||
|
if strings.Contains(server, "/") {
|
||||||
|
addr, err := net.ResolveUnixAddr("unix", server)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
naddr[i] = newStaticAddr(addr)
|
||||||
|
} else {
|
||||||
|
tcpaddr, err := net.ResolveTCPAddr("tcp", server)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
naddr[i] = newStaticAddr(tcpaddr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ss.mu.Lock()
|
||||||
|
defer ss.mu.Unlock()
|
||||||
|
ss.addrs = naddr
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Each iterates over each server calling the given function
|
||||||
|
func (ss *ServerList) Each(f func(net.Addr) error) error {
|
||||||
|
ss.mu.RLock()
|
||||||
|
defer ss.mu.RUnlock()
|
||||||
|
for _, a := range ss.addrs {
|
||||||
|
if err := f(a); nil != err {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// keyBufPool returns []byte buffers for use by PickServer's call to
|
||||||
|
// crc32.ChecksumIEEE to avoid allocations. (but doesn't avoid the
|
||||||
|
// copies, which at least are bounded in size and small)
|
||||||
|
var keyBufPool = sync.Pool{
|
||||||
|
New: func() interface{} {
|
||||||
|
b := make([]byte, 256)
|
||||||
|
return &b
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ss *ServerList) PickServer(key string) (net.Addr, error) {
|
||||||
|
ss.mu.RLock()
|
||||||
|
defer ss.mu.RUnlock()
|
||||||
|
if len(ss.addrs) == 0 {
|
||||||
|
return nil, ErrNoServers
|
||||||
|
}
|
||||||
|
if len(ss.addrs) == 1 {
|
||||||
|
return ss.addrs[0], nil
|
||||||
|
}
|
||||||
|
bufp := keyBufPool.Get().(*[]byte)
|
||||||
|
n := copy(*bufp, key)
|
||||||
|
cs := crc32.ChecksumIEEE((*bufp)[:n])
|
||||||
|
keyBufPool.Put(bufp)
|
||||||
|
|
||||||
|
return ss.addrs[cs%uint32(len(ss.addrs))], nil
|
||||||
|
}
|
|
@ -0,0 +1,175 @@
|
||||||
|
|
||||||
|
Apache License
|
||||||
|
Version 2.0, January 2004
|
||||||
|
http://www.apache.org/licenses/
|
||||||
|
|
||||||
|
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||||
|
|
||||||
|
1. Definitions.
|
||||||
|
|
||||||
|
"License" shall mean the terms and conditions for use, reproduction,
|
||||||
|
and distribution as defined by Sections 1 through 9 of this document.
|
||||||
|
|
||||||
|
"Licensor" shall mean the copyright owner or entity authorized by
|
||||||
|
the copyright owner that is granting the License.
|
||||||
|
|
||||||
|
"Legal Entity" shall mean the union of the acting entity and all
|
||||||
|
other entities that control, are controlled by, or are under common
|
||||||
|
control with that entity. For the purposes of this definition,
|
||||||
|
"control" means (i) the power, direct or indirect, to cause the
|
||||||
|
direction or management of such entity, whether by contract or
|
||||||
|
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||||
|
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||||
|
|
||||||
|
"You" (or "Your") shall mean an individual or Legal Entity
|
||||||
|
exercising permissions granted by this License.
|
||||||
|
|
||||||
|
"Source" form shall mean the preferred form for making modifications,
|
||||||
|
including but not limited to software source code, documentation
|
||||||
|
source, and configuration files.
|
||||||
|
|
||||||
|
"Object" form shall mean any form resulting from mechanical
|
||||||
|
transformation or translation of a Source form, including but
|
||||||
|
not limited to compiled object code, generated documentation,
|
||||||
|
and conversions to other media types.
|
||||||
|
|
||||||
|
"Work" shall mean the work of authorship, whether in Source or
|
||||||
|
Object form, made available under the License, as indicated by a
|
||||||
|
copyright notice that is included in or attached to the work
|
||||||
|
(an example is provided in the Appendix below).
|
||||||
|
|
||||||
|
"Derivative Works" shall mean any work, whether in Source or Object
|
||||||
|
form, that is based on (or derived from) the Work and for which the
|
||||||
|
editorial revisions, annotations, elaborations, or other modifications
|
||||||
|
represent, as a whole, an original work of authorship. For the purposes
|
||||||
|
of this License, Derivative Works shall not include works that remain
|
||||||
|
separable from, or merely link (or bind by name) to the interfaces of,
|
||||||
|
the Work and Derivative Works thereof.
|
||||||
|
|
||||||
|
"Contribution" shall mean any work of authorship, including
|
||||||
|
the original version of the Work and any modifications or additions
|
||||||
|
to that Work or Derivative Works thereof, that is intentionally
|
||||||
|
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||||
|
or by an individual or Legal Entity authorized to submit on behalf of
|
||||||
|
the copyright owner. For the purposes of this definition, "submitted"
|
||||||
|
means any form of electronic, verbal, or written communication sent
|
||||||
|
to the Licensor or its representatives, including but not limited to
|
||||||
|
communication on electronic mailing lists, source code control systems,
|
||||||
|
and issue tracking systems that are managed by, or on behalf of, the
|
||||||
|
Licensor for the purpose of discussing and improving the Work, but
|
||||||
|
excluding communication that is conspicuously marked or otherwise
|
||||||
|
designated in writing by the copyright owner as "Not a Contribution."
|
||||||
|
|
||||||
|
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||||
|
on behalf of whom a Contribution has been received by Licensor and
|
||||||
|
subsequently incorporated within the Work.
|
||||||
|
|
||||||
|
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
copyright license to reproduce, prepare Derivative Works of,
|
||||||
|
publicly display, publicly perform, sublicense, and distribute the
|
||||||
|
Work and such Derivative Works in Source or Object form.
|
||||||
|
|
||||||
|
3. Grant of Patent License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
(except as stated in this section) patent license to make, have made,
|
||||||
|
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||||
|
where such license applies only to those patent claims licensable
|
||||||
|
by such Contributor that are necessarily infringed by their
|
||||||
|
Contribution(s) alone or by combination of their Contribution(s)
|
||||||
|
with the Work to which such Contribution(s) was submitted. If You
|
||||||
|
institute patent litigation against any entity (including a
|
||||||
|
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||||
|
or a Contribution incorporated within the Work constitutes direct
|
||||||
|
or contributory patent infringement, then any patent licenses
|
||||||
|
granted to You under this License for that Work shall terminate
|
||||||
|
as of the date such litigation is filed.
|
||||||
|
|
||||||
|
4. Redistribution. You may reproduce and distribute copies of the
|
||||||
|
Work or Derivative Works thereof in any medium, with or without
|
||||||
|
modifications, and in Source or Object form, provided that You
|
||||||
|
meet the following conditions:
|
||||||
|
|
||||||
|
(a) You must give any other recipients of the Work or
|
||||||
|
Derivative Works a copy of this License; and
|
||||||
|
|
||||||
|
(b) You must cause any modified files to carry prominent notices
|
||||||
|
stating that You changed the files; and
|
||||||
|
|
||||||
|
(c) You must retain, in the Source form of any Derivative Works
|
||||||
|
that You distribute, all copyright, patent, trademark, and
|
||||||
|
attribution notices from the Source form of the Work,
|
||||||
|
excluding those notices that do not pertain to any part of
|
||||||
|
the Derivative Works; and
|
||||||
|
|
||||||
|
(d) If the Work includes a "NOTICE" text file as part of its
|
||||||
|
distribution, then any Derivative Works that You distribute must
|
||||||
|
include a readable copy of the attribution notices contained
|
||||||
|
within such NOTICE file, excluding those notices that do not
|
||||||
|
pertain to any part of the Derivative Works, in at least one
|
||||||
|
of the following places: within a NOTICE text file distributed
|
||||||
|
as part of the Derivative Works; within the Source form or
|
||||||
|
documentation, if provided along with the Derivative Works; or,
|
||||||
|
within a display generated by the Derivative Works, if and
|
||||||
|
wherever such third-party notices normally appear. The contents
|
||||||
|
of the NOTICE file are for informational purposes only and
|
||||||
|
do not modify the License. You may add Your own attribution
|
||||||
|
notices within Derivative Works that You distribute, alongside
|
||||||
|
or as an addendum to the NOTICE text from the Work, provided
|
||||||
|
that such additional attribution notices cannot be construed
|
||||||
|
as modifying the License.
|
||||||
|
|
||||||
|
You may add Your own copyright statement to Your modifications and
|
||||||
|
may provide additional or different license terms and conditions
|
||||||
|
for use, reproduction, or distribution of Your modifications, or
|
||||||
|
for any such Derivative Works as a whole, provided Your use,
|
||||||
|
reproduction, and distribution of the Work otherwise complies with
|
||||||
|
the conditions stated in this License.
|
||||||
|
|
||||||
|
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||||
|
any Contribution intentionally submitted for inclusion in the Work
|
||||||
|
by You to the Licensor shall be under the terms and conditions of
|
||||||
|
this License, without any additional terms or conditions.
|
||||||
|
Notwithstanding the above, nothing herein shall supersede or modify
|
||||||
|
the terms of any separate license agreement you may have executed
|
||||||
|
with Licensor regarding such Contributions.
|
||||||
|
|
||||||
|
6. Trademarks. This License does not grant permission to use the trade
|
||||||
|
names, trademarks, service marks, or product names of the Licensor,
|
||||||
|
except as required for reasonable and customary use in describing the
|
||||||
|
origin of the Work and reproducing the content of the NOTICE file.
|
||||||
|
|
||||||
|
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||||
|
agreed to in writing, Licensor provides the Work (and each
|
||||||
|
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||||
|
implied, including, without limitation, any warranties or conditions
|
||||||
|
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||||
|
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||||
|
appropriateness of using or redistributing the Work and assume any
|
||||||
|
risks associated with Your exercise of permissions under this License.
|
||||||
|
|
||||||
|
8. Limitation of Liability. In no event and under no legal theory,
|
||||||
|
whether in tort (including negligence), contract, or otherwise,
|
||||||
|
unless required by applicable law (such as deliberate and grossly
|
||||||
|
negligent acts) or agreed to in writing, shall any Contributor be
|
||||||
|
liable to You for damages, including any direct, indirect, special,
|
||||||
|
incidental, or consequential damages of any character arising as a
|
||||||
|
result of this License or out of the use or inability to use the
|
||||||
|
Work (including but not limited to damages for loss of goodwill,
|
||||||
|
work stoppage, computer failure or malfunction, or any and all
|
||||||
|
other commercial damages or losses), even if such Contributor
|
||||||
|
has been advised of the possibility of such damages.
|
||||||
|
|
||||||
|
9. Accepting Warranty or Additional Liability. While redistributing
|
||||||
|
the Work or Derivative Works thereof, You may choose to offer,
|
||||||
|
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||||
|
or other liability obligations and/or rights consistent with this
|
||||||
|
License. However, in accepting such obligations, You may act only
|
||||||
|
on Your own behalf and on Your sole responsibility, not on behalf
|
||||||
|
of any other Contributor, and only if You agree to indemnify,
|
||||||
|
defend, and hold each Contributor harmless for any liability
|
||||||
|
incurred by, or claims asserted against, such Contributor by reason
|
||||||
|
of your accepting any such warranty or additional liability.
|
|
@ -0,0 +1,54 @@
|
||||||
|
// Copyright 2014 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package internal
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
WatchState = 1 << iota
|
||||||
|
MultiState
|
||||||
|
SubscribeState
|
||||||
|
MonitorState
|
||||||
|
)
|
||||||
|
|
||||||
|
type CommandInfo struct {
|
||||||
|
Set, Clear int
|
||||||
|
}
|
||||||
|
|
||||||
|
var commandInfos = map[string]CommandInfo{
|
||||||
|
"WATCH": {Set: WatchState},
|
||||||
|
"UNWATCH": {Clear: WatchState},
|
||||||
|
"MULTI": {Set: MultiState},
|
||||||
|
"EXEC": {Clear: WatchState | MultiState},
|
||||||
|
"DISCARD": {Clear: WatchState | MultiState},
|
||||||
|
"PSUBSCRIBE": {Set: SubscribeState},
|
||||||
|
"SUBSCRIBE": {Set: SubscribeState},
|
||||||
|
"MONITOR": {Set: MonitorState},
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
for n, ci := range commandInfos {
|
||||||
|
commandInfos[strings.ToLower(n)] = ci
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func LookupCommandInfo(commandName string) CommandInfo {
|
||||||
|
if ci, ok := commandInfos[commandName]; ok {
|
||||||
|
return ci
|
||||||
|
}
|
||||||
|
return commandInfos[strings.ToUpper(commandName)]
|
||||||
|
}
|
|
@ -0,0 +1,618 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"bytes"
|
||||||
|
"crypto/tls"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net"
|
||||||
|
"net/url"
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// conn is the low-level implementation of Conn
|
||||||
|
type conn struct {
|
||||||
|
|
||||||
|
// Shared
|
||||||
|
mu sync.Mutex
|
||||||
|
pending int
|
||||||
|
err error
|
||||||
|
conn net.Conn
|
||||||
|
|
||||||
|
// Read
|
||||||
|
readTimeout time.Duration
|
||||||
|
br *bufio.Reader
|
||||||
|
|
||||||
|
// Write
|
||||||
|
writeTimeout time.Duration
|
||||||
|
bw *bufio.Writer
|
||||||
|
|
||||||
|
// Scratch space for formatting argument length.
|
||||||
|
// '*' or '$', length, "\r\n"
|
||||||
|
lenScratch [32]byte
|
||||||
|
|
||||||
|
// Scratch space for formatting integers and floats.
|
||||||
|
numScratch [40]byte
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialTimeout acts like Dial but takes timeouts for establishing the
|
||||||
|
// connection to the server, writing a command and reading a reply.
|
||||||
|
//
|
||||||
|
// Deprecated: Use Dial with options instead.
|
||||||
|
func DialTimeout(network, address string, connectTimeout, readTimeout, writeTimeout time.Duration) (Conn, error) {
|
||||||
|
return Dial(network, address,
|
||||||
|
DialConnectTimeout(connectTimeout),
|
||||||
|
DialReadTimeout(readTimeout),
|
||||||
|
DialWriteTimeout(writeTimeout))
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialOption specifies an option for dialing a Redis server.
|
||||||
|
type DialOption struct {
|
||||||
|
f func(*dialOptions)
|
||||||
|
}
|
||||||
|
|
||||||
|
type dialOptions struct {
|
||||||
|
readTimeout time.Duration
|
||||||
|
writeTimeout time.Duration
|
||||||
|
dial func(network, addr string) (net.Conn, error)
|
||||||
|
db int
|
||||||
|
password string
|
||||||
|
dialTLS bool
|
||||||
|
skipVerify bool
|
||||||
|
tlsConfig *tls.Config
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialReadTimeout specifies the timeout for reading a single command reply.
|
||||||
|
func DialReadTimeout(d time.Duration) DialOption {
|
||||||
|
return DialOption{func(do *dialOptions) {
|
||||||
|
do.readTimeout = d
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialWriteTimeout specifies the timeout for writing a single command.
|
||||||
|
func DialWriteTimeout(d time.Duration) DialOption {
|
||||||
|
return DialOption{func(do *dialOptions) {
|
||||||
|
do.writeTimeout = d
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialConnectTimeout specifies the timeout for connecting to the Redis server.
|
||||||
|
func DialConnectTimeout(d time.Duration) DialOption {
|
||||||
|
return DialOption{func(do *dialOptions) {
|
||||||
|
dialer := net.Dialer{Timeout: d}
|
||||||
|
do.dial = dialer.Dial
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialNetDial specifies a custom dial function for creating TCP
|
||||||
|
// connections. If this option is left out, then net.Dial is
|
||||||
|
// used. DialNetDial overrides DialConnectTimeout.
|
||||||
|
func DialNetDial(dial func(network, addr string) (net.Conn, error)) DialOption {
|
||||||
|
return DialOption{func(do *dialOptions) {
|
||||||
|
do.dial = dial
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialDatabase specifies the database to select when dialing a connection.
|
||||||
|
func DialDatabase(db int) DialOption {
|
||||||
|
return DialOption{func(do *dialOptions) {
|
||||||
|
do.db = db
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialPassword specifies the password to use when connecting to
|
||||||
|
// the Redis server.
|
||||||
|
func DialPassword(password string) DialOption {
|
||||||
|
return DialOption{func(do *dialOptions) {
|
||||||
|
do.password = password
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialTLSConfig specifies the config to use when a TLS connection is dialed.
|
||||||
|
// Has no effect when not dialing a TLS connection.
|
||||||
|
func DialTLSConfig(c *tls.Config) DialOption {
|
||||||
|
return DialOption{func(do *dialOptions) {
|
||||||
|
do.tlsConfig = c
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// DialTLSSkipVerify to disable server name verification when connecting
|
||||||
|
// over TLS. Has no effect when not dialing a TLS connection.
|
||||||
|
func DialTLSSkipVerify(skip bool) DialOption {
|
||||||
|
return DialOption{func(do *dialOptions) {
|
||||||
|
do.skipVerify = skip
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Dial connects to the Redis server at the given network and
|
||||||
|
// address using the specified options.
|
||||||
|
func Dial(network, address string, options ...DialOption) (Conn, error) {
|
||||||
|
do := dialOptions{
|
||||||
|
dial: net.Dial,
|
||||||
|
}
|
||||||
|
for _, option := range options {
|
||||||
|
option.f(&do)
|
||||||
|
}
|
||||||
|
|
||||||
|
netConn, err := do.dial(network, address)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if do.dialTLS {
|
||||||
|
tlsConfig := cloneTLSClientConfig(do.tlsConfig, do.skipVerify)
|
||||||
|
if tlsConfig.ServerName == "" {
|
||||||
|
host, _, err := net.SplitHostPort(address)
|
||||||
|
if err != nil {
|
||||||
|
netConn.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
tlsConfig.ServerName = host
|
||||||
|
}
|
||||||
|
|
||||||
|
tlsConn := tls.Client(netConn, tlsConfig)
|
||||||
|
if err := tlsConn.Handshake(); err != nil {
|
||||||
|
netConn.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
netConn = tlsConn
|
||||||
|
}
|
||||||
|
|
||||||
|
c := &conn{
|
||||||
|
conn: netConn,
|
||||||
|
bw: bufio.NewWriter(netConn),
|
||||||
|
br: bufio.NewReader(netConn),
|
||||||
|
readTimeout: do.readTimeout,
|
||||||
|
writeTimeout: do.writeTimeout,
|
||||||
|
}
|
||||||
|
|
||||||
|
if do.password != "" {
|
||||||
|
if _, err := c.Do("AUTH", do.password); err != nil {
|
||||||
|
netConn.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if do.db != 0 {
|
||||||
|
if _, err := c.Do("SELECT", do.db); err != nil {
|
||||||
|
netConn.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return c, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func dialTLS(do *dialOptions) {
|
||||||
|
do.dialTLS = true
|
||||||
|
}
|
||||||
|
|
||||||
|
var pathDBRegexp = regexp.MustCompile(`/(\d*)\z`)
|
||||||
|
|
||||||
|
// DialURL connects to a Redis server at the given URL using the Redis
|
||||||
|
// URI scheme. URLs should follow the draft IANA specification for the
|
||||||
|
// scheme (https://www.iana.org/assignments/uri-schemes/prov/redis).
|
||||||
|
func DialURL(rawurl string, options ...DialOption) (Conn, error) {
|
||||||
|
u, err := url.Parse(rawurl)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if u.Scheme != "redis" && u.Scheme != "rediss" {
|
||||||
|
return nil, fmt.Errorf("invalid redis URL scheme: %s", u.Scheme)
|
||||||
|
}
|
||||||
|
|
||||||
|
// As per the IANA draft spec, the host defaults to localhost and
|
||||||
|
// the port defaults to 6379.
|
||||||
|
host, port, err := net.SplitHostPort(u.Host)
|
||||||
|
if err != nil {
|
||||||
|
// assume port is missing
|
||||||
|
host = u.Host
|
||||||
|
port = "6379"
|
||||||
|
}
|
||||||
|
if host == "" {
|
||||||
|
host = "localhost"
|
||||||
|
}
|
||||||
|
address := net.JoinHostPort(host, port)
|
||||||
|
|
||||||
|
if u.User != nil {
|
||||||
|
password, isSet := u.User.Password()
|
||||||
|
if isSet {
|
||||||
|
options = append(options, DialPassword(password))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
match := pathDBRegexp.FindStringSubmatch(u.Path)
|
||||||
|
if len(match) == 2 {
|
||||||
|
db := 0
|
||||||
|
if len(match[1]) > 0 {
|
||||||
|
db, err = strconv.Atoi(match[1])
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("invalid database: %s", u.Path[1:])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if db != 0 {
|
||||||
|
options = append(options, DialDatabase(db))
|
||||||
|
}
|
||||||
|
} else if u.Path != "" {
|
||||||
|
return nil, fmt.Errorf("invalid database: %s", u.Path[1:])
|
||||||
|
}
|
||||||
|
|
||||||
|
if u.Scheme == "rediss" {
|
||||||
|
options = append([]DialOption{{dialTLS}}, options...)
|
||||||
|
}
|
||||||
|
|
||||||
|
return Dial("tcp", address, options...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewConn returns a new Redigo connection for the given net connection.
|
||||||
|
func NewConn(netConn net.Conn, readTimeout, writeTimeout time.Duration) Conn {
|
||||||
|
return &conn{
|
||||||
|
conn: netConn,
|
||||||
|
bw: bufio.NewWriter(netConn),
|
||||||
|
br: bufio.NewReader(netConn),
|
||||||
|
readTimeout: readTimeout,
|
||||||
|
writeTimeout: writeTimeout,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) Close() error {
|
||||||
|
c.mu.Lock()
|
||||||
|
err := c.err
|
||||||
|
if c.err == nil {
|
||||||
|
c.err = errors.New("redigo: closed")
|
||||||
|
err = c.conn.Close()
|
||||||
|
}
|
||||||
|
c.mu.Unlock()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) fatal(err error) error {
|
||||||
|
c.mu.Lock()
|
||||||
|
if c.err == nil {
|
||||||
|
c.err = err
|
||||||
|
// Close connection to force errors on subsequent calls and to unblock
|
||||||
|
// other reader or writer.
|
||||||
|
c.conn.Close()
|
||||||
|
}
|
||||||
|
c.mu.Unlock()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) Err() error {
|
||||||
|
c.mu.Lock()
|
||||||
|
err := c.err
|
||||||
|
c.mu.Unlock()
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) writeLen(prefix byte, n int) error {
|
||||||
|
c.lenScratch[len(c.lenScratch)-1] = '\n'
|
||||||
|
c.lenScratch[len(c.lenScratch)-2] = '\r'
|
||||||
|
i := len(c.lenScratch) - 3
|
||||||
|
for {
|
||||||
|
c.lenScratch[i] = byte('0' + n%10)
|
||||||
|
i -= 1
|
||||||
|
n = n / 10
|
||||||
|
if n == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
c.lenScratch[i] = prefix
|
||||||
|
_, err := c.bw.Write(c.lenScratch[i:])
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) writeString(s string) error {
|
||||||
|
c.writeLen('$', len(s))
|
||||||
|
c.bw.WriteString(s)
|
||||||
|
_, err := c.bw.WriteString("\r\n")
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) writeBytes(p []byte) error {
|
||||||
|
c.writeLen('$', len(p))
|
||||||
|
c.bw.Write(p)
|
||||||
|
_, err := c.bw.WriteString("\r\n")
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) writeInt64(n int64) error {
|
||||||
|
return c.writeBytes(strconv.AppendInt(c.numScratch[:0], n, 10))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) writeFloat64(n float64) error {
|
||||||
|
return c.writeBytes(strconv.AppendFloat(c.numScratch[:0], n, 'g', -1, 64))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) writeCommand(cmd string, args []interface{}) (err error) {
|
||||||
|
c.writeLen('*', 1+len(args))
|
||||||
|
err = c.writeString(cmd)
|
||||||
|
for _, arg := range args {
|
||||||
|
if err != nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
switch arg := arg.(type) {
|
||||||
|
case string:
|
||||||
|
err = c.writeString(arg)
|
||||||
|
case []byte:
|
||||||
|
err = c.writeBytes(arg)
|
||||||
|
case int:
|
||||||
|
err = c.writeInt64(int64(arg))
|
||||||
|
case int64:
|
||||||
|
err = c.writeInt64(arg)
|
||||||
|
case float64:
|
||||||
|
err = c.writeFloat64(arg)
|
||||||
|
case bool:
|
||||||
|
if arg {
|
||||||
|
err = c.writeString("1")
|
||||||
|
} else {
|
||||||
|
err = c.writeString("0")
|
||||||
|
}
|
||||||
|
case nil:
|
||||||
|
err = c.writeString("")
|
||||||
|
default:
|
||||||
|
var buf bytes.Buffer
|
||||||
|
fmt.Fprint(&buf, arg)
|
||||||
|
err = c.writeBytes(buf.Bytes())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
type protocolError string
|
||||||
|
|
||||||
|
func (pe protocolError) Error() string {
|
||||||
|
return fmt.Sprintf("redigo: %s (possible server error or unsupported concurrent read by application)", string(pe))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) readLine() ([]byte, error) {
|
||||||
|
p, err := c.br.ReadSlice('\n')
|
||||||
|
if err == bufio.ErrBufferFull {
|
||||||
|
return nil, protocolError("long response line")
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
i := len(p) - 2
|
||||||
|
if i < 0 || p[i] != '\r' {
|
||||||
|
return nil, protocolError("bad response line terminator")
|
||||||
|
}
|
||||||
|
return p[:i], nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// parseLen parses bulk string and array lengths.
|
||||||
|
func parseLen(p []byte) (int, error) {
|
||||||
|
if len(p) == 0 {
|
||||||
|
return -1, protocolError("malformed length")
|
||||||
|
}
|
||||||
|
|
||||||
|
if p[0] == '-' && len(p) == 2 && p[1] == '1' {
|
||||||
|
// handle $-1 and $-1 null replies.
|
||||||
|
return -1, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var n int
|
||||||
|
for _, b := range p {
|
||||||
|
n *= 10
|
||||||
|
if b < '0' || b > '9' {
|
||||||
|
return -1, protocolError("illegal bytes in length")
|
||||||
|
}
|
||||||
|
n += int(b - '0')
|
||||||
|
}
|
||||||
|
|
||||||
|
return n, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// parseInt parses an integer reply.
|
||||||
|
func parseInt(p []byte) (interface{}, error) {
|
||||||
|
if len(p) == 0 {
|
||||||
|
return 0, protocolError("malformed integer")
|
||||||
|
}
|
||||||
|
|
||||||
|
var negate bool
|
||||||
|
if p[0] == '-' {
|
||||||
|
negate = true
|
||||||
|
p = p[1:]
|
||||||
|
if len(p) == 0 {
|
||||||
|
return 0, protocolError("malformed integer")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var n int64
|
||||||
|
for _, b := range p {
|
||||||
|
n *= 10
|
||||||
|
if b < '0' || b > '9' {
|
||||||
|
return 0, protocolError("illegal bytes in length")
|
||||||
|
}
|
||||||
|
n += int64(b - '0')
|
||||||
|
}
|
||||||
|
|
||||||
|
if negate {
|
||||||
|
n = -n
|
||||||
|
}
|
||||||
|
return n, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
okReply interface{} = "OK"
|
||||||
|
pongReply interface{} = "PONG"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c *conn) readReply() (interface{}, error) {
|
||||||
|
line, err := c.readLine()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if len(line) == 0 {
|
||||||
|
return nil, protocolError("short response line")
|
||||||
|
}
|
||||||
|
switch line[0] {
|
||||||
|
case '+':
|
||||||
|
switch {
|
||||||
|
case len(line) == 3 && line[1] == 'O' && line[2] == 'K':
|
||||||
|
// Avoid allocation for frequent "+OK" response.
|
||||||
|
return okReply, nil
|
||||||
|
case len(line) == 5 && line[1] == 'P' && line[2] == 'O' && line[3] == 'N' && line[4] == 'G':
|
||||||
|
// Avoid allocation in PING command benchmarks :)
|
||||||
|
return pongReply, nil
|
||||||
|
default:
|
||||||
|
return string(line[1:]), nil
|
||||||
|
}
|
||||||
|
case '-':
|
||||||
|
return Error(string(line[1:])), nil
|
||||||
|
case ':':
|
||||||
|
return parseInt(line[1:])
|
||||||
|
case '$':
|
||||||
|
n, err := parseLen(line[1:])
|
||||||
|
if n < 0 || err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
p := make([]byte, n)
|
||||||
|
_, err = io.ReadFull(c.br, p)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if line, err := c.readLine(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
} else if len(line) != 0 {
|
||||||
|
return nil, protocolError("bad bulk string format")
|
||||||
|
}
|
||||||
|
return p, nil
|
||||||
|
case '*':
|
||||||
|
n, err := parseLen(line[1:])
|
||||||
|
if n < 0 || err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
r := make([]interface{}, n)
|
||||||
|
for i := range r {
|
||||||
|
r[i], err = c.readReply()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return r, nil
|
||||||
|
}
|
||||||
|
return nil, protocolError("unexpected response line")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) Send(cmd string, args ...interface{}) error {
|
||||||
|
c.mu.Lock()
|
||||||
|
c.pending += 1
|
||||||
|
c.mu.Unlock()
|
||||||
|
if c.writeTimeout != 0 {
|
||||||
|
c.conn.SetWriteDeadline(time.Now().Add(c.writeTimeout))
|
||||||
|
}
|
||||||
|
if err := c.writeCommand(cmd, args); err != nil {
|
||||||
|
return c.fatal(err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) Flush() error {
|
||||||
|
if c.writeTimeout != 0 {
|
||||||
|
c.conn.SetWriteDeadline(time.Now().Add(c.writeTimeout))
|
||||||
|
}
|
||||||
|
if err := c.bw.Flush(); err != nil {
|
||||||
|
return c.fatal(err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) Receive() (reply interface{}, err error) {
|
||||||
|
if c.readTimeout != 0 {
|
||||||
|
c.conn.SetReadDeadline(time.Now().Add(c.readTimeout))
|
||||||
|
}
|
||||||
|
if reply, err = c.readReply(); err != nil {
|
||||||
|
return nil, c.fatal(err)
|
||||||
|
}
|
||||||
|
// When using pub/sub, the number of receives can be greater than the
|
||||||
|
// number of sends. To enable normal use of the connection after
|
||||||
|
// unsubscribing from all channels, we do not decrement pending to a
|
||||||
|
// negative value.
|
||||||
|
//
|
||||||
|
// The pending field is decremented after the reply is read to handle the
|
||||||
|
// case where Receive is called before Send.
|
||||||
|
c.mu.Lock()
|
||||||
|
if c.pending > 0 {
|
||||||
|
c.pending -= 1
|
||||||
|
}
|
||||||
|
c.mu.Unlock()
|
||||||
|
if err, ok := reply.(Error); ok {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *conn) Do(cmd string, args ...interface{}) (interface{}, error) {
|
||||||
|
c.mu.Lock()
|
||||||
|
pending := c.pending
|
||||||
|
c.pending = 0
|
||||||
|
c.mu.Unlock()
|
||||||
|
|
||||||
|
if cmd == "" && pending == 0 {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.writeTimeout != 0 {
|
||||||
|
c.conn.SetWriteDeadline(time.Now().Add(c.writeTimeout))
|
||||||
|
}
|
||||||
|
|
||||||
|
if cmd != "" {
|
||||||
|
if err := c.writeCommand(cmd, args); err != nil {
|
||||||
|
return nil, c.fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.bw.Flush(); err != nil {
|
||||||
|
return nil, c.fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.readTimeout != 0 {
|
||||||
|
c.conn.SetReadDeadline(time.Now().Add(c.readTimeout))
|
||||||
|
}
|
||||||
|
|
||||||
|
if cmd == "" {
|
||||||
|
reply := make([]interface{}, pending)
|
||||||
|
for i := range reply {
|
||||||
|
r, e := c.readReply()
|
||||||
|
if e != nil {
|
||||||
|
return nil, c.fatal(e)
|
||||||
|
}
|
||||||
|
reply[i] = r
|
||||||
|
}
|
||||||
|
return reply, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
var reply interface{}
|
||||||
|
for i := 0; i <= pending; i++ {
|
||||||
|
var e error
|
||||||
|
if reply, e = c.readReply(); e != nil {
|
||||||
|
return nil, c.fatal(e)
|
||||||
|
}
|
||||||
|
if e, ok := reply.(Error); ok && err == nil {
|
||||||
|
err = e
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return reply, err
|
||||||
|
}
|
|
@ -0,0 +1,177 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
// Package redis is a client for the Redis database.
|
||||||
|
//
|
||||||
|
// The Redigo FAQ (https://github.com/garyburd/redigo/wiki/FAQ) contains more
|
||||||
|
// documentation about this package.
|
||||||
|
//
|
||||||
|
// Connections
|
||||||
|
//
|
||||||
|
// The Conn interface is the primary interface for working with Redis.
|
||||||
|
// Applications create connections by calling the Dial, DialWithTimeout or
|
||||||
|
// NewConn functions. In the future, functions will be added for creating
|
||||||
|
// sharded and other types of connections.
|
||||||
|
//
|
||||||
|
// The application must call the connection Close method when the application
|
||||||
|
// is done with the connection.
|
||||||
|
//
|
||||||
|
// Executing Commands
|
||||||
|
//
|
||||||
|
// The Conn interface has a generic method for executing Redis commands:
|
||||||
|
//
|
||||||
|
// Do(commandName string, args ...interface{}) (reply interface{}, err error)
|
||||||
|
//
|
||||||
|
// The Redis command reference (http://redis.io/commands) lists the available
|
||||||
|
// commands. An example of using the Redis APPEND command is:
|
||||||
|
//
|
||||||
|
// n, err := conn.Do("APPEND", "key", "value")
|
||||||
|
//
|
||||||
|
// The Do method converts command arguments to binary strings for transmission
|
||||||
|
// to the server as follows:
|
||||||
|
//
|
||||||
|
// Go Type Conversion
|
||||||
|
// []byte Sent as is
|
||||||
|
// string Sent as is
|
||||||
|
// int, int64 strconv.FormatInt(v)
|
||||||
|
// float64 strconv.FormatFloat(v, 'g', -1, 64)
|
||||||
|
// bool true -> "1", false -> "0"
|
||||||
|
// nil ""
|
||||||
|
// all other types fmt.Print(v)
|
||||||
|
//
|
||||||
|
// Redis command reply types are represented using the following Go types:
|
||||||
|
//
|
||||||
|
// Redis type Go type
|
||||||
|
// error redis.Error
|
||||||
|
// integer int64
|
||||||
|
// simple string string
|
||||||
|
// bulk string []byte or nil if value not present.
|
||||||
|
// array []interface{} or nil if value not present.
|
||||||
|
//
|
||||||
|
// Use type assertions or the reply helper functions to convert from
|
||||||
|
// interface{} to the specific Go type for the command result.
|
||||||
|
//
|
||||||
|
// Pipelining
|
||||||
|
//
|
||||||
|
// Connections support pipelining using the Send, Flush and Receive methods.
|
||||||
|
//
|
||||||
|
// Send(commandName string, args ...interface{}) error
|
||||||
|
// Flush() error
|
||||||
|
// Receive() (reply interface{}, err error)
|
||||||
|
//
|
||||||
|
// Send writes the command to the connection's output buffer. Flush flushes the
|
||||||
|
// connection's output buffer to the server. Receive reads a single reply from
|
||||||
|
// the server. The following example shows a simple pipeline.
|
||||||
|
//
|
||||||
|
// c.Send("SET", "foo", "bar")
|
||||||
|
// c.Send("GET", "foo")
|
||||||
|
// c.Flush()
|
||||||
|
// c.Receive() // reply from SET
|
||||||
|
// v, err = c.Receive() // reply from GET
|
||||||
|
//
|
||||||
|
// The Do method combines the functionality of the Send, Flush and Receive
|
||||||
|
// methods. The Do method starts by writing the command and flushing the output
|
||||||
|
// buffer. Next, the Do method receives all pending replies including the reply
|
||||||
|
// for the command just sent by Do. If any of the received replies is an error,
|
||||||
|
// then Do returns the error. If there are no errors, then Do returns the last
|
||||||
|
// reply. If the command argument to the Do method is "", then the Do method
|
||||||
|
// will flush the output buffer and receive pending replies without sending a
|
||||||
|
// command.
|
||||||
|
//
|
||||||
|
// Use the Send and Do methods to implement pipelined transactions.
|
||||||
|
//
|
||||||
|
// c.Send("MULTI")
|
||||||
|
// c.Send("INCR", "foo")
|
||||||
|
// c.Send("INCR", "bar")
|
||||||
|
// r, err := c.Do("EXEC")
|
||||||
|
// fmt.Println(r) // prints [1, 1]
|
||||||
|
//
|
||||||
|
// Concurrency
|
||||||
|
//
|
||||||
|
// Connections support one concurrent caller to the Receive method and one
|
||||||
|
// concurrent caller to the Send and Flush methods. No other concurrency is
|
||||||
|
// supported including concurrent calls to the Do method.
|
||||||
|
//
|
||||||
|
// For full concurrent access to Redis, use the thread-safe Pool to get, use
|
||||||
|
// and release a connection from within a goroutine. Connections returned from
|
||||||
|
// a Pool have the concurrency restrictions described in the previous
|
||||||
|
// paragraph.
|
||||||
|
//
|
||||||
|
// Publish and Subscribe
|
||||||
|
//
|
||||||
|
// Use the Send, Flush and Receive methods to implement Pub/Sub subscribers.
|
||||||
|
//
|
||||||
|
// c.Send("SUBSCRIBE", "example")
|
||||||
|
// c.Flush()
|
||||||
|
// for {
|
||||||
|
// reply, err := c.Receive()
|
||||||
|
// if err != nil {
|
||||||
|
// return err
|
||||||
|
// }
|
||||||
|
// // process pushed message
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// The PubSubConn type wraps a Conn with convenience methods for implementing
|
||||||
|
// subscribers. The Subscribe, PSubscribe, Unsubscribe and PUnsubscribe methods
|
||||||
|
// send and flush a subscription management command. The receive method
|
||||||
|
// converts a pushed message to convenient types for use in a type switch.
|
||||||
|
//
|
||||||
|
// psc := redis.PubSubConn{Conn: c}
|
||||||
|
// psc.Subscribe("example")
|
||||||
|
// for {
|
||||||
|
// switch v := psc.Receive().(type) {
|
||||||
|
// case redis.Message:
|
||||||
|
// fmt.Printf("%s: message: %s\n", v.Channel, v.Data)
|
||||||
|
// case redis.Subscription:
|
||||||
|
// fmt.Printf("%s: %s %d\n", v.Channel, v.Kind, v.Count)
|
||||||
|
// case error:
|
||||||
|
// return v
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Reply Helpers
|
||||||
|
//
|
||||||
|
// The Bool, Int, Bytes, String, Strings and Values functions convert a reply
|
||||||
|
// to a value of a specific type. To allow convenient wrapping of calls to the
|
||||||
|
// connection Do and Receive methods, the functions take a second argument of
|
||||||
|
// type error. If the error is non-nil, then the helper function returns the
|
||||||
|
// error. If the error is nil, the function converts the reply to the specified
|
||||||
|
// type:
|
||||||
|
//
|
||||||
|
// exists, err := redis.Bool(c.Do("EXISTS", "foo"))
|
||||||
|
// if err != nil {
|
||||||
|
// // handle error return from c.Do or type conversion error.
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// The Scan function converts elements of a array reply to Go types:
|
||||||
|
//
|
||||||
|
// var value1 int
|
||||||
|
// var value2 string
|
||||||
|
// reply, err := redis.Values(c.Do("MGET", "key1", "key2"))
|
||||||
|
// if err != nil {
|
||||||
|
// // handle error
|
||||||
|
// }
|
||||||
|
// if _, err := redis.Scan(reply, &value1, &value2); err != nil {
|
||||||
|
// // handle error
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Errors
|
||||||
|
//
|
||||||
|
// Connection methods return error replies from the server as type redis.Error.
|
||||||
|
//
|
||||||
|
// Call the connection Err() method to determine if the connection encountered
|
||||||
|
// non-recoverable error such as a network error or protocol parsing error. If
|
||||||
|
// Err() returns a non-nil value, then the connection is not usable and should
|
||||||
|
// be closed.
|
||||||
|
package redis
|
|
@ -0,0 +1,33 @@
|
||||||
|
// +build go1.7
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import "crypto/tls"
|
||||||
|
|
||||||
|
// similar cloneTLSClientConfig in the stdlib, but also honor skipVerify for the nil case
|
||||||
|
func cloneTLSClientConfig(cfg *tls.Config, skipVerify bool) *tls.Config {
|
||||||
|
if cfg == nil {
|
||||||
|
return &tls.Config{InsecureSkipVerify: skipVerify}
|
||||||
|
}
|
||||||
|
return &tls.Config{
|
||||||
|
Rand: cfg.Rand,
|
||||||
|
Time: cfg.Time,
|
||||||
|
Certificates: cfg.Certificates,
|
||||||
|
NameToCertificate: cfg.NameToCertificate,
|
||||||
|
GetCertificate: cfg.GetCertificate,
|
||||||
|
RootCAs: cfg.RootCAs,
|
||||||
|
NextProtos: cfg.NextProtos,
|
||||||
|
ServerName: cfg.ServerName,
|
||||||
|
ClientAuth: cfg.ClientAuth,
|
||||||
|
ClientCAs: cfg.ClientCAs,
|
||||||
|
InsecureSkipVerify: cfg.InsecureSkipVerify,
|
||||||
|
CipherSuites: cfg.CipherSuites,
|
||||||
|
PreferServerCipherSuites: cfg.PreferServerCipherSuites,
|
||||||
|
ClientSessionCache: cfg.ClientSessionCache,
|
||||||
|
MinVersion: cfg.MinVersion,
|
||||||
|
MaxVersion: cfg.MaxVersion,
|
||||||
|
CurvePreferences: cfg.CurvePreferences,
|
||||||
|
DynamicRecordSizingDisabled: cfg.DynamicRecordSizingDisabled,
|
||||||
|
Renegotiation: cfg.Renegotiation,
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,117 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NewLoggingConn returns a logging wrapper around a connection.
|
||||||
|
func NewLoggingConn(conn Conn, logger *log.Logger, prefix string) Conn {
|
||||||
|
if prefix != "" {
|
||||||
|
prefix = prefix + "."
|
||||||
|
}
|
||||||
|
return &loggingConn{conn, logger, prefix}
|
||||||
|
}
|
||||||
|
|
||||||
|
type loggingConn struct {
|
||||||
|
Conn
|
||||||
|
logger *log.Logger
|
||||||
|
prefix string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *loggingConn) Close() error {
|
||||||
|
err := c.Conn.Close()
|
||||||
|
var buf bytes.Buffer
|
||||||
|
fmt.Fprintf(&buf, "%sClose() -> (%v)", c.prefix, err)
|
||||||
|
c.logger.Output(2, buf.String())
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *loggingConn) printValue(buf *bytes.Buffer, v interface{}) {
|
||||||
|
const chop = 32
|
||||||
|
switch v := v.(type) {
|
||||||
|
case []byte:
|
||||||
|
if len(v) > chop {
|
||||||
|
fmt.Fprintf(buf, "%q...", v[:chop])
|
||||||
|
} else {
|
||||||
|
fmt.Fprintf(buf, "%q", v)
|
||||||
|
}
|
||||||
|
case string:
|
||||||
|
if len(v) > chop {
|
||||||
|
fmt.Fprintf(buf, "%q...", v[:chop])
|
||||||
|
} else {
|
||||||
|
fmt.Fprintf(buf, "%q", v)
|
||||||
|
}
|
||||||
|
case []interface{}:
|
||||||
|
if len(v) == 0 {
|
||||||
|
buf.WriteString("[]")
|
||||||
|
} else {
|
||||||
|
sep := "["
|
||||||
|
fin := "]"
|
||||||
|
if len(v) > chop {
|
||||||
|
v = v[:chop]
|
||||||
|
fin = "...]"
|
||||||
|
}
|
||||||
|
for _, vv := range v {
|
||||||
|
buf.WriteString(sep)
|
||||||
|
c.printValue(buf, vv)
|
||||||
|
sep = ", "
|
||||||
|
}
|
||||||
|
buf.WriteString(fin)
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
fmt.Fprint(buf, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *loggingConn) print(method, commandName string, args []interface{}, reply interface{}, err error) {
|
||||||
|
var buf bytes.Buffer
|
||||||
|
fmt.Fprintf(&buf, "%s%s(", c.prefix, method)
|
||||||
|
if method != "Receive" {
|
||||||
|
buf.WriteString(commandName)
|
||||||
|
for _, arg := range args {
|
||||||
|
buf.WriteString(", ")
|
||||||
|
c.printValue(&buf, arg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
buf.WriteString(") -> (")
|
||||||
|
if method != "Send" {
|
||||||
|
c.printValue(&buf, reply)
|
||||||
|
buf.WriteString(", ")
|
||||||
|
}
|
||||||
|
fmt.Fprintf(&buf, "%v)", err)
|
||||||
|
c.logger.Output(3, buf.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *loggingConn) Do(commandName string, args ...interface{}) (interface{}, error) {
|
||||||
|
reply, err := c.Conn.Do(commandName, args...)
|
||||||
|
c.print("Do", commandName, args, reply, err)
|
||||||
|
return reply, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *loggingConn) Send(commandName string, args ...interface{}) error {
|
||||||
|
err := c.Conn.Send(commandName, args...)
|
||||||
|
c.print("Send", commandName, args, nil, err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *loggingConn) Receive() (interface{}, error) {
|
||||||
|
reply, err := c.Conn.Receive()
|
||||||
|
c.print("Receive", "", nil, reply, err)
|
||||||
|
return reply, err
|
||||||
|
}
|
|
@ -0,0 +1,416 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"container/list"
|
||||||
|
"crypto/rand"
|
||||||
|
"crypto/sha1"
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"strconv"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/garyburd/redigo/internal"
|
||||||
|
)
|
||||||
|
|
||||||
|
var nowFunc = time.Now // for testing
|
||||||
|
|
||||||
|
// ErrPoolExhausted is returned from a pool connection method (Do, Send,
|
||||||
|
// Receive, Flush, Err) when the maximum number of database connections in the
|
||||||
|
// pool has been reached.
|
||||||
|
var ErrPoolExhausted = errors.New("redigo: connection pool exhausted")
|
||||||
|
|
||||||
|
var (
|
||||||
|
errPoolClosed = errors.New("redigo: connection pool closed")
|
||||||
|
errConnClosed = errors.New("redigo: connection closed")
|
||||||
|
)
|
||||||
|
|
||||||
|
// Pool maintains a pool of connections. The application calls the Get method
|
||||||
|
// to get a connection from the pool and the connection's Close method to
|
||||||
|
// return the connection's resources to the pool.
|
||||||
|
//
|
||||||
|
// The following example shows how to use a pool in a web application. The
|
||||||
|
// application creates a pool at application startup and makes it available to
|
||||||
|
// request handlers using a package level variable. The pool configuration used
|
||||||
|
// here is an example, not a recommendation.
|
||||||
|
//
|
||||||
|
// func newPool(addr string) *redis.Pool {
|
||||||
|
// return &redis.Pool{
|
||||||
|
// MaxIdle: 3,
|
||||||
|
// IdleTimeout: 240 * time.Second,
|
||||||
|
// Dial: func () (redis.Conn, error) { return redis.Dial("tcp", addr) },
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// var (
|
||||||
|
// pool *redis.Pool
|
||||||
|
// redisServer = flag.String("redisServer", ":6379", "")
|
||||||
|
// )
|
||||||
|
//
|
||||||
|
// func main() {
|
||||||
|
// flag.Parse()
|
||||||
|
// pool = newPool(*redisServer)
|
||||||
|
// ...
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// A request handler gets a connection from the pool and closes the connection
|
||||||
|
// when the handler is done:
|
||||||
|
//
|
||||||
|
// func serveHome(w http.ResponseWriter, r *http.Request) {
|
||||||
|
// conn := pool.Get()
|
||||||
|
// defer conn.Close()
|
||||||
|
// ...
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Use the Dial function to authenticate connections with the AUTH command or
|
||||||
|
// select a database with the SELECT command:
|
||||||
|
//
|
||||||
|
// pool := &redis.Pool{
|
||||||
|
// // Other pool configuration not shown in this example.
|
||||||
|
// Dial: func () (redis.Conn, error) {
|
||||||
|
// c, err := redis.Dial("tcp", server)
|
||||||
|
// if err != nil {
|
||||||
|
// return nil, err
|
||||||
|
// }
|
||||||
|
// if _, err := c.Do("AUTH", password); err != nil {
|
||||||
|
// c.Close()
|
||||||
|
// return nil, err
|
||||||
|
// }
|
||||||
|
// if _, err := c.Do("SELECT", db); err != nil {
|
||||||
|
// c.Close()
|
||||||
|
// return nil, err
|
||||||
|
// }
|
||||||
|
// return c, nil
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// Use the TestOnBorrow function to check the health of an idle connection
|
||||||
|
// before the connection is returned to the application. This example PINGs
|
||||||
|
// connections that have been idle more than a minute:
|
||||||
|
//
|
||||||
|
// pool := &redis.Pool{
|
||||||
|
// // Other pool configuration not shown in this example.
|
||||||
|
// TestOnBorrow: func(c redis.Conn, t time.Time) error {
|
||||||
|
// if time.Since(t) < time.Minute {
|
||||||
|
// return nil
|
||||||
|
// }
|
||||||
|
// _, err := c.Do("PING")
|
||||||
|
// return err
|
||||||
|
// },
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
type Pool struct {
|
||||||
|
|
||||||
|
// Dial is an application supplied function for creating and configuring a
|
||||||
|
// connection.
|
||||||
|
//
|
||||||
|
// The connection returned from Dial must not be in a special state
|
||||||
|
// (subscribed to pubsub channel, transaction started, ...).
|
||||||
|
Dial func() (Conn, error)
|
||||||
|
|
||||||
|
// TestOnBorrow is an optional application supplied function for checking
|
||||||
|
// the health of an idle connection before the connection is used again by
|
||||||
|
// the application. Argument t is the time that the connection was returned
|
||||||
|
// to the pool. If the function returns an error, then the connection is
|
||||||
|
// closed.
|
||||||
|
TestOnBorrow func(c Conn, t time.Time) error
|
||||||
|
|
||||||
|
// Maximum number of idle connections in the pool.
|
||||||
|
MaxIdle int
|
||||||
|
|
||||||
|
// Maximum number of connections allocated by the pool at a given time.
|
||||||
|
// When zero, there is no limit on the number of connections in the pool.
|
||||||
|
MaxActive int
|
||||||
|
|
||||||
|
// Close connections after remaining idle for this duration. If the value
|
||||||
|
// is zero, then idle connections are not closed. Applications should set
|
||||||
|
// the timeout to a value less than the server's timeout.
|
||||||
|
IdleTimeout time.Duration
|
||||||
|
|
||||||
|
// If Wait is true and the pool is at the MaxActive limit, then Get() waits
|
||||||
|
// for a connection to be returned to the pool before returning.
|
||||||
|
Wait bool
|
||||||
|
|
||||||
|
// mu protects fields defined below.
|
||||||
|
mu sync.Mutex
|
||||||
|
cond *sync.Cond
|
||||||
|
closed bool
|
||||||
|
active int
|
||||||
|
|
||||||
|
// Stack of idleConn with most recently used at the front.
|
||||||
|
idle list.List
|
||||||
|
}
|
||||||
|
|
||||||
|
type idleConn struct {
|
||||||
|
c Conn
|
||||||
|
t time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewPool creates a new pool.
|
||||||
|
//
|
||||||
|
// Deprecated: Initialize the Pool directory as shown in the example.
|
||||||
|
func NewPool(newFn func() (Conn, error), maxIdle int) *Pool {
|
||||||
|
return &Pool{Dial: newFn, MaxIdle: maxIdle}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get gets a connection. The application must close the returned connection.
|
||||||
|
// This method always returns a valid connection so that applications can defer
|
||||||
|
// error handling to the first use of the connection. If there is an error
|
||||||
|
// getting an underlying connection, then the connection Err, Do, Send, Flush
|
||||||
|
// and Receive methods return that error.
|
||||||
|
func (p *Pool) Get() Conn {
|
||||||
|
c, err := p.get()
|
||||||
|
if err != nil {
|
||||||
|
return errorConnection{err}
|
||||||
|
}
|
||||||
|
return &pooledConnection{p: p, c: c}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ActiveCount returns the number of active connections in the pool.
|
||||||
|
func (p *Pool) ActiveCount() int {
|
||||||
|
p.mu.Lock()
|
||||||
|
active := p.active
|
||||||
|
p.mu.Unlock()
|
||||||
|
return active
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close releases the resources used by the pool.
|
||||||
|
func (p *Pool) Close() error {
|
||||||
|
p.mu.Lock()
|
||||||
|
idle := p.idle
|
||||||
|
p.idle.Init()
|
||||||
|
p.closed = true
|
||||||
|
p.active -= idle.Len()
|
||||||
|
if p.cond != nil {
|
||||||
|
p.cond.Broadcast()
|
||||||
|
}
|
||||||
|
p.mu.Unlock()
|
||||||
|
for e := idle.Front(); e != nil; e = e.Next() {
|
||||||
|
e.Value.(idleConn).c.Close()
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// release decrements the active count and signals waiters. The caller must
|
||||||
|
// hold p.mu during the call.
|
||||||
|
func (p *Pool) release() {
|
||||||
|
p.active -= 1
|
||||||
|
if p.cond != nil {
|
||||||
|
p.cond.Signal()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// get prunes stale connections and returns a connection from the idle list or
|
||||||
|
// creates a new connection.
|
||||||
|
func (p *Pool) get() (Conn, error) {
|
||||||
|
p.mu.Lock()
|
||||||
|
|
||||||
|
// Prune stale connections.
|
||||||
|
|
||||||
|
if timeout := p.IdleTimeout; timeout > 0 {
|
||||||
|
for i, n := 0, p.idle.Len(); i < n; i++ {
|
||||||
|
e := p.idle.Back()
|
||||||
|
if e == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
ic := e.Value.(idleConn)
|
||||||
|
if ic.t.Add(timeout).After(nowFunc()) {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
p.idle.Remove(e)
|
||||||
|
p.release()
|
||||||
|
p.mu.Unlock()
|
||||||
|
ic.c.Close()
|
||||||
|
p.mu.Lock()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
|
||||||
|
// Get idle connection.
|
||||||
|
|
||||||
|
for i, n := 0, p.idle.Len(); i < n; i++ {
|
||||||
|
e := p.idle.Front()
|
||||||
|
if e == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
ic := e.Value.(idleConn)
|
||||||
|
p.idle.Remove(e)
|
||||||
|
test := p.TestOnBorrow
|
||||||
|
p.mu.Unlock()
|
||||||
|
if test == nil || test(ic.c, ic.t) == nil {
|
||||||
|
return ic.c, nil
|
||||||
|
}
|
||||||
|
ic.c.Close()
|
||||||
|
p.mu.Lock()
|
||||||
|
p.release()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check for pool closed before dialing a new connection.
|
||||||
|
|
||||||
|
if p.closed {
|
||||||
|
p.mu.Unlock()
|
||||||
|
return nil, errors.New("redigo: get on closed pool")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Dial new connection if under limit.
|
||||||
|
|
||||||
|
if p.MaxActive == 0 || p.active < p.MaxActive {
|
||||||
|
dial := p.Dial
|
||||||
|
p.active += 1
|
||||||
|
p.mu.Unlock()
|
||||||
|
c, err := dial()
|
||||||
|
if err != nil {
|
||||||
|
p.mu.Lock()
|
||||||
|
p.release()
|
||||||
|
p.mu.Unlock()
|
||||||
|
c = nil
|
||||||
|
}
|
||||||
|
return c, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !p.Wait {
|
||||||
|
p.mu.Unlock()
|
||||||
|
return nil, ErrPoolExhausted
|
||||||
|
}
|
||||||
|
|
||||||
|
if p.cond == nil {
|
||||||
|
p.cond = sync.NewCond(&p.mu)
|
||||||
|
}
|
||||||
|
p.cond.Wait()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Pool) put(c Conn, forceClose bool) error {
|
||||||
|
err := c.Err()
|
||||||
|
p.mu.Lock()
|
||||||
|
if !p.closed && err == nil && !forceClose {
|
||||||
|
p.idle.PushFront(idleConn{t: nowFunc(), c: c})
|
||||||
|
if p.idle.Len() > p.MaxIdle {
|
||||||
|
c = p.idle.Remove(p.idle.Back()).(idleConn).c
|
||||||
|
} else {
|
||||||
|
c = nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if c == nil {
|
||||||
|
if p.cond != nil {
|
||||||
|
p.cond.Signal()
|
||||||
|
}
|
||||||
|
p.mu.Unlock()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
p.release()
|
||||||
|
p.mu.Unlock()
|
||||||
|
return c.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
type pooledConnection struct {
|
||||||
|
p *Pool
|
||||||
|
c Conn
|
||||||
|
state int
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
sentinel []byte
|
||||||
|
sentinelOnce sync.Once
|
||||||
|
)
|
||||||
|
|
||||||
|
func initSentinel() {
|
||||||
|
p := make([]byte, 64)
|
||||||
|
if _, err := rand.Read(p); err == nil {
|
||||||
|
sentinel = p
|
||||||
|
} else {
|
||||||
|
h := sha1.New()
|
||||||
|
io.WriteString(h, "Oops, rand failed. Use time instead.")
|
||||||
|
io.WriteString(h, strconv.FormatInt(time.Now().UnixNano(), 10))
|
||||||
|
sentinel = h.Sum(nil)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *pooledConnection) Close() error {
|
||||||
|
c := pc.c
|
||||||
|
if _, ok := c.(errorConnection); ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
pc.c = errorConnection{errConnClosed}
|
||||||
|
|
||||||
|
if pc.state&internal.MultiState != 0 {
|
||||||
|
c.Send("DISCARD")
|
||||||
|
pc.state &^= (internal.MultiState | internal.WatchState)
|
||||||
|
} else if pc.state&internal.WatchState != 0 {
|
||||||
|
c.Send("UNWATCH")
|
||||||
|
pc.state &^= internal.WatchState
|
||||||
|
}
|
||||||
|
if pc.state&internal.SubscribeState != 0 {
|
||||||
|
c.Send("UNSUBSCRIBE")
|
||||||
|
c.Send("PUNSUBSCRIBE")
|
||||||
|
// To detect the end of the message stream, ask the server to echo
|
||||||
|
// a sentinel value and read until we see that value.
|
||||||
|
sentinelOnce.Do(initSentinel)
|
||||||
|
c.Send("ECHO", sentinel)
|
||||||
|
c.Flush()
|
||||||
|
for {
|
||||||
|
p, err := c.Receive()
|
||||||
|
if err != nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if p, ok := p.([]byte); ok && bytes.Equal(p, sentinel) {
|
||||||
|
pc.state &^= internal.SubscribeState
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
c.Do("")
|
||||||
|
pc.p.put(c, pc.state != 0)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *pooledConnection) Err() error {
|
||||||
|
return pc.c.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *pooledConnection) Do(commandName string, args ...interface{}) (reply interface{}, err error) {
|
||||||
|
ci := internal.LookupCommandInfo(commandName)
|
||||||
|
pc.state = (pc.state | ci.Set) &^ ci.Clear
|
||||||
|
return pc.c.Do(commandName, args...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *pooledConnection) Send(commandName string, args ...interface{}) error {
|
||||||
|
ci := internal.LookupCommandInfo(commandName)
|
||||||
|
pc.state = (pc.state | ci.Set) &^ ci.Clear
|
||||||
|
return pc.c.Send(commandName, args...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *pooledConnection) Flush() error {
|
||||||
|
return pc.c.Flush()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (pc *pooledConnection) Receive() (reply interface{}, err error) {
|
||||||
|
return pc.c.Receive()
|
||||||
|
}
|
||||||
|
|
||||||
|
type errorConnection struct{ err error }
|
||||||
|
|
||||||
|
func (ec errorConnection) Do(string, ...interface{}) (interface{}, error) { return nil, ec.err }
|
||||||
|
func (ec errorConnection) Send(string, ...interface{}) error { return ec.err }
|
||||||
|
func (ec errorConnection) Err() error { return ec.err }
|
||||||
|
func (ec errorConnection) Close() error { return ec.err }
|
||||||
|
func (ec errorConnection) Flush() error { return ec.err }
|
||||||
|
func (ec errorConnection) Receive() (interface{}, error) { return nil, ec.err }
|
|
@ -0,0 +1,31 @@
|
||||||
|
// +build !go1.7
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import "crypto/tls"
|
||||||
|
|
||||||
|
// similar cloneTLSClientConfig in the stdlib, but also honor skipVerify for the nil case
|
||||||
|
func cloneTLSClientConfig(cfg *tls.Config, skipVerify bool) *tls.Config {
|
||||||
|
if cfg == nil {
|
||||||
|
return &tls.Config{InsecureSkipVerify: skipVerify}
|
||||||
|
}
|
||||||
|
return &tls.Config{
|
||||||
|
Rand: cfg.Rand,
|
||||||
|
Time: cfg.Time,
|
||||||
|
Certificates: cfg.Certificates,
|
||||||
|
NameToCertificate: cfg.NameToCertificate,
|
||||||
|
GetCertificate: cfg.GetCertificate,
|
||||||
|
RootCAs: cfg.RootCAs,
|
||||||
|
NextProtos: cfg.NextProtos,
|
||||||
|
ServerName: cfg.ServerName,
|
||||||
|
ClientAuth: cfg.ClientAuth,
|
||||||
|
ClientCAs: cfg.ClientCAs,
|
||||||
|
InsecureSkipVerify: cfg.InsecureSkipVerify,
|
||||||
|
CipherSuites: cfg.CipherSuites,
|
||||||
|
PreferServerCipherSuites: cfg.PreferServerCipherSuites,
|
||||||
|
ClientSessionCache: cfg.ClientSessionCache,
|
||||||
|
MinVersion: cfg.MinVersion,
|
||||||
|
MaxVersion: cfg.MaxVersion,
|
||||||
|
CurvePreferences: cfg.CurvePreferences,
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,144 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import "errors"
|
||||||
|
|
||||||
|
// Subscription represents a subscribe or unsubscribe notification.
|
||||||
|
type Subscription struct {
|
||||||
|
|
||||||
|
// Kind is "subscribe", "unsubscribe", "psubscribe" or "punsubscribe"
|
||||||
|
Kind string
|
||||||
|
|
||||||
|
// The channel that was changed.
|
||||||
|
Channel string
|
||||||
|
|
||||||
|
// The current number of subscriptions for connection.
|
||||||
|
Count int
|
||||||
|
}
|
||||||
|
|
||||||
|
// Message represents a message notification.
|
||||||
|
type Message struct {
|
||||||
|
|
||||||
|
// The originating channel.
|
||||||
|
Channel string
|
||||||
|
|
||||||
|
// The message data.
|
||||||
|
Data []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
// PMessage represents a pmessage notification.
|
||||||
|
type PMessage struct {
|
||||||
|
|
||||||
|
// The matched pattern.
|
||||||
|
Pattern string
|
||||||
|
|
||||||
|
// The originating channel.
|
||||||
|
Channel string
|
||||||
|
|
||||||
|
// The message data.
|
||||||
|
Data []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pong represents a pubsub pong notification.
|
||||||
|
type Pong struct {
|
||||||
|
Data string
|
||||||
|
}
|
||||||
|
|
||||||
|
// PubSubConn wraps a Conn with convenience methods for subscribers.
|
||||||
|
type PubSubConn struct {
|
||||||
|
Conn Conn
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close closes the connection.
|
||||||
|
func (c PubSubConn) Close() error {
|
||||||
|
return c.Conn.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Subscribe subscribes the connection to the specified channels.
|
||||||
|
func (c PubSubConn) Subscribe(channel ...interface{}) error {
|
||||||
|
c.Conn.Send("SUBSCRIBE", channel...)
|
||||||
|
return c.Conn.Flush()
|
||||||
|
}
|
||||||
|
|
||||||
|
// PSubscribe subscribes the connection to the given patterns.
|
||||||
|
func (c PubSubConn) PSubscribe(channel ...interface{}) error {
|
||||||
|
c.Conn.Send("PSUBSCRIBE", channel...)
|
||||||
|
return c.Conn.Flush()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unsubscribe unsubscribes the connection from the given channels, or from all
|
||||||
|
// of them if none is given.
|
||||||
|
func (c PubSubConn) Unsubscribe(channel ...interface{}) error {
|
||||||
|
c.Conn.Send("UNSUBSCRIBE", channel...)
|
||||||
|
return c.Conn.Flush()
|
||||||
|
}
|
||||||
|
|
||||||
|
// PUnsubscribe unsubscribes the connection from the given patterns, or from all
|
||||||
|
// of them if none is given.
|
||||||
|
func (c PubSubConn) PUnsubscribe(channel ...interface{}) error {
|
||||||
|
c.Conn.Send("PUNSUBSCRIBE", channel...)
|
||||||
|
return c.Conn.Flush()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ping sends a PING to the server with the specified data.
|
||||||
|
func (c PubSubConn) Ping(data string) error {
|
||||||
|
c.Conn.Send("PING", data)
|
||||||
|
return c.Conn.Flush()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Receive returns a pushed message as a Subscription, Message, PMessage, Pong
|
||||||
|
// or error. The return value is intended to be used directly in a type switch
|
||||||
|
// as illustrated in the PubSubConn example.
|
||||||
|
func (c PubSubConn) Receive() interface{} {
|
||||||
|
reply, err := Values(c.Conn.Receive())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var kind string
|
||||||
|
reply, err = Scan(reply, &kind)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
switch kind {
|
||||||
|
case "message":
|
||||||
|
var m Message
|
||||||
|
if _, err := Scan(reply, &m.Channel, &m.Data); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return m
|
||||||
|
case "pmessage":
|
||||||
|
var pm PMessage
|
||||||
|
if _, err := Scan(reply, &pm.Pattern, &pm.Channel, &pm.Data); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return pm
|
||||||
|
case "subscribe", "psubscribe", "unsubscribe", "punsubscribe":
|
||||||
|
s := Subscription{Kind: kind}
|
||||||
|
if _, err := Scan(reply, &s.Channel, &s.Count); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
case "pong":
|
||||||
|
var p Pong
|
||||||
|
if _, err := Scan(reply, &p.Data); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return p
|
||||||
|
}
|
||||||
|
return errors.New("redigo: unknown pubsub notification")
|
||||||
|
}
|
|
@ -0,0 +1,41 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
// Error represents an error returned in a command reply.
|
||||||
|
type Error string
|
||||||
|
|
||||||
|
func (err Error) Error() string { return string(err) }
|
||||||
|
|
||||||
|
// Conn represents a connection to a Redis server.
|
||||||
|
type Conn interface {
|
||||||
|
// Close closes the connection.
|
||||||
|
Close() error
|
||||||
|
|
||||||
|
// Err returns a non-nil value when the connection is not usable.
|
||||||
|
Err() error
|
||||||
|
|
||||||
|
// Do sends a command to the server and returns the received reply.
|
||||||
|
Do(commandName string, args ...interface{}) (reply interface{}, err error)
|
||||||
|
|
||||||
|
// Send writes the command to the client's output buffer.
|
||||||
|
Send(commandName string, args ...interface{}) error
|
||||||
|
|
||||||
|
// Flush flushes the output buffer to the Redis server.
|
||||||
|
Flush() error
|
||||||
|
|
||||||
|
// Receive receives a single reply from the Redis server
|
||||||
|
Receive() (reply interface{}, err error)
|
||||||
|
}
|
|
@ -0,0 +1,425 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ErrNil indicates that a reply value is nil.
|
||||||
|
var ErrNil = errors.New("redigo: nil returned")
|
||||||
|
|
||||||
|
// Int is a helper that converts a command reply to an integer. If err is not
|
||||||
|
// equal to nil, then Int returns 0, err. Otherwise, Int converts the
|
||||||
|
// reply to an int as follows:
|
||||||
|
//
|
||||||
|
// Reply type Result
|
||||||
|
// integer int(reply), nil
|
||||||
|
// bulk string parsed reply, nil
|
||||||
|
// nil 0, ErrNil
|
||||||
|
// other 0, error
|
||||||
|
func Int(reply interface{}, err error) (int, error) {
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case int64:
|
||||||
|
x := int(reply)
|
||||||
|
if int64(x) != reply {
|
||||||
|
return 0, strconv.ErrRange
|
||||||
|
}
|
||||||
|
return x, nil
|
||||||
|
case []byte:
|
||||||
|
n, err := strconv.ParseInt(string(reply), 10, 0)
|
||||||
|
return int(n), err
|
||||||
|
case nil:
|
||||||
|
return 0, ErrNil
|
||||||
|
case Error:
|
||||||
|
return 0, reply
|
||||||
|
}
|
||||||
|
return 0, fmt.Errorf("redigo: unexpected type for Int, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Int64 is a helper that converts a command reply to 64 bit integer. If err is
|
||||||
|
// not equal to nil, then Int returns 0, err. Otherwise, Int64 converts the
|
||||||
|
// reply to an int64 as follows:
|
||||||
|
//
|
||||||
|
// Reply type Result
|
||||||
|
// integer reply, nil
|
||||||
|
// bulk string parsed reply, nil
|
||||||
|
// nil 0, ErrNil
|
||||||
|
// other 0, error
|
||||||
|
func Int64(reply interface{}, err error) (int64, error) {
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case int64:
|
||||||
|
return reply, nil
|
||||||
|
case []byte:
|
||||||
|
n, err := strconv.ParseInt(string(reply), 10, 64)
|
||||||
|
return n, err
|
||||||
|
case nil:
|
||||||
|
return 0, ErrNil
|
||||||
|
case Error:
|
||||||
|
return 0, reply
|
||||||
|
}
|
||||||
|
return 0, fmt.Errorf("redigo: unexpected type for Int64, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
var errNegativeInt = errors.New("redigo: unexpected value for Uint64")
|
||||||
|
|
||||||
|
// Uint64 is a helper that converts a command reply to 64 bit integer. If err is
|
||||||
|
// not equal to nil, then Int returns 0, err. Otherwise, Int64 converts the
|
||||||
|
// reply to an int64 as follows:
|
||||||
|
//
|
||||||
|
// Reply type Result
|
||||||
|
// integer reply, nil
|
||||||
|
// bulk string parsed reply, nil
|
||||||
|
// nil 0, ErrNil
|
||||||
|
// other 0, error
|
||||||
|
func Uint64(reply interface{}, err error) (uint64, error) {
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case int64:
|
||||||
|
if reply < 0 {
|
||||||
|
return 0, errNegativeInt
|
||||||
|
}
|
||||||
|
return uint64(reply), nil
|
||||||
|
case []byte:
|
||||||
|
n, err := strconv.ParseUint(string(reply), 10, 64)
|
||||||
|
return n, err
|
||||||
|
case nil:
|
||||||
|
return 0, ErrNil
|
||||||
|
case Error:
|
||||||
|
return 0, reply
|
||||||
|
}
|
||||||
|
return 0, fmt.Errorf("redigo: unexpected type for Uint64, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Float64 is a helper that converts a command reply to 64 bit float. If err is
|
||||||
|
// not equal to nil, then Float64 returns 0, err. Otherwise, Float64 converts
|
||||||
|
// the reply to an int as follows:
|
||||||
|
//
|
||||||
|
// Reply type Result
|
||||||
|
// bulk string parsed reply, nil
|
||||||
|
// nil 0, ErrNil
|
||||||
|
// other 0, error
|
||||||
|
func Float64(reply interface{}, err error) (float64, error) {
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case []byte:
|
||||||
|
n, err := strconv.ParseFloat(string(reply), 64)
|
||||||
|
return n, err
|
||||||
|
case nil:
|
||||||
|
return 0, ErrNil
|
||||||
|
case Error:
|
||||||
|
return 0, reply
|
||||||
|
}
|
||||||
|
return 0, fmt.Errorf("redigo: unexpected type for Float64, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// String is a helper that converts a command reply to a string. If err is not
|
||||||
|
// equal to nil, then String returns "", err. Otherwise String converts the
|
||||||
|
// reply to a string as follows:
|
||||||
|
//
|
||||||
|
// Reply type Result
|
||||||
|
// bulk string string(reply), nil
|
||||||
|
// simple string reply, nil
|
||||||
|
// nil "", ErrNil
|
||||||
|
// other "", error
|
||||||
|
func String(reply interface{}, err error) (string, error) {
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case []byte:
|
||||||
|
return string(reply), nil
|
||||||
|
case string:
|
||||||
|
return reply, nil
|
||||||
|
case nil:
|
||||||
|
return "", ErrNil
|
||||||
|
case Error:
|
||||||
|
return "", reply
|
||||||
|
}
|
||||||
|
return "", fmt.Errorf("redigo: unexpected type for String, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Bytes is a helper that converts a command reply to a slice of bytes. If err
|
||||||
|
// is not equal to nil, then Bytes returns nil, err. Otherwise Bytes converts
|
||||||
|
// the reply to a slice of bytes as follows:
|
||||||
|
//
|
||||||
|
// Reply type Result
|
||||||
|
// bulk string reply, nil
|
||||||
|
// simple string []byte(reply), nil
|
||||||
|
// nil nil, ErrNil
|
||||||
|
// other nil, error
|
||||||
|
func Bytes(reply interface{}, err error) ([]byte, error) {
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case []byte:
|
||||||
|
return reply, nil
|
||||||
|
case string:
|
||||||
|
return []byte(reply), nil
|
||||||
|
case nil:
|
||||||
|
return nil, ErrNil
|
||||||
|
case Error:
|
||||||
|
return nil, reply
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("redigo: unexpected type for Bytes, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Bool is a helper that converts a command reply to a boolean. If err is not
|
||||||
|
// equal to nil, then Bool returns false, err. Otherwise Bool converts the
|
||||||
|
// reply to boolean as follows:
|
||||||
|
//
|
||||||
|
// Reply type Result
|
||||||
|
// integer value != 0, nil
|
||||||
|
// bulk string strconv.ParseBool(reply)
|
||||||
|
// nil false, ErrNil
|
||||||
|
// other false, error
|
||||||
|
func Bool(reply interface{}, err error) (bool, error) {
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case int64:
|
||||||
|
return reply != 0, nil
|
||||||
|
case []byte:
|
||||||
|
return strconv.ParseBool(string(reply))
|
||||||
|
case nil:
|
||||||
|
return false, ErrNil
|
||||||
|
case Error:
|
||||||
|
return false, reply
|
||||||
|
}
|
||||||
|
return false, fmt.Errorf("redigo: unexpected type for Bool, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MultiBulk is a helper that converts an array command reply to a []interface{}.
|
||||||
|
//
|
||||||
|
// Deprecated: Use Values instead.
|
||||||
|
func MultiBulk(reply interface{}, err error) ([]interface{}, error) { return Values(reply, err) }
|
||||||
|
|
||||||
|
// Values is a helper that converts an array command reply to a []interface{}.
|
||||||
|
// If err is not equal to nil, then Values returns nil, err. Otherwise, Values
|
||||||
|
// converts the reply as follows:
|
||||||
|
//
|
||||||
|
// Reply type Result
|
||||||
|
// array reply, nil
|
||||||
|
// nil nil, ErrNil
|
||||||
|
// other nil, error
|
||||||
|
func Values(reply interface{}, err error) ([]interface{}, error) {
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case []interface{}:
|
||||||
|
return reply, nil
|
||||||
|
case nil:
|
||||||
|
return nil, ErrNil
|
||||||
|
case Error:
|
||||||
|
return nil, reply
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("redigo: unexpected type for Values, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Strings is a helper that converts an array command reply to a []string. If
|
||||||
|
// err is not equal to nil, then Strings returns nil, err. Nil array items are
|
||||||
|
// converted to "" in the output slice. Strings returns an error if an array
|
||||||
|
// item is not a bulk string or nil.
|
||||||
|
func Strings(reply interface{}, err error) ([]string, error) {
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case []interface{}:
|
||||||
|
result := make([]string, len(reply))
|
||||||
|
for i := range reply {
|
||||||
|
if reply[i] == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
p, ok := reply[i].([]byte)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("redigo: unexpected element type for Strings, got type %T", reply[i])
|
||||||
|
}
|
||||||
|
result[i] = string(p)
|
||||||
|
}
|
||||||
|
return result, nil
|
||||||
|
case nil:
|
||||||
|
return nil, ErrNil
|
||||||
|
case Error:
|
||||||
|
return nil, reply
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("redigo: unexpected type for Strings, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ByteSlices is a helper that converts an array command reply to a [][]byte.
|
||||||
|
// If err is not equal to nil, then ByteSlices returns nil, err. Nil array
|
||||||
|
// items are stay nil. ByteSlices returns an error if an array item is not a
|
||||||
|
// bulk string or nil.
|
||||||
|
func ByteSlices(reply interface{}, err error) ([][]byte, error) {
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
switch reply := reply.(type) {
|
||||||
|
case []interface{}:
|
||||||
|
result := make([][]byte, len(reply))
|
||||||
|
for i := range reply {
|
||||||
|
if reply[i] == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
p, ok := reply[i].([]byte)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("redigo: unexpected element type for ByteSlices, got type %T", reply[i])
|
||||||
|
}
|
||||||
|
result[i] = p
|
||||||
|
}
|
||||||
|
return result, nil
|
||||||
|
case nil:
|
||||||
|
return nil, ErrNil
|
||||||
|
case Error:
|
||||||
|
return nil, reply
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("redigo: unexpected type for ByteSlices, got type %T", reply)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ints is a helper that converts an array command reply to a []int. If
|
||||||
|
// err is not equal to nil, then Ints returns nil, err.
|
||||||
|
func Ints(reply interface{}, err error) ([]int, error) {
|
||||||
|
var ints []int
|
||||||
|
values, err := Values(reply, err)
|
||||||
|
if err != nil {
|
||||||
|
return ints, err
|
||||||
|
}
|
||||||
|
if err := ScanSlice(values, &ints); err != nil {
|
||||||
|
return ints, err
|
||||||
|
}
|
||||||
|
return ints, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// StringMap is a helper that converts an array of strings (alternating key, value)
|
||||||
|
// into a map[string]string. The HGETALL and CONFIG GET commands return replies in this format.
|
||||||
|
// Requires an even number of values in result.
|
||||||
|
func StringMap(result interface{}, err error) (map[string]string, error) {
|
||||||
|
values, err := Values(result, err)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if len(values)%2 != 0 {
|
||||||
|
return nil, errors.New("redigo: StringMap expects even number of values result")
|
||||||
|
}
|
||||||
|
m := make(map[string]string, len(values)/2)
|
||||||
|
for i := 0; i < len(values); i += 2 {
|
||||||
|
key, okKey := values[i].([]byte)
|
||||||
|
value, okValue := values[i+1].([]byte)
|
||||||
|
if !okKey || !okValue {
|
||||||
|
return nil, errors.New("redigo: ScanMap key not a bulk string value")
|
||||||
|
}
|
||||||
|
m[string(key)] = string(value)
|
||||||
|
}
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// IntMap is a helper that converts an array of strings (alternating key, value)
|
||||||
|
// into a map[string]int. The HGETALL commands return replies in this format.
|
||||||
|
// Requires an even number of values in result.
|
||||||
|
func IntMap(result interface{}, err error) (map[string]int, error) {
|
||||||
|
values, err := Values(result, err)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if len(values)%2 != 0 {
|
||||||
|
return nil, errors.New("redigo: IntMap expects even number of values result")
|
||||||
|
}
|
||||||
|
m := make(map[string]int, len(values)/2)
|
||||||
|
for i := 0; i < len(values); i += 2 {
|
||||||
|
key, ok := values[i].([]byte)
|
||||||
|
if !ok {
|
||||||
|
return nil, errors.New("redigo: ScanMap key not a bulk string value")
|
||||||
|
}
|
||||||
|
value, err := Int(values[i+1], nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
m[string(key)] = value
|
||||||
|
}
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Int64Map is a helper that converts an array of strings (alternating key, value)
|
||||||
|
// into a map[string]int64. The HGETALL commands return replies in this format.
|
||||||
|
// Requires an even number of values in result.
|
||||||
|
func Int64Map(result interface{}, err error) (map[string]int64, error) {
|
||||||
|
values, err := Values(result, err)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if len(values)%2 != 0 {
|
||||||
|
return nil, errors.New("redigo: Int64Map expects even number of values result")
|
||||||
|
}
|
||||||
|
m := make(map[string]int64, len(values)/2)
|
||||||
|
for i := 0; i < len(values); i += 2 {
|
||||||
|
key, ok := values[i].([]byte)
|
||||||
|
if !ok {
|
||||||
|
return nil, errors.New("redigo: ScanMap key not a bulk string value")
|
||||||
|
}
|
||||||
|
value, err := Int64(values[i+1], nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
m[string(key)] = value
|
||||||
|
}
|
||||||
|
return m, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Positions is a helper that converts an array of positions (lat, long)
|
||||||
|
// into a [][2]float64. The GEOPOS command returns replies in this format.
|
||||||
|
func Positions(result interface{}, err error) ([]*[2]float64, error) {
|
||||||
|
values, err := Values(result, err)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
positions := make([]*[2]float64, len(values))
|
||||||
|
for i := range values {
|
||||||
|
if values[i] == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
p, ok := values[i].([]interface{})
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("redigo: unexpected element type for interface slice, got type %T", values[i])
|
||||||
|
}
|
||||||
|
if len(p) != 2 {
|
||||||
|
return nil, fmt.Errorf("redigo: unexpected number of values for a member position, got %d", len(p))
|
||||||
|
}
|
||||||
|
lat, err := Float64(p[0], nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
long, err := Float64(p[1], nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
positions[i] = &[2]float64{lat, long}
|
||||||
|
}
|
||||||
|
return positions, nil
|
||||||
|
}
|
|
@ -0,0 +1,555 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
func ensureLen(d reflect.Value, n int) {
|
||||||
|
if n > d.Cap() {
|
||||||
|
d.Set(reflect.MakeSlice(d.Type(), n, n))
|
||||||
|
} else {
|
||||||
|
d.SetLen(n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func cannotConvert(d reflect.Value, s interface{}) error {
|
||||||
|
var sname string
|
||||||
|
switch s.(type) {
|
||||||
|
case string:
|
||||||
|
sname = "Redis simple string"
|
||||||
|
case Error:
|
||||||
|
sname = "Redis error"
|
||||||
|
case int64:
|
||||||
|
sname = "Redis integer"
|
||||||
|
case []byte:
|
||||||
|
sname = "Redis bulk string"
|
||||||
|
case []interface{}:
|
||||||
|
sname = "Redis array"
|
||||||
|
default:
|
||||||
|
sname = reflect.TypeOf(s).String()
|
||||||
|
}
|
||||||
|
return fmt.Errorf("cannot convert from %s to %s", sname, d.Type())
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertAssignBulkString(d reflect.Value, s []byte) (err error) {
|
||||||
|
switch d.Type().Kind() {
|
||||||
|
case reflect.Float32, reflect.Float64:
|
||||||
|
var x float64
|
||||||
|
x, err = strconv.ParseFloat(string(s), d.Type().Bits())
|
||||||
|
d.SetFloat(x)
|
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
||||||
|
var x int64
|
||||||
|
x, err = strconv.ParseInt(string(s), 10, d.Type().Bits())
|
||||||
|
d.SetInt(x)
|
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
||||||
|
var x uint64
|
||||||
|
x, err = strconv.ParseUint(string(s), 10, d.Type().Bits())
|
||||||
|
d.SetUint(x)
|
||||||
|
case reflect.Bool:
|
||||||
|
var x bool
|
||||||
|
x, err = strconv.ParseBool(string(s))
|
||||||
|
d.SetBool(x)
|
||||||
|
case reflect.String:
|
||||||
|
d.SetString(string(s))
|
||||||
|
case reflect.Slice:
|
||||||
|
if d.Type().Elem().Kind() != reflect.Uint8 {
|
||||||
|
err = cannotConvert(d, s)
|
||||||
|
} else {
|
||||||
|
d.SetBytes(s)
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
err = cannotConvert(d, s)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertAssignInt(d reflect.Value, s int64) (err error) {
|
||||||
|
switch d.Type().Kind() {
|
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
||||||
|
d.SetInt(s)
|
||||||
|
if d.Int() != s {
|
||||||
|
err = strconv.ErrRange
|
||||||
|
d.SetInt(0)
|
||||||
|
}
|
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
||||||
|
if s < 0 {
|
||||||
|
err = strconv.ErrRange
|
||||||
|
} else {
|
||||||
|
x := uint64(s)
|
||||||
|
d.SetUint(x)
|
||||||
|
if d.Uint() != x {
|
||||||
|
err = strconv.ErrRange
|
||||||
|
d.SetUint(0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case reflect.Bool:
|
||||||
|
d.SetBool(s != 0)
|
||||||
|
default:
|
||||||
|
err = cannotConvert(d, s)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertAssignValue(d reflect.Value, s interface{}) (err error) {
|
||||||
|
switch s := s.(type) {
|
||||||
|
case []byte:
|
||||||
|
err = convertAssignBulkString(d, s)
|
||||||
|
case int64:
|
||||||
|
err = convertAssignInt(d, s)
|
||||||
|
default:
|
||||||
|
err = cannotConvert(d, s)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertAssignArray(d reflect.Value, s []interface{}) error {
|
||||||
|
if d.Type().Kind() != reflect.Slice {
|
||||||
|
return cannotConvert(d, s)
|
||||||
|
}
|
||||||
|
ensureLen(d, len(s))
|
||||||
|
for i := 0; i < len(s); i++ {
|
||||||
|
if err := convertAssignValue(d.Index(i), s[i]); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertAssign(d interface{}, s interface{}) (err error) {
|
||||||
|
// Handle the most common destination types using type switches and
|
||||||
|
// fall back to reflection for all other types.
|
||||||
|
switch s := s.(type) {
|
||||||
|
case nil:
|
||||||
|
// ingore
|
||||||
|
case []byte:
|
||||||
|
switch d := d.(type) {
|
||||||
|
case *string:
|
||||||
|
*d = string(s)
|
||||||
|
case *int:
|
||||||
|
*d, err = strconv.Atoi(string(s))
|
||||||
|
case *bool:
|
||||||
|
*d, err = strconv.ParseBool(string(s))
|
||||||
|
case *[]byte:
|
||||||
|
*d = s
|
||||||
|
case *interface{}:
|
||||||
|
*d = s
|
||||||
|
case nil:
|
||||||
|
// skip value
|
||||||
|
default:
|
||||||
|
if d := reflect.ValueOf(d); d.Type().Kind() != reflect.Ptr {
|
||||||
|
err = cannotConvert(d, s)
|
||||||
|
} else {
|
||||||
|
err = convertAssignBulkString(d.Elem(), s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case int64:
|
||||||
|
switch d := d.(type) {
|
||||||
|
case *int:
|
||||||
|
x := int(s)
|
||||||
|
if int64(x) != s {
|
||||||
|
err = strconv.ErrRange
|
||||||
|
x = 0
|
||||||
|
}
|
||||||
|
*d = x
|
||||||
|
case *bool:
|
||||||
|
*d = s != 0
|
||||||
|
case *interface{}:
|
||||||
|
*d = s
|
||||||
|
case nil:
|
||||||
|
// skip value
|
||||||
|
default:
|
||||||
|
if d := reflect.ValueOf(d); d.Type().Kind() != reflect.Ptr {
|
||||||
|
err = cannotConvert(d, s)
|
||||||
|
} else {
|
||||||
|
err = convertAssignInt(d.Elem(), s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case string:
|
||||||
|
switch d := d.(type) {
|
||||||
|
case *string:
|
||||||
|
*d = string(s)
|
||||||
|
default:
|
||||||
|
err = cannotConvert(reflect.ValueOf(d), s)
|
||||||
|
}
|
||||||
|
case []interface{}:
|
||||||
|
switch d := d.(type) {
|
||||||
|
case *[]interface{}:
|
||||||
|
*d = s
|
||||||
|
case *interface{}:
|
||||||
|
*d = s
|
||||||
|
case nil:
|
||||||
|
// skip value
|
||||||
|
default:
|
||||||
|
if d := reflect.ValueOf(d); d.Type().Kind() != reflect.Ptr {
|
||||||
|
err = cannotConvert(d, s)
|
||||||
|
} else {
|
||||||
|
err = convertAssignArray(d.Elem(), s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case Error:
|
||||||
|
err = s
|
||||||
|
default:
|
||||||
|
err = cannotConvert(reflect.ValueOf(d), s)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Scan copies from src to the values pointed at by dest.
|
||||||
|
//
|
||||||
|
// The values pointed at by dest must be an integer, float, boolean, string,
|
||||||
|
// []byte, interface{} or slices of these types. Scan uses the standard strconv
|
||||||
|
// package to convert bulk strings to numeric and boolean types.
|
||||||
|
//
|
||||||
|
// If a dest value is nil, then the corresponding src value is skipped.
|
||||||
|
//
|
||||||
|
// If a src element is nil, then the corresponding dest value is not modified.
|
||||||
|
//
|
||||||
|
// To enable easy use of Scan in a loop, Scan returns the slice of src
|
||||||
|
// following the copied values.
|
||||||
|
func Scan(src []interface{}, dest ...interface{}) ([]interface{}, error) {
|
||||||
|
if len(src) < len(dest) {
|
||||||
|
return nil, errors.New("redigo.Scan: array short")
|
||||||
|
}
|
||||||
|
var err error
|
||||||
|
for i, d := range dest {
|
||||||
|
err = convertAssign(d, src[i])
|
||||||
|
if err != nil {
|
||||||
|
err = fmt.Errorf("redigo.Scan: cannot assign to dest %d: %v", i, err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return src[len(dest):], err
|
||||||
|
}
|
||||||
|
|
||||||
|
type fieldSpec struct {
|
||||||
|
name string
|
||||||
|
index []int
|
||||||
|
omitEmpty bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type structSpec struct {
|
||||||
|
m map[string]*fieldSpec
|
||||||
|
l []*fieldSpec
|
||||||
|
}
|
||||||
|
|
||||||
|
func (ss *structSpec) fieldSpec(name []byte) *fieldSpec {
|
||||||
|
return ss.m[string(name)]
|
||||||
|
}
|
||||||
|
|
||||||
|
func compileStructSpec(t reflect.Type, depth map[string]int, index []int, ss *structSpec) {
|
||||||
|
for i := 0; i < t.NumField(); i++ {
|
||||||
|
f := t.Field(i)
|
||||||
|
switch {
|
||||||
|
case f.PkgPath != "" && !f.Anonymous:
|
||||||
|
// Ignore unexported fields.
|
||||||
|
case f.Anonymous:
|
||||||
|
// TODO: Handle pointers. Requires change to decoder and
|
||||||
|
// protection against infinite recursion.
|
||||||
|
if f.Type.Kind() == reflect.Struct {
|
||||||
|
compileStructSpec(f.Type, depth, append(index, i), ss)
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
fs := &fieldSpec{name: f.Name}
|
||||||
|
tag := f.Tag.Get("redis")
|
||||||
|
p := strings.Split(tag, ",")
|
||||||
|
if len(p) > 0 {
|
||||||
|
if p[0] == "-" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if len(p[0]) > 0 {
|
||||||
|
fs.name = p[0]
|
||||||
|
}
|
||||||
|
for _, s := range p[1:] {
|
||||||
|
switch s {
|
||||||
|
case "omitempty":
|
||||||
|
fs.omitEmpty = true
|
||||||
|
default:
|
||||||
|
panic(fmt.Errorf("redigo: unknown field tag %s for type %s", s, t.Name()))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
d, found := depth[fs.name]
|
||||||
|
if !found {
|
||||||
|
d = 1 << 30
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case len(index) == d:
|
||||||
|
// At same depth, remove from result.
|
||||||
|
delete(ss.m, fs.name)
|
||||||
|
j := 0
|
||||||
|
for i := 0; i < len(ss.l); i++ {
|
||||||
|
if fs.name != ss.l[i].name {
|
||||||
|
ss.l[j] = ss.l[i]
|
||||||
|
j += 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ss.l = ss.l[:j]
|
||||||
|
case len(index) < d:
|
||||||
|
fs.index = make([]int, len(index)+1)
|
||||||
|
copy(fs.index, index)
|
||||||
|
fs.index[len(index)] = i
|
||||||
|
depth[fs.name] = len(index)
|
||||||
|
ss.m[fs.name] = fs
|
||||||
|
ss.l = append(ss.l, fs)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
structSpecMutex sync.RWMutex
|
||||||
|
structSpecCache = make(map[reflect.Type]*structSpec)
|
||||||
|
defaultFieldSpec = &fieldSpec{}
|
||||||
|
)
|
||||||
|
|
||||||
|
func structSpecForType(t reflect.Type) *structSpec {
|
||||||
|
|
||||||
|
structSpecMutex.RLock()
|
||||||
|
ss, found := structSpecCache[t]
|
||||||
|
structSpecMutex.RUnlock()
|
||||||
|
if found {
|
||||||
|
return ss
|
||||||
|
}
|
||||||
|
|
||||||
|
structSpecMutex.Lock()
|
||||||
|
defer structSpecMutex.Unlock()
|
||||||
|
ss, found = structSpecCache[t]
|
||||||
|
if found {
|
||||||
|
return ss
|
||||||
|
}
|
||||||
|
|
||||||
|
ss = &structSpec{m: make(map[string]*fieldSpec)}
|
||||||
|
compileStructSpec(t, make(map[string]int), nil, ss)
|
||||||
|
structSpecCache[t] = ss
|
||||||
|
return ss
|
||||||
|
}
|
||||||
|
|
||||||
|
var errScanStructValue = errors.New("redigo.ScanStruct: value must be non-nil pointer to a struct")
|
||||||
|
|
||||||
|
// ScanStruct scans alternating names and values from src to a struct. The
|
||||||
|
// HGETALL and CONFIG GET commands return replies in this format.
|
||||||
|
//
|
||||||
|
// ScanStruct uses exported field names to match values in the response. Use
|
||||||
|
// 'redis' field tag to override the name:
|
||||||
|
//
|
||||||
|
// Field int `redis:"myName"`
|
||||||
|
//
|
||||||
|
// Fields with the tag redis:"-" are ignored.
|
||||||
|
//
|
||||||
|
// Integer, float, boolean, string and []byte fields are supported. Scan uses the
|
||||||
|
// standard strconv package to convert bulk string values to numeric and
|
||||||
|
// boolean types.
|
||||||
|
//
|
||||||
|
// If a src element is nil, then the corresponding field is not modified.
|
||||||
|
func ScanStruct(src []interface{}, dest interface{}) error {
|
||||||
|
d := reflect.ValueOf(dest)
|
||||||
|
if d.Kind() != reflect.Ptr || d.IsNil() {
|
||||||
|
return errScanStructValue
|
||||||
|
}
|
||||||
|
d = d.Elem()
|
||||||
|
if d.Kind() != reflect.Struct {
|
||||||
|
return errScanStructValue
|
||||||
|
}
|
||||||
|
ss := structSpecForType(d.Type())
|
||||||
|
|
||||||
|
if len(src)%2 != 0 {
|
||||||
|
return errors.New("redigo.ScanStruct: number of values not a multiple of 2")
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < len(src); i += 2 {
|
||||||
|
s := src[i+1]
|
||||||
|
if s == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
name, ok := src[i].([]byte)
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("redigo.ScanStruct: key %d not a bulk string value", i)
|
||||||
|
}
|
||||||
|
fs := ss.fieldSpec(name)
|
||||||
|
if fs == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := convertAssignValue(d.FieldByIndex(fs.index), s); err != nil {
|
||||||
|
return fmt.Errorf("redigo.ScanStruct: cannot assign field %s: %v", fs.name, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
errScanSliceValue = errors.New("redigo.ScanSlice: dest must be non-nil pointer to a struct")
|
||||||
|
)
|
||||||
|
|
||||||
|
// ScanSlice scans src to the slice pointed to by dest. The elements the dest
|
||||||
|
// slice must be integer, float, boolean, string, struct or pointer to struct
|
||||||
|
// values.
|
||||||
|
//
|
||||||
|
// Struct fields must be integer, float, boolean or string values. All struct
|
||||||
|
// fields are used unless a subset is specified using fieldNames.
|
||||||
|
func ScanSlice(src []interface{}, dest interface{}, fieldNames ...string) error {
|
||||||
|
d := reflect.ValueOf(dest)
|
||||||
|
if d.Kind() != reflect.Ptr || d.IsNil() {
|
||||||
|
return errScanSliceValue
|
||||||
|
}
|
||||||
|
d = d.Elem()
|
||||||
|
if d.Kind() != reflect.Slice {
|
||||||
|
return errScanSliceValue
|
||||||
|
}
|
||||||
|
|
||||||
|
isPtr := false
|
||||||
|
t := d.Type().Elem()
|
||||||
|
if t.Kind() == reflect.Ptr && t.Elem().Kind() == reflect.Struct {
|
||||||
|
isPtr = true
|
||||||
|
t = t.Elem()
|
||||||
|
}
|
||||||
|
|
||||||
|
if t.Kind() != reflect.Struct {
|
||||||
|
ensureLen(d, len(src))
|
||||||
|
for i, s := range src {
|
||||||
|
if s == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := convertAssignValue(d.Index(i), s); err != nil {
|
||||||
|
return fmt.Errorf("redigo.ScanSlice: cannot assign element %d: %v", i, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
ss := structSpecForType(t)
|
||||||
|
fss := ss.l
|
||||||
|
if len(fieldNames) > 0 {
|
||||||
|
fss = make([]*fieldSpec, len(fieldNames))
|
||||||
|
for i, name := range fieldNames {
|
||||||
|
fss[i] = ss.m[name]
|
||||||
|
if fss[i] == nil {
|
||||||
|
return fmt.Errorf("redigo.ScanSlice: ScanSlice bad field name %s", name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(fss) == 0 {
|
||||||
|
return errors.New("redigo.ScanSlice: no struct fields")
|
||||||
|
}
|
||||||
|
|
||||||
|
n := len(src) / len(fss)
|
||||||
|
if n*len(fss) != len(src) {
|
||||||
|
return errors.New("redigo.ScanSlice: length not a multiple of struct field count")
|
||||||
|
}
|
||||||
|
|
||||||
|
ensureLen(d, n)
|
||||||
|
for i := 0; i < n; i++ {
|
||||||
|
d := d.Index(i)
|
||||||
|
if isPtr {
|
||||||
|
if d.IsNil() {
|
||||||
|
d.Set(reflect.New(t))
|
||||||
|
}
|
||||||
|
d = d.Elem()
|
||||||
|
}
|
||||||
|
for j, fs := range fss {
|
||||||
|
s := src[i*len(fss)+j]
|
||||||
|
if s == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if err := convertAssignValue(d.FieldByIndex(fs.index), s); err != nil {
|
||||||
|
return fmt.Errorf("redigo.ScanSlice: cannot assign element %d to field %s: %v", i*len(fss)+j, fs.name, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Args is a helper for constructing command arguments from structured values.
|
||||||
|
type Args []interface{}
|
||||||
|
|
||||||
|
// Add returns the result of appending value to args.
|
||||||
|
func (args Args) Add(value ...interface{}) Args {
|
||||||
|
return append(args, value...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddFlat returns the result of appending the flattened value of v to args.
|
||||||
|
//
|
||||||
|
// Maps are flattened by appending the alternating keys and map values to args.
|
||||||
|
//
|
||||||
|
// Slices are flattened by appending the slice elements to args.
|
||||||
|
//
|
||||||
|
// Structs are flattened by appending the alternating names and values of
|
||||||
|
// exported fields to args. If v is a nil struct pointer, then nothing is
|
||||||
|
// appended. The 'redis' field tag overrides struct field names. See ScanStruct
|
||||||
|
// for more information on the use of the 'redis' field tag.
|
||||||
|
//
|
||||||
|
// Other types are appended to args as is.
|
||||||
|
func (args Args) AddFlat(v interface{}) Args {
|
||||||
|
rv := reflect.ValueOf(v)
|
||||||
|
switch rv.Kind() {
|
||||||
|
case reflect.Struct:
|
||||||
|
args = flattenStruct(args, rv)
|
||||||
|
case reflect.Slice:
|
||||||
|
for i := 0; i < rv.Len(); i++ {
|
||||||
|
args = append(args, rv.Index(i).Interface())
|
||||||
|
}
|
||||||
|
case reflect.Map:
|
||||||
|
for _, k := range rv.MapKeys() {
|
||||||
|
args = append(args, k.Interface(), rv.MapIndex(k).Interface())
|
||||||
|
}
|
||||||
|
case reflect.Ptr:
|
||||||
|
if rv.Type().Elem().Kind() == reflect.Struct {
|
||||||
|
if !rv.IsNil() {
|
||||||
|
args = flattenStruct(args, rv.Elem())
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
args = append(args, v)
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
args = append(args, v)
|
||||||
|
}
|
||||||
|
return args
|
||||||
|
}
|
||||||
|
|
||||||
|
func flattenStruct(args Args, v reflect.Value) Args {
|
||||||
|
ss := structSpecForType(v.Type())
|
||||||
|
for _, fs := range ss.l {
|
||||||
|
fv := v.FieldByIndex(fs.index)
|
||||||
|
if fs.omitEmpty {
|
||||||
|
var empty = false
|
||||||
|
switch fv.Kind() {
|
||||||
|
case reflect.Array, reflect.Map, reflect.Slice, reflect.String:
|
||||||
|
empty = fv.Len() == 0
|
||||||
|
case reflect.Bool:
|
||||||
|
empty = !fv.Bool()
|
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
||||||
|
empty = fv.Int() == 0
|
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
|
||||||
|
empty = fv.Uint() == 0
|
||||||
|
case reflect.Float32, reflect.Float64:
|
||||||
|
empty = fv.Float() == 0
|
||||||
|
case reflect.Interface, reflect.Ptr:
|
||||||
|
empty = fv.IsNil()
|
||||||
|
}
|
||||||
|
if empty {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
args = append(args, fs.name, fv.Interface())
|
||||||
|
}
|
||||||
|
return args
|
||||||
|
}
|
|
@ -0,0 +1,86 @@
|
||||||
|
// Copyright 2012 Gary Burd
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
package redis
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/sha1"
|
||||||
|
"encoding/hex"
|
||||||
|
"io"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Script encapsulates the source, hash and key count for a Lua script. See
|
||||||
|
// http://redis.io/commands/eval for information on scripts in Redis.
|
||||||
|
type Script struct {
|
||||||
|
keyCount int
|
||||||
|
src string
|
||||||
|
hash string
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewScript returns a new script object. If keyCount is greater than or equal
|
||||||
|
// to zero, then the count is automatically inserted in the EVAL command
|
||||||
|
// argument list. If keyCount is less than zero, then the application supplies
|
||||||
|
// the count as the first value in the keysAndArgs argument to the Do, Send and
|
||||||
|
// SendHash methods.
|
||||||
|
func NewScript(keyCount int, src string) *Script {
|
||||||
|
h := sha1.New()
|
||||||
|
io.WriteString(h, src)
|
||||||
|
return &Script{keyCount, src, hex.EncodeToString(h.Sum(nil))}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Script) args(spec string, keysAndArgs []interface{}) []interface{} {
|
||||||
|
var args []interface{}
|
||||||
|
if s.keyCount < 0 {
|
||||||
|
args = make([]interface{}, 1+len(keysAndArgs))
|
||||||
|
args[0] = spec
|
||||||
|
copy(args[1:], keysAndArgs)
|
||||||
|
} else {
|
||||||
|
args = make([]interface{}, 2+len(keysAndArgs))
|
||||||
|
args[0] = spec
|
||||||
|
args[1] = s.keyCount
|
||||||
|
copy(args[2:], keysAndArgs)
|
||||||
|
}
|
||||||
|
return args
|
||||||
|
}
|
||||||
|
|
||||||
|
// Do evaluates the script. Under the covers, Do optimistically evaluates the
|
||||||
|
// script using the EVALSHA command. If the command fails because the script is
|
||||||
|
// not loaded, then Do evaluates the script using the EVAL command (thus
|
||||||
|
// causing the script to load).
|
||||||
|
func (s *Script) Do(c Conn, keysAndArgs ...interface{}) (interface{}, error) {
|
||||||
|
v, err := c.Do("EVALSHA", s.args(s.hash, keysAndArgs)...)
|
||||||
|
if e, ok := err.(Error); ok && strings.HasPrefix(string(e), "NOSCRIPT ") {
|
||||||
|
v, err = c.Do("EVAL", s.args(s.src, keysAndArgs)...)
|
||||||
|
}
|
||||||
|
return v, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendHash evaluates the script without waiting for the reply. The script is
|
||||||
|
// evaluated with the EVALSHA command. The application must ensure that the
|
||||||
|
// script is loaded by a previous call to Send, Do or Load methods.
|
||||||
|
func (s *Script) SendHash(c Conn, keysAndArgs ...interface{}) error {
|
||||||
|
return c.Send("EVALSHA", s.args(s.hash, keysAndArgs)...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send evaluates the script without waiting for the reply.
|
||||||
|
func (s *Script) Send(c Conn, keysAndArgs ...interface{}) error {
|
||||||
|
return c.Send("EVAL", s.args(s.src, keysAndArgs)...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load loads the script without evaluating it.
|
||||||
|
func (s *Script) Load(c Conn) error {
|
||||||
|
_, err := c.Do("SCRIPT", "LOAD", s.src)
|
||||||
|
return err
|
||||||
|
}
|
|
@ -0,0 +1,20 @@
|
||||||
|
# This is the official list of Freetype-Go authors for copyright purposes.
|
||||||
|
# This file is distinct from the CONTRIBUTORS files.
|
||||||
|
# See the latter for an explanation.
|
||||||
|
#
|
||||||
|
# Freetype-Go is derived from Freetype, which is written in C. The latter
|
||||||
|
# is copyright 1996-2010 David Turner, Robert Wilhelm, and Werner Lemberg.
|
||||||
|
|
||||||
|
# Names should be added to this file as
|
||||||
|
# Name or Organization <email address>
|
||||||
|
# The email address is not required for organizations.
|
||||||
|
|
||||||
|
# Please keep the list sorted.
|
||||||
|
|
||||||
|
Google Inc.
|
||||||
|
Jeff R. Allen <jra@nella.org>
|
||||||
|
Maksim Kochkin <maxxarts@gmail.com>
|
||||||
|
Michael Fogleman <fogleman@gmail.com>
|
||||||
|
Rémy Oudompheng <oudomphe@phare.normalesup.org>
|
||||||
|
Roger Peppe <rogpeppe@gmail.com>
|
||||||
|
Steven Edwards <steven@stephenwithav.com>
|
|
@ -0,0 +1,38 @@
|
||||||
|
# This is the official list of people who can contribute
|
||||||
|
# (and typically have contributed) code to the Freetype-Go repository.
|
||||||
|
# The AUTHORS file lists the copyright holders; this file
|
||||||
|
# lists people. For example, Google employees are listed here
|
||||||
|
# but not in AUTHORS, because Google holds the copyright.
|
||||||
|
#
|
||||||
|
# The submission process automatically checks to make sure
|
||||||
|
# that people submitting code are listed in this file (by email address).
|
||||||
|
#
|
||||||
|
# Names should be added to this file only after verifying that
|
||||||
|
# the individual or the individual's organization has agreed to
|
||||||
|
# the appropriate Contributor License Agreement, found here:
|
||||||
|
#
|
||||||
|
# http://code.google.com/legal/individual-cla-v1.0.html
|
||||||
|
# http://code.google.com/legal/corporate-cla-v1.0.html
|
||||||
|
#
|
||||||
|
# The agreement for individuals can be filled out on the web.
|
||||||
|
#
|
||||||
|
# When adding J Random Contributor's name to this file,
|
||||||
|
# either J's name or J's organization's name should be
|
||||||
|
# added to the AUTHORS file, depending on whether the
|
||||||
|
# individual or corporate CLA was used.
|
||||||
|
|
||||||
|
# Names should be added to this file like so:
|
||||||
|
# Name <email address>
|
||||||
|
|
||||||
|
# Please keep the list sorted.
|
||||||
|
|
||||||
|
Andrew Gerrand <adg@golang.org>
|
||||||
|
Jeff R. Allen <jra@nella.org> <jeff.allen@gmail.com>
|
||||||
|
Maksim Kochkin <maxxarts@gmail.com>
|
||||||
|
Michael Fogleman <fogleman@gmail.com>
|
||||||
|
Nigel Tao <nigeltao@golang.org>
|
||||||
|
Rémy Oudompheng <oudomphe@phare.normalesup.org> <remyoudompheng@gmail.com>
|
||||||
|
Rob Pike <r@golang.org>
|
||||||
|
Roger Peppe <rogpeppe@gmail.com>
|
||||||
|
Russ Cox <rsc@golang.org>
|
||||||
|
Steven Edwards <steven@stephenwithav.com>
|
|
@ -0,0 +1,12 @@
|
||||||
|
Use of the Freetype-Go software is subject to your choice of exactly one of
|
||||||
|
the following two licenses:
|
||||||
|
* The FreeType License, which is similar to the original BSD license with
|
||||||
|
an advertising clause, or
|
||||||
|
* The GNU General Public License (GPL), version 2 or later.
|
||||||
|
|
||||||
|
The text of these licenses are available in the licenses/ftl.txt and the
|
||||||
|
licenses/gpl.txt files respectively. They are also available at
|
||||||
|
http://freetype.sourceforge.net/license.html
|
||||||
|
|
||||||
|
The Luxi fonts in the testdata directory are licensed separately. See the
|
||||||
|
testdata/COPYING file for details.
|
|
@ -0,0 +1,21 @@
|
||||||
|
The Freetype font rasterizer in the Go programming language.
|
||||||
|
|
||||||
|
To download and install from source:
|
||||||
|
$ go get github.com/golang/freetype
|
||||||
|
|
||||||
|
It is an incomplete port:
|
||||||
|
* It only supports TrueType fonts, and not Type 1 fonts nor bitmap fonts.
|
||||||
|
* It only supports the Unicode encoding.
|
||||||
|
|
||||||
|
There are also some implementation differences:
|
||||||
|
* It uses a 26.6 fixed point co-ordinate system everywhere internally,
|
||||||
|
as opposed to the original Freetype's mix of 26.6 (or 10.6 for 16-bit
|
||||||
|
systems) in some places, and 24.8 in the "smooth" rasterizer.
|
||||||
|
|
||||||
|
Freetype-Go is derived from Freetype, which is written in C. Freetype is
|
||||||
|
copyright 1996-2010 David Turner, Robert Wilhelm, and Werner Lemberg.
|
||||||
|
Freetype-Go is copyright The Freetype-Go Authors, who are listed in the
|
||||||
|
AUTHORS file.
|
||||||
|
|
||||||
|
Unless otherwise noted, the Freetype-Go source files are distributed
|
||||||
|
under the BSD-style license found in the LICENSE file.
|
|
@ -0,0 +1,341 @@
|
||||||
|
// Copyright 2010 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
// The freetype package provides a convenient API to draw text onto an image.
|
||||||
|
// Use the freetype/raster and freetype/truetype packages for lower level
|
||||||
|
// control over rasterization and TrueType parsing.
|
||||||
|
package freetype
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"image"
|
||||||
|
"image/draw"
|
||||||
|
|
||||||
|
"github.com/golang/freetype/raster"
|
||||||
|
"github.com/golang/freetype/truetype"
|
||||||
|
"golang.org/x/image/font"
|
||||||
|
"golang.org/x/image/math/fixed"
|
||||||
|
)
|
||||||
|
|
||||||
|
// These constants determine the size of the glyph cache. The cache is keyed
|
||||||
|
// primarily by the glyph index modulo nGlyphs, and secondarily by sub-pixel
|
||||||
|
// position for the mask image. Sub-pixel positions are quantized to
|
||||||
|
// nXFractions possible values in both the x and y directions.
|
||||||
|
const (
|
||||||
|
nGlyphs = 256
|
||||||
|
nXFractions = 4
|
||||||
|
nYFractions = 1
|
||||||
|
)
|
||||||
|
|
||||||
|
// An entry in the glyph cache is keyed explicitly by the glyph index and
|
||||||
|
// implicitly by the quantized x and y fractional offset. It maps to a mask
|
||||||
|
// image and an offset.
|
||||||
|
type cacheEntry struct {
|
||||||
|
valid bool
|
||||||
|
glyph truetype.Index
|
||||||
|
advanceWidth fixed.Int26_6
|
||||||
|
mask *image.Alpha
|
||||||
|
offset image.Point
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseFont just calls the Parse function from the freetype/truetype package.
|
||||||
|
// It is provided here so that code that imports this package doesn't need
|
||||||
|
// to also include the freetype/truetype package.
|
||||||
|
func ParseFont(b []byte) (*truetype.Font, error) {
|
||||||
|
return truetype.Parse(b)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pt converts from a co-ordinate pair measured in pixels to a fixed.Point26_6
|
||||||
|
// co-ordinate pair measured in fixed.Int26_6 units.
|
||||||
|
func Pt(x, y int) fixed.Point26_6 {
|
||||||
|
return fixed.Point26_6{
|
||||||
|
X: fixed.Int26_6(x << 6),
|
||||||
|
Y: fixed.Int26_6(y << 6),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// A Context holds the state for drawing text in a given font and size.
|
||||||
|
type Context struct {
|
||||||
|
r *raster.Rasterizer
|
||||||
|
f *truetype.Font
|
||||||
|
glyphBuf truetype.GlyphBuf
|
||||||
|
// clip is the clip rectangle for drawing.
|
||||||
|
clip image.Rectangle
|
||||||
|
// dst and src are the destination and source images for drawing.
|
||||||
|
dst draw.Image
|
||||||
|
src image.Image
|
||||||
|
// fontSize and dpi are used to calculate scale. scale is the number of
|
||||||
|
// 26.6 fixed point units in 1 em. hinting is the hinting policy.
|
||||||
|
fontSize, dpi float64
|
||||||
|
scale fixed.Int26_6
|
||||||
|
hinting font.Hinting
|
||||||
|
// cache is the glyph cache.
|
||||||
|
cache [nGlyphs * nXFractions * nYFractions]cacheEntry
|
||||||
|
}
|
||||||
|
|
||||||
|
// PointToFixed converts the given number of points (as in "a 12 point font")
|
||||||
|
// into a 26.6 fixed point number of pixels.
|
||||||
|
func (c *Context) PointToFixed(x float64) fixed.Int26_6 {
|
||||||
|
return fixed.Int26_6(x * float64(c.dpi) * (64.0 / 72.0))
|
||||||
|
}
|
||||||
|
|
||||||
|
// drawContour draws the given closed contour with the given offset.
|
||||||
|
func (c *Context) drawContour(ps []truetype.Point, dx, dy fixed.Int26_6) {
|
||||||
|
if len(ps) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// The low bit of each point's Flags value is whether the point is on the
|
||||||
|
// curve. Truetype fonts only have quadratic Bézier curves, not cubics.
|
||||||
|
// Thus, two consecutive off-curve points imply an on-curve point in the
|
||||||
|
// middle of those two.
|
||||||
|
//
|
||||||
|
// See http://chanae.walon.org/pub/ttf/ttf_glyphs.htm for more details.
|
||||||
|
|
||||||
|
// ps[0] is a truetype.Point measured in FUnits and positive Y going
|
||||||
|
// upwards. start is the same thing measured in fixed point units and
|
||||||
|
// positive Y going downwards, and offset by (dx, dy).
|
||||||
|
start := fixed.Point26_6{
|
||||||
|
X: dx + ps[0].X,
|
||||||
|
Y: dy - ps[0].Y,
|
||||||
|
}
|
||||||
|
others := []truetype.Point(nil)
|
||||||
|
if ps[0].Flags&0x01 != 0 {
|
||||||
|
others = ps[1:]
|
||||||
|
} else {
|
||||||
|
last := fixed.Point26_6{
|
||||||
|
X: dx + ps[len(ps)-1].X,
|
||||||
|
Y: dy - ps[len(ps)-1].Y,
|
||||||
|
}
|
||||||
|
if ps[len(ps)-1].Flags&0x01 != 0 {
|
||||||
|
start = last
|
||||||
|
others = ps[:len(ps)-1]
|
||||||
|
} else {
|
||||||
|
start = fixed.Point26_6{
|
||||||
|
X: (start.X + last.X) / 2,
|
||||||
|
Y: (start.Y + last.Y) / 2,
|
||||||
|
}
|
||||||
|
others = ps
|
||||||
|
}
|
||||||
|
}
|
||||||
|
c.r.Start(start)
|
||||||
|
q0, on0 := start, true
|
||||||
|
for _, p := range others {
|
||||||
|
q := fixed.Point26_6{
|
||||||
|
X: dx + p.X,
|
||||||
|
Y: dy - p.Y,
|
||||||
|
}
|
||||||
|
on := p.Flags&0x01 != 0
|
||||||
|
if on {
|
||||||
|
if on0 {
|
||||||
|
c.r.Add1(q)
|
||||||
|
} else {
|
||||||
|
c.r.Add2(q0, q)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if on0 {
|
||||||
|
// No-op.
|
||||||
|
} else {
|
||||||
|
mid := fixed.Point26_6{
|
||||||
|
X: (q0.X + q.X) / 2,
|
||||||
|
Y: (q0.Y + q.Y) / 2,
|
||||||
|
}
|
||||||
|
c.r.Add2(q0, mid)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
q0, on0 = q, on
|
||||||
|
}
|
||||||
|
// Close the curve.
|
||||||
|
if on0 {
|
||||||
|
c.r.Add1(start)
|
||||||
|
} else {
|
||||||
|
c.r.Add2(q0, start)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// rasterize returns the advance width, glyph mask and integer-pixel offset
|
||||||
|
// to render the given glyph at the given sub-pixel offsets.
|
||||||
|
// The 26.6 fixed point arguments fx and fy must be in the range [0, 1).
|
||||||
|
func (c *Context) rasterize(glyph truetype.Index, fx, fy fixed.Int26_6) (
|
||||||
|
fixed.Int26_6, *image.Alpha, image.Point, error) {
|
||||||
|
|
||||||
|
if err := c.glyphBuf.Load(c.f, c.scale, glyph, c.hinting); err != nil {
|
||||||
|
return 0, nil, image.Point{}, err
|
||||||
|
}
|
||||||
|
// Calculate the integer-pixel bounds for the glyph.
|
||||||
|
xmin := int(fx+c.glyphBuf.Bounds.Min.X) >> 6
|
||||||
|
ymin := int(fy-c.glyphBuf.Bounds.Max.Y) >> 6
|
||||||
|
xmax := int(fx+c.glyphBuf.Bounds.Max.X+0x3f) >> 6
|
||||||
|
ymax := int(fy-c.glyphBuf.Bounds.Min.Y+0x3f) >> 6
|
||||||
|
if xmin > xmax || ymin > ymax {
|
||||||
|
return 0, nil, image.Point{}, errors.New("freetype: negative sized glyph")
|
||||||
|
}
|
||||||
|
// A TrueType's glyph's nodes can have negative co-ordinates, but the
|
||||||
|
// rasterizer clips anything left of x=0 or above y=0. xmin and ymin are
|
||||||
|
// the pixel offsets, based on the font's FUnit metrics, that let a
|
||||||
|
// negative co-ordinate in TrueType space be non-negative in rasterizer
|
||||||
|
// space. xmin and ymin are typically <= 0.
|
||||||
|
fx -= fixed.Int26_6(xmin << 6)
|
||||||
|
fy -= fixed.Int26_6(ymin << 6)
|
||||||
|
// Rasterize the glyph's vectors.
|
||||||
|
c.r.Clear()
|
||||||
|
e0 := 0
|
||||||
|
for _, e1 := range c.glyphBuf.Ends {
|
||||||
|
c.drawContour(c.glyphBuf.Points[e0:e1], fx, fy)
|
||||||
|
e0 = e1
|
||||||
|
}
|
||||||
|
a := image.NewAlpha(image.Rect(0, 0, xmax-xmin, ymax-ymin))
|
||||||
|
c.r.Rasterize(raster.NewAlphaSrcPainter(a))
|
||||||
|
return c.glyphBuf.AdvanceWidth, a, image.Point{xmin, ymin}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// glyph returns the advance width, glyph mask and integer-pixel offset to
|
||||||
|
// render the given glyph at the given sub-pixel point. It is a cache for the
|
||||||
|
// rasterize method. Unlike rasterize, p's co-ordinates do not have to be in
|
||||||
|
// the range [0, 1).
|
||||||
|
func (c *Context) glyph(glyph truetype.Index, p fixed.Point26_6) (
|
||||||
|
fixed.Int26_6, *image.Alpha, image.Point, error) {
|
||||||
|
|
||||||
|
// Split p.X and p.Y into their integer and fractional parts.
|
||||||
|
ix, fx := int(p.X>>6), p.X&0x3f
|
||||||
|
iy, fy := int(p.Y>>6), p.Y&0x3f
|
||||||
|
// Calculate the index t into the cache array.
|
||||||
|
tg := int(glyph) % nGlyphs
|
||||||
|
tx := int(fx) / (64 / nXFractions)
|
||||||
|
ty := int(fy) / (64 / nYFractions)
|
||||||
|
t := ((tg*nXFractions)+tx)*nYFractions + ty
|
||||||
|
// Check for a cache hit.
|
||||||
|
if e := c.cache[t]; e.valid && e.glyph == glyph {
|
||||||
|
return e.advanceWidth, e.mask, e.offset.Add(image.Point{ix, iy}), nil
|
||||||
|
}
|
||||||
|
// Rasterize the glyph and put the result into the cache.
|
||||||
|
advanceWidth, mask, offset, err := c.rasterize(glyph, fx, fy)
|
||||||
|
if err != nil {
|
||||||
|
return 0, nil, image.Point{}, err
|
||||||
|
}
|
||||||
|
c.cache[t] = cacheEntry{true, glyph, advanceWidth, mask, offset}
|
||||||
|
return advanceWidth, mask, offset.Add(image.Point{ix, iy}), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DrawString draws s at p and returns p advanced by the text extent. The text
|
||||||
|
// is placed so that the left edge of the em square of the first character of s
|
||||||
|
// and the baseline intersect at p. The majority of the affected pixels will be
|
||||||
|
// above and to the right of the point, but some may be below or to the left.
|
||||||
|
// For example, drawing a string that starts with a 'J' in an italic font may
|
||||||
|
// affect pixels below and left of the point.
|
||||||
|
//
|
||||||
|
// p is a fixed.Point26_6 and can therefore represent sub-pixel positions.
|
||||||
|
func (c *Context) DrawString(s string, p fixed.Point26_6) (fixed.Point26_6, error) {
|
||||||
|
if c.f == nil {
|
||||||
|
return fixed.Point26_6{}, errors.New("freetype: DrawText called with a nil font")
|
||||||
|
}
|
||||||
|
prev, hasPrev := truetype.Index(0), false
|
||||||
|
for _, rune := range s {
|
||||||
|
index := c.f.Index(rune)
|
||||||
|
if hasPrev {
|
||||||
|
kern := c.f.Kern(c.scale, prev, index)
|
||||||
|
if c.hinting != font.HintingNone {
|
||||||
|
kern = (kern + 32) &^ 63
|
||||||
|
}
|
||||||
|
p.X += kern
|
||||||
|
}
|
||||||
|
advanceWidth, mask, offset, err := c.glyph(index, p)
|
||||||
|
if err != nil {
|
||||||
|
return fixed.Point26_6{}, err
|
||||||
|
}
|
||||||
|
p.X += advanceWidth
|
||||||
|
glyphRect := mask.Bounds().Add(offset)
|
||||||
|
dr := c.clip.Intersect(glyphRect)
|
||||||
|
if !dr.Empty() {
|
||||||
|
mp := image.Point{0, dr.Min.Y - glyphRect.Min.Y}
|
||||||
|
draw.DrawMask(c.dst, dr, c.src, image.ZP, mask, mp, draw.Over)
|
||||||
|
}
|
||||||
|
prev, hasPrev = index, true
|
||||||
|
}
|
||||||
|
return p, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// recalc recalculates scale and bounds values from the font size, screen
|
||||||
|
// resolution and font metrics, and invalidates the glyph cache.
|
||||||
|
func (c *Context) recalc() {
|
||||||
|
c.scale = fixed.Int26_6(c.fontSize * c.dpi * (64.0 / 72.0))
|
||||||
|
if c.f == nil {
|
||||||
|
c.r.SetBounds(0, 0)
|
||||||
|
} else {
|
||||||
|
// Set the rasterizer's bounds to be big enough to handle the largest glyph.
|
||||||
|
b := c.f.Bounds(c.scale)
|
||||||
|
xmin := +int(b.Min.X) >> 6
|
||||||
|
ymin := -int(b.Max.Y) >> 6
|
||||||
|
xmax := +int(b.Max.X+63) >> 6
|
||||||
|
ymax := -int(b.Min.Y-63) >> 6
|
||||||
|
c.r.SetBounds(xmax-xmin, ymax-ymin)
|
||||||
|
}
|
||||||
|
for i := range c.cache {
|
||||||
|
c.cache[i] = cacheEntry{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetDPI sets the screen resolution in dots per inch.
|
||||||
|
func (c *Context) SetDPI(dpi float64) {
|
||||||
|
if c.dpi == dpi {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.dpi = dpi
|
||||||
|
c.recalc()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetFont sets the font used to draw text.
|
||||||
|
func (c *Context) SetFont(f *truetype.Font) {
|
||||||
|
if c.f == f {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.f = f
|
||||||
|
c.recalc()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetFontSize sets the font size in points (as in "a 12 point font").
|
||||||
|
func (c *Context) SetFontSize(fontSize float64) {
|
||||||
|
if c.fontSize == fontSize {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.fontSize = fontSize
|
||||||
|
c.recalc()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetHinting sets the hinting policy.
|
||||||
|
func (c *Context) SetHinting(hinting font.Hinting) {
|
||||||
|
c.hinting = hinting
|
||||||
|
for i := range c.cache {
|
||||||
|
c.cache[i] = cacheEntry{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetDst sets the destination image for draw operations.
|
||||||
|
func (c *Context) SetDst(dst draw.Image) {
|
||||||
|
c.dst = dst
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetSrc sets the source image for draw operations. This is typically an
|
||||||
|
// image.Uniform.
|
||||||
|
func (c *Context) SetSrc(src image.Image) {
|
||||||
|
c.src = src
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetClip sets the clip rectangle for drawing.
|
||||||
|
func (c *Context) SetClip(clip image.Rectangle) {
|
||||||
|
c.clip = clip
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO(nigeltao): implement Context.SetGamma.
|
||||||
|
|
||||||
|
// NewContext creates a new Context.
|
||||||
|
func NewContext() *Context {
|
||||||
|
return &Context{
|
||||||
|
r: raster.NewRasterizer(0, 0),
|
||||||
|
fontSize: 12,
|
||||||
|
dpi: 72,
|
||||||
|
scale: 12 << 6,
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,245 @@
|
||||||
|
// Copyright 2010 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package raster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"math"
|
||||||
|
|
||||||
|
"golang.org/x/image/math/fixed"
|
||||||
|
)
|
||||||
|
|
||||||
|
// maxAbs returns the maximum of abs(a) and abs(b).
|
||||||
|
func maxAbs(a, b fixed.Int26_6) fixed.Int26_6 {
|
||||||
|
if a < 0 {
|
||||||
|
a = -a
|
||||||
|
}
|
||||||
|
if b < 0 {
|
||||||
|
b = -b
|
||||||
|
}
|
||||||
|
if a < b {
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
return a
|
||||||
|
}
|
||||||
|
|
||||||
|
// pNeg returns the vector -p, or equivalently p rotated by 180 degrees.
|
||||||
|
func pNeg(p fixed.Point26_6) fixed.Point26_6 {
|
||||||
|
return fixed.Point26_6{-p.X, -p.Y}
|
||||||
|
}
|
||||||
|
|
||||||
|
// pDot returns the dot product p·q.
|
||||||
|
func pDot(p fixed.Point26_6, q fixed.Point26_6) fixed.Int52_12 {
|
||||||
|
px, py := int64(p.X), int64(p.Y)
|
||||||
|
qx, qy := int64(q.X), int64(q.Y)
|
||||||
|
return fixed.Int52_12(px*qx + py*qy)
|
||||||
|
}
|
||||||
|
|
||||||
|
// pLen returns the length of the vector p.
|
||||||
|
func pLen(p fixed.Point26_6) fixed.Int26_6 {
|
||||||
|
// TODO(nigeltao): use fixed point math.
|
||||||
|
x := float64(p.X)
|
||||||
|
y := float64(p.Y)
|
||||||
|
return fixed.Int26_6(math.Sqrt(x*x + y*y))
|
||||||
|
}
|
||||||
|
|
||||||
|
// pNorm returns the vector p normalized to the given length, or zero if p is
|
||||||
|
// degenerate.
|
||||||
|
func pNorm(p fixed.Point26_6, length fixed.Int26_6) fixed.Point26_6 {
|
||||||
|
d := pLen(p)
|
||||||
|
if d == 0 {
|
||||||
|
return fixed.Point26_6{}
|
||||||
|
}
|
||||||
|
s, t := int64(length), int64(d)
|
||||||
|
x := int64(p.X) * s / t
|
||||||
|
y := int64(p.Y) * s / t
|
||||||
|
return fixed.Point26_6{fixed.Int26_6(x), fixed.Int26_6(y)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// pRot45CW returns the vector p rotated clockwise by 45 degrees.
|
||||||
|
//
|
||||||
|
// Note that the Y-axis grows downwards, so {1, 0}.Rot45CW is {1/√2, 1/√2}.
|
||||||
|
func pRot45CW(p fixed.Point26_6) fixed.Point26_6 {
|
||||||
|
// 181/256 is approximately 1/√2, or sin(π/4).
|
||||||
|
px, py := int64(p.X), int64(p.Y)
|
||||||
|
qx := (+px - py) * 181 / 256
|
||||||
|
qy := (+px + py) * 181 / 256
|
||||||
|
return fixed.Point26_6{fixed.Int26_6(qx), fixed.Int26_6(qy)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// pRot90CW returns the vector p rotated clockwise by 90 degrees.
|
||||||
|
//
|
||||||
|
// Note that the Y-axis grows downwards, so {1, 0}.Rot90CW is {0, 1}.
|
||||||
|
func pRot90CW(p fixed.Point26_6) fixed.Point26_6 {
|
||||||
|
return fixed.Point26_6{-p.Y, p.X}
|
||||||
|
}
|
||||||
|
|
||||||
|
// pRot135CW returns the vector p rotated clockwise by 135 degrees.
|
||||||
|
//
|
||||||
|
// Note that the Y-axis grows downwards, so {1, 0}.Rot135CW is {-1/√2, 1/√2}.
|
||||||
|
func pRot135CW(p fixed.Point26_6) fixed.Point26_6 {
|
||||||
|
// 181/256 is approximately 1/√2, or sin(π/4).
|
||||||
|
px, py := int64(p.X), int64(p.Y)
|
||||||
|
qx := (-px - py) * 181 / 256
|
||||||
|
qy := (+px - py) * 181 / 256
|
||||||
|
return fixed.Point26_6{fixed.Int26_6(qx), fixed.Int26_6(qy)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// pRot45CCW returns the vector p rotated counter-clockwise by 45 degrees.
|
||||||
|
//
|
||||||
|
// Note that the Y-axis grows downwards, so {1, 0}.Rot45CCW is {1/√2, -1/√2}.
|
||||||
|
func pRot45CCW(p fixed.Point26_6) fixed.Point26_6 {
|
||||||
|
// 181/256 is approximately 1/√2, or sin(π/4).
|
||||||
|
px, py := int64(p.X), int64(p.Y)
|
||||||
|
qx := (+px + py) * 181 / 256
|
||||||
|
qy := (-px + py) * 181 / 256
|
||||||
|
return fixed.Point26_6{fixed.Int26_6(qx), fixed.Int26_6(qy)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// pRot90CCW returns the vector p rotated counter-clockwise by 90 degrees.
|
||||||
|
//
|
||||||
|
// Note that the Y-axis grows downwards, so {1, 0}.Rot90CCW is {0, -1}.
|
||||||
|
func pRot90CCW(p fixed.Point26_6) fixed.Point26_6 {
|
||||||
|
return fixed.Point26_6{p.Y, -p.X}
|
||||||
|
}
|
||||||
|
|
||||||
|
// pRot135CCW returns the vector p rotated counter-clockwise by 135 degrees.
|
||||||
|
//
|
||||||
|
// Note that the Y-axis grows downwards, so {1, 0}.Rot135CCW is {-1/√2, -1/√2}.
|
||||||
|
func pRot135CCW(p fixed.Point26_6) fixed.Point26_6 {
|
||||||
|
// 181/256 is approximately 1/√2, or sin(π/4).
|
||||||
|
px, py := int64(p.X), int64(p.Y)
|
||||||
|
qx := (-px + py) * 181 / 256
|
||||||
|
qy := (-px - py) * 181 / 256
|
||||||
|
return fixed.Point26_6{fixed.Int26_6(qx), fixed.Int26_6(qy)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// An Adder accumulates points on a curve.
|
||||||
|
type Adder interface {
|
||||||
|
// Start starts a new curve at the given point.
|
||||||
|
Start(a fixed.Point26_6)
|
||||||
|
// Add1 adds a linear segment to the current curve.
|
||||||
|
Add1(b fixed.Point26_6)
|
||||||
|
// Add2 adds a quadratic segment to the current curve.
|
||||||
|
Add2(b, c fixed.Point26_6)
|
||||||
|
// Add3 adds a cubic segment to the current curve.
|
||||||
|
Add3(b, c, d fixed.Point26_6)
|
||||||
|
}
|
||||||
|
|
||||||
|
// A Path is a sequence of curves, and a curve is a start point followed by a
|
||||||
|
// sequence of linear, quadratic or cubic segments.
|
||||||
|
type Path []fixed.Int26_6
|
||||||
|
|
||||||
|
// String returns a human-readable representation of a Path.
|
||||||
|
func (p Path) String() string {
|
||||||
|
s := ""
|
||||||
|
for i := 0; i < len(p); {
|
||||||
|
if i != 0 {
|
||||||
|
s += " "
|
||||||
|
}
|
||||||
|
switch p[i] {
|
||||||
|
case 0:
|
||||||
|
s += "S0" + fmt.Sprint([]fixed.Int26_6(p[i+1:i+3]))
|
||||||
|
i += 4
|
||||||
|
case 1:
|
||||||
|
s += "A1" + fmt.Sprint([]fixed.Int26_6(p[i+1:i+3]))
|
||||||
|
i += 4
|
||||||
|
case 2:
|
||||||
|
s += "A2" + fmt.Sprint([]fixed.Int26_6(p[i+1:i+5]))
|
||||||
|
i += 6
|
||||||
|
case 3:
|
||||||
|
s += "A3" + fmt.Sprint([]fixed.Int26_6(p[i+1:i+7]))
|
||||||
|
i += 8
|
||||||
|
default:
|
||||||
|
panic("freetype/raster: bad path")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
// Clear cancels any previous calls to p.Start or p.AddXxx.
|
||||||
|
func (p *Path) Clear() {
|
||||||
|
*p = (*p)[:0]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start starts a new curve at the given point.
|
||||||
|
func (p *Path) Start(a fixed.Point26_6) {
|
||||||
|
*p = append(*p, 0, a.X, a.Y, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add1 adds a linear segment to the current curve.
|
||||||
|
func (p *Path) Add1(b fixed.Point26_6) {
|
||||||
|
*p = append(*p, 1, b.X, b.Y, 1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add2 adds a quadratic segment to the current curve.
|
||||||
|
func (p *Path) Add2(b, c fixed.Point26_6) {
|
||||||
|
*p = append(*p, 2, b.X, b.Y, c.X, c.Y, 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add3 adds a cubic segment to the current curve.
|
||||||
|
func (p *Path) Add3(b, c, d fixed.Point26_6) {
|
||||||
|
*p = append(*p, 3, b.X, b.Y, c.X, c.Y, d.X, d.Y, 3)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddPath adds the Path q to p.
|
||||||
|
func (p *Path) AddPath(q Path) {
|
||||||
|
*p = append(*p, q...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddStroke adds a stroked Path.
|
||||||
|
func (p *Path) AddStroke(q Path, width fixed.Int26_6, cr Capper, jr Joiner) {
|
||||||
|
Stroke(p, q, width, cr, jr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// firstPoint returns the first point in a non-empty Path.
|
||||||
|
func (p Path) firstPoint() fixed.Point26_6 {
|
||||||
|
return fixed.Point26_6{p[1], p[2]}
|
||||||
|
}
|
||||||
|
|
||||||
|
// lastPoint returns the last point in a non-empty Path.
|
||||||
|
func (p Path) lastPoint() fixed.Point26_6 {
|
||||||
|
return fixed.Point26_6{p[len(p)-3], p[len(p)-2]}
|
||||||
|
}
|
||||||
|
|
||||||
|
// addPathReversed adds q reversed to p.
|
||||||
|
// For example, if q consists of a linear segment from A to B followed by a
|
||||||
|
// quadratic segment from B to C to D, then the values of q looks like:
|
||||||
|
// index: 01234567890123
|
||||||
|
// value: 0AA01BB12CCDD2
|
||||||
|
// So, when adding q backwards to p, we want to Add2(C, B) followed by Add1(A).
|
||||||
|
func addPathReversed(p Adder, q Path) {
|
||||||
|
if len(q) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
i := len(q) - 1
|
||||||
|
for {
|
||||||
|
switch q[i] {
|
||||||
|
case 0:
|
||||||
|
return
|
||||||
|
case 1:
|
||||||
|
i -= 4
|
||||||
|
p.Add1(
|
||||||
|
fixed.Point26_6{q[i-2], q[i-1]},
|
||||||
|
)
|
||||||
|
case 2:
|
||||||
|
i -= 6
|
||||||
|
p.Add2(
|
||||||
|
fixed.Point26_6{q[i+2], q[i+3]},
|
||||||
|
fixed.Point26_6{q[i-2], q[i-1]},
|
||||||
|
)
|
||||||
|
case 3:
|
||||||
|
i -= 8
|
||||||
|
p.Add3(
|
||||||
|
fixed.Point26_6{q[i+4], q[i+5]},
|
||||||
|
fixed.Point26_6{q[i+2], q[i+3]},
|
||||||
|
fixed.Point26_6{q[i-2], q[i-1]},
|
||||||
|
)
|
||||||
|
default:
|
||||||
|
panic("freetype/raster: bad path")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,287 @@
|
||||||
|
// Copyright 2010 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package raster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"image"
|
||||||
|
"image/color"
|
||||||
|
"image/draw"
|
||||||
|
"math"
|
||||||
|
)
|
||||||
|
|
||||||
|
// A Span is a horizontal segment of pixels with constant alpha. X0 is an
|
||||||
|
// inclusive bound and X1 is exclusive, the same as for slices. A fully opaque
|
||||||
|
// Span has Alpha == 0xffff.
|
||||||
|
type Span struct {
|
||||||
|
Y, X0, X1 int
|
||||||
|
Alpha uint32
|
||||||
|
}
|
||||||
|
|
||||||
|
// A Painter knows how to paint a batch of Spans. Rasterization may involve
|
||||||
|
// Painting multiple batches, and done will be true for the final batch. The
|
||||||
|
// Spans' Y values are monotonically increasing during a rasterization. Paint
|
||||||
|
// may use all of ss as scratch space during the call.
|
||||||
|
type Painter interface {
|
||||||
|
Paint(ss []Span, done bool)
|
||||||
|
}
|
||||||
|
|
||||||
|
// The PainterFunc type adapts an ordinary function to the Painter interface.
|
||||||
|
type PainterFunc func(ss []Span, done bool)
|
||||||
|
|
||||||
|
// Paint just delegates the call to f.
|
||||||
|
func (f PainterFunc) Paint(ss []Span, done bool) { f(ss, done) }
|
||||||
|
|
||||||
|
// An AlphaOverPainter is a Painter that paints Spans onto a *image.Alpha using
|
||||||
|
// the Over Porter-Duff composition operator.
|
||||||
|
type AlphaOverPainter struct {
|
||||||
|
Image *image.Alpha
|
||||||
|
}
|
||||||
|
|
||||||
|
// Paint satisfies the Painter interface.
|
||||||
|
func (r AlphaOverPainter) Paint(ss []Span, done bool) {
|
||||||
|
b := r.Image.Bounds()
|
||||||
|
for _, s := range ss {
|
||||||
|
if s.Y < b.Min.Y {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if s.Y >= b.Max.Y {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.X0 < b.Min.X {
|
||||||
|
s.X0 = b.Min.X
|
||||||
|
}
|
||||||
|
if s.X1 > b.Max.X {
|
||||||
|
s.X1 = b.Max.X
|
||||||
|
}
|
||||||
|
if s.X0 >= s.X1 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
base := (s.Y-r.Image.Rect.Min.Y)*r.Image.Stride - r.Image.Rect.Min.X
|
||||||
|
p := r.Image.Pix[base+s.X0 : base+s.X1]
|
||||||
|
a := int(s.Alpha >> 8)
|
||||||
|
for i, c := range p {
|
||||||
|
v := int(c)
|
||||||
|
p[i] = uint8((v*255 + (255-v)*a) / 255)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewAlphaOverPainter creates a new AlphaOverPainter for the given image.
|
||||||
|
func NewAlphaOverPainter(m *image.Alpha) AlphaOverPainter {
|
||||||
|
return AlphaOverPainter{m}
|
||||||
|
}
|
||||||
|
|
||||||
|
// An AlphaSrcPainter is a Painter that paints Spans onto a *image.Alpha using
|
||||||
|
// the Src Porter-Duff composition operator.
|
||||||
|
type AlphaSrcPainter struct {
|
||||||
|
Image *image.Alpha
|
||||||
|
}
|
||||||
|
|
||||||
|
// Paint satisfies the Painter interface.
|
||||||
|
func (r AlphaSrcPainter) Paint(ss []Span, done bool) {
|
||||||
|
b := r.Image.Bounds()
|
||||||
|
for _, s := range ss {
|
||||||
|
if s.Y < b.Min.Y {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if s.Y >= b.Max.Y {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.X0 < b.Min.X {
|
||||||
|
s.X0 = b.Min.X
|
||||||
|
}
|
||||||
|
if s.X1 > b.Max.X {
|
||||||
|
s.X1 = b.Max.X
|
||||||
|
}
|
||||||
|
if s.X0 >= s.X1 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
base := (s.Y-r.Image.Rect.Min.Y)*r.Image.Stride - r.Image.Rect.Min.X
|
||||||
|
p := r.Image.Pix[base+s.X0 : base+s.X1]
|
||||||
|
color := uint8(s.Alpha >> 8)
|
||||||
|
for i := range p {
|
||||||
|
p[i] = color
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewAlphaSrcPainter creates a new AlphaSrcPainter for the given image.
|
||||||
|
func NewAlphaSrcPainter(m *image.Alpha) AlphaSrcPainter {
|
||||||
|
return AlphaSrcPainter{m}
|
||||||
|
}
|
||||||
|
|
||||||
|
// An RGBAPainter is a Painter that paints Spans onto a *image.RGBA.
|
||||||
|
type RGBAPainter struct {
|
||||||
|
// Image is the image to compose onto.
|
||||||
|
Image *image.RGBA
|
||||||
|
// Op is the Porter-Duff composition operator.
|
||||||
|
Op draw.Op
|
||||||
|
// cr, cg, cb and ca are the 16-bit color to paint the spans.
|
||||||
|
cr, cg, cb, ca uint32
|
||||||
|
}
|
||||||
|
|
||||||
|
// Paint satisfies the Painter interface.
|
||||||
|
func (r *RGBAPainter) Paint(ss []Span, done bool) {
|
||||||
|
b := r.Image.Bounds()
|
||||||
|
for _, s := range ss {
|
||||||
|
if s.Y < b.Min.Y {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if s.Y >= b.Max.Y {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.X0 < b.Min.X {
|
||||||
|
s.X0 = b.Min.X
|
||||||
|
}
|
||||||
|
if s.X1 > b.Max.X {
|
||||||
|
s.X1 = b.Max.X
|
||||||
|
}
|
||||||
|
if s.X0 >= s.X1 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// This code mimics drawGlyphOver in $GOROOT/src/image/draw/draw.go.
|
||||||
|
ma := s.Alpha
|
||||||
|
const m = 1<<16 - 1
|
||||||
|
i0 := (s.Y-r.Image.Rect.Min.Y)*r.Image.Stride + (s.X0-r.Image.Rect.Min.X)*4
|
||||||
|
i1 := i0 + (s.X1-s.X0)*4
|
||||||
|
if r.Op == draw.Over {
|
||||||
|
for i := i0; i < i1; i += 4 {
|
||||||
|
dr := uint32(r.Image.Pix[i+0])
|
||||||
|
dg := uint32(r.Image.Pix[i+1])
|
||||||
|
db := uint32(r.Image.Pix[i+2])
|
||||||
|
da := uint32(r.Image.Pix[i+3])
|
||||||
|
a := (m - (r.ca * ma / m)) * 0x101
|
||||||
|
r.Image.Pix[i+0] = uint8((dr*a + r.cr*ma) / m >> 8)
|
||||||
|
r.Image.Pix[i+1] = uint8((dg*a + r.cg*ma) / m >> 8)
|
||||||
|
r.Image.Pix[i+2] = uint8((db*a + r.cb*ma) / m >> 8)
|
||||||
|
r.Image.Pix[i+3] = uint8((da*a + r.ca*ma) / m >> 8)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
for i := i0; i < i1; i += 4 {
|
||||||
|
r.Image.Pix[i+0] = uint8(r.cr * ma / m >> 8)
|
||||||
|
r.Image.Pix[i+1] = uint8(r.cg * ma / m >> 8)
|
||||||
|
r.Image.Pix[i+2] = uint8(r.cb * ma / m >> 8)
|
||||||
|
r.Image.Pix[i+3] = uint8(r.ca * ma / m >> 8)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetColor sets the color to paint the spans.
|
||||||
|
func (r *RGBAPainter) SetColor(c color.Color) {
|
||||||
|
r.cr, r.cg, r.cb, r.ca = c.RGBA()
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewRGBAPainter creates a new RGBAPainter for the given image.
|
||||||
|
func NewRGBAPainter(m *image.RGBA) *RGBAPainter {
|
||||||
|
return &RGBAPainter{Image: m}
|
||||||
|
}
|
||||||
|
|
||||||
|
// A MonochromePainter wraps another Painter, quantizing each Span's alpha to
|
||||||
|
// be either fully opaque or fully transparent.
|
||||||
|
type MonochromePainter struct {
|
||||||
|
Painter Painter
|
||||||
|
y, x0, x1 int
|
||||||
|
}
|
||||||
|
|
||||||
|
// Paint delegates to the wrapped Painter after quantizing each Span's alpha
|
||||||
|
// value and merging adjacent fully opaque Spans.
|
||||||
|
func (m *MonochromePainter) Paint(ss []Span, done bool) {
|
||||||
|
// We compact the ss slice, discarding any Spans whose alpha quantizes to zero.
|
||||||
|
j := 0
|
||||||
|
for _, s := range ss {
|
||||||
|
if s.Alpha >= 0x8000 {
|
||||||
|
if m.y == s.Y && m.x1 == s.X0 {
|
||||||
|
m.x1 = s.X1
|
||||||
|
} else {
|
||||||
|
ss[j] = Span{m.y, m.x0, m.x1, 1<<16 - 1}
|
||||||
|
j++
|
||||||
|
m.y, m.x0, m.x1 = s.Y, s.X0, s.X1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if done {
|
||||||
|
// Flush the accumulated Span.
|
||||||
|
finalSpan := Span{m.y, m.x0, m.x1, 1<<16 - 1}
|
||||||
|
if j < len(ss) {
|
||||||
|
ss[j] = finalSpan
|
||||||
|
j++
|
||||||
|
m.Painter.Paint(ss[:j], true)
|
||||||
|
} else if j == len(ss) {
|
||||||
|
m.Painter.Paint(ss, false)
|
||||||
|
if cap(ss) > 0 {
|
||||||
|
ss = ss[:1]
|
||||||
|
} else {
|
||||||
|
ss = make([]Span, 1)
|
||||||
|
}
|
||||||
|
ss[0] = finalSpan
|
||||||
|
m.Painter.Paint(ss, true)
|
||||||
|
} else {
|
||||||
|
panic("unreachable")
|
||||||
|
}
|
||||||
|
// Reset the accumulator, so that this Painter can be re-used.
|
||||||
|
m.y, m.x0, m.x1 = 0, 0, 0
|
||||||
|
} else {
|
||||||
|
m.Painter.Paint(ss[:j], false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewMonochromePainter creates a new MonochromePainter that wraps the given
|
||||||
|
// Painter.
|
||||||
|
func NewMonochromePainter(p Painter) *MonochromePainter {
|
||||||
|
return &MonochromePainter{Painter: p}
|
||||||
|
}
|
||||||
|
|
||||||
|
// A GammaCorrectionPainter wraps another Painter, performing gamma-correction
|
||||||
|
// on each Span's alpha value.
|
||||||
|
type GammaCorrectionPainter struct {
|
||||||
|
// Painter is the wrapped Painter.
|
||||||
|
Painter Painter
|
||||||
|
// a is the precomputed alpha values for linear interpolation, with fully
|
||||||
|
// opaque == 0xffff.
|
||||||
|
a [256]uint16
|
||||||
|
// gammaIsOne is whether gamma correction is a no-op.
|
||||||
|
gammaIsOne bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// Paint delegates to the wrapped Painter after performing gamma-correction on
|
||||||
|
// each Span.
|
||||||
|
func (g *GammaCorrectionPainter) Paint(ss []Span, done bool) {
|
||||||
|
if !g.gammaIsOne {
|
||||||
|
const n = 0x101
|
||||||
|
for i, s := range ss {
|
||||||
|
if s.Alpha == 0 || s.Alpha == 0xffff {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
p, q := s.Alpha/n, s.Alpha%n
|
||||||
|
// The resultant alpha is a linear interpolation of g.a[p] and g.a[p+1].
|
||||||
|
a := uint32(g.a[p])*(n-q) + uint32(g.a[p+1])*q
|
||||||
|
ss[i].Alpha = (a + n/2) / n
|
||||||
|
}
|
||||||
|
}
|
||||||
|
g.Painter.Paint(ss, done)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetGamma sets the gamma value.
|
||||||
|
func (g *GammaCorrectionPainter) SetGamma(gamma float64) {
|
||||||
|
g.gammaIsOne = gamma == 1
|
||||||
|
if g.gammaIsOne {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for i := 0; i < 256; i++ {
|
||||||
|
a := float64(i) / 0xff
|
||||||
|
a = math.Pow(a, gamma)
|
||||||
|
g.a[i] = uint16(0xffff * a)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewGammaCorrectionPainter creates a new GammaCorrectionPainter that wraps
|
||||||
|
// the given Painter.
|
||||||
|
func NewGammaCorrectionPainter(p Painter, gamma float64) *GammaCorrectionPainter {
|
||||||
|
g := &GammaCorrectionPainter{Painter: p}
|
||||||
|
g.SetGamma(gamma)
|
||||||
|
return g
|
||||||
|
}
|
|
@ -0,0 +1,601 @@
|
||||||
|
// Copyright 2010 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
// Package raster provides an anti-aliasing 2-D rasterizer.
|
||||||
|
//
|
||||||
|
// It is part of the larger Freetype suite of font-related packages, but the
|
||||||
|
// raster package is not specific to font rasterization, and can be used
|
||||||
|
// standalone without any other Freetype package.
|
||||||
|
//
|
||||||
|
// Rasterization is done by the same area/coverage accumulation algorithm as
|
||||||
|
// the Freetype "smooth" module, and the Anti-Grain Geometry library. A
|
||||||
|
// description of the area/coverage algorithm is at
|
||||||
|
// http://projects.tuxee.net/cl-vectors/section-the-cl-aa-algorithm
|
||||||
|
package raster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"golang.org/x/image/math/fixed"
|
||||||
|
)
|
||||||
|
|
||||||
|
// A cell is part of a linked list (for a given yi co-ordinate) of accumulated
|
||||||
|
// area/coverage for the pixel at (xi, yi).
|
||||||
|
type cell struct {
|
||||||
|
xi int
|
||||||
|
area, cover int
|
||||||
|
next int
|
||||||
|
}
|
||||||
|
|
||||||
|
type Rasterizer struct {
|
||||||
|
// If false, the default behavior is to use the even-odd winding fill
|
||||||
|
// rule during Rasterize.
|
||||||
|
UseNonZeroWinding bool
|
||||||
|
// An offset (in pixels) to the painted spans.
|
||||||
|
Dx, Dy int
|
||||||
|
|
||||||
|
// The width of the Rasterizer. The height is implicit in len(cellIndex).
|
||||||
|
width int
|
||||||
|
// splitScaleN is the scaling factor used to determine how many times
|
||||||
|
// to decompose a quadratic or cubic segment into a linear approximation.
|
||||||
|
splitScale2, splitScale3 int
|
||||||
|
|
||||||
|
// The current pen position.
|
||||||
|
a fixed.Point26_6
|
||||||
|
// The current cell and its area/coverage being accumulated.
|
||||||
|
xi, yi int
|
||||||
|
area, cover int
|
||||||
|
|
||||||
|
// Saved cells.
|
||||||
|
cell []cell
|
||||||
|
// Linked list of cells, one per row.
|
||||||
|
cellIndex []int
|
||||||
|
// Buffers.
|
||||||
|
cellBuf [256]cell
|
||||||
|
cellIndexBuf [64]int
|
||||||
|
spanBuf [64]Span
|
||||||
|
}
|
||||||
|
|
||||||
|
// findCell returns the index in r.cell for the cell corresponding to
|
||||||
|
// (r.xi, r.yi). The cell is created if necessary.
|
||||||
|
func (r *Rasterizer) findCell() int {
|
||||||
|
if r.yi < 0 || r.yi >= len(r.cellIndex) {
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
xi := r.xi
|
||||||
|
if xi < 0 {
|
||||||
|
xi = -1
|
||||||
|
} else if xi > r.width {
|
||||||
|
xi = r.width
|
||||||
|
}
|
||||||
|
i, prev := r.cellIndex[r.yi], -1
|
||||||
|
for i != -1 && r.cell[i].xi <= xi {
|
||||||
|
if r.cell[i].xi == xi {
|
||||||
|
return i
|
||||||
|
}
|
||||||
|
i, prev = r.cell[i].next, i
|
||||||
|
}
|
||||||
|
c := len(r.cell)
|
||||||
|
if c == cap(r.cell) {
|
||||||
|
buf := make([]cell, c, 4*c)
|
||||||
|
copy(buf, r.cell)
|
||||||
|
r.cell = buf[0 : c+1]
|
||||||
|
} else {
|
||||||
|
r.cell = r.cell[0 : c+1]
|
||||||
|
}
|
||||||
|
r.cell[c] = cell{xi, 0, 0, i}
|
||||||
|
if prev == -1 {
|
||||||
|
r.cellIndex[r.yi] = c
|
||||||
|
} else {
|
||||||
|
r.cell[prev].next = c
|
||||||
|
}
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// saveCell saves any accumulated r.area/r.cover for (r.xi, r.yi).
|
||||||
|
func (r *Rasterizer) saveCell() {
|
||||||
|
if r.area != 0 || r.cover != 0 {
|
||||||
|
i := r.findCell()
|
||||||
|
if i != -1 {
|
||||||
|
r.cell[i].area += r.area
|
||||||
|
r.cell[i].cover += r.cover
|
||||||
|
}
|
||||||
|
r.area = 0
|
||||||
|
r.cover = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// setCell sets the (xi, yi) cell that r is accumulating area/coverage for.
|
||||||
|
func (r *Rasterizer) setCell(xi, yi int) {
|
||||||
|
if r.xi != xi || r.yi != yi {
|
||||||
|
r.saveCell()
|
||||||
|
r.xi, r.yi = xi, yi
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// scan accumulates area/coverage for the yi'th scanline, going from
|
||||||
|
// x0 to x1 in the horizontal direction (in 26.6 fixed point co-ordinates)
|
||||||
|
// and from y0f to y1f fractional vertical units within that scanline.
|
||||||
|
func (r *Rasterizer) scan(yi int, x0, y0f, x1, y1f fixed.Int26_6) {
|
||||||
|
// Break the 26.6 fixed point X co-ordinates into integral and fractional parts.
|
||||||
|
x0i := int(x0) / 64
|
||||||
|
x0f := x0 - fixed.Int26_6(64*x0i)
|
||||||
|
x1i := int(x1) / 64
|
||||||
|
x1f := x1 - fixed.Int26_6(64*x1i)
|
||||||
|
|
||||||
|
// A perfectly horizontal scan.
|
||||||
|
if y0f == y1f {
|
||||||
|
r.setCell(x1i, yi)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
dx, dy := x1-x0, y1f-y0f
|
||||||
|
// A single cell scan.
|
||||||
|
if x0i == x1i {
|
||||||
|
r.area += int((x0f + x1f) * dy)
|
||||||
|
r.cover += int(dy)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// There are at least two cells. Apart from the first and last cells,
|
||||||
|
// all intermediate cells go through the full width of the cell,
|
||||||
|
// or 64 units in 26.6 fixed point format.
|
||||||
|
var (
|
||||||
|
p, q, edge0, edge1 fixed.Int26_6
|
||||||
|
xiDelta int
|
||||||
|
)
|
||||||
|
if dx > 0 {
|
||||||
|
p, q = (64-x0f)*dy, dx
|
||||||
|
edge0, edge1, xiDelta = 0, 64, 1
|
||||||
|
} else {
|
||||||
|
p, q = x0f*dy, -dx
|
||||||
|
edge0, edge1, xiDelta = 64, 0, -1
|
||||||
|
}
|
||||||
|
yDelta, yRem := p/q, p%q
|
||||||
|
if yRem < 0 {
|
||||||
|
yDelta -= 1
|
||||||
|
yRem += q
|
||||||
|
}
|
||||||
|
// Do the first cell.
|
||||||
|
xi, y := x0i, y0f
|
||||||
|
r.area += int((x0f + edge1) * yDelta)
|
||||||
|
r.cover += int(yDelta)
|
||||||
|
xi, y = xi+xiDelta, y+yDelta
|
||||||
|
r.setCell(xi, yi)
|
||||||
|
if xi != x1i {
|
||||||
|
// Do all the intermediate cells.
|
||||||
|
p = 64 * (y1f - y + yDelta)
|
||||||
|
fullDelta, fullRem := p/q, p%q
|
||||||
|
if fullRem < 0 {
|
||||||
|
fullDelta -= 1
|
||||||
|
fullRem += q
|
||||||
|
}
|
||||||
|
yRem -= q
|
||||||
|
for xi != x1i {
|
||||||
|
yDelta = fullDelta
|
||||||
|
yRem += fullRem
|
||||||
|
if yRem >= 0 {
|
||||||
|
yDelta += 1
|
||||||
|
yRem -= q
|
||||||
|
}
|
||||||
|
r.area += int(64 * yDelta)
|
||||||
|
r.cover += int(yDelta)
|
||||||
|
xi, y = xi+xiDelta, y+yDelta
|
||||||
|
r.setCell(xi, yi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Do the last cell.
|
||||||
|
yDelta = y1f - y
|
||||||
|
r.area += int((edge0 + x1f) * yDelta)
|
||||||
|
r.cover += int(yDelta)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start starts a new curve at the given point.
|
||||||
|
func (r *Rasterizer) Start(a fixed.Point26_6) {
|
||||||
|
r.setCell(int(a.X/64), int(a.Y/64))
|
||||||
|
r.a = a
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add1 adds a linear segment to the current curve.
|
||||||
|
func (r *Rasterizer) Add1(b fixed.Point26_6) {
|
||||||
|
x0, y0 := r.a.X, r.a.Y
|
||||||
|
x1, y1 := b.X, b.Y
|
||||||
|
dx, dy := x1-x0, y1-y0
|
||||||
|
// Break the 26.6 fixed point Y co-ordinates into integral and fractional
|
||||||
|
// parts.
|
||||||
|
y0i := int(y0) / 64
|
||||||
|
y0f := y0 - fixed.Int26_6(64*y0i)
|
||||||
|
y1i := int(y1) / 64
|
||||||
|
y1f := y1 - fixed.Int26_6(64*y1i)
|
||||||
|
|
||||||
|
if y0i == y1i {
|
||||||
|
// There is only one scanline.
|
||||||
|
r.scan(y0i, x0, y0f, x1, y1f)
|
||||||
|
|
||||||
|
} else if dx == 0 {
|
||||||
|
// This is a vertical line segment. We avoid calling r.scan and instead
|
||||||
|
// manipulate r.area and r.cover directly.
|
||||||
|
var (
|
||||||
|
edge0, edge1 fixed.Int26_6
|
||||||
|
yiDelta int
|
||||||
|
)
|
||||||
|
if dy > 0 {
|
||||||
|
edge0, edge1, yiDelta = 0, 64, 1
|
||||||
|
} else {
|
||||||
|
edge0, edge1, yiDelta = 64, 0, -1
|
||||||
|
}
|
||||||
|
x0i, yi := int(x0)/64, y0i
|
||||||
|
x0fTimes2 := (int(x0) - (64 * x0i)) * 2
|
||||||
|
// Do the first pixel.
|
||||||
|
dcover := int(edge1 - y0f)
|
||||||
|
darea := int(x0fTimes2 * dcover)
|
||||||
|
r.area += darea
|
||||||
|
r.cover += dcover
|
||||||
|
yi += yiDelta
|
||||||
|
r.setCell(x0i, yi)
|
||||||
|
// Do all the intermediate pixels.
|
||||||
|
dcover = int(edge1 - edge0)
|
||||||
|
darea = int(x0fTimes2 * dcover)
|
||||||
|
for yi != y1i {
|
||||||
|
r.area += darea
|
||||||
|
r.cover += dcover
|
||||||
|
yi += yiDelta
|
||||||
|
r.setCell(x0i, yi)
|
||||||
|
}
|
||||||
|
// Do the last pixel.
|
||||||
|
dcover = int(y1f - edge0)
|
||||||
|
darea = int(x0fTimes2 * dcover)
|
||||||
|
r.area += darea
|
||||||
|
r.cover += dcover
|
||||||
|
|
||||||
|
} else {
|
||||||
|
// There are at least two scanlines. Apart from the first and last
|
||||||
|
// scanlines, all intermediate scanlines go through the full height of
|
||||||
|
// the row, or 64 units in 26.6 fixed point format.
|
||||||
|
var (
|
||||||
|
p, q, edge0, edge1 fixed.Int26_6
|
||||||
|
yiDelta int
|
||||||
|
)
|
||||||
|
if dy > 0 {
|
||||||
|
p, q = (64-y0f)*dx, dy
|
||||||
|
edge0, edge1, yiDelta = 0, 64, 1
|
||||||
|
} else {
|
||||||
|
p, q = y0f*dx, -dy
|
||||||
|
edge0, edge1, yiDelta = 64, 0, -1
|
||||||
|
}
|
||||||
|
xDelta, xRem := p/q, p%q
|
||||||
|
if xRem < 0 {
|
||||||
|
xDelta -= 1
|
||||||
|
xRem += q
|
||||||
|
}
|
||||||
|
// Do the first scanline.
|
||||||
|
x, yi := x0, y0i
|
||||||
|
r.scan(yi, x, y0f, x+xDelta, edge1)
|
||||||
|
x, yi = x+xDelta, yi+yiDelta
|
||||||
|
r.setCell(int(x)/64, yi)
|
||||||
|
if yi != y1i {
|
||||||
|
// Do all the intermediate scanlines.
|
||||||
|
p = 64 * dx
|
||||||
|
fullDelta, fullRem := p/q, p%q
|
||||||
|
if fullRem < 0 {
|
||||||
|
fullDelta -= 1
|
||||||
|
fullRem += q
|
||||||
|
}
|
||||||
|
xRem -= q
|
||||||
|
for yi != y1i {
|
||||||
|
xDelta = fullDelta
|
||||||
|
xRem += fullRem
|
||||||
|
if xRem >= 0 {
|
||||||
|
xDelta += 1
|
||||||
|
xRem -= q
|
||||||
|
}
|
||||||
|
r.scan(yi, x, edge0, x+xDelta, edge1)
|
||||||
|
x, yi = x+xDelta, yi+yiDelta
|
||||||
|
r.setCell(int(x)/64, yi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Do the last scanline.
|
||||||
|
r.scan(yi, x, edge0, x1, y1f)
|
||||||
|
}
|
||||||
|
// The next lineTo starts from b.
|
||||||
|
r.a = b
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add2 adds a quadratic segment to the current curve.
|
||||||
|
func (r *Rasterizer) Add2(b, c fixed.Point26_6) {
|
||||||
|
// Calculate nSplit (the number of recursive decompositions) based on how
|
||||||
|
// 'curvy' it is. Specifically, how much the middle point b deviates from
|
||||||
|
// (a+c)/2.
|
||||||
|
dev := maxAbs(r.a.X-2*b.X+c.X, r.a.Y-2*b.Y+c.Y) / fixed.Int26_6(r.splitScale2)
|
||||||
|
nsplit := 0
|
||||||
|
for dev > 0 {
|
||||||
|
dev /= 4
|
||||||
|
nsplit++
|
||||||
|
}
|
||||||
|
// dev is 32-bit, and nsplit++ every time we shift off 2 bits, so maxNsplit
|
||||||
|
// is 16.
|
||||||
|
const maxNsplit = 16
|
||||||
|
if nsplit > maxNsplit {
|
||||||
|
panic("freetype/raster: Add2 nsplit too large: " + strconv.Itoa(nsplit))
|
||||||
|
}
|
||||||
|
// Recursively decompose the curve nSplit levels deep.
|
||||||
|
var (
|
||||||
|
pStack [2*maxNsplit + 3]fixed.Point26_6
|
||||||
|
sStack [maxNsplit + 1]int
|
||||||
|
i int
|
||||||
|
)
|
||||||
|
sStack[0] = nsplit
|
||||||
|
pStack[0] = c
|
||||||
|
pStack[1] = b
|
||||||
|
pStack[2] = r.a
|
||||||
|
for i >= 0 {
|
||||||
|
s := sStack[i]
|
||||||
|
p := pStack[2*i:]
|
||||||
|
if s > 0 {
|
||||||
|
// Split the quadratic curve p[:3] into an equivalent set of two
|
||||||
|
// shorter curves: p[:3] and p[2:5]. The new p[4] is the old p[2],
|
||||||
|
// and p[0] is unchanged.
|
||||||
|
mx := p[1].X
|
||||||
|
p[4].X = p[2].X
|
||||||
|
p[3].X = (p[4].X + mx) / 2
|
||||||
|
p[1].X = (p[0].X + mx) / 2
|
||||||
|
p[2].X = (p[1].X + p[3].X) / 2
|
||||||
|
my := p[1].Y
|
||||||
|
p[4].Y = p[2].Y
|
||||||
|
p[3].Y = (p[4].Y + my) / 2
|
||||||
|
p[1].Y = (p[0].Y + my) / 2
|
||||||
|
p[2].Y = (p[1].Y + p[3].Y) / 2
|
||||||
|
// The two shorter curves have one less split to do.
|
||||||
|
sStack[i] = s - 1
|
||||||
|
sStack[i+1] = s - 1
|
||||||
|
i++
|
||||||
|
} else {
|
||||||
|
// Replace the level-0 quadratic with a two-linear-piece
|
||||||
|
// approximation.
|
||||||
|
midx := (p[0].X + 2*p[1].X + p[2].X) / 4
|
||||||
|
midy := (p[0].Y + 2*p[1].Y + p[2].Y) / 4
|
||||||
|
r.Add1(fixed.Point26_6{midx, midy})
|
||||||
|
r.Add1(p[0])
|
||||||
|
i--
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add3 adds a cubic segment to the current curve.
|
||||||
|
func (r *Rasterizer) Add3(b, c, d fixed.Point26_6) {
|
||||||
|
// Calculate nSplit (the number of recursive decompositions) based on how
|
||||||
|
// 'curvy' it is.
|
||||||
|
dev2 := maxAbs(r.a.X-3*(b.X+c.X)+d.X, r.a.Y-3*(b.Y+c.Y)+d.Y) / fixed.Int26_6(r.splitScale2)
|
||||||
|
dev3 := maxAbs(r.a.X-2*b.X+d.X, r.a.Y-2*b.Y+d.Y) / fixed.Int26_6(r.splitScale3)
|
||||||
|
nsplit := 0
|
||||||
|
for dev2 > 0 || dev3 > 0 {
|
||||||
|
dev2 /= 8
|
||||||
|
dev3 /= 4
|
||||||
|
nsplit++
|
||||||
|
}
|
||||||
|
// devN is 32-bit, and nsplit++ every time we shift off 2 bits, so
|
||||||
|
// maxNsplit is 16.
|
||||||
|
const maxNsplit = 16
|
||||||
|
if nsplit > maxNsplit {
|
||||||
|
panic("freetype/raster: Add3 nsplit too large: " + strconv.Itoa(nsplit))
|
||||||
|
}
|
||||||
|
// Recursively decompose the curve nSplit levels deep.
|
||||||
|
var (
|
||||||
|
pStack [3*maxNsplit + 4]fixed.Point26_6
|
||||||
|
sStack [maxNsplit + 1]int
|
||||||
|
i int
|
||||||
|
)
|
||||||
|
sStack[0] = nsplit
|
||||||
|
pStack[0] = d
|
||||||
|
pStack[1] = c
|
||||||
|
pStack[2] = b
|
||||||
|
pStack[3] = r.a
|
||||||
|
for i >= 0 {
|
||||||
|
s := sStack[i]
|
||||||
|
p := pStack[3*i:]
|
||||||
|
if s > 0 {
|
||||||
|
// Split the cubic curve p[:4] into an equivalent set of two
|
||||||
|
// shorter curves: p[:4] and p[3:7]. The new p[6] is the old p[3],
|
||||||
|
// and p[0] is unchanged.
|
||||||
|
m01x := (p[0].X + p[1].X) / 2
|
||||||
|
m12x := (p[1].X + p[2].X) / 2
|
||||||
|
m23x := (p[2].X + p[3].X) / 2
|
||||||
|
p[6].X = p[3].X
|
||||||
|
p[5].X = m23x
|
||||||
|
p[1].X = m01x
|
||||||
|
p[2].X = (m01x + m12x) / 2
|
||||||
|
p[4].X = (m12x + m23x) / 2
|
||||||
|
p[3].X = (p[2].X + p[4].X) / 2
|
||||||
|
m01y := (p[0].Y + p[1].Y) / 2
|
||||||
|
m12y := (p[1].Y + p[2].Y) / 2
|
||||||
|
m23y := (p[2].Y + p[3].Y) / 2
|
||||||
|
p[6].Y = p[3].Y
|
||||||
|
p[5].Y = m23y
|
||||||
|
p[1].Y = m01y
|
||||||
|
p[2].Y = (m01y + m12y) / 2
|
||||||
|
p[4].Y = (m12y + m23y) / 2
|
||||||
|
p[3].Y = (p[2].Y + p[4].Y) / 2
|
||||||
|
// The two shorter curves have one less split to do.
|
||||||
|
sStack[i] = s - 1
|
||||||
|
sStack[i+1] = s - 1
|
||||||
|
i++
|
||||||
|
} else {
|
||||||
|
// Replace the level-0 cubic with a two-linear-piece approximation.
|
||||||
|
midx := (p[0].X + 3*(p[1].X+p[2].X) + p[3].X) / 8
|
||||||
|
midy := (p[0].Y + 3*(p[1].Y+p[2].Y) + p[3].Y) / 8
|
||||||
|
r.Add1(fixed.Point26_6{midx, midy})
|
||||||
|
r.Add1(p[0])
|
||||||
|
i--
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddPath adds the given Path.
|
||||||
|
func (r *Rasterizer) AddPath(p Path) {
|
||||||
|
for i := 0; i < len(p); {
|
||||||
|
switch p[i] {
|
||||||
|
case 0:
|
||||||
|
r.Start(
|
||||||
|
fixed.Point26_6{p[i+1], p[i+2]},
|
||||||
|
)
|
||||||
|
i += 4
|
||||||
|
case 1:
|
||||||
|
r.Add1(
|
||||||
|
fixed.Point26_6{p[i+1], p[i+2]},
|
||||||
|
)
|
||||||
|
i += 4
|
||||||
|
case 2:
|
||||||
|
r.Add2(
|
||||||
|
fixed.Point26_6{p[i+1], p[i+2]},
|
||||||
|
fixed.Point26_6{p[i+3], p[i+4]},
|
||||||
|
)
|
||||||
|
i += 6
|
||||||
|
case 3:
|
||||||
|
r.Add3(
|
||||||
|
fixed.Point26_6{p[i+1], p[i+2]},
|
||||||
|
fixed.Point26_6{p[i+3], p[i+4]},
|
||||||
|
fixed.Point26_6{p[i+5], p[i+6]},
|
||||||
|
)
|
||||||
|
i += 8
|
||||||
|
default:
|
||||||
|
panic("freetype/raster: bad path")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddStroke adds a stroked Path.
|
||||||
|
func (r *Rasterizer) AddStroke(q Path, width fixed.Int26_6, cr Capper, jr Joiner) {
|
||||||
|
Stroke(r, q, width, cr, jr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// areaToAlpha converts an area value to a uint32 alpha value. A completely
|
||||||
|
// filled pixel corresponds to an area of 64*64*2, and an alpha of 0xffff. The
|
||||||
|
// conversion of area values greater than this depends on the winding rule:
|
||||||
|
// even-odd or non-zero.
|
||||||
|
func (r *Rasterizer) areaToAlpha(area int) uint32 {
|
||||||
|
// The C Freetype implementation (version 2.3.12) does "alpha := area>>1"
|
||||||
|
// without the +1. Round-to-nearest gives a more symmetric result than
|
||||||
|
// round-down. The C implementation also returns 8-bit alpha, not 16-bit
|
||||||
|
// alpha.
|
||||||
|
a := (area + 1) >> 1
|
||||||
|
if a < 0 {
|
||||||
|
a = -a
|
||||||
|
}
|
||||||
|
alpha := uint32(a)
|
||||||
|
if r.UseNonZeroWinding {
|
||||||
|
if alpha > 0x0fff {
|
||||||
|
alpha = 0x0fff
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
alpha &= 0x1fff
|
||||||
|
if alpha > 0x1000 {
|
||||||
|
alpha = 0x2000 - alpha
|
||||||
|
} else if alpha == 0x1000 {
|
||||||
|
alpha = 0x0fff
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// alpha is now in the range [0x0000, 0x0fff]. Convert that 12-bit alpha to
|
||||||
|
// 16-bit alpha.
|
||||||
|
return alpha<<4 | alpha>>8
|
||||||
|
}
|
||||||
|
|
||||||
|
// Rasterize converts r's accumulated curves into Spans for p. The Spans passed
|
||||||
|
// to p are non-overlapping, and sorted by Y and then X. They all have non-zero
|
||||||
|
// width (and 0 <= X0 < X1 <= r.width) and non-zero A, except for the final
|
||||||
|
// Span, which has Y, X0, X1 and A all equal to zero.
|
||||||
|
func (r *Rasterizer) Rasterize(p Painter) {
|
||||||
|
r.saveCell()
|
||||||
|
s := 0
|
||||||
|
for yi := 0; yi < len(r.cellIndex); yi++ {
|
||||||
|
xi, cover := 0, 0
|
||||||
|
for c := r.cellIndex[yi]; c != -1; c = r.cell[c].next {
|
||||||
|
if cover != 0 && r.cell[c].xi > xi {
|
||||||
|
alpha := r.areaToAlpha(cover * 64 * 2)
|
||||||
|
if alpha != 0 {
|
||||||
|
xi0, xi1 := xi, r.cell[c].xi
|
||||||
|
if xi0 < 0 {
|
||||||
|
xi0 = 0
|
||||||
|
}
|
||||||
|
if xi1 >= r.width {
|
||||||
|
xi1 = r.width
|
||||||
|
}
|
||||||
|
if xi0 < xi1 {
|
||||||
|
r.spanBuf[s] = Span{yi + r.Dy, xi0 + r.Dx, xi1 + r.Dx, alpha}
|
||||||
|
s++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
cover += r.cell[c].cover
|
||||||
|
alpha := r.areaToAlpha(cover*64*2 - r.cell[c].area)
|
||||||
|
xi = r.cell[c].xi + 1
|
||||||
|
if alpha != 0 {
|
||||||
|
xi0, xi1 := r.cell[c].xi, xi
|
||||||
|
if xi0 < 0 {
|
||||||
|
xi0 = 0
|
||||||
|
}
|
||||||
|
if xi1 >= r.width {
|
||||||
|
xi1 = r.width
|
||||||
|
}
|
||||||
|
if xi0 < xi1 {
|
||||||
|
r.spanBuf[s] = Span{yi + r.Dy, xi0 + r.Dx, xi1 + r.Dx, alpha}
|
||||||
|
s++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if s > len(r.spanBuf)-2 {
|
||||||
|
p.Paint(r.spanBuf[:s], false)
|
||||||
|
s = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
p.Paint(r.spanBuf[:s], true)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Clear cancels any previous calls to r.Start or r.AddXxx.
|
||||||
|
func (r *Rasterizer) Clear() {
|
||||||
|
r.a = fixed.Point26_6{}
|
||||||
|
r.xi = 0
|
||||||
|
r.yi = 0
|
||||||
|
r.area = 0
|
||||||
|
r.cover = 0
|
||||||
|
r.cell = r.cell[:0]
|
||||||
|
for i := 0; i < len(r.cellIndex); i++ {
|
||||||
|
r.cellIndex[i] = -1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetBounds sets the maximum width and height of the rasterized image and
|
||||||
|
// calls Clear. The width and height are in pixels, not fixed.Int26_6 units.
|
||||||
|
func (r *Rasterizer) SetBounds(width, height int) {
|
||||||
|
if width < 0 {
|
||||||
|
width = 0
|
||||||
|
}
|
||||||
|
if height < 0 {
|
||||||
|
height = 0
|
||||||
|
}
|
||||||
|
// Use the same ssN heuristic as the C Freetype (version 2.4.0)
|
||||||
|
// implementation.
|
||||||
|
ss2, ss3 := 32, 16
|
||||||
|
if width > 24 || height > 24 {
|
||||||
|
ss2, ss3 = 2*ss2, 2*ss3
|
||||||
|
if width > 120 || height > 120 {
|
||||||
|
ss2, ss3 = 2*ss2, 2*ss3
|
||||||
|
}
|
||||||
|
}
|
||||||
|
r.width = width
|
||||||
|
r.splitScale2 = ss2
|
||||||
|
r.splitScale3 = ss3
|
||||||
|
r.cell = r.cellBuf[:0]
|
||||||
|
if height > len(r.cellIndexBuf) {
|
||||||
|
r.cellIndex = make([]int, height)
|
||||||
|
} else {
|
||||||
|
r.cellIndex = r.cellIndexBuf[:height]
|
||||||
|
}
|
||||||
|
r.Clear()
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewRasterizer creates a new Rasterizer with the given bounds.
|
||||||
|
func NewRasterizer(width, height int) *Rasterizer {
|
||||||
|
r := new(Rasterizer)
|
||||||
|
r.SetBounds(width, height)
|
||||||
|
return r
|
||||||
|
}
|
|
@ -0,0 +1,483 @@
|
||||||
|
// Copyright 2010 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package raster
|
||||||
|
|
||||||
|
import (
|
||||||
|
"golang.org/x/image/math/fixed"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Two points are considered practically equal if the square of the distance
|
||||||
|
// between them is less than one quarter (i.e. 1024 / 4096).
|
||||||
|
const epsilon = fixed.Int52_12(1024)
|
||||||
|
|
||||||
|
// A Capper signifies how to begin or end a stroked path.
|
||||||
|
type Capper interface {
|
||||||
|
// Cap adds a cap to p given a pivot point and the normal vector of a
|
||||||
|
// terminal segment. The normal's length is half of the stroke width.
|
||||||
|
Cap(p Adder, halfWidth fixed.Int26_6, pivot, n1 fixed.Point26_6)
|
||||||
|
}
|
||||||
|
|
||||||
|
// The CapperFunc type adapts an ordinary function to be a Capper.
|
||||||
|
type CapperFunc func(Adder, fixed.Int26_6, fixed.Point26_6, fixed.Point26_6)
|
||||||
|
|
||||||
|
func (f CapperFunc) Cap(p Adder, halfWidth fixed.Int26_6, pivot, n1 fixed.Point26_6) {
|
||||||
|
f(p, halfWidth, pivot, n1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// A Joiner signifies how to join interior nodes of a stroked path.
|
||||||
|
type Joiner interface {
|
||||||
|
// Join adds a join to the two sides of a stroked path given a pivot
|
||||||
|
// point and the normal vectors of the trailing and leading segments.
|
||||||
|
// Both normals have length equal to half of the stroke width.
|
||||||
|
Join(lhs, rhs Adder, halfWidth fixed.Int26_6, pivot, n0, n1 fixed.Point26_6)
|
||||||
|
}
|
||||||
|
|
||||||
|
// The JoinerFunc type adapts an ordinary function to be a Joiner.
|
||||||
|
type JoinerFunc func(lhs, rhs Adder, halfWidth fixed.Int26_6, pivot, n0, n1 fixed.Point26_6)
|
||||||
|
|
||||||
|
func (f JoinerFunc) Join(lhs, rhs Adder, halfWidth fixed.Int26_6, pivot, n0, n1 fixed.Point26_6) {
|
||||||
|
f(lhs, rhs, halfWidth, pivot, n0, n1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RoundCapper adds round caps to a stroked path.
|
||||||
|
var RoundCapper Capper = CapperFunc(roundCapper)
|
||||||
|
|
||||||
|
func roundCapper(p Adder, halfWidth fixed.Int26_6, pivot, n1 fixed.Point26_6) {
|
||||||
|
// The cubic Bézier approximation to a circle involves the magic number
|
||||||
|
// (√2 - 1) * 4/3, which is approximately 35/64.
|
||||||
|
const k = 35
|
||||||
|
e := pRot90CCW(n1)
|
||||||
|
side := pivot.Add(e)
|
||||||
|
start, end := pivot.Sub(n1), pivot.Add(n1)
|
||||||
|
d, e := n1.Mul(k), e.Mul(k)
|
||||||
|
p.Add3(start.Add(e), side.Sub(d), side)
|
||||||
|
p.Add3(side.Add(d), end.Add(e), end)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ButtCapper adds butt caps to a stroked path.
|
||||||
|
var ButtCapper Capper = CapperFunc(buttCapper)
|
||||||
|
|
||||||
|
func buttCapper(p Adder, halfWidth fixed.Int26_6, pivot, n1 fixed.Point26_6) {
|
||||||
|
p.Add1(pivot.Add(n1))
|
||||||
|
}
|
||||||
|
|
||||||
|
// SquareCapper adds square caps to a stroked path.
|
||||||
|
var SquareCapper Capper = CapperFunc(squareCapper)
|
||||||
|
|
||||||
|
func squareCapper(p Adder, halfWidth fixed.Int26_6, pivot, n1 fixed.Point26_6) {
|
||||||
|
e := pRot90CCW(n1)
|
||||||
|
side := pivot.Add(e)
|
||||||
|
p.Add1(side.Sub(n1))
|
||||||
|
p.Add1(side.Add(n1))
|
||||||
|
p.Add1(pivot.Add(n1))
|
||||||
|
}
|
||||||
|
|
||||||
|
// RoundJoiner adds round joins to a stroked path.
|
||||||
|
var RoundJoiner Joiner = JoinerFunc(roundJoiner)
|
||||||
|
|
||||||
|
func roundJoiner(lhs, rhs Adder, haflWidth fixed.Int26_6, pivot, n0, n1 fixed.Point26_6) {
|
||||||
|
dot := pDot(pRot90CW(n0), n1)
|
||||||
|
if dot >= 0 {
|
||||||
|
addArc(lhs, pivot, n0, n1)
|
||||||
|
rhs.Add1(pivot.Sub(n1))
|
||||||
|
} else {
|
||||||
|
lhs.Add1(pivot.Add(n1))
|
||||||
|
addArc(rhs, pivot, pNeg(n0), pNeg(n1))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BevelJoiner adds bevel joins to a stroked path.
|
||||||
|
var BevelJoiner Joiner = JoinerFunc(bevelJoiner)
|
||||||
|
|
||||||
|
func bevelJoiner(lhs, rhs Adder, haflWidth fixed.Int26_6, pivot, n0, n1 fixed.Point26_6) {
|
||||||
|
lhs.Add1(pivot.Add(n1))
|
||||||
|
rhs.Add1(pivot.Sub(n1))
|
||||||
|
}
|
||||||
|
|
||||||
|
// addArc adds a circular arc from pivot+n0 to pivot+n1 to p. The shorter of
|
||||||
|
// the two possible arcs is taken, i.e. the one spanning <= 180 degrees. The
|
||||||
|
// two vectors n0 and n1 must be of equal length.
|
||||||
|
func addArc(p Adder, pivot, n0, n1 fixed.Point26_6) {
|
||||||
|
// r2 is the square of the length of n0.
|
||||||
|
r2 := pDot(n0, n0)
|
||||||
|
if r2 < epsilon {
|
||||||
|
// The arc radius is so small that we collapse to a straight line.
|
||||||
|
p.Add1(pivot.Add(n1))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// We approximate the arc by 0, 1, 2 or 3 45-degree quadratic segments plus
|
||||||
|
// a final quadratic segment from s to n1. Each 45-degree segment has
|
||||||
|
// control points {1, 0}, {1, tan(π/8)} and {1/√2, 1/√2} suitably scaled,
|
||||||
|
// rotated and translated. tan(π/8) is approximately 27/64.
|
||||||
|
const tpo8 = 27
|
||||||
|
var s fixed.Point26_6
|
||||||
|
// We determine which octant the angle between n0 and n1 is in via three
|
||||||
|
// dot products. m0, m1 and m2 are n0 rotated clockwise by 45, 90 and 135
|
||||||
|
// degrees.
|
||||||
|
m0 := pRot45CW(n0)
|
||||||
|
m1 := pRot90CW(n0)
|
||||||
|
m2 := pRot90CW(m0)
|
||||||
|
if pDot(m1, n1) >= 0 {
|
||||||
|
if pDot(n0, n1) >= 0 {
|
||||||
|
if pDot(m2, n1) <= 0 {
|
||||||
|
// n1 is between 0 and 45 degrees clockwise of n0.
|
||||||
|
s = n0
|
||||||
|
} else {
|
||||||
|
// n1 is between 45 and 90 degrees clockwise of n0.
|
||||||
|
p.Add2(pivot.Add(n0).Add(m1.Mul(tpo8)), pivot.Add(m0))
|
||||||
|
s = m0
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
pm1, n0t := pivot.Add(m1), n0.Mul(tpo8)
|
||||||
|
p.Add2(pivot.Add(n0).Add(m1.Mul(tpo8)), pivot.Add(m0))
|
||||||
|
p.Add2(pm1.Add(n0t), pm1)
|
||||||
|
if pDot(m0, n1) >= 0 {
|
||||||
|
// n1 is between 90 and 135 degrees clockwise of n0.
|
||||||
|
s = m1
|
||||||
|
} else {
|
||||||
|
// n1 is between 135 and 180 degrees clockwise of n0.
|
||||||
|
p.Add2(pm1.Sub(n0t), pivot.Add(m2))
|
||||||
|
s = m2
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if pDot(n0, n1) >= 0 {
|
||||||
|
if pDot(m0, n1) >= 0 {
|
||||||
|
// n1 is between 0 and 45 degrees counter-clockwise of n0.
|
||||||
|
s = n0
|
||||||
|
} else {
|
||||||
|
// n1 is between 45 and 90 degrees counter-clockwise of n0.
|
||||||
|
p.Add2(pivot.Add(n0).Sub(m1.Mul(tpo8)), pivot.Sub(m2))
|
||||||
|
s = pNeg(m2)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
pm1, n0t := pivot.Sub(m1), n0.Mul(tpo8)
|
||||||
|
p.Add2(pivot.Add(n0).Sub(m1.Mul(tpo8)), pivot.Sub(m2))
|
||||||
|
p.Add2(pm1.Add(n0t), pm1)
|
||||||
|
if pDot(m2, n1) <= 0 {
|
||||||
|
// n1 is between 90 and 135 degrees counter-clockwise of n0.
|
||||||
|
s = pNeg(m1)
|
||||||
|
} else {
|
||||||
|
// n1 is between 135 and 180 degrees counter-clockwise of n0.
|
||||||
|
p.Add2(pm1.Sub(n0t), pivot.Sub(m0))
|
||||||
|
s = pNeg(m0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// The final quadratic segment has two endpoints s and n1 and the middle
|
||||||
|
// control point is a multiple of s.Add(n1), i.e. it is on the angle
|
||||||
|
// bisector of those two points. The multiple ranges between 128/256 and
|
||||||
|
// 150/256 as the angle between s and n1 ranges between 0 and 45 degrees.
|
||||||
|
//
|
||||||
|
// When the angle is 0 degrees (i.e. s and n1 are coincident) then
|
||||||
|
// s.Add(n1) is twice s and so the middle control point of the degenerate
|
||||||
|
// quadratic segment should be half s.Add(n1), and half = 128/256.
|
||||||
|
//
|
||||||
|
// When the angle is 45 degrees then 150/256 is the ratio of the lengths of
|
||||||
|
// the two vectors {1, tan(π/8)} and {1 + 1/√2, 1/√2}.
|
||||||
|
//
|
||||||
|
// d is the normalized dot product between s and n1. Since the angle ranges
|
||||||
|
// between 0 and 45 degrees then d ranges between 256/256 and 181/256.
|
||||||
|
d := 256 * pDot(s, n1) / r2
|
||||||
|
multiple := fixed.Int26_6(150-(150-128)*(d-181)/(256-181)) >> 2
|
||||||
|
p.Add2(pivot.Add(s.Add(n1).Mul(multiple)), pivot.Add(n1))
|
||||||
|
}
|
||||||
|
|
||||||
|
// midpoint returns the midpoint of two Points.
|
||||||
|
func midpoint(a, b fixed.Point26_6) fixed.Point26_6 {
|
||||||
|
return fixed.Point26_6{(a.X + b.X) / 2, (a.Y + b.Y) / 2}
|
||||||
|
}
|
||||||
|
|
||||||
|
// angleGreaterThan45 returns whether the angle between two vectors is more
|
||||||
|
// than 45 degrees.
|
||||||
|
func angleGreaterThan45(v0, v1 fixed.Point26_6) bool {
|
||||||
|
v := pRot45CCW(v0)
|
||||||
|
return pDot(v, v1) < 0 || pDot(pRot90CW(v), v1) < 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// interpolate returns the point (1-t)*a + t*b.
|
||||||
|
func interpolate(a, b fixed.Point26_6, t fixed.Int52_12) fixed.Point26_6 {
|
||||||
|
s := 1<<12 - t
|
||||||
|
x := s*fixed.Int52_12(a.X) + t*fixed.Int52_12(b.X)
|
||||||
|
y := s*fixed.Int52_12(a.Y) + t*fixed.Int52_12(b.Y)
|
||||||
|
return fixed.Point26_6{fixed.Int26_6(x >> 12), fixed.Int26_6(y >> 12)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// curviest2 returns the value of t for which the quadratic parametric curve
|
||||||
|
// (1-t)²*a + 2*t*(1-t).b + t²*c has maximum curvature.
|
||||||
|
//
|
||||||
|
// The curvature of the parametric curve f(t) = (x(t), y(t)) is
|
||||||
|
// |x′y″-y′x″| / (x′²+y′²)^(3/2).
|
||||||
|
//
|
||||||
|
// Let d = b-a and e = c-2*b+a, so that f′(t) = 2*d+2*e*t and f″(t) = 2*e.
|
||||||
|
// The curvature's numerator is (2*dx+2*ex*t)*(2*ey)-(2*dy+2*ey*t)*(2*ex),
|
||||||
|
// which simplifies to 4*dx*ey-4*dy*ex, which is constant with respect to t.
|
||||||
|
//
|
||||||
|
// Thus, curvature is extreme where the denominator is extreme, i.e. where
|
||||||
|
// (x′²+y′²) is extreme. The first order condition is that
|
||||||
|
// 2*x′*x″+2*y′*y″ = 0, or (dx+ex*t)*ex + (dy+ey*t)*ey = 0.
|
||||||
|
// Solving for t gives t = -(dx*ex+dy*ey) / (ex*ex+ey*ey).
|
||||||
|
func curviest2(a, b, c fixed.Point26_6) fixed.Int52_12 {
|
||||||
|
dx := int64(b.X - a.X)
|
||||||
|
dy := int64(b.Y - a.Y)
|
||||||
|
ex := int64(c.X - 2*b.X + a.X)
|
||||||
|
ey := int64(c.Y - 2*b.Y + a.Y)
|
||||||
|
if ex == 0 && ey == 0 {
|
||||||
|
return 2048
|
||||||
|
}
|
||||||
|
return fixed.Int52_12(-4096 * (dx*ex + dy*ey) / (ex*ex + ey*ey))
|
||||||
|
}
|
||||||
|
|
||||||
|
// A stroker holds state for stroking a path.
|
||||||
|
type stroker struct {
|
||||||
|
// p is the destination that records the stroked path.
|
||||||
|
p Adder
|
||||||
|
// u is the half-width of the stroke.
|
||||||
|
u fixed.Int26_6
|
||||||
|
// cr and jr specify how to end and connect path segments.
|
||||||
|
cr Capper
|
||||||
|
jr Joiner
|
||||||
|
// r is the reverse path. Stroking a path involves constructing two
|
||||||
|
// parallel paths 2*u apart. The first path is added immediately to p,
|
||||||
|
// the second path is accumulated in r and eventually added in reverse.
|
||||||
|
r Path
|
||||||
|
// a is the most recent segment point. anorm is the segment normal of
|
||||||
|
// length u at that point.
|
||||||
|
a, anorm fixed.Point26_6
|
||||||
|
}
|
||||||
|
|
||||||
|
// addNonCurvy2 adds a quadratic segment to the stroker, where the segment
|
||||||
|
// defined by (k.a, b, c) achieves maximum curvature at either k.a or c.
|
||||||
|
func (k *stroker) addNonCurvy2(b, c fixed.Point26_6) {
|
||||||
|
// We repeatedly divide the segment at its middle until it is straight
|
||||||
|
// enough to approximate the stroke by just translating the control points.
|
||||||
|
// ds and ps are stacks of depths and points. t is the top of the stack.
|
||||||
|
const maxDepth = 5
|
||||||
|
var (
|
||||||
|
ds [maxDepth + 1]int
|
||||||
|
ps [2*maxDepth + 3]fixed.Point26_6
|
||||||
|
t int
|
||||||
|
)
|
||||||
|
// Initially the ps stack has one quadratic segment of depth zero.
|
||||||
|
ds[0] = 0
|
||||||
|
ps[2] = k.a
|
||||||
|
ps[1] = b
|
||||||
|
ps[0] = c
|
||||||
|
anorm := k.anorm
|
||||||
|
var cnorm fixed.Point26_6
|
||||||
|
|
||||||
|
for {
|
||||||
|
depth := ds[t]
|
||||||
|
a := ps[2*t+2]
|
||||||
|
b := ps[2*t+1]
|
||||||
|
c := ps[2*t+0]
|
||||||
|
ab := b.Sub(a)
|
||||||
|
bc := c.Sub(b)
|
||||||
|
abIsSmall := pDot(ab, ab) < fixed.Int52_12(1<<12)
|
||||||
|
bcIsSmall := pDot(bc, bc) < fixed.Int52_12(1<<12)
|
||||||
|
if abIsSmall && bcIsSmall {
|
||||||
|
// Approximate the segment by a circular arc.
|
||||||
|
cnorm = pRot90CCW(pNorm(bc, k.u))
|
||||||
|
mac := midpoint(a, c)
|
||||||
|
addArc(k.p, mac, anorm, cnorm)
|
||||||
|
addArc(&k.r, mac, pNeg(anorm), pNeg(cnorm))
|
||||||
|
} else if depth < maxDepth && angleGreaterThan45(ab, bc) {
|
||||||
|
// Divide the segment in two and push both halves on the stack.
|
||||||
|
mab := midpoint(a, b)
|
||||||
|
mbc := midpoint(b, c)
|
||||||
|
t++
|
||||||
|
ds[t+0] = depth + 1
|
||||||
|
ds[t-1] = depth + 1
|
||||||
|
ps[2*t+2] = a
|
||||||
|
ps[2*t+1] = mab
|
||||||
|
ps[2*t+0] = midpoint(mab, mbc)
|
||||||
|
ps[2*t-1] = mbc
|
||||||
|
continue
|
||||||
|
} else {
|
||||||
|
// Translate the control points.
|
||||||
|
bnorm := pRot90CCW(pNorm(c.Sub(a), k.u))
|
||||||
|
cnorm = pRot90CCW(pNorm(bc, k.u))
|
||||||
|
k.p.Add2(b.Add(bnorm), c.Add(cnorm))
|
||||||
|
k.r.Add2(b.Sub(bnorm), c.Sub(cnorm))
|
||||||
|
}
|
||||||
|
if t == 0 {
|
||||||
|
k.a, k.anorm = c, cnorm
|
||||||
|
return
|
||||||
|
}
|
||||||
|
t--
|
||||||
|
anorm = cnorm
|
||||||
|
}
|
||||||
|
panic("unreachable")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add1 adds a linear segment to the stroker.
|
||||||
|
func (k *stroker) Add1(b fixed.Point26_6) {
|
||||||
|
bnorm := pRot90CCW(pNorm(b.Sub(k.a), k.u))
|
||||||
|
if len(k.r) == 0 {
|
||||||
|
k.p.Start(k.a.Add(bnorm))
|
||||||
|
k.r.Start(k.a.Sub(bnorm))
|
||||||
|
} else {
|
||||||
|
k.jr.Join(k.p, &k.r, k.u, k.a, k.anorm, bnorm)
|
||||||
|
}
|
||||||
|
k.p.Add1(b.Add(bnorm))
|
||||||
|
k.r.Add1(b.Sub(bnorm))
|
||||||
|
k.a, k.anorm = b, bnorm
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add2 adds a quadratic segment to the stroker.
|
||||||
|
func (k *stroker) Add2(b, c fixed.Point26_6) {
|
||||||
|
ab := b.Sub(k.a)
|
||||||
|
bc := c.Sub(b)
|
||||||
|
abnorm := pRot90CCW(pNorm(ab, k.u))
|
||||||
|
if len(k.r) == 0 {
|
||||||
|
k.p.Start(k.a.Add(abnorm))
|
||||||
|
k.r.Start(k.a.Sub(abnorm))
|
||||||
|
} else {
|
||||||
|
k.jr.Join(k.p, &k.r, k.u, k.a, k.anorm, abnorm)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Approximate nearly-degenerate quadratics by linear segments.
|
||||||
|
abIsSmall := pDot(ab, ab) < epsilon
|
||||||
|
bcIsSmall := pDot(bc, bc) < epsilon
|
||||||
|
if abIsSmall || bcIsSmall {
|
||||||
|
acnorm := pRot90CCW(pNorm(c.Sub(k.a), k.u))
|
||||||
|
k.p.Add1(c.Add(acnorm))
|
||||||
|
k.r.Add1(c.Sub(acnorm))
|
||||||
|
k.a, k.anorm = c, acnorm
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// The quadratic segment (k.a, b, c) has a point of maximum curvature.
|
||||||
|
// If this occurs at an end point, we process the segment as a whole.
|
||||||
|
t := curviest2(k.a, b, c)
|
||||||
|
if t <= 0 || 4096 <= t {
|
||||||
|
k.addNonCurvy2(b, c)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Otherwise, we perform a de Casteljau decomposition at the point of
|
||||||
|
// maximum curvature and process the two straighter parts.
|
||||||
|
mab := interpolate(k.a, b, t)
|
||||||
|
mbc := interpolate(b, c, t)
|
||||||
|
mabc := interpolate(mab, mbc, t)
|
||||||
|
|
||||||
|
// If the vectors ab and bc are close to being in opposite directions,
|
||||||
|
// then the decomposition can become unstable, so we approximate the
|
||||||
|
// quadratic segment by two linear segments joined by an arc.
|
||||||
|
bcnorm := pRot90CCW(pNorm(bc, k.u))
|
||||||
|
if pDot(abnorm, bcnorm) < -fixed.Int52_12(k.u)*fixed.Int52_12(k.u)*2047/2048 {
|
||||||
|
pArc := pDot(abnorm, bc) < 0
|
||||||
|
|
||||||
|
k.p.Add1(mabc.Add(abnorm))
|
||||||
|
if pArc {
|
||||||
|
z := pRot90CW(abnorm)
|
||||||
|
addArc(k.p, mabc, abnorm, z)
|
||||||
|
addArc(k.p, mabc, z, bcnorm)
|
||||||
|
}
|
||||||
|
k.p.Add1(mabc.Add(bcnorm))
|
||||||
|
k.p.Add1(c.Add(bcnorm))
|
||||||
|
|
||||||
|
k.r.Add1(mabc.Sub(abnorm))
|
||||||
|
if !pArc {
|
||||||
|
z := pRot90CW(abnorm)
|
||||||
|
addArc(&k.r, mabc, pNeg(abnorm), z)
|
||||||
|
addArc(&k.r, mabc, z, pNeg(bcnorm))
|
||||||
|
}
|
||||||
|
k.r.Add1(mabc.Sub(bcnorm))
|
||||||
|
k.r.Add1(c.Sub(bcnorm))
|
||||||
|
|
||||||
|
k.a, k.anorm = c, bcnorm
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Process the decomposed parts.
|
||||||
|
k.addNonCurvy2(mab, mabc)
|
||||||
|
k.addNonCurvy2(mbc, c)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add3 adds a cubic segment to the stroker.
|
||||||
|
func (k *stroker) Add3(b, c, d fixed.Point26_6) {
|
||||||
|
panic("freetype/raster: stroke unimplemented for cubic segments")
|
||||||
|
}
|
||||||
|
|
||||||
|
// stroke adds the stroked Path q to p, where q consists of exactly one curve.
|
||||||
|
func (k *stroker) stroke(q Path) {
|
||||||
|
// Stroking is implemented by deriving two paths each k.u apart from q.
|
||||||
|
// The left-hand-side path is added immediately to k.p; the right-hand-side
|
||||||
|
// path is accumulated in k.r. Once we've finished adding the LHS to k.p,
|
||||||
|
// we add the RHS in reverse order.
|
||||||
|
k.r = make(Path, 0, len(q))
|
||||||
|
k.a = fixed.Point26_6{q[1], q[2]}
|
||||||
|
for i := 4; i < len(q); {
|
||||||
|
switch q[i] {
|
||||||
|
case 1:
|
||||||
|
k.Add1(
|
||||||
|
fixed.Point26_6{q[i+1], q[i+2]},
|
||||||
|
)
|
||||||
|
i += 4
|
||||||
|
case 2:
|
||||||
|
k.Add2(
|
||||||
|
fixed.Point26_6{q[i+1], q[i+2]},
|
||||||
|
fixed.Point26_6{q[i+3], q[i+4]},
|
||||||
|
)
|
||||||
|
i += 6
|
||||||
|
case 3:
|
||||||
|
k.Add3(
|
||||||
|
fixed.Point26_6{q[i+1], q[i+2]},
|
||||||
|
fixed.Point26_6{q[i+3], q[i+4]},
|
||||||
|
fixed.Point26_6{q[i+5], q[i+6]},
|
||||||
|
)
|
||||||
|
i += 8
|
||||||
|
default:
|
||||||
|
panic("freetype/raster: bad path")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if len(k.r) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// TODO(nigeltao): if q is a closed curve then we should join the first and
|
||||||
|
// last segments instead of capping them.
|
||||||
|
k.cr.Cap(k.p, k.u, q.lastPoint(), pNeg(k.anorm))
|
||||||
|
addPathReversed(k.p, k.r)
|
||||||
|
pivot := q.firstPoint()
|
||||||
|
k.cr.Cap(k.p, k.u, pivot, pivot.Sub(fixed.Point26_6{k.r[1], k.r[2]}))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Stroke adds q stroked with the given width to p. The result is typically
|
||||||
|
// self-intersecting and should be rasterized with UseNonZeroWinding.
|
||||||
|
// cr and jr may be nil, which defaults to a RoundCapper or RoundJoiner.
|
||||||
|
func Stroke(p Adder, q Path, width fixed.Int26_6, cr Capper, jr Joiner) {
|
||||||
|
if len(q) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if cr == nil {
|
||||||
|
cr = RoundCapper
|
||||||
|
}
|
||||||
|
if jr == nil {
|
||||||
|
jr = RoundJoiner
|
||||||
|
}
|
||||||
|
if q[0] != 0 {
|
||||||
|
panic("freetype/raster: bad path")
|
||||||
|
}
|
||||||
|
s := stroker{p: p, u: width / 2, cr: cr, jr: jr}
|
||||||
|
i := 0
|
||||||
|
for j := 4; j < len(q); {
|
||||||
|
switch q[j] {
|
||||||
|
case 0:
|
||||||
|
s.stroke(q[i:j])
|
||||||
|
i, j = j, j+4
|
||||||
|
case 1:
|
||||||
|
j += 4
|
||||||
|
case 2:
|
||||||
|
j += 6
|
||||||
|
case 3:
|
||||||
|
j += 8
|
||||||
|
default:
|
||||||
|
panic("freetype/raster: bad path")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
s.stroke(q[i:])
|
||||||
|
}
|
|
@ -0,0 +1,507 @@
|
||||||
|
// Copyright 2015 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package truetype
|
||||||
|
|
||||||
|
import (
|
||||||
|
"image"
|
||||||
|
"math"
|
||||||
|
|
||||||
|
"github.com/golang/freetype/raster"
|
||||||
|
"golang.org/x/image/font"
|
||||||
|
"golang.org/x/image/math/fixed"
|
||||||
|
)
|
||||||
|
|
||||||
|
func powerOf2(i int) bool {
|
||||||
|
return i != 0 && (i&(i-1)) == 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// Options are optional arguments to NewFace.
|
||||||
|
type Options struct {
|
||||||
|
// Size is the font size in points, as in "a 10 point font size".
|
||||||
|
//
|
||||||
|
// A zero value means to use a 12 point font size.
|
||||||
|
Size float64
|
||||||
|
|
||||||
|
// DPI is the dots-per-inch resolution.
|
||||||
|
//
|
||||||
|
// A zero value means to use 72 DPI.
|
||||||
|
DPI float64
|
||||||
|
|
||||||
|
// Hinting is how to quantize the glyph nodes.
|
||||||
|
//
|
||||||
|
// A zero value means to use no hinting.
|
||||||
|
Hinting font.Hinting
|
||||||
|
|
||||||
|
// GlyphCacheEntries is the number of entries in the glyph mask image
|
||||||
|
// cache.
|
||||||
|
//
|
||||||
|
// If non-zero, it must be a power of 2.
|
||||||
|
//
|
||||||
|
// A zero value means to use 512 entries.
|
||||||
|
GlyphCacheEntries int
|
||||||
|
|
||||||
|
// SubPixelsX is the number of sub-pixel locations a glyph's dot is
|
||||||
|
// quantized to, in the horizontal direction. For example, a value of 8
|
||||||
|
// means that the dot is quantized to 1/8th of a pixel. This quantization
|
||||||
|
// only affects the glyph mask image, not its bounding box or advance
|
||||||
|
// width. A higher value gives a more faithful glyph image, but reduces the
|
||||||
|
// effectiveness of the glyph cache.
|
||||||
|
//
|
||||||
|
// If non-zero, it must be a power of 2, and be between 1 and 64 inclusive.
|
||||||
|
//
|
||||||
|
// A zero value means to use 4 sub-pixel locations.
|
||||||
|
SubPixelsX int
|
||||||
|
|
||||||
|
// SubPixelsY is the number of sub-pixel locations a glyph's dot is
|
||||||
|
// quantized to, in the vertical direction. For example, a value of 8
|
||||||
|
// means that the dot is quantized to 1/8th of a pixel. This quantization
|
||||||
|
// only affects the glyph mask image, not its bounding box or advance
|
||||||
|
// width. A higher value gives a more faithful glyph image, but reduces the
|
||||||
|
// effectiveness of the glyph cache.
|
||||||
|
//
|
||||||
|
// If non-zero, it must be a power of 2, and be between 1 and 64 inclusive.
|
||||||
|
//
|
||||||
|
// A zero value means to use 1 sub-pixel location.
|
||||||
|
SubPixelsY int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *Options) size() float64 {
|
||||||
|
if o != nil && o.Size > 0 {
|
||||||
|
return o.Size
|
||||||
|
}
|
||||||
|
return 12
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *Options) dpi() float64 {
|
||||||
|
if o != nil && o.DPI > 0 {
|
||||||
|
return o.DPI
|
||||||
|
}
|
||||||
|
return 72
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *Options) hinting() font.Hinting {
|
||||||
|
if o != nil {
|
||||||
|
switch o.Hinting {
|
||||||
|
case font.HintingVertical, font.HintingFull:
|
||||||
|
// TODO: support vertical hinting.
|
||||||
|
return font.HintingFull
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return font.HintingNone
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *Options) glyphCacheEntries() int {
|
||||||
|
if o != nil && powerOf2(o.GlyphCacheEntries) {
|
||||||
|
return o.GlyphCacheEntries
|
||||||
|
}
|
||||||
|
// 512 is 128 * 4 * 1, which lets us cache 128 glyphs at 4 * 1 subpixel
|
||||||
|
// locations in the X and Y direction.
|
||||||
|
return 512
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *Options) subPixelsX() (value uint32, halfQuantum, mask fixed.Int26_6) {
|
||||||
|
if o != nil {
|
||||||
|
switch o.SubPixelsX {
|
||||||
|
case 1, 2, 4, 8, 16, 32, 64:
|
||||||
|
return subPixels(o.SubPixelsX)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// This default value of 4 isn't based on anything scientific, merely as
|
||||||
|
// small a number as possible that looks almost as good as no quantization,
|
||||||
|
// or returning subPixels(64).
|
||||||
|
return subPixels(4)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *Options) subPixelsY() (value uint32, halfQuantum, mask fixed.Int26_6) {
|
||||||
|
if o != nil {
|
||||||
|
switch o.SubPixelsX {
|
||||||
|
case 1, 2, 4, 8, 16, 32, 64:
|
||||||
|
return subPixels(o.SubPixelsX)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// This default value of 1 isn't based on anything scientific, merely that
|
||||||
|
// vertical sub-pixel glyph rendering is pretty rare. Baseline locations
|
||||||
|
// can usually afford to snap to the pixel grid, so the vertical direction
|
||||||
|
// doesn't have the deal with the horizontal's fractional advance widths.
|
||||||
|
return subPixels(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// subPixels returns q and the bias and mask that leads to q quantized
|
||||||
|
// sub-pixel locations per full pixel.
|
||||||
|
//
|
||||||
|
// For example, q == 4 leads to a bias of 8 and a mask of 0xfffffff0, or -16,
|
||||||
|
// because we want to round fractions of fixed.Int26_6 as:
|
||||||
|
// - 0 to 7 rounds to 0.
|
||||||
|
// - 8 to 23 rounds to 16.
|
||||||
|
// - 24 to 39 rounds to 32.
|
||||||
|
// - 40 to 55 rounds to 48.
|
||||||
|
// - 56 to 63 rounds to 64.
|
||||||
|
// which means to add 8 and then bitwise-and with -16, in two's complement
|
||||||
|
// representation.
|
||||||
|
//
|
||||||
|
// When q == 1, we want bias == 32 and mask == -64.
|
||||||
|
// When q == 2, we want bias == 16 and mask == -32.
|
||||||
|
// When q == 4, we want bias == 8 and mask == -16.
|
||||||
|
// ...
|
||||||
|
// When q == 64, we want bias == 0 and mask == -1. (The no-op case).
|
||||||
|
// The pattern is clear.
|
||||||
|
func subPixels(q int) (value uint32, bias, mask fixed.Int26_6) {
|
||||||
|
return uint32(q), 32 / fixed.Int26_6(q), -64 / fixed.Int26_6(q)
|
||||||
|
}
|
||||||
|
|
||||||
|
// glyphCacheEntry caches the arguments and return values of rasterize.
|
||||||
|
type glyphCacheEntry struct {
|
||||||
|
key glyphCacheKey
|
||||||
|
val glyphCacheVal
|
||||||
|
}
|
||||||
|
|
||||||
|
type glyphCacheKey struct {
|
||||||
|
index Index
|
||||||
|
fx, fy uint8
|
||||||
|
}
|
||||||
|
|
||||||
|
type glyphCacheVal struct {
|
||||||
|
advanceWidth fixed.Int26_6
|
||||||
|
offset image.Point
|
||||||
|
gw int
|
||||||
|
gh int
|
||||||
|
}
|
||||||
|
|
||||||
|
type indexCacheEntry struct {
|
||||||
|
rune rune
|
||||||
|
index Index
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewFace returns a new font.Face for the given Font.
|
||||||
|
func NewFace(f *Font, opts *Options) font.Face {
|
||||||
|
a := &face{
|
||||||
|
f: f,
|
||||||
|
hinting: opts.hinting(),
|
||||||
|
scale: fixed.Int26_6(0.5 + (opts.size() * opts.dpi() * 64 / 72)),
|
||||||
|
glyphCache: make([]glyphCacheEntry, opts.glyphCacheEntries()),
|
||||||
|
}
|
||||||
|
a.subPixelX, a.subPixelBiasX, a.subPixelMaskX = opts.subPixelsX()
|
||||||
|
a.subPixelY, a.subPixelBiasY, a.subPixelMaskY = opts.subPixelsY()
|
||||||
|
|
||||||
|
// Fill the cache with invalid entries. Valid glyph cache entries have fx
|
||||||
|
// and fy in the range [0, 64). Valid index cache entries have rune >= 0.
|
||||||
|
for i := range a.glyphCache {
|
||||||
|
a.glyphCache[i].key.fy = 0xff
|
||||||
|
}
|
||||||
|
for i := range a.indexCache {
|
||||||
|
a.indexCache[i].rune = -1
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set the rasterizer's bounds to be big enough to handle the largest glyph.
|
||||||
|
b := f.Bounds(a.scale)
|
||||||
|
xmin := +int(b.Min.X) >> 6
|
||||||
|
ymin := -int(b.Max.Y) >> 6
|
||||||
|
xmax := +int(b.Max.X+63) >> 6
|
||||||
|
ymax := -int(b.Min.Y-63) >> 6
|
||||||
|
a.maxw = xmax - xmin
|
||||||
|
a.maxh = ymax - ymin
|
||||||
|
a.masks = image.NewAlpha(image.Rect(0, 0, a.maxw, a.maxh*len(a.glyphCache)))
|
||||||
|
a.r.SetBounds(a.maxw, a.maxh)
|
||||||
|
a.p = facePainter{a}
|
||||||
|
|
||||||
|
return a
|
||||||
|
}
|
||||||
|
|
||||||
|
type face struct {
|
||||||
|
f *Font
|
||||||
|
hinting font.Hinting
|
||||||
|
scale fixed.Int26_6
|
||||||
|
subPixelX uint32
|
||||||
|
subPixelBiasX fixed.Int26_6
|
||||||
|
subPixelMaskX fixed.Int26_6
|
||||||
|
subPixelY uint32
|
||||||
|
subPixelBiasY fixed.Int26_6
|
||||||
|
subPixelMaskY fixed.Int26_6
|
||||||
|
masks *image.Alpha
|
||||||
|
glyphCache []glyphCacheEntry
|
||||||
|
r raster.Rasterizer
|
||||||
|
p raster.Painter
|
||||||
|
paintOffset int
|
||||||
|
maxw int
|
||||||
|
maxh int
|
||||||
|
glyphBuf GlyphBuf
|
||||||
|
indexCache [indexCacheLen]indexCacheEntry
|
||||||
|
|
||||||
|
// TODO: clip rectangle?
|
||||||
|
}
|
||||||
|
|
||||||
|
const indexCacheLen = 256
|
||||||
|
|
||||||
|
func (a *face) index(r rune) Index {
|
||||||
|
const mask = indexCacheLen - 1
|
||||||
|
c := &a.indexCache[r&mask]
|
||||||
|
if c.rune == r {
|
||||||
|
return c.index
|
||||||
|
}
|
||||||
|
i := a.f.Index(r)
|
||||||
|
c.rune = r
|
||||||
|
c.index = i
|
||||||
|
return i
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close satisfies the font.Face interface.
|
||||||
|
func (a *face) Close() error { return nil }
|
||||||
|
|
||||||
|
// Metrics satisfies the font.Face interface.
|
||||||
|
func (a *face) Metrics() font.Metrics {
|
||||||
|
scale := float64(a.scale)
|
||||||
|
fupe := float64(a.f.FUnitsPerEm())
|
||||||
|
return font.Metrics{
|
||||||
|
Height: a.scale,
|
||||||
|
Ascent: fixed.Int26_6(math.Ceil(scale * float64(+a.f.ascent) / fupe)),
|
||||||
|
Descent: fixed.Int26_6(math.Ceil(scale * float64(-a.f.descent) / fupe)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Kern satisfies the font.Face interface.
|
||||||
|
func (a *face) Kern(r0, r1 rune) fixed.Int26_6 {
|
||||||
|
i0 := a.index(r0)
|
||||||
|
i1 := a.index(r1)
|
||||||
|
kern := a.f.Kern(a.scale, i0, i1)
|
||||||
|
if a.hinting != font.HintingNone {
|
||||||
|
kern = (kern + 32) &^ 63
|
||||||
|
}
|
||||||
|
return kern
|
||||||
|
}
|
||||||
|
|
||||||
|
// Glyph satisfies the font.Face interface.
|
||||||
|
func (a *face) Glyph(dot fixed.Point26_6, r rune) (
|
||||||
|
dr image.Rectangle, mask image.Image, maskp image.Point, advance fixed.Int26_6, ok bool) {
|
||||||
|
|
||||||
|
// Quantize to the sub-pixel granularity.
|
||||||
|
dotX := (dot.X + a.subPixelBiasX) & a.subPixelMaskX
|
||||||
|
dotY := (dot.Y + a.subPixelBiasY) & a.subPixelMaskY
|
||||||
|
|
||||||
|
// Split the coordinates into their integer and fractional parts.
|
||||||
|
ix, fx := int(dotX>>6), dotX&0x3f
|
||||||
|
iy, fy := int(dotY>>6), dotY&0x3f
|
||||||
|
|
||||||
|
index := a.index(r)
|
||||||
|
cIndex := uint32(index)
|
||||||
|
cIndex = cIndex*a.subPixelX - uint32(fx/a.subPixelMaskX)
|
||||||
|
cIndex = cIndex*a.subPixelY - uint32(fy/a.subPixelMaskY)
|
||||||
|
cIndex &= uint32(len(a.glyphCache) - 1)
|
||||||
|
a.paintOffset = a.maxh * int(cIndex)
|
||||||
|
k := glyphCacheKey{
|
||||||
|
index: index,
|
||||||
|
fx: uint8(fx),
|
||||||
|
fy: uint8(fy),
|
||||||
|
}
|
||||||
|
var v glyphCacheVal
|
||||||
|
if a.glyphCache[cIndex].key != k {
|
||||||
|
var ok bool
|
||||||
|
v, ok = a.rasterize(index, fx, fy)
|
||||||
|
if !ok {
|
||||||
|
return image.Rectangle{}, nil, image.Point{}, 0, false
|
||||||
|
}
|
||||||
|
a.glyphCache[cIndex] = glyphCacheEntry{k, v}
|
||||||
|
} else {
|
||||||
|
v = a.glyphCache[cIndex].val
|
||||||
|
}
|
||||||
|
|
||||||
|
dr.Min = image.Point{
|
||||||
|
X: ix + v.offset.X,
|
||||||
|
Y: iy + v.offset.Y,
|
||||||
|
}
|
||||||
|
dr.Max = image.Point{
|
||||||
|
X: dr.Min.X + v.gw,
|
||||||
|
Y: dr.Min.Y + v.gh,
|
||||||
|
}
|
||||||
|
return dr, a.masks, image.Point{Y: a.paintOffset}, v.advanceWidth, true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *face) GlyphBounds(r rune) (bounds fixed.Rectangle26_6, advance fixed.Int26_6, ok bool) {
|
||||||
|
if err := a.glyphBuf.Load(a.f, a.scale, a.index(r), a.hinting); err != nil {
|
||||||
|
return fixed.Rectangle26_6{}, 0, false
|
||||||
|
}
|
||||||
|
xmin := +a.glyphBuf.Bounds.Min.X
|
||||||
|
ymin := -a.glyphBuf.Bounds.Max.Y
|
||||||
|
xmax := +a.glyphBuf.Bounds.Max.X
|
||||||
|
ymax := -a.glyphBuf.Bounds.Min.Y
|
||||||
|
if xmin > xmax || ymin > ymax {
|
||||||
|
return fixed.Rectangle26_6{}, 0, false
|
||||||
|
}
|
||||||
|
return fixed.Rectangle26_6{
|
||||||
|
Min: fixed.Point26_6{
|
||||||
|
X: xmin,
|
||||||
|
Y: ymin,
|
||||||
|
},
|
||||||
|
Max: fixed.Point26_6{
|
||||||
|
X: xmax,
|
||||||
|
Y: ymax,
|
||||||
|
},
|
||||||
|
}, a.glyphBuf.AdvanceWidth, true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *face) GlyphAdvance(r rune) (advance fixed.Int26_6, ok bool) {
|
||||||
|
if err := a.glyphBuf.Load(a.f, a.scale, a.index(r), a.hinting); err != nil {
|
||||||
|
return 0, false
|
||||||
|
}
|
||||||
|
return a.glyphBuf.AdvanceWidth, true
|
||||||
|
}
|
||||||
|
|
||||||
|
// rasterize returns the advance width, integer-pixel offset to render at, and
|
||||||
|
// the width and height of the given glyph at the given sub-pixel offsets.
|
||||||
|
//
|
||||||
|
// The 26.6 fixed point arguments fx and fy must be in the range [0, 1).
|
||||||
|
func (a *face) rasterize(index Index, fx, fy fixed.Int26_6) (v glyphCacheVal, ok bool) {
|
||||||
|
if err := a.glyphBuf.Load(a.f, a.scale, index, a.hinting); err != nil {
|
||||||
|
return glyphCacheVal{}, false
|
||||||
|
}
|
||||||
|
// Calculate the integer-pixel bounds for the glyph.
|
||||||
|
xmin := int(fx+a.glyphBuf.Bounds.Min.X) >> 6
|
||||||
|
ymin := int(fy-a.glyphBuf.Bounds.Max.Y) >> 6
|
||||||
|
xmax := int(fx+a.glyphBuf.Bounds.Max.X+0x3f) >> 6
|
||||||
|
ymax := int(fy-a.glyphBuf.Bounds.Min.Y+0x3f) >> 6
|
||||||
|
if xmin > xmax || ymin > ymax {
|
||||||
|
return glyphCacheVal{}, false
|
||||||
|
}
|
||||||
|
// A TrueType's glyph's nodes can have negative co-ordinates, but the
|
||||||
|
// rasterizer clips anything left of x=0 or above y=0. xmin and ymin are
|
||||||
|
// the pixel offsets, based on the font's FUnit metrics, that let a
|
||||||
|
// negative co-ordinate in TrueType space be non-negative in rasterizer
|
||||||
|
// space. xmin and ymin are typically <= 0.
|
||||||
|
fx -= fixed.Int26_6(xmin << 6)
|
||||||
|
fy -= fixed.Int26_6(ymin << 6)
|
||||||
|
// Rasterize the glyph's vectors.
|
||||||
|
a.r.Clear()
|
||||||
|
pixOffset := a.paintOffset * a.maxw
|
||||||
|
clear(a.masks.Pix[pixOffset : pixOffset+a.maxw*a.maxh])
|
||||||
|
e0 := 0
|
||||||
|
for _, e1 := range a.glyphBuf.Ends {
|
||||||
|
a.drawContour(a.glyphBuf.Points[e0:e1], fx, fy)
|
||||||
|
e0 = e1
|
||||||
|
}
|
||||||
|
a.r.Rasterize(a.p)
|
||||||
|
return glyphCacheVal{
|
||||||
|
a.glyphBuf.AdvanceWidth,
|
||||||
|
image.Point{xmin, ymin},
|
||||||
|
xmax - xmin,
|
||||||
|
ymax - ymin,
|
||||||
|
}, true
|
||||||
|
}
|
||||||
|
|
||||||
|
func clear(pix []byte) {
|
||||||
|
for i := range pix {
|
||||||
|
pix[i] = 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// drawContour draws the given closed contour with the given offset.
|
||||||
|
func (a *face) drawContour(ps []Point, dx, dy fixed.Int26_6) {
|
||||||
|
if len(ps) == 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// The low bit of each point's Flags value is whether the point is on the
|
||||||
|
// curve. Truetype fonts only have quadratic Bézier curves, not cubics.
|
||||||
|
// Thus, two consecutive off-curve points imply an on-curve point in the
|
||||||
|
// middle of those two.
|
||||||
|
//
|
||||||
|
// See http://chanae.walon.org/pub/ttf/ttf_glyphs.htm for more details.
|
||||||
|
|
||||||
|
// ps[0] is a truetype.Point measured in FUnits and positive Y going
|
||||||
|
// upwards. start is the same thing measured in fixed point units and
|
||||||
|
// positive Y going downwards, and offset by (dx, dy).
|
||||||
|
start := fixed.Point26_6{
|
||||||
|
X: dx + ps[0].X,
|
||||||
|
Y: dy - ps[0].Y,
|
||||||
|
}
|
||||||
|
var others []Point
|
||||||
|
if ps[0].Flags&0x01 != 0 {
|
||||||
|
others = ps[1:]
|
||||||
|
} else {
|
||||||
|
last := fixed.Point26_6{
|
||||||
|
X: dx + ps[len(ps)-1].X,
|
||||||
|
Y: dy - ps[len(ps)-1].Y,
|
||||||
|
}
|
||||||
|
if ps[len(ps)-1].Flags&0x01 != 0 {
|
||||||
|
start = last
|
||||||
|
others = ps[:len(ps)-1]
|
||||||
|
} else {
|
||||||
|
start = fixed.Point26_6{
|
||||||
|
X: (start.X + last.X) / 2,
|
||||||
|
Y: (start.Y + last.Y) / 2,
|
||||||
|
}
|
||||||
|
others = ps
|
||||||
|
}
|
||||||
|
}
|
||||||
|
a.r.Start(start)
|
||||||
|
q0, on0 := start, true
|
||||||
|
for _, p := range others {
|
||||||
|
q := fixed.Point26_6{
|
||||||
|
X: dx + p.X,
|
||||||
|
Y: dy - p.Y,
|
||||||
|
}
|
||||||
|
on := p.Flags&0x01 != 0
|
||||||
|
if on {
|
||||||
|
if on0 {
|
||||||
|
a.r.Add1(q)
|
||||||
|
} else {
|
||||||
|
a.r.Add2(q0, q)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if on0 {
|
||||||
|
// No-op.
|
||||||
|
} else {
|
||||||
|
mid := fixed.Point26_6{
|
||||||
|
X: (q0.X + q.X) / 2,
|
||||||
|
Y: (q0.Y + q.Y) / 2,
|
||||||
|
}
|
||||||
|
a.r.Add2(q0, mid)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
q0, on0 = q, on
|
||||||
|
}
|
||||||
|
// Close the curve.
|
||||||
|
if on0 {
|
||||||
|
a.r.Add1(start)
|
||||||
|
} else {
|
||||||
|
a.r.Add2(q0, start)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// facePainter is like a raster.AlphaSrcPainter, with an additional Y offset
|
||||||
|
// (face.paintOffset) to the painted spans.
|
||||||
|
type facePainter struct {
|
||||||
|
a *face
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p facePainter) Paint(ss []raster.Span, done bool) {
|
||||||
|
m := p.a.masks
|
||||||
|
b := m.Bounds()
|
||||||
|
b.Min.Y = p.a.paintOffset
|
||||||
|
b.Max.Y = p.a.paintOffset + p.a.maxh
|
||||||
|
for _, s := range ss {
|
||||||
|
s.Y += p.a.paintOffset
|
||||||
|
if s.Y < b.Min.Y {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if s.Y >= b.Max.Y {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if s.X0 < b.Min.X {
|
||||||
|
s.X0 = b.Min.X
|
||||||
|
}
|
||||||
|
if s.X1 > b.Max.X {
|
||||||
|
s.X1 = b.Max.X
|
||||||
|
}
|
||||||
|
if s.X0 >= s.X1 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
base := (s.Y-m.Rect.Min.Y)*m.Stride - m.Rect.Min.X
|
||||||
|
p := m.Pix[base+s.X0 : base+s.X1]
|
||||||
|
color := uint8(s.Alpha >> 8)
|
||||||
|
for i := range p {
|
||||||
|
p[i] = color
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,522 @@
|
||||||
|
// Copyright 2010 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package truetype
|
||||||
|
|
||||||
|
import (
|
||||||
|
"golang.org/x/image/font"
|
||||||
|
"golang.org/x/image/math/fixed"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TODO: implement VerticalHinting.
|
||||||
|
|
||||||
|
// A Point is a co-ordinate pair plus whether it is 'on' a contour or an 'off'
|
||||||
|
// control point.
|
||||||
|
type Point struct {
|
||||||
|
X, Y fixed.Int26_6
|
||||||
|
// The Flags' LSB means whether or not this Point is 'on' the contour.
|
||||||
|
// Other bits are reserved for internal use.
|
||||||
|
Flags uint32
|
||||||
|
}
|
||||||
|
|
||||||
|
// A GlyphBuf holds a glyph's contours. A GlyphBuf can be re-used to load a
|
||||||
|
// series of glyphs from a Font.
|
||||||
|
type GlyphBuf struct {
|
||||||
|
// AdvanceWidth is the glyph's advance width.
|
||||||
|
AdvanceWidth fixed.Int26_6
|
||||||
|
// Bounds is the glyph's bounding box.
|
||||||
|
Bounds fixed.Rectangle26_6
|
||||||
|
// Points contains all Points from all contours of the glyph. If hinting
|
||||||
|
// was used to load a glyph then Unhinted contains those Points before they
|
||||||
|
// were hinted, and InFontUnits contains those Points before they were
|
||||||
|
// hinted and scaled.
|
||||||
|
Points, Unhinted, InFontUnits []Point
|
||||||
|
// Ends is the point indexes of the end point of each contour. The length
|
||||||
|
// of Ends is the number of contours in the glyph. The i'th contour
|
||||||
|
// consists of points Points[Ends[i-1]:Ends[i]], where Ends[-1] is
|
||||||
|
// interpreted to mean zero.
|
||||||
|
Ends []int
|
||||||
|
|
||||||
|
font *Font
|
||||||
|
scale fixed.Int26_6
|
||||||
|
hinting font.Hinting
|
||||||
|
hinter hinter
|
||||||
|
// phantomPoints are the co-ordinates of the synthetic phantom points
|
||||||
|
// used for hinting and bounding box calculations.
|
||||||
|
phantomPoints [4]Point
|
||||||
|
// pp1x is the X co-ordinate of the first phantom point. The '1' is
|
||||||
|
// using 1-based indexing; pp1x is almost always phantomPoints[0].X.
|
||||||
|
// TODO: eliminate this and consistently use phantomPoints[0].X.
|
||||||
|
pp1x fixed.Int26_6
|
||||||
|
// metricsSet is whether the glyph's metrics have been set yet. For a
|
||||||
|
// compound glyph, a sub-glyph may override the outer glyph's metrics.
|
||||||
|
metricsSet bool
|
||||||
|
// tmp is a scratch buffer.
|
||||||
|
tmp []Point
|
||||||
|
}
|
||||||
|
|
||||||
|
// Flags for decoding a glyph's contours. These flags are documented at
|
||||||
|
// http://developer.apple.com/fonts/TTRefMan/RM06/Chap6glyf.html.
|
||||||
|
const (
|
||||||
|
flagOnCurve = 1 << iota
|
||||||
|
flagXShortVector
|
||||||
|
flagYShortVector
|
||||||
|
flagRepeat
|
||||||
|
flagPositiveXShortVector
|
||||||
|
flagPositiveYShortVector
|
||||||
|
|
||||||
|
// The remaining flags are for internal use.
|
||||||
|
flagTouchedX
|
||||||
|
flagTouchedY
|
||||||
|
)
|
||||||
|
|
||||||
|
// The same flag bits (0x10 and 0x20) are overloaded to have two meanings,
|
||||||
|
// dependent on the value of the flag{X,Y}ShortVector bits.
|
||||||
|
const (
|
||||||
|
flagThisXIsSame = flagPositiveXShortVector
|
||||||
|
flagThisYIsSame = flagPositiveYShortVector
|
||||||
|
)
|
||||||
|
|
||||||
|
// Load loads a glyph's contours from a Font, overwriting any previously loaded
|
||||||
|
// contours for this GlyphBuf. scale is the number of 26.6 fixed point units in
|
||||||
|
// 1 em, i is the glyph index, and h is the hinting policy.
|
||||||
|
func (g *GlyphBuf) Load(f *Font, scale fixed.Int26_6, i Index, h font.Hinting) error {
|
||||||
|
g.Points = g.Points[:0]
|
||||||
|
g.Unhinted = g.Unhinted[:0]
|
||||||
|
g.InFontUnits = g.InFontUnits[:0]
|
||||||
|
g.Ends = g.Ends[:0]
|
||||||
|
g.font = f
|
||||||
|
g.hinting = h
|
||||||
|
g.scale = scale
|
||||||
|
g.pp1x = 0
|
||||||
|
g.phantomPoints = [4]Point{}
|
||||||
|
g.metricsSet = false
|
||||||
|
|
||||||
|
if h != font.HintingNone {
|
||||||
|
if err := g.hinter.init(f, scale); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err := g.load(0, i, true); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
// TODO: this selection of either g.pp1x or g.phantomPoints[0].X isn't ideal,
|
||||||
|
// and should be cleaned up once we have all the testScaling tests passing,
|
||||||
|
// plus additional tests for Freetype-Go's bounding boxes matching C Freetype's.
|
||||||
|
pp1x := g.pp1x
|
||||||
|
if h != font.HintingNone {
|
||||||
|
pp1x = g.phantomPoints[0].X
|
||||||
|
}
|
||||||
|
if pp1x != 0 {
|
||||||
|
for i := range g.Points {
|
||||||
|
g.Points[i].X -= pp1x
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
advanceWidth := g.phantomPoints[1].X - g.phantomPoints[0].X
|
||||||
|
if h != font.HintingNone {
|
||||||
|
if len(f.hdmx) >= 8 {
|
||||||
|
if n := u32(f.hdmx, 4); n > 3+uint32(i) {
|
||||||
|
for hdmx := f.hdmx[8:]; uint32(len(hdmx)) >= n; hdmx = hdmx[n:] {
|
||||||
|
if fixed.Int26_6(hdmx[0]) == scale>>6 {
|
||||||
|
advanceWidth = fixed.Int26_6(hdmx[2+i]) << 6
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
advanceWidth = (advanceWidth + 32) &^ 63
|
||||||
|
}
|
||||||
|
g.AdvanceWidth = advanceWidth
|
||||||
|
|
||||||
|
// Set g.Bounds to the 'control box', which is the bounding box of the
|
||||||
|
// Bézier curves' control points. This is easier to calculate, no smaller
|
||||||
|
// than and often equal to the tightest possible bounding box of the curves
|
||||||
|
// themselves. This approach is what C Freetype does. We can't just scale
|
||||||
|
// the nominal bounding box in the glyf data as the hinting process and
|
||||||
|
// phantom point adjustment may move points outside of that box.
|
||||||
|
if len(g.Points) == 0 {
|
||||||
|
g.Bounds = fixed.Rectangle26_6{}
|
||||||
|
} else {
|
||||||
|
p := g.Points[0]
|
||||||
|
g.Bounds.Min.X = p.X
|
||||||
|
g.Bounds.Max.X = p.X
|
||||||
|
g.Bounds.Min.Y = p.Y
|
||||||
|
g.Bounds.Max.Y = p.Y
|
||||||
|
for _, p := range g.Points[1:] {
|
||||||
|
if g.Bounds.Min.X > p.X {
|
||||||
|
g.Bounds.Min.X = p.X
|
||||||
|
} else if g.Bounds.Max.X < p.X {
|
||||||
|
g.Bounds.Max.X = p.X
|
||||||
|
}
|
||||||
|
if g.Bounds.Min.Y > p.Y {
|
||||||
|
g.Bounds.Min.Y = p.Y
|
||||||
|
} else if g.Bounds.Max.Y < p.Y {
|
||||||
|
g.Bounds.Max.Y = p.Y
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Snap the box to the grid, if hinting is on.
|
||||||
|
if h != font.HintingNone {
|
||||||
|
g.Bounds.Min.X &^= 63
|
||||||
|
g.Bounds.Min.Y &^= 63
|
||||||
|
g.Bounds.Max.X += 63
|
||||||
|
g.Bounds.Max.X &^= 63
|
||||||
|
g.Bounds.Max.Y += 63
|
||||||
|
g.Bounds.Max.Y &^= 63
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *GlyphBuf) load(recursion uint32, i Index, useMyMetrics bool) (err error) {
|
||||||
|
// The recursion limit here is arbitrary, but defends against malformed glyphs.
|
||||||
|
if recursion >= 32 {
|
||||||
|
return UnsupportedError("excessive compound glyph recursion")
|
||||||
|
}
|
||||||
|
// Find the relevant slice of g.font.glyf.
|
||||||
|
var g0, g1 uint32
|
||||||
|
if g.font.locaOffsetFormat == locaOffsetFormatShort {
|
||||||
|
g0 = 2 * uint32(u16(g.font.loca, 2*int(i)))
|
||||||
|
g1 = 2 * uint32(u16(g.font.loca, 2*int(i)+2))
|
||||||
|
} else {
|
||||||
|
g0 = u32(g.font.loca, 4*int(i))
|
||||||
|
g1 = u32(g.font.loca, 4*int(i)+4)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode the contour count and nominal bounding box, from the first
|
||||||
|
// 10 bytes of the glyf data. boundsYMin and boundsXMax, at offsets 4
|
||||||
|
// and 6, are unused.
|
||||||
|
glyf, ne, boundsXMin, boundsYMax := []byte(nil), 0, fixed.Int26_6(0), fixed.Int26_6(0)
|
||||||
|
if g0+10 <= g1 {
|
||||||
|
glyf = g.font.glyf[g0:g1]
|
||||||
|
ne = int(int16(u16(glyf, 0)))
|
||||||
|
boundsXMin = fixed.Int26_6(int16(u16(glyf, 2)))
|
||||||
|
boundsYMax = fixed.Int26_6(int16(u16(glyf, 8)))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create the phantom points.
|
||||||
|
uhm, pp1x := g.font.unscaledHMetric(i), fixed.Int26_6(0)
|
||||||
|
uvm := g.font.unscaledVMetric(i, boundsYMax)
|
||||||
|
g.phantomPoints = [4]Point{
|
||||||
|
{X: boundsXMin - uhm.LeftSideBearing},
|
||||||
|
{X: boundsXMin - uhm.LeftSideBearing + uhm.AdvanceWidth},
|
||||||
|
{X: uhm.AdvanceWidth / 2, Y: boundsYMax + uvm.TopSideBearing},
|
||||||
|
{X: uhm.AdvanceWidth / 2, Y: boundsYMax + uvm.TopSideBearing - uvm.AdvanceHeight},
|
||||||
|
}
|
||||||
|
if len(glyf) == 0 {
|
||||||
|
g.addPhantomsAndScale(len(g.Points), len(g.Points), true, true)
|
||||||
|
copy(g.phantomPoints[:], g.Points[len(g.Points)-4:])
|
||||||
|
g.Points = g.Points[:len(g.Points)-4]
|
||||||
|
// TODO: also trim g.InFontUnits and g.Unhinted?
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Load and hint the contours.
|
||||||
|
if ne < 0 {
|
||||||
|
if ne != -1 {
|
||||||
|
// http://developer.apple.com/fonts/TTRefMan/RM06/Chap6glyf.html says that
|
||||||
|
// "the values -2, -3, and so forth, are reserved for future use."
|
||||||
|
return UnsupportedError("negative number of contours")
|
||||||
|
}
|
||||||
|
pp1x = g.font.scale(g.scale * (boundsXMin - uhm.LeftSideBearing))
|
||||||
|
if err := g.loadCompound(recursion, uhm, i, glyf, useMyMetrics); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
np0, ne0 := len(g.Points), len(g.Ends)
|
||||||
|
program := g.loadSimple(glyf, ne)
|
||||||
|
g.addPhantomsAndScale(np0, np0, true, true)
|
||||||
|
pp1x = g.Points[len(g.Points)-4].X
|
||||||
|
if g.hinting != font.HintingNone {
|
||||||
|
if len(program) != 0 {
|
||||||
|
err := g.hinter.run(
|
||||||
|
program,
|
||||||
|
g.Points[np0:],
|
||||||
|
g.Unhinted[np0:],
|
||||||
|
g.InFontUnits[np0:],
|
||||||
|
g.Ends[ne0:],
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Drop the four phantom points.
|
||||||
|
g.InFontUnits = g.InFontUnits[:len(g.InFontUnits)-4]
|
||||||
|
g.Unhinted = g.Unhinted[:len(g.Unhinted)-4]
|
||||||
|
}
|
||||||
|
if useMyMetrics {
|
||||||
|
copy(g.phantomPoints[:], g.Points[len(g.Points)-4:])
|
||||||
|
}
|
||||||
|
g.Points = g.Points[:len(g.Points)-4]
|
||||||
|
if np0 != 0 {
|
||||||
|
// The hinting program expects the []Ends values to be indexed
|
||||||
|
// relative to the inner glyph, not the outer glyph, so we delay
|
||||||
|
// adding np0 until after the hinting program (if any) has run.
|
||||||
|
for i := ne0; i < len(g.Ends); i++ {
|
||||||
|
g.Ends[i] += np0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if useMyMetrics && !g.metricsSet {
|
||||||
|
g.metricsSet = true
|
||||||
|
g.pp1x = pp1x
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// loadOffset is the initial offset for loadSimple and loadCompound. The first
|
||||||
|
// 10 bytes are the number of contours and the bounding box.
|
||||||
|
const loadOffset = 10
|
||||||
|
|
||||||
|
func (g *GlyphBuf) loadSimple(glyf []byte, ne int) (program []byte) {
|
||||||
|
offset := loadOffset
|
||||||
|
for i := 0; i < ne; i++ {
|
||||||
|
g.Ends = append(g.Ends, 1+int(u16(glyf, offset)))
|
||||||
|
offset += 2
|
||||||
|
}
|
||||||
|
|
||||||
|
// Note the TrueType hinting instructions.
|
||||||
|
instrLen := int(u16(glyf, offset))
|
||||||
|
offset += 2
|
||||||
|
program = glyf[offset : offset+instrLen]
|
||||||
|
offset += instrLen
|
||||||
|
|
||||||
|
if ne == 0 {
|
||||||
|
return program
|
||||||
|
}
|
||||||
|
|
||||||
|
np0 := len(g.Points)
|
||||||
|
np1 := np0 + int(g.Ends[len(g.Ends)-1])
|
||||||
|
|
||||||
|
// Decode the flags.
|
||||||
|
for i := np0; i < np1; {
|
||||||
|
c := uint32(glyf[offset])
|
||||||
|
offset++
|
||||||
|
g.Points = append(g.Points, Point{Flags: c})
|
||||||
|
i++
|
||||||
|
if c&flagRepeat != 0 {
|
||||||
|
count := glyf[offset]
|
||||||
|
offset++
|
||||||
|
for ; count > 0; count-- {
|
||||||
|
g.Points = append(g.Points, Point{Flags: c})
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode the co-ordinates.
|
||||||
|
var x int16
|
||||||
|
for i := np0; i < np1; i++ {
|
||||||
|
f := g.Points[i].Flags
|
||||||
|
if f&flagXShortVector != 0 {
|
||||||
|
dx := int16(glyf[offset])
|
||||||
|
offset++
|
||||||
|
if f&flagPositiveXShortVector == 0 {
|
||||||
|
x -= dx
|
||||||
|
} else {
|
||||||
|
x += dx
|
||||||
|
}
|
||||||
|
} else if f&flagThisXIsSame == 0 {
|
||||||
|
x += int16(u16(glyf, offset))
|
||||||
|
offset += 2
|
||||||
|
}
|
||||||
|
g.Points[i].X = fixed.Int26_6(x)
|
||||||
|
}
|
||||||
|
var y int16
|
||||||
|
for i := np0; i < np1; i++ {
|
||||||
|
f := g.Points[i].Flags
|
||||||
|
if f&flagYShortVector != 0 {
|
||||||
|
dy := int16(glyf[offset])
|
||||||
|
offset++
|
||||||
|
if f&flagPositiveYShortVector == 0 {
|
||||||
|
y -= dy
|
||||||
|
} else {
|
||||||
|
y += dy
|
||||||
|
}
|
||||||
|
} else if f&flagThisYIsSame == 0 {
|
||||||
|
y += int16(u16(glyf, offset))
|
||||||
|
offset += 2
|
||||||
|
}
|
||||||
|
g.Points[i].Y = fixed.Int26_6(y)
|
||||||
|
}
|
||||||
|
|
||||||
|
return program
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *GlyphBuf) loadCompound(recursion uint32, uhm HMetric, i Index,
|
||||||
|
glyf []byte, useMyMetrics bool) error {
|
||||||
|
|
||||||
|
// Flags for decoding a compound glyph. These flags are documented at
|
||||||
|
// http://developer.apple.com/fonts/TTRefMan/RM06/Chap6glyf.html.
|
||||||
|
const (
|
||||||
|
flagArg1And2AreWords = 1 << iota
|
||||||
|
flagArgsAreXYValues
|
||||||
|
flagRoundXYToGrid
|
||||||
|
flagWeHaveAScale
|
||||||
|
flagUnused
|
||||||
|
flagMoreComponents
|
||||||
|
flagWeHaveAnXAndYScale
|
||||||
|
flagWeHaveATwoByTwo
|
||||||
|
flagWeHaveInstructions
|
||||||
|
flagUseMyMetrics
|
||||||
|
flagOverlapCompound
|
||||||
|
)
|
||||||
|
np0, ne0 := len(g.Points), len(g.Ends)
|
||||||
|
offset := loadOffset
|
||||||
|
for {
|
||||||
|
flags := u16(glyf, offset)
|
||||||
|
component := Index(u16(glyf, offset+2))
|
||||||
|
dx, dy, transform, hasTransform := fixed.Int26_6(0), fixed.Int26_6(0), [4]int16{}, false
|
||||||
|
if flags&flagArg1And2AreWords != 0 {
|
||||||
|
dx = fixed.Int26_6(int16(u16(glyf, offset+4)))
|
||||||
|
dy = fixed.Int26_6(int16(u16(glyf, offset+6)))
|
||||||
|
offset += 8
|
||||||
|
} else {
|
||||||
|
dx = fixed.Int26_6(int16(int8(glyf[offset+4])))
|
||||||
|
dy = fixed.Int26_6(int16(int8(glyf[offset+5])))
|
||||||
|
offset += 6
|
||||||
|
}
|
||||||
|
if flags&flagArgsAreXYValues == 0 {
|
||||||
|
return UnsupportedError("compound glyph transform vector")
|
||||||
|
}
|
||||||
|
if flags&(flagWeHaveAScale|flagWeHaveAnXAndYScale|flagWeHaveATwoByTwo) != 0 {
|
||||||
|
hasTransform = true
|
||||||
|
switch {
|
||||||
|
case flags&flagWeHaveAScale != 0:
|
||||||
|
transform[0] = int16(u16(glyf, offset+0))
|
||||||
|
transform[3] = transform[0]
|
||||||
|
offset += 2
|
||||||
|
case flags&flagWeHaveAnXAndYScale != 0:
|
||||||
|
transform[0] = int16(u16(glyf, offset+0))
|
||||||
|
transform[3] = int16(u16(glyf, offset+2))
|
||||||
|
offset += 4
|
||||||
|
case flags&flagWeHaveATwoByTwo != 0:
|
||||||
|
transform[0] = int16(u16(glyf, offset+0))
|
||||||
|
transform[1] = int16(u16(glyf, offset+2))
|
||||||
|
transform[2] = int16(u16(glyf, offset+4))
|
||||||
|
transform[3] = int16(u16(glyf, offset+6))
|
||||||
|
offset += 8
|
||||||
|
}
|
||||||
|
}
|
||||||
|
savedPP := g.phantomPoints
|
||||||
|
np0 := len(g.Points)
|
||||||
|
componentUMM := useMyMetrics && (flags&flagUseMyMetrics != 0)
|
||||||
|
if err := g.load(recursion+1, component, componentUMM); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if flags&flagUseMyMetrics == 0 {
|
||||||
|
g.phantomPoints = savedPP
|
||||||
|
}
|
||||||
|
if hasTransform {
|
||||||
|
for j := np0; j < len(g.Points); j++ {
|
||||||
|
p := &g.Points[j]
|
||||||
|
newX := 0 +
|
||||||
|
fixed.Int26_6((int64(p.X)*int64(transform[0])+1<<13)>>14) +
|
||||||
|
fixed.Int26_6((int64(p.Y)*int64(transform[2])+1<<13)>>14)
|
||||||
|
newY := 0 +
|
||||||
|
fixed.Int26_6((int64(p.X)*int64(transform[1])+1<<13)>>14) +
|
||||||
|
fixed.Int26_6((int64(p.Y)*int64(transform[3])+1<<13)>>14)
|
||||||
|
p.X, p.Y = newX, newY
|
||||||
|
}
|
||||||
|
}
|
||||||
|
dx = g.font.scale(g.scale * dx)
|
||||||
|
dy = g.font.scale(g.scale * dy)
|
||||||
|
if flags&flagRoundXYToGrid != 0 {
|
||||||
|
dx = (dx + 32) &^ 63
|
||||||
|
dy = (dy + 32) &^ 63
|
||||||
|
}
|
||||||
|
for j := np0; j < len(g.Points); j++ {
|
||||||
|
p := &g.Points[j]
|
||||||
|
p.X += dx
|
||||||
|
p.Y += dy
|
||||||
|
}
|
||||||
|
// TODO: also adjust g.InFontUnits and g.Unhinted?
|
||||||
|
if flags&flagMoreComponents == 0 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
instrLen := 0
|
||||||
|
if g.hinting != font.HintingNone && offset+2 <= len(glyf) {
|
||||||
|
instrLen = int(u16(glyf, offset))
|
||||||
|
offset += 2
|
||||||
|
}
|
||||||
|
|
||||||
|
g.addPhantomsAndScale(np0, len(g.Points), false, instrLen > 0)
|
||||||
|
points, ends := g.Points[np0:], g.Ends[ne0:]
|
||||||
|
g.Points = g.Points[:len(g.Points)-4]
|
||||||
|
for j := range points {
|
||||||
|
points[j].Flags &^= flagTouchedX | flagTouchedY
|
||||||
|
}
|
||||||
|
|
||||||
|
if instrLen == 0 {
|
||||||
|
if !g.metricsSet {
|
||||||
|
copy(g.phantomPoints[:], points[len(points)-4:])
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Hint the compound glyph.
|
||||||
|
program := glyf[offset : offset+instrLen]
|
||||||
|
// Temporarily adjust the ends to be relative to this compound glyph.
|
||||||
|
if np0 != 0 {
|
||||||
|
for i := range ends {
|
||||||
|
ends[i] -= np0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Hinting instructions of a composite glyph completely refer to the
|
||||||
|
// (already) hinted subglyphs.
|
||||||
|
g.tmp = append(g.tmp[:0], points...)
|
||||||
|
if err := g.hinter.run(program, points, g.tmp, g.tmp, ends); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if np0 != 0 {
|
||||||
|
for i := range ends {
|
||||||
|
ends[i] += np0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !g.metricsSet {
|
||||||
|
copy(g.phantomPoints[:], points[len(points)-4:])
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *GlyphBuf) addPhantomsAndScale(np0, np1 int, simple, adjust bool) {
|
||||||
|
// Add the four phantom points.
|
||||||
|
g.Points = append(g.Points, g.phantomPoints[:]...)
|
||||||
|
// Scale the points.
|
||||||
|
if simple && g.hinting != font.HintingNone {
|
||||||
|
g.InFontUnits = append(g.InFontUnits, g.Points[np1:]...)
|
||||||
|
}
|
||||||
|
for i := np1; i < len(g.Points); i++ {
|
||||||
|
p := &g.Points[i]
|
||||||
|
p.X = g.font.scale(g.scale * p.X)
|
||||||
|
p.Y = g.font.scale(g.scale * p.Y)
|
||||||
|
}
|
||||||
|
if g.hinting == font.HintingNone {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// Round the 1st phantom point to the grid, shifting all other points equally.
|
||||||
|
// Note that "all other points" starts from np0, not np1.
|
||||||
|
// TODO: delete this adjustment and the np0/np1 distinction, when
|
||||||
|
// we update the compatibility tests to C Freetype 2.5.3.
|
||||||
|
// See http://git.savannah.gnu.org/cgit/freetype/freetype2.git/commit/?id=05c786d990390a7ca18e62962641dac740bacb06
|
||||||
|
if adjust {
|
||||||
|
pp1x := g.Points[len(g.Points)-4].X
|
||||||
|
if dx := ((pp1x + 32) &^ 63) - pp1x; dx != 0 {
|
||||||
|
for i := np0; i < len(g.Points); i++ {
|
||||||
|
g.Points[i].X += dx
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if simple {
|
||||||
|
g.Unhinted = append(g.Unhinted, g.Points[np1:]...)
|
||||||
|
}
|
||||||
|
// Round the 2nd and 4th phantom point to the grid.
|
||||||
|
p := &g.Points[len(g.Points)-3]
|
||||||
|
p.X = (p.X + 32) &^ 63
|
||||||
|
p = &g.Points[len(g.Points)-1]
|
||||||
|
p.Y = (p.Y + 32) &^ 63
|
||||||
|
}
|
File diff suppressed because it is too large
Load Diff
|
@ -0,0 +1,289 @@
|
||||||
|
// Copyright 2012 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package truetype
|
||||||
|
|
||||||
|
// The Truetype opcodes are summarized at
|
||||||
|
// https://developer.apple.com/fonts/TTRefMan/RM07/appendixA.html
|
||||||
|
|
||||||
|
const (
|
||||||
|
opSVTCA0 = 0x00 // Set freedom and projection Vectors To Coordinate Axis
|
||||||
|
opSVTCA1 = 0x01 // .
|
||||||
|
opSPVTCA0 = 0x02 // Set Projection Vector To Coordinate Axis
|
||||||
|
opSPVTCA1 = 0x03 // .
|
||||||
|
opSFVTCA0 = 0x04 // Set Freedom Vector to Coordinate Axis
|
||||||
|
opSFVTCA1 = 0x05 // .
|
||||||
|
opSPVTL0 = 0x06 // Set Projection Vector To Line
|
||||||
|
opSPVTL1 = 0x07 // .
|
||||||
|
opSFVTL0 = 0x08 // Set Freedom Vector To Line
|
||||||
|
opSFVTL1 = 0x09 // .
|
||||||
|
opSPVFS = 0x0a // Set Projection Vector From Stack
|
||||||
|
opSFVFS = 0x0b // Set Freedom Vector From Stack
|
||||||
|
opGPV = 0x0c // Get Projection Vector
|
||||||
|
opGFV = 0x0d // Get Freedom Vector
|
||||||
|
opSFVTPV = 0x0e // Set Freedom Vector To Projection Vector
|
||||||
|
opISECT = 0x0f // moves point p to the InterSECTion of two lines
|
||||||
|
opSRP0 = 0x10 // Set Reference Point 0
|
||||||
|
opSRP1 = 0x11 // Set Reference Point 1
|
||||||
|
opSRP2 = 0x12 // Set Reference Point 2
|
||||||
|
opSZP0 = 0x13 // Set Zone Pointer 0
|
||||||
|
opSZP1 = 0x14 // Set Zone Pointer 1
|
||||||
|
opSZP2 = 0x15 // Set Zone Pointer 2
|
||||||
|
opSZPS = 0x16 // Set Zone PointerS
|
||||||
|
opSLOOP = 0x17 // Set LOOP variable
|
||||||
|
opRTG = 0x18 // Round To Grid
|
||||||
|
opRTHG = 0x19 // Round To Half Grid
|
||||||
|
opSMD = 0x1a // Set Minimum Distance
|
||||||
|
opELSE = 0x1b // ELSE clause
|
||||||
|
opJMPR = 0x1c // JuMP Relative
|
||||||
|
opSCVTCI = 0x1d // Set Control Value Table Cut-In
|
||||||
|
opSSWCI = 0x1e // Set Single Width Cut-In
|
||||||
|
opSSW = 0x1f // Set Single Width
|
||||||
|
opDUP = 0x20 // DUPlicate top stack element
|
||||||
|
opPOP = 0x21 // POP top stack element
|
||||||
|
opCLEAR = 0x22 // CLEAR the stack
|
||||||
|
opSWAP = 0x23 // SWAP the top two elements on the stack
|
||||||
|
opDEPTH = 0x24 // DEPTH of the stack
|
||||||
|
opCINDEX = 0x25 // Copy the INDEXed element to the top of the stack
|
||||||
|
opMINDEX = 0x26 // Move the INDEXed element to the top of the stack
|
||||||
|
opALIGNPTS = 0x27 // ALIGN PoinTS
|
||||||
|
op_0x28 = 0x28 // deprecated
|
||||||
|
opUTP = 0x29 // UnTouch Point
|
||||||
|
opLOOPCALL = 0x2a // LOOP and CALL function
|
||||||
|
opCALL = 0x2b // CALL function
|
||||||
|
opFDEF = 0x2c // Function DEFinition
|
||||||
|
opENDF = 0x2d // END Function definition
|
||||||
|
opMDAP0 = 0x2e // Move Direct Absolute Point
|
||||||
|
opMDAP1 = 0x2f // .
|
||||||
|
opIUP0 = 0x30 // Interpolate Untouched Points through the outline
|
||||||
|
opIUP1 = 0x31 // .
|
||||||
|
opSHP0 = 0x32 // SHift Point using reference point
|
||||||
|
opSHP1 = 0x33 // .
|
||||||
|
opSHC0 = 0x34 // SHift Contour using reference point
|
||||||
|
opSHC1 = 0x35 // .
|
||||||
|
opSHZ0 = 0x36 // SHift Zone using reference point
|
||||||
|
opSHZ1 = 0x37 // .
|
||||||
|
opSHPIX = 0x38 // SHift point by a PIXel amount
|
||||||
|
opIP = 0x39 // Interpolate Point
|
||||||
|
opMSIRP0 = 0x3a // Move Stack Indirect Relative Point
|
||||||
|
opMSIRP1 = 0x3b // .
|
||||||
|
opALIGNRP = 0x3c // ALIGN to Reference Point
|
||||||
|
opRTDG = 0x3d // Round To Double Grid
|
||||||
|
opMIAP0 = 0x3e // Move Indirect Absolute Point
|
||||||
|
opMIAP1 = 0x3f // .
|
||||||
|
opNPUSHB = 0x40 // PUSH N Bytes
|
||||||
|
opNPUSHW = 0x41 // PUSH N Words
|
||||||
|
opWS = 0x42 // Write Store
|
||||||
|
opRS = 0x43 // Read Store
|
||||||
|
opWCVTP = 0x44 // Write Control Value Table in Pixel units
|
||||||
|
opRCVT = 0x45 // Read Control Value Table entry
|
||||||
|
opGC0 = 0x46 // Get Coordinate projected onto the projection vector
|
||||||
|
opGC1 = 0x47 // .
|
||||||
|
opSCFS = 0x48 // Sets Coordinate From the Stack using projection vector and freedom vector
|
||||||
|
opMD0 = 0x49 // Measure Distance
|
||||||
|
opMD1 = 0x4a // .
|
||||||
|
opMPPEM = 0x4b // Measure Pixels Per EM
|
||||||
|
opMPS = 0x4c // Measure Point Size
|
||||||
|
opFLIPON = 0x4d // set the auto FLIP Boolean to ON
|
||||||
|
opFLIPOFF = 0x4e // set the auto FLIP Boolean to OFF
|
||||||
|
opDEBUG = 0x4f // DEBUG call
|
||||||
|
opLT = 0x50 // Less Than
|
||||||
|
opLTEQ = 0x51 // Less Than or EQual
|
||||||
|
opGT = 0x52 // Greater Than
|
||||||
|
opGTEQ = 0x53 // Greater Than or EQual
|
||||||
|
opEQ = 0x54 // EQual
|
||||||
|
opNEQ = 0x55 // Not EQual
|
||||||
|
opODD = 0x56 // ODD
|
||||||
|
opEVEN = 0x57 // EVEN
|
||||||
|
opIF = 0x58 // IF test
|
||||||
|
opEIF = 0x59 // End IF
|
||||||
|
opAND = 0x5a // logical AND
|
||||||
|
opOR = 0x5b // logical OR
|
||||||
|
opNOT = 0x5c // logical NOT
|
||||||
|
opDELTAP1 = 0x5d // DELTA exception P1
|
||||||
|
opSDB = 0x5e // Set Delta Base in the graphics state
|
||||||
|
opSDS = 0x5f // Set Delta Shift in the graphics state
|
||||||
|
opADD = 0x60 // ADD
|
||||||
|
opSUB = 0x61 // SUBtract
|
||||||
|
opDIV = 0x62 // DIVide
|
||||||
|
opMUL = 0x63 // MULtiply
|
||||||
|
opABS = 0x64 // ABSolute value
|
||||||
|
opNEG = 0x65 // NEGate
|
||||||
|
opFLOOR = 0x66 // FLOOR
|
||||||
|
opCEILING = 0x67 // CEILING
|
||||||
|
opROUND00 = 0x68 // ROUND value
|
||||||
|
opROUND01 = 0x69 // .
|
||||||
|
opROUND10 = 0x6a // .
|
||||||
|
opROUND11 = 0x6b // .
|
||||||
|
opNROUND00 = 0x6c // No ROUNDing of value
|
||||||
|
opNROUND01 = 0x6d // .
|
||||||
|
opNROUND10 = 0x6e // .
|
||||||
|
opNROUND11 = 0x6f // .
|
||||||
|
opWCVTF = 0x70 // Write Control Value Table in Funits
|
||||||
|
opDELTAP2 = 0x71 // DELTA exception P2
|
||||||
|
opDELTAP3 = 0x72 // DELTA exception P3
|
||||||
|
opDELTAC1 = 0x73 // DELTA exception C1
|
||||||
|
opDELTAC2 = 0x74 // DELTA exception C2
|
||||||
|
opDELTAC3 = 0x75 // DELTA exception C3
|
||||||
|
opSROUND = 0x76 // Super ROUND
|
||||||
|
opS45ROUND = 0x77 // Super ROUND 45 degrees
|
||||||
|
opJROT = 0x78 // Jump Relative On True
|
||||||
|
opJROF = 0x79 // Jump Relative On False
|
||||||
|
opROFF = 0x7a // Round OFF
|
||||||
|
op_0x7b = 0x7b // deprecated
|
||||||
|
opRUTG = 0x7c // Round Up To Grid
|
||||||
|
opRDTG = 0x7d // Round Down To Grid
|
||||||
|
opSANGW = 0x7e // Set ANGle Weight
|
||||||
|
opAA = 0x7f // Adjust Angle
|
||||||
|
opFLIPPT = 0x80 // FLIP PoinT
|
||||||
|
opFLIPRGON = 0x81 // FLIP RanGe ON
|
||||||
|
opFLIPRGOFF = 0x82 // FLIP RanGe OFF
|
||||||
|
op_0x83 = 0x83 // deprecated
|
||||||
|
op_0x84 = 0x84 // deprecated
|
||||||
|
opSCANCTRL = 0x85 // SCAN conversion ConTRoL
|
||||||
|
opSDPVTL0 = 0x86 // Set Dual Projection Vector To Line
|
||||||
|
opSDPVTL1 = 0x87 // .
|
||||||
|
opGETINFO = 0x88 // GET INFOrmation
|
||||||
|
opIDEF = 0x89 // Instruction DEFinition
|
||||||
|
opROLL = 0x8a // ROLL the top three stack elements
|
||||||
|
opMAX = 0x8b // MAXimum of top two stack elements
|
||||||
|
opMIN = 0x8c // MINimum of top two stack elements
|
||||||
|
opSCANTYPE = 0x8d // SCANTYPE
|
||||||
|
opINSTCTRL = 0x8e // INSTRuction execution ConTRoL
|
||||||
|
op_0x8f = 0x8f
|
||||||
|
op_0x90 = 0x90
|
||||||
|
op_0x91 = 0x91
|
||||||
|
op_0x92 = 0x92
|
||||||
|
op_0x93 = 0x93
|
||||||
|
op_0x94 = 0x94
|
||||||
|
op_0x95 = 0x95
|
||||||
|
op_0x96 = 0x96
|
||||||
|
op_0x97 = 0x97
|
||||||
|
op_0x98 = 0x98
|
||||||
|
op_0x99 = 0x99
|
||||||
|
op_0x9a = 0x9a
|
||||||
|
op_0x9b = 0x9b
|
||||||
|
op_0x9c = 0x9c
|
||||||
|
op_0x9d = 0x9d
|
||||||
|
op_0x9e = 0x9e
|
||||||
|
op_0x9f = 0x9f
|
||||||
|
op_0xa0 = 0xa0
|
||||||
|
op_0xa1 = 0xa1
|
||||||
|
op_0xa2 = 0xa2
|
||||||
|
op_0xa3 = 0xa3
|
||||||
|
op_0xa4 = 0xa4
|
||||||
|
op_0xa5 = 0xa5
|
||||||
|
op_0xa6 = 0xa6
|
||||||
|
op_0xa7 = 0xa7
|
||||||
|
op_0xa8 = 0xa8
|
||||||
|
op_0xa9 = 0xa9
|
||||||
|
op_0xaa = 0xaa
|
||||||
|
op_0xab = 0xab
|
||||||
|
op_0xac = 0xac
|
||||||
|
op_0xad = 0xad
|
||||||
|
op_0xae = 0xae
|
||||||
|
op_0xaf = 0xaf
|
||||||
|
opPUSHB000 = 0xb0 // PUSH Bytes
|
||||||
|
opPUSHB001 = 0xb1 // .
|
||||||
|
opPUSHB010 = 0xb2 // .
|
||||||
|
opPUSHB011 = 0xb3 // .
|
||||||
|
opPUSHB100 = 0xb4 // .
|
||||||
|
opPUSHB101 = 0xb5 // .
|
||||||
|
opPUSHB110 = 0xb6 // .
|
||||||
|
opPUSHB111 = 0xb7 // .
|
||||||
|
opPUSHW000 = 0xb8 // PUSH Words
|
||||||
|
opPUSHW001 = 0xb9 // .
|
||||||
|
opPUSHW010 = 0xba // .
|
||||||
|
opPUSHW011 = 0xbb // .
|
||||||
|
opPUSHW100 = 0xbc // .
|
||||||
|
opPUSHW101 = 0xbd // .
|
||||||
|
opPUSHW110 = 0xbe // .
|
||||||
|
opPUSHW111 = 0xbf // .
|
||||||
|
opMDRP00000 = 0xc0 // Move Direct Relative Point
|
||||||
|
opMDRP00001 = 0xc1 // .
|
||||||
|
opMDRP00010 = 0xc2 // .
|
||||||
|
opMDRP00011 = 0xc3 // .
|
||||||
|
opMDRP00100 = 0xc4 // .
|
||||||
|
opMDRP00101 = 0xc5 // .
|
||||||
|
opMDRP00110 = 0xc6 // .
|
||||||
|
opMDRP00111 = 0xc7 // .
|
||||||
|
opMDRP01000 = 0xc8 // .
|
||||||
|
opMDRP01001 = 0xc9 // .
|
||||||
|
opMDRP01010 = 0xca // .
|
||||||
|
opMDRP01011 = 0xcb // .
|
||||||
|
opMDRP01100 = 0xcc // .
|
||||||
|
opMDRP01101 = 0xcd // .
|
||||||
|
opMDRP01110 = 0xce // .
|
||||||
|
opMDRP01111 = 0xcf // .
|
||||||
|
opMDRP10000 = 0xd0 // .
|
||||||
|
opMDRP10001 = 0xd1 // .
|
||||||
|
opMDRP10010 = 0xd2 // .
|
||||||
|
opMDRP10011 = 0xd3 // .
|
||||||
|
opMDRP10100 = 0xd4 // .
|
||||||
|
opMDRP10101 = 0xd5 // .
|
||||||
|
opMDRP10110 = 0xd6 // .
|
||||||
|
opMDRP10111 = 0xd7 // .
|
||||||
|
opMDRP11000 = 0xd8 // .
|
||||||
|
opMDRP11001 = 0xd9 // .
|
||||||
|
opMDRP11010 = 0xda // .
|
||||||
|
opMDRP11011 = 0xdb // .
|
||||||
|
opMDRP11100 = 0xdc // .
|
||||||
|
opMDRP11101 = 0xdd // .
|
||||||
|
opMDRP11110 = 0xde // .
|
||||||
|
opMDRP11111 = 0xdf // .
|
||||||
|
opMIRP00000 = 0xe0 // Move Indirect Relative Point
|
||||||
|
opMIRP00001 = 0xe1 // .
|
||||||
|
opMIRP00010 = 0xe2 // .
|
||||||
|
opMIRP00011 = 0xe3 // .
|
||||||
|
opMIRP00100 = 0xe4 // .
|
||||||
|
opMIRP00101 = 0xe5 // .
|
||||||
|
opMIRP00110 = 0xe6 // .
|
||||||
|
opMIRP00111 = 0xe7 // .
|
||||||
|
opMIRP01000 = 0xe8 // .
|
||||||
|
opMIRP01001 = 0xe9 // .
|
||||||
|
opMIRP01010 = 0xea // .
|
||||||
|
opMIRP01011 = 0xeb // .
|
||||||
|
opMIRP01100 = 0xec // .
|
||||||
|
opMIRP01101 = 0xed // .
|
||||||
|
opMIRP01110 = 0xee // .
|
||||||
|
opMIRP01111 = 0xef // .
|
||||||
|
opMIRP10000 = 0xf0 // .
|
||||||
|
opMIRP10001 = 0xf1 // .
|
||||||
|
opMIRP10010 = 0xf2 // .
|
||||||
|
opMIRP10011 = 0xf3 // .
|
||||||
|
opMIRP10100 = 0xf4 // .
|
||||||
|
opMIRP10101 = 0xf5 // .
|
||||||
|
opMIRP10110 = 0xf6 // .
|
||||||
|
opMIRP10111 = 0xf7 // .
|
||||||
|
opMIRP11000 = 0xf8 // .
|
||||||
|
opMIRP11001 = 0xf9 // .
|
||||||
|
opMIRP11010 = 0xfa // .
|
||||||
|
opMIRP11011 = 0xfb // .
|
||||||
|
opMIRP11100 = 0xfc // .
|
||||||
|
opMIRP11101 = 0xfd // .
|
||||||
|
opMIRP11110 = 0xfe // .
|
||||||
|
opMIRP11111 = 0xff // .
|
||||||
|
)
|
||||||
|
|
||||||
|
// popCount is the number of stack elements that each opcode pops.
|
||||||
|
var popCount = [256]uint8{
|
||||||
|
// 1, 2, 3, 4, 5, 6, 7, 8, 9, a, b, c, d, e, f
|
||||||
|
0, 0, 0, 0, 0, 0, 2, 2, 2, 2, 2, 2, 0, 0, 0, 5, // 0x00 - 0x0f
|
||||||
|
1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 0, 1, 1, 1, 1, // 0x10 - 0x1f
|
||||||
|
1, 1, 0, 2, 0, 1, 1, 2, 0, 1, 2, 1, 1, 0, 1, 1, // 0x20 - 0x2f
|
||||||
|
0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 2, 2, 0, 0, 2, 2, // 0x30 - 0x3f
|
||||||
|
0, 0, 2, 1, 2, 1, 1, 1, 2, 2, 2, 0, 0, 0, 0, 0, // 0x40 - 0x4f
|
||||||
|
2, 2, 2, 2, 2, 2, 1, 1, 1, 0, 2, 2, 1, 1, 1, 1, // 0x50 - 0x5f
|
||||||
|
2, 2, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, // 0x60 - 0x6f
|
||||||
|
2, 1, 1, 1, 1, 1, 1, 1, 2, 2, 0, 0, 0, 0, 1, 1, // 0x70 - 0x7f
|
||||||
|
0, 2, 2, 0, 0, 1, 2, 2, 1, 1, 3, 2, 2, 1, 2, 0, // 0x80 - 0x8f
|
||||||
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // 0x90 - 0x9f
|
||||||
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // 0xa0 - 0xaf
|
||||||
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // 0xb0 - 0xbf
|
||||||
|
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, // 0xc0 - 0xcf
|
||||||
|
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, // 0xd0 - 0xdf
|
||||||
|
2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, // 0xe0 - 0xef
|
||||||
|
2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, // 0xf0 - 0xff
|
||||||
|
}
|
|
@ -0,0 +1,643 @@
|
||||||
|
// Copyright 2010 The Freetype-Go Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by your choice of either the
|
||||||
|
// FreeType License or the GNU General Public License version 2 (or
|
||||||
|
// any later version), both of which can be found in the LICENSE file.
|
||||||
|
|
||||||
|
// Package truetype provides a parser for the TTF and TTC file formats.
|
||||||
|
// Those formats are documented at http://developer.apple.com/fonts/TTRefMan/
|
||||||
|
// and http://www.microsoft.com/typography/otspec/
|
||||||
|
//
|
||||||
|
// Some of a font's methods provide lengths or co-ordinates, e.g. bounds, font
|
||||||
|
// metrics and control points. All these methods take a scale parameter, which
|
||||||
|
// is the number of pixels in 1 em, expressed as a 26.6 fixed point value. For
|
||||||
|
// example, if 1 em is 10 pixels then scale is fixed.I(10), which is equal to
|
||||||
|
// fixed.Int26_6(10 << 6).
|
||||||
|
//
|
||||||
|
// To measure a TrueType font in ideal FUnit space, use scale equal to
|
||||||
|
// font.FUnitsPerEm().
|
||||||
|
package truetype
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"golang.org/x/image/math/fixed"
|
||||||
|
)
|
||||||
|
|
||||||
|
// An Index is a Font's index of a rune.
|
||||||
|
type Index uint16
|
||||||
|
|
||||||
|
// A NameID identifies a name table entry.
|
||||||
|
//
|
||||||
|
// See https://developer.apple.com/fonts/TrueType-Reference-Manual/RM06/Chap6name.html
|
||||||
|
type NameID uint16
|
||||||
|
|
||||||
|
const (
|
||||||
|
NameIDCopyright NameID = 0
|
||||||
|
NameIDFontFamily = 1
|
||||||
|
NameIDFontSubfamily = 2
|
||||||
|
NameIDUniqueSubfamilyID = 3
|
||||||
|
NameIDFontFullName = 4
|
||||||
|
NameIDNameTableVersion = 5
|
||||||
|
NameIDPostscriptName = 6
|
||||||
|
NameIDTrademarkNotice = 7
|
||||||
|
NameIDManufacturerName = 8
|
||||||
|
NameIDDesignerName = 9
|
||||||
|
NameIDFontDescription = 10
|
||||||
|
NameIDFontVendorURL = 11
|
||||||
|
NameIDFontDesignerURL = 12
|
||||||
|
NameIDFontLicense = 13
|
||||||
|
NameIDFontLicenseURL = 14
|
||||||
|
NameIDPreferredFamily = 16
|
||||||
|
NameIDPreferredSubfamily = 17
|
||||||
|
NameIDCompatibleName = 18
|
||||||
|
NameIDSampleText = 19
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// A 32-bit encoding consists of a most-significant 16-bit Platform ID and a
|
||||||
|
// least-significant 16-bit Platform Specific ID. The magic numbers are
|
||||||
|
// specified at https://www.microsoft.com/typography/otspec/name.htm
|
||||||
|
unicodeEncoding = 0x00000003 // PID = 0 (Unicode), PSID = 3 (Unicode 2.0)
|
||||||
|
microsoftSymbolEncoding = 0x00030000 // PID = 3 (Microsoft), PSID = 0 (Symbol)
|
||||||
|
microsoftUCS2Encoding = 0x00030001 // PID = 3 (Microsoft), PSID = 1 (UCS-2)
|
||||||
|
microsoftUCS4Encoding = 0x0003000a // PID = 3 (Microsoft), PSID = 10 (UCS-4)
|
||||||
|
)
|
||||||
|
|
||||||
|
// An HMetric holds the horizontal metrics of a single glyph.
|
||||||
|
type HMetric struct {
|
||||||
|
AdvanceWidth, LeftSideBearing fixed.Int26_6
|
||||||
|
}
|
||||||
|
|
||||||
|
// A VMetric holds the vertical metrics of a single glyph.
|
||||||
|
type VMetric struct {
|
||||||
|
AdvanceHeight, TopSideBearing fixed.Int26_6
|
||||||
|
}
|
||||||
|
|
||||||
|
// A FormatError reports that the input is not a valid TrueType font.
|
||||||
|
type FormatError string
|
||||||
|
|
||||||
|
func (e FormatError) Error() string {
|
||||||
|
return "freetype: invalid TrueType format: " + string(e)
|
||||||
|
}
|
||||||
|
|
||||||
|
// An UnsupportedError reports that the input uses a valid but unimplemented
|
||||||
|
// TrueType feature.
|
||||||
|
type UnsupportedError string
|
||||||
|
|
||||||
|
func (e UnsupportedError) Error() string {
|
||||||
|
return "freetype: unsupported TrueType feature: " + string(e)
|
||||||
|
}
|
||||||
|
|
||||||
|
// u32 returns the big-endian uint32 at b[i:].
|
||||||
|
func u32(b []byte, i int) uint32 {
|
||||||
|
return uint32(b[i])<<24 | uint32(b[i+1])<<16 | uint32(b[i+2])<<8 | uint32(b[i+3])
|
||||||
|
}
|
||||||
|
|
||||||
|
// u16 returns the big-endian uint16 at b[i:].
|
||||||
|
func u16(b []byte, i int) uint16 {
|
||||||
|
return uint16(b[i])<<8 | uint16(b[i+1])
|
||||||
|
}
|
||||||
|
|
||||||
|
// readTable returns a slice of the TTF data given by a table's directory entry.
|
||||||
|
func readTable(ttf []byte, offsetLength []byte) ([]byte, error) {
|
||||||
|
offset := int(u32(offsetLength, 0))
|
||||||
|
if offset < 0 {
|
||||||
|
return nil, FormatError(fmt.Sprintf("offset too large: %d", uint32(offset)))
|
||||||
|
}
|
||||||
|
length := int(u32(offsetLength, 4))
|
||||||
|
if length < 0 {
|
||||||
|
return nil, FormatError(fmt.Sprintf("length too large: %d", uint32(length)))
|
||||||
|
}
|
||||||
|
end := offset + length
|
||||||
|
if end < 0 || end > len(ttf) {
|
||||||
|
return nil, FormatError(fmt.Sprintf("offset + length too large: %d", uint32(offset)+uint32(length)))
|
||||||
|
}
|
||||||
|
return ttf[offset:end], nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// parseSubtables returns the offset and platformID of the best subtable in
|
||||||
|
// table, where best favors a Unicode cmap encoding, and failing that, a
|
||||||
|
// Microsoft cmap encoding. offset is the offset of the first subtable in
|
||||||
|
// table, and size is the size of each subtable.
|
||||||
|
//
|
||||||
|
// If pred is non-nil, then only subtables that satisfy that predicate will be
|
||||||
|
// considered.
|
||||||
|
func parseSubtables(table []byte, name string, offset, size int, pred func([]byte) bool) (
|
||||||
|
bestOffset int, bestPID uint32, retErr error) {
|
||||||
|
|
||||||
|
if len(table) < 4 {
|
||||||
|
return 0, 0, FormatError(name + " too short")
|
||||||
|
}
|
||||||
|
nSubtables := int(u16(table, 2))
|
||||||
|
if len(table) < size*nSubtables+offset {
|
||||||
|
return 0, 0, FormatError(name + " too short")
|
||||||
|
}
|
||||||
|
ok := false
|
||||||
|
for i := 0; i < nSubtables; i, offset = i+1, offset+size {
|
||||||
|
if pred != nil && !pred(table[offset:]) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// We read the 16-bit Platform ID and 16-bit Platform Specific ID as a single uint32.
|
||||||
|
// All values are big-endian.
|
||||||
|
pidPsid := u32(table, offset)
|
||||||
|
// We prefer the Unicode cmap encoding. Failing to find that, we fall
|
||||||
|
// back onto the Microsoft cmap encoding.
|
||||||
|
if pidPsid == unicodeEncoding {
|
||||||
|
bestOffset, bestPID, ok = offset, pidPsid>>16, true
|
||||||
|
break
|
||||||
|
|
||||||
|
} else if pidPsid == microsoftSymbolEncoding ||
|
||||||
|
pidPsid == microsoftUCS2Encoding ||
|
||||||
|
pidPsid == microsoftUCS4Encoding {
|
||||||
|
|
||||||
|
bestOffset, bestPID, ok = offset, pidPsid>>16, true
|
||||||
|
// We don't break out of the for loop, so that Unicode can override Microsoft.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !ok {
|
||||||
|
return 0, 0, UnsupportedError(name + " encoding")
|
||||||
|
}
|
||||||
|
return bestOffset, bestPID, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
locaOffsetFormatUnknown int = iota
|
||||||
|
locaOffsetFormatShort
|
||||||
|
locaOffsetFormatLong
|
||||||
|
)
|
||||||
|
|
||||||
|
// A cm holds a parsed cmap entry.
|
||||||
|
type cm struct {
|
||||||
|
start, end, delta, offset uint32
|
||||||
|
}
|
||||||
|
|
||||||
|
// A Font represents a Truetype font.
|
||||||
|
type Font struct {
|
||||||
|
// Tables sliced from the TTF data. The different tables are documented
|
||||||
|
// at http://developer.apple.com/fonts/TTRefMan/RM06/Chap6.html
|
||||||
|
cmap, cvt, fpgm, glyf, hdmx, head, hhea, hmtx, kern, loca, maxp, name, os2, prep, vmtx []byte
|
||||||
|
|
||||||
|
cmapIndexes []byte
|
||||||
|
|
||||||
|
// Cached values derived from the raw ttf data.
|
||||||
|
cm []cm
|
||||||
|
locaOffsetFormat int
|
||||||
|
nGlyph, nHMetric, nKern int
|
||||||
|
fUnitsPerEm int32
|
||||||
|
ascent int32 // In FUnits.
|
||||||
|
descent int32 // In FUnits; typically negative.
|
||||||
|
bounds fixed.Rectangle26_6 // In FUnits.
|
||||||
|
// Values from the maxp section.
|
||||||
|
maxTwilightPoints, maxStorage, maxFunctionDefs, maxStackElements uint16
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Font) parseCmap() error {
|
||||||
|
const (
|
||||||
|
cmapFormat4 = 4
|
||||||
|
cmapFormat12 = 12
|
||||||
|
languageIndependent = 0
|
||||||
|
)
|
||||||
|
|
||||||
|
offset, _, err := parseSubtables(f.cmap, "cmap", 4, 8, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
offset = int(u32(f.cmap, offset+4))
|
||||||
|
if offset <= 0 || offset > len(f.cmap) {
|
||||||
|
return FormatError("bad cmap offset")
|
||||||
|
}
|
||||||
|
|
||||||
|
cmapFormat := u16(f.cmap, offset)
|
||||||
|
switch cmapFormat {
|
||||||
|
case cmapFormat4:
|
||||||
|
language := u16(f.cmap, offset+4)
|
||||||
|
if language != languageIndependent {
|
||||||
|
return UnsupportedError(fmt.Sprintf("language: %d", language))
|
||||||
|
}
|
||||||
|
segCountX2 := int(u16(f.cmap, offset+6))
|
||||||
|
if segCountX2%2 == 1 {
|
||||||
|
return FormatError(fmt.Sprintf("bad segCountX2: %d", segCountX2))
|
||||||
|
}
|
||||||
|
segCount := segCountX2 / 2
|
||||||
|
offset += 14
|
||||||
|
f.cm = make([]cm, segCount)
|
||||||
|
for i := 0; i < segCount; i++ {
|
||||||
|
f.cm[i].end = uint32(u16(f.cmap, offset))
|
||||||
|
offset += 2
|
||||||
|
}
|
||||||
|
offset += 2
|
||||||
|
for i := 0; i < segCount; i++ {
|
||||||
|
f.cm[i].start = uint32(u16(f.cmap, offset))
|
||||||
|
offset += 2
|
||||||
|
}
|
||||||
|
for i := 0; i < segCount; i++ {
|
||||||
|
f.cm[i].delta = uint32(u16(f.cmap, offset))
|
||||||
|
offset += 2
|
||||||
|
}
|
||||||
|
for i := 0; i < segCount; i++ {
|
||||||
|
f.cm[i].offset = uint32(u16(f.cmap, offset))
|
||||||
|
offset += 2
|
||||||
|
}
|
||||||
|
f.cmapIndexes = f.cmap[offset:]
|
||||||
|
return nil
|
||||||
|
|
||||||
|
case cmapFormat12:
|
||||||
|
if u16(f.cmap, offset+2) != 0 {
|
||||||
|
return FormatError(fmt.Sprintf("cmap format: % x", f.cmap[offset:offset+4]))
|
||||||
|
}
|
||||||
|
length := u32(f.cmap, offset+4)
|
||||||
|
language := u32(f.cmap, offset+8)
|
||||||
|
if language != languageIndependent {
|
||||||
|
return UnsupportedError(fmt.Sprintf("language: %d", language))
|
||||||
|
}
|
||||||
|
nGroups := u32(f.cmap, offset+12)
|
||||||
|
if length != 12*nGroups+16 {
|
||||||
|
return FormatError("inconsistent cmap length")
|
||||||
|
}
|
||||||
|
offset += 16
|
||||||
|
f.cm = make([]cm, nGroups)
|
||||||
|
for i := uint32(0); i < nGroups; i++ {
|
||||||
|
f.cm[i].start = u32(f.cmap, offset+0)
|
||||||
|
f.cm[i].end = u32(f.cmap, offset+4)
|
||||||
|
f.cm[i].delta = u32(f.cmap, offset+8) - f.cm[i].start
|
||||||
|
offset += 12
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return UnsupportedError(fmt.Sprintf("cmap format: %d", cmapFormat))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Font) parseHead() error {
|
||||||
|
if len(f.head) != 54 {
|
||||||
|
return FormatError(fmt.Sprintf("bad head length: %d", len(f.head)))
|
||||||
|
}
|
||||||
|
f.fUnitsPerEm = int32(u16(f.head, 18))
|
||||||
|
f.bounds.Min.X = fixed.Int26_6(int16(u16(f.head, 36)))
|
||||||
|
f.bounds.Min.Y = fixed.Int26_6(int16(u16(f.head, 38)))
|
||||||
|
f.bounds.Max.X = fixed.Int26_6(int16(u16(f.head, 40)))
|
||||||
|
f.bounds.Max.Y = fixed.Int26_6(int16(u16(f.head, 42)))
|
||||||
|
switch i := u16(f.head, 50); i {
|
||||||
|
case 0:
|
||||||
|
f.locaOffsetFormat = locaOffsetFormatShort
|
||||||
|
case 1:
|
||||||
|
f.locaOffsetFormat = locaOffsetFormatLong
|
||||||
|
default:
|
||||||
|
return FormatError(fmt.Sprintf("bad indexToLocFormat: %d", i))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Font) parseHhea() error {
|
||||||
|
if len(f.hhea) != 36 {
|
||||||
|
return FormatError(fmt.Sprintf("bad hhea length: %d", len(f.hhea)))
|
||||||
|
}
|
||||||
|
f.ascent = int32(int16(u16(f.hhea, 4)))
|
||||||
|
f.descent = int32(int16(u16(f.hhea, 6)))
|
||||||
|
f.nHMetric = int(u16(f.hhea, 34))
|
||||||
|
if 4*f.nHMetric+2*(f.nGlyph-f.nHMetric) != len(f.hmtx) {
|
||||||
|
return FormatError(fmt.Sprintf("bad hmtx length: %d", len(f.hmtx)))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Font) parseKern() error {
|
||||||
|
// Apple's TrueType documentation (http://developer.apple.com/fonts/TTRefMan/RM06/Chap6kern.html) says:
|
||||||
|
// "Previous versions of the 'kern' table defined both the version and nTables fields in the header
|
||||||
|
// as UInt16 values and not UInt32 values. Use of the older format on the Mac OS is discouraged
|
||||||
|
// (although AAT can sense an old kerning table and still make correct use of it). Microsoft
|
||||||
|
// Windows still uses the older format for the 'kern' table and will not recognize the newer one.
|
||||||
|
// Fonts targeted for the Mac OS only should use the new format; fonts targeted for both the Mac OS
|
||||||
|
// and Windows should use the old format."
|
||||||
|
// Since we expect that almost all fonts aim to be Windows-compatible, we only parse the "older" format,
|
||||||
|
// just like the C Freetype implementation.
|
||||||
|
if len(f.kern) == 0 {
|
||||||
|
if f.nKern != 0 {
|
||||||
|
return FormatError("bad kern table length")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if len(f.kern) < 18 {
|
||||||
|
return FormatError("kern data too short")
|
||||||
|
}
|
||||||
|
version, offset := u16(f.kern, 0), 2
|
||||||
|
if version != 0 {
|
||||||
|
return UnsupportedError(fmt.Sprintf("kern version: %d", version))
|
||||||
|
}
|
||||||
|
n, offset := u16(f.kern, offset), offset+2
|
||||||
|
if n != 1 {
|
||||||
|
return UnsupportedError(fmt.Sprintf("kern nTables: %d", n))
|
||||||
|
}
|
||||||
|
offset += 2
|
||||||
|
length, offset := int(u16(f.kern, offset)), offset+2
|
||||||
|
coverage, offset := u16(f.kern, offset), offset+2
|
||||||
|
if coverage != 0x0001 {
|
||||||
|
// We only support horizontal kerning.
|
||||||
|
return UnsupportedError(fmt.Sprintf("kern coverage: 0x%04x", coverage))
|
||||||
|
}
|
||||||
|
f.nKern, offset = int(u16(f.kern, offset)), offset+2
|
||||||
|
if 6*f.nKern != length-14 {
|
||||||
|
return FormatError("bad kern table length")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *Font) parseMaxp() error {
|
||||||
|
if len(f.maxp) != 32 {
|
||||||
|
return FormatError(fmt.Sprintf("bad maxp length: %d", len(f.maxp)))
|
||||||
|
}
|
||||||
|
f.nGlyph = int(u16(f.maxp, 4))
|
||||||
|
f.maxTwilightPoints = u16(f.maxp, 16)
|
||||||
|
f.maxStorage = u16(f.maxp, 18)
|
||||||
|
f.maxFunctionDefs = u16(f.maxp, 20)
|
||||||
|
f.maxStackElements = u16(f.maxp, 24)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// scale returns x divided by f.fUnitsPerEm, rounded to the nearest integer.
|
||||||
|
func (f *Font) scale(x fixed.Int26_6) fixed.Int26_6 {
|
||||||
|
if x >= 0 {
|
||||||
|
x += fixed.Int26_6(f.fUnitsPerEm) / 2
|
||||||
|
} else {
|
||||||
|
x -= fixed.Int26_6(f.fUnitsPerEm) / 2
|
||||||
|
}
|
||||||
|
return x / fixed.Int26_6(f.fUnitsPerEm)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Bounds returns the union of a Font's glyphs' bounds.
|
||||||
|
func (f *Font) Bounds(scale fixed.Int26_6) fixed.Rectangle26_6 {
|
||||||
|
b := f.bounds
|
||||||
|
b.Min.X = f.scale(scale * b.Min.X)
|
||||||
|
b.Min.Y = f.scale(scale * b.Min.Y)
|
||||||
|
b.Max.X = f.scale(scale * b.Max.X)
|
||||||
|
b.Max.Y = f.scale(scale * b.Max.Y)
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
|
||||||
|
// FUnitsPerEm returns the number of FUnits in a Font's em-square's side.
|
||||||
|
func (f *Font) FUnitsPerEm() int32 {
|
||||||
|
return f.fUnitsPerEm
|
||||||
|
}
|
||||||
|
|
||||||
|
// Index returns a Font's index for the given rune.
|
||||||
|
func (f *Font) Index(x rune) Index {
|
||||||
|
c := uint32(x)
|
||||||
|
for i, j := 0, len(f.cm); i < j; {
|
||||||
|
h := i + (j-i)/2
|
||||||
|
cm := &f.cm[h]
|
||||||
|
if c < cm.start {
|
||||||
|
j = h
|
||||||
|
} else if cm.end < c {
|
||||||
|
i = h + 1
|
||||||
|
} else if cm.offset == 0 {
|
||||||
|
return Index(c + cm.delta)
|
||||||
|
} else {
|
||||||
|
offset := int(cm.offset) + 2*(h-len(f.cm)+int(c-cm.start))
|
||||||
|
return Index(u16(f.cmapIndexes, offset))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// Name returns the Font's name value for the given NameID. It returns "" if
|
||||||
|
// there was an error, or if that name was not found.
|
||||||
|
func (f *Font) Name(id NameID) string {
|
||||||
|
x, platformID, err := parseSubtables(f.name, "name", 6, 12, func(b []byte) bool {
|
||||||
|
return NameID(u16(b, 6)) == id
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
offset, length := u16(f.name, 4)+u16(f.name, x+10), u16(f.name, x+8)
|
||||||
|
// Return the ASCII value of the encoded string.
|
||||||
|
// The string is encoded as UTF-16 on non-Apple platformIDs; Apple is platformID 1.
|
||||||
|
src := f.name[offset : offset+length]
|
||||||
|
var dst []byte
|
||||||
|
if platformID != 1 { // UTF-16.
|
||||||
|
if len(src)&1 != 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
dst = make([]byte, len(src)/2)
|
||||||
|
for i := range dst {
|
||||||
|
dst[i] = printable(u16(src, 2*i))
|
||||||
|
}
|
||||||
|
} else { // ASCII.
|
||||||
|
dst = make([]byte, len(src))
|
||||||
|
for i, c := range src {
|
||||||
|
dst[i] = printable(uint16(c))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return string(dst)
|
||||||
|
}
|
||||||
|
|
||||||
|
func printable(r uint16) byte {
|
||||||
|
if 0x20 <= r && r < 0x7f {
|
||||||
|
return byte(r)
|
||||||
|
}
|
||||||
|
return '?'
|
||||||
|
}
|
||||||
|
|
||||||
|
// unscaledHMetric returns the unscaled horizontal metrics for the glyph with
|
||||||
|
// the given index.
|
||||||
|
func (f *Font) unscaledHMetric(i Index) (h HMetric) {
|
||||||
|
j := int(i)
|
||||||
|
if j < 0 || f.nGlyph <= j {
|
||||||
|
return HMetric{}
|
||||||
|
}
|
||||||
|
if j >= f.nHMetric {
|
||||||
|
p := 4 * (f.nHMetric - 1)
|
||||||
|
return HMetric{
|
||||||
|
AdvanceWidth: fixed.Int26_6(u16(f.hmtx, p)),
|
||||||
|
LeftSideBearing: fixed.Int26_6(int16(u16(f.hmtx, p+2*(j-f.nHMetric)+4))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return HMetric{
|
||||||
|
AdvanceWidth: fixed.Int26_6(u16(f.hmtx, 4*j)),
|
||||||
|
LeftSideBearing: fixed.Int26_6(int16(u16(f.hmtx, 4*j+2))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HMetric returns the horizontal metrics for the glyph with the given index.
|
||||||
|
func (f *Font) HMetric(scale fixed.Int26_6, i Index) HMetric {
|
||||||
|
h := f.unscaledHMetric(i)
|
||||||
|
h.AdvanceWidth = f.scale(scale * h.AdvanceWidth)
|
||||||
|
h.LeftSideBearing = f.scale(scale * h.LeftSideBearing)
|
||||||
|
return h
|
||||||
|
}
|
||||||
|
|
||||||
|
// unscaledVMetric returns the unscaled vertical metrics for the glyph with
|
||||||
|
// the given index. yMax is the top of the glyph's bounding box.
|
||||||
|
func (f *Font) unscaledVMetric(i Index, yMax fixed.Int26_6) (v VMetric) {
|
||||||
|
j := int(i)
|
||||||
|
if j < 0 || f.nGlyph <= j {
|
||||||
|
return VMetric{}
|
||||||
|
}
|
||||||
|
if 4*j+4 <= len(f.vmtx) {
|
||||||
|
return VMetric{
|
||||||
|
AdvanceHeight: fixed.Int26_6(u16(f.vmtx, 4*j)),
|
||||||
|
TopSideBearing: fixed.Int26_6(int16(u16(f.vmtx, 4*j+2))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// The OS/2 table has grown over time.
|
||||||
|
// https://developer.apple.com/fonts/TTRefMan/RM06/Chap6OS2.html
|
||||||
|
// says that it was originally 68 bytes. Optional fields, including
|
||||||
|
// the ascender and descender, are described at
|
||||||
|
// http://www.microsoft.com/typography/otspec/os2.htm
|
||||||
|
if len(f.os2) >= 72 {
|
||||||
|
sTypoAscender := fixed.Int26_6(int16(u16(f.os2, 68)))
|
||||||
|
sTypoDescender := fixed.Int26_6(int16(u16(f.os2, 70)))
|
||||||
|
return VMetric{
|
||||||
|
AdvanceHeight: sTypoAscender - sTypoDescender,
|
||||||
|
TopSideBearing: sTypoAscender - yMax,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return VMetric{
|
||||||
|
AdvanceHeight: fixed.Int26_6(f.fUnitsPerEm),
|
||||||
|
TopSideBearing: 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// VMetric returns the vertical metrics for the glyph with the given index.
|
||||||
|
func (f *Font) VMetric(scale fixed.Int26_6, i Index) VMetric {
|
||||||
|
// TODO: should 0 be bounds.YMax?
|
||||||
|
v := f.unscaledVMetric(i, 0)
|
||||||
|
v.AdvanceHeight = f.scale(scale * v.AdvanceHeight)
|
||||||
|
v.TopSideBearing = f.scale(scale * v.TopSideBearing)
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
|
||||||
|
// Kern returns the horizontal adjustment for the given glyph pair. A positive
|
||||||
|
// kern means to move the glyphs further apart.
|
||||||
|
func (f *Font) Kern(scale fixed.Int26_6, i0, i1 Index) fixed.Int26_6 {
|
||||||
|
if f.nKern == 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
g := uint32(i0)<<16 | uint32(i1)
|
||||||
|
lo, hi := 0, f.nKern
|
||||||
|
for lo < hi {
|
||||||
|
i := (lo + hi) / 2
|
||||||
|
ig := u32(f.kern, 18+6*i)
|
||||||
|
if ig < g {
|
||||||
|
lo = i + 1
|
||||||
|
} else if ig > g {
|
||||||
|
hi = i
|
||||||
|
} else {
|
||||||
|
return f.scale(scale * fixed.Int26_6(int16(u16(f.kern, 22+6*i))))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse returns a new Font for the given TTF or TTC data.
|
||||||
|
//
|
||||||
|
// For TrueType Collections, the first font in the collection is parsed.
|
||||||
|
func Parse(ttf []byte) (font *Font, err error) {
|
||||||
|
return parse(ttf, 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
func parse(ttf []byte, offset int) (font *Font, err error) {
|
||||||
|
if len(ttf)-offset < 12 {
|
||||||
|
err = FormatError("TTF data is too short")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
originalOffset := offset
|
||||||
|
magic, offset := u32(ttf, offset), offset+4
|
||||||
|
switch magic {
|
||||||
|
case 0x00010000:
|
||||||
|
// No-op.
|
||||||
|
case 0x74746366: // "ttcf" as a big-endian uint32.
|
||||||
|
if originalOffset != 0 {
|
||||||
|
err = FormatError("recursive TTC")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ttcVersion, offset := u32(ttf, offset), offset+4
|
||||||
|
if ttcVersion != 0x00010000 {
|
||||||
|
// TODO: support TTC version 2.0, once I have such a .ttc file to test with.
|
||||||
|
err = FormatError("bad TTC version")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
numFonts, offset := int(u32(ttf, offset)), offset+4
|
||||||
|
if numFonts <= 0 {
|
||||||
|
err = FormatError("bad number of TTC fonts")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len(ttf[offset:])/4 < numFonts {
|
||||||
|
err = FormatError("TTC offset table is too short")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// TODO: provide an API to select which font in a TrueType collection to return,
|
||||||
|
// not just the first one. This may require an API to parse a TTC's name tables,
|
||||||
|
// so users of this package can select the font in a TTC by name.
|
||||||
|
offset = int(u32(ttf, offset))
|
||||||
|
if offset <= 0 || offset > len(ttf) {
|
||||||
|
err = FormatError("bad TTC offset")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return parse(ttf, offset)
|
||||||
|
default:
|
||||||
|
err = FormatError("bad TTF version")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
n, offset := int(u16(ttf, offset)), offset+2
|
||||||
|
if len(ttf) < 16*n+12 {
|
||||||
|
err = FormatError("TTF data is too short")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
f := new(Font)
|
||||||
|
// Assign the table slices.
|
||||||
|
for i := 0; i < n; i++ {
|
||||||
|
x := 16*i + 12
|
||||||
|
switch string(ttf[x : x+4]) {
|
||||||
|
case "cmap":
|
||||||
|
f.cmap, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "cvt ":
|
||||||
|
f.cvt, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "fpgm":
|
||||||
|
f.fpgm, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "glyf":
|
||||||
|
f.glyf, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "hdmx":
|
||||||
|
f.hdmx, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "head":
|
||||||
|
f.head, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "hhea":
|
||||||
|
f.hhea, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "hmtx":
|
||||||
|
f.hmtx, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "kern":
|
||||||
|
f.kern, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "loca":
|
||||||
|
f.loca, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "maxp":
|
||||||
|
f.maxp, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "name":
|
||||||
|
f.name, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "OS/2":
|
||||||
|
f.os2, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "prep":
|
||||||
|
f.prep, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
case "vmtx":
|
||||||
|
f.vmtx, err = readTable(ttf, ttf[x+8:x+16])
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Parse and sanity-check the TTF data.
|
||||||
|
if err = f.parseHead(); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err = f.parseMaxp(); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err = f.parseCmap(); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err = f.parseKern(); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err = f.parseHhea(); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
font = f
|
||||||
|
return
|
||||||
|
}
|
|
@ -0,0 +1,26 @@
|
||||||
|
# Compiled Object files, Static and Dynamic libs (Shared Objects)
|
||||||
|
*.o
|
||||||
|
*.a
|
||||||
|
*.so
|
||||||
|
|
||||||
|
# Folders
|
||||||
|
_obj
|
||||||
|
_test
|
||||||
|
.idea
|
||||||
|
.git
|
||||||
|
|
||||||
|
# Architecture specific extensions/prefixes
|
||||||
|
*.[568vq]
|
||||||
|
[568vq].out
|
||||||
|
|
||||||
|
*.cgo1.go
|
||||||
|
*.cgo2.c
|
||||||
|
_cgo_defun.c
|
||||||
|
_cgo_gotypes.go
|
||||||
|
_cgo_export.*
|
||||||
|
|
||||||
|
_testmain.go
|
||||||
|
|
||||||
|
*.exe
|
||||||
|
*.test
|
||||||
|
*.prof
|
|
@ -0,0 +1,201 @@
|
||||||
|
Apache License
|
||||||
|
Version 2.0, January 2004
|
||||||
|
http://www.apache.org/licenses/
|
||||||
|
|
||||||
|
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||||
|
|
||||||
|
1. Definitions.
|
||||||
|
|
||||||
|
"License" shall mean the terms and conditions for use, reproduction,
|
||||||
|
and distribution as defined by Sections 1 through 9 of this document.
|
||||||
|
|
||||||
|
"Licensor" shall mean the copyright owner or entity authorized by
|
||||||
|
the copyright owner that is granting the License.
|
||||||
|
|
||||||
|
"Legal Entity" shall mean the union of the acting entity and all
|
||||||
|
other entities that control, are controlled by, or are under common
|
||||||
|
control with that entity. For the purposes of this definition,
|
||||||
|
"control" means (i) the power, direct or indirect, to cause the
|
||||||
|
direction or management of such entity, whether by contract or
|
||||||
|
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||||
|
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||||
|
|
||||||
|
"You" (or "Your") shall mean an individual or Legal Entity
|
||||||
|
exercising permissions granted by this License.
|
||||||
|
|
||||||
|
"Source" form shall mean the preferred form for making modifications,
|
||||||
|
including but not limited to software source code, documentation
|
||||||
|
source, and configuration files.
|
||||||
|
|
||||||
|
"Object" form shall mean any form resulting from mechanical
|
||||||
|
transformation or translation of a Source form, including but
|
||||||
|
not limited to compiled object code, generated documentation,
|
||||||
|
and conversions to other media types.
|
||||||
|
|
||||||
|
"Work" shall mean the work of authorship, whether in Source or
|
||||||
|
Object form, made available under the License, as indicated by a
|
||||||
|
copyright notice that is included in or attached to the work
|
||||||
|
(an example is provided in the Appendix below).
|
||||||
|
|
||||||
|
"Derivative Works" shall mean any work, whether in Source or Object
|
||||||
|
form, that is based on (or derived from) the Work and for which the
|
||||||
|
editorial revisions, annotations, elaborations, or other modifications
|
||||||
|
represent, as a whole, an original work of authorship. For the purposes
|
||||||
|
of this License, Derivative Works shall not include works that remain
|
||||||
|
separable from, or merely link (or bind by name) to the interfaces of,
|
||||||
|
the Work and Derivative Works thereof.
|
||||||
|
|
||||||
|
"Contribution" shall mean any work of authorship, including
|
||||||
|
the original version of the Work and any modifications or additions
|
||||||
|
to that Work or Derivative Works thereof, that is intentionally
|
||||||
|
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||||
|
or by an individual or Legal Entity authorized to submit on behalf of
|
||||||
|
the copyright owner. For the purposes of this definition, "submitted"
|
||||||
|
means any form of electronic, verbal, or written communication sent
|
||||||
|
to the Licensor or its representatives, including but not limited to
|
||||||
|
communication on electronic mailing lists, source code control systems,
|
||||||
|
and issue tracking systems that are managed by, or on behalf of, the
|
||||||
|
Licensor for the purpose of discussing and improving the Work, but
|
||||||
|
excluding communication that is conspicuously marked or otherwise
|
||||||
|
designated in writing by the copyright owner as "Not a Contribution."
|
||||||
|
|
||||||
|
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||||
|
on behalf of whom a Contribution has been received by Licensor and
|
||||||
|
subsequently incorporated within the Work.
|
||||||
|
|
||||||
|
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
copyright license to reproduce, prepare Derivative Works of,
|
||||||
|
publicly display, publicly perform, sublicense, and distribute the
|
||||||
|
Work and such Derivative Works in Source or Object form.
|
||||||
|
|
||||||
|
3. Grant of Patent License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
(except as stated in this section) patent license to make, have made,
|
||||||
|
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||||
|
where such license applies only to those patent claims licensable
|
||||||
|
by such Contributor that are necessarily infringed by their
|
||||||
|
Contribution(s) alone or by combination of their Contribution(s)
|
||||||
|
with the Work to which such Contribution(s) was submitted. If You
|
||||||
|
institute patent litigation against any entity (including a
|
||||||
|
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||||
|
or a Contribution incorporated within the Work constitutes direct
|
||||||
|
or contributory patent infringement, then any patent licenses
|
||||||
|
granted to You under this License for that Work shall terminate
|
||||||
|
as of the date such litigation is filed.
|
||||||
|
|
||||||
|
4. Redistribution. You may reproduce and distribute copies of the
|
||||||
|
Work or Derivative Works thereof in any medium, with or without
|
||||||
|
modifications, and in Source or Object form, provided that You
|
||||||
|
meet the following conditions:
|
||||||
|
|
||||||
|
(a) You must give any other recipients of the Work or
|
||||||
|
Derivative Works a copy of this License; and
|
||||||
|
|
||||||
|
(b) You must cause any modified files to carry prominent notices
|
||||||
|
stating that You changed the files; and
|
||||||
|
|
||||||
|
(c) You must retain, in the Source form of any Derivative Works
|
||||||
|
that You distribute, all copyright, patent, trademark, and
|
||||||
|
attribution notices from the Source form of the Work,
|
||||||
|
excluding those notices that do not pertain to any part of
|
||||||
|
the Derivative Works; and
|
||||||
|
|
||||||
|
(d) If the Work includes a "NOTICE" text file as part of its
|
||||||
|
distribution, then any Derivative Works that You distribute must
|
||||||
|
include a readable copy of the attribution notices contained
|
||||||
|
within such NOTICE file, excluding those notices that do not
|
||||||
|
pertain to any part of the Derivative Works, in at least one
|
||||||
|
of the following places: within a NOTICE text file distributed
|
||||||
|
as part of the Derivative Works; within the Source form or
|
||||||
|
documentation, if provided along with the Derivative Works; or,
|
||||||
|
within a display generated by the Derivative Works, if and
|
||||||
|
wherever such third-party notices normally appear. The contents
|
||||||
|
of the NOTICE file are for informational purposes only and
|
||||||
|
do not modify the License. You may add Your own attribution
|
||||||
|
notices within Derivative Works that You distribute, alongside
|
||||||
|
or as an addendum to the NOTICE text from the Work, provided
|
||||||
|
that such additional attribution notices cannot be construed
|
||||||
|
as modifying the License.
|
||||||
|
|
||||||
|
You may add Your own copyright statement to Your modifications and
|
||||||
|
may provide additional or different license terms and conditions
|
||||||
|
for use, reproduction, or distribution of Your modifications, or
|
||||||
|
for any such Derivative Works as a whole, provided Your use,
|
||||||
|
reproduction, and distribution of the Work otherwise complies with
|
||||||
|
the conditions stated in this License.
|
||||||
|
|
||||||
|
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||||
|
any Contribution intentionally submitted for inclusion in the Work
|
||||||
|
by You to the Licensor shall be under the terms and conditions of
|
||||||
|
this License, without any additional terms or conditions.
|
||||||
|
Notwithstanding the above, nothing herein shall supersede or modify
|
||||||
|
the terms of any separate license agreement you may have executed
|
||||||
|
with Licensor regarding such Contributions.
|
||||||
|
|
||||||
|
6. Trademarks. This License does not grant permission to use the trade
|
||||||
|
names, trademarks, service marks, or product names of the Licensor,
|
||||||
|
except as required for reasonable and customary use in describing the
|
||||||
|
origin of the Work and reproducing the content of the NOTICE file.
|
||||||
|
|
||||||
|
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||||
|
agreed to in writing, Licensor provides the Work (and each
|
||||||
|
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||||
|
implied, including, without limitation, any warranties or conditions
|
||||||
|
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||||
|
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||||
|
appropriateness of using or redistributing the Work and assume any
|
||||||
|
risks associated with Your exercise of permissions under this License.
|
||||||
|
|
||||||
|
8. Limitation of Liability. In no event and under no legal theory,
|
||||||
|
whether in tort (including negligence), contract, or otherwise,
|
||||||
|
unless required by applicable law (such as deliberate and grossly
|
||||||
|
negligent acts) or agreed to in writing, shall any Contributor be
|
||||||
|
liable to You for damages, including any direct, indirect, special,
|
||||||
|
incidental, or consequential damages of any character arising as a
|
||||||
|
result of this License or out of the use or inability to use the
|
||||||
|
Work (including but not limited to damages for loss of goodwill,
|
||||||
|
work stoppage, computer failure or malfunction, or any and all
|
||||||
|
other commercial damages or losses), even if such Contributor
|
||||||
|
has been advised of the possibility of such damages.
|
||||||
|
|
||||||
|
9. Accepting Warranty or Additional Liability. While redistributing
|
||||||
|
the Work or Derivative Works thereof, You may choose to offer,
|
||||||
|
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||||
|
or other liability obligations and/or rights consistent with this
|
||||||
|
License. However, in accepting such obligations, You may act only
|
||||||
|
on Your own behalf and on Your sole responsibility, not on behalf
|
||||||
|
of any other Contributor, and only if You agree to indemnify,
|
||||||
|
defend, and hold each Contributor harmless for any liability
|
||||||
|
incurred by, or claims asserted against, such Contributor by reason
|
||||||
|
of your accepting any such warranty or additional liability.
|
||||||
|
|
||||||
|
END OF TERMS AND CONDITIONS
|
||||||
|
|
||||||
|
APPENDIX: How to apply the Apache License to your work.
|
||||||
|
|
||||||
|
To apply the Apache License to your work, attach the following
|
||||||
|
boilerplate notice, with the fields enclosed by brackets "{}"
|
||||||
|
replaced with your own identifying information. (Don't include
|
||||||
|
the brackets!) The text should be enclosed in the appropriate
|
||||||
|
comment syntax for the file format. We also recommend that a
|
||||||
|
file or class name and description of purpose be included on the
|
||||||
|
same "printed page" as the copyright notice for easier
|
||||||
|
identification within third-party archives.
|
||||||
|
|
||||||
|
Copyright {yyyy} {name of copyright owner}
|
||||||
|
|
||||||
|
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.
|
|
@ -0,0 +1,65 @@
|
||||||
|
# gocaptcha
|
||||||
|
一个简单的Go语言实现的验证码
|
||||||
|
|
||||||
|
##图片实例
|
||||||
|
|
||||||
|
![image](https://raw.githubusercontent.com/lifei6671/gocaptcha/master/example/image_1.jpg)
|
||||||
|
![image](https://raw.githubusercontent.com/lifei6671/gocaptcha/master/example/image_2.jpg)
|
||||||
|
![image](https://raw.githubusercontent.com/lifei6671/gocaptcha/master/example/image_3.jpg)
|
||||||
|
![image](https://raw.githubusercontent.com/lifei6671/gocaptcha/master/example/image_4.jpg)
|
||||||
|
|
||||||
|
##简介
|
||||||
|
|
||||||
|
基于Golang实现的图片验证码生成库,可以实现随机字母个数,随机直线,随机噪点等。可以设置任意多字体,每个验证码随机选一种字体展示。
|
||||||
|
|
||||||
|
##实例
|
||||||
|
|
||||||
|
####使用:
|
||||||
|
|
||||||
|
```
|
||||||
|
go get github.com/lifei6671/gocaptcha/
|
||||||
|
```
|
||||||
|
|
||||||
|
####使用的类库
|
||||||
|
|
||||||
|
```
|
||||||
|
go get github.com/golang/freetype
|
||||||
|
go get github.com/golang/freetype/truetype
|
||||||
|
go get golang.org/x/image
|
||||||
|
```
|
||||||
|
天朝可以去 http://www.golangtc.com/download/package 或 https://gopm.io 下载
|
||||||
|
|
||||||
|
####代码
|
||||||
|
具体实例可以查看example目录,有生成的验证码图片。
|
||||||
|
|
||||||
|
```
|
||||||
|
|
||||||
|
func Get(w http.ResponseWriter, r *http.Request) {
|
||||||
|
//初始化一个验证码对象
|
||||||
|
captchaImage,err := gocaptcha.NewCaptchaImage(dx,dy,gocaptcha.RandLightColor());
|
||||||
|
|
||||||
|
//画上三条随机直线
|
||||||
|
captchaImage.Drawline(3);
|
||||||
|
|
||||||
|
//画边框
|
||||||
|
captchaImage.DrawBorder(gocaptcha.ColorToRGB(0x17A7A7A));
|
||||||
|
|
||||||
|
//画随机噪点
|
||||||
|
captchaImage.DrawNoise(gocaptcha.CaptchaComplexHigh);
|
||||||
|
|
||||||
|
//画随机文字噪点
|
||||||
|
captchaImage.DrawTextNoise(gocaptcha.CaptchaComplexLower);
|
||||||
|
//画验证码文字,可以预先保持到Session种或其他储存容器种
|
||||||
|
captchaImage.DrawText(gocaptcha.RandText(4));
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
//将验证码保持到输出流种,可以是文件或HTTP流等
|
||||||
|
captchaImage.SaveImage(w,gocaptcha.ImageFormatJpeg);
|
||||||
|
}
|
||||||
|
|
||||||
|
```
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,429 @@
|
||||||
|
package gocaptcha
|
||||||
|
|
||||||
|
import (
|
||||||
|
"image"
|
||||||
|
"image/color"
|
||||||
|
"image/draw"
|
||||||
|
"math"
|
||||||
|
"io"
|
||||||
|
"image/png"
|
||||||
|
"image/jpeg"
|
||||||
|
"errors"
|
||||||
|
"time"
|
||||||
|
"math/rand"
|
||||||
|
"github.com/golang/freetype"
|
||||||
|
"flag"
|
||||||
|
"golang.org/x/image/font"
|
||||||
|
"log"
|
||||||
|
"io/ioutil"
|
||||||
|
"github.com/golang/freetype/truetype"
|
||||||
|
"image/gif"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
dpi = flag.Float64("dpi", 72, "screen resolution in Dots Per Inch")
|
||||||
|
r = rand.New(rand.NewSource(time.Now().UnixNano()));
|
||||||
|
FontFamily []string = make([]string,0);
|
||||||
|
)
|
||||||
|
|
||||||
|
const txtChars = "AaCcDdEeFfGgHhJjKkLMmNnPpQqRrSsTtUuVvWwXxYtZ2346789";
|
||||||
|
|
||||||
|
|
||||||
|
const(
|
||||||
|
//图片格式
|
||||||
|
ImageFormatPng = iota
|
||||||
|
ImageFormatJpeg
|
||||||
|
ImageFormatGif
|
||||||
|
//验证码噪点强度
|
||||||
|
CaptchaComplexLower = iota
|
||||||
|
CaptchaComplexMedium
|
||||||
|
CaptchaComplexHigh
|
||||||
|
)
|
||||||
|
|
||||||
|
type CaptchaImage struct {
|
||||||
|
nrgba *image.NRGBA
|
||||||
|
width int
|
||||||
|
height int
|
||||||
|
Complex int
|
||||||
|
}
|
||||||
|
|
||||||
|
//获取指定目录下的所有文件,不进入下一级目录搜索,可以匹配后缀过滤。
|
||||||
|
func ReadFonts(dirPth string, suffix string) (err error) {
|
||||||
|
files := make([]string, 0, 10)
|
||||||
|
dir, err := ioutil.ReadDir(dirPth)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
PthSep := string(os.PathSeparator)
|
||||||
|
suffix = strings.ToUpper(suffix) //忽略后缀匹配的大小写
|
||||||
|
for _, fi := range dir {
|
||||||
|
if fi.IsDir() { // 忽略目录
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) { //匹配文件
|
||||||
|
files = append(files, dirPth+PthSep+fi.Name())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
SetFontFamily(files...)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
//新建一个图片对象
|
||||||
|
func NewCaptchaImage(width int,height int,bgColor color.RGBA) (*CaptchaImage ,error){
|
||||||
|
|
||||||
|
m := image.NewNRGBA(image.Rect(0, 0, width, height));
|
||||||
|
|
||||||
|
draw.Draw(m, m.Bounds(), &image.Uniform{ bgColor }, image.ZP, draw.Src);
|
||||||
|
|
||||||
|
return &CaptchaImage{
|
||||||
|
nrgba: m,
|
||||||
|
height : height,
|
||||||
|
width : width,
|
||||||
|
},nil;
|
||||||
|
}
|
||||||
|
|
||||||
|
//保存图片对象
|
||||||
|
func (this *CaptchaImage)SaveImage(w io.Writer ,imageFormat int) error{
|
||||||
|
|
||||||
|
if(imageFormat == ImageFormatPng){
|
||||||
|
return png.Encode(w, this.nrgba);
|
||||||
|
};
|
||||||
|
if(imageFormat == ImageFormatJpeg){
|
||||||
|
return jpeg.Encode(w,this.nrgba, &jpeg.Options{ 100 });
|
||||||
|
}
|
||||||
|
if(imageFormat == ImageFormatGif){
|
||||||
|
return gif.Encode(w,this.nrgba, &gif.Options{ NumColors : 256});
|
||||||
|
}
|
||||||
|
|
||||||
|
return errors.New("Not supported image format");
|
||||||
|
}
|
||||||
|
//添加一个较粗的空白直线
|
||||||
|
func (captcha *CaptchaImage) DrawHollowLine()(*CaptchaImage){
|
||||||
|
|
||||||
|
first := (captcha.width / 20);
|
||||||
|
end := first * 19;
|
||||||
|
|
||||||
|
lineColor := color.RGBA{ R : 245,G:250,B:251,A:255};
|
||||||
|
|
||||||
|
x1 := float64(r.Intn(first));
|
||||||
|
//y1 := float64(r.Intn(y)+y);
|
||||||
|
|
||||||
|
x2 := float64( r.Intn(first)+end);
|
||||||
|
|
||||||
|
multiple := float64(r.Intn(5)+3)/float64(5);
|
||||||
|
if(int(multiple*10) % 3 == 0){
|
||||||
|
multiple = multiple * -1.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
w := captcha.height / 20;
|
||||||
|
|
||||||
|
for ;x1 < x2; x1 ++{
|
||||||
|
|
||||||
|
y := math.Sin(x1*math.Pi*multiple/float64(captcha.width)) * float64(captcha.height/3);
|
||||||
|
|
||||||
|
if(multiple < 0){
|
||||||
|
y = y + float64(captcha.height/2);
|
||||||
|
}
|
||||||
|
captcha.nrgba.Set(int(x1),int(y),lineColor);
|
||||||
|
|
||||||
|
for i:=0;i<=w;i++{
|
||||||
|
captcha.nrgba.Set(int(x1),int(y)+i,lineColor);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return captcha;
|
||||||
|
}
|
||||||
|
//画一条直线
|
||||||
|
func (captcha *CaptchaImage)Drawline(num int) (*CaptchaImage) {
|
||||||
|
|
||||||
|
first := (captcha.width / 10);
|
||||||
|
end := first * 9;
|
||||||
|
|
||||||
|
y := captcha.height / 3;
|
||||||
|
|
||||||
|
for i:=0;i<num;i++{
|
||||||
|
|
||||||
|
point1 := Point{ X : r.Intn(first),Y:r.Intn(y)};
|
||||||
|
point2 := Point{X : r.Intn(first)+end,Y:r.Intn(y)};
|
||||||
|
|
||||||
|
if(i % 2 == 0){
|
||||||
|
point1.Y = r.Intn(y)+y*2;
|
||||||
|
point2.Y = r.Intn(y);
|
||||||
|
}else{
|
||||||
|
point1.Y = r.Intn(y)+y*(i%2);
|
||||||
|
point2.Y = r.Intn(y) + y*2;
|
||||||
|
}
|
||||||
|
|
||||||
|
captcha.drawBeeline(point1,point2,RandDeepColor());
|
||||||
|
|
||||||
|
}
|
||||||
|
return captcha;
|
||||||
|
}
|
||||||
|
|
||||||
|
func (captcha *CaptchaImage)drawBeeline(point1 Point,point2 Point,lineColor color.RGBA){
|
||||||
|
dx := math.Abs(float64(point1.X - point2.X));
|
||||||
|
|
||||||
|
dy := math.Abs(float64(point2.Y - point1.Y))
|
||||||
|
sx, sy := 1, 1
|
||||||
|
if point1.X >= point2.X {
|
||||||
|
sx = -1
|
||||||
|
}
|
||||||
|
if point1.Y >= point2.Y {
|
||||||
|
sy = -1
|
||||||
|
}
|
||||||
|
err := dx - dy
|
||||||
|
for {
|
||||||
|
captcha.nrgba.Set(point1.X,point1.Y,lineColor);
|
||||||
|
captcha.nrgba.Set(point1.X+1,point1.Y,lineColor);
|
||||||
|
captcha.nrgba.Set(point1.X-1,point1.Y,lineColor);
|
||||||
|
captcha.nrgba.Set(point1.X+2,point1.Y,lineColor);
|
||||||
|
captcha.nrgba.Set(point1.X-2,point1.Y,lineColor);
|
||||||
|
if point1.X == point2.X && point1.Y == point2.Y {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
e2 := err * 2
|
||||||
|
if e2 > -dy {
|
||||||
|
err -= dy
|
||||||
|
point1.X += sx
|
||||||
|
}
|
||||||
|
if e2 < dx {
|
||||||
|
err += dx
|
||||||
|
point1.Y += sy
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//画边框
|
||||||
|
func (captcha *CaptchaImage) DrawBorder(borderColor color.RGBA) (*CaptchaImage){
|
||||||
|
for x :=0;x<captcha.width;x++{
|
||||||
|
captcha.nrgba.Set(x,0,borderColor);
|
||||||
|
captcha.nrgba.Set(x,captcha.height-1,borderColor);
|
||||||
|
}
|
||||||
|
for y:=0;y<captcha.height;y++{
|
||||||
|
captcha.nrgba.Set(0,y,borderColor);
|
||||||
|
captcha.nrgba.Set(captcha.width-1,y,borderColor);
|
||||||
|
}
|
||||||
|
return captcha;
|
||||||
|
}
|
||||||
|
|
||||||
|
//画噪点
|
||||||
|
func (captcha *CaptchaImage) DrawNoise(complex int) (*CaptchaImage){
|
||||||
|
density := 18;
|
||||||
|
if(complex == CaptchaComplexLower){
|
||||||
|
density = 28;
|
||||||
|
}else if(complex == CaptchaComplexMedium){
|
||||||
|
density = 18;
|
||||||
|
}else if(complex == CaptchaComplexHigh){
|
||||||
|
density = 8;
|
||||||
|
}
|
||||||
|
maxSize := (captcha.height * captcha.width)/density;
|
||||||
|
|
||||||
|
for i:=0;i<maxSize;i++{
|
||||||
|
|
||||||
|
rw := r.Intn(captcha.width);
|
||||||
|
rh := r.Intn(captcha.height);
|
||||||
|
|
||||||
|
captcha.nrgba.Set(rw,rh,RandColor());
|
||||||
|
size := r.Intn(maxSize);
|
||||||
|
if size%3 == 0{
|
||||||
|
captcha.nrgba.Set(rw+1,rh+1,RandColor());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return captcha;
|
||||||
|
}
|
||||||
|
//画文字噪点
|
||||||
|
func (captcha *CaptchaImage) DrawTextNoise(complex int)(error){
|
||||||
|
density := 1500;
|
||||||
|
if(complex == CaptchaComplexLower){
|
||||||
|
density = 2000;
|
||||||
|
}else if(complex == CaptchaComplexMedium){
|
||||||
|
density = 1500;
|
||||||
|
}else if(complex == CaptchaComplexHigh){
|
||||||
|
density = 1000;
|
||||||
|
}
|
||||||
|
|
||||||
|
maxSize := (captcha.height * captcha.width)/density;
|
||||||
|
|
||||||
|
fmt.Println(maxSize);
|
||||||
|
|
||||||
|
r := rand.New(rand.NewSource(time.Now().UnixNano()));
|
||||||
|
|
||||||
|
c := freetype.NewContext();
|
||||||
|
c.SetDPI(*dpi)
|
||||||
|
|
||||||
|
c.SetClip(captcha.nrgba.Bounds());
|
||||||
|
c.SetDst(captcha.nrgba)
|
||||||
|
c.SetHinting(font.HintingFull)
|
||||||
|
rawFontSize := float64(captcha.height) / (1+ float64(r.Intn(7))/float64(10))
|
||||||
|
|
||||||
|
for i:=0;i<maxSize;i++{
|
||||||
|
|
||||||
|
rw := r.Intn(captcha.width);
|
||||||
|
rh := r.Intn(captcha.height);
|
||||||
|
|
||||||
|
text := RandText(1);
|
||||||
|
fontSize := rawFontSize/2 + float64(r.Intn(5));
|
||||||
|
|
||||||
|
c.SetSrc(image.NewUniform(RandLightColor()))
|
||||||
|
c.SetFontSize(fontSize);
|
||||||
|
f,err := RandFontFamily();
|
||||||
|
|
||||||
|
if(err != nil){
|
||||||
|
log.Println(err);
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
c.SetFont(f)
|
||||||
|
pt := freetype.Pt(rw, rh);
|
||||||
|
|
||||||
|
_, err = c.DrawString(text, pt)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil;
|
||||||
|
}
|
||||||
|
//写字
|
||||||
|
func(captcha *CaptchaImage) DrawText(text string)(error){
|
||||||
|
c := freetype.NewContext();
|
||||||
|
c.SetDPI(*dpi)
|
||||||
|
|
||||||
|
c.SetClip(captcha.nrgba.Bounds());
|
||||||
|
c.SetDst(captcha.nrgba)
|
||||||
|
c.SetHinting(font.HintingFull)
|
||||||
|
|
||||||
|
fontWidth := captcha.width/ len(text);
|
||||||
|
|
||||||
|
|
||||||
|
for i,s := range text{
|
||||||
|
|
||||||
|
fontSize := float64(captcha.height) / (1+ float64(r.Intn(7))/float64(9))
|
||||||
|
|
||||||
|
c.SetSrc(image.NewUniform(RandDeepColor()))
|
||||||
|
c.SetFontSize(fontSize);
|
||||||
|
f,err := RandFontFamily();
|
||||||
|
|
||||||
|
if(err != nil){
|
||||||
|
log.Println(err);
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
c.SetFont(f)
|
||||||
|
|
||||||
|
x := int(fontWidth)*i + int(fontWidth)/int(fontSize);
|
||||||
|
|
||||||
|
y := 5 + r.Intn(captcha.height/2) + int(fontSize/2);
|
||||||
|
|
||||||
|
pt := freetype.Pt(x, y);
|
||||||
|
|
||||||
|
_, err = c.DrawString(string(s), pt)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return err;
|
||||||
|
}
|
||||||
|
//pt.Y += c.PointToFixed(*size * *spacing)
|
||||||
|
//pt.X += c.PointToFixed(*size);
|
||||||
|
}
|
||||||
|
return nil;
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
//获取所及字体
|
||||||
|
func RandFontFamily() (*truetype.Font ,error ){
|
||||||
|
fontfile := FontFamily[r.Intn(len(FontFamily))];
|
||||||
|
|
||||||
|
fontBytes, err := ioutil.ReadFile(fontfile)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return &truetype.Font{},err;
|
||||||
|
}
|
||||||
|
f, err := freetype.ParseFont(fontBytes)
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return &truetype.Font{},err;
|
||||||
|
}
|
||||||
|
return f,nil;
|
||||||
|
}
|
||||||
|
|
||||||
|
//随机生成深色系
|
||||||
|
func RandDeepColor() color.RGBA{
|
||||||
|
|
||||||
|
randColor := RandColor();
|
||||||
|
|
||||||
|
increase := float64(30 + r.Intn(255));
|
||||||
|
|
||||||
|
red := math.Abs(math.Min(float64(randColor.R) - increase,255));
|
||||||
|
|
||||||
|
green := math.Abs(math.Min(float64(randColor.G) - increase,255));
|
||||||
|
blue := math.Abs(math.Min(float64(randColor.B) - increase,255));
|
||||||
|
|
||||||
|
|
||||||
|
return color.RGBA{R:uint8(red), G : uint8(green),B:uint8(blue),A:uint8(255)};
|
||||||
|
}
|
||||||
|
//随机生成浅色
|
||||||
|
func RandLightColor() color.RGBA{
|
||||||
|
|
||||||
|
red := r.Intn(55) + 200;
|
||||||
|
green := r.Intn(55)+200;
|
||||||
|
blue := r.Intn(55) + 200;
|
||||||
|
|
||||||
|
|
||||||
|
return color.RGBA{R:uint8(red), G : uint8(green),B:uint8(blue),A:uint8(255)};
|
||||||
|
}
|
||||||
|
|
||||||
|
//生成随机颜色
|
||||||
|
func RandColor() color.RGBA{
|
||||||
|
|
||||||
|
red := r.Intn(255);
|
||||||
|
green := r.Intn(255);
|
||||||
|
blue := r.Intn(255);
|
||||||
|
if((red + green) > 400){
|
||||||
|
blue = 0;
|
||||||
|
}else{
|
||||||
|
blue = 400 - green -red;
|
||||||
|
}
|
||||||
|
if(blue > 255){
|
||||||
|
blue = 255;
|
||||||
|
}
|
||||||
|
return color.RGBA{R:uint8(red), G : uint8(green),B:uint8(blue),A:uint8(255)};
|
||||||
|
}
|
||||||
|
|
||||||
|
//生成随机字体
|
||||||
|
func RandText(num int) string {
|
||||||
|
textNum := len(txtChars);
|
||||||
|
text := "";
|
||||||
|
r := rand.New(rand.NewSource(time.Now().UnixNano()));
|
||||||
|
|
||||||
|
for i:=0;i<num ;i++ {
|
||||||
|
text = text + string(txtChars[r.Intn(textNum)]);
|
||||||
|
}
|
||||||
|
return text;
|
||||||
|
}
|
||||||
|
|
||||||
|
//添加一个字体路径到字体库
|
||||||
|
func SetFontFamily(fontPath ... string){
|
||||||
|
|
||||||
|
FontFamily= append(FontFamily,fontPath...);
|
||||||
|
}
|
||||||
|
/**
|
||||||
|
* 颜色代码转换为RGB
|
||||||
|
* input int
|
||||||
|
* output int red, green, blue
|
||||||
|
**/
|
||||||
|
func ColorToRGB(colorVal int) (color.RGBA) {
|
||||||
|
|
||||||
|
red := colorVal >> 16
|
||||||
|
green := (colorVal & 0x00FF00) >> 8
|
||||||
|
blue := colorVal & 0x0000FF
|
||||||
|
|
||||||
|
|
||||||
|
return color.RGBA{
|
||||||
|
R:uint8(red),
|
||||||
|
G:uint8(green),
|
||||||
|
B:uint8(blue),
|
||||||
|
A:uint8(255),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -0,0 +1,10 @@
|
||||||
|
package gocaptcha
|
||||||
|
|
||||||
|
type Point struct {
|
||||||
|
X int
|
||||||
|
Y int
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewPoint(x int,y int) *Point{
|
||||||
|
return &Point{ X :x,Y:y};
|
||||||
|
}
|
Loading…
Reference in New Issue