com.ibm.wala.util.tables.Query Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of com.ibm.wala.util Show documentation
T. J. Watson Libraries for Analysis
The newest version!
/*
 * Copyright (c) 2006 IBM Corporation.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 */
package com.ibm.wala.util.tables;

import com.ibm.wala.util.collections.HashSetFactory;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Map;
import java.util.function.Predicate;
import org.jspecify.annotations.NullUnmarked;

/** Misc SQL-like support for queries on tables */
public class Query {

  /**
   * SELECT * from t where column=value
   *
   * @throws IllegalArgumentException if t == null
   */
  @NullUnmarked
  public static  Collection> selectStarWhereEquals(
      Table t, String column, T value) throws IllegalArgumentException {
    if (t == null) {
      throw new IllegalArgumentException("t == null");
    }
    Collection> result = new ArrayList<>();
    for (int i = 0; i < t.getNumberOfRows(); i++) {
      Map p = t.row2Map(i);
      if (p.get(column).equals(value)) {
        result.add(p);
      }
    }
    return result;
  }

  /** SELECT attribute FROM t where column=value */
  public static  Collection selectWhereEquals(
      Table t, String attribute, String column, T value) {
    Collection> rows = selectStarWhereEquals(t, column, value);
    Collection result = HashSetFactory.make();
    for (Map p : rows) {
      result.add(p.get(attribute));
    }
    return result;
  }

  /** SELECT attribute FROM t where P(column) */
  public static  Collection> selectStarWhere(
      Table t, String column, Predicate P) {
    if (t == null) {
      throw new IllegalArgumentException("t == null");
    }
    Collection> c = new ArrayList<>();
    for (int i = 0; i < t.getNumberOfRows(); i++) {
      Map p = t.row2Map(i);
      T s = p.get(column);
      if (P.test(s)) {
        c.add(p);
      }
    }
    return c;
  }

  public static  Table viewWhereEquals(Table t, String column, T value) {
    Collection> c = selectStarWhereEquals(t, column, value);
    Table result = new Table<>(t);
    for (Map p : c) {
      result.addRow(p);
    }
    return result;
  }

  public static StringTable viewWhereEquals(StringTable t, String column, String value) {
    Collection> c = selectStarWhereEquals(t, column, value);
    StringTable result = new StringTable(t);
    for (Map p : c) {
      result.addRow(p);
    }
    return result;
  }
}