This commit is contained in:
tovjemam 2024-10-13 19:32:55 +02:00
commit 704979f9bb
8 changed files with 873 additions and 0 deletions

29
CMakeLists.txt Normal file
View File

@ -0,0 +1,29 @@
# Minimum required CMake version
cmake_minimum_required(VERSION 3.10)
# Project name and version
project(Graph VERSION 1.0)
# Specify the C standard
set(CMAKE_C_STANDARD 90)
set(CMAKE_C_STANDARD_REQUIRED True)
# Add the executable target
add_executable(Graph
"main.c"
"lex.c"
"parser.c"
"tree.c"
)
# Optionally, you can set compiler warnings
if (MSVC)
target_compile_options(Graph PRIVATE /W4)
else()
target_compile_options(Graph PRIVATE -Wall -Wextra -pedantic)
endif()
# Optionally, set the output directory for the executable
# set_target_properties(Graph PROPERTIES
# RUNTIME_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/bin"
# )

191
lex.c Normal file
View File

@ -0,0 +1,191 @@
#include <stdlib.h>
#include <stdio.h>
#include "lex.h"
struct lexer {
const char* start;
const char* prev_p;
const char* p;
struct token tok;
};
struct lexer *lex_create(const char *str) {
struct lexer *lex = malloc(sizeof(struct lexer));
if (!lex)
return NULL;
lex->start = str;
lex->prev_p = str;
lex->p = str;
lex_next(lex);
return lex;
}
void lex_free(struct lexer *lex) {
free(lex);
}
static int is_whitespace(char p) {
return p == ' ' || p == '\t' || p == '\n' || p == '\r';
}
static void skip_whitespace(struct lexer *lex) {
while (is_whitespace(*lex->p))
++lex->p;
}
static int try_advance(struct lexer *lex, const char *str) {
const char *temp_p = lex->p;
while (1) {
if (!*str) {
lex->p = temp_p;
return 1;
}
if (*str != *temp_p) {
return 0;
}
++temp_p;
++str;
}
}
static int try_token(struct lexer *lex, const char *tok_str, enum token_type type) {
if (try_advance(lex, tok_str)) {
lex->tok.type = type;
return 1;
}
return 0;
}
static int try_fn(struct lexer *lex, const char *tok_str, enum math_fn fn) {
if (try_advance(lex, tok_str)) {
lex->tok.type = TOK_FUNCTION;
lex->tok.val.fn = fn;
return 1;
}
return 0;
}
static int try_number(struct lexer* lex) {
char *end;
double val;
val = strtod(lex->p, &end);
if (lex->p == end) return 0;
lex->tok.type = TOK_NUMBER;
lex->tok.val.num = val;
lex->p = end;
return 1;
}
void lex_next(struct lexer *lex) {
skip_whitespace(lex);
lex->prev_p = lex->p;
if (!*lex->p) {
lex->tok.type = TOK_EOF;
return;
}
if (try_token(lex, "+", TOK_PLUS)) return;
if (try_token(lex, "-", TOK_MINUS)) return;
if (try_token(lex, "*", TOK_MULTIPLY)) return;
if (try_token(lex, "/", TOK_DIVIDE)) return;
if (try_token(lex, "^", TOK_POWER)) return;
if (try_token(lex, "x", TOK_X)) return;
if (try_token(lex, "(", TOK_LEFT_PAREN)) return;
if (try_token(lex, ")", TOK_RIGHT_PAREN)) return;
if (try_fn(lex, "abs", FN_ABS)) return;
if (try_fn(lex, "exp", FN_EXP)) return;
if (try_fn(lex, "ln", FN_LN)) return;
if (try_fn(lex, "log", FN_LOG)) return;
if (try_fn(lex, "sinh", FN_SINH)) return;
if (try_fn(lex, "sin", FN_SIN)) return;
if (try_fn(lex, "cosh", FN_COSH)) return;
if (try_fn(lex, "cos", FN_COS)) return;
if (try_fn(lex, "tanh", FN_TANH)) return;
if (try_fn(lex, "tan", FN_TAN)) return;
if (try_fn(lex, "asin", FN_ASIN)) return;
if (try_fn(lex, "acos", FN_ACOS)) return;
if (try_fn(lex, "atan", FN_ATAN)) return;
if (try_number(lex)) return;
fprintf(stderr, "Lexer error - unrecognized sequence \"%s\"\n", lex->p);
lex_print_position(lex);
lex->tok.type = TOK_ERROR;
return;
}
struct token *lex_token(struct lexer *lex) {
return &lex->tok;
}
void lex_print_position(struct lexer *lex) {
int i;
int pos = lex->prev_p - lex->start;
fprintf(stderr, "At character %d\n", pos);
fprintf(stderr, " %s\n", lex->start);
fprintf(stderr, " ");
for (i = 0; i < pos; ++i)
fprintf(stderr, " ");
fprintf(stderr, "^\n----------------------------------------------------\n");
}
#ifdef LEX_DEBUG
void lex_debug_print_token(struct token *tok) {
static const char *token_str[] = {
"TOK_EOF",
"TOK_ERROR",
"TOK_NUMBER",
"TOK_PLUS",
"TOK_MINUS",
"TOK_MULTIPLY",
"TOK_DIVIDE",
"TOK_POWER",
"TOK_X",
"TOK_FUNCTION",
"TOK_LEFT_PAREN",
"TOK_RIGHT_PAREN"
};
static const char *fn_str[] = {
"FN_ABS",
"FN_EXP",
"FN_LN",
"FN_LOG",
"FN_SIN",
"FN_COS",
"FN_TAN",
"FN_ASIN",
"FN_ACOS",
"FN_ATAN",
"FN_SINH",
"FN_COSH",
"FN_TANH"
};
printf("%-20s ", token_str[tok->type]);
if (tok->type == TOK_NUMBER)
printf("%.2f\n", tok->val.num);
else if (tok->type == TOK_FUNCTION)
printf("%s\n", fn_str[tok->val.fn]);
else
printf("\n");
}
#endif /* LEX_DEBUG */

61
lex.h Normal file
View File

@ -0,0 +1,61 @@
#ifndef LEX_H
#define LEX_H
#define LEX_DEBUG
enum token_type {
TOK_EOF,
TOK_ERROR,
TOK_NUMBER,
TOK_PLUS,
TOK_MINUS,
TOK_MULTIPLY,
TOK_DIVIDE,
TOK_POWER,
TOK_X,
TOK_FUNCTION,
TOK_LEFT_PAREN,
TOK_RIGHT_PAREN
};
enum math_fn {
FN_ABS,
FN_EXP,
FN_LN,
FN_LOG,
FN_SIN,
FN_COS,
FN_TAN,
FN_ASIN,
FN_ACOS,
FN_ATAN,
FN_SINH,
FN_COSH,
FN_TANH
};
struct token {
enum token_type type;
union token_val {
double num;
enum math_fn fn;
} val;
};
struct lexer;
extern struct lexer *lex_create(const char *str);
extern void lex_free(struct lexer *lex);
extern void lex_next(struct lexer *lex);
extern struct token *lex_token(struct lexer *lex);
extern void lex_print_position(struct lexer *lex);
#ifdef LEX_DEBUG
extern void lex_debug_print_token(struct token *tok);
#endif /* LEX_DEBUG */
#endif /* LEX_H */

38
main.c Normal file
View File

@ -0,0 +1,38 @@
#include <stdio.h>
#include <stdlib.h>
#include "lex.h"
#include "parser.h"
int main(int argc, char *argv[]) {
struct lexer *lex;
struct expr_node *node;
if (argc < 2) {
printf("Usage: %s <expression>\n", argv[0]);
return 1;
}
/*
//struct lexer *lex = lex_create("x* sin (x ^ 2) * 5.5e+2* cosh (x )");
//struct lexer *lex = lex_create("-1 + -2 - 3 + 4");
*/
lex = lex_create(argv[1]);
node = parse_expression(lex);
lex_free(lex);
if (node) {
double x;
node_debug_print(node);
printf("polygon(");
for (x = -10.0; x < 10.0; x += 0.1) {
printf("(%.2f,%.2f),", x, node_eval(node, x));
}
printf("(10, -1000),(-10,-1000))\n");
}
node_free(node);
return 0;
}

263
parser.c Normal file
View File

@ -0,0 +1,263 @@
#include <stdlib.h>
#include <stdio.h>
#include "parser.h"
static int token_is(struct lexer *lex, enum token_type type) {
return lex_token(lex)->type == type;
}
static int accept(struct lexer *lex, enum token_type type) {
if (token_is(lex, type)) {
lex_next(lex);
return 1;
}
return 0;
}
static double token_num(struct lexer *lex) {
return lex_token(lex)->val.num;
}
static enum math_fn token_fn(struct lexer *lex) {
return lex_token(lex)->val.fn;
}
static void error_expected_one_of(struct lexer *lex, const int expected[], enum token_type got) {
static const char* token_name[] = {
"end of expression",
"lexer error",
"constant",
"+",
"-",
"*",
"/",
"^",
"x",
"function name",
"(",
")"
};
int separate = 0;
fprintf(stderr, "Syntax error - expected ");
for (; *expected >= 0; ++expected) {
if (separate)
fprintf(stderr, ", ");
else
separate = 1;
fprintf(stderr, "'%s'", token_name[*expected]);
}
fprintf(stderr, " - but got '%s'\n", token_name[got]);
lex_print_position(lex);
}
static void error_expected_single(struct lexer *lex, enum token_type expected, enum token_type got) {
static int expected_arr[] = { -1, -1 };
expected_arr[0] = expected;
error_expected_one_of(lex, expected_arr, got);
}
struct expr_node *parse_subexpression(struct lexer *lex);
static struct expr_node *parse_bracketed(struct lexer* lex) {
struct expr_node *node;
if (!accept(lex, TOK_LEFT_PAREN)) {
error_expected_single(lex, TOK_LEFT_PAREN, lex_token(lex)->type);
return NULL;
}
if (!(node = parse_subexpression(lex)))
return NULL;
if (!accept(lex, TOK_RIGHT_PAREN)) {
error_expected_single(lex, TOK_RIGHT_PAREN, lex_token(lex)->type);
node_free(node);
return NULL;
}
return node;
}
static struct expr_node *parse_base(struct lexer *lex) {
struct expr_node *inner;
if (token_is(lex, TOK_NUMBER)) {
double val = token_num(lex);
lex_next(lex);
return node_create_const(val);
}
if (accept(lex, TOK_X)) {
return node_create_x();
}
if (token_is(lex, TOK_FUNCTION)) {
enum math_fn fn = token_fn(lex);
lex_next(lex);
if (!(inner = parse_bracketed(lex)))
return NULL;
return node_create_fn(fn, inner);
}
if (token_is(lex, TOK_LEFT_PAREN)) {
return parse_bracketed(lex);
}
{
static const int expected[] = { TOK_NUMBER, TOK_X, TOK_FUNCTION, TOK_LEFT_PAREN, -1 };
error_expected_one_of(lex, expected, lex_token(lex)->type);
}
return NULL;
}
static struct expr_node* parse_unary(struct lexer *lex) {
if (accept(lex, TOK_MINUS)) {
struct expr_node *node, *inner;
if (!(inner = parse_base(lex)))
return NULL;
if (!(node = node_create_neg(inner))) {
node_free(inner);
return NULL;
}
return node;
}
accept(lex, TOK_PLUS);
return parse_base(lex);
}
static struct expr_node *parse_factor(struct lexer *lex) {
struct expr_node *node, *new_node, *inner;
if (!(node = parse_unary(lex)))
return NULL;
while (accept(lex, TOK_POWER)) {
if (!(inner = parse_unary(lex))) {
node_free(node);
return NULL;
}
if (!(new_node = node_create_pow(node, inner))) {
node_free(node);
node_free(inner);
return NULL;
}
node = new_node;
}
return node;
}
static struct expr_node *parse_term(struct lexer *lex) {
struct expr_node *node, *new_node, *inner;
if (!(node = parse_factor(lex)))
return NULL;
while (1) {
struct expr_node *(*create_node)(struct expr_node *left, struct expr_node *right);
if (accept(lex, TOK_MULTIPLY))
create_node = node_create_mult;
else if (accept(lex, TOK_DIVIDE))
create_node = node_create_div;
else
break;
if (!(inner = parse_factor(lex))) {
node_free(node);
return NULL;
}
if (!(new_node = create_node(node, inner))) {
node_free(node);
node_free(inner);
return NULL;
}
node = new_node;
}
return node;
}
struct expr_node *parse_subexpression(struct lexer *lex) {
struct expr_node *node, *new_node, *inner;
/*
if (accept(lex, TOK_MINUS)) {
if (!(term = parse_term(lex)))
return NULL;
if (!(node = node_create_neg(term))) {
node_free(term);
return NULL;
}
}
else {
accept(lex, TOK_PLUS);
if (!(node = parse_term(lex)))
return NULL;
}
*/
if (!(node = parse_term(lex)))
return NULL;
while (1) {
struct expr_node *(*create_node)(struct expr_node *left, struct expr_node *right);
if (accept(lex, TOK_PLUS))
create_node = node_create_add;
else if (accept(lex, TOK_MINUS))
create_node = node_create_sub;
else
break;
if (!(inner = parse_term(lex))) {
node_free(node);
return NULL;
}
if (!(new_node = create_node(node, inner))) {
node_free(node);
node_free(inner);
return NULL;
}
node = new_node;
}
return node;
}
struct expr_node *parse_expression(struct lexer *lex) {
struct expr_node *node;
if (!(node = parse_subexpression(lex)))
return NULL;
if (!token_is(lex, TOK_EOF)) {
error_expected_single(lex, TOK_EOF, lex_token(lex)->type);
node_free(node);
return NULL;
}
return node;
}

8
parser.h Normal file
View File

@ -0,0 +1,8 @@
#ifndef PARSER_H
#define PARSER_H
#include "tree.h"
extern struct expr_node *parse_expression(struct lexer *lex);
#endif /* PARSER_H */

232
tree.c Normal file
View File

@ -0,0 +1,232 @@
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include "tree.h"
static struct expr_node *alloc_node(void) {
return malloc(sizeof(struct expr_node));
}
struct expr_node *node_create_const(double val) {
struct expr_node *node = alloc_node();
if (!node) return NULL;
node->type = EXPR_CONST;
node->vals.num = val;
return node;
}
struct expr_node *node_create_neg(struct expr_node *unop) {
struct expr_node *node = alloc_node();
if (!node) return NULL;
node->type = EXPR_NEG;
node->vals.unop = unop;
return node;
}
static struct expr_node *create_binary_node(enum expr_type type, struct expr_node *left, struct expr_node *right) {
struct expr_node *node = alloc_node();
if (!node) return NULL;
node->type = type;
node->vals.binop.left = left;
node->vals.binop.right = right;
return node;
}
struct expr_node *node_create_add(struct expr_node *left, struct expr_node *right) {
return create_binary_node(EXPR_ADD, left, right);
}
struct expr_node *node_create_sub(struct expr_node *left, struct expr_node *right) {
return create_binary_node(EXPR_SUB, left, right);
}
struct expr_node *node_create_mult(struct expr_node *left, struct expr_node *right) {
return create_binary_node(EXPR_MULT, left, right);
}
struct expr_node *node_create_div(struct expr_node *left, struct expr_node *right) {
return create_binary_node(EXPR_DIV, left, right);
}
struct expr_node *node_create_pow(struct expr_node *base, struct expr_node *power) {
return create_binary_node(EXPR_POW, base, power);
}
struct expr_node *node_create_x(void) {
struct expr_node *node = alloc_node();
if (!node) return NULL;
node->type = EXPR_X;
return node;
}
struct expr_node *node_create_fn(enum math_fn fn, struct expr_node *arg) {
struct expr_node *node = alloc_node();
if (!node) return NULL;
node->type = EXPR_FN;
node->vals.fn.fn = fn;
node->vals.fn.arg = arg;
return node;
}
void node_free(struct expr_node *node) {
if (!node) return;
switch (node->type) {
case EXPR_ADD:
case EXPR_SUB:
case EXPR_MULT:
case EXPR_DIV:
case EXPR_POW:
node_free(node->vals.binop.left);
node_free(node->vals.binop.right);
break;
case EXPR_NEG:
node_free(node->vals.unop);
break;
case EXPR_FN:
node_free(node->vals.fn.arg);
break;
default:
break;
}
free(node);
}
static void debug_indent(int indent) {
int i;
for (i = 0; i < indent; ++i)
printf(" ");
}
static void debug_print(struct expr_node *node, int indent);
static void debug_print_binop(struct expr_node *node, const char* name, int indent) {
debug_indent(indent); printf("[%s]\n", name);
/*debug_indent(indent); printf("left:\n");*/
debug_print(node->vals.binop.left, indent + 1);
/*debug_indent(indent); printf("right:\n");*/
debug_print(node->vals.binop.right, indent + 1);
}
static void debug_print(struct expr_node *node, int indent) {
static const char* fn_str[] = {
"FN_ABS",
"FN_EXP",
"FN_LN",
"FN_LOG",
"FN_SIN",
"FN_COS",
"FN_TAN",
"FN_ASIN",
"FN_ACOS",
"FN_ATAN",
"FN_SINH",
"FN_COSH",
"FN_TANH"
};
switch (node->type) {
case EXPR_ADD:
debug_print_binop(node, "ADD", indent);
break;
case EXPR_SUB:
debug_print_binop(node, "SUB", indent);
break;
case EXPR_MULT:
debug_print_binop(node, "MULT", indent);
break;
case EXPR_DIV:
debug_print_binop(node, "DIV", indent);
break;
case EXPR_POW:
debug_print_binop(node, "POW", indent);
break;
case EXPR_NEG:
debug_indent(indent); printf("[NEG]\n");
/*debug_indent(indent); printf("unop:\n");*/
debug_print(node->vals.unop, indent + 1);
break;
case EXPR_CONST:
debug_indent(indent); printf("[CONST] %.2f\n", node->vals.num);
break;
case EXPR_X:
debug_indent(indent); printf("[X]\n");
break;
case EXPR_FN:
debug_indent(indent); printf("[FN] %s\n", fn_str[node->vals.fn.fn]);
/*debug_indent(indent); printf("arg:\n");*/
debug_print(node->vals.fn.arg, indent + 1);
break;
default:
break;
}
}
void node_debug_print(struct expr_node *node) {
debug_print(node, 0);
}
double node_eval(struct expr_node *node, double x) {
switch (node->type) {
case EXPR_CONST:
return node->vals.num;
case EXPR_X:
return x;
case EXPR_NEG:
return -node_eval(node->vals.unop, x);
case EXPR_ADD:
return node_eval(node->vals.binop.left, x) + node_eval(node->vals.binop.right, x);
case EXPR_SUB:
return node_eval(node->vals.binop.left, x) - node_eval(node->vals.binop.right, x);
case EXPR_MULT:
return node_eval(node->vals.binop.left, x) * node_eval(node->vals.binop.right, x);
case EXPR_DIV:
return node_eval(node->vals.binop.left, x) / node_eval(node->vals.binop.right, x);
case EXPR_POW:
return pow(node_eval(node->vals.binop.left, x), node_eval(node->vals.binop.right, x));
case EXPR_FN:
{
double inner = node_eval(node->vals.fn.arg, x);
switch (node->vals.fn.fn) {
case FN_ABS: return fabs(inner);
case FN_EXP: return exp(inner);
case FN_LN: return log(inner);
case FN_LOG: return log10(inner);
case FN_SIN: return sin(inner);
case FN_COS: return cos(inner);
case FN_TAN: return tan(inner);
case FN_ASIN: return asin(inner);
case FN_ACOS: return acos(inner);
case FN_ATAN: return atan(inner);
case FN_SINH: return sinh(inner);
case FN_COSH: return cosh(inner);
case FN_TANH: return tanh(inner);
}
}
}
return 0.0;
}

51
tree.h Normal file
View File

@ -0,0 +1,51 @@
#ifndef TREE_H
#define TREE_H
#include "lex.h"
enum expr_type {
EXPR_CONST,
EXPR_NEG,
EXPR_ADD,
EXPR_SUB,
EXPR_MULT,
EXPR_DIV,
EXPR_POW,
EXPR_X,
EXPR_FN
};
struct expr_node {
enum expr_type type;
union expr_vals {
struct expr_binop_vals {
struct expr_node *left;
struct expr_node *right;
} binop;
struct expr_fn_vals {
enum math_fn fn;
struct expr_node *arg;
} fn;
struct expr_node *unop;
double num;
} vals;
};
extern struct expr_node *node_create_const(double val);
extern struct expr_node *node_create_neg(struct expr_node *unop);
extern struct expr_node *node_create_add(struct expr_node *left, struct expr_node *right);
extern struct expr_node *node_create_sub(struct expr_node *left, struct expr_node *right);
extern struct expr_node *node_create_mult(struct expr_node *left, struct expr_node *right);
extern struct expr_node *node_create_div(struct expr_node *left, struct expr_node *right);
extern struct expr_node *node_create_pow(struct expr_node *base, struct expr_node *power);
extern struct expr_node *node_create_x(void);
extern struct expr_node *node_create_fn(enum math_fn fn, struct expr_node *arg);
extern void node_debug_print(struct expr_node *node);
extern double node_eval(struct expr_node *node, double x);
extern void node_free(struct expr_node *node);
#endif /* TREE_H */