sqlite-parquet-vtable/parquet/parquet_filter.h

#ifndef PARQUET_FILTER_H
#define PARQUET_FILTER_H

#include <vector>
#include <cstdint>

enum ConstraintOperator {
  Equal,
  GreaterThan,
  LessThanOrEqual,
  LessThan,
  GreaterThanOrEqual,
  Match,
  Like,
  Glob,
  Regexp,
  NotEqual,
  IsNot,
  IsNotNull,
  IsNull,
  Is
};

enum ValueType {
  Null,
  Boolean,
  Integer,
  Double,
  Blob,
  Text
};

class Constraint {
  int column; // underlying column in the query
  ConstraintOperator op;
  ValueType type;

  bool boolValue;
  uintptr_t intValue;
  double doubleValue;
  // Doubles as string value
  std::vector<unsigned char> blobValue;

public:
  // Kind of a messy constructor function, but it's just for internal use, so whatever.
  Constraint(
    int column,
    ConstraintOperator op,
    ValueType type,
    bool boolValue,
    uintptr_t intValue,
    double doubleValue,
    std::vector<unsigned char> blobValue
  );

  int getColumn();
  ConstraintOperator getOperator();
  ValueType getType();
  bool getBool();
  uintptr_t getInt();
  double getDouble();
  std::vector<unsigned char> getBytes();
};

#endif