sharedochiai.cpp

   1 /*
   2  *  sharedochiai.cpp
   3  *  Mothur
   4  *
   5  *  Created by Sarah Westcott on 3/23/09.
   6  *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
   7  *
   8  */
   9
  10 #include "sharedochiai.h"
  11
  12 /***********************************************************************/
  13
  14 EstOutput Ochiai::getValues(vector<SharedRAbundVector*> shared) {
  15         try {
  16                 double S1, S2, S12, tempA, tempB;
  17                 S1 = 0; S2 = 0; S12 = 0; tempA = 0; tempB = 0;
  18
  19                 /*S1, S2 = number of OTUs observed or estimated in A and B
  20                 S12=number of OTUs shared between A and B */
  21
  22                 data.resize(1,0);
  23
  24                 for (int i = 0; i < shared[0]->getNumBins(); i++) {
  25                         //store in temps to avoid multiple repetitive function calls
  26                         tempA = shared[0]->getAbundance(i);
  27                         tempB = shared[1]->getAbundance(i);
  28
  29                         if (tempA != 0) { S1++; }
  30                         if (tempB != 0) { S2++; }
  31
  32                         //they are shared
  33                         if ((tempA != 0) && (tempB != 0)) {     S12++; }
  34                 }
  35
  36                 data[0] = S12 / ((float)pow((S1 * S2), 0.5));
  37
  38                 if (isnan(data[0]) || isinf(data[0])) { data[0] = 0; }
  39
  40                 return data;
  41         }
  42         catch(exception& e) {
  43                 m->errorOut(e, "Ochiai", "getValues");
  44                 exit(1);
  45         }
  46 }
  47
  48 /***********************************************************************/