import numpy as np
import sf3dmodels.filament as sf
import sf3dmodels.Plot_model as pm
from sf3dmodels.utils.units import pc

f1 = sf.FilamentModel([0,0,0], [0,0,1], -0.2*pc, 0.2*pc, 0.01*pc) # Instantiate the FilamentModel class with geometrical parameters for the grid construction
f1.cylinder(0.1*pc, 1e-3*pc, temp_pars = [500, 0.02*pc, -0.3], abund_pars = 1e-4) # Specify the method and the physical parameters

lims=np.array([-0.3,0.3])*pc

pm.scatter3D(f1.GRID, f1.density, np.mean(f1.density), axisunit = pc,
             colordim = f1.temperature, 
             colorlabel = 'T [K]',
             xlim=lims, ylim=lims, zlim=lims,
             NRand = 10000, show=True)