// tcusto2.cpp -- Regression test program, custom view tests
// $Id$
// This is part of Metakit, the homepage is http://www.equi4.com/metakit/

#include "regress.h"

void TestCustom2()
{
  B(c11, Unique operation, 0);
  {
    c4_IntProp p1 ("p1"), p2 ("p2");

    c4_View v1, v2, v3;

    v1.Add(p1 [1] + p2 [11]);
    v1.Add(p1 [1] + p2 [22]);
    v1.Add(p1 [2] + p2 [33]);
    v1.Add(p1 [2] + p2 [33]);
    v1.Add(p1 [3] + p2 [44]);
    v1.Add(p1 [4] + p2 [55]);
    v1.Add(p1 [4] + p2 [55]);
    v1.Add(p1 [4] + p2 [55]);

    v2 = v1.Unique();
      A(v2.GetSize() == 5);
      A(p1 (v2[0]) == 1);
      A(p1 (v2[1]) == 1);
      A(p1 (v2[2]) == 2);
      A(p1 (v2[3]) == 3);
      A(p1 (v2[4]) == 4);

      A(p2 (v2[0]) == 11);
      A(p2 (v2[1]) == 22);
      A(p2 (v2[2]) == 33);
      A(p2 (v2[3]) == 44);
      A(p2 (v2[4]) == 55);

  } E;

  B(c12, Union operation, 0)
  {
    c4_IntProp p1 ("p1");

    c4_View v1, v2, v3;

    v1.Add(p1 [1]);
    v1.Add(p1 [2]);
    v1.Add(p1 [3]);

    v2.Add(p1 [2]);
    v2.Add(p1 [3]);
    v2.Add(p1 [4]);
    v2.Add(p1 [5]);

    v3 = v1.Union(v2);
      A(v3.GetSize() == 5);
      A(p1 (v3[0]) == 1);
      A(p1 (v3[1]) == 2);
      A(p1 (v3[2]) == 3);
      A(p1 (v3[3]) == 4);
      A(p1 (v3[4]) == 5);
  } E;

  B(c13, Intersect operation, 0)
  {
    c4_IntProp p1 ("p1");

    c4_View v1, v2, v3;

    v1.Add(p1 [1]);
    v1.Add(p1 [2]);
    v1.Add(p1 [3]);

    v2.Add(p1 [2]);
    v2.Add(p1 [3]);
    v2.Add(p1 [4]);
    v2.Add(p1 [5]);

    v3 = v1.Intersect(v2);
      A(v3.GetSize() == 2);
      A(p1 (v3[0]) == 2);
      A(p1 (v3[1]) == 3);
  } E;

  B(c14, Different operation, 0)
  {
    c4_IntProp p1 ("p1");

    c4_View v1, v2, v3;

    v1.Add(p1 [1]);
    v1.Add(p1 [2]);
    v1.Add(p1 [3]);

    v2.Add(p1 [2]);
    v2.Add(p1 [3]);
    v2.Add(p1 [4]);
    v2.Add(p1 [5]);

    v3 = v1.Different(v2);
      A(v3.GetSize() == 3);
      A(p1 (v3[0]) == 1);
      A(p1 (v3[1]) == 4);
      A(p1 (v3[2]) == 5);
  } E;

  B(c15, Minus operation, 0)
  {
    c4_IntProp p1 ("p1");

    c4_View v1, v2, v3;

    v1.Add(p1 [1]);
    v1.Add(p1 [2]);
    v1.Add(p1 [3]);

    v2.Add(p1 [2]);
    v2.Add(p1 [3]);
    v2.Add(p1 [4]);
    v2.Add(p1 [5]);

    v3 = v1.Minus(v2);
      A(v3.GetSize() == 1);
      A(p1 (v3[0]) == 1);
  } E;

  B(c16, View comparisons, 0)
  {
    c4_IntProp p1 ("p1");

    c4_View v1;
    v1.Add(p1 [1]);
    v1.Add(p1 [2]);
    v1.Add(p1 [3]);
    v1.Add(p1 [4]);
    v1.Add(p1 [5]);

      A(v1 == v1);
      A(v1 == v1.Slice(0));
      A(v1.Slice(0,2) < v1.Slice(0,3));
      A(v1.Slice(0,3) == v1.Slice(0,3));
      A(v1.Slice(0,4) > v1.Slice(0,3));
      A(v1.Slice(0,3) < v1.Slice(1,3));
      A(v1.Slice(0,3) < v1.Slice(1,4));
      A(v1.Slice(1,3) > v1.Slice(0,3));
      A(v1.Slice(1,4) > v1.Slice(0,3));
  } E;

  B(c17, Join operation, 0)
  {
    c4_StringProp p1 ("p1"), p2 ("p2");
    c4_IntProp p3 ("p3");

    c4_View v1, v2, v3;

    v1.Add(p1 [""]);
    v1.Add(p1 ["1"] + p2 ["a"]);
    v1.Add(p1 ["12"] + p2 ["ab"]);
    v1.Add(p1 ["123"] + p2 ["abc"]);

    v2.Add(p1 ["1"] + p3 [1]);
    v2.Add(p1 ["12"] + p3 [1]);
    v2.Add(p1 ["12"] + p3 [2]);
    v2.Add(p1 ["123"] + p3 [1]);
    v2.Add(p1 ["123"] + p3 [2]);
    v2.Add(p1 ["123"] + p3 [3]);

    v3 = v1.Join(p1, v2);     // inner join
      A(v3.GetSize() == 6);

      A(p1 (v3[0]) == (c4_String) "1");
      A(p1 (v3[1]) == (c4_String) "12");
      A(p1 (v3[2]) == (c4_String) "12");
      A(p1 (v3[3]) == (c4_String) "123");
      A(p1 (v3[4]) == (c4_String) "123");
      A(p1 (v3[5]) == (c4_String) "123");

      A(p2 (v3[0]) == (c4_String) "a");
      A(p2 (v3[1]) == (c4_String) "ab");
      A(p2 (v3[2]) == (c4_String) "ab");
      A(p2 (v3[3]) == (c4_String) "abc");
      A(p2 (v3[4]) == (c4_String) "abc");
      A(p2 (v3[5]) == (c4_String) "abc");

      A(p3 (v3[0]) == 1);
      A(p3 (v3[1]) == 1);
      A(p3 (v3[2]) == 2);
      A(p3 (v3[3]) == 1);
      A(p3 (v3[4]) == 2);
      A(p3 (v3[5]) == 3);

    v3 = v1.Join(p1, v2, true);   // outer join
      A(v3.GetSize() == 7);

      A(p1 (v3[0]) == (c4_String) "");
      A(p1 (v3[1]) == (c4_String) "1");
      A(p1 (v3[2]) == (c4_String) "12");
      A(p1 (v3[3]) == (c4_String) "12");
      A(p1 (v3[4]) == (c4_String) "123");
      A(p1 (v3[5]) == (c4_String) "123");
      A(p1 (v3[6]) == (c4_String) "123");

      A(p2 (v3[0]) == (c4_String) "");
      A(p2 (v3[1]) == (c4_String) "a");
      A(p2 (v3[2]) == (c4_String) "ab");
      A(p2 (v3[3]) == (c4_String) "ab");
      A(p2 (v3[4]) == (c4_String) "abc");
      A(p2 (v3[5]) == (c4_String) "abc");
      A(p2 (v3[6]) == (c4_String) "abc");

      A(p3 (v3[0]) == 0);
      A(p3 (v3[1]) == 1);
      A(p3 (v3[2]) == 1);
      A(p3 (v3[3]) == 2);
      A(p3 (v3[4]) == 1);
      A(p3 (v3[5]) == 2);
      A(p3 (v3[6]) == 3);
  } E;

  B(c18, Groupby sort fix, 0) // fails in 1.8.4 (from P. Ritter, 14-10-1998)
  {
    c4_StringProp p1 ("Country");
    c4_StringProp p2 ("City");
    c4_ViewProp p3 ("SubList");

    c4_View v1, v2, v3;

    v1.Add(p1 ["US"]   + p2 ["Philadelphia"]);
    v1.Add(p1 ["France"]   + p2 ["Bordeaux"]);
    v1.Add(p1 ["US"]   + p2 ["Miami"]);
    v1.Add(p1 ["France"]   + p2 ["Paris"]);
    v1.Add(p1 ["US"]   + p2 ["Boston"]);
    v1.Add(p1 ["France"]   + p2 ["Nice"]);
    v1.Add(p1 ["US"]   + p2 ["NY"]);
    v1.Add(p1 ["US"]   + p2 ["Miami"]);

    v2 = v1.GroupBy(p1, p3);
      A(v2.GetSize() == 2);
      A(p1 (v2[0]) == (c4_String) "France");
      A(p1 (v2[1]) == (c4_String) "US");

    v3 = p3 (v2[0]);
      A(v3.GetSize() == 3);
      A(p2 (v3[0]) == (c4_String) "Bordeaux");
      A(p2 (v3[1]) == (c4_String) "Nice");
      A(p2 (v3[2]) == (c4_String) "Paris");
    v3 = p3 (v2[1]);
      A(v3.GetSize() == 5);
      A(p2 (v3[0]) == (c4_String) "Boston");
      A(p2 (v3[1]) == (c4_String) "Miami");
      A(p2 (v3[2]) == (c4_String) "Miami");
      A(p2 (v3[3]) == (c4_String) "NY");
      A(p2 (v3[4]) == (c4_String) "Philadelphia");
  } E;

  B(c19, JoinProp operation, 0) // moved, used to also be called c15
  {
    c4_StringProp p1 ("p1");
    c4_ViewProp p2 ("p2");
    c4_IntProp p3 ("p3");

    c4_View v1, v2a, v2b, v2c, v3;

    v2a.Add(p3 [1]);
    v2a.Add(p3 [2]);
    v2a.Add(p3 [3]);
    v1.Add(p1 ["123"] + p2 [v2a]);

    v2b.Add(p3 [1]);
    v2b.Add(p3 [2]);
    v1.Add(p1 ["12"] + p2 [v2b]);

    v2c.Add(p3 [1]);
    v1.Add(p1 ["1"] + p2 [v2c]);

    v1.Add(p1 [""]);

    v3 = v1.JoinProp(p2);     // inner join
      A(v3.GetSize() == 6);

      A(p1 (v3[0]) == (c4_String) "123");
      A(p1 (v3[1]) == (c4_String) "123");
      A(p1 (v3[2]) == (c4_String) "123");
      A(p1 (v3[3]) == (c4_String) "12");
      A(p1 (v3[4]) == (c4_String) "12");
      A(p1 (v3[5]) == (c4_String) "1");

      A(p3 (v3[0]) == 1);
      A(p3 (v3[1]) == 2);
      A(p3 (v3[2]) == 3);
      A(p3 (v3[3]) == 1);
      A(p3 (v3[4]) == 2);
      A(p3 (v3[5]) == 1);

    v3 = v1.JoinProp(p2, true);   // outer join
      A(v3.GetSize() == 7);

      A(p1 (v3[0]) == (c4_String) "123");
      A(p1 (v3[1]) == (c4_String) "123");
      A(p1 (v3[2]) == (c4_String) "123");
      A(p1 (v3[3]) == (c4_String) "12");
      A(p1 (v3[4]) == (c4_String) "12");
      A(p1 (v3[5]) == (c4_String) "1");
      A(p1 (v3[6]) == (c4_String) "");

      A(p3 (v3[0]) == 1);
      A(p3 (v3[1]) == 2);
      A(p3 (v3[2]) == 3);
      A(p3 (v3[3]) == 1);
      A(p3 (v3[4]) == 2);
      A(p3 (v3[5]) == 1);
      A(p3 (v3[6]) == 0);
  } E;

  B(c20, Wide cartesian product, 0)
  {
      // added 2nd prop's to do a better test - 1999-12-23
    c4_IntProp p1 ("p1");
    c4_IntProp p2 ("p2");
    c4_IntProp p3 ("p3");
    c4_IntProp p4 ("p4");

    c4_View v1;
    v1.Add(p1 [123] + p2 [321]);
    v1.Add(p1 [234] + p2 [432]);
    v1.Add(p1 [345] + p2 [543]);

    c4_View v2;
    v2.Add(p3 [111] + p4 [11]);
    v2.Add(p3 [222] + p4 [22]);

    c4_View v3 = v1.Product(v2);
      A(v3.GetSize() == 6);
      A(p1 (v3[0]) == 123);
      A(p2 (v3[0]) == 321);
      A(p3 (v3[0]) == 111);
      A(p4 (v3[0]) == 11);
      A(p1 (v3[1]) == 123);
      A(p2 (v3[1]) == 321);
      A(p3 (v3[1]) == 222);
      A(p4 (v3[1]) == 22);
      A(p1 (v3[2]) == 234);
      A(p2 (v3[2]) == 432);
      A(p3 (v3[2]) == 111);
      A(p4 (v3[2]) == 11);
      A(p1 (v3[3]) == 234);
      A(p2 (v3[3]) == 432);
      A(p3 (v3[3]) == 222);
      A(p4 (v3[3]) == 22);
      A(p1 (v3[4]) == 345);
      A(p2 (v3[4]) == 543);
      A(p3 (v3[4]) == 111);
      A(p4 (v3[4]) == 11);
      A(p1 (v3[5]) == 345);
      A(p2 (v3[5]) == 543);
      A(p3 (v3[5]) == 222);
      A(p4 (v3[5]) == 22);

    v1.Add(p1 [456]);
      A(v3.GetSize() == 8);
    v2.Add(p2 [333]);
      A(v3.GetSize() == 12);
  } E;

  B(c21, Join on compound key, 0)
  {
    c4_IntProp p1 ("p1"), p2 ("p2"), p3 ("p3"), p4 ("p4");

    c4_View v1, v2, v3;

    v1.Add(p1 [1] + p2 [11] + p3 [111]);
    v1.Add(p1 [2] + p2 [22] + p3 [222]);
    v1.Add(p1 [3] + p2 [22] + p3 [111]);

    v2.Add(p2 [11] + p3 [111] + p4 [1111]);
    v2.Add(p2 [22] + p3 [222] + p4 [2222]);
    v2.Add(p2 [22] + p3 [222] + p4 [3333]);
    v2.Add(p2 [22] + p3 [333] + p4 [4444]);

  // this works here, but it fails in Python, i.e. Mk4py 2.4.0
  v3 = v1.Join((p2, p3), v2);

      A(v3.GetSize() == 3);

      A(p1 (v3[0]) == 1);
      A(p1 (v3[1]) == 2);
      A(p1 (v3[2]) == 2);

      A(p2 (v3[0]) == 11);
      A(p2 (v3[1]) == 22);
      A(p2 (v3[2]) == 22);

      A(p3 (v3[0]) == 111);
      A(p3 (v3[1]) == 222);
      A(p3 (v3[2]) == 222);

      A(p4 (v3[0]) == 1111);
      A(p4 (v3[1]) == 2222);
      A(p4 (v3[2]) == 3333);
  } E;

  B(c22, Groupby with selection, 0)
  {
    c4_Storage s1;
    c4_View v1 = s1.GetAs("v1[p1:I,p2:I,p3:I]");
    c4_IntProp p1 ("p1"), p2 ("p2"), p3 ("p3");
    c4_ViewProp p4 ("p4");

    v1.Add(p1[0] + p2[1] + p3[10]);
    v1.Add(p1[1] + p2[1] + p3[20]);
    v1.Add(p1[2] + p2[2] + p3[30]);
    v1.Add(p1[3] + p2[3] + p3[40]);
    v1.Add(p1[4] + p2[3] + p3[50]);

    s1.Commit();
      A(v1.GetSize() == 5);

    c4_View v2 = v1.GroupBy(p2,p4);
      A(v2.GetSize() == 3);

    c4_View v3 = p4 (v2[0]);
      A(v3.GetSize() == 2);
      A(p3 (v3[0]) == 10);
      A(p3 (v3[1]) == 20);

    c4_View v4 = p4 (v2[1]);
      A(v4.GetSize() == 1);
      A(p3 (v4[0]) == 30);

    c4_View v5 = p4 (v2[2]);
      A(v5.GetSize() == 2);
      A(p3 (v5[0]) == 40);
      A(p3 (v5[1]) == 50);

    c4_View v6 = v4.Sort();
      A(v6.GetSize() == 1);
      A(p1 (v6[0]) == 2);
      A(p3 (v6[0]) == 30);

  } E;
}