edu.stanford.nlp.tagger.maxent.CountWrapper Maven / Gradle / Ivy
Show all versions of stanford-parser Show documentation
/**
* Title: StanfordMaxEnt
* Description: A Maximum Entropy Toolkit
* Copyright: Copyright (c) Kristina Toutanova
* Company: Stanford University
*/
package edu.stanford.nlp.tagger.maxent;
import edu.stanford.nlp.util.logging.Redwood;
import java.io.IOException;
import java.io.DataInputStream;
import java.io.DataOutputStream;
/** A simple data structure for some tag counts.
* @author Kristina Toutanova
* @version 1.0
*/
public class CountWrapper {
/** A logger for this class */
private static Redwood.RedwoodChannels log = Redwood.channels(CountWrapper.class);
private String word;
private int countPart;
private int countThat;
private int countIn;
private int countRB;
//private Dictionary dictLocal = new Dictionary();
//private static final String rpTag = "RP";
//private static final String inTag = "IN";
//private static final String rbTag = "RB";
public CountWrapper() {
}
protected CountWrapper(String word, int countPart, int countThat, int countIn, int countRB) {
assert(word != null);
this.word = word;
this.countPart = countPart;
this.countThat = countThat;
this.countIn = countIn;
this.countRB = countRB;
}
protected void incThat() {
this.countThat++;
}
public int getCountPart() {
return countPart;
}
public int getCountThat() {
return countThat;
}
public int getCountIn() {
return countIn;
}
public int getCountRB() {
return countRB;
}
public String getWord() {
return word;
}
@Override
public int hashCode() {
return word.hashCode();
}
/** Equality is tested only on the word, and not the various counts
* that are maintained.
*
* @param obj Item tested for equality
* @return Whether equal
*/
@Override
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if ( ! (obj instanceof CountWrapper)) {
return false;
}
CountWrapper cw = (CountWrapper) obj;
return word.equals(cw.word);
}
protected void save(DataOutputStream rf) {
try {
rf.writeInt(word.length());
rf.write(word.getBytes());
rf.writeInt(countPart);
rf.writeInt(countThat);
rf.writeInt(countIn);
rf.writeInt(countRB);
} catch (Exception e) {
e.printStackTrace();
}
}
protected void read(DataInputStream rf) {
try {
int len = rf.readInt();
byte[] buff = new byte[len];
if (rf.read(buff) != len) { log.info("Error: rewrite CountWrapper.read"); }
word = new String(buff);
assert(word != null);
countPart = rf.readInt();
countThat = rf.readInt();
countIn = rf.readInt();
countRB = rf.readInt();
} catch (IOException e) {
e.printStackTrace();
}
}
}