granicus.if.org Git - postgresql/blob - src/test/regress/regress.c

   1 /*
   2  * $Header: /cvsroot/pgsql/src/test/regress/regress.c,v 1.33 1999/07/17 20:18:52 momjian Exp $
   3  */
   4
   5 #include <float.h>                              /* faked on sunos */
   6
   7 #include "postgres.h"
   8
   9 #include "utils/geo_decls.h"    /* includes <math.h> */
  10 #include "executor/executor.h"  /* For GetAttributeByName */
  11
  12 #define P_MAXDIG 12
  13 #define LDELIM                  '('
  14 #define RDELIM                  ')'
  15 #define DELIM                   ','
  16
  17 typedef void *TUPLE;
  18
  19 extern double *regress_dist_ptpath(Point *pt, PATH *path);
  20 extern double *regress_path_dist(PATH *p1, PATH *p2);
  21 extern PATH *poly2path(POLYGON *poly);
  22 extern Point *interpt_pp(PATH *p1, PATH *p2);
  23 extern void regress_lseg_construct(LSEG *lseg, Point *pt1, Point *pt2);
  24 extern char overpaid(TUPLE tuple);
  25 extern int      boxarea(BOX *box);
  26 extern char *reverse_name(char *string);
  27
  28 /*
  29 ** Distance from a point to a path
  30 */
  31 double *
  32 regress_dist_ptpath(pt, path)
  33 Point      *pt;
  34 PATH       *path;
  35 {
  36         double     *result;
  37         double     *tmp;
  38         int                     i;
  39         LSEG            lseg;
  40
  41         switch (path->npts)
  42         {
  43                 case 0:
  44                         result = palloc(sizeof(double));
  45                         *result = Abs((double) DBL_MAX);        /* +infinity */
  46                         break;
  47                 case 1:
  48                         result = point_distance(pt, &path->p[0]);
  49                         break;
  50                 default:
  51
  52                         /*
  53                          * the distance from a point to a path is the smallest
  54                          * distance from the point to any of its constituent segments.
  55                          */
  56                         Assert(path->npts > 1);
  57                         result = palloc(sizeof(double));
  58                         for (i = 0; i < path->npts - 1; ++i)
  59                         {
  60                                 regress_lseg_construct(&lseg, &path->p[i], &path->p[i + 1]);
  61                                 tmp = dist_ps(pt, &lseg);
  62                                 if (i == 0 || *tmp < *result)
  63                                         *result = *tmp;
  64                                 pfree(tmp);
  65
  66                         }
  67                         break;
  68         }
  69         return result;
  70 }
  71
  72 /* this essentially does a cartesian product of the lsegs in the
  73    two paths, and finds the min distance between any two lsegs */
  74 double *
  75 regress_path_dist(p1, p2)
  76 PATH       *p1;
  77 PATH       *p2;
  78 {
  79         double     *min,
  80                            *tmp;
  81         int                     i,
  82                                 j;
  83         LSEG            seg1,
  84                                 seg2;
  85
  86         regress_lseg_construct(&seg1, &p1->p[0], &p1->p[1]);
  87         regress_lseg_construct(&seg2, &p2->p[0], &p2->p[1]);
  88         min = lseg_distance(&seg1, &seg2);
  89
  90         for (i = 0; i < p1->npts - 1; i++)
  91                 for (j = 0; j < p2->npts - 1; j++)
  92                 {
  93                         regress_lseg_construct(&seg1, &p1->p[i], &p1->p[i + 1]);
  94                         regress_lseg_construct(&seg2, &p2->p[j], &p2->p[j + 1]);
  95
  96                         if (*min < *(tmp = lseg_distance(&seg1, &seg2)))
  97                                 *min = *tmp;
  98                         pfree(tmp);
  99                 }
 100
 101         return min;
 102 }
 103
 104 PATH *
 105 poly2path(poly)
 106 POLYGON    *poly;
 107 {
 108         int                     i;
 109         char       *output = (char *) palloc(2 * (P_MAXDIG + 1) * poly->npts + 64);
 110         char            buf[2 * (P_MAXDIG) + 20];
 111
 112         sprintf(output, "(1, %*d", P_MAXDIG, poly->npts);
 113
 114         for (i = 0; i < poly->npts; i++)
 115         {
 116                 sprintf(buf, ",%*g,%*g", P_MAXDIG, poly->p[i].x, P_MAXDIG, poly->p[i].y);
 117                 strcat(output, buf);
 118         }
 119
 120         sprintf(buf, "%c", RDELIM);
 121         strcat(output, buf);
 122         return path_in(output);
 123 }
 124
 125 /* return the point where two paths intersect.  Assumes that they do. */
 126 Point *
 127 interpt_pp(p1, p2)
 128 PATH       *p1;
 129 PATH       *p2;
 130 {
 131
 132         Point      *retval;
 133         int                     i,
 134                                 j;
 135         LSEG            seg1,
 136                                 seg2;
 137
 138 #ifdef NOT_USED
 139         LINE       *ln;
 140
 141 #endif
 142         bool            found;                  /* We've found the intersection */
 143
 144         found = false;                          /* Haven't found it yet */
 145
 146         for (i = 0; i < p1->npts - 1 && !found; i++)
 147                 for (j = 0; j < p2->npts - 1 && !found; j++)
 148                 {
 149                         regress_lseg_construct(&seg1, &p1->p[i], &p1->p[i + 1]);
 150                         regress_lseg_construct(&seg2, &p2->p[j], &p2->p[j + 1]);
 151                         if (lseg_intersect(&seg1, &seg2))
 152                                 found = true;
 153                 }
 154
 155 #ifdef NOT_USED
 156         ln = line_construct_pp(&seg2.p[0], &seg2.p[1]);
 157         retval = interpt_sl(&seg1, ln);
 158 #endif
 159         retval = lseg_interpt(&seg1, &seg2);
 160
 161         return retval;
 162 }
 163
 164
 165 /* like lseg_construct, but assume space already allocated */
 166 void
 167 regress_lseg_construct(lseg, pt1, pt2)
 168 LSEG       *lseg;
 169 Point      *pt1;
 170 Point      *pt2;
 171 {
 172         lseg->p[0].x = pt1->x;
 173         lseg->p[0].y = pt1->y;
 174         lseg->p[1].x = pt2->x;
 175         lseg->p[1].y = pt2->y;
 176         lseg->m = point_sl(pt1, pt2);
 177 }
 178
 179
 180 char
 181 overpaid(tuple)
 182 TUPLE           tuple;
 183 {
 184         bool            isnull;
 185         long            salary;
 186
 187         salary = (long) GetAttributeByName(tuple, "salary", &isnull);
 188         return salary > 699;
 189 }
 190
 191 /* New type "widget"
 192  * This used to be "circle", but I added circle to builtins,
 193  *      so needed to make sure the names do not collide. - tgl 97/04/21
 194  */
 195
 196 typedef struct
 197 {
 198         Point           center;
 199         double          radius;
 200 }                       WIDGET;
 201
 202 WIDGET     *widget_in(char *str);
 203 char       *widget_out(WIDGET * widget);
 204 int                     pt_in_widget(Point *point, WIDGET * widget);
 205
 206 #define NARGS   3
 207
 208 WIDGET *
 209 widget_in(str)
 210 char       *str;
 211 {
 212         char       *p,
 213                            *coord[NARGS],
 214                                 buf2[1000];
 215         int                     i;
 216         WIDGET     *result;
 217
 218         if (str == NULL)
 219                 return NULL;
 220         for (i = 0, p = str; *p && i < NARGS && *p != RDELIM; p++)
 221                 if (*p == ',' || (*p == LDELIM && !i))
 222                         coord[i++] = p + 1;
 223         if (i < NARGS - 1)
 224                 return NULL;
 225         result = (WIDGET *) palloc(sizeof(WIDGET));
 226         result->center.x = atof(coord[0]);
 227         result->center.y = atof(coord[1]);
 228         result->radius = atof(coord[2]);
 229
 230         sprintf(buf2, "widget_in: read (%f, %f, %f)\n", result->center.x,
 231                         result->center.y, result->radius);
 232         return result;
 233 }
 234
 235 char *
 236 widget_out(widget)
 237 WIDGET     *widget;
 238 {
 239         char       *result;
 240
 241         if (widget == NULL)
 242                 return NULL;
 243
 244         result = (char *) palloc(60);
 245         sprintf(result, "(%g,%g,%g)",
 246                         widget->center.x, widget->center.y, widget->radius);
 247         return result;
 248 }
 249
 250 int
 251 pt_in_widget(point, widget)
 252 Point      *point;
 253 WIDGET     *widget;
 254 {
 255         extern double point_dt();
 256
 257         return point_dt(point, &widget->center) < widget->radius;
 258 }
 259
 260 #define ABS(X) ((X) > 0 ? (X) : -(X))
 261
 262 int
 263 boxarea(box)
 264
 265 BOX                *box;
 266
 267 {
 268         int                     width,
 269                                 height;
 270
 271         width = ABS(box->high.x - box->low.x);
 272         height = ABS(box->high.y - box->low.y);
 273         return width * height;
 274 }
 275
 276 char *
 277 reverse_name(string)
 278 char       *string;
 279 {
 280         int                     i;
 281         int                     len;
 282         char       *new_string;
 283
 284         if (!(new_string = palloc(NAMEDATALEN)))
 285         {
 286                 fprintf(stderr, "reverse_name: palloc failed\n");
 287                 return NULL;
 288         }
 289         MemSet(new_string, 0, NAMEDATALEN);
 290         for (i = 0; i < NAMEDATALEN && string[i]; ++i)
 291                 ;
 292         if (i == NAMEDATALEN || !string[i])
 293                 --i;
 294         len = i;
 295         for (; i >= 0; --i)
 296                 new_string[len - i] = string[i];
 297         return new_string;
 298 }
 299
 300 #include "executor/spi.h"               /* this is what you need to work with SPI */
 301 #include "commands/trigger.h"   /* -"- and triggers */
 302
 303 static TransactionId fd17b_xid = InvalidTransactionId;
 304 static TransactionId fd17a_xid = InvalidTransactionId;
 305 static int      fd17b_level = 0;
 306 static int      fd17a_level = 0;
 307 static bool fd17b_recursion = true;
 308 static bool fd17a_recursion = true;
 309 HeapTuple       funny_dup17(void);
 310
 311 HeapTuple                                               /* have to return HeapTuple to Executor */
 312 funny_dup17()
 313 {
 314         TransactionId *xid;
 315         int                *level;
 316         bool       *recursion;
 317         Relation        rel;
 318         TupleDesc       tupdesc;
 319         HeapTuple       tuple;
 320         char            sql[MAX_QUERY_SIZE];
 321         char       *when;
 322         int                     inserted;
 323         int                     selected = 0;
 324         int                     ret;
 325
 326         tuple = CurrentTriggerData->tg_trigtuple;
 327         rel = CurrentTriggerData->tg_relation;
 328         tupdesc = rel->rd_att;
 329         if (TRIGGER_FIRED_BEFORE(CurrentTriggerData->tg_event))
 330         {
 331                 xid = &fd17b_xid;
 332                 level = &fd17b_level;
 333                 recursion = &fd17b_recursion;
 334                 when = "BEFORE";
 335         }
 336         else
 337         {
 338                 xid = &fd17a_xid;
 339                 level = &fd17a_level;
 340                 recursion = &fd17a_recursion;
 341                 when = "AFTER ";
 342         }
 343
 344         CurrentTriggerData = NULL;
 345
 346         if (!TransactionIdIsCurrentTransactionId(*xid))
 347         {
 348                 *xid = GetCurrentTransactionId();
 349                 *level = 0;
 350                 *recursion = true;
 351         }
 352
 353         if (*level == 17)
 354         {
 355                 *recursion = false;
 356                 return tuple;
 357         }
 358
 359         if (!(*recursion))
 360                 return tuple;
 361
 362         (*level)++;
 363
 364         SPI_connect();
 365
 366         sprintf(sql, "insert into %s select * from %s where %s = '%s'::%s",
 367                         SPI_getrelname(rel), SPI_getrelname(rel),
 368                         SPI_fname(tupdesc, 1),
 369                         SPI_getvalue(tuple, tupdesc, 1),
 370                         SPI_gettype(tupdesc, 1));
 371
 372         if ((ret = SPI_exec(sql, 0)) < 0)
 373                 elog(ERROR, "funny_dup17 (fired %s) on level %3d: SPI_exec (insert ...) returned %d",
 374                          when, *level, ret);
 375
 376         inserted = SPI_processed;
 377
 378         sprintf(sql, "select count (*) from %s where %s = '%s'::%s",
 379                         SPI_getrelname(rel),
 380                         SPI_fname(tupdesc, 1),
 381                         SPI_getvalue(tuple, tupdesc, 1),
 382                         SPI_gettype(tupdesc, 1));
 383
 384         if ((ret = SPI_exec(sql, 0)) < 0)
 385                 elog(ERROR, "funny_dup17 (fired %s) on level %3d: SPI_exec (select ...) returned %d",
 386                          when, *level, ret);
 387
 388         if (SPI_processed > 0)
 389         {
 390                 selected = int4in(
 391                                                   SPI_getvalue(
 392                                                                            SPI_tuptable->vals[0],
 393                                                                            SPI_tuptable->tupdesc,
 394                                                                            1
 395                                                                            )
 396                         );
 397         }
 398
 399         elog(NOTICE, "funny_dup17 (fired %s) on level %3d: %d/%d tuples inserted/selected",
 400                  when, *level, inserted, selected);
 401
 402         SPI_finish();
 403
 404         (*level)--;
 405
 406         if (*level == 0)
 407                 *xid = InvalidTransactionId;
 408
 409         return tuple;
 410 }
 411
 412 HeapTuple       ttdummy(void);
 413 int32           set_ttdummy(int32 on);
 414
 415 extern int4 nextval(struct varlena * seqin);
 416
 417 #define TTDUMMY_INFINITY        999999
 418
 419 static void *splan = NULL;
 420 static bool ttoff = false;
 421
 422 HeapTuple
 423 ttdummy()
 424 {
 425         Trigger    *trigger;            /* to get trigger name */
 426         char      **args;                       /* arguments */
 427         int                     attnum[2];              /* fnumbers of start/stop columns */
 428         Datum           oldon,
 429                                 oldoff;
 430         Datum           newon,
 431                                 newoff;
 432         Datum      *cvals;                      /* column values */
 433         char       *cnulls;                     /* column nulls */
 434         char       *relname;            /* triggered relation name */
 435         Relation        rel;                    /* triggered relation */
 436         HeapTuple       trigtuple;
 437         HeapTuple       newtuple = NULL;
 438         HeapTuple       rettuple;
 439         TupleDesc       tupdesc;                /* tuple description */
 440         int                     natts;                  /* # of attributes */
 441         bool            isnull;                 /* to know is some column NULL or not */
 442         int                     ret;
 443         int                     i;
 444
 445         if (!CurrentTriggerData)
 446                 elog(ERROR, "ttdummy: triggers are not initialized");
 447         if (TRIGGER_FIRED_FOR_STATEMENT(CurrentTriggerData->tg_event))
 448                 elog(ERROR, "ttdummy: can't process STATEMENT events");
 449         if (TRIGGER_FIRED_AFTER(CurrentTriggerData->tg_event))
 450                 elog(ERROR, "ttdummy: must be fired before event");
 451         if (TRIGGER_FIRED_BY_INSERT(CurrentTriggerData->tg_event))
 452                 elog(ERROR, "ttdummy: can't process INSERT event");
 453         if (TRIGGER_FIRED_BY_UPDATE(CurrentTriggerData->tg_event))
 454                 newtuple = CurrentTriggerData->tg_newtuple;
 455
 456         trigtuple = CurrentTriggerData->tg_trigtuple;
 457
 458         rel = CurrentTriggerData->tg_relation;
 459         relname = SPI_getrelname(rel);
 460
 461         /* check if TT is OFF for this relation */
 462         if (ttoff)                                      /* OFF - nothing to do */
 463         {
 464                 pfree(relname);
 465                 return (newtuple != NULL) ? newtuple : trigtuple;
 466         }
 467
 468         trigger = CurrentTriggerData->tg_trigger;
 469
 470         if (trigger->tgnargs != 2)
 471                 elog(ERROR, "ttdummy (%s): invalid (!= 2) number of arguments %d",
 472                          relname, trigger->tgnargs);
 473
 474         args = trigger->tgargs;
 475         tupdesc = rel->rd_att;
 476         natts = tupdesc->natts;
 477
 478         CurrentTriggerData = NULL;
 479
 480         for (i = 0; i < 2; i++)
 481         {
 482                 attnum[i] = SPI_fnumber(tupdesc, args[i]);
 483                 if (attnum[i] < 0)
 484                         elog(ERROR, "ttdummy (%s): there is no attribute %s", relname, args[i]);
 485                 if (SPI_gettypeid(tupdesc, attnum[i]) != INT4OID)
 486                         elog(ERROR, "ttdummy (%s): attributes %s and %s must be of abstime type",
 487                                  relname, args[0], args[1]);
 488         }
 489
 490         oldon = SPI_getbinval(trigtuple, tupdesc, attnum[0], &isnull);
 491         if (isnull)
 492                 elog(ERROR, "ttdummy (%s): %s must be NOT NULL", relname, args[0]);
 493
 494         oldoff = SPI_getbinval(trigtuple, tupdesc, attnum[1], &isnull);
 495         if (isnull)
 496                 elog(ERROR, "ttdummy (%s): %s must be NOT NULL", relname, args[1]);
 497
 498         if (newtuple != NULL)           /* UPDATE */
 499         {
 500                 newon = SPI_getbinval(newtuple, tupdesc, attnum[0], &isnull);
 501                 if (isnull)
 502                         elog(ERROR, "ttdummy (%s): %s must be NOT NULL", relname, args[0]);
 503                 newoff = SPI_getbinval(newtuple, tupdesc, attnum[1], &isnull);
 504                 if (isnull)
 505                         elog(ERROR, "ttdummy (%s): %s must be NOT NULL", relname, args[1]);
 506
 507                 if (oldon != newon || oldoff != newoff)
 508                         elog(ERROR, "ttdummy (%s): you can't change %s and/or %s columns (use set_ttdummy)",
 509                                  relname, args[0], args[1]);
 510
 511                 if (newoff != TTDUMMY_INFINITY)
 512                 {
 513                         pfree(relname);         /* allocated in upper executor context */
 514                         return NULL;
 515                 }
 516         }
 517         else if (oldoff != TTDUMMY_INFINITY)            /* DELETE */
 518         {
 519                 pfree(relname);
 520                 return NULL;
 521         }
 522
 523         {
 524                 struct varlena *seqname = textin("ttdummy_seq");
 525
 526                 newoff = nextval(seqname);
 527                 pfree(seqname);
 528         }
 529
 530         /* Connect to SPI manager */
 531         if ((ret = SPI_connect()) < 0)
 532                 elog(ERROR, "ttdummy (%s): SPI_connect returned %d", relname, ret);
 533
 534         /* Fetch tuple values and nulls */
 535         cvals = (Datum *) palloc(natts * sizeof(Datum));
 536         cnulls = (char *) palloc(natts * sizeof(char));
 537         for (i = 0; i < natts; i++)
 538         {
 539                 cvals[i] = SPI_getbinval((newtuple != NULL) ? newtuple : trigtuple,
 540                                                                  tupdesc, i + 1, &isnull);
 541                 cnulls[i] = (isnull) ? 'n' : ' ';
 542         }
 543
 544         /* change date column(s) */
 545         if (newtuple)                           /* UPDATE */
 546         {
 547                 cvals[attnum[0] - 1] = newoff;  /* start_date eq current date */
 548                 cnulls[attnum[0] - 1] = ' ';
 549                 cvals[attnum[1] - 1] = TTDUMMY_INFINITY;                /* stop_date eq INFINITY */
 550                 cnulls[attnum[1] - 1] = ' ';
 551         }
 552         else
 553 /* DELETE */
 554         {
 555                 cvals[attnum[1] - 1] = newoff;  /* stop_date eq current date */
 556                 cnulls[attnum[1] - 1] = ' ';
 557         }
 558
 559         /* if there is no plan ... */
 560         if (splan == NULL)
 561         {
 562                 void       *pplan;
 563                 Oid                *ctypes;
 564                 char            sql[MAX_QUERY_SIZE];
 565
 566                 /* allocate ctypes for preparation */
 567                 ctypes = (Oid *) palloc(natts * sizeof(Oid));
 568
 569                 /*
 570                  * Construct query: INSERT INTO _relation_ VALUES ($1, ...)
 571                  */
 572                 sprintf(sql, "INSERT INTO %s VALUES (", relname);
 573                 for (i = 1; i <= natts; i++)
 574                 {
 575                         sprintf(sql + strlen(sql), "$%d%s",
 576                                         i, (i < natts) ? ", " : ")");
 577                         ctypes[i - 1] = SPI_gettypeid(tupdesc, i);
 578                 }
 579
 580                 /* Prepare plan for query */
 581                 pplan = SPI_prepare(sql, natts, ctypes);
 582                 if (pplan == NULL)
 583                         elog(ERROR, "ttdummy (%s): SPI_prepare returned %d", relname, SPI_result);
 584
 585                 pplan = SPI_saveplan(pplan);
 586                 if (pplan == NULL)
 587                         elog(ERROR, "ttdummy (%s): SPI_saveplan returned %d", relname, SPI_result);
 588
 589                 splan = pplan;
 590         }
 591
 592         ret = SPI_execp(splan, cvals, cnulls, 0);
 593
 594         if (ret < 0)
 595                 elog(ERROR, "ttdummy (%s): SPI_execp returned %d", relname, ret);
 596
 597         /* Tuple to return to upper Executor ... */
 598         if (newtuple)                           /* UPDATE */
 599         {
 600                 HeapTuple       tmptuple;
 601
 602                 tmptuple = SPI_copytuple(trigtuple);
 603                 rettuple = SPI_modifytuple(rel, tmptuple, 1, &(attnum[1]), &newoff, NULL);
 604                 SPI_pfree(tmptuple);
 605         }
 606         else
 607 /* DELETE */
 608                 rettuple = trigtuple;
 609
 610         SPI_finish();                           /* don't forget say Bye to SPI mgr */
 611
 612         pfree(relname);
 613
 614         return rettuple;
 615 }
 616
 617 int32
 618 set_ttdummy(int32 on)
 619 {
 620
 621         if (ttoff)                                      /* OFF currently */
 622         {
 623                 if (on == 0)
 624                         return 0;
 625
 626                 /* turn ON */
 627                 ttoff = false;
 628                 return 0;
 629         }
 630
 631         /* ON currently */
 632         if (on != 0)
 633                 return 1;
 634
 635         /* turn OFF */
 636         ttoff = true;
 637
 638         return 1;
 639
 640 }