add MapToG2

include/mcl/bn.hpp

@@ -119,6 +119,12 @@ struct MapTo {
 	{
 		return gmp::legendre(x.getMpz(), Fp::getOp().mp);
 	}
+	int legendre(const Fp2& x) const
+	{
+		Fp y;
+		Fp2::norm(y, x);
+		return legendre(y);
+	}
 	MapTo()
 	{
 		if (!Fp::squareRoot(c1, -3)) throw cybozu::Exception("MapTo:c1");
@@ -156,6 +162,35 @@ struct MapTo {
 	ERR_POINT:
 		throw cybozu::Exception("MapTo:calcG1:bad") << t;
 	}
+	void calcG2(G2& P, const Fp2& t) const
+	{
+		Fp2 x, y, w;
+		bool negative = legendre(t) < 0;
+		if (t.isZero()) goto ERR_POINT;
+		Fp2::sqr(w, t);
+		w += G2::b_;
+		w.a += Fp::one();
+		if (w.isZero()) goto ERR_POINT;
+		Fp2::inv(w, w);
+		w.a *= c1;
+		w.b *= c1;
+		w *= t;
+		for (int i = 0; i < 3; i++) {
+			switch (i) {
+			case 0: x = - t * w; x.a += c2; break;
+			case 1: Fp2::neg(x, x); x.a -= Fp::one(); break;
+			case 2: Fp2::sqr(x, w); Fp2::inv(x, x); x.a += Fp::one(); break;
+			}
+			G2::getWeierstrass(y, x);
+			if (Fp2::squareRoot(y, y)) {
+				if (negative) Fp2::neg(y, y);
+				P.set(x, y);
+				return;
+			}
+		}
+	ERR_POINT:
+		throw cybozu::Exception("MapTo:calcG2:bad") << t;
+	}
 };
 
 template<class Fp>

include/mcl/fp_tower.hpp

@@ -153,6 +153,13 @@ public:
 		Fp::mul(y.b, x.b, t2);
 		return true;
 	}
+	static void inline norm(Fp& y, const Fp2T& x)
+	{
+		Fp aa, bb;
+		Fp::sqr(aa, x.a);
+		Fp::sqr(bb, x.b);
+		Fp::add(y, aa, bb);
+	}
 
 	static const Fp& getXi_a() { return xi_a_; }
 	static void init(uint32_t xi_a)

test/bn_test.cpp

@@ -95,7 +95,7 @@ CYBOZU_TEST_AUTO(naive)
 	CYBOZU_BENCH("finalExp", BN::finalExp, e1, e1); // 1.3Mclk
 }
 
-CYBOZU_TEST_AUTO(MapTo)
+CYBOZU_TEST_AUTO(MapToG1)
 {
 	mcl::bn::MapTo<Fp> mapTo;
 
@@ -108,6 +108,16 @@ CYBOZU_TEST_AUTO(MapTo)
 	CYBOZU_TEST_EXCEPTION(mapTo.calcG1(g, -mapTo.c1), cybozu::Exception);
 }
 
+CYBOZU_TEST_AUTO(MapToG2)
+{
+	mcl::bn::MapTo<Fp> mapTo;
+
+	G2 g;
+	for (int i = 1; i < 10; i++) {
+		mapTo.calcG2(g, i);
+	}
+}
+
 CYBOZU_TEST_AUTO(stream)
 {
 	const TestSet& ts = g_testSetTbl[0];