@RBuiltin(
name = "dimnames",
kind = PRIMITIVE,
parameterNames = {"x"},
dispatch = INTERNAL_GENERIC)
public abstract class DimNames extends RBuiltinNode {
private final RAttributeProfiles attrProfiles = RAttributeProfiles.create();
private final ConditionProfile nullProfile = ConditionProfile.createBinaryProfile();
private final BranchProfile dataframeProfile = BranchProfile.create();
private final BranchProfile factorProfile = BranchProfile.create();
private final BranchProfile otherProfile = BranchProfile.create();
@Specialization
protected RNull getDimNames(@SuppressWarnings("unused") RNull operand) {
controlVisibility();
return RNull.instance;
}
@Specialization
protected Object getDimNames(RAbstractContainer container) {
controlVisibility();
RList names;
if (container instanceof RDataFrame) {
dataframeProfile.enter();
names = ((RDataFrame) container).getVector().getDimNames();
} else if (container instanceof RFactor) {
factorProfile.enter();
names = ((RFactor) container).getVector().getDimNames();
} else {
otherProfile.enter();
names = container.getDimNames(attrProfiles);
}
return nullProfile.profile(names == null) ? RNull.instance : names;
}
}
/**
* Simple attribute access node that specializes on the position at which the attribute was found
* last time.
*/
public abstract class UnaryCopyAttributesNode extends RBaseNode {
protected final boolean copyAllAttributes;
protected final RAttributeProfiles attrSourceProfiles = RAttributeProfiles.create();
protected UnaryCopyAttributesNode(boolean copyAllAttributes) {
this.copyAllAttributes = copyAllAttributes;
}
public abstract RAbstractVector execute(RAbstractVector target, RAbstractVector left);
protected boolean containsMetadata(RAbstractVector vector, RAttributeProfiles attrProfiles) {
return vector instanceof RVector && vector.hasDimensions()
|| (copyAllAttributes && vector.getAttributes() != null)
|| vector.getNames(attrProfiles) != null
|| vector.getDimNames(attrProfiles) != null;
}
@SuppressWarnings("unused")
@Specialization(guards = "!containsMetadata(source, attrSourceProfiles)")
public RAbstractVector copyNoMetadata(RAbstractVector target, RAbstractVector source) {
return target;
}
@SuppressWarnings("unused")
@Specialization(guards = {"copyAllAttributes", "target == source"})
public RAbstractVector copySameVector(RAbstractVector target, RAbstractVector source) {
return target;
}
@Specialization(
guards = {
"!copyAllAttributes || target != source",
"containsMetadata(source, attrSourceProfiles)"
})
public RAbstractVector copySameLength(
RAbstractVector target,
RAbstractVector source, //
@Cached("create()") CopyOfRegAttributesNode copyOfReg, //
@Cached("createDim()") RemoveAttributeNode removeDim, //
@Cached("createDimNames()") RemoveAttributeNode removeDimNames, //
@Cached("create()") InitAttributesNode initAttributes, //
@Cached("createNames()") PutAttributeNode putNames, //
@Cached("createDim()") PutAttributeNode putDim, //
@Cached("createBinaryProfile()") ConditionProfile noDimensions, //
@Cached("createBinaryProfile()") ConditionProfile hasNamesSource, //
@Cached("createBinaryProfile()") ConditionProfile hasDimNames) {
RVector result = target.materialize();
if (copyAllAttributes) {
copyOfReg.execute(source, result);
}
int[] newDimensions = source.getDimensions();
if (noDimensions.profile(newDimensions == null)) {
RAttributes attributes = result.getAttributes();
if (attributes != null) {
removeDim.execute(attributes);
removeDimNames.execute(attributes);
result.setInternalDimNames(null);
}
result.setInternalDimensions(null);
RStringVector vecNames = source.getNames(attrSourceProfiles);
if (hasNamesSource.profile(vecNames != null)) {
putNames.execute(initAttributes.execute(result), vecNames);
result.setInternalNames(vecNames);
return result;
}
return result;
}
putDim.execute(
initAttributes.execute(result),
RDataFactory.createIntVector(newDimensions, RDataFactory.COMPLETE_VECTOR));
result.setInternalDimensions(newDimensions);
RList newDimNames = source.getDimNames(attrSourceProfiles);
if (hasDimNames.profile(newDimNames != null)) {
result.getAttributes().put(RRuntime.DIMNAMES_ATTR_KEY, newDimNames);
newDimNames.elementNamePrefix = RRuntime.DIMNAMES_LIST_ELEMENT_NAME_PREFIX;
result.setInternalDimNames(newDimNames);
return result;
}
return result;
}
}
