Stamps for mathematical optrations, to be used while building ASTree

Author: nataraj-hates-MS-for-stealing-github

blobstamper/blobstamper.h

@@ -24,3 +24,4 @@
 #include "galley.h"
 #include "stamp_enumerator.h"
 #include "stamp_lottery.h"
+#include "stamp_math_op.h"

blobstamper/stamp_lottery.h

@@ -16,6 +16,9 @@
  *
  ******************************************************************************/
 
+#ifndef STAMP_LOTTERY_H
+#define STAMP_LOTTERY_H
+
 template<class StampT> class StampLottery: public StampT
 {
   protected:
@@ -225,3 +228,5 @@ StampLottery<StampT>::Append(StampT & stamp)
   stamps.push_back(stamp);
   oracle_size = init_oracle_size(stamps);
 }
+
+#endif  // STAMP_LOTTERY_H

blobstamper/stamp_math_op.cpp

@@ -0,0 +1,35 @@
+/******************************************************************************
+ *
+ * Copyright 2021 Nikolay Shaplov (Postgres Professional)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ ******************************************************************************/
+
+#include <string>
+#include"stamp_math_op.h"
+
+std::string
+StampMathUnaryOp::ExtractStr(Blob &blob)
+{
+  return  op_name + "(" +  stamp.ExtractStr(blob) + ")";
+}
+
+
+std::string
+StampMathBinaryOp::ExtractStr(Blob &blob)
+{
+  std::vector<Blob> blobs = extract_internal(blob);
+  return (std::string)"(" +  stamp1.ExtractStr(blobs[0]) + " "+  op_name + " " + stamp2.ExtractStr(blobs[1]) + ")";
+}
+

blobstamper/stamp_math_op.h

@@ -0,0 +1,58 @@
+/******************************************************************************
+ *
+ * Copyright 2021 Nikolay Shaplov (Postgres Professional)
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ ******************************************************************************/
+
+#ifndef STAMP_MATH_OP_H
+#define STAMP_MATH_OP_H
+
+#include"blob.h"
+#include"stamp.h"
+#include"galley.h"
+
+
+class StampMathUnaryOp: public StampBaseStr
+{
+  protected:
+    std::string op_name;
+    StampBaseStr &stamp;
+  public:
+    virtual std::string ExtractStr(Blob &blob) override;
+    StampMathUnaryOp(std::string arg_op_name, StampBaseStr& arg_stamp) : op_name(arg_op_name), stamp(arg_stamp) {};
+    virtual int maxSize() override {return -1;};
+    virtual int minSize() override {return stamp.minSize();};
+};
+
+
+
+class StampMathBinaryOp: public StampBaseStr, public GalleySetBase
+{
+  protected:
+    std::string op_name;
+    StampBaseStr &stamp1;
+    StampBaseStr &stamp2;
+  public:
+    virtual std::string ExtractStr(Blob &blob) override;
+    StampMathBinaryOp(std::string arg_op_name, StampBaseStr& arg_stamp1, StampBaseStr& arg_stamp2) :
+        GalleySetBase({arg_stamp1, arg_stamp2}),
+        op_name(arg_op_name),
+        stamp1(arg_stamp1),
+        stamp2(arg_stamp2) {};
+    virtual int maxSize() override {return -1;};
+
+};
+
+#endif // STAMP_MATH_OP_H