Added getters for distance and kernel at DR preprocessor

shogun-toolbox · Sep 26, 2011 · 140237b · 140237b
1 parent af3211a
commit 140237b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/src/shogun/preprocessor/DimensionReductionPreprocessor.cpp b/src/shogun/preprocessor/DimensionReductionPreprocessor.cpp
@@ -61,24 +61,34 @@ int32_t CDimensionReductionPreprocessor::get_target_dim() const
 
 void CDimensionReductionPreprocessor::set_distance(CDistance* distance)
 {
-	SG_UNREF(m_distance->parallel);
 	SG_UNREF(m_distance);
 	SG_REF(distance);
 	m_distance = distance;
 	m_distance->parallel = this->parallel;
 	SG_REF(this->parallel);
 }
 
+CDistance* CDimensionReductionPreprocessor::get_distance() const
+{
+	SG_REF(m_distance);
+	return m_distance;
+}
+
 void CDimensionReductionPreprocessor::set_kernel(CKernel* kernel)
 {
-	SG_UNREF(m_kernel->parallel);
 	SG_UNREF(m_kernel);
 	SG_REF(kernel);
 	m_kernel = kernel;
 	m_kernel->parallel = this->parallel;
 	SG_REF(this->parallel);
 }
 
+CKernel* CDimensionReductionPreprocessor::get_kernel() const
+{
+	SG_REF(m_kernel);
+	return m_kernel;
+}
+
 int32_t CDimensionReductionPreprocessor::detect_dim(SGMatrix<float64_t> distance_matrix)
 {
 	SG_NOTIMPLEMENTED;

diff --git a/src/shogun/preprocessor/DimensionReductionPreprocessor.h b/src/shogun/preprocessor/DimensionReductionPreprocessor.h
@@ -79,11 +79,21 @@ class CDimensionReductionPreprocessor: public CSimplePreprocessor<float64_t>
 	 */
 	void set_distance(CDistance* distance);
 
+	/** getter for distance
+	 * @return distance
+	 */
+	CDistance* get_distance() const;
+
 	/** setter for kernel
 	 * @param kernel kernel to set
 	 */
 	void set_kernel(CKernel* kernel);
 
+	/** getter for kernel
+	 * @return kernel
+	 */
+	CKernel* get_kernel() const;
+
 public:
 
 	/** const indicating target dimensionality should be determined automagically */