package org.rosuda.JRI;
// JRclient library - client interface to Rserve, see http://www.rosuda.org/Rserve/
// Copyright (C) 2004 Simon Urbanek
// --- for licensing information see LICENSE file in the original JRclient distribution ---
import java.util.*;
/** representation of a factor variable. In R there is no actual xpression
type called "factor", instead it is coded as an int vector with a list
attribute. The parser code of REXP converts such constructs directly into
the RFactor objects and defines an own XT_FACTOR type
@version $Id: RFactor.java 2909 2008-07-15 15:06:49Z urbanek $
*/
public class RFactor extends Object {
/** IDs (content: Integer) each entry corresponds to a case, ID specifies the category */
Vector id;
/** values (content: String), ergo category names */
Vector val;
/** create a new, empty factor var */
public RFactor() { id=new Vector(); val=new Vector(); }
/** create a new factor variable, based on the supplied arrays.
@param i array of IDs (0..v.length-1)
@param v values - category names */
public RFactor(int[] i, String[] v) {
this(i, v, 0);
}
/** create a new factor variable, based on the supplied arrays.
@param i array of IDs (base .. v.length-1+base)
@param v values - cotegory names
@param base of the indexing
*/
RFactor(int[] i, String[] v, int base) {
id=new Vector(); val=new Vector();
int j;
if (i!=null && i.length>0)
for(j=0;j<i.length;j++)
id.addElement(new Integer(i[j]-base));
if (v!=null && v.length>0)
for(j=0;j<v.length;j++)
val.addElement(v[j]);
}
/** add a new element (by name)
@param v value */
public void add(String v) {
int i=val.indexOf(v);
if (i<0) {
i=val.size();
val.addElement(v);
};
id.addElement(new Integer(i));
}
/** returns name for a specific ID
@param i ID
@return name. may throw exception if out of range */
public String at(int i) {
if (i < 0 || i >= id.size()) return null;
int j = ((Integer)id.elementAt(i)).intValue();
/* due to the index shift NA (INT_MIN) will turn into INT_MAX if base is 1 */
return (j < 0 || j > 2147483640) ? null : ((String)val.elementAt(j));
}
/** returns the number of caes */
public int size() { return id.size(); }
/** displayable representation of the factor variable */
public String toString() {
//return "{"+((val==null)?"<null>;":("levels="+val.size()+";"))+((id==null)?"<null>":("cases="+id.size()))+"}";
StringBuffer sb=new StringBuffer("{levels=(");
if (val==null)
sb.append("null");
else
for (int i=0;i<val.size();i++) {
sb.append((i>0)?",\"":"\"");
sb.append((String)val.elementAt(i));
sb.append("\"");
};
sb.append("),ids=(");
if (id==null)
sb.append("null");
else
for (int i=0;i<id.size();i++) {
if (i>0) sb.append(",");
sb.append((Integer)id.elementAt(i));
};
sb.append(")}");
return sb.toString();
}
}