00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00045
#ifndef SelectColumnsVMatrix_INC
00046
#define SelectColumnsVMatrix_INC
00047
00048
#include "SourceVMatrix.h"
00049
00050
namespace PLearn {
00051
using namespace std;
00052
00056 class SelectColumnsVMatrix:
public SourceVMatrix
00057 {
00058
00059
private:
00060
00061 typedef SourceVMatrix inherited;
00062
00063
public:
00064
00066 TVec<int> indices;
00067 TVec<string> fields;
00068
00069
public:
00070
00072
SelectColumnsVMatrix();
00073
00076
SelectColumnsVMatrix(
VMat the_source,
TVec<string> the_fields);
00077
00080
SelectColumnsVMatrix(
VMat the_source,
TVec<int> the_indices);
00081
00083
SelectColumnsVMatrix(
VMat the_source,
Vec the_indices);
00084
00085
PLEARN_DECLARE_OBJECT(
SelectColumnsVMatrix);
00086
00087
virtual void build();
00088
virtual void makeDeepCopyFromShallowCopy(map<const void*, void*>& copies);
00089
00090
protected:
00091
00092
static void declareOptions(
OptionList &ol);
00093 void getNewRow(
int i,
const Vec& v)
const {
getSubRow(i, 0, v); }
00094
00095
public:
00096
00097
virtual real get(
int i,
int j)
const;
00098
virtual void getSubRow(
int i,
int j,
Vec v)
const;
00099 virtual void reset_dimensions()
00100 {
00101 source->reset_dimensions(); length_=source->
length();
00102
for (
int i=0;
indices.
length();i++)
00103
if (
indices[i]>=source->
width())
00104
PLERROR(
"SelectColumnsVMatrix::reset_dimensions, underlying source not wide enough (%d>=%d)",
00105
indices[i],source->
width());
00106 }
00107
00108
virtual const map<string,real>& getStringToRealMapping(
int col)
const;
00109
virtual const map<real,string>& getRealToStringMapping(
int col)
const;
00110
00111
virtual real getStringVal(
int col,
const string & str)
const;
00112
virtual string getValString(
int col,
real val)
const;
00113
00114
private:
00115
void build_();
00116
00117 };
00118
00119
DECLARE_OBJECT_PTR(SelectColumnsVMatrix);
00120
00121 }
00122
#endif