From ecca282f1fb1881b5d42ed329f227036f0937819 Mon Sep 17 00:00:00 2001
From: xuewei cao <36172337+xueweic@users.noreply.github.com>
Date: Wed, 26 Mar 2025 11:20:39 -0400
Subject: [PATCH] add colocboost object

---
 R/colocboost.R          | 1 +
 R/colocboost_assemble.R | 4 ++--
 R/colocboost_plot.R     | 3 +++
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/R/colocboost.R b/R/colocboost.R
index 19904ae..ed9e022 100644
--- a/R/colocboost.R
+++ b/R/colocboost.R
@@ -582,6 +582,7 @@ colocboost <- function(X = NULL, Y = NULL, # individual data
                                      merging =  merging,
                                      tol = tol,
                                      output_level = output_level)
+    class(cb_output) <- "colocboost"
     
     return(cb_output)
 }
diff --git a/R/colocboost_assemble.R b/R/colocboost_assemble.R
index 4677859..fb3e79f 100644
--- a/R/colocboost_assemble.R
+++ b/R/colocboost_assemble.R
@@ -40,7 +40,7 @@ colocboost_assemble <- function(cb_obj,
                                 output_level = 1){
 
     if (class(cb_obj) != "colocboost"){
-        stop("Input must from colocboost function!")}
+        stop("Input must from colocboost object!")}
   
     # - data information
     data_info <- get_data_info(cb_obj)
@@ -232,7 +232,7 @@ colocboost_assemble <- function(cb_obj,
           }
         }
     }
-
+    
     return(cb_output)
 
 }
diff --git a/R/colocboost_plot.R b/R/colocboost_plot.R
index 1afe502..736a5fb 100644
--- a/R/colocboost_plot.R
+++ b/R/colocboost_plot.R
@@ -28,6 +28,9 @@ colocboost_plot <- function(cb_output, y = "log10p",
                             ...){
                     
   
+    if (class(cb_output) != "colocboost"){
+        stop("Input of colocboost_plot must be a 'colocboost' object!")}
+  
     # get cb_plot_input data from colocboost results
     cb_plot_input <- get_input_plot(cb_output, plot_cos_idx = plot_cos_idx, 
                                     plot_target_only = plot_target_only,