Exemplo n.º 1
0
  /**
   * Read in a list of ExactCall objects from reader, keeping only those with starts in startsToKeep
   * or all sites (if this is empty)
   *
   * @param reader a just-opened reader sitting at the start of the file
   * @param startsToKeep a list of start position of the calls to keep, or empty if all calls should
   *     be kept
   * @param parser a genome loc parser to create genome locs
   * @return a list of ExactCall objects in reader
   * @throws IOException
   */
  public static List<ExactCall> readExactLog(
      final BufferedReader reader, final List<Integer> startsToKeep, GenomeLocParser parser)
      throws IOException {
    if (reader == null) throw new IllegalArgumentException("reader cannot be null");
    if (startsToKeep == null) throw new IllegalArgumentException("startsToKeep cannot be null");
    if (parser == null) throw new IllegalArgumentException("GenomeLocParser cannot be null");

    List<ExactCall> calls = new LinkedList<ExactCall>();

    // skip the header line
    reader.readLine();

    // skip the first "type" line
    reader.readLine();

    while (true) {
      final VariantContextBuilder builder = new VariantContextBuilder();
      final List<Allele> alleles = new ArrayList<Allele>();
      final List<Genotype> genotypes = new ArrayList<Genotype>();
      final double[] posteriors = new double[2];
      final double[] priors = MathUtils.normalizeFromLog10(new double[] {0.5, 0.5}, true);
      final List<Integer> mle = new ArrayList<Integer>();
      final Map<Allele, Double> log10pNonRefByAllele = new HashMap<Allele, Double>();
      long runtimeNano = -1;

      GenomeLoc currentLoc = null;
      while (true) {
        final String line = reader.readLine();
        if (line == null) return calls;

        final String[] parts = line.split("\t");
        final GenomeLoc lineLoc = parser.parseGenomeLoc(parts[0]);
        final String variable = parts[1];
        final String key = parts[2];
        final String value = parts[3];

        if (currentLoc == null) currentLoc = lineLoc;

        if (variable.equals("type")) {
          if (startsToKeep.isEmpty() || startsToKeep.contains(currentLoc.getStart())) {
            builder.alleles(alleles);
            final int stop = currentLoc.getStart() + alleles.get(0).length() - 1;
            builder.chr(currentLoc.getContig()).start(currentLoc.getStart()).stop(stop);
            builder.genotypes(genotypes);
            final int[] mleInts = ArrayUtils.toPrimitive(mle.toArray(new Integer[] {}));
            final AFCalcResult result =
                new AFCalcResult(mleInts, 1, alleles, posteriors, priors, log10pNonRefByAllele);
            calls.add(new ExactCall(builder.make(), runtimeNano, result));
          }
          break;
        } else if (variable.equals("allele")) {
          final boolean isRef = key.equals("0");
          alleles.add(Allele.create(value, isRef));
        } else if (variable.equals("PL")) {
          final GenotypeBuilder gb = new GenotypeBuilder(key);
          gb.PL(GenotypeLikelihoods.fromPLField(value).getAsPLs());
          genotypes.add(gb.make());
        } else if (variable.equals("log10PosteriorOfAFEq0")) {
          posteriors[0] = Double.valueOf(value);
        } else if (variable.equals("log10PosteriorOfAFGt0")) {
          posteriors[1] = Double.valueOf(value);
        } else if (variable.equals("MLE")) {
          mle.add(Integer.valueOf(value));
        } else if (variable.equals("pNonRefByAllele")) {
          final Allele a = Allele.create(key);
          log10pNonRefByAllele.put(a, Double.valueOf(value));
        } else if (variable.equals("runtime.nano")) {
          runtimeNano = Long.valueOf(value);
        } else {
          // nothing to do
        }
      }
    }
  }
Exemplo n.º 2
0
 private final double getRefBinomialProb(final int depth) {
   if (depth < binomialProbabilityDepthCache.length) return binomialProbabilityDepthCache[depth];
   else return MathUtils.binomialProbability(0, depth, 0.5);
 }
  /**
   * query='dbConn.((0 as i close by (i + 100 where i <= max(Trener.getPremia())) as i join
   * count((Trener where getPremia() >= i and getPremia() < i + 100)) as c).(c + " Trenerow uzyskuje
   * premie pomiedzy " + i + " i " + i + 100) as message)' '
   */
  public java.util.List<java.lang.String> executeQuery(
      final ObjectContainerBase ocb, final Transaction t) {
    final LocalTransaction transLocal = (LocalTransaction) t;
    java.lang.Integer _asResult_i = 0;
    java.util.List<java.lang.Integer> _closeByResult = new ArrayList<java.lang.Integer>();
    _closeByResult.add(_asResult_i);

    int _i3 = 0;

    while (_i3 < _closeByResult.size()) {
      java.lang.Integer _closeByEl = _closeByResult.get(_i3);

      if (_closeByEl != null) {
        ocb.activate(_closeByEl, 1);
      }

      java.lang.Integer _ident_i = _closeByEl;
      java.lang.Integer _plusResult = _ident_i + 100;

      if (_plusResult != null) {
        ocb.activate(_plusResult, 2);
      }

      java.lang.Integer _whereEl = _plusResult;
      java.lang.Integer _ident_i1 = _closeByEl;
      final java.util.Collection<pl.wcislo.sbql4j.javac.test.football.model.Trener> _ident_Trener =
          new java.util.ArrayList<pl.wcislo.sbql4j.javac.test.football.model.Trener>();
      ClassMetadata _classMeta31 =
          ocb.classCollection()
              .getClassMetadata("pl.wcislo.sbql4j.javac.test.football.model.Trener");
      long[] _ids31 = _classMeta31.getIDs(transLocal);

      for (long _id31 : _ids31) {
        LazyObjectReference _ref31 = transLocal.lazyReferenceFor((int) _id31);
        _ident_Trener.add((pl.wcislo.sbql4j.javac.test.football.model.Trener) _ref31.getObject());
      }

      java.util.Collection<java.lang.Integer> _dotResult =
          new java.util.ArrayList<java.lang.Integer>();
      int _dotIndex = 0;

      if (_ident_Trener != null) {
        for (pl.wcislo.sbql4j.javac.test.football.model.Trener _dotEl : _ident_Trener) {
          if (_dotEl == null) {
            continue;
          }

          if (_dotEl != null) {
            ocb.activate(_dotEl, 1);
          }

          java.lang.Integer _mth_getPremiaResult = _dotEl.getPremia();

          if (_mth_getPremiaResult != null) {
            ocb.activate(_mth_getPremiaResult, 1);
          }

          if (_mth_getPremiaResult != null) {
            ocb.activate(_mth_getPremiaResult, 1);
          }

          _dotResult.add(_mth_getPremiaResult);
          _dotIndex++;
        }
      }

      Number _max7 = null;

      for (Number _maxEl7 : _dotResult) {
        _max7 = MathUtils.max(_max7, _maxEl7);
      }

      java.lang.Integer _maxResult = (java.lang.Integer) _max7;

      Boolean _less_or_equalResult = _ident_i1 <= _maxResult;
      java.lang.Integer _whereResult = null;

      if (_less_or_equalResult) {
        _whereResult = _plusResult;
      }

      java.lang.Integer _asResult_i1 = _whereResult;

      if (_asResult_i1 != null) {
        _closeByResult.add(_asResult_i1);
      }

      _i3++;
    }

    java.util.List<pl.wcislo.sbql4j.java.model.runtime.Struct> _joinResult =
        new java.util.ArrayList<pl.wcislo.sbql4j.java.model.runtime.Struct>();
    int _joinIndex = 0;

    for (java.lang.Integer _joinEl : _closeByResult) {
      if (_joinEl != null) {
        ocb.activate(_joinEl, 1);
      }

      final java.util.Collection<pl.wcislo.sbql4j.javac.test.football.model.Trener> _ident_Trener1 =
          new java.util.ArrayList<pl.wcislo.sbql4j.javac.test.football.model.Trener>();
      ClassMetadata _classMeta32 =
          ocb.classCollection()
              .getClassMetadata("pl.wcislo.sbql4j.javac.test.football.model.Trener");
      long[] _ids32 = _classMeta32.getIDs(transLocal);

      for (long _id32 : _ids32) {
        LazyObjectReference _ref32 = transLocal.lazyReferenceFor((int) _id32);
        _ident_Trener1.add((pl.wcislo.sbql4j.javac.test.football.model.Trener) _ref32.getObject());
      }

      java.util.Collection<pl.wcislo.sbql4j.javac.test.football.model.Trener> _whereResult1 =
          new java.util.ArrayList<pl.wcislo.sbql4j.javac.test.football.model.Trener>();
      int _whereLoopIndex1 = 0;

      for (pl.wcislo.sbql4j.javac.test.football.model.Trener _whereEl1 : _ident_Trener1) {
        if (_whereEl1 == null) {
          continue;
        }

        if (_whereEl1 != null) {
          ocb.activate(_whereEl1, 1);
        }

        java.lang.Integer _mth_getPremiaResult1 = _whereEl1.getPremia();

        if (_mth_getPremiaResult1 != null) {
          ocb.activate(_mth_getPremiaResult1, 1);
        }

        java.lang.Integer _ident_i2 = _joinEl;

        if (_ident_i2 != null) {
          ocb.activate(_ident_i2, 1);
        }

        Boolean _more_or_equalResult = _mth_getPremiaResult1 >= _ident_i2;
        java.lang.Boolean _andResult;

        if (!_more_or_equalResult) {
          _andResult = false;
        } else {
          java.lang.Integer _mth_getPremiaResult2 = _whereEl1.getPremia();

          if (_mth_getPremiaResult2 != null) {
            ocb.activate(_mth_getPremiaResult2, 1);
          }

          java.lang.Integer _ident_i3 = _joinEl;

          if (_ident_i3 != null) {
            ocb.activate(_ident_i3, 1);
          }

          java.lang.Integer _plusResult1 = _ident_i3 + 100;

          Boolean _lessResult = _mth_getPremiaResult2 < _plusResult1;
          _andResult = _lessResult;
        }

        if (_andResult) {
          _whereResult1.add(_whereEl1);
        }

        _whereLoopIndex1++;
      }

      java.lang.Integer _countResult = _whereResult1.size();
      java.lang.Integer _asResult_c = _countResult;
      _joinResult.add(OperatorUtils.cartesianProductSS(_joinEl, _asResult_c, "i", "c"));
      _joinIndex++;
    }

    java.util.List<java.lang.String> _dotResult1 = new java.util.ArrayList<java.lang.String>();
    int _dotIndex1 = 0;

    if (_joinResult != null) {
      for (pl.wcislo.sbql4j.java.model.runtime.Struct _dotEl1 : _joinResult) {
        if (_dotEl1 == null) {
          continue;
        }

        if (_dotEl1 != null) {
          ocb.activate(_dotEl1, 1);
        }

        java.lang.Integer _ident_c = (java.lang.Integer) _dotEl1.get("c");

        if (_ident_c != null) {
          ocb.activate(_ident_c, 1);
        }

        java.lang.String _plusResult2 = _ident_c + " Trenerow uzyskuje premie pomiedzy  ";
        java.lang.Integer _ident_i4 = (java.lang.Integer) _dotEl1.get("i");

        if (_ident_i4 != null) {
          ocb.activate(_ident_i4, 1);
        }

        java.lang.String _plusResult3 = _plusResult2 + _ident_i4;
        java.lang.String _plusResult4 = _plusResult3 + " i ";
        java.lang.Integer _ident_i5 = (java.lang.Integer) _dotEl1.get("i");

        if (_ident_i5 != null) {
          ocb.activate(_ident_i5, 1);
        }

        java.lang.Integer _plusResult5 = _ident_i5 + 100;
        java.lang.String _plusResult6 = _plusResult4 + _plusResult5;

        if (_plusResult6 != null) {
          ocb.activate(_plusResult6, 1);
        }

        _dotResult1.add(_plusResult6);
        _dotIndex1++;
      }
    }

    java.util.List<java.lang.String> _asResult_message = _dotResult1;
    pl.wcislo.sbql4j.db4o.utils.DerefUtils.activateResult(_asResult_message, ocb);

    return _asResult_message;
  }
Exemplo n.º 4
0
 static {
   for (int i = 1; i < binomialProbabilityDepthCache.length; i++) {
     binomialProbabilityDepthCache[i] = MathUtils.binomialProbability(0, i, 0.5);
   }
 }