o N#iV@s dZddlZddlZddlZddlZddlZddlZddlZgdZdZ dZ dZ dZ dZ d Zd ZejejeZd Zd,d d ZddZddZddZddZddZd-ddZddZddZdd Zd!d"Z d#d$Z!d%d&Z"d'd(Z#d)d*Z$e%d+kre$dSdS).av Phase 2 CFD: Control vane parametric sweep. Creates a 3D duct with a deflected vane at various angles, runs simpleFoam, and extracts forces. Duct: cylinder R=20mm, L=137mm (approximated as O-grid in blockMesh) Vane: span=38mm (diameter, so half-span=19mm from center), chord=18mm, thickness=1.5mm Vane center at Z=36mm from inlet (midpoint of Z=34..38mm in real geometry) N)r g{Gz?gK7?g~jt?g;On?g~jtX?g;On?g>@z0source /usr/lib/openfoam/openfoam2406/etc/bashrccCsFtd|d|}|r|d|d7}tj|dd|duddd }|S) z$Run an OpenFOAM command in case_dir.z && cd z && z > z 2>&1Tz /bin/bashNiX)shell executablecapture_outputtexttimeout) OF_SOURCE subprocessrun)cmdcase_dirlog_namefull_cmdresultr1/home/zenith/canfly/cfd/phase2_vanes/run_sweep.pyrun_of"s  rc Cst|dO}|d|tdt||D]3\}}}}|D] }|td|q |||fD]}|D] }|td|q5q1|tddqWddS1sWwYdS)zBWrite binary STL. triangles = list of (normal, v1, v2, v3) tuples.wbsPz.rotate_ycsg|]}|qSrr).0c)r1rr \sz%generate_vane_stl..) )rr()rr()r)r7r8)r6r(r8)r6r8r9)rrr5)rr7r)rr6r9)rr9r7)r5rr8)r5r8r(cSs^|d|d|d|d|d|d|d|d|d|d|d|dfS)Nr5r(rrabrrrcrossns^z generate_vane_stl..crosscSs.|d|d|d|d|d|dfS)Nrr5r(rr:rrrsubqs.zgenerate_vane_stl..subcs:ttdd|DdkrdStfdd|DS)Ncss|]}||VqdSNrr2r*rrr uz7generate_vane_stl..normalize..g-q=)rrr5c3s|]}|VqdSr?rr@magrrrAxrB)mathsqrtsumtuple)r&rrCr normalizetsz$generate_vane_stl..normalizeN) rEradianscossin VANE_CHORD VANE_SPAN VANE_THICKappendr') angle_degr angle_rad half_chord half_span half_thick corners_localcornersfacesr=r>rIri0i1i2v0r"r#nr)r/r1r0rgenerate_vane_stl:s6          r^cCsTtd}td}dgd| d| d|d| d|d|d| d|d | d| d |d |d| d |d |d|d |d | d|d |d}tjtj|dddttj|ddd}||WddS1swYdS)zWrite blockMeshDict for a 3D rectangular duct approximating the cylinder. Using a simple box: X=[-R,R], Y=[-R,R], Z=[0,L] This is a square duct approximation - good enough for force trends. izFoamFile { version 2.0; format ascii; class dictionary; object blockMeshDict; } scale 0.001; vertices ( (z z 0) // 0 ( z 0) // 1 ( z z 0) // 2 (z 0) // 3 ( z) // 4 ( z) // 5 ( z) // 6 (a) // 7 ); blocks ( hex (0 1 2 3 4 5 6 7) (20 20 70) simpleGrading (1 1 1) ); boundary ( inlet { type patch; faces ( (0 1 2 3) ); } outlet { type patch; faces ( (4 5 6 7) ); } ductWall { type wall; faces ( (0 1 5 4) (1 2 6 5) (2 3 7 6) (3 0 4 7) ); } ); systemTexist_ok blockMeshDictwN)DUCT_RDUCT_Ljoinosmakedirspathrr)rRLcontentr rrrwrite_blockmesh_dicts^      6 "rocCsVtj|}d}ttj|ddd}||WddS1s$wYdS)z=Write snappyHexMeshDict to refine around vane and snap to it.aFoamFile { version 2.0; format ascii; class dictionary; object snappyHexMeshDict; } castellatedMesh true; snap true; addLayers false; geometry { vane.stl { type triSurfaceMesh; name vane; } } castellatedMeshControls { maxLocalCells 50000; maxGlobalCells 200000; minRefinementCells 10; maxLoadUnbalance 0.10; nCellsBetweenLevels 3; resolveFeatureAngle 30; features ( ); refinementSurfaces { vane { level (2 2); patchInfo { type wall; } } } refinementRegions { } locationInMesh (0 0 0.001); // point inside the duct (not inside vane) allowFreeStandingZoneFaces true; } snapControls { nSmoothPatch 3; tolerance 2.0; nSolveIter 100; nRelaxIter 5; nFeatureSnapIter 10; implicitFeatureSnap false; explicitFeatureSnap true; multiRegionFeatureSnap false; } addLayersControls { relativeSizes true; layers {} expansionRatio 1.0; finalLayerThickness 0.3; minThickness 0.1; nGrow 0; featureAngle 60; nRelaxIter 3; nSmoothSurfaceNormals 1; nSmoothNormals 3; nSmoothThickness 10; maxFaceThicknessRatio 0.5; maxThicknessToMedialRatio 0.3; minMedialAxisAngle 90; nBufferCellsNoExtrude 0; nLayerIter 50; } meshQualityControls { maxNonOrtho 65; maxBoundarySkewness 20; maxInternalSkewness 4; maxConcave 80; minVol 1e-13; minTetQuality -1e30; minArea -1; minTwist 0.02; minDeterminant 0.001; minFaceWeight 0.02; minVolRatio 0.01; minTriangleTwist -1; nSmoothScale 4; errorReduction 0.75; relaxed { maxNonOrtho 75; } } debug 0; mergeTolerance 1e-6; rasnappyHexMeshDictreN)rirkbasenamerrhr)rstl_filestl_namernr rrrwrite_snappy_dicts o "rtcCsJd}ttj|ddd}||WddS1swYdS)z*Write decomposeParDict (single processor).zFoamFile { version 2.0; format ascii; class dictionary; object decomposeParDict; } numberOfSubdomains 1; method simple; simpleCoeffs { n (1 1 1); delta 0.001; } radecomposeParDictreNrrirkrhrrrnr rrrwrite_decompose_dict<s "rxcCsXd|dtd}ttj|ddd}||WddS1s%wYdS)z.)keyz/ WARNING: No time directories in forces outputz force.datz moment.datz forces.datz# WARNING: No force data file foundz z/: r)FxFyFzMxMyMzrcS*g|]}|r|ds|qS#strip startswithr2lrrrr45*z[-+]?\d*\.?\d+(?:[eE][-+]?\d+)?r7r5rr(rr6rz% WARNING: Error parsing force file: cSrrrrrrrr4Frrrrz& WARNING: Error parsing moment file: )rirkrhrrsortedlistdirfloatr readlinesrefindallr Exception)r forces_dirtimes latest_time force_file moment_filettdforcesr lines last_linernumserrrextract_forcess~         rc CsTg}tD]}tjtd|d}td|d|t||t||}|rzt|}|red|i|}| |td|ddd |d dd |d dtd |ddd|ddd|ddqtd|| |dddddddqtd|| |dddddddqtjtd}t |ddd}t j |gdd}| ||Wdn1swYtddtd|tdtddd d!dd"d!d d"d!d d"d!dd"d!dd"d!dd"td#|D]/} t| dd d!| dd$d!| d d$d!| d d$d!| dd$d!| dd$d!| dd$ qdS)%Nangle_02dz >>> Setting up case for angle=z deg at anglez Forces: Fx=rz.6fz Fy=rz Fz=rz Moments: Mx=rz My=rz Mz=rz% Could not extract forces for angle r)rrrrrrrz Simulation failed for angle zvane_forces.csvrer_)newline) fieldnamesrrzResults saved to Anglez>6r`z>12zT------------------------------------------------------------------------------------z>12.6f)ANGLESrirkrhBASE_DIRrrrrrPrcsv DictWriter writeheader writerows) resultsrrsuccessrrowcsv_filer writerrrrrmainVsR    ,.      @2 r__main__r?)ry)&__doc__risysrrrErrrrfrgrNrMrOr)rrkdirnameabspath__file__rr rr'r^rortrxr|rrrrrrrr__name__rrrrsJ   ICv 06;B% F-