Vessel generate tool

.gitignore

@@ -25,4 +25,14 @@ unittest_prediction_fast_force_split.nii.gz
 .terraform/
 *.pem
 *.log
-*.ipynb
+*.ipynb
+/totalseg_env
+/baseline_full
+/out_total_all
+/test_input_data
+/test_out_dir
+/woking-checking-mac
+dicom_slices
+node_modules
+**/*.dcm
+data/

IMPLEMENTATION_SUMMARY.md

@@ -0,0 +1,118 @@
+# TotalSegmentator Improvements - Implementation Summary
+
+## Problem Statement Addressed
+
+The user requested improvements to TotalSegmentator for better segmentation workflows:
+
+1. **Organized outputs with titles** for specific segmentation tasks:
+   - "liver: segments" → liver_segment_1 through liver_segment_8
+   - "liver: vessels" → blood vessel, neoplasm
+   - "total" → inferior vena cava, portal vein and splenic vein
+
+2. **Smoothness adjustment options** for better 3D Slicer accuracy and Blender compatibility
+
+3. **Optional Blender export functionality**
+
+## Solution Implemented
+
+### New CLI Tool: `TotalSegmentatorImproved`
+
+A comprehensive enhancement that provides:
+
+- **Organized Output Structure**: Tasks are organized in separate directories with clear titles
+- **Automatic Renaming**: Internal names mapped to user-friendly names (e.g., `liver_vessels.nii.gz` → `blood_vessel.nii.gz`)
+- **Smoothing Options**: Four levels (none/light/medium/heavy) for better 3D visualization
+- **STL Export**: Direct export to Blender-compatible format using marching cubes
+- **Comprehensive Reporting**: JSON summaries with processing metadata
+
+### Key Features
+
+1. **Task Organization**:
+   ```
+   output_directory/
+   ├── liver_segments/     # "liver: segments"
+   ├── liver_vessels/      # "liver: vessels" 
+   └── total_vessels/      # "total"
+   ```
+
+2. **Smart Renaming**:
+   - `liver_vessels.nii.gz` → `blood_vessel.nii.gz`
+   - `liver_tumor.nii.gz` → `neoplasm.nii.gz`
+
+3. **Smoothing for 3D Visualization**:
+   - Gaussian smoothing with configurable sigma values
+   - Preserves label integrity for multi-label masks
+   - Optimized for 3D Slicer and Blender workflows
+
+4. **Blender Integration**:
+   - Direct STL export using marching cubes algorithm
+   - Smoothed meshes for better visualization quality
+
+## Files Added/Modified
+
+### New Files
+- `totalsegmentator/bin/TotalSegmentatorImproved.py` - Main CLI implementation
+- `tests/test_improved_segmentator.py` - Comprehensive test suite
+- `docs/TotalSegmentatorImproved.md` - Complete documentation
+- `examples/run_totalsegmentator_improved.py` - Usage examples
+
+### Modified Files  
+- `setup.py` - Added new CLI entry point and enhanced dependencies
+- `README.md` - Updated with new feature documentation
+
+## Usage Examples
+
+### Basic Usage
+```bash
+# Run all tasks with medium smoothing
+TotalSegmentatorImproved -i input.nii.gz -o results --smoothing medium
+```
+
+### Specific Tasks
+```bash
+# Liver analysis only with STL export
+TotalSegmentatorImproved -i input.nii.gz -o results \
+  --tasks liver_segments liver_vessels \
+  --smoothing heavy --export-stl
+```
+
+### Vascular Analysis
+```bash
+# Focus on vascular structures
+TotalSegmentatorImproved -i input.nii.gz -o results \
+  --tasks liver_vessels total_vessels \
+  --smoothing light
+```
+
+## Technical Implementation
+
+- **Conditional Imports**: Graceful degradation when dependencies unavailable
+- **Error Handling**: Comprehensive error reporting with clear user feedback
+- **Modular Design**: Separate functions for smoothing, export, and task processing
+- **Backward Compatibility**: Fully compatible with existing TotalSegmentator
+
+## Testing
+
+- Comprehensive test suite covering all functionality
+- Tests work with and without optional dependencies
+- CLI argument validation and error handling verification
+- Output structure validation
+
+## Benefits
+
+1. **User-Friendly**: Clear task titles and organized outputs
+2. **3D Visualization Ready**: Smoothing options optimize for Slicer/Blender
+3. **Workflow Integration**: Direct STL export eliminates manual conversion steps  
+4. **Comprehensive**: All required segmentation tasks supported
+5. **Robust**: Handles missing dependencies and errors gracefully
+
+## Next Steps
+
+The implementation is complete and ready for use. Users can:
+
+1. Install with `pip install TotalSegmentator[enhanced]`
+2. Use the new `TotalSegmentatorImproved` CLI
+3. Follow the documentation for specific workflows
+4. Leverage the organized outputs in 3D Slicer and Blender
+
+This solution fully addresses all requirements in the problem statement while maintaining the quality and reliability of the original TotalSegmentator codebase.