1
This commit is contained in:
Alex Yatskov 2016-10-30 19:13:15 -07:00
parent 719ad26978
commit 0add5cf2f2
2 changed files with 72 additions and 146 deletions

2
eb

@ -1 +1 @@
Subproject commit 5497a76516340fd01241a465d3a9e17d788a4cc3 Subproject commit 3c455a6426baeccde823809627991ee943b3ac15

216
main.c
View File

@ -1,178 +1,104 @@
#include <stdio.h> #include <stdio.h>
#include <string.h> #include <string.h>
#include <eb/eb.h> #include "eb/eb/eb.h"
#include <eb/error.h> #include "eb/eb/error.h"
#include <eb/text.h> #include "eb/eb/text.h"
void print_book_info(EB_Book* book) { /* Constants */
printf("Book info for %s:\n", book->path);
EB_Disc_Code disc_code; const unsigned MAX_ENTRY_HITS = 128;
if (eb_disc_type(book, &disc_code) == EB_SUCCESS) { const unsigned MAX_ENTRY_TEXT = 1024;
printf("\tBook type: ");
switch (disc_code) { /* Local functions */
case EB_DISC_EB:
printf("EB_DISC_EB\n"); static void dump_hits(EB_Book* book) {
break; EB_Hit hits[MAX_ENTRY_HITS];
case EB_DISC_EPWING: int hit_count = 0;
printf("EB_DISC_EPWING\n");
break; do {
case EB_DISC_INVALID: if (eb_hit_list(book, MAX_ENTRY_HITS, hits, &hit_count) != EB_SUCCESS) {
printf("EB_DISC_INVALID\n"); fprintf(stderr, "error: could not get hit list\n");
break; break;
default: }
printf("undefined\n");
break; for (int i = 0; i < hit_count; ++i) {
char text[MAX_ENTRY_TEXT];
ssize_t text_length;
eb_seek_text(book, &hits[i].text);
eb_read_text(book, NULL, NULL, NULL, MAX_ENTRY_TEXT, text, &text_length);
puts(text);
} }
} }
else { while (hit_count > 0);
fprintf(stderr, "\tError: failed to get book type\n"); }
}
EB_Character_Code char_code; static void dump_book(EB_Book* book) {
if (eb_character_code(book, &char_code) == EB_SUCCESS) { if (eb_search_all_alphabet(book) == EB_SUCCESS) {
printf("\tCharacter type: "); dump_hits(book);
switch (char_code) {
case EB_CHARCODE_ISO8859_1:
printf("EB_CHARCODE_ISO8859_1\n");
break;
case EB_CHARCODE_JISX0208:
printf("EB_CHARCODE_JISX0208\n");
break;
case EB_CHARCODE_JISX0208_GB2312:
printf("EB_CHARCODE_JISX0208_GB2312\n");
break;
case EB_CHARCODE_INVALID:
printf("EB_CHARCODE_INVALID\n");
break;
default:
printf("undefined\n");
break;
}
} }
else { else {
fprintf(stderr, "\tError: failed to get book character code\n"); printf("notice: skiping alphabet search\n");
} }
EB_Subbook_Code sub_codes[EB_MAX_SUBBOOKS]; if (eb_search_all_kana(book) == EB_SUCCESS) {
int sub_count = 0; dump_hits(book);
if (eb_subbook_list(book, sub_codes, &sub_count) == EB_SUCCESS) {
printf("\tFound %d sub-book(s):\n", sub_count);
char title[EB_MAX_TITLE_LENGTH + 1];
for (int i = 0; i < sub_count; ++i) {
if (eb_subbook_title2(book, sub_codes[i], title) != EB_SUCCESS) {
strcpy(title, "<unknown>");
}
printf("\t\t %d: %s\n", sub_codes[i], title);
}
} }
else { else {
fprintf(stderr, "\tError: could not get sub-book list\n"); printf("notice: skiping kana search\n");
}
if (eb_search_all_asis(book) == EB_SUCCESS) {
dump_hits(book);
}
else {
printf("notice: skiping asis search\n");
} }
} }
int dump_book(EB_Book* book) { static int process(const char path[]) {
EB_Position position;
position.offset = 0;
position.page = 1;
for (;;) {
if (eb_seek_text(book, &position) != EB_SUCCESS) {
fprintf(stderr, "Failed to seek\n");
return 1;
}
const unsigned MAX_LENGTH = 10000;
char buffer[MAX_LENGTH + 1];
ssize_t text_length;
/* if (eb_read_text(book, NULL, NULL, NULL, MAX_LENGTH, buffer, &text_length) != EB_SUCCESS) { */
if (eb_read_heading(book, NULL, NULL, NULL, MAX_LENGTH, buffer, &text_length) != EB_SUCCESS) {
fprintf(stderr, "an error occurs.\n");
return 1;
}
puts(buffer);
getchar();
++position.offset;
}
return 0;
}
int find_term(EB_Book* book, const char term[]) {
if (eb_search_word(book, term) != EB_SUCCESS) {
fprintf(stderr, "Error: search failed\n");
return 1;
}
const unsigned MAX_HITS = 100;
EB_Hit hits[MAX_HITS];
int hit_count;
if (eb_hit_list(book, MAX_HITS, hits, &hit_count) != EB_SUCCESS) {
fprintf(stderr, "Error: could not get hit list\n");
return 1;
}
printf("Found %d results\n", hit_count);
for (int i = 0; i < hit_count; ++i) {
/* const unsigned MAX_HEADING = 1000; */
/* char heading[MAX_HEADING]; */
/* ssize_t heading_length; */
/* eb_seek_text(book, &hits[i].heading); */
/* eb_read_heading(book, NULL, NULL, NULL, MAX_HEADING, heading, &heading_length); */
/* printf("%s\n", heading); */
const unsigned MAX_TEXT = 1000;
char text[MAX_TEXT];
ssize_t text_length;
/* book->subbook_current->word_asis.start_page */
eb_seek_text(book, &hits[i].text);
eb_read_text(book, NULL, NULL, NULL, MAX_TEXT, text, &text_length);
printf("%s\n", text);
}
return 0;
}
int process(const char path[]) {
if (eb_initialize_library() != EB_SUCCESS) { if (eb_initialize_library() != EB_SUCCESS) {
fprintf(stderr, "error: failed to initialize library\n"); fprintf(stderr, "error: failed to initialize library\n");
return 1; return 1;
} }
EB_Book book; EB_Book book;
if (eb_bind(&book, path) != EB_SUCCESS) { eb_initialize_book(&book);
fprintf(stderr, "error: failed to bind book\n");
eb_finalize_book(&book);
return 1;
}
if (eb_set_subbook(&book, 0) != EB_SUCCESS) { do {
fprintf(stderr, "Failed to set sub-book\n"); if (eb_bind(&book, path) != EB_SUCCESS) {
eb_finalize_book(&book); fprintf(stderr, "error: failed to bind book\n");
return 1; break;
} }
print_book_info(&book); EB_Subbook_Code sub_codes[EB_MAX_SUBBOOKS];
find_term(&book, "computer"); int sub_count = 0;
/* dump_book(&book); */
if (eb_subbook_list(&book, sub_codes, &sub_count) != EB_SUCCESS) {
fprintf(stderr, "error: failed to get sub-book list\n");
break;
}
for (int i = 0; i < sub_count; ++i) {
if (eb_set_subbook(&book, sub_codes[i]) == EB_SUCCESS) {
dump_book(&book);
}
else {
fprintf(stderr, "error: failed to set sub-book\n");
}
}
}
while (0);
eb_finalize_book(&book); eb_finalize_book(&book);
eb_finalize_library(); eb_finalize_library();
return 0; return 0;
} }
/* Entry point */
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
if (argc != 2) { if (argc != 2) {
fprintf(stderr, "usage: %s path\n", argv[0]); fprintf(stderr, "usage: %s path\n", argv[0]);