Add new trigonometric functions to postgres

aipescience · Sep 11, 2024 · 4dbf02b · 4dbf02b
1 parent f8f304c
commit 4dbf02b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/src/queryparser/postgresql/PostgreSQLLexer.g4 b/src/queryparser/postgresql/PostgreSQLLexer.g4
@@ -31,13 +31,17 @@ fragment Z_ :	'z' | 'Z';
 
 ABS				    : A_ B_ S_ ;
 ACOS				: A_ C_ O_ S_ ;
+ACOSD				: A_ C_ O_ S_ D_ ;
 ALL				    : A_ L_ L_  ;
 ANY				    : A_ N_ Y_ ;
 ASC				    : A_ S_ C_  ;
 ASIN				: A_ S_ I_ N_  ;
+ASIND				: A_ S_ I_ N_ D_ ;
 AS_SYM				: A_ S_  ;
 ATAN				: A_ T_ A_ N_  ;
+ATAND				: A_ T_ A_ N_ D_  ;
 ATAN2				: A_ T_ A_ N_ '2'  ;
+ATAN2D				: A_ T_ A_ N_ '2' D_  ;
 AVG				    : A_ V_ G_;
 BETWEEN				: B_ E_ T_ W_ E_ E_ N_  ;
 BIGINT              : B_ I_ G_ I_ N_ T_ ;
@@ -59,7 +63,9 @@ CONCAT				: C_ O_ N_ C_ A_ T_  ;
 CONCAT_WS			: C_ O_ N_ C_ A_ T_ '_' W_ S_  ;
 CONVERT_SYM			: C_ O_ N_ V_ E_ R_ T_  ;
 COS				    : C_ O_ S_  ;
+COSD			    : C_ O_ S_ D_  ;
 COT				    : C_ O_ T_  ;
+COTD			    : C_ O_ T_ D_  ;
 COUNT				: C_ O_ U_ N_ T_  ;
 CROSS				: C_ R_ O_ S_ S_  ;
 DATE_PART           : D_ A_ T_ E_ '_' P_ A_ R_ T_ ;
@@ -107,6 +113,7 @@ LIKE_SYM			: L_ I_ K_ E_  ;
 LIMIT				: L_ I_ M_ I_ T_  ;
 LN				    : L_ N_  ;
 LOG				    : L_ O_ G_  ;
+LOG10               : L_ O_ G_ '10' ;
 LOWER				: (L_ O_ W_ E_ R_) | (L_ C_ A_ S_ E_) ;
 LPAD				: L_ P_ A_ D_  ;
 LTRIM				: L_ T_ R_ I_ M_  ;
@@ -154,6 +161,7 @@ SHARE_SYM			: S_ H_ A_ R_ E_  ;
 SIGN				: S_ I_ G_ N_  ;
 SIGNED_SYM			: S_ I_ G_ N_ E_ D_  ;
 SIN				    : S_ I_ N_  ;
+SIND			    : S_ I_ N_ D_  ;
 SOUNDS_SYM          : S_ O_ U_ N_ D_ S_ ;
 SQUARE_DEGREES      : S_ Q_ U_ A_ R_ E_ '_' D_ E_ G_ R_ E_ E_ S_ ;
 SQRT				: S_ Q_ R_ T_  ;
@@ -166,6 +174,7 @@ SUBSTRING			: (S_ U_ B_ S_ T_ R_ I_ N_ G_) | (S_ U_ B_ S_ T_ R_) ;
 SUM				    : S_ U_ M_  ;
 SYMMETRIC           : S_ Y_ M_ M_ E_ T_ R_ I_ C_ ;
 TAN				    : T_ A_ N_  ;
+TAND			    : T_ A_ N_ D_  ;
 THEN_SYM			: T_ H_ E_ N_  ;
 TIME_SYM			: T_ I_ M_ E_  ;
 TIMESTAMP			: T_ I_ M_ E_ S_ T_ A_ M_ P_  ;

diff --git a/src/queryparser/postgresql/PostgreSQLParser.g4 b/src/queryparser/postgresql/PostgreSQLParser.g4
@@ -48,10 +48,11 @@ group_functions:
 	| STDDEV | STDDEV_POP | STDDEV_SAMP | VAR_POP | VAR_SAMP | VARIANCE ;
 
 number_functions:
-	  ABS | ACOS | ASIN | ATAN2 | ATAN | CBRT | CEIL | CEILING | COS | COT
-    | DEGREES | DIV | EXP | FLOOR | LN | LOG | MOD | PI | POW
-    | POWER | RADIANS | RANDOM | ROUND | SIGN | SIN | SQUARE_DEGREES | SQRT
-    | STERADIANS | TAN | TRUNC ;
+      ABS | ACOS | ACOSD | ASIN | ASIND | ATAN2 | ATAN2D | ATAN | ATAND
+    | CBRT | CEIL | CEILING | COS | COSD | COT | COTD
+    | DEGREES | DIV | EXP | FLOOR | LN | LOG | LOG10 | MOD | PI | POW
+    | POWER | RADIANS | RANDOM | ROUND | SIGN | SIN | SIND | SQUARE_DEGREES
+    | SQRT | STERADIANS | TAN | TAND | TRUNC ;
 
 other_functions:
       ENCODE | MD5 ;