From abb94f836b21e1fe76e83675dd09b4236a689a45 Mon Sep 17 00:00:00 2001 From: Fredrik Tolf Date: Fri, 20 Mar 2015 06:14:10 +0100 Subject: [PATCH] Added more indexed types. --- didex/index.py | 22 ++++++++++++++++++++-- 1 file changed, 20 insertions(+), 2 deletions(-) diff --git a/didex/index.py b/didex/index.py index a1ca061..4d42506 100644 --- a/didex/index.py +++ b/didex/index.py @@ -1,4 +1,4 @@ -import struct, contextlib +import struct, contextlib, math from . import db, lib from .db import bd, txnfun @@ -47,7 +47,25 @@ class maybe(object): else: return self.bk.compare(a[1:], b[1:]) -t_int = simpletype.struct(">Q") +def floatcmp(a, b): + if math.isnan(a) and math.isnan(b): + return 0 + elif math.isnan(a): + return -1 + elif math.isnan(b): + return 1 + elif a < b: + return -1 + elif a > b: + return 1 + else: + return 0 + +t_int = simpletype.struct(">q") +t_uint = simpletype.struct(">Q") +t_float = simpletype.struct(">d") +t_float.compare = floatcmp +t_str = simpletype((lambda ob: ob.encode("utf-8")), (lambda dat: dat.decode("utf-8"))) class index(object): def __init__(self, db, name, datatype): -- 2.11.0