/**
* Copyright (c) 2011, Regents of the University of Colorado All rights
* reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* Redistributions of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer. Redistributions in binary
* form must reproduce the above copyright notice, this list of conditions and
* the following disclaimer in the documentation and/or other materials provided
* with the distribution. Neither the name of the University of Colorado at
* Boulder nor the names of its contributors may be used to endorse or promote
* products derived from this software without specific prior written
* permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/
package clear.dep.srl;
import clear.dep.DepLib;
import clear.ftr.FtrLib;
/**
* Semantic role labeling head.
*
* @author Jinho D. Choi <b>Last update:</b> 4/19/2011
*/
public class SRLHead implements Comparable<SRLHead> {
static final public String DELIM = ":";
public int headId;
public String label;
public double score;
public SRLHead() {
set(DepLib.NULL_HEAD_ID, FtrLib.TAG_NULL);
}
/**
* @param headInfo "headId:label[:score]"
*/
public SRLHead(String headInfo) {
String[] tmp = headInfo.split(DELIM);
set(Integer.parseInt(tmp[0]), tmp[1]);
if (tmp.length > 2) {
score = Double.parseDouble(tmp[2]);
} else {
score = 1;
}
}
public SRLHead(int headId, String label) {
set(headId, label);
}
public SRLHead(int headId, String label, double score) {
set(headId, label, score);
}
public void set(int headId, String label) {
this.headId = headId;
this.label = label;
}
public void set(int headId, String label, double score) {
this.headId = headId;
this.label = label;
this.score = score;
}
public String stripCLabel(String label) {
if (label.startsWith("C-")) {
return label.substring(2);
}
return label;
}
public boolean equals(SRLHead head) {
return equals(head.headId, head.label);
}
public boolean equals(int headId) {
return this.headId == headId;
}
public boolean equals(String label) {
return stripCLabel(this.label).equals(stripCLabel(label));
}
public boolean equals(int headId, String label) {
return equals(headId) && equals(label);
}
public boolean labelMatches(String regex) {
return label.matches(regex);
}
public void copy(SRLHead head) {
set(head.headId, head.label, head.score);
}
@Override
public SRLHead clone() {
return new SRLHead(headId, label, score);
}
@Override
public String toString() {
StringBuilder build = new StringBuilder();
build.append(headId);
build.append(DELIM);
build.append(label);
// build.append(DELIM);
// build.append(score);
return build.toString();
}
@Override
public int compareTo(SRLHead head) {
float diff = headId - head.headId;
if (diff < 0) {
return -1;
} else if (diff > 0) {
return 1;
}
return 0;
}
}